@agentsky/sky 2.10.0 → 2.12.0

package/dist/harness-connect-DUC7GIBL.js

@@ -0,0 +1,102 @@
+import "./chunk-MLKGABMK.js";
+
+// src/harness-connect.ts
+import { spawn } from "child_process";
+import { createInterface } from "readline";
+function agentCommandFrom(argv) {
+  const separator = argv.indexOf("--");
+  return separator >= 0 ? argv.slice(separator + 1) : [];
+}
+function harnessRelayUrl(apiBase, session) {
+  const base = apiBase.replace(/\/$/, "").replace(/^http(s?):\/\//, (_, secure) => `ws${secure}://`);
+  return `${base}/api/v1/sessions/${encodeURIComponent(session)}/harness`;
+}
+async function connectHarness(options) {
+  const [program, ...args] = options.command;
+  if (!program) throw new Error("give the ACP agent command after --, e.g. sky harness connect sess_\u2026 -- npx my-agent --acp");
+  if (!options.token.startsWith("hct_")) throw new Error("the connect token must start with hct_");
+  const url = options.url ?? (options.apiBase && options.session ? harnessRelayUrl(options.apiBase, options.session) : null);
+  if (!url) throw new Error("give --session (or set AGENTSKY_HARNESS_RELAY_URL)");
+  const log = options.log ?? ((line) => console.error(line));
+  const { default: WebSocket } = await import("ws");
+  return new Promise((resolve, reject) => {
+    const socket = new WebSocket(url, { headers: { authorization: `Bearer ${options.token}` } });
+    let child = null;
+    let settled = false;
+    const finish = (code) => {
+      if (settled) return;
+      settled = true;
+      if (child && child.exitCode === null) child.kill("SIGTERM");
+      if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) socket.close();
+      resolve(code);
+    };
+    options.signal?.addEventListener("abort", () => finish(0), { once: true });
+    socket.on("unexpected-response", (_request, response) => {
+      settled = true;
+      reject(new RelayRefused(response.statusCode ?? 0, `the relay refused the connection: ${response.statusCode} ${response.statusMessage ?? ""}`.trim()));
+    });
+    socket.on("error", (error) => {
+      if (!settled) {
+        settled = true;
+        reject(error);
+      }
+    });
+    socket.on("open", () => {
+      log(`[sky harness] connected; running ${program}`);
+      child = spawn(program, args, { stdio: ["pipe", "pipe", "inherit"] });
+      child.on("error", (error) => {
+        log(`[sky harness] agent failed to start: ${error.message}`);
+        finish(1);
+      });
+      child.on("exit", (code) => {
+        log(`[sky harness] agent exited (${code ?? "signal"})`);
+        finish(code ?? 1);
+      });
+      createInterface({ input: child.stdout }).on("line", (line) => {
+        if (line.trim() && socket.readyState === WebSocket.OPEN) socket.send(line);
+      });
+    });
+    socket.on("message", (data) => {
+      child?.stdin?.write(`${data.toString()}
+`);
+    });
+    socket.on("close", (code) => {
+      log(`[sky harness] relay closed (${code})`);
+      finish(0);
+    });
+  });
+}
+var RelayRefused = class extends Error {
+  constructor(status, message) {
+    super(message);
+    this.status = status;
+    this.name = "RelayRefused";
+  }
+  status;
+};
+async function serveHarness(options, wait = (ms) => new Promise((r) => setTimeout(r, ms))) {
+  const log = options.log ?? ((line) => console.error(line));
+  let backoff = 1e3;
+  while (!options.signal?.aborted) {
+    const started = Date.now();
+    try {
+      await connectHarness(options);
+    } catch (error) {
+      if (error instanceof RelayRefused && [401, 403, 404].includes(error.status)) throw error;
+      log(`[sky harness] ${error instanceof Error ? error.message : String(error)}`);
+    }
+    if (options.signal?.aborted) break;
+    if (Date.now() - started > 6e4) backoff = 1e3;
+    log(`[sky harness] reconnecting in ${Math.round(backoff / 1e3)} s`);
+    await wait(backoff);
+    backoff = Math.min(backoff * 2, 3e4);
+  }
+  return 0;
+}
+export {
+  RelayRefused,
+  agentCommandFrom,
+  connectHarness,
+  harnessRelayUrl,
+  serveHarness
+};