@devness/useai 0.5.26 → 0.5.28

package/dist/index.js

@@ -2681,7 +2681,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.5.26";
+    VERSION = "0.5.28";
   }
 });
 
@@ -17070,13 +17070,13 @@ function showManualHints(installedTools) {
   for (const { name, hint } of hints) {
     console.log(`  ${source_default.bold(name)}: ${hint}`);
   }
-  console.log(source_default.dim("\n  \u250C\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2510"));
+  console.log();
   for (const line of USEAI_INSTRUCTIONS_TEXT.split("\n")) {
-    console.log(source_default.dim("  \u2502 ") + line);
+    console.log(`  ${line}`);
   }
-  console.log(source_default.dim("  \u2514\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2518\n"));
+  console.log();
 }
-async function daemonInstallFlow(tools, explicit) {
+async function daemonInstallFlow(tools, autoYes, explicit) {
   console.log(source_default.dim("  Ensuring UseAI daemon is running..."));
   const daemonOk = await ensureDaemon();
   let useDaemon = true;
@@ -17099,30 +17099,105 @@ async function daemonInstallFlow(tools, explicit) {
       }
     }
   }
-  const targetTools = explicit ? tools : tools.filter((t) => t.detect());
-  if (targetTools.length === 0) {
-    console.log(source_default.red("\n  No AI tools detected on this machine."));
+  if (explicit) {
+    console.log();
+    let configuredCount2 = 0;
+    for (const tool of tools) {
+      try {
+        configureToolDaemon(tool, useDaemon);
+        configuredCount2++;
+      } catch (e) {
+        console.log(source_default.red(`  \u2717 ${tool.name.padEnd(18)} \u2014 ${e.message}`));
+      }
+    }
+    installHooksAndFinish(tools, configuredCount2, useDaemon);
     return;
   }
-  let configuredCount = 0;
+  console.log(source_default.dim("\n  Scanning for AI tools...\n"));
+  const detected = tools.filter((t) => t.detect());
+  if (detected.length === 0) {
+    console.log(source_default.red("  No AI tools detected on this machine."));
+    return;
+  }
+  const alreadyConfigured = detected.filter((t) => t.isConfigured());
+  const unconfigured = detected.filter((t) => !t.isConfigured());
+  console.log(`  Found ${source_default.bold(String(detected.length))} AI tool${detected.length === 1 ? "" : "s"} on this machine:
+`);
+  for (const tool of alreadyConfigured) {
+    console.log(source_default.green(`  \u2705 ${tool.name}`) + source_default.dim("  (already configured)"));
+  }
+  for (const tool of unconfigured) {
+    console.log(source_default.dim(`  \u2610  ${tool.name}`));
+  }
   console.log();
-  for (const tool of targetTools) {
-    try {
-      if (useDaemon && tool.supportsUrl) {
-        tool.installHttp();
-        console.log(source_default.green(`  \u2713 ${tool.name.padEnd(18)} \u2192 ${source_default.dim("HTTP (daemon)")}`));
-      } else if (useDaemon && !tool.supportsUrl) {
-        tool.install();
-        console.log(source_default.green(`  \u2713 ${tool.name.padEnd(18)} \u2192 ${source_default.dim("stdio (no URL support)")}`));
-      } else {
-        tool.install();
-        console.log(source_default.green(`  \u2713 ${tool.name.padEnd(18)} \u2192 ${source_default.dim("stdio")}`));
+  if (unconfigured.length === 0) {
+    console.log(source_default.green("  All detected tools are already configured."));
+    for (const tool of alreadyConfigured) {
+      try {
+        configureToolDaemon(tool, useDaemon);
+      } catch {
       }
+    }
+    installHooksAndFinish(alreadyConfigured, alreadyConfigured.length, useDaemon);
+    return;
+  }
+  let toInstall;
+  if (autoYes) {
+    toInstall = unconfigured;
+  } else {
+    let selected;
+    try {
+      selected = await dist_default4({
+        message: "Select tools to configure:",
+        choices: unconfigured.map((t) => ({
+          name: t.name,
+          value: t.id,
+          checked: true
+        }))
+      });
+    } catch {
+      console.log("\n");
+      return;
+    }
+    toInstall = unconfigured.filter((t) => selected.includes(t.id));
+  }
+  if (toInstall.length === 0) {
+    console.log(source_default.dim("  No tools selected."));
+    return;
+  }
+  console.log(`
+  Configuring ${toInstall.length} tool${toInstall.length === 1 ? "" : "s"}...
+`);
+  let configuredCount = 0;
+  for (const tool of toInstall) {
+    try {
+      configureToolDaemon(tool, useDaemon);
       configuredCount++;
     } catch (e) {
       console.log(source_default.red(`  \u2717 ${tool.name.padEnd(18)} \u2014 ${e.message}`));
     }
   }
+  for (const tool of alreadyConfigured) {
+    try {
+      configureToolDaemon(tool, useDaemon);
+    } catch {
+    }
+  }
+  installHooksAndFinish([...toInstall, ...alreadyConfigured], configuredCount, useDaemon);
+}
+function configureToolDaemon(tool, useDaemon) {
+  if (useDaemon && tool.supportsUrl) {
+    tool.installHttp();
+    console.log(source_default.green(`  \u2713 ${tool.name.padEnd(18)} \u2192 ${source_default.dim("HTTP (daemon)")}`));
+  } else if (useDaemon && !tool.supportsUrl) {
+    tool.install();
+    console.log(source_default.green(`  \u2713 ${tool.name.padEnd(18)} \u2192 ${source_default.dim("stdio (no URL support)")}`));
+  } else {
+    tool.install();
+    console.log(source_default.green(`  \u2713 ${tool.name.padEnd(18)} \u2192 ${source_default.dim("stdio")}`));
+  }
+}
+function installHooksAndFinish(allTools, configuredCount, useDaemon) {
   try {
     const hooksInstalled = installClaudeCodeHooks();
     if (hooksInstalled) {
@@ -17131,7 +17206,7 @@ async function daemonInstallFlow(tools, explicit) {
   } catch {
     console.log(source_default.yellow("  \u26A0 Could not install Claude Code hooks"));
   }
-  showManualHints(targetTools);
+  showManualHints(allTools);
   const mode = useDaemon ? "daemon mode" : "stdio mode";
   console.log(`
   Done! UseAI configured in ${source_default.bold(String(configuredCount))} tool${configuredCount === 1 ? "" : "s"} (${mode}).`);
@@ -17301,7 +17376,7 @@ async function runSetup(args) {
   } else if (isStdio) {
     await shared.installFlow(tools, autoYes, explicit);
   } else {
-    await daemonInstallFlow(tools, explicit);
+    await daemonInstallFlow(tools, autoYes, explicit);
   }
 }
 var shared;
@@ -37343,32 +37418,95 @@ function parseBody(req) {
     req.on("error", reject);
   });
 }
-function tryRecoverEndSession(staleMcpSessionId, body, res) {
+function sendJsonRpcResult(res, rpcId, text) {
+  res.writeHead(200, { "Content-Type": "application/json" });
+  res.end(JSON.stringify({
+    jsonrpc: "2.0",
+    result: { content: [{ type: "text", text }] },
+    id: rpcId
+  }));
+}
+function readChainMetadata(useaiSessionId) {
+  const activePath = join10(ACTIVE_DIR, `${useaiSessionId}.jsonl`);
+  const sealedPath = join10(SEALED_DIR, `${useaiSessionId}.jsonl`);
+  const chainPath = existsSync11(activePath) ? activePath : existsSync11(sealedPath) ? sealedPath : null;
+  if (!chainPath) return null;
+  try {
+    const firstLine = readFileSync5(chainPath, "utf-8").split("\n")[0];
+    if (!firstLine) return null;
+    const record2 = JSON.parse(firstLine);
+    const d = record2.data;
+    return {
+      client: d["client"] ?? "unknown",
+      startTime: record2.timestamp,
+      taskType: d["task_type"] ?? "coding",
+      title: d["title"] ?? void 0,
+      privateTitle: d["private_title"] ?? void 0,
+      project: d["project"] ?? void 0,
+      convId: d["conversation_id"] ?? void 0,
+      convIdx: d["conversation_index"] ?? void 0
+    };
+  } catch {
+    return null;
+  }
+}
+function recoverStartSession(staleMcpSessionId, args, rpcId, res) {
+  const map = readMcpMap();
+  const prevSessionId = map[staleMcpSessionId];
+  if (prevSessionId) {
+    const prevActivePath = join10(ACTIVE_DIR, `${prevSessionId}.jsonl`);
+    if (existsSync11(prevActivePath)) {
+      sealOrphanFile(prevSessionId);
+    }
+  }
+  let client = "unknown";
+  if (prevSessionId) {
+    const meta = readChainMetadata(prevSessionId);
+    if (meta) client = meta.client;
+  }
+  const newSessionId = randomUUID4();
+  const taskType = args["task_type"] ?? "coding";
+  const title = args["title"];
+  const privateTitle = args["private_title"];
+  const project = args["project"];
+  const convId = prevSessionId ? readChainMetadata(prevSessionId)?.convId ?? randomUUID4() : randomUUID4();
+  const chainData = {
+    client,
+    task_type: taskType,
+    project,
+    conversation_id: convId,
+    version: VERSION,
+    recovered: true
+  };
+  if (title) chainData["title"] = title;
+  if (privateTitle) chainData["private_title"] = privateTitle;
+  const record2 = buildChainRecord("session_start", newSessionId, chainData, "GENESIS", daemonSigningKey);
+  const chainPath = join10(ACTIVE_DIR, `${newSessionId}.jsonl`);
+  appendFileSync2(chainPath, JSON.stringify(record2) + "\n");
+  writeMcpMapping(staleMcpSessionId, newSessionId);
+  sendJsonRpcResult(
+    res,
+    rpcId,
+    `useai session started \u2014 ${taskType} on ${client} \xB7 ${newSessionId.slice(0, 8)} \xB7 conv ${convId.slice(0, 8)} \xB7 recovered \xB7 ${daemonSigningKey ? "signed" : "unsigned"}`
+  );
+  console.log(`Recovered useai_start: new session ${newSessionId.slice(0, 8)} (MCP ${staleMcpSessionId.slice(0, 8)})`);
+  return true;
+}
+function recoverHeartbeat(staleMcpSessionId, rpcId, res) {
+  const map = readMcpMap();
+  const useaiSessionId = map[staleMcpSessionId];
+  if (!useaiSessionId) return false;
+  const chainPath = join10(ACTIVE_DIR, `${useaiSessionId}.jsonl`);
+  if (!existsSync11(chainPath)) {
+    sendJsonRpcResult(res, rpcId, "Session already ended (recovered).");
+    return true;
+  }
   try {
-    const rpc = body;
-    if (rpc?.method !== "tools/call" || rpc?.params?.name !== "useai_end") return false;
-    const args = rpc.params?.arguments ?? {};
-    const rpcId = rpc.id;
-    const map = readMcpMap();
-    const useaiSessionId = map[staleMcpSessionId];
-    if (!useaiSessionId) return false;
-    const chainPath = join10(ACTIVE_DIR, `${useaiSessionId}.jsonl`);
-    if (!existsSync11(chainPath)) return false;
     const content = readFileSync5(chainPath, "utf-8").trim();
-    if (!content) return false;
     const lines = content.split("\n").filter(Boolean);
     if (lines.length === 0) return false;
     const firstRecord = JSON.parse(lines[0]);
     const lastRecord = JSON.parse(lines[lines.length - 1]);
-    if (lastRecord.type === "session_end" || lastRecord.type === "session_seal") return false;
-    const startData = firstRecord.data;
-    const client = startData["client"] ?? "unknown";
-    const startTime = firstRecord.timestamp;
-    const sessionTitle = startData["title"] ?? void 0;
-    const sessionPrivateTitle = startData["private_title"] ?? void 0;
-    const sessionProject = startData["project"] ?? void 0;
-    const convId = startData["conversation_id"] ?? void 0;
-    const convIdx = startData["conversation_index"] ?? void 0;
     let heartbeatCount = 0;
     for (const line of lines) {
       try {
@@ -37376,34 +37514,58 @@ function tryRecoverEndSession(staleMcpSessionId, body, res) {
       } catch {
       }
     }
-    let chainTip = lastRecord.hash;
-    const duration3 = Math.round((Date.now() - new Date(startTime).getTime()) / 1e3);
-    const now = (/* @__PURE__ */ new Date()).toISOString();
-    let recordCount = lines.length;
-    const taskType = args["task_type"] ?? startData["task_type"] ?? "coding";
-    const languages = args["languages"] ?? [];
-    const filesTouched = args["files_touched_count"] ?? 0;
-    const milestonesInput = args["milestones"];
-    const evaluation = args["evaluation"];
-    let milestoneCount = 0;
+    heartbeatCount++;
+    const duration3 = Math.round((Date.now() - new Date(firstRecord.timestamp).getTime()) / 1e3);
+    const record2 = buildChainRecord("heartbeat", useaiSessionId, {
+      heartbeat_number: heartbeatCount,
+      cumulative_seconds: duration3,
+      recovered: true
+    }, lastRecord.hash, daemonSigningKey);
+    appendFileSync2(chainPath, JSON.stringify(record2) + "\n");
+    sendJsonRpcResult(res, rpcId, `Heartbeat recorded. Session active for ${formatDuration(duration3)}.`);
+    return true;
+  } catch {
+    return false;
+  }
+}
+function recoverEndSession(staleMcpSessionId, args, rpcId, res) {
+  const map = readMcpMap();
+  const useaiSessionId = map[staleMcpSessionId];
+  if (!useaiSessionId) return false;
+  const activePath = join10(ACTIVE_DIR, `${useaiSessionId}.jsonl`);
+  const sealedPath = join10(SEALED_DIR, `${useaiSessionId}.jsonl`);
+  const chainPath = existsSync11(activePath) ? activePath : existsSync11(sealedPath) ? sealedPath : null;
+  if (!chainPath) return false;
+  const isAlreadySealed = chainPath === sealedPath;
+  const content = readFileSync5(chainPath, "utf-8").trim();
+  if (!content) return false;
+  const lines = content.split("\n").filter(Boolean);
+  if (lines.length === 0) return false;
+  const firstRecord = JSON.parse(lines[0]);
+  const startData = firstRecord.data;
+  const client = startData["client"] ?? "unknown";
+  const startTime = firstRecord.timestamp;
+  const sessionTitle = startData["title"] ?? void 0;
+  const sessionPrivateTitle = startData["private_title"] ?? void 0;
+  const sessionProject = startData["project"] ?? void 0;
+  const convId = startData["conversation_id"] ?? void 0;
+  const convIdx = startData["conversation_index"] ?? void 0;
+  const taskType = args["task_type"] ?? startData["task_type"] ?? "coding";
+  const languages = args["languages"] ?? [];
+  const filesTouched = args["files_touched_count"] ?? 0;
+  const milestonesInput = args["milestones"];
+  const evaluation = args["evaluation"];
+  const now = (/* @__PURE__ */ new Date()).toISOString();
+  const duration3 = Math.round((Date.now() - new Date(startTime).getTime()) / 1e3);
+  if (isAlreadySealed) {
+    let milestoneCount2 = 0;
     if (milestonesInput && milestonesInput.length > 0) {
       const config2 = readJson(CONFIG_FILE, { milestone_tracking: true, auto_sync: true });
       if (config2.milestone_tracking) {
         const durationMinutes = Math.round(duration3 / 60);
         const allMilestones = readJson(MILESTONES_FILE, []);
         for (const m of milestonesInput) {
-          const mRecord = buildChainRecord("milestone", useaiSessionId, {
-            title: m.title,
-            private_title: m.private_title,
-            category: m.category,
-            complexity: m.complexity ?? "medium",
-            duration_minutes: durationMinutes,
-            languages
-          }, chainTip, daemonSigningKey);
-          appendFileSync2(chainPath, JSON.stringify(mRecord) + "\n");
-          chainTip = mRecord.hash;
-          recordCount++;
-          const milestone = {
+          allMilestones.push({
             id: `m_${randomUUID4().slice(0, 8)}`,
             session_id: useaiSessionId,
             title: m.title,
@@ -37417,26 +37579,14 @@ function tryRecoverEndSession(staleMcpSessionId, body, res) {
             created_at: now,
             published: false,
             published_at: null,
-            chain_hash: mRecord.hash
-          };
-          allMilestones.push(milestone);
-          milestoneCount++;
+            chain_hash: ""
+          });
+          milestoneCount2++;
         }
         writeJson(MILESTONES_FILE, allMilestones);
       }
     }
-    const endRecord = buildChainRecord("session_end", useaiSessionId, {
-      duration_seconds: duration3,
-      task_type: taskType,
-      languages,
-      files_touched: filesTouched,
-      heartbeat_count: heartbeatCount,
-      recovered: true,
-      ...evaluation ? { evaluation } : {}
-    }, chainTip, daemonSigningKey);
-    appendFileSync2(chainPath, JSON.stringify(endRecord) + "\n");
-    recordCount++;
-    const sealData = JSON.stringify({
+    const richSeal = {
       session_id: useaiSessionId,
       conversation_id: convId,
       conversation_index: convIdx,
@@ -37451,60 +37601,172 @@ function tryRecoverEndSession(staleMcpSessionId, body, res) {
       started_at: startTime,
       ended_at: now,
       duration_seconds: duration3,
-      heartbeat_count: heartbeatCount,
-      record_count: recordCount + 1,
-      // +1 for the seal record itself
-      chain_end_hash: endRecord.hash
-    });
-    const sealSignature = signHash(
-      createHash4("sha256").update(sealData).digest("hex"),
-      daemonSigningKey
+      heartbeat_count: 0,
+      record_count: lines.length,
+      chain_start_hash: firstRecord.prev_hash,
+      chain_end_hash: "",
+      seal_signature: ""
+    };
+    upsertSessionSeal(richSeal);
+    removeMcpMapping(staleMcpSessionId);
+    const durationStr2 = formatDuration(duration3);
+    sendJsonRpcResult(
+      res,
+      rpcId,
+      `Session ended (recovered): ${durationStr2} ${taskType}` + (milestoneCount2 > 0 ? ` \xB7 ${milestoneCount2} milestone${milestoneCount2 > 1 ? "s" : ""} recorded` : "") + (evaluation ? ` \xB7 eval: ${evaluation.task_outcome}` : "")
     );
-    const sealRecord = buildChainRecord("session_seal", useaiSessionId, {
+    console.log(`Recovered useai_end for already-sealed session ${useaiSessionId.slice(0, 8)} (MCP ${staleMcpSessionId.slice(0, 8)})`);
+    return true;
+  }
+  const lastRecord = JSON.parse(lines[lines.length - 1]);
+  if (lastRecord.type === "session_end" || lastRecord.type === "session_seal") {
+    removeMcpMapping(staleMcpSessionId);
+    sendJsonRpcResult(res, rpcId, "Session already ended.");
+    return true;
+  }
+  let heartbeatCount = 0;
+  for (const line of lines) {
+    try {
+      if (JSON.parse(line).type === "heartbeat") heartbeatCount++;
+    } catch {
+    }
+  }
+  let chainTip = lastRecord.hash;
+  let recordCount = lines.length;
+  let milestoneCount = 0;
+  if (milestonesInput && milestonesInput.length > 0) {
+    const config2 = readJson(CONFIG_FILE, { milestone_tracking: true, auto_sync: true });
+    if (config2.milestone_tracking) {
+      const durationMinutes = Math.round(duration3 / 60);
+      const allMilestones = readJson(MILESTONES_FILE, []);
+      for (const m of milestonesInput) {
+        const mRecord = buildChainRecord("milestone", useaiSessionId, {
+          title: m.title,
+          private_title: m.private_title,
+          category: m.category,
+          complexity: m.complexity ?? "medium",
+          duration_minutes: durationMinutes,
+          languages
+        }, chainTip, daemonSigningKey);
+        appendFileSync2(activePath, JSON.stringify(mRecord) + "\n");
+        chainTip = mRecord.hash;
+        recordCount++;
+        allMilestones.push({
+          id: `m_${randomUUID4().slice(0, 8)}`,
+          session_id: useaiSessionId,
+          title: m.title,
+          private_title: m.private_title,
+          project: sessionProject,
+          category: m.category,
+          complexity: m.complexity ?? "medium",
+          duration_minutes: durationMinutes,
+          languages,
+          client,
+          created_at: now,
+          published: false,
+          published_at: null,
+          chain_hash: mRecord.hash
+        });
+        milestoneCount++;
+      }
+      writeJson(MILESTONES_FILE, allMilestones);
+    }
+  }
+  const endRecord = buildChainRecord("session_end", useaiSessionId, {
+    duration_seconds: duration3,
+    task_type: taskType,
+    languages,
+    files_touched: filesTouched,
+    heartbeat_count: heartbeatCount,
+    recovered: true,
+    ...evaluation ? { evaluation } : {}
+  }, chainTip, daemonSigningKey);
+  appendFileSync2(activePath, JSON.stringify(endRecord) + "\n");
+  recordCount++;
+  const sealData = JSON.stringify({
+    session_id: useaiSessionId,
+    conversation_id: convId,
+    conversation_index: convIdx,
+    client,
+    task_type: taskType,
+    languages,
+    files_touched: filesTouched,
+    project: sessionProject,
+    title: sessionTitle,
+    private_title: sessionPrivateTitle,
+    evaluation: evaluation ?? void 0,
+    started_at: startTime,
+    ended_at: now,
+    duration_seconds: duration3,
+    heartbeat_count: heartbeatCount,
+    record_count: recordCount + 1,
+    chain_end_hash: endRecord.hash
+  });
+  const sealSignature = signHash(
+    createHash4("sha256").update(sealData).digest("hex"),
+    daemonSigningKey
+  );
+  appendFileSync2(activePath, JSON.stringify(
+    buildChainRecord("session_seal", useaiSessionId, {
       seal: sealData,
       seal_signature: sealSignature,
       recovered: true
-    }, endRecord.hash, daemonSigningKey);
-    appendFileSync2(chainPath, JSON.stringify(sealRecord) + "\n");
-    try {
-      renameSync3(chainPath, join10(SEALED_DIR, `${useaiSessionId}.jsonl`));
-    } catch {
+    }, endRecord.hash, daemonSigningKey)
+  ) + "\n");
+  try {
+    renameSync3(activePath, sealedPath);
+  } catch {
+  }
+  upsertSessionSeal({
+    session_id: useaiSessionId,
+    conversation_id: convId,
+    conversation_index: convIdx,
+    client,
+    task_type: taskType,
+    languages,
+    files_touched: filesTouched,
+    project: sessionProject,
+    title: sessionTitle,
+    private_title: sessionPrivateTitle,
+    evaluation: evaluation ?? void 0,
+    started_at: startTime,
+    ended_at: now,
+    duration_seconds: duration3,
+    heartbeat_count: heartbeatCount,
+    record_count: recordCount + 1,
+    chain_start_hash: firstRecord.prev_hash,
+    chain_end_hash: endRecord.hash,
+    seal_signature: sealSignature
+  });
+  removeMcpMapping(staleMcpSessionId);
+  const durationStr = formatDuration(duration3);
+  sendJsonRpcResult(
+    res,
+    rpcId,
+    `Session ended (recovered): ${durationStr} ${taskType}` + (milestoneCount > 0 ? ` \xB7 ${milestoneCount} milestone${milestoneCount > 1 ? "s" : ""} recorded` : "") + (evaluation ? ` \xB7 eval: ${evaluation.task_outcome}` : "")
+  );
+  console.log(`Recovered useai_end for stale session ${useaiSessionId.slice(0, 8)} (MCP ${staleMcpSessionId.slice(0, 8)})`);
+  return true;
+}
+function tryRecoverStaleSession(staleMcpSessionId, body, res) {
+  try {
+    const rpc = body;
+    if (rpc?.method !== "tools/call") return false;
+    const toolName = rpc.params?.name;
+    const args = rpc.params?.arguments ?? {};
+    const rpcId = rpc.id;
+    switch (toolName) {
+      case "useai_start":
+        return recoverStartSession(staleMcpSessionId, args, rpcId, res);
+      case "useai_heartbeat":
+        return recoverHeartbeat(staleMcpSessionId, rpcId, res);
+      case "useai_end":
+        return recoverEndSession(staleMcpSessionId, args, rpcId, res);
+      default:
+        return false;
     }
-    const seal = {
-      session_id: useaiSessionId,
-      conversation_id: convId,
-      conversation_index: convIdx,
-      client,
-      task_type: taskType,
-      languages,
-      files_touched: filesTouched,
-      project: sessionProject,
-      title: sessionTitle,
-      private_title: sessionPrivateTitle,
-      evaluation: evaluation ?? void 0,
-      started_at: startTime,
-      ended_at: now,
-      duration_seconds: duration3,
-      heartbeat_count: heartbeatCount,
-      record_count: recordCount + 1,
-      chain_start_hash: firstRecord.prev_hash,
-      chain_end_hash: endRecord.hash,
-      seal_signature: sealSignature
-    };
-    upsertSessionSeal(seal);
-    removeMcpMapping(staleMcpSessionId);
-    const durationMin = Math.round(duration3 / 60);
-    const responseText = `Session ended (recovered): ${durationMin}m ${taskType}` + (milestoneCount > 0 ? ` \xB7 ${milestoneCount} milestone${milestoneCount > 1 ? "s" : ""} recorded` : "") + (evaluation ? ` \xB7 eval: ${evaluation.task_outcome}` : "");
-    res.writeHead(200, { "Content-Type": "application/json" });
-    res.end(JSON.stringify({
-      jsonrpc: "2.0",
-      result: { content: [{ type: "text", text: responseText }] },
-      id: rpcId
-    }));
-    console.log(`Recovered useai_end for stale session ${useaiSessionId.slice(0, 8)} (MCP ${staleMcpSessionId.slice(0, 8)})`);
-    return true;
   } catch (err) {
-    console.error("Recovery failed:", err.message);
+    console.error("Stale session recovery failed:", err.message);
     return false;
   }
 }
@@ -37674,7 +37936,7 @@ async function startDaemon(port) {
           resetIdleTimer(sid);
           await sessions.get(sid).transport.handleRequest(req, res, body);
         } else if (sid && !sessions.has(sid)) {
-          if (!tryRecoverEndSession(sid, body, res)) {
+          if (!tryRecoverStaleSession(sid, body, res)) {
             res.writeHead(404, { "Content-Type": "application/json" });
             res.end(JSON.stringify({
               jsonrpc: "2.0",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devness/useai",
-  "version": "0.5.26",
+  "version": "0.5.28",
   "description": "Track your AI-assisted development workflow. MCP server that records usage metrics across all your AI tools.",
   "keywords": [
     "mcp",