@vheins/local-memory-mcp 0.7.3 → 0.8.0

package/dist/{chunk-ZCK6RZFX.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");
@@ -2170,7 +2077,8 @@ var MemoryStoreSchema = z.object({
   supersedes: z.string().uuid().optional(),
   tags: z.array(z.string()).optional(),
   metadata: z.record(z.string(), z.any()).optional(),
-  is_global: z.boolean().default(false)
+  is_global: z.boolean().default(false),
+  structured: z.boolean().default(false)
 });
 var MemoryUpdateSchema = z.object({
   id: z.string().uuid(),
@@ -2185,7 +2093,8 @@ var MemoryUpdateSchema = z.object({
   tags: z.array(z.string()).optional(),
   metadata: z.record(z.string(), z.any()).optional(),
   is_global: z.boolean().optional(),
-  completed_at: z.string().optional()
+  completed_at: z.string().optional(),
+  structured: z.boolean().default(false)
 }).refine(
   (data) => data.type !== void 0 || data.content !== void 0 || data.title !== void 0 || data.importance !== void 0 || data.status !== void 0 || data.supersedes !== void 0 || data.tags !== void 0 || data.metadata !== void 0 || data.is_global !== void 0 || data.agent !== void 0 || data.role !== void 0 || data.completed_at !== void 0,
   { message: "At least one field must be provided for update" }
@@ -2208,7 +2117,8 @@ var MemorySearchSchema = z.object({
 var MemoryAcknowledgeSchema = z.object({
   memory_id: z.string().uuid(),
   status: z.enum(["used", "irrelevant", "contradictory"]),
-  application_context: z.string().min(10).optional()
+  application_context: z.string().min(10).optional(),
+  structured: z.boolean().default(false)
 });
 var MemoryRecapSchema = z.object({
   repo: z.string().min(1).transform(normalizeRepo),
@@ -2219,13 +2129,15 @@ var MemoryRecapSchema = z.object({
 var MemoryDeleteSchema = z.object({
   repo: z.string().min(1).transform(normalizeRepo).optional(),
   id: z.string().uuid().optional(),
-  ids: z.array(z.string().uuid()).min(1).optional()
+  ids: z.array(z.string().uuid()).min(1).optional(),
+  structured: z.boolean().default(false)
 }).refine((data) => data.id !== void 0 || data.ids !== void 0, {
   message: "Either 'id' or 'ids' must be provided for deletion"
 });
 var MemorySummarizeSchema = z.object({
   repo: z.string().min(1).transform(normalizeRepo),
-  signals: z.array(z.string().max(200)).min(1)
+  signals: z.array(z.string().max(200)).min(1),
+  structured: z.boolean().default(false)
 });
 var MemorySynthesizeSchema = z.object({
   repo: z.string().min(1).transform(normalizeRepo).optional(),
@@ -2235,7 +2147,8 @@ var MemorySynthesizeSchema = z.object({
   include_tasks: z.boolean().default(true),
   use_tools: z.boolean().default(true),
   max_iterations: z.number().int().min(1).max(5).default(3),
-  max_tokens: z.number().int().min(128).max(4e3).default(1200)
+  max_tokens: z.number().int().min(128).max(4e3).default(1200),
+  structured: z.boolean().default(false)
 });
 var TaskStatusSchema = z.enum(["backlog", "pending", "in_progress", "completed", "canceled", "blocked"]);
 var TaskPrioritySchema = z.number().min(1).max(5);
@@ -2273,7 +2186,8 @@ var TaskCreateSchema = z.object({
   depends_on: z.string().uuid().optional(),
   est_tokens: z.number().int().min(0).optional(),
   // Allow bulk tasks
-  tasks: z.array(SingleTaskCreateSchema).min(1).optional()
+  tasks: z.array(SingleTaskCreateSchema).min(1).optional(),
+  structured: z.boolean().default(false)
 }).refine(
   (data) => {
     if (data.tasks) return true;
@@ -2282,7 +2196,8 @@ var TaskCreateSchema = z.object({
   { message: "Either 'tasks' array or single task fields (task_code, phase, title, description) must be provided" }
 );
 var TaskCreateInteractiveSchema = SingleTaskCreateSchema.partial().extend({
-  repo: z.string().min(1).transform(normalizeRepo).optional()
+  repo: z.string().min(1).transform(normalizeRepo).optional(),
+  structured: z.boolean().default(false)
 });
 var TaskUpdateSchema = z.object({
   repo: z.string().min(1).transform(normalizeRepo),
@@ -2304,7 +2219,8 @@ var TaskUpdateSchema = z.object({
   parent_id: z.string().uuid().optional(),
   depends_on: z.string().uuid().optional(),
   est_tokens: z.number().int().min(0).optional(),
-  force: z.boolean().optional()
+  force: z.boolean().optional(),
+  structured: z.boolean().default(false)
 }).refine((data) => data.id !== void 0 || data.ids !== void 0, {
   message: "Either 'id' or 'ids' must be provided for update"
 }).refine((data) => Object.keys(data).length > 2, {
@@ -2330,13 +2246,15 @@ var TaskSearchSchema = z.object({
 var TaskDeleteSchema = z.object({
   repo: z.string().min(1).transform(normalizeRepo),
   id: z.string().uuid().optional(),
-  ids: z.array(z.string().uuid()).min(1).optional()
+  ids: z.array(z.string().uuid()).min(1).optional(),
+  structured: z.boolean().default(false)
 }).refine((data) => data.id !== void 0 || data.ids !== void 0, {
   message: "Either 'id' or 'ids' must be provided for deletion"
 });
 var MemoryDetailSchema = z.object({
   id: z.string().uuid().optional(),
-  code: z.string().max(20).optional()
+  code: z.string().max(20).optional(),
+  structured: z.boolean().default(false)
 }).refine((data) => data.id !== void 0 || data.code !== void 0, {
   message: "Either id or code must be provided"
 });
@@ -2378,7 +2296,8 @@ var TOOL_DEFINITIONS = [
           description: "Allow the sampled model to call local memory/task tools during synthesis when the client supports sampling.tools."
         },
         max_iterations: { type: "number", minimum: 1, maximum: 5, default: 3 },
-        max_tokens: { type: "number", minimum: 128, maximum: 4e3, default: 1200 }
+        max_tokens: { type: "number", minimum: 128, maximum: 4e3, default: 1200 },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON results." }
       },
       required: ["objective"]
     },
@@ -2421,7 +2340,8 @@ var TOOL_DEFINITIONS = [
         priority: { type: "number", minimum: 1, maximum: 5, default: 3 },
         agent: { type: "string" },
         role: { type: "string" },
-        doc_path: { type: "string" }
+        doc_path: { type: "string" },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON result." }
       }
     },
     outputSchema: {
@@ -2444,7 +2364,8 @@ var TOOL_DEFINITIONS = [
     inputSchema: {
       type: "object",
       properties: {
-        id: { type: "string", format: "uuid", description: "Memory entry ID" }
+        id: { type: "string", format: "uuid", description: "Memory entry ID" },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON details." }
       },
       required: ["id"]
     }
@@ -2548,7 +2469,8 @@ var TOOL_DEFINITIONS = [
           description: "If true, this memory is shared across all repositories"
         },
         ttlDays: { type: "number", minimum: 1 },
-        supersedes: { type: "string", format: "uuid" }
+        supersedes: { type: "string", format: "uuid" },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON of the stored memory." }
       },
       required: ["type", "title", "content", "importance", "scope", "agent", "model"]
     },
@@ -2581,7 +2503,8 @@ var TOOL_DEFINITIONS = [
       properties: {
         memory_id: { type: "string", format: "uuid" },
         status: { type: "string", enum: ["used", "irrelevant", "contradictory"] },
-        application_context: { type: "string", minLength: 10 }
+        application_context: { type: "string", minLength: 10 },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON result." }
       },
       required: ["memory_id", "status"]
     },
@@ -2632,7 +2555,8 @@ var TOOL_DEFINITIONS = [
         tags: { type: "array", items: { type: "string" } },
         metadata: { type: "object" },
         is_global: { type: "boolean" },
-        completed_at: { type: "string" }
+        completed_at: { type: "string" },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON of the updated memory." }
       },
       required: ["id"]
     },
@@ -2756,7 +2680,8 @@ var TOOL_DEFINITIONS = [
           items: { type: "string", maxLength: 200 },
           minItems: 1,
           description: "High-level signals to include in summary"
-        }
+        },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON of the summary." }
       },
       required: ["repo", "signals"]
     },
@@ -2791,7 +2716,8 @@ var TOOL_DEFINITIONS = [
           items: { type: "string", format: "uuid" },
           minItems: 1,
           description: "Array of memory IDs to delete"
-        }
+        },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON result." }
       }
     },
     outputSchema: {
@@ -2939,7 +2865,8 @@ var TOOL_DEFINITIONS = [
             required: ["task_code", "phase", "title", "description"]
           },
           description: "Array of tasks for bulk creation"
-        }
+        },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON result." }
       },
       required: ["repo"]
     },
@@ -3005,7 +2932,8 @@ var TOOL_DEFINITIONS = [
         force: {
           type: "boolean",
           description: "If true, bypasses status transition validation (e.g. pending -> completed)."
-        }
+        },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON result." }
       },
       required: ["repo"]
     },
@@ -3042,7 +2970,8 @@ var TOOL_DEFINITIONS = [
       properties: {
         repo: { type: "string", description: "Repository name" },
         id: { type: "string", format: "uuid", description: "Task ID (for single deletion)" },
-        ids: { type: "array", items: { type: "string", format: "uuid" }, description: "Task IDs (for bulk deletion)" }
+        ids: { type: "array", items: { type: "string", format: "uuid" }, description: "Task IDs (for bulk deletion)" },
+        structured: { type: "boolean", default: false, description: "If true, returns structured JSON result." }
       },
       required: ["repo"]
     },