@lesscap/baton-cli 0.9.4 → 0.9.5

package/dist/baton.mjs

@@ -2286,7 +2286,7 @@ var require_websocket = __commonJS({
     var net = __require("net");
     var tls = __require("tls");
     var { randomBytes, createHash: createHash2 } = __require("crypto");
-    var { Duplex, Readable: Readable2 } = __require("stream");
+    var { Duplex, Readable: Readable4 } = __require("stream");
     var { URL: URL2 } = __require("url");
     var PerMessageDeflate2 = require_permessage_deflate();
     var Receiver2 = require_receiver();
@@ -5437,7 +5437,9 @@ var rawRequest = async (url, init2) => {
   const res = await fetch(url, {
     method: init2.method,
     headers: { ...authHeaders, ...init2.headers ?? {} },
-    ...init2.body !== void 0 ? { body: init2.body } : {}
+    ...init2.body !== void 0 ? { body: init2.body } : {},
+    ...init2.body instanceof ReadableStream ? { duplex: "half" } : {},
+    ...init2.signal ? { signal: init2.signal } : {}
   });
   if (!res.ok)
     throw new HttpError(res.status, `${init2.method} ${url} \u2192 ${res.status}: ${await res.text()}`);
@@ -5493,9 +5495,19 @@ var primeLogin = (baseUrl) => {
 };
 
 // src/client/files.ts
-var filesClient = (baseUrl) => ({
-  report: (result) => request(`${baseUrl}/workers/me/files/results`, { method: "PUT", body: result })
-});
+var filesClient = (baseUrl) => {
+  const downloadUrl = (requestId) => `${baseUrl}/workers/me/files/downloads/${encodeURIComponent(requestId)}`;
+  return {
+    report: (result) => request(`${baseUrl}/workers/me/files/results`, { method: "PUT", body: result }),
+    uploadDownload: (requestId, size, body, signal) => rawRequest(downloadUrl(requestId), {
+      method: "PUT",
+      body,
+      headers: { "content-type": "application/octet-stream", "content-length": String(size) },
+      ...signal ? { signal } : {}
+    }),
+    rejectDownload: (requestId, failure) => request(downloadUrl(requestId), { method: "PUT", body: failure })
+  };
+};
 
 // src/client/loops.ts
 var loopsClient = (baseUrl) => {
@@ -6776,6 +6788,8 @@ var mediaTypeIn = (types3) => (path) => {
 };
 var imageMediaType = mediaTypeIn(IMAGE_MEDIA_TYPES);
 var rawMediaType = mediaTypeIn(RAW_MEDIA_TYPES);
+var FILE_DOWNLOAD_MAX_BYTES = 100 * 1024 * 1024;
+var FILE_DOWNLOAD_TIMEOUT_MS = 3e5;
 var normalizeFilePath = (value, allowRoot = false) => {
   if (value === "") return allowRoot ? "" : null;
   if (value.includes("\0") || value.includes("\\") || value.startsWith("/")) return null;
@@ -10131,6 +10145,10 @@ var discoverBranches = async (baseDir) => {
   }).sort((a, b) => rank(a) - rank(b) || (a.name < b.name ? -1 : a.name > b.name ? 1 : 0));
 };
 
+// src/worker/files/commit.ts
+import { spawn as spawn2 } from "node:child_process";
+import { Readable as Readable2 } from "node:stream";
+
 // src/worker/files/common.ts
 var FileError = class extends Error {
   constructor(code, message) {
@@ -10140,6 +10158,10 @@ var FileError = class extends Error {
   code;
 };
 var notFound = (path) => new FileError("file_not_found", `not found: ${path}`);
+var checkDownloadSize = (path, size) => {
+  if (size > FILE_DOWNLOAD_MAX_BYTES)
+    throw new FileError("file_too_large", `too large to download (${size} bytes): ${path}`);
+};
 var checkedPath = (path, allowRoot) => {
   const rel = normalizeFilePath(path, allowRoot);
   if (rel === null) throw new FileError("invalid_request", `invalid path: ${path}`);
@@ -10282,11 +10304,38 @@ var readRawCommitFile = async (repo, commit, path) => {
   const bytes = await runGit(repo, ["cat-file", "blob", entry.oid], FILE_RAW_MAX_BYTES + 1);
   return rawPayload(rel, mediaType, bytes);
 };
+var openCommitDownload = async (repo, commit, path, signal) => {
+  const rel = checkedPath(path, false);
+  const entry = await regularEntry(repo, commit, rel);
+  checkDownloadSize(rel, entry.size);
+  const child = spawn2("git", ["--literal-pathspecs", "-C", repo, "cat-file", "blob", entry.oid], {
+    stdio: ["ignore", "pipe", "ignore"],
+    signal,
+    env: { ...process.env, LC_ALL: "C", GIT_OPTIONAL_LOCKS: "0", GIT_TERMINAL_PROMPT: "0" }
+  });
+  const exited = new Promise((resolve4) => {
+    child.once("error", resolve4);
+    child.once(
+      "close",
+      (code, killed) => resolve4(code === 0 ? null : new Error(`git cat-file exited with ${code ?? killed}`))
+    );
+  });
+  const body = Readable2.toWeb(child.stdout).pipeThrough(
+    new TransformStream({
+      async flush() {
+        const failure = await exited;
+        if (failure) throw failure;
+      }
+    })
+  );
+  return { size: entry.size, body, close: () => child.kill() };
+};
 
 // src/worker/files/disk.ts
 import { constants as constants2 } from "node:fs";
 import { open as open2, readdir, realpath as realpath3, stat as stat3 } from "node:fs/promises";
 import { isAbsolute as isAbsolute4, join as join15, relative, sep } from "node:path";
+import { Readable as Readable3 } from "node:stream";
 var isMissing = (error) => ["ENOENT", "ENOTDIR", "ELOOP"].includes(error.code ?? "");
 var childPath = (parent, name) => parent ? `${parent}/${name}` : name;
 var inside = (root, real) => {
@@ -10407,6 +10456,72 @@ var readRawFile = async (root, path) => {
   if ("oversize" in read) throw new FileError("unsupported", `too large to serve raw: ${file.path}`);
   return rawPayload(file.path, mediaType, read.bytes);
 };
+var openDiskDownload = async (root, path, signal) => {
+  const file = await resolveInside(root, path, false);
+  const handle = await open2(file.real, constants2.O_RDONLY | constants2.O_NONBLOCK);
+  try {
+    const opened = await handle.stat();
+    if (!opened.isFile()) throw notFound(file.path);
+    checkDownloadSize(file.path, opened.size);
+    const stream = handle.createReadStream({ signal });
+    return {
+      size: opened.size,
+      body: Readable3.toWeb(stream),
+      close: () => stream.destroy()
+    };
+  } catch (error) {
+    await handle.close();
+    throw error;
+  }
+};
+
+// src/worker/files/download.ts
+var exactLength = (size) => {
+  let seen = 0;
+  return new TransformStream({
+    transform(chunk, controller) {
+      seen += chunk.byteLength;
+      if (seen > size) throw new Error(`file grew past its ${size} bytes while downloading`);
+      controller.enqueue(chunk);
+    },
+    flush() {
+      if (seen !== size) throw new Error(`file ended at ${seen} of its ${size} bytes`);
+    }
+  });
+};
+var openSource = async (baseDir, branch, path, signal) => {
+  if (branch.worktree !== null) return openDiskDownload(branch.worktree, path, signal);
+  if (branch.commit === null)
+    throw new FileError("branch_not_found", `branch ${branch.name} is empty`);
+  return openCommitDownload(baseDir, branch.commit, path, signal);
+};
+var openFileDownload = async (baseDir, query3, signal) => {
+  const branches = await discoverBranches(baseDir);
+  const branch = branches.find((candidate) => candidate.name === query3.branch);
+  if (!branch) throw new FileError("branch_not_found", `branch ${query3.branch} does not exist`);
+  const source = await openSource(baseDir, branch, query3.path, signal);
+  return { ...source, body: source.body.pipeThrough(exactLength(source.size)) };
+};
+var failureOf = (error) => {
+  if (error instanceof FileError) return { ok: false, code: error.code, message: error.message };
+  console.error("[files] opening a download failed", error);
+  return { ok: false, code: "unavailable", message: "the worker could not read the file" };
+};
+var sendFileDownload = async (baseDir, command, client, signal) => {
+  const bounded = AbortSignal.any([signal, AbortSignal.timeout(FILE_DOWNLOAD_TIMEOUT_MS)]);
+  let source;
+  try {
+    source = await openFileDownload(baseDir, command.query, bounded);
+  } catch (error) {
+    await client.rejectDownload(command.requestId, failureOf(error));
+    return;
+  }
+  try {
+    await client.uploadDownload(command.requestId, source.size, source.body, bounded);
+  } finally {
+    source.close();
+  }
+};
 
 // src/worker/files/handler.ts
 var DEFAULT_CONCURRENCY = 4;
@@ -10534,7 +10649,7 @@ import { stat as stat4 } from "node:fs/promises";
 import { join as join17 } from "node:path";
 
 // src/worker/service-process.ts
-import { spawn as spawn2 } from "node:child_process";
+import { spawn as spawn3 } from "node:child_process";
 import { closeSync, existsSync as existsSync6, openSync } from "node:fs";
 import { dirname as dirname4, join as join16 } from "node:path";
 import { fileURLToPath as fileURLToPath2 } from "node:url";
@@ -10554,7 +10669,7 @@ var startServiceProcess = (input) => {
   const logFd = openSync(input.logPath, "a");
   let wrapper;
   try {
-    wrapper = spawn2(process.execPath, [binPath(), "service", "child"], {
+    wrapper = spawn3(process.execPath, [binPath(), "service", "child"], {
       detached: false,
       stdio: ["ignore", logFd, logFd, "ipc"],
       env: process.env
@@ -10704,7 +10819,7 @@ var createServiceSupervisor = (deps) => {
 };
 
 // src/worker/session-supervisor.ts
-import { spawn as spawn3 } from "node:child_process";
+import { spawn as spawn4 } from "node:child_process";
 import { randomUUID as randomUUID2 } from "node:crypto";
 import { existsSync as existsSync7 } from "node:fs";
 import { mkdir as mkdir6, rm as rm3, stat as stat5 } from "node:fs/promises";
@@ -11009,7 +11124,7 @@ var createSessionSupervisor = (deps) => {
     if (children.has(sessionId)) return;
     await stopRecord(sessionId);
     const runnerToken = randomUUID2();
-    const child = spawn3(
+    const child = spawn4(
       process.execPath,
       [binPath2(), "session", "run", String(sessionId), "--runner-token", runnerToken],
       {
@@ -11201,7 +11316,7 @@ var createSessionSupervisor = (deps) => {
 
 // src/worker/terminal-manager.ts
 import { statSync as statSync2 } from "node:fs";
-import { spawn as spawn4 } from "node-pty";
+import { spawn as spawn5 } from "node-pty";
 
 // ../../node_modules/.pnpm/ws@8.21.0/node_modules/ws/wrapper.mjs
 var import_stream2 = __toESM(require_stream(), 1);
@@ -11266,7 +11381,7 @@ var createTerminalManager = (deps) => {
       return;
     }
     const command = ptyCommand(cfg.agentKind, agentSessionId);
-    const term = spawn4(command.file, command.args, {
+    const term = spawn5(command.file, command.args, {
       name: "xterm-color",
       cols: 80,
       rows: 24,
@@ -11356,6 +11471,7 @@ var runWorkerDaemon = async (cfg, client, signal) => {
     hasChild: (id) => supervisor.has(id)
   });
   const files = createFileQueryRunner(cwd);
+  const downloads = new AbortController();
   const services = createServiceSupervisor({
     sendReport: (report) => client.services.report(report),
     cfg,
@@ -11413,6 +11529,10 @@ var runWorkerDaemon = async (cfg, client, signal) => {
         void services.stop(cmd.requestId, cmd.name).catch((err) => log(`service stop failed: ${String(err)}`));
       else if (cmd.cmd === "files.query")
         void files(cmd.query).then((reply) => client.files.report({ requestId: cmd.requestId, ...reply })).catch((err) => log(`files query failed: ${String(err)}`));
+      else if (cmd.cmd === "files.download")
+        void sendFileDownload(cwd, cmd, client.files, downloads.signal).catch(
+          (err) => log(`files download failed: ${String(err)}`)
+        );
     } catch (error) {
       log(`invalid command: ${String(error)}`);
     }
@@ -11426,6 +11546,7 @@ var runWorkerDaemon = async (cfg, client, signal) => {
     signal.addEventListener("abort", () => resolve4(), { once: true });
   });
   es.close();
+  downloads.abort();
   liveness.stop();
   services.killAll();
   await supervisor.killAll();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lesscap/baton-cli",
-  "version": "0.9.4",
+  "version": "0.9.5",
   "license": "MIT",
   "publishConfig": {
     "access": "public"
@@ -15,15 +15,23 @@
     "scripts",
     "skills"
   ],
+  "scripts": {
+    "baton": "tsx src/index.ts",
+    "build": "node build.mjs && chmod +x dist/baton.mjs",
+    "postinstall": "node scripts/postinstall.mjs",
+    "prepublishOnly": "pnpm build",
+    "typecheck": "tsc --noEmit -p tsconfig.json",
+    "test": "tsx --test 'src/**/*.test.ts'"
+  },
   "devDependencies": {
+    "@baton/shared": "workspace:*",
     "@types/node": "^24.19.2",
     "@types/ws": "^8.18.1",
     "citty": "^0.2.2",
     "esbuild": "^0.28.2",
     "eventsource": "^4.1.0",
     "tsx": "^4.23.15",
-    "typescript": "^7.0.2",
-    "@baton/shared": "0.0.0"
+    "typescript": "^7.0.2"
   },
   "dependencies": {
     "@anthropic-ai/claude-agent-sdk": "^0.3.316",
@@ -33,12 +41,5 @@
     "mime": "^4.1.0",
     "node-pty": "^1.1.0",
     "ws": "^8.21.0"
-  },
-  "scripts": {
-    "baton": "tsx src/index.ts",
-    "build": "node build.mjs && chmod +x dist/baton.mjs",
-    "postinstall": "node scripts/postinstall.mjs",
-    "typecheck": "tsc --noEmit -p tsconfig.json",
-    "test": "tsx --test 'src/**/*.test.ts'"
   }
-}
+}

package/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2026 bencode
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.