context-mode 1.0.74 → 1.0.76

package/build/server.js

@@ -1602,38 +1602,51 @@ server.registerTool("ctx_doctor", {
     else {
         lines.push("- [-] Performance: NORMAL — install Bun for 3-5x speed boost");
     }
-    // Server test
-    try {
+    // Server test — cleanup executor to prevent resource leaks (#247)
+    {
         const testExecutor = new PolyglotExecutor({ runtimes });
-        const result = await testExecutor.execute({ language: "javascript", code: 'console.log("ok");', timeout: 5000 });
-        if (result.exitCode === 0 && result.stdout.trim() === "ok") {
-            lines.push("- [x] Server test: PASS");
+        try {
+            const result = await testExecutor.execute({ language: "javascript", code: 'console.log("ok");', timeout: 5000 });
+            if (result.exitCode === 0 && result.stdout.trim() === "ok") {
+                lines.push("- [x] Server test: PASS");
+            }
+            else {
+                const detail = result.stderr?.trim() ? ` (${result.stderr.trim().slice(0, 200)})` : "";
+                lines.push(`- [ ] Server test: FAIL — exit ${result.exitCode}${detail}`);
+            }
         }
-        else {
-            const detail = result.stderr?.trim() ? ` (${result.stderr.trim().slice(0, 200)})` : "";
-            lines.push(`- [ ] Server test: FAIL — exit ${result.exitCode}${detail}`);
+        catch (err) {
+            lines.push(`- [ ] Server test: FAIL — ${err instanceof Error ? err.message : err}`);
+        }
+        finally {
+            testExecutor.cleanupBackgrounded();
         }
     }
-    catch (err) {
-        lines.push(`- [ ] Server test: FAIL — ${err instanceof Error ? err.message : err}`);
-    }
-    // FTS5 / SQLite
-    try {
-        const Database = loadDatabase();
-        const db = new Database(":memory:");
-        db.exec("CREATE VIRTUAL TABLE fts_test USING fts5(content)");
-        db.exec("INSERT INTO fts_test(content) VALUES ('hello world')");
-        const row = db.prepare("SELECT * FROM fts_test WHERE fts_test MATCH 'hello'").get();
-        db.close();
-        if (row && row.content === "hello world") {
-            lines.push("- [x] FTS5 / SQLite: PASS — native module works");
+    // FTS5 / SQLite — close in finally to prevent GC segfault (#247)
+    {
+        let testDb;
+        try {
+            const Database = loadDatabase();
+            testDb = new Database(":memory:");
+            testDb.exec("CREATE VIRTUAL TABLE fts_test USING fts5(content)");
+            testDb.exec("INSERT INTO fts_test(content) VALUES ('hello world')");
+            const row = testDb.prepare("SELECT * FROM fts_test WHERE fts_test MATCH 'hello'").get();
+            if (row && row.content === "hello world") {
+                lines.push("- [x] FTS5 / SQLite: PASS — native module works");
+            }
+            else {
+                lines.push("- [ ] FTS5 / SQLite: FAIL — unexpected result");
+            }
         }
-        else {
-            lines.push("- [ ] FTS5 / SQLite: FAIL — unexpected result");
+        catch (err) {
+            lines.push(`- [ ] FTS5 / SQLite: FAIL — ${err instanceof Error ? err.message : err}`);
+        }
+        finally {
+            try {
+                testDb?.close();
+            }
+            catch { /* best effort */ }
         }
-    }
-    catch (err) {
-        lines.push(`- [ ] FTS5 / SQLite: FAIL — ${err instanceof Error ? err.message : err}`);
     }
     // Hook script
     const hookPath = resolve(pluginRoot, "hooks", "pretooluse.mjs");
@@ -1849,6 +1862,8 @@ async function main() {
     if (cleaned > 0) {
         console.error(`Cleaned up ${cleaned} stale DB file(s) from previous sessions`);
     }
+    // MCP readiness sentinel path (#230)
+    const mcpSentinel = join(tmpdir(), `context-mode-mcp-ready-${process.ppid}`);
     // Clean up own DB + backgrounded processes + preload script on shutdown
     const shutdown = () => {
         executor.cleanupBackgrounded();
@@ -1858,6 +1873,11 @@ async function main() {
             unlinkSync(CM_FS_PRELOAD);
         }
         catch { /* best effort */ }
+        // Remove MCP readiness sentinel (#230)
+        try {
+            unlinkSync(mcpSentinel);
+        }
+        catch { /* best effort */ }
     };
     const gracefulShutdown = async () => {
         shutdown();
@@ -1870,6 +1890,11 @@ async function main() {
     startLifecycleGuard({ onShutdown: () => gracefulShutdown() });
     const transport = new StdioServerTransport();
     await server.connect(transport);
+    // Write MCP readiness sentinel (#230)
+    try {
+        writeFileSync(mcpSentinel, String(process.pid));
+    }
+    catch { /* best effort */ }
     // Detect platform adapter — stored for platform-aware session paths
     try {
         const { detectPlatform, getAdapter } = await import("./adapters/detect.js");

package/build/session/db.js

@@ -225,7 +225,7 @@ export class SessionDB extends SQLiteBase {
             // Update meta if session exists
             this.stmt(S.updateMetaLastEvent).run(sessionId);
         });
-        transaction();
+        this.withRetry(() => transaction());
     }
     /**
      * Retrieve events for a session with optional filtering.

package/build/store.js

@@ -7,7 +7,7 @@
  * Use for documentation, API references, and any content where
  * you need EXACT text later — not summaries.
  */
-import { loadDatabase, applyWALPragmas, closeDB, withRetry } from "./db-base.js";
+import { loadDatabase, applyWALPragmas, closeDB, cleanOrphanedWALFiles, withRetry, deleteDBFiles, isSQLiteCorruptionError } from "./db-base.js";
 import { readFileSync, readdirSync, unlinkSync, existsSync, statSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
@@ -267,8 +267,30 @@ export class ContentStore {
         const Database = loadDatabase();
         this.#dbPath =
             dbPath ?? join(tmpdir(), `context-mode-${process.pid}.db`);
-        this.#db = new Database(this.#dbPath, { timeout: 30000 });
-        applyWALPragmas(this.#db);
+        cleanOrphanedWALFiles(this.#dbPath);
+        let db;
+        try {
+            db = new Database(this.#dbPath, { timeout: 30000 });
+            applyWALPragmas(db);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            if (isSQLiteCorruptionError(msg)) {
+                deleteDBFiles(this.#dbPath);
+                cleanOrphanedWALFiles(this.#dbPath);
+                try {
+                    db = new Database(this.#dbPath, { timeout: 30000 });
+                    applyWALPragmas(db);
+                }
+                catch (retryErr) {
+                    throw new Error(`Failed to create fresh DB after deleting corrupt file: ${retryErr instanceof Error ? retryErr.message : String(retryErr)}`);
+                }
+            }
+            else {
+                throw err;
+            }
+        }
+        this.#db = db;
         this.#initSchema();
         this.#prepareStatements();
     }