@lesscap/baton-cli 0.2.2 → 0.2.4

package/dist/baton.mjs

@@ -550,6 +550,7 @@ var requirementClient = (baseUrl) => {
     listByProject: (projectId) => request(u(`/projects/${projectId}/requirements`), { method: "GET" }),
     get: (id) => request(u(`/requirements/${id}`), { method: "GET" }),
     getByCode: (projectId, code) => fetchItemByCode(baseUrl, projectId, code, "requirement"),
+    update: (id, patch2) => request(u(`/requirements/${id}`), { method: "PATCH", body: patch2 }),
     setStatus: (id, status) => request(u(`/requirements/${id}`), { method: "PATCH", body: { status } }),
     remove: (id) => request(u(`/requirements/${id}`), { method: "DELETE" })
   };
@@ -587,9 +588,13 @@ var sessionsClient = (baseUrl) => {
       const matches = all.filter((s) => s.name === name);
       return matches.length === 0 ? null : matches[matches.length - 1] ?? null;
     },
-    sendMessage: (id, text, attachments) => request(u(`/sessions/${id}/messages`), {
+    sendMessage: (id, text, attachments, planMode) => request(u(`/sessions/${id}/messages`), {
       method: "POST",
-      body: attachments && attachments.length > 0 ? { text, attachments } : { text }
+      body: {
+        text,
+        ...attachments && attachments.length > 0 ? { attachments } : {},
+        ...planMode ? { planMode: true } : {}
+      }
     }),
     // Raw-body upload: the file streams as the request body (no multipart), so
     // a Blob backed by a file on disk uploads without buffering. filename rides
@@ -618,6 +623,7 @@ var taskClient = (baseUrl) => {
     listByRequirement: (requirementId) => request(u(`/requirements/${requirementId}/tasks`), { method: "GET" }),
     get: (id) => request(u(`/tasks/${id}`), { method: "GET" }),
     getByCode: (projectId, code) => fetchItemByCode(baseUrl, projectId, code, "task"),
+    update: (id, patch2) => request(u(`/tasks/${id}`), { method: "PATCH", body: patch2 }),
     setStatus: (id, status) => request(u(`/tasks/${id}`), { method: "PATCH", body: { status } }),
     remove: (id) => request(u(`/tasks/${id}`), { method: "DELETE" }),
     listComments: (id) => request(u(`/tasks/${id}/comments`), { method: "GET" }),
@@ -734,6 +740,11 @@ var viewWorker = (cfg) => {
     apiToken: worker2.apiToken
   };
 };
+var worktreeConfig = (cfg) => ({
+  server: cfg.server,
+  project: cfg.projectId,
+  worker: { id: cfg.workerId, name: cfg.name, machineId: cfg.machineId, apiToken: cfg.apiToken }
+});
 
 // src/config.ts
 var resolveBaseUrl = (urlArg, env = process.env) => {
@@ -772,6 +783,12 @@ var resolveWorkspaceId = (args) => {
     "no workspace in scope. pass --workspace <id> or run `baton init` in this directory."
   );
 };
+var parseIssueUrl = (url) => {
+  const m = url.match(/^https:\/\/github\.com\/[^/]+\/[^/]+\/issues\/(\d+)/);
+  if (!m)
+    throw new Error(`expected a GitHub issue url (https://github.com/o/r/issues/N), got "${url}"`);
+  return { source: "github", number: Number(m[1]), url: url.split(/[?#]/)[0] };
+};
 var splitCsv = (s) => s ? s.split(",").map((x) => x.trim()).filter(Boolean) : void 0;
 
 // src/commands/init.ts
@@ -931,6 +948,7 @@ var project = defineCommand({
 
 // src/commands/requirement.ts
 var createRequirement = (c, input, json) => c.requirements.create(input).then((r) => renderOne(r, fmtRequirement, json));
+var updateRequirement = (c, id, patch2, json) => c.requirements.update(id, patch2).then((r) => renderOne(r, fmtRequirement, json));
 var listRequirements = (c, projectId, json) => c.requirements.listByProject(projectId).then((rs) => renderList(rs, fmtRequirement, json));
 var getRequirement = (c, id, json) => c.requirements.get(id).then((r) => renderOne(r, fmtRequirement, json));
 var setRequirementStatus = (c, id, status, json) => c.requirements.setStatus(id, status).then((r) => renderOne(r, fmtRequirement, json));
@@ -946,6 +964,7 @@ var requirement = defineCommand({
         project: { type: "string", description: "project id (overrides .baton.json)" },
         desc: { type: "string", description: "description" },
         body: { type: "string", description: "requirement body (markdown)" },
+        github: { type: "string", description: "link a GitHub issue url (light association)" },
         ...common
       },
       run: async ({ args }) => {
@@ -956,13 +975,69 @@ var requirement = defineCommand({
               projectId: resolveProjectId(args),
               title: args.title,
               description: args.desc,
-              body: args.body
+              body: args.body,
+              external: args.github ? parseIssueUrl(args.github) : void 0
             },
             Boolean(args.json)
           )
         );
       }
     }),
+    update: defineCommand({
+      meta: { name: "update", description: "update a requirement title / body" },
+      args: {
+        code: { type: "positional", required: true, description: "requirement code (R-N)" },
+        title: { type: "string", description: "new title" },
+        body: { type: "string", description: "new body (markdown)" },
+        project: { type: "string", description: "project id (overrides .baton.json)" },
+        ...common
+      },
+      run: async ({ args }) => {
+        const patch2 = {
+          ...args.title !== void 0 ? { title: args.title } : {},
+          ...args.body !== void 0 ? { body: args.body } : {}
+        };
+        if (Object.keys(patch2).length === 0) throw new Error("pass --title and/or --body");
+        const c = clientFor(args);
+        const id = await resolveByCode(c, resolveProjectId(args), args.code);
+        console.log(await updateRequirement(c, id, patch2, Boolean(args.json)));
+      }
+    }),
+    link: defineCommand({
+      meta: { name: "link", description: "link a requirement to a GitHub issue url" },
+      args: {
+        code: { type: "positional", required: true, description: "requirement code (R-N)" },
+        // Named `issue` (not `url`) — `url` would collide with the common --url flag.
+        issue: { type: "positional", required: true, description: "GitHub issue url" },
+        project: { type: "string", description: "project id (overrides .baton.json)" },
+        ...common
+      },
+      run: async ({ args }) => {
+        const c = clientFor(args);
+        const id = await resolveByCode(c, resolveProjectId(args), args.code);
+        console.log(
+          await updateRequirement(
+            c,
+            id,
+            { external: parseIssueUrl(args.issue) },
+            Boolean(args.json)
+          )
+        );
+      }
+    }),
+    unlink: defineCommand({
+      meta: { name: "unlink", description: "clear a requirement\u2019s GitHub issue association" },
+      args: {
+        code: { type: "positional", required: true, description: "requirement code (R-N)" },
+        project: { type: "string", description: "project id (overrides .baton.json)" },
+        ...common
+      },
+      run: async ({ args }) => {
+        const c = clientFor(args);
+        const id = await resolveByCode(c, resolveProjectId(args), args.code);
+        console.log(await updateRequirement(c, id, { external: null }, Boolean(args.json)));
+      }
+    }),
     ls: defineCommand({
       meta: { name: "ls", description: "list requirements in a project" },
       args: {
@@ -1795,14 +1870,15 @@ ${text}` : header;
 };
 
 // src/session/runner/query.ts
-var startQuery = (config, text, resuming, queryFn, abortController, log, envOverlay) => {
+var startQuery = (config, text, resuming, queryFn, abortController, log, envOverlay, planMode = false) => {
   const env = buildSdkEnv(envOverlay);
   const exe = claudeExecutable();
   const addDirs = additionalDirs();
   if (addDirs) log(`[add-dir] ${addDirs.join(", ")}`);
+  if (planMode) log("[plan] read-only planning turn");
   const options = {
     cwd: config.worktreePath,
-    permissionMode: "bypassPermissions",
+    permissionMode: planMode ? "plan" : "bypassPermissions",
     includePartialMessages: false,
     abortController,
     stderr: (line) => log(`[claude] ${line.trimEnd()}`),
@@ -1868,6 +1944,7 @@ var runTurn = async (config, worker2, msg, resuming, queryFn, log = (m) => conso
   const payload = msg.payload;
   const rawText = typeof payload?.text === "string" ? payload.text : "";
   const attachments = Array.isArray(payload?.attachments) ? payload.attachments : [];
+  const planMode = payload?.planMode === true;
   if (rawText.length === 0 && attachments.length === 0) {
     await worker2.emitEvent("turn_error", { message: "user_message missing text and attachments" });
     return -1;
@@ -1883,15 +1960,15 @@ var runTurn = async (config, worker2, msg, resuming, queryFn, log = (m) => conso
   const abort = new AbortController();
   const ceiling = turnTimeoutMs();
   let timer;
-  const timeout = new Promise((resolve) => {
-    timer = setTimeout(() => resolve(TIMEOUT), ceiling);
+  const timeout = new Promise((resolve2) => {
+    timer = setTimeout(() => resolve2(TIMEOUT), ceiling);
   });
-  const interrupted = new Promise((resolve) => {
-    if (externalSignal?.aborted) resolve(ABORTED);
-    else externalSignal?.addEventListener("abort", () => resolve(ABORTED), { once: true });
+  const interrupted = new Promise((resolve2) => {
+    if (externalSignal?.aborted) resolve2(ABORTED);
+    else externalSignal?.addEventListener("abort", () => resolve2(ABORTED), { once: true });
   });
   try {
-    const messages = startQuery(config, text, resuming, queryFn, abort, log, envOverlay);
+    const messages = startQuery(config, text, resuming, queryFn, abort, log, envOverlay, planMode);
     const consume = streamSdkEvents(messages, worker2);
     void consume.catch(() => {
     });
@@ -1906,7 +1983,9 @@ var runTurn = async (config, worker2, msg, resuming, queryFn, log = (m) => conso
     if (outcome === TIMEOUT) {
       log(`[watchdog] turn exceeded ${ceiling}ms \u2014 aborting claude`);
       abort.abort();
-      await worker2.emitEvent("turn_error", { message: `turn exceeded ${ceiling}ms \u2014 claude aborted` });
+      await worker2.emitEvent("turn_error", {
+        message: `turn exceeded ${ceiling}ms \u2014 claude aborted`
+      });
       await worker2.emitEvent("turn_complete", {});
       return -1;
     }
@@ -1940,9 +2019,9 @@ var subscribeStream = (url, ESCtor, seen, log, onUserMessage, onInterrupt, onRea
   es.onerror = () => log("sse error (eventsource will retry)");
   return es;
 };
-var waitForAbort = (signal) => new Promise((resolve) => {
-  if (signal.aborted) return resolve();
-  signal.addEventListener("abort", () => resolve(), { once: true });
+var waitForAbort = (signal) => new Promise((resolve2) => {
+  if (signal.aborted) return resolve2();
+  signal.addEventListener("abort", () => resolve2(), { once: true });
 });
 var shouldReap = (lastActivity, now, busy, queueLen, idleMs) => !busy && queueLen === 0 && now - lastActivity >= idleMs;
 var idleMsFromEnv = () => Number(process.env.BATON_SESSION_IDLE_MS) || 30 * 6e4;
@@ -2025,12 +2104,12 @@ var runDaemon = async (config, deps, signal) => {
   );
   const idleMs = idleMsFromEnv();
   let idleTimer;
-  const reaped = new Promise((resolve) => {
+  const reaped = new Promise((resolve2) => {
     idleTimer = setInterval(
       () => {
         if (shouldReap(lastActivity, Date.now(), busy, pendingQueue.length, idleMs)) {
           log("idle \u2014 shutting down (resumes on next message)");
-          resolve();
+          resolve2();
         }
       },
       Math.min(idleMs, 3e4)
@@ -2209,6 +2288,7 @@ var session = defineCommand({
 
 // src/commands/task.ts
 var createTask = (c, input, json) => c.tasks.create(input).then((t) => renderOne(t, fmtTask, json));
+var updateTask = (c, id, patch2, json) => c.tasks.update(id, patch2).then((t) => renderOne(t, fmtTask, json));
 var listTasks = (c, requirementId, json) => c.tasks.listByRequirement(requirementId).then((ts) => renderList(ts, fmtTask, json));
 var getTask = (c, id, json) => c.tasks.get(id).then((t) => renderOne(t, fmtTask, json));
 var setTaskStatus = (c, id, status, json) => c.tasks.setStatus(id, status).then((t) => renderOne(t, fmtTask, json));
@@ -2228,6 +2308,7 @@ var task = defineCommand({
         project: { type: "string", description: "project id (overrides .baton.json)" },
         body: { type: "string", description: "task body (markdown)" },
         deps: { type: "string", description: "comma-separated dependency task codes (T-N,T-N)" },
+        github: { type: "string", description: "link a GitHub issue url (light association)" },
         ...common
       },
       run: async ({ args }) => {
@@ -2245,13 +2326,64 @@ var task = defineCommand({
               requirementId,
               title: args.title,
               body: args.body,
-              dependsOn: dependsOn.length ? dependsOn : void 0
+              dependsOn: dependsOn.length ? dependsOn : void 0,
+              external: args.github ? parseIssueUrl(args.github) : void 0
             },
             Boolean(args.json)
           )
         );
       }
     }),
+    update: defineCommand({
+      meta: { name: "update", description: "update a task title / body" },
+      args: {
+        code: { type: "positional", required: true, description: "task code (T-N)" },
+        title: { type: "string", description: "new title" },
+        body: { type: "string", description: "new body (markdown)" },
+        project: { type: "string", description: "project id (overrides .baton.json)" },
+        ...common
+      },
+      run: async ({ args }) => {
+        const patch2 = {
+          ...args.title !== void 0 ? { title: args.title } : {},
+          ...args.body !== void 0 ? { body: args.body } : {}
+        };
+        if (Object.keys(patch2).length === 0) throw new Error("pass --title and/or --body");
+        const c = clientFor(args);
+        const id = await resolveTaskByCode(c, resolveProjectId(args), args.code);
+        console.log(await updateTask(c, id, patch2, Boolean(args.json)));
+      }
+    }),
+    link: defineCommand({
+      meta: { name: "link", description: "link a task to a GitHub issue url" },
+      args: {
+        code: { type: "positional", required: true, description: "task code (T-N)" },
+        // Named `issue` (not `url`) — `url` would collide with the common --url flag.
+        issue: { type: "positional", required: true, description: "GitHub issue url" },
+        project: { type: "string", description: "project id (overrides .baton.json)" },
+        ...common
+      },
+      run: async ({ args }) => {
+        const c = clientFor(args);
+        const id = await resolveTaskByCode(c, resolveProjectId(args), args.code);
+        console.log(
+          await updateTask(c, id, { external: parseIssueUrl(args.issue) }, Boolean(args.json))
+        );
+      }
+    }),
+    unlink: defineCommand({
+      meta: { name: "unlink", description: "clear a task\u2019s GitHub issue association" },
+      args: {
+        code: { type: "positional", required: true, description: "task code (T-N)" },
+        project: { type: "string", description: "project id (overrides .baton.json)" },
+        ...common
+      },
+      run: async ({ args }) => {
+        const c = clientFor(args);
+        const id = await resolveTaskByCode(c, resolveProjectId(args), args.code);
+        console.log(await updateTask(c, id, { external: null }, Boolean(args.json)));
+      }
+    }),
     ls: defineCommand({
       meta: { name: "ls", description: "list tasks in a requirement" },
       args: {
@@ -2281,7 +2413,7 @@ var task = defineCommand({
     "set-status": defineCommand({
       meta: {
         name: "set-status",
-        description: "set task status (todo|in_progress|done|failed|cancelled)"
+        description: "set task status (todo|in_progress|blocked|done|failed|cancelled)"
       },
       args: {
         code: { type: "positional", required: true, description: "task code (T-N)" },
@@ -2359,7 +2491,7 @@ import { query as query2 } from "@anthropic-ai/claude-agent-sdk";
 // src/session/runner/title.ts
 var clip = (s, max) => s.length > max ? s.slice(0, max) : s;
 var isDeclined = (title) => title.replace(/[^a-z]/gi, "").toUpperCase() === "NONE";
-var sanitizeTitle = (raw) => raw.replace(/[\r\n]+/g, " ").replace(/["'`]/g, "").replace(/^\s*(title|session)\s*[:-]\s*/i, "").replace(/\s+/g, " ").trim().slice(0, 40).trim();
+var sanitizeTitle = (raw) => raw.replace(/[\r\n]+/g, " ").replace(/[*_`~#>]/g, "").replace(/["']/g, "").replace(/^\s*(?:title|session|标题)\s*[:：-]\s*/i, "").replace(/^\s*(?:[-*•]|\d+[.)、])\s+/, "").replace(/\s+/g, " ").trim().slice(0, 30).replace(/[\s。．，,、；;：:!！?？.]+$/, "").trim();
 var TITLE_TIMEOUT_MS = 3e4;
 var generateTitle = async (input) => {
   const { worktreePath, userText, assistantText, queryFn } = input;
@@ -2368,7 +2500,8 @@ var generateTitle = async (input) => {
     `User: ${clip(userText, 800)}`,
     assistantText ? `Assistant: ${clip(assistantText, 800)}` : ""
   ].filter(Boolean).join("\n");
-  const prompt = `Reply with ONLY a 2-5 word title (no quotes, no punctuation) summarizing this coding session. If it is too short or generic to title meaningfully, reply with exactly NONE:
+  const prompt = `Reply with ONLY a short title for this session: a noun phrase like a filename \u2014 at most 6 words or ~12 Chinese characters. It must NOT be a sentence, no punctuation, no markdown, no quotes, and never an opener like "\u597D\u7684"/"\u6536\u5230"/"I'll". If it is too thin or generic to title meaningfully, reply with exactly NONE.
+
 ${exchange}`;
   const exe = claudeExecutable();
   const abort = new AbortController();
@@ -2379,6 +2512,9 @@ ${exchange}`;
       prompt,
       options: {
         cwd: worktreePath,
+        // Don't load the worktree's CLAUDE.md / project settings — they steer the
+        // model into verbose summaries instead of a terse title.
+        settingSources: [],
         permissionMode: "bypassPermissions",
         allowedTools: [],
         includePartialMessages: false,
@@ -2401,8 +2537,8 @@ ${exchange}`;
 
 // src/session/worktree.ts
 import { execFileSync, spawnSync as spawnSync2 } from "node:child_process";
-import { existsSync as existsSync4, mkdirSync } from "node:fs";
-import { dirname } from "node:path";
+import { appendFileSync, existsSync as existsSync4, mkdirSync, readFileSync as readFileSync3 } from "node:fs";
+import { dirname, isAbsolute as isAbsolute2, resolve } from "node:path";
 var createWorktree = (input) => {
   if (existsSync4(input.worktreePath))
     throw new Error(`worktree path already exists: ${input.worktreePath}`);
@@ -2425,6 +2561,24 @@ var repoHeadBranch = (repo) => {
   const branch = r.status === 0 ? r.stdout.trim() : "";
   return branch || "main";
 };
+var ensureExcluded = (repo, pattern) => {
+  try {
+    const r = spawnSync2("git", ["-C", repo, "rev-parse", "--git-path", "info/exclude"], {
+      stdio: "pipe",
+      encoding: "utf8"
+    });
+    if (r.status !== 0) return;
+    const rel = r.stdout.trim();
+    const path = isAbsolute2(rel) ? rel : resolve(repo, rel);
+    const current = existsSync4(path) ? readFileSync3(path, "utf8") : "";
+    if (current.split("\n").includes(pattern)) return;
+    mkdirSync(dirname(path), { recursive: true });
+    const sep = current === "" || current.endsWith("\n") ? "" : "\n";
+    appendFileSync(path, `${sep}${pattern}
+`, "utf8");
+  } catch {
+  }
+};
 var removeWorktree = (repo, worktreePath) => {
   if (!existsSync4(worktreePath)) return;
   try {
@@ -2494,6 +2648,8 @@ var runWorkerDaemon = async (cfg, client, signal) => {
       await client.sessions.materialize(sessionId, { agentSessionId, worktreePath }, cfg.apiToken);
       log(`materialized session #${sessionId} \u2192 ${worktreePath}`);
     }
+    ensureExcluded(repo, PROJECT_CONFIG_NAME);
+    saveProjectConfig(join5(worktreePath, PROJECT_CONFIG_NAME), worktreeConfig(cfg));
     spawnChild(sessionId, worktreePath);
   };
   const onStop = (sessionId) => {
@@ -2565,9 +2721,9 @@ var runWorkerDaemon = async (cfg, client, signal) => {
   };
   es.onerror = () => log("command stream error (eventsource will retry)");
   log(`listening for commands (server ${cfg.server}, repo ${repo})`);
-  await new Promise((resolve) => {
-    if (signal.aborted) return resolve();
-    signal.addEventListener("abort", () => resolve(), { once: true });
+  await new Promise((resolve2) => {
+    if (signal.aborted) return resolve2();
+    signal.addEventListener("abort", () => resolve2(), { once: true });
   });
   es.close();
   clearInterval(hb);
@@ -2576,13 +2732,13 @@ var runWorkerDaemon = async (cfg, client, signal) => {
 
 // src/worker/machine-id.ts
 import { randomUUID as randomUUID2 } from "node:crypto";
-import { existsSync as existsSync6, mkdirSync as mkdirSync2, readFileSync as readFileSync3, writeFileSync as writeFileSync2 } from "node:fs";
+import { existsSync as existsSync6, mkdirSync as mkdirSync2, readFileSync as readFileSync4, writeFileSync as writeFileSync2 } from "node:fs";
 import { homedir as homedir3 } from "node:os";
 import { dirname as dirname3, join as join6 } from "node:path";
 var machineIdPath = (env = process.env) => join6(env.XDG_DATA_HOME ?? join6(env.HOME ?? homedir3(), ".local/share"), "baton", "machine-id");
 var readOrCreateMachineId = (path = machineIdPath()) => {
   if (existsSync6(path)) {
-    const s = readFileSync3(path, "utf8").trim();
+    const s = readFileSync4(path, "utf8").trim();
     if (s) return s;
   }
   const id = randomUUID2();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lesscap/baton-cli",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "type": "module",
   "description": "baton CLI — register/run a worker, manage projects/requirements/tasks/sessions",
   "bin": {