context-vault 2.3.0 → 2.4.0

package/scripts/postinstall.js

@@ -0,0 +1,45 @@
+#!/usr/bin/env node
+
+/**
+ * postinstall.js — Auto-rebuild native modules on install
+ *
+ * Detects NODE_MODULE_VERSION mismatches and attempts a rebuild.
+ */
+
+import { execSync } from "node:child_process";
+
+async function main() {
+  let needsRebuild = false;
+
+  try {
+    await import("better-sqlite3");
+  } catch (e) {
+    if (e.message?.includes("NODE_MODULE_VERSION")) {
+      needsRebuild = true;
+    }
+  }
+
+  try {
+    await import("sqlite-vec");
+  } catch (e) {
+    if (e.message?.includes("NODE_MODULE_VERSION")) {
+      needsRebuild = true;
+    }
+  }
+
+  if (needsRebuild) {
+    console.log("[context-vault] Rebuilding native modules for Node.js " + process.version + "...");
+    try {
+      execSync("npm rebuild better-sqlite3 sqlite-vec", {
+        stdio: "inherit",
+        timeout: 60000,
+      });
+      console.log("[context-vault] Native modules rebuilt successfully.");
+    } catch {
+      console.error("[context-vault] Warning: native module rebuild failed.");
+      console.error("[context-vault] Try manually: npm rebuild better-sqlite3 sqlite-vec");
+    }
+  }
+}
+
+main().catch(() => {});

package/scripts/prepack.js

@@ -0,0 +1,31 @@
+#!/usr/bin/env node
+
+/**
+ * prepack.js — Cross-platform bundle preparation
+ *
+ * Copies @context-vault/core into node_modules for npm pack bundling.
+ * Replaces the Unix shell script in package.json "prepack".
+ */
+
+import { cpSync, rmSync, mkdirSync } from "node:fs";
+import { join, dirname } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const LOCAL_ROOT = join(__dirname, "..");
+const CORE_SRC = join(LOCAL_ROOT, "..", "core");
+const CORE_DEST = join(LOCAL_ROOT, "node_modules", "@context-vault", "core");
+
+// Ensure target directory exists
+mkdirSync(join(LOCAL_ROOT, "node_modules", "@context-vault"), { recursive: true });
+
+// Remove old copy if present
+rmSync(CORE_DEST, { recursive: true, force: true });
+
+// Copy core package (dereference symlinks)
+cpSync(CORE_SRC, CORE_DEST, { recursive: true, dereference: true });
+
+// Remove nested node_modules from the copy
+rmSync(join(CORE_DEST, "node_modules"), { recursive: true, force: true });
+
+console.log("[prepack] Bundled @context-vault/core into node_modules");

package/src/server/index.js

@@ -11,89 +11,131 @@ const pkg = JSON.parse(readFileSync(join(__dirname, "..", "..", "package.json"),
 
 import { resolveConfig } from "@context-vault/core/core/config";
 import { embed } from "@context-vault/core/index/embed";
-import { initDatabase, prepareStatements, insertVec, deleteVec } from "@context-vault/core/index/db";
+import { initDatabase, NativeModuleError, prepareStatements, insertVec, deleteVec } from "@context-vault/core/index/db";
 import { registerTools } from "@context-vault/core/server/tools";
 
-// ─── Config Resolution ──────────────────────────────────────────────────────
+// ─── Phased Startup ─────────────────────────────────────────────────────────
 
-const config = resolveConfig();
+async function main() {
+  let phase = "CONFIG";
+  let db;
 
-// Create directories
-mkdirSync(config.dataDir, { recursive: true });
-mkdirSync(config.vaultDir, { recursive: true });
+  try {
+    // ── Phase: CONFIG ────────────────────────────────────────────────────────
+    const config = resolveConfig();
 
-// Write .context-mcp marker (always update to reflect current version)
-const markerPath = join(config.vaultDir, ".context-mcp");
-const markerData = existsSync(markerPath) ? JSON.parse(readFileSync(markerPath, "utf-8")) : {};
-writeFileSync(markerPath, JSON.stringify({ created: markerData.created || new Date().toISOString(), version: pkg.version }, null, 2) + "\n");
+    // ── Phase: DIRS ──────────────────────────────────────────────────────────
+    phase = "DIRS";
+    mkdirSync(config.dataDir, { recursive: true });
+    mkdirSync(config.vaultDir, { recursive: true });
 
-// Update existence flag after directory creation
-config.vaultDirExists = existsSync(config.vaultDir);
-
-// Startup diagnostics
-console.error(`[context-mcp] Vault: ${config.vaultDir}`);
-console.error(`[context-mcp] Database: ${config.dbPath}`);
-console.error(`[context-mcp] Dev dir: ${config.devDir}`);
-if (!config.vaultDirExists) {
-  console.error(`[context-mcp] WARNING: Vault directory not found!`);
+    // Write .context-mcp marker (non-fatal)
+    try {
+      const markerPath = join(config.vaultDir, ".context-mcp");
+      const markerData = existsSync(markerPath) ? JSON.parse(readFileSync(markerPath, "utf-8")) : {};
+      writeFileSync(markerPath, JSON.stringify({ created: markerData.created || new Date().toISOString(), version: pkg.version }, null, 2) + "\n");
+    } catch (markerErr) {
+      console.error(`[context-mcp] Warning: could not write marker file: ${markerErr.message}`);
+    }
+
+    config.vaultDirExists = existsSync(config.vaultDir);
+
+    // Startup diagnostics
+    console.error(`[context-mcp] Vault: ${config.vaultDir}`);
+    console.error(`[context-mcp] Database: ${config.dbPath}`);
+    console.error(`[context-mcp] Dev dir: ${config.devDir}`);
+    if (!config.vaultDirExists) {
+      console.error(`[context-mcp] WARNING: Vault directory not found!`);
+    }
+
+    // ── Phase: DB ────────────────────────────────────────────────────────────
+    phase = "DB";
+    db = await initDatabase(config.dbPath);
+    const stmts = prepareStatements(db);
+
+    const ctx = {
+      db,
+      config,
+      stmts,
+      embed,
+      insertVec: (rowid, embedding) => insertVec(stmts, rowid, embedding),
+      deleteVec: (rowid) => deleteVec(stmts, rowid),
+    };
+
+    // ── Phase: SERVER ────────────────────────────────────────────────────────
+    phase = "SERVER";
+    const server = new McpServer(
+      { name: "context-mcp", version: pkg.version },
+      { capabilities: { tools: {} } }
+    );
+
+    registerTools(server, ctx);
+
+    // ── Graceful Shutdown ────────────────────────────────────────────────────
+    function shutdown(signal) {
+      console.error(`[context-mcp] Received ${signal}, shutting down...`);
+      try {
+        if (db.inTransaction) {
+          console.error("[context-mcp] Rolling back active transaction...");
+          db.exec("ROLLBACK");
+        }
+        db.pragma("wal_checkpoint(TRUNCATE)");
+        db.close();
+        console.error("[context-mcp] Database closed cleanly.");
+      } catch (shutdownErr) {
+        console.error(`[context-mcp] Shutdown error: ${shutdownErr.message}`);
+      }
+      process.exit(0);
+    }
+    process.on("SIGINT", () => shutdown("SIGINT"));
+    process.on("SIGTERM", () => shutdown("SIGTERM"));
+
+    // ── Phase: CONNECTED ─────────────────────────────────────────────────────
+    phase = "CONNECTED";
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+
+    // ── Non-blocking Update Check ────────────────────────────────────────────
+    setTimeout(() => {
+      import("node:child_process").then(({ execSync }) => {
+        try {
+          const latest = execSync("npm view context-vault version", {
+            encoding: "utf-8",
+            timeout: 5000,
+            stdio: ["pipe", "pipe", "pipe"],
+          }).trim();
+          if (latest && latest !== pkg.version) {
+            console.error(`[context-mcp] Update available: v${pkg.version} → v${latest}. Run: context-mcp update`);
+          }
+        } catch {}
+      }).catch(() => {});
+    }, 3000);
+
+  } catch (err) {
+    if (err instanceof NativeModuleError) {
+      // Boxed diagnostic for native module mismatch
+      console.error("");
+      console.error("╔══════════════════════════════════════════════════════════════╗");
+      console.error("║  context-mcp: Native Module Error                           ║");
+      console.error("╚══════════════════════════════════════════════════════════════╝");
+      console.error("");
+      console.error(err.message);
+      console.error("");
+      console.error(`  Node.js path:    ${process.execPath}`);
+      console.error(`  Node.js version: ${process.version}`);
+      console.error("");
+      process.exit(78); // EX_CONFIG
+    }
+
+    console.error(`[context-mcp] Fatal error during ${phase} phase: ${err.message}`);
+    if (phase === "DB") {
+      console.error(`[context-mcp] Try deleting the DB file and restarting: rm "${err.dbPath || "vault.db"}"`);
+    }
+    process.exit(1);
+  }
 }
 
-// ─── Database Init ───────────────────────────────────────────────────────────
-
-let db, stmts;
-try {
-  db = initDatabase(config.dbPath);
-  stmts = prepareStatements(db);
-} catch (e) {
-  console.error(`[context-mcp] Database init failed: ${e.message}`);
-  console.error(`[context-mcp] DB path: ${config.dbPath}`);
-  console.error(`[context-mcp] Try deleting the DB file and restarting: rm "${config.dbPath}"`);
+main().catch((err) => {
+  console.error(`[context-mcp] Unexpected fatal error: ${err.message}`);
   process.exit(1);
-}
-
-const ctx = {
-  db,
-  config,
-  stmts,
-  embed,
-  insertVec: (rowid, embedding) => insertVec(stmts, rowid, embedding),
-  deleteVec: (rowid) => deleteVec(stmts, rowid),
-};
-
-// ─── MCP Server ──────────────────────────────────────────────────────────────
-
-const server = new McpServer(
-  { name: "context-mcp", version: pkg.version },
-  { capabilities: { tools: {} } }
-);
-
-registerTools(server, ctx);
-
-// ─── Graceful Shutdown ───────────────────────────────────────────────────────
-
-function shutdown() {
-  try { db.close(); } catch {}
-  process.exit(0);
-}
-process.on("SIGINT", shutdown);
-process.on("SIGTERM", shutdown);
-
-const transport = new StdioServerTransport();
-await server.connect(transport);
-
-// ─── Non-blocking Update Check ──────────────────────────────────────────────
-
-setTimeout(() => {
-  import("node:child_process").then(({ execSync }) => {
-    try {
-      const latest = execSync("npm view context-vault version", {
-        encoding: "utf-8",
-        timeout: 5000,
-        stdio: ["pipe", "pipe", "pipe"],
-      }).trim();
-      if (latest && latest !== pkg.version) {
-        console.error(`[context-mcp] Update available: v${pkg.version} → v${latest}. Run: context-mcp update`);
-      }
-    } catch {}
-  }).catch(() => {});
-}, 3000);
+});