kongbrain 0.1.1 → 0.1.3

package/src/surreal.ts

@@ -482,6 +482,30 @@ export class SurrealStore {
     }
   }
 
+  async markSessionActive(sessionId: string): Promise<void> {
+    assertRecordId(sessionId);
+    await this.queryExec(
+      `UPDATE ${sessionId} SET cleanup_completed = false, last_active = time::now()`,
+    );
+  }
+
+  async markSessionEnded(sessionId: string): Promise<void> {
+    assertRecordId(sessionId);
+    await this.queryExec(
+      `UPDATE ${sessionId} SET ended_at = time::now(), cleanup_completed = true`,
+    );
+  }
+
+  async getOrphanedSessions(limit = 3): Promise<{ id: string; started_at: string }[]> {
+    return this.queryFirst<{ id: string; started_at: string }>(
+      `SELECT id, started_at FROM session
+       WHERE cleanup_completed != true
+         AND started_at < time::now() - 2m
+       ORDER BY started_at DESC LIMIT $lim`,
+      { lim: limit },
+    );
+  }
+
   async linkSessionToTask(sessionId: string, taskId: string): Promise<void> {
     await this.queryExec(`RELATE ${sessionId}->session_task->${taskId}`);
   }

package/src/wakeup.ts

@@ -13,6 +13,7 @@ import type { CompleteFn } from "./state.js";
 import type { SurrealStore } from "./surreal.js";
 import { hasSoul, getSoul, checkGraduation } from "./soul.js";
 import type { MaturityStage } from "./soul.js";
+import { readAndDeleteHandoffFile } from "./handoff-file.js";
 import { swallow } from "./errors.js";
 
 // --- Types ---
@@ -63,6 +64,7 @@ export async function synthesizeWakeup(
   store: SurrealStore,
   complete: CompleteFn,
   currentSessionId?: string,
+  workspaceDir?: string,
 ): Promise<string | null> {
   if (!store.isAvailable()) return null;
 
@@ -75,7 +77,10 @@ export async function synthesizeWakeup(
     hasSoul(store),
   ]);
 
-  if (!handoff && monologues.length === 0 && identityChunks.length === 0 && previousTurns.length === 0) return null;
+  // Check for sync handoff file (written on abrupt exit)
+  const handoffFile = workspaceDir ? readAndDeleteHandoffFile(workspaceDir) : null;
+
+  if (!handoff && !handoffFile && monologues.length === 0 && identityChunks.length === 0 && previousTurns.length === 0) return null;
 
   const sections: string[] = [];
 
@@ -98,6 +103,13 @@ export async function synthesizeWakeup(
     }
     annotation += ")";
     sections.push(`[LAST HANDOFF] ${annotation}\n${handoff.text}`);
+  } else if (handoffFile) {
+    // Fallback: sync handoff file from abrupt exit (no DB handoff note exists)
+    const lines: string[] = [];
+    lines.push(`Session ended abruptly (${handoffFile.userTurnCount} turns, ${handoffFile.unextractedTokens} unextracted tokens)`);
+    if (handoffFile.lastUserText) lines.push(`Last user message: ${handoffFile.lastUserText}`);
+    if (handoffFile.lastAssistantText) lines.push(`Last assistant message: ${handoffFile.lastAssistantText}`);
+    sections.push(`[LAST SESSION EXIT]\n${lines.join("\n")}`);
   }
 
   if (previousTurns.length > 0) {