@keepgoingdev/mcp-server 0.8.0 → 0.9.1

package/dist/index.js

@@ -339,7 +339,8 @@ function generateEnrichedBriefing(opts) {
       timestamp: s.timestamp,
       summary: s.summary || "",
       nextStep: s.nextStep || "",
-      branch: s.gitBranch
+      branch: s.gitBranch,
+      sessionPhase: s.sessionPhase
     }));
   }
   if (opts.recentCommits && opts.recentCommits.length > 0) {
@@ -372,10 +373,13 @@ function buildCurrentFocus(lastSession, projectState, gitBranch) {
 function buildRecentActivity(lastSession, recentSessions, recentCommitMessages) {
   const parts = [];
   const sessionCount = recentSessions.length;
+  const planCount = recentSessions.filter((s) => s.sessionPhase === "planning").length;
   if (sessionCount > 1) {
-    parts.push(`${sessionCount} recent sessions`);
+    const planSuffix = planCount > 0 ? ` (${planCount} plan-only)` : "";
+    parts.push(`${sessionCount} recent sessions${planSuffix}`);
   } else if (sessionCount === 1) {
-    parts.push("1 recent session");
+    const planSuffix = planCount === 1 ? " (plan-only)" : "";
+    parts.push(`1 recent session${planSuffix}`);
   }
   if (lastSession.summary) {
     const brief = lastSession.summary.length > 120 ? lastSession.summary.slice(0, 117) + "..." : lastSession.summary;
@@ -574,7 +578,8 @@ function formatEnrichedBriefing(briefing) {
     for (const s of briefing.sessionHistory) {
       const relTime = formatRelativeTime(s.timestamp);
       const branch = s.branch ? ` (${s.branch})` : "";
-      lines.push(`- **${relTime}${branch}:** ${s.summary || "No summary"}. Next: ${s.nextStep || "Not specified"}`);
+      const planTag = s.sessionPhase === "planning" ? " (plan)" : "";
+      lines.push(`- **${relTime}${branch}${planTag}:** ${s.summary || "No summary"}. Next: ${s.nextStep || "Not specified"}`);
     }
   }
   if (briefing.recentCommits && briefing.recentCommits.length > 0) {
@@ -2099,6 +2104,15 @@ var POST_TOOL_USE_HOOK = {
     }
   ]
 };
+var PLAN_MODE_HOOK = {
+  matcher: "Read|Grep|Glob|Bash|WebSearch",
+  hooks: [
+    {
+      type: "command",
+      command: "npx -y @keepgoingdev/mcp-server --heartbeat"
+    }
+  ]
+};
 var SESSION_END_HOOK = {
   matcher: "",
   hooks: [
@@ -2108,7 +2122,7 @@ var SESSION_END_HOOK = {
     }
   ]
 };
-var KEEPGOING_RULES_VERSION = 2;
+var KEEPGOING_RULES_VERSION = 3;
 var KEEPGOING_RULES_CONTENT = `<!-- @keepgoingdev/mcp-server v${KEEPGOING_RULES_VERSION} -->
 ## KeepGoing
 
@@ -2118,6 +2132,8 @@ After completing a task or meaningful piece of work, call the \`save_checkpoint\
 - \`summary\`: 1-2 sentences. What changed and why, no file paths, no implementation details (those are captured from git).
 - \`nextStep\`: What to do next
 - \`blocker\`: Any blocker (if applicable)
+
+When working in plan mode (investigating, designing, iterating on an approach before any edits), call \`save_checkpoint\` when you reach a significant milestone or conclusion. Use the summary to capture what was investigated and decided. This preserves planning context for future sessions.
 `;
 function getRulesFileVersion(content) {
   const match = content.match(/<!-- @keepgoingdev\/mcp-server v(\d+) -->/);
@@ -2178,6 +2194,13 @@ function writeHooksToSettings(settings) {
     settings.hooks.PostToolUse.push(POST_TOOL_USE_HOOK);
     changed = true;
   }
+  const hasHeartbeat = settings.hooks.PostToolUse.some(
+    (entry) => entry?.hooks?.some((h) => typeof h?.command === "string" && h.command.includes("--heartbeat"))
+  );
+  if (!hasHeartbeat) {
+    settings.hooks.PostToolUse.push(PLAN_MODE_HOOK);
+    changed = true;
+  }
   if (!Array.isArray(settings.hooks.SessionEnd)) {
     settings.hooks.SessionEnd = [];
   }
@@ -2741,7 +2764,11 @@ function registerSaveCheckpoint(server, reader, workspacePath) {
       const touchedFiles = getTouchedFiles(workspacePath);
       const commitHashes = getCommitsSince(workspacePath, lastSession?.timestamp);
       const projectName = path10.basename(resolveStorageRoot(workspacePath));
+      const writer = new KeepGoingWriter(workspacePath);
       const sessionId = generateSessionId({ workspaceRoot: workspacePath, branch: gitBranch ?? void 0, worktreePath: workspacePath });
+      const existingTasks = writer.readCurrentTasks();
+      const existingSession = existingTasks.find((t) => t.sessionId === sessionId);
+      const sessionPhase = existingSession?.sessionPhase;
       const checkpoint = createCheckpoint({
         summary,
         nextStep: nextStep || "",
@@ -2751,10 +2778,19 @@ function registerSaveCheckpoint(server, reader, workspacePath) {
         commitHashes,
         workspaceRoot: workspacePath,
         source: "manual",
-        sessionId
+        sessionId,
+        ...sessionPhase ? { sessionPhase } : {},
+        ...sessionPhase === "planning" ? { tags: ["plan"] } : {}
       });
-      const writer = new KeepGoingWriter(workspacePath);
       writer.saveCheckpoint(checkpoint, projectName);
+      writer.upsertSession({
+        sessionId,
+        sessionActive: true,
+        branch: gitBranch ?? void 0,
+        updatedAt: checkpoint.timestamp,
+        taskSummary: summary,
+        nextStep: nextStep || void 0
+      });
       const lines = [
         `Checkpoint saved.`,
         `- **ID:** ${checkpoint.id}`,
@@ -2907,8 +2943,9 @@ function registerGetCurrentTask(server, reader) {
         lines.push("");
       }
       for (const task of [...activeTasks, ...finishedTasks]) {
+        const phaseLabel = task.sessionPhase === "planning" ? "Planning" : void 0;
         const statusIcon = task.sessionActive ? "\u{1F7E2}" : "\u2705";
-        const statusLabel = task.sessionActive ? "Active" : "Finished";
+        const statusLabel = task.sessionActive ? phaseLabel || "Active" : "Finished";
         const sessionLabel = task.sessionLabel || task.agentLabel || task.sessionId || "Session";
         lines.push(`### ${statusIcon} ${sessionLabel} (${statusLabel})`);
         lines.push(`- **Updated:** ${formatRelativeTime(task.updatedAt)}`);
@@ -3470,6 +3507,7 @@ import path12 from "path";
 async function handleSaveCheckpoint() {
   const wsPath = resolveWsPath();
   const reader = new KeepGoingReader(wsPath);
+  const writer = new KeepGoingWriter(wsPath);
   const { session: lastSession } = reader.getScopedLastSession();
   if (lastSession?.timestamp) {
     const ageMs = Date.now() - new Date(lastSession.timestamp).getTime();
@@ -3479,10 +3517,44 @@ async function handleSaveCheckpoint() {
   }
   const touchedFiles = getTouchedFiles(wsPath);
   const commitHashes = getCommitsSince(wsPath, lastSession?.timestamp);
-  if (touchedFiles.length === 0 && commitHashes.length === 0) {
+  const gitBranch = getCurrentBranch(wsPath);
+  const existingTasks = writer.readCurrentTasks();
+  const existingSession = existingTasks.find(
+    (t) => t.sessionActive && t.worktreePath === wsPath && t.branch === (gitBranch ?? void 0)
+  ) ?? existingTasks.find(
+    (t) => t.sessionActive && t.worktreePath === wsPath
+  );
+  const sessionId = existingSession?.sessionId ?? generateSessionId({ workspaceRoot: wsPath, branch: gitBranch ?? void 0, worktreePath: wsPath });
+  const isPlanning = existingSession?.sessionPhase === "planning";
+  if (touchedFiles.length === 0 && commitHashes.length === 0 && !isPlanning) {
+    process.exit(0);
+  }
+  const projectName = path12.basename(resolveStorageRoot(wsPath));
+  if (touchedFiles.length === 0 && commitHashes.length === 0 && isPlanning) {
+    const summary2 = existingSession?.sessionLabel || existingSession?.taskSummary || "Planning session";
+    const checkpoint2 = createCheckpoint({
+      summary: summary2,
+      nextStep: existingSession?.nextStep || "",
+      gitBranch,
+      touchedFiles: [],
+      commitHashes: [],
+      workspaceRoot: wsPath,
+      source: "auto",
+      sessionId,
+      sessionPhase: "planning",
+      tags: ["plan"]
+    });
+    writer.saveCheckpoint(checkpoint2, projectName);
+    writer.upsertSession({
+      sessionId,
+      sessionActive: false,
+      nextStep: checkpoint2.nextStep || void 0,
+      branch: gitBranch ?? void 0,
+      updatedAt: checkpoint2.timestamp
+    });
+    console.log(`[KeepGoing] Plan checkpoint saved: ${summary2}`);
     process.exit(0);
   }
-  const gitBranch = getCurrentBranch(wsPath);
   const commitMessages = getCommitMessagesSince(wsPath, lastSession?.timestamp);
   const now = (/* @__PURE__ */ new Date()).toISOString();
   const events = buildSessionEvents({
@@ -3496,8 +3568,7 @@ async function handleSaveCheckpoint() {
   });
   const summary = buildSmartSummary(events) ?? `Worked on ${touchedFiles.slice(0, 5).map((f) => path12.basename(f)).join(", ")}`;
   const nextStep = buildSmartNextStep(events);
-  const projectName = path12.basename(resolveStorageRoot(wsPath));
-  const sessionId = generateSessionId({ workspaceRoot: wsPath, branch: gitBranch ?? void 0, worktreePath: wsPath });
+  const sessionPhase = existingSession?.sessionPhase;
   const checkpoint = createCheckpoint({
     summary,
     nextStep,
@@ -3506,9 +3577,10 @@ async function handleSaveCheckpoint() {
     commitHashes,
     workspaceRoot: wsPath,
     source: "auto",
-    sessionId
+    sessionId,
+    ...sessionPhase ? { sessionPhase } : {},
+    ...sessionPhase === "planning" ? { tags: ["plan"] } : {}
   });
-  const writer = new KeepGoingWriter(wsPath);
   writer.saveCheckpoint(checkpoint, projectName);
   writer.upsertSession({
     sessionId,
@@ -3776,7 +3848,8 @@ async function handleUpdateTaskFromHook() {
         branch,
         worktreePath: wsPath,
         sessionActive: true,
-        updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+        updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+        sessionPhase: "active"
       };
       const sessionId = hookData.session_id || generateSessionId({ ...task, workspaceRoot: wsPath });
       task.sessionId = sessionId;
@@ -3923,6 +3996,59 @@ async function handleDetectDecisions() {
   process.exit(0);
 }
 
+// src/cli/heartbeat.ts
+var STDIN_TIMEOUT_MS3 = 3e3;
+var THROTTLE_MS = 3e4;
+async function handleHeartbeat() {
+  const wsPath = resolveWsPath();
+  const chunks = [];
+  const timeout = setTimeout(() => process.exit(0), STDIN_TIMEOUT_MS3);
+  process.stdin.on("error", () => {
+    clearTimeout(timeout);
+    process.exit(0);
+  });
+  process.stdin.on("data", (chunk) => chunks.push(chunk));
+  process.stdin.on("end", () => {
+    clearTimeout(timeout);
+    try {
+      const raw = Buffer.concat(chunks).toString("utf-8").trim();
+      if (!raw) {
+        process.exit(0);
+      }
+      const hookData = JSON.parse(raw);
+      const writer = new KeepGoingWriter(wsPath);
+      const existing = writer.readCurrentTasks();
+      const sessionIdFromHook = hookData.session_id;
+      const sessionId = sessionIdFromHook || generateSessionId({ workspaceRoot: wsPath, worktreePath: wsPath, branch: getCurrentBranch(wsPath) ?? void 0 });
+      const existingSession = existing.find((t) => t.sessionId === sessionId);
+      if (existingSession?.updatedAt) {
+        const ageMs = Date.now() - new Date(existingSession.updatedAt).getTime();
+        if (ageMs < THROTTLE_MS) {
+          process.exit(0);
+        }
+      }
+      const sessionPhase = existingSession?.sessionPhase === "active" ? "active" : "planning";
+      const branch = existingSession?.branch ?? getCurrentBranch(wsPath) ?? void 0;
+      const task = {
+        sessionId,
+        sessionActive: true,
+        updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+        sessionPhase,
+        worktreePath: wsPath,
+        branch
+      };
+      if (!existingSession?.sessionLabel && hookData.transcript_path) {
+        const label = extractSessionLabel(hookData.transcript_path);
+        if (label) task.sessionLabel = label;
+      }
+      writer.upsertSession(task);
+    } catch {
+    }
+    process.exit(0);
+  });
+  process.stdin.resume();
+}
+
 // src/index.ts
 var CLI_HANDLERS = {
   "--print-momentum": handlePrintMomentum,
@@ -3932,7 +4058,8 @@ var CLI_HANDLERS = {
   "--print-current": handlePrintCurrent,
   "--statusline": handleStatusline,
   "--continue-on": handleContinueOn,
-  "--detect-decisions": handleDetectDecisions
+  "--detect-decisions": handleDetectDecisions,
+  "--heartbeat": handleHeartbeat
 };
 var flag = process.argv.slice(2).find((a) => a in CLI_HANDLERS);
 if (flag) {