@sma1lboy/kobe 0.5.8 → 0.5.9

package/dist/bin/kobed.js

@@ -362,7 +362,8 @@ var init_binary = __esm(() => {
 });
 
 // src/engine/claude-code-local/history.ts
-import { readFile, readdir, unlink } from "fs/promises";
+import { randomUUID } from "crypto";
+import { appendFile, mkdir, readFile, readdir, unlink, writeFile } from "fs/promises";
 import { homedir as homedir3 } from "os";
 import path2 from "path";
 function encodeCwd(cwd) {
@@ -459,6 +460,76 @@ function extractUsage(v) {
 function isObject(v) {
   return typeof v === "object" && v !== null && !Array.isArray(v);
 }
+async function appendInterruptedUserPrompt(sessionId, cwd, prompt, deps = defaultDeps2) {
+  if (!prompt || prompt.trim().length === 0)
+    return;
+  const projectDir = path2.join(deps.projectsDir(), encodeCwd(cwd));
+  const filePath = path2.join(projectDir, `${sessionId}.jsonl`);
+  let lines = [];
+  try {
+    const raw = await readFile(filePath, "utf8");
+    lines = raw.split(`
+`).filter((l) => l.length > 0);
+  } catch (err) {
+    if (err.code !== "ENOENT")
+      throw err;
+    await mkdir(projectDir, { recursive: true });
+  }
+  let lastConvIdx = -1;
+  let lastConvRecord = null;
+  let lastConvRole = null;
+  for (let i = lines.length - 1;i >= 0; i--) {
+    let parsed;
+    try {
+      parsed = JSON.parse(lines[i]);
+    } catch {
+      continue;
+    }
+    if (!isObject(parsed))
+      continue;
+    const inner = isObject(parsed.message) ? parsed.message : parsed;
+    const role = inner.role;
+    if (role === "user" || role === "assistant") {
+      lastConvIdx = i;
+      lastConvRecord = parsed;
+      lastConvRole = role;
+      break;
+    }
+  }
+  const now = new Date().toISOString();
+  if (lastConvRole === "user" && lastConvRecord && lastConvIdx >= 0) {
+    const inner = isObject(lastConvRecord.message) ? lastConvRecord.message : lastConvRecord;
+    const existing = typeof inner.content === "string" ? inner.content : "";
+    if (existing === prompt || existing.endsWith(`
+
+${prompt}`))
+      return;
+    inner.content = existing.length > 0 ? `${existing}
+
+${prompt}` : prompt;
+    lastConvRecord.timestamp = now;
+    lines[lastConvIdx] = JSON.stringify(lastConvRecord);
+    await writeFile(filePath, `${lines.join(`
+`)}
+`);
+    return;
+  }
+  const parentUuid = lastConvRecord && typeof lastConvRecord.uuid === "string" ? lastConvRecord.uuid : null;
+  const record = {
+    type: "user",
+    message: { role: "user", content: prompt },
+    uuid: randomUUID(),
+    parentUuid,
+    sessionId,
+    cwd,
+    timestamp: now,
+    isSidechain: false,
+    userType: "external",
+    version: "1.0.0"
+  };
+  await appendFile(filePath, `${JSON.stringify(record)}
+`);
+}
 var defaultDeps2;
 var init_history = __esm(() => {
   defaultDeps2 = {
@@ -866,13 +937,21 @@ class ClaudeCodeLocal {
   }
   async stop(handle) {
     const sid = handle.sessionId;
-    await this.registry.kill(sid, this.stopGraceMs);
     const session = this.running.get(sid);
+    const shouldRescue = !!session && !session.completedNaturally && session.prompt.trim().length > 0;
+    const rescuePrompt = session?.prompt ?? "";
+    const rescueCwd = session?.cwd ?? handle.cwd;
+    await this.registry.kill(sid, this.stopGraceMs);
     if (session) {
       session.closed = true;
       this.notify(session);
       this.running.delete(sid);
     }
+    if (shouldRescue) {
+      try {
+        await appendInterruptedUserPrompt(sid, rescueCwd, rescuePrompt);
+      } catch {}
+    }
   }
   async start(args) {
     const binaryPath = await this.binaryPathResolver();
@@ -905,14 +984,17 @@ class ClaudeCodeLocal {
         spawned,
         queue,
         waiters: [],
-        closed: false
+        closed: false,
+        completedNaturally: false,
+        prompt: args.prompt
       };
       this.running.set(sessionId, session);
       this.registry.register({
         sessionId,
         cwd: args.cwd,
         proc: spawned.proc,
-        startedAt: Date.now()
+        startedAt: Date.now(),
+        prompt: args.prompt
       });
       resolveHandle({ sessionId, cwd: args.cwd });
     };
@@ -934,6 +1016,9 @@ class ClaudeCodeLocal {
       try {
         for await (const ev of events) {
           queue.push(ev);
+          if (ev.type === "done" && session) {
+            session.completedNaturally = true;
+          }
           if (session)
             this.notify(session);
         }
@@ -988,11 +1073,11 @@ var init_claude_code_local = __esm(() => {
 });
 
 // src/orchestrator/bridge/server.ts
-import { mkdir, unlink as unlink2 } from "fs/promises";
+import { mkdir as mkdir2, unlink as unlink2 } from "fs/promises";
 import { createServer } from "net";
 import { dirname as dirname2 } from "path";
 async function startBridgeServer(orch, socketPath) {
-  await mkdir(dirname2(socketPath), { recursive: true });
+  await mkdir2(dirname2(socketPath), { recursive: true });
   await unlink2(socketPath).catch(() => {});
   const conns = new Set;
   const server = createServer((conn) => {
@@ -1125,7 +1210,7 @@ var exports_bridge = {};
 __export(exports_bridge, {
   startBridge: () => startBridge
 });
-import { writeFile } from "fs/promises";
+import { writeFile as writeFile2 } from "fs/promises";
 import { homedir as homedir5 } from "os";
 import { join as join3 } from "path";
 import { fileURLToPath as fileURLToPath2 } from "url";
@@ -1145,7 +1230,7 @@ async function startBridge(orch, opts = {}) {
       }
     }
   };
-  await writeFile(mcpConfigPath, JSON.stringify(mcpConfig, null, 2), "utf8");
+  await writeFile2(mcpConfigPath, JSON.stringify(mcpConfig, null, 2), "utf8");
   process.env.KOBE_MCP_CONFIG = mcpConfigPath;
   return {
     socketPath,
@@ -3034,6 +3119,7 @@ class Orchestrator {
   worktrees;
   metadataSuggester;
   handles = new Map;
+  firstSpawnLatches = new Map;
   subscribers = new Map;
   pumps = new Map;
   pendingInputBroker = new InMemoryPendingInputBroker;
@@ -3266,8 +3352,16 @@ class Orchestrator {
       const renameTabId = this.resolveTab(task, tabId).id;
       this.maybeRenameTempBranch(task.id, renameTabId, prompt);
     }
-    const targetTab = this.resolveTab(task, tabId);
+    let targetTab = this.resolveTab(task, tabId);
     const key = tabKey(task.id, targetTab.id);
+    if (!targetTab.sessionId) {
+      const inflight = this.firstSpawnLatches.get(key);
+      if (inflight) {
+        await inflight.catch(() => {});
+        task = this.requireTask(id);
+        targetTab = this.resolveTab(task, tabId);
+      }
+    }
     if (this.handles.has(key) === false) {
       const running = this.countRunning();
       if (running >= CONCURRENCY_CAP) {
@@ -3288,18 +3382,28 @@ class Orchestrator {
         model: modelToUse
       });
     } else {
-      handle = await this.engine.spawn(task.worktreePath, promptToSend, {
-        permissionMode: task.permissionMode,
-        model: modelToUse
+      let releaseLatch = () => {};
+      const latch = new Promise((resolve2) => {
+        releaseLatch = resolve2;
       });
-      await this.updateTab(task.id, targetTab.id, { sessionId: handle.sessionId });
-      if (task.title === PLACEHOLDER_TASK_TITLE && prompt && prompt.trim().length > 0) {
-        const derived = deriveTitleFromPrompt(prompt);
-        if (derived)
-          await this.store.update(task.id, { title: derived });
-      }
-      if (prompt && prompt.trim().length > 0) {
-        this.maybeUpgradeTitle(task.id, prompt);
+      this.firstSpawnLatches.set(key, latch);
+      try {
+        handle = await this.engine.spawn(task.worktreePath, promptToSend, {
+          permissionMode: task.permissionMode,
+          model: modelToUse
+        });
+        await this.updateTab(task.id, targetTab.id, { sessionId: handle.sessionId });
+        if (task.title === PLACEHOLDER_TASK_TITLE && prompt && prompt.trim().length > 0) {
+          const derived = deriveTitleFromPrompt(prompt);
+          if (derived)
+            await this.store.update(task.id, { title: derived });
+        }
+        if (prompt && prompt.trim().length > 0) {
+          this.maybeUpgradeTitle(task.id, prompt);
+        }
+      } finally {
+        releaseLatch();
+        this.firstSpawnLatches.delete(key);
       }
     }
     this.handles.set(key, handle);
@@ -3366,6 +3470,12 @@ class Orchestrator {
     }
     this.dispatchEvent(task.id, targetTab.id, { type: "done" });
   }
+  async steerTask(id, prompt, tabId) {
+    const task = this.requireTask(id);
+    const targetTab = this.resolveTab(task, tabId);
+    await this.interruptTask(task.id, targetTab.id);
+    await this.runTask(task.id, prompt, targetTab.id);
+  }
   async pauseTask(id) {
     const task = this.requireTask(id);
     if (task.status !== "in_progress") {
@@ -3796,7 +3906,7 @@ var init_core = __esm(() => {
 });
 
 // src/orchestrator/index/store.ts
-import { mkdir as mkdir2, open, readFile as readFile3, rename, unlink as unlink3 } from "fs/promises";
+import { mkdir as mkdir3, open, readFile as readFile3, rename, unlink as unlink3 } from "fs/promises";
 import { homedir as homedir8 } from "os";
 import { dirname as dirname4, join as join6 } from "path";
 
@@ -3870,7 +3980,7 @@ class TaskIndexStore {
     return next;
   }
   async doSave() {
-    await mkdir2(dirname4(this.path), { recursive: true });
+    await mkdir3(dirname4(this.path), { recursive: true });
     const payload = this.snapshot();
     const json = `${JSON.stringify(payload, null, 2)}
 `;
@@ -4421,7 +4531,7 @@ init_paths();
 // src/daemon/server.ts
 init_repos();
 init_paths();
-import { mkdir as mkdir3, readFile as readFile5, unlink as unlink4, writeFile as writeFile3 } from "fs/promises";
+import { mkdir as mkdir4, readFile as readFile5, unlink as unlink4, writeFile as writeFile4 } from "fs/promises";
 import { createServer as createServer2 } from "net";
 import { dirname as dirname5 } from "path";
 
@@ -4761,8 +4871,8 @@ async function startDaemonServer(orch, options = {}) {
   const startedAt = options.startedAt ?? new Date;
   const clients = new Set;
   let nextClientId = 1;
-  await mkdir3(dirname5(socketPath), { recursive: true });
-  await mkdir3(dirname5(pidPath), { recursive: true });
+  await mkdir4(dirname5(socketPath), { recursive: true });
+  await mkdir4(dirname5(pidPath), { recursive: true });
   await unlink4(socketPath).catch(() => {});
   const server = createServer2((socket) => {
     const client = {
@@ -4820,7 +4930,7 @@ async function startDaemonServer(orch, options = {}) {
       resolve2();
     });
   });
-  await writeFile3(pidPath, `${process.pid}
+  await writeFile4(pidPath, `${process.pid}
 `, "utf8");
   async function stopSoon() {
     await options.onStop?.();
@@ -4990,6 +5100,10 @@ async function startDaemonServer(orch, options = {}) {
         await orch.interruptTask(requireString2(payload, "taskId"), optionalString2(payload, "tabId"));
         return {};
       }
+      case "chat.steer": {
+        await orch.steerTask(requireString2(payload, "taskId"), requireString2(payload, "text"), optionalString2(payload, "tabId"));
+        return {};
+      }
       case "chat.input.pending": {
         return { pending: orch.peekPendingInput(requireString2(payload, "taskId")) };
       }