@ricsam/r5dctl 0.0.21 → 0.0.23

package/dist/cjs/shell.cjs

@@ -0,0 +1,245 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var shell_exports = {};
+__export(shell_exports, {
+  createShellWebSocketUrl: () => createShellWebSocketUrl,
+  runR5dctlShell: () => runR5dctlShell
+});
+module.exports = __toCommonJS(shell_exports);
+var import_ws = __toESM(require("ws"), 1);
+function clampTerminalSize(value, fallback) {
+  if (!Number.isFinite(value) || (value ?? 0) <= 0) {
+    return fallback;
+  }
+  return Math.max(1, Math.min(Math.floor(value), 500));
+}
+function createShellWebSocketUrl(baseUrl) {
+  const url = new URL("/ws", baseUrl);
+  if (url.protocol === "https:") {
+    url.protocol = "wss:";
+  } else if (url.protocol === "http:") {
+    url.protocol = "ws:";
+  } else {
+    throw new Error(`Unsupported r5d base URL protocol: ${url.protocol}`);
+  }
+  return url.toString();
+}
+function decodeMessage(data) {
+  if (typeof data === "string") {
+    return data;
+  }
+  if (Buffer.isBuffer(data)) {
+    return data.toString("utf8");
+  }
+  if (data instanceof ArrayBuffer) {
+    return Buffer.from(data).toString("utf8");
+  }
+  if (ArrayBuffer.isView(data)) {
+    return Buffer.from(data.buffer, data.byteOffset, data.byteLength).toString("utf8");
+  }
+  if (Array.isArray(data)) {
+    return Buffer.concat(data).toString("utf8");
+  }
+  return String(data);
+}
+async function runR5dctlShell(options) {
+  const stdin = options.stdin ?? process.stdin;
+  const stdout = options.stdout ?? process.stdout;
+  const signalTarget = options.signalTarget ?? process;
+  const interactive = options.command === void 0;
+  if (interactive && !stdin.isTTY) {
+    throw new Error("Interactive shell requires a TTY. Use `shell -c <command>` for non-interactive use.");
+  }
+  if (!options.credential) {
+    throw new Error("Authentication required. Run `r5dctl auth login` or provide a token or API key.");
+  }
+  const createWebSocket = options.createWebSocket ?? ((url, clientOptions) => new import_ws.default(url, clientOptions));
+  const socket = createWebSocket(createShellWebSocketUrl(options.baseUrl), {
+    headers: {
+      Authorization: `Bearer ${options.credential}`
+    }
+  });
+  return await new Promise((resolve, reject) => {
+    let ptyId;
+    let openRequested = false;
+    let settled = false;
+    let inputAttached = false;
+    const priorRawMode = stdin.isRaw ?? false;
+    const send = (message) => {
+      if (socket.readyState !== import_ws.default.OPEN) {
+        return;
+      }
+      socket.send(JSON.stringify(message));
+    };
+    const sendResize = () => {
+      if (!ptyId) {
+        return;
+      }
+      send({
+        type: "shell_pty_resize",
+        ptyId,
+        cols: clampTerminalSize(stdout.columns, 80),
+        rows: clampTerminalSize(stdout.rows, 24)
+      });
+    };
+    const onInput = (chunk) => {
+      if (ptyId) {
+        send({ type: "shell_pty_input", ptyId, data: Buffer.isBuffer(chunk) ? chunk.toString("utf8") : chunk });
+      }
+    };
+    const onInputEnd = () => {
+      if (!interactive && ptyId) {
+        send({ type: "shell_pty_input", ptyId, data: "" });
+      }
+    };
+    const onSigint = () => {
+      if (ptyId) {
+        send({ type: "shell_pty_input", ptyId, data: "" });
+      }
+    };
+    const onSigterm = () => {
+      if (ptyId) {
+        send({ type: "shell_pty_close", ptyId });
+        ptyId = void 0;
+      }
+      finish({ exitCode: 143 });
+    };
+    const cleanup = () => {
+      socket.off("message", onMessage);
+      socket.off("error", onSocketError);
+      socket.off("close", onSocketClose);
+      if (inputAttached) {
+        stdin.off("data", onInput);
+        stdin.off("end", onInputEnd);
+      }
+      signalTarget.off("SIGWINCH", sendResize);
+      signalTarget.off("SIGINT", onSigint);
+      signalTarget.off("SIGTERM", onSigterm);
+      if (stdin.isTTY && stdin.setRawMode) {
+        stdin.setRawMode(priorRawMode);
+      }
+    };
+    const finish = (result) => {
+      if (settled) {
+        return;
+      }
+      settled = true;
+      cleanup();
+      if (socket.readyState === import_ws.default.OPEN || socket.readyState === import_ws.default.CONNECTING) {
+        socket.close(1e3, "r5dctl shell finished");
+      }
+      if ("error" in result) {
+        reject(result.error);
+      } else {
+        resolve(result.exitCode);
+      }
+    };
+    function attachInput() {
+      if (inputAttached) {
+        return;
+      }
+      inputAttached = true;
+      if (stdin.isTTY && stdin.setRawMode) {
+        stdin.setRawMode(true);
+      }
+      stdin.on("data", onInput);
+      stdin.on("end", onInputEnd);
+      signalTarget.on("SIGWINCH", sendResize);
+      signalTarget.on("SIGTERM", onSigterm);
+      if (!stdin.isTTY) {
+        signalTarget.on("SIGINT", onSigint);
+      }
+    }
+    function onMessage(data) {
+      let message;
+      try {
+        const parsed = JSON.parse(decodeMessage(data));
+        if (!parsed || typeof parsed !== "object" || !("type" in parsed)) {
+          throw new Error("Invalid message");
+        }
+        message = parsed;
+      } catch {
+        finish({ error: new Error("Received an invalid response from r5d.dev") });
+        return;
+      }
+      if (message.type === "connected") {
+        if (!openRequested) {
+          openRequested = true;
+          send({
+            type: "shell_pty_open",
+            projectId: options.projectId,
+            branchName: options.branchName,
+            cols: clampTerminalSize(stdout.columns, 80),
+            rows: clampTerminalSize(stdout.rows, 24),
+            ...options.command === void 0 ? {} : { command: options.command }
+          });
+        }
+        return;
+      }
+      if (message.type === "shell_pty_opened") {
+        ptyId = message.ptyId;
+        attachInput();
+        return;
+      }
+      if (message.type === "shell_pty_output") {
+        if (message.ptyId === ptyId) {
+          stdout.write(message.data);
+        }
+        return;
+      }
+      if (message.type === "shell_pty_exit") {
+        if (message.ptyId === ptyId) {
+          ptyId = void 0;
+          finish({ exitCode: message.exitCode });
+        }
+        return;
+      }
+      if (message.type === "shell_pty_error" && (!message.ptyId || message.ptyId === ptyId)) {
+        finish({ error: new Error(message.message) });
+      }
+    }
+    function onSocketError(error) {
+      finish({ error: new Error(`r5d.dev shell connection failed: ${error.message}`) });
+    }
+    function onSocketClose(code, reason) {
+      if (!settled) {
+        const suffix = reason.length > 0 ? `: ${reason.toString("utf8")}` : "";
+        finish({ error: new Error(`r5d.dev shell connection closed (${code})${suffix}`) });
+      }
+    }
+    socket.on("message", onMessage);
+    socket.on("error", onSocketError);
+    socket.on("close", onSocketClose);
+  });
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createShellWebSocketUrl,
+  runR5dctlShell
+});