@vheins/local-memory-mcp 0.7.4 → 0.8.0

package/dist/{chunk-BSASVWKJ.js → chunk-CQK2B6SF.js}

@@ -172,40 +172,28 @@ function addLogSink(sink) {
 var LOG_LEVEL_VALUES = Object.keys(LEVELS);
 
 // src/mcp/storage/sqlite.ts
-import initSqlJs from "sql.js";
+import Database from "better-sqlite3";
 import path2 from "path";
 import fs2 from "fs";
 import os from "os";
 
 // src/mcp/storage/migrations.ts
 var MigrationManager = class {
-  constructor(db, saveDb) {
+  constructor(db) {
     this.db = db;
-    this.saveDb = saveDb;
   }
   db;
-  saveDb;
   run(sql) {
-    this.db.run(sql);
+    this.db.prepare(sql).run();
   }
   exec(sql) {
     this.db.exec(sql);
   }
   all(sql) {
-    const result = this.db.exec(sql);
-    if (result.length === 0) return [];
-    const { columns, values } = result[0];
-    return values.map((row) => {
-      const obj = {};
-      columns.forEach((col, i) => {
-        obj[col] = row[i];
-      });
-      return obj;
-    });
+    return this.db.prepare(sql).all();
   }
   get(sql) {
-    const rows = this.all(sql);
-    return rows[0];
+    return this.db.prepare(sql).get();
   }
   migrate() {
     this.exec(`
@@ -427,7 +415,6 @@ var MigrationManager = class {
       this.run("UPDATE tasks SET task_code = substr(id, 1, 8) WHERE task_code IS NULL");
     } catch {
     }
-    if (this.saveDb) this.saveDb();
   }
   ensureMemoryTypeConstraint() {
     const tableSql = this.get("SELECT sql FROM sqlite_master WHERE type = 'table' AND name = 'memories'");
@@ -531,7 +518,7 @@ var MigrationManager = class {
   }
   ensureMemoryStatusConstraintRemoved() {
     const tableSql = this.get("SELECT sql FROM sqlite_master WHERE type = 'table' AND name = 'memories'");
-    if (tableSql?.sql?.includes("status TEXT NOT NULL DEFAULT 'active' CHECK")) {
+    if (typeof tableSql?.sql === "string" && tableSql.sql.includes("status TEXT NOT NULL DEFAULT 'active' CHECK")) {
       this.ensureMemoryTypeConstraint();
     }
   }
@@ -792,36 +779,25 @@ function tokenize(text) {
 
 // src/mcp/storage/base.ts
 var BaseEntity = class {
-  constructor(db, saveDb) {
+  constructor(db) {
     this.db = db;
-    this.saveDb = saveDb;
   }
   db;
-  saveDb;
   run(sql, params = []) {
-    this.db.run(sql, params);
-    if (this.saveDb) this.saveDb();
-    return { changes: this.db.getRowsModified() };
+    const stmt = this.db.prepare(sql);
+    const result = stmt.run(...params);
+    return { changes: result.changes };
   }
   exec(sql) {
     this.db.exec(sql);
-    if (this.saveDb) this.saveDb();
   }
   all(sql, params = []) {
-    const result = this.db.exec(sql, params);
-    if (result.length === 0) return [];
-    const { columns, values } = result[0];
-    return values.map((row) => {
-      const obj = {};
-      columns.forEach((col, i) => {
-        obj[col] = row[i];
-      });
-      return obj;
-    });
+    const stmt = this.db.prepare(sql);
+    return stmt.all(...params);
   }
   get(sql, params = []) {
-    const rows = this.all(sql, params);
-    return rows[0];
+    const stmt = this.db.prepare(sql);
+    return stmt.get(...params);
   }
   safeJSONParse(json, defaultValue) {
     if (!json) return defaultValue;
@@ -1135,7 +1111,7 @@ var MemoryEntity = class extends BaseEntity {
     }
     return count;
   }
-  getRecentMemories(repo, limit, offset = 0, includeArchived = false, excludeTypes = []) {
+  getRecentMemories(repo, limit, offset = 0, includeArchived = false, excludeTypes = [], sortOrder = "DESC") {
     let query = "SELECT * FROM memories WHERE repo = ?";
     const params = [repo];
     if (!includeArchived) {
@@ -1145,7 +1121,7 @@ var MemoryEntity = class extends BaseEntity {
       query += ` AND type NOT IN (${excludeTypes.map(() => "?").join(",")})`;
       params.push(...excludeTypes);
     }
-    query += " ORDER BY created_at DESC LIMIT ? OFFSET ?";
+    query += ` ORDER BY importance DESC, created_at ${sortOrder} LIMIT ? OFFSET ?`;
     params.push(limit, offset);
     const rows = this.all(query, params);
     return rows.map((row) => this.rowToMemoryEntry(row));
@@ -1380,8 +1356,28 @@ var TaskEntity = class extends BaseEntity {
     const fields = [];
     const values = [];
     const anyUpdates = updates;
+    const VALID_COLUMNS = /* @__PURE__ */ new Set([
+      "repo",
+      "task_code",
+      "phase",
+      "title",
+      "description",
+      "status",
+      "priority",
+      "agent",
+      "role",
+      "doc_path",
+      "finished_at",
+      "canceled_at",
+      "tags",
+      "metadata",
+      "parent_id",
+      "depends_on",
+      "est_tokens",
+      "in_progress_at"
+    ]);
     Object.keys(updates).forEach((key) => {
-      if (key !== "comment" && key !== "model" && anyUpdates[key] !== void 0) {
+      if (VALID_COLUMNS.has(key) && anyUpdates[key] !== void 0) {
         if (key === "tags" || key === "metadata") {
           fields.push(`${key} = ?`);
           values.push(JSON.stringify(anyUpdates[key]));
@@ -1899,32 +1895,6 @@ function resolveDbPath() {
   return standardPath;
 }
 var DB_PATH = resolveDbPath();
-var sqlJsReady = null;
-var sqlJsModule = null;
-async function getSqlJs() {
-  if (!sqlJsModule) {
-    sqlJsModule = await initSqlJs();
-  }
-  await sqlJsReady;
-  return sqlJsModule;
-}
-function createSaveFunction(db, dbPath) {
-  return () => {
-    if (dbPath && dbPath !== ":memory:") {
-      const data = db.export();
-      const buffer = Buffer.from(data);
-      fs2.writeFileSync(dbPath, buffer);
-    }
-  };
-}
-function warmUpSqlJs() {
-  if (!sqlJsReady) {
-    sqlJsReady = (async () => {
-      sqlJsModule = await initSqlJs();
-    })();
-  }
-  return sqlJsReady;
-}
 var SQLiteStore = class _SQLiteStore {
   db;
   memories;
@@ -1932,106 +1902,42 @@ var SQLiteStore = class _SQLiteStore {
   actions;
   system;
   summaries;
-  _ready;
-  lastLoadedAt = 0;
-  saveDbPtr;
   dbPathInstance;
   constructor(dbPath) {
     const finalPath = dbPath ?? DB_PATH;
     this.dbPathInstance = finalPath;
-    warmUpSqlJs();
-    this.db = {};
-    this.memories = {};
-    this.tasks = {};
-    this.actions = {};
-    this.system = {};
-    this.summaries = {};
-    this._ready = this._init(finalPath);
-  }
-  async _init(finalPath) {
-    const SQL = await getSqlJs();
-    let db;
-    if (finalPath === ":memory:") {
-      db = new SQL.Database();
-    } else {
+    if (finalPath !== ":memory:") {
       const dbDir = path2.dirname(finalPath);
       if (!fs2.existsSync(dbDir)) {
         fs2.mkdirSync(dbDir, { recursive: true });
       }
-      if (fs2.existsSync(finalPath)) {
-        const fileBuffer = fs2.readFileSync(finalPath);
-        db = new SQL.Database(fileBuffer);
-      } else {
-        db = new SQL.Database();
-      }
     }
-    this.saveDbPtr = createSaveFunction(db, finalPath);
-    const wrappedSaveDb = () => {
-      if (this.saveDbPtr) {
-        this.saveDbPtr();
-        this.lastLoadedAt = Date.now();
-      }
-    };
-    db.run("PRAGMA journal_mode = WAL");
-    db.run("PRAGMA synchronous = NORMAL");
-    db.run("PRAGMA busy_timeout = 5000");
-    const migrator = new MigrationManager(db, wrappedSaveDb);
+    this.db = new Database(finalPath);
+    this.db.pragma("journal_mode = WAL");
+    this.db.pragma("synchronous = NORMAL");
+    this.db.pragma("busy_timeout = 5000");
+    this.db.pragma("foreign_keys = ON");
+    const migrator = new MigrationManager(this.db);
     migrator.migrate();
     migrator.addMemoryCodeColumn();
-    Object.assign(this, {
-      db,
-      memories: new MemoryEntity(db, wrappedSaveDb),
-      tasks: new TaskEntity(db, wrappedSaveDb),
-      actions: new ActionEntity(db, wrappedSaveDb),
-      system: new SystemEntity(db, wrappedSaveDb),
-      summaries: new SummaryEntity(db, wrappedSaveDb)
-    });
-    this.lastLoadedAt = Date.now();
-    if (finalPath !== ":memory:") {
-      wrappedSaveDb();
-    }
-  }
-  async refresh(force = false) {
-    const path5 = this.getDbPath();
-    if (path5 === ":memory:") return;
-    try {
-      const stats = fs2.statSync(path5);
-      const mtime = stats.mtimeMs;
-      if (force || mtime > this.lastLoadedAt) {
-        const SQL = await getSqlJs();
-        const fileBuffer = fs2.readFileSync(path5);
-        const newDb = new SQL.Database(fileBuffer);
-        const wrappedSaveDb = () => {
-          if (this.saveDbPtr) {
-            this.saveDbPtr();
-            this.lastLoadedAt = Date.now();
-          }
-        };
-        this.db = newDb;
-        this.memories = new MemoryEntity(newDb, wrappedSaveDb);
-        this.tasks = new TaskEntity(newDb, wrappedSaveDb);
-        this.actions = new ActionEntity(newDb, wrappedSaveDb);
-        this.system = new SystemEntity(newDb, wrappedSaveDb);
-        this.summaries = new SummaryEntity(newDb, wrappedSaveDb);
-        this.lastLoadedAt = Date.now();
-      }
-    } catch (e) {
-      console.error("Failed to refresh database from disk:", e);
-    }
+    this.memories = new MemoryEntity(this.db);
+    this.tasks = new TaskEntity(this.db);
+    this.actions = new ActionEntity(this.db);
+    this.system = new SystemEntity(this.db);
+    this.summaries = new SummaryEntity(this.db);
   }
   async ready() {
-    await this._ready;
+  }
+  async refresh() {
   }
   static async create(dbPath) {
-    const store = new _SQLiteStore(dbPath);
-    await store.ready();
-    return store;
+    return new _SQLiteStore(dbPath);
   }
   getDbPath() {
     return this.dbPathInstance;
   }
   close() {
-    if (this.db && this.db.close) {
+    if (this.db && this.db.open) {
       this.db.close();
     }
   }
@@ -2057,7 +1963,8 @@ function updateSessionFromInitialize(session, params) {
   session.clientCapabilities = capabilities;
   session.supportsRoots = Boolean(capabilities.roots);
   session.supportsSampling = Boolean(capabilities.sampling);
-  session.supportsSamplingTools = Boolean(capabilities.sampling?.tools);
+  const sampling = capabilities.sampling;
+  session.supportsSamplingTools = Boolean(sampling?.tools);
   session.supportsElicitation = Boolean(capabilities.elicitation);
   session.supportsElicitationForm = supportsElicitationMode(capabilities.elicitation, "form");
   session.supportsElicitationUrl = supportsElicitationMode(capabilities.elicitation, "url");