context-mode 1.0.104 → 1.0.105

package/build/session/db.js

@@ -87,6 +87,7 @@ const S = {
     upsertResume: "upsertResume",
     getResume: "getResume",
     markResumeConsumed: "markResumeConsumed",
+    claimLatestUnconsumedResume: "claimLatestUnconsumedResume",
     deleteEvents: "deleteEvents",
     deleteMeta: "deleteMeta",
     deleteResume: "deleteResume",
@@ -251,6 +252,19 @@ export class SessionDB extends SQLiteBase {
          consumed = 0`);
         p(S.getResume, `SELECT snapshot, event_count, consumed FROM session_resume WHERE session_id = ?`);
         p(S.markResumeConsumed, `UPDATE session_resume SET consumed = 1 WHERE session_id = ?`);
+        // Atomic "pick newest unconsumed snapshot AND mark it consumed in one
+        // statement". Required for race-safe cross-session resume injection
+        // (Mickey / PR #376) — two parallel chat-turn hooks must not both read
+        // the same row before either one writes consumed=1.
+        p(S.claimLatestUnconsumedResume, `UPDATE session_resume
+       SET consumed = 1
+       WHERE id = (
+         SELECT id FROM session_resume
+         WHERE consumed = 0
+         ORDER BY created_at DESC, id DESC
+         LIMIT 1
+       )
+       RETURNING session_id, snapshot`);
         // ── Delete ──
         p(S.deleteEvents, `DELETE FROM session_events WHERE session_id = ?`);
         p(S.deleteMeta, `DELETE FROM session_meta WHERE session_id = ?`);
@@ -478,6 +492,22 @@ export class SessionDB extends SQLiteBase {
     markResumeConsumed(sessionId) {
         this.stmt(S.markResumeConsumed).run(sessionId);
     }
+    /**
+     * Atomically claim the most recent unconsumed resume snapshot in this DB.
+     *
+     * `SessionDB` is sharded per project (see `getSessionDBPath` — SHA-256 of
+     * project dir), so "this DB" already implies "this project". The atomic
+     * `UPDATE … RETURNING` ensures concurrent processes for the same project
+     * cannot both inject the same snapshot (Mickey / PR #376 race).
+     *
+     * Returns null when no unconsumed snapshot exists.
+     */
+    claimLatestUnconsumedResume() {
+        const row = this.stmt(S.claimLatestUnconsumedResume).get();
+        if (!row)
+            return null;
+        return { sessionId: row.session_id, snapshot: row.snapshot };
+    }
     /**
      * Return the most recent session_id from session_meta, or null if none.
      * Used by the runtime to attach persistent counters to the right session