@agentsky/sky 2.10.0 → 2.12.0

package/dist/worker-6Q4K4HZW.js

@@ -0,0 +1,190 @@
+import "./chunk-MLKGABMK.js";
+
+// src/worker.ts
+import { mkdirSync, writeFileSync, rmSync } from "fs";
+import { tmpdir } from "os";
+import { join } from "path";
+import { randomBytes } from "crypto";
+var RealDockerRunner = class {
+  async run({ image, envFile, port, signal }) {
+    const { spawn } = await import("child_process");
+    return new Promise((resolve, reject) => {
+      const child = spawn(
+        "docker",
+        [
+          "run",
+          "--rm",
+          "--pull=always",
+          `--env-file=${envFile}`,
+          `-p`,
+          `127.0.0.1:${port}:8080`,
+          image
+        ],
+        { stdio: "inherit" }
+      );
+      signal.addEventListener("abort", () => {
+        child.kill("SIGTERM");
+      });
+      child.on("error", reject);
+      child.on("exit", (code) => {
+        if (code === 0 || signal.aborted) resolve();
+        else reject(new Error(`docker exited with code ${code}`));
+      });
+    });
+  }
+};
+var POD_PORT = 18080;
+var HEARTBEAT_INTERVAL_MS = 15e3;
+async function startWorker(opts) {
+  const docker = opts.docker ?? new RealDockerRunner();
+  const signal = opts.signal ?? new AbortController().signal;
+  const wsUrl = opts.apiBase.replace(/^https?:\/\//, (m) => m === "https://" ? "wss://" : "ws://").replace(/\/$/, "");
+  const connectUrl = `${wsUrl}/api/v1/worker-pools/${opts.poolId}/connect`;
+  return new Promise((resolve, reject) => {
+    import("ws").then(({ default: WS }) => {
+      const ws = new WS(connectUrl, {
+        headers: { authorization: `Bearer ${opts.token}` }
+      });
+      let currentContainer = null;
+      const heartbeatTimer = setInterval(() => {
+        if (ws.readyState === WS.OPEN) {
+          ws.send(JSON.stringify({ type: "heartbeat" }));
+        }
+      }, HEARTBEAT_INTERVAL_MS);
+      signal.addEventListener("abort", () => {
+        clearInterval(heartbeatTimer);
+        currentContainer?.abort();
+        ws.close();
+        resolve();
+      });
+      ws.on("open", () => {
+        ws.send(JSON.stringify({
+          type: "hello",
+          label: opts.label,
+          imageDigest: "unknown",
+          capacity: 1
+        }));
+        console.log(`[sky worker] connected to pool ${opts.poolId}`);
+      });
+      ws.on("message", async (raw) => {
+        let msg;
+        try {
+          msg = JSON.parse(raw.toString());
+        } catch {
+          return;
+        }
+        if (msg.type === "start") {
+          await handleStart(msg, ws, WS, docker, currentContainer);
+        } else if (msg.type === "stop") {
+          currentContainer?.abort();
+          currentContainer = null;
+        } else if (msg.type === "http") {
+          await handleHttp(msg, ws);
+        } else if (msg.type === "stream_open") {
+          void handleStreamOpen(msg, ws);
+        }
+      });
+      ws.on("close", () => {
+        clearInterval(heartbeatTimer);
+        currentContainer?.abort();
+        resolve();
+      });
+      ws.on("error", (err) => {
+        clearInterval(heartbeatTimer);
+        currentContainer?.abort();
+        reject(err);
+      });
+    }).catch(reject);
+  });
+  async function handleStart(msg, ws, WS, docker2, existingContainer) {
+    if (existingContainer) existingContainer.abort();
+    const dir = join(tmpdir(), `sky-worker-${randomBytes(6).toString("hex")}`);
+    mkdirSync(dir, { mode: 448 });
+    const envFile = join(dir, "pod.env");
+    const envLines = Object.entries(msg.env).map(([k, v]) => `${k}=${v.replace(/\n/g, "\\n")}`).join("\n");
+    writeFileSync(envFile, envLines, { mode: 384 });
+    const containerAbort = new AbortController();
+    try {
+      void docker2.run({
+        image: msg.image,
+        envFile,
+        port: POD_PORT,
+        signal: containerAbort.signal
+      }).finally(() => {
+        try {
+          rmSync(dir, { recursive: true });
+        } catch {
+        }
+      });
+    } catch (err) {
+      try {
+        rmSync(dir, { recursive: true });
+      } catch {
+      }
+      console.error("[sky worker] docker run failed:", err);
+    }
+  }
+  async function handleHttp(msg, ws) {
+    const url = `http://127.0.0.1:${POD_PORT}${msg.path}`;
+    try {
+      const response = await fetch(url, {
+        method: msg.method,
+        headers: msg.headers,
+        body: msg.body || void 0,
+        signal: AbortSignal.timeout(3e4)
+      });
+      const body = await response.text();
+      const headers = {};
+      response.headers.forEach((v, k) => {
+        headers[k] = v;
+      });
+      const resp = {
+        type: "http_response",
+        id: msg.id,
+        status: response.status,
+        headers,
+        body
+      };
+      ws.send(JSON.stringify(resp));
+    } catch (err) {
+      ws.send(JSON.stringify({
+        type: "http_response",
+        id: msg.id,
+        status: 502,
+        headers: {},
+        body: String(err)
+      }));
+    }
+  }
+  async function handleStreamOpen(msg, ws) {
+    const url = `http://127.0.0.1:${POD_PORT}${msg.path}`;
+    try {
+      const response = await fetch(url, {
+        headers: { accept: "text/event-stream" },
+        signal: AbortSignal.timeout(3e5)
+      });
+      const reader = response.body?.getReader();
+      if (!reader) {
+        ws.send(JSON.stringify({ type: "stream_close", id: msg.id }));
+        return;
+      }
+      const decoder = new TextDecoder();
+      for (; ; ) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        ws.send(JSON.stringify({
+          type: "stream_event",
+          id: msg.id,
+          data: decoder.decode(value, { stream: true })
+        }));
+      }
+      ws.send(JSON.stringify({ type: "stream_close", id: msg.id }));
+    } catch {
+      ws.send(JSON.stringify({ type: "stream_close", id: msg.id }));
+    }
+  }
+}
+export {
+  RealDockerRunner,
+  startWorker
+};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@agentsky/sky",
-  "version": "2.10.0",
-  "description": "sky — the AgentSky developer CLI: launch and talk to long-lived agents from your terminal.",
+  "version": "2.12.0",
+  "description": "sky \u2014 the AgentSky developer CLI: launch and talk to long-lived agents from your terminal.",
   "license": "MIT",
   "type": "module",
   "bin": {
@@ -22,7 +22,8 @@
   "dependencies": {
     "commander": "^14.0.0",
     "smol-toml": "^1.4.2",
-    "zod": "4.5.4"
+    "zod": "4.5.4",
+    "ws": "^8.18.0"
   },
   "devDependencies": {
     "@agent/chat": "*",