copilot-proxy-web 1.0.0

package/bin/wss-client.js

@@ -0,0 +1,227 @@
+#!/usr/bin/env node
+
+const WebSocket = require("ws");
+
+function usage() {
+  process.stdout.write(
+    [
+      "Usage:",
+      "  copilot-proxy-ws --url wss://proxy.example.com/ws --sessionId default",
+      "",
+      "Options:",
+      "  --url URL              WebSocket endpoint (wss://.../ws)",
+      "  --sessionId ID         Session id to attach",
+      "  --token TOKEN          AUTH_TOKEN (or set AUTH_TOKEN env)",
+      "  --origin ORIGIN        Set Origin header (for servers that require it)",
+      "  --header KEY:VALUE     Add custom request header (repeatable)",
+      "  --cf-access-id ID      Cloudflare Access Client ID",
+      "  --cf-access-secret ID  Cloudflare Access Client Secret",
+      "  --text TEXT            Send a text command (adds CR on server)",
+      "  --keys KEYS            Send raw keys",
+      "  --resize COLSxROWS     Send resize event once",
+      "  --no-stdin             Do not forward stdin",
+    ].join("\n") + "\n"
+  );
+}
+
+function parseArgs(argv) {
+  const out = {
+    url: "",
+    sessionId: "",
+    token: "",
+    text: "",
+    keys: "",
+    resize: "",
+    noStdin: false,
+    headers: [],
+    cfAccessId: "",
+    cfAccessSecret: "",
+  };
+  for (let i = 0; i < argv.length; i += 1) {
+    const arg = argv[i];
+    if (arg === "--url" && i + 1 < argv.length) {
+      out.url = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--sessionId" && i + 1 < argv.length) {
+      out.sessionId = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--token" && i + 1 < argv.length) {
+      out.token = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--origin" && i + 1 < argv.length) {
+      out.origin = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--header" && i + 1 < argv.length) {
+      out.headers.push(argv[i + 1]);
+      i += 1;
+      continue;
+    }
+    if (arg === "--cf-access-id" && i + 1 < argv.length) {
+      out.cfAccessId = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--cf-access-secret" && i + 1 < argv.length) {
+      out.cfAccessSecret = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--text" && i + 1 < argv.length) {
+      out.text = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--keys" && i + 1 < argv.length) {
+      out.keys = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--resize" && i + 1 < argv.length) {
+      out.resize = argv[i + 1];
+      i += 1;
+      continue;
+    }
+    if (arg === "--no-stdin") {
+      out.noStdin = true;
+      continue;
+    }
+    if (arg === "--help" || arg === "-h") {
+      usage();
+      process.exit(0);
+    }
+  }
+  return out;
+}
+
+function buildUrl(baseUrl, sessionId) {
+  if (!sessionId) return baseUrl;
+  const hasQuery = baseUrl.includes("?");
+  const joiner = hasQuery ? "&" : "?";
+  return `${baseUrl}${joiner}sessionId=${encodeURIComponent(sessionId)}`;
+}
+
+function encodeAuthProtocol(token) {
+  if (!token) return null;
+  return `auth.${Buffer.from(token, "utf8").toString("base64url")}`;
+}
+
+function parseResize(value) {
+  if (!value) return null;
+  const match = value.match(/^(\d+)x(\d+)$/);
+  if (!match) return null;
+  const cols = Number(match[1]);
+  const rows = Number(match[2]);
+  if (!Number.isFinite(cols) || !Number.isFinite(rows)) return null;
+  return { cols, rows };
+}
+
+function sendBinary(ws, type, text) {
+  const payload = Buffer.from(text, "utf8");
+  const buf = Buffer.alloc(payload.length + 1);
+  buf[0] = type;
+  payload.copy(buf, 1);
+  ws.send(buf);
+}
+
+function parseHeaders(entries, origin) {
+  const customHeaders = {};
+  for (const entry of entries || []) {
+    const idx = entry.indexOf(":");
+    if (idx > 0) {
+      const key = entry.slice(0, idx).trim();
+      const value = entry.slice(idx + 1).trim();
+      if (key) customHeaders[key] = value;
+    }
+  }
+  if (!origin && Object.keys(customHeaders).length === 0) return undefined;
+  return { ...(origin ? { Origin: origin } : {}), ...customHeaders };
+}
+
+if (require.main !== module) {
+  module.exports = {
+    parseArgs,
+    parseHeaders,
+  };
+} else {
+  const args = parseArgs(process.argv.slice(2));
+  const token = args.token || process.env.AUTH_TOKEN || "";
+  if (!args.url) {
+    usage();
+    process.exit(1);
+  }
+
+  const url = buildUrl(args.url, args.sessionId);
+  const protocols = [];
+  const authProto = encodeAuthProtocol(token);
+  if (authProto) protocols.push(authProto);
+
+  const headers = parseHeaders(args.headers, args.origin) || {};
+  if (args.cfAccessId) headers["CF-Access-Client-Id"] = args.cfAccessId;
+  if (args.cfAccessSecret) headers["CF-Access-Client-Secret"] = args.cfAccessSecret;
+
+  const ws = new WebSocket(url, protocols.length > 0 ? protocols : undefined, {
+    headers: Object.keys(headers).length > 0 ? headers : undefined,
+  });
+
+  ws.on("open", () => {
+  if (args.text) {
+    ws.send(JSON.stringify({ type: "text", data: args.text }));
+  }
+  if (args.keys) {
+    ws.send(JSON.stringify({ type: "keys", data: args.keys }));
+  }
+  const resize = parseResize(args.resize);
+  if (resize) {
+    ws.send(JSON.stringify({ type: "resize", cols: resize.cols, rows: resize.rows }));
+  }
+  if (!args.noStdin) {
+    if (process.stdin.isTTY) {
+      process.stdin.setRawMode(true);
+    }
+    process.stdin.resume();
+    process.stdin.on("data", (data) => {
+      if (data && data.includes && data.includes("\u0003")) {
+        ws.close();
+        process.exit(0);
+      }
+      sendBinary(ws, 1, data.toString());
+    });
+  }
+  if (process.stdout.isTTY) {
+    process.stdout.on("resize", () => {
+      ws.send(
+        JSON.stringify({
+          type: "resize",
+          cols: process.stdout.columns || 80,
+          rows: process.stdout.rows || 24,
+        })
+      );
+    });
+  }
+  });
+
+  ws.on("message", (data, isBinary) => {
+    if (isBinary) {
+      process.stdout.write(Buffer.from(data));
+    } else {
+      process.stdout.write(data.toString());
+    }
+  });
+
+  ws.on("close", () => {
+    process.exit(0);
+  });
+
+  ws.on("error", (err) => {
+    process.stderr.write(`[copilot-proxy-ws] ${err.message}\n`);
+    process.exit(1);
+  });
+}