@hasna/mementos 0.14.17 → 0.14.18

package/dist/index.js

@@ -10142,6 +10142,35 @@ ALTER TABLE memories ADD COLUMN sequence_group TEXT DEFAULT NULL;
 ALTER TABLE memories ADD COLUMN sequence_order INTEGER DEFAULT NULL;
 CREATE INDEX IF NOT EXISTS idx_memories_sequence_group ON memories(sequence_group) WHERE sequence_group IS NOT NULL;
 INSERT OR IGNORE INTO _migrations (id) VALUES (32);
+`,
+    `
+ALTER TABLE machines ADD COLUMN is_primary INTEGER NOT NULL DEFAULT 0;
+CREATE INDEX IF NOT EXISTS idx_machines_primary ON machines(is_primary);
+CREATE TRIGGER IF NOT EXISTS machines_single_primary_insert
+AFTER INSERT ON machines
+WHEN NEW.is_primary = 1
+BEGIN
+  UPDATE machines
+  SET is_primary = 0,
+      last_seen_at = COALESCE(NEW.last_seen_at, datetime('now'))
+  WHERE id != NEW.id AND is_primary = 1;
+END;
+CREATE TRIGGER IF NOT EXISTS machines_single_primary_update
+AFTER UPDATE OF is_primary ON machines
+WHEN NEW.is_primary = 1
+BEGIN
+  UPDATE machines
+  SET is_primary = 0,
+      last_seen_at = COALESCE(NEW.last_seen_at, datetime('now'))
+  WHERE id != NEW.id AND is_primary = 1;
+END;
+CREATE TRIGGER IF NOT EXISTS machines_prevent_delete_primary
+BEFORE DELETE ON machines
+WHEN OLD.is_primary = 1
+BEGIN
+  SELECT RAISE(ABORT, 'Primary machine cannot be deleted');
+END;
+INSERT OR IGNORE INTO _migrations (id) VALUES (33);
 `
   ];
 });
@@ -10225,11 +10254,16 @@ function ensureDir(filePath) {
   }
 }
 function getDatabase(dbPath) {
-  if (_db)
-    return _db;
   const path = dbPath || getDbPath2();
+  if (_db) {
+    if (_dbPath === path)
+      return _db;
+    _db.close();
+    _db = null;
+  }
+  _dbPath = path;
   ensureDir(path);
-  _db = new SqliteAdapter(path, { create: true });
+  _db = new SqliteAdapter(path);
   _db.run("PRAGMA journal_mode = WAL");
   _db.run("PRAGMA busy_timeout = 5000");
   _db.run("PRAGMA foreign_keys = ON");
@@ -10253,13 +10287,17 @@ function runMigrations(db) {
     for (let i = currentLevel;i < MIGRATIONS.length; i++) {
       try {
         db.exec(MIGRATIONS[i]);
-      } catch {}
+      } catch (e) {
+        console.warn(`[mementos] Migration ${i + 1} failed: ${e instanceof Error ? e.message : String(e)}`);
+      }
     }
   } catch {
-    for (const migration of MIGRATIONS) {
+    for (let i = 0;i < MIGRATIONS.length; i++) {
       try {
-        db.exec(migration);
-      } catch {}
+        db.exec(MIGRATIONS[i]);
+      } catch (e) {
+        console.warn(`[mementos] Migration ${i + 1} failed: ${e instanceof Error ? e.message : String(e)}`);
+      }
     }
   }
 }
@@ -10282,6 +10320,9 @@ function shortUuid() {
   return crypto.randomUUID().slice(0, 8);
 }
 function resolvePartialId(db, table, partialId) {
+  if (!ALLOWED_TABLES.has(table)) {
+    throw new Error(`Invalid table name: ${table}`);
+  }
   if (partialId.length >= 36) {
     const row = db.query(`SELECT id FROM ${table} WHERE id = ?`).get(partialId);
     return row?.id ?? null;
@@ -10292,10 +10333,25 @@ function resolvePartialId(db, table, partialId) {
   }
   return null;
 }
-var _db = null;
+var _db = null, _dbPath = null, ALLOWED_TABLES;
 var init_database = __esm(() => {
   init_dist();
   init_migrations();
+  ALLOWED_TABLES = new Set([
+    "memories",
+    "agents",
+    "entities",
+    "projects",
+    "relations",
+    "memory_audit_log",
+    "locks",
+    "sessions",
+    "session_memory_jobs",
+    "synthesis_runs",
+    "synthesis_proposals",
+    "tool_events",
+    "webhook_hooks"
+  ]);
 });
 
 // src/types/index.ts
@@ -10999,6 +11055,22 @@ function listMemories(filter, db) {
       conditions.push("session_id = ?");
       params.push(filter.session_id);
     }
+    if ("machine_id" in filter) {
+      if (filter.machine_id === null) {
+        conditions.push("machine_id IS NULL");
+      } else if (filter.machine_id) {
+        conditions.push("machine_id = ?");
+        params.push(filter.machine_id);
+      }
+    }
+    if ("visible_to_machine_id" in filter) {
+      if (filter.visible_to_machine_id === null) {
+        conditions.push("machine_id IS NULL");
+      } else if (filter.visible_to_machine_id !== undefined) {
+        conditions.push("(machine_id IS NULL OR machine_id = ?)");
+        params.push(filter.visible_to_machine_id);
+      }
+    }
     if (filter.min_importance) {
       conditions.push("importance >= ?");
       params.push(filter.min_importance);
@@ -11605,6 +11677,102 @@ function listProjects(db) {
   const rows = d.query("SELECT * FROM projects ORDER BY updated_at DESC").all();
   return rows.map(parseProjectRow);
 }
+// src/db/machines.ts
+init_database();
+import { hostname, platform as platform2 } from "os";
+function parseMachine(row) {
+  if (!row)
+    return null;
+  return {
+    ...row,
+    is_primary: Boolean(row.is_primary)
+  };
+}
+function registerMachine(name, db = getDatabase()) {
+  const host = hostname();
+  const plat = platform2();
+  const machineName = name?.trim() || host;
+  const existing = parseMachine(db.query("SELECT * FROM machines WHERE hostname = ?").get(host));
+  if (existing) {
+    db.run("UPDATE machines SET last_seen_at = ? WHERE id = ?", [now(), existing.id]);
+    return parseMachine(db.query("SELECT * FROM machines WHERE id = ?").get(existing.id));
+  }
+  let finalName = machineName;
+  let suffix = 2;
+  while (db.query("SELECT id FROM machines WHERE name = ?").get(finalName)) {
+    finalName = `${machineName}-${suffix++}`;
+  }
+  const id = uuid();
+  db.run("INSERT INTO machines (id, name, hostname, platform) VALUES (?, ?, ?, ?)", [id, finalName, host, plat]);
+  return parseMachine(db.query("SELECT * FROM machines WHERE id = ?").get(id));
+}
+function listMachines(db = getDatabase()) {
+  const rows = db.query("SELECT * FROM machines ORDER BY is_primary DESC, last_seen_at DESC, created_at ASC").all();
+  return rows.map((row) => parseMachine(row));
+}
+function getMachine(id, db = getDatabase()) {
+  return parseMachine(db.query("SELECT * FROM machines WHERE id = ? OR name = ?").get(id, id));
+}
+function renameMachine(id, newName, db = getDatabase()) {
+  const m = parseMachine(db.query("SELECT * FROM machines WHERE id = ?").get(id));
+  if (!m)
+    throw new Error(`Machine not found: ${id}`);
+  const clash = db.query("SELECT id FROM machines WHERE name = ? AND id != ?").get(newName, id);
+  if (clash)
+    throw new Error(`Machine name already taken: ${newName}`);
+  db.run("UPDATE machines SET name = ?, last_seen_at = ? WHERE id = ?", [newName, now(), id]);
+  return parseMachine(db.query("SELECT * FROM machines WHERE id = ?").get(id));
+}
+function getPrimaryMachine(db = getDatabase()) {
+  return parseMachine(db.query("SELECT * FROM machines WHERE is_primary = 1 LIMIT 1").get());
+}
+function getPrimaryMachineCandidate(db = getDatabase()) {
+  if (getPrimaryMachine(db))
+    return null;
+  return parseMachine(db.query("SELECT * FROM machines ORDER BY created_at ASC, id ASC LIMIT 1").get());
+}
+function setPrimaryMachine(id, db = getDatabase()) {
+  const machine = getMachine(id, db);
+  if (!machine)
+    throw new Error(`Machine not found: ${id}`);
+  const updatedAt = now();
+  db.run("UPDATE machines SET is_primary = 0, last_seen_at = ? WHERE is_primary = 1 AND id != ?", [updatedAt, machine.id]);
+  db.run("UPDATE machines SET is_primary = 1, last_seen_at = ? WHERE id = ?", [updatedAt, machine.id]);
+  return getMachine(machine.id, db);
+}
+function deleteMachine(id, db = getDatabase()) {
+  const machine = getMachine(id, db);
+  if (!machine)
+    throw new Error(`Machine not found: ${id}`);
+  if (machine.is_primary) {
+    throw new Error(`Primary machine cannot be deleted: ${machine.name}`);
+  }
+  db.run("DELETE FROM machines WHERE id = ?", [machine.id]);
+}
+function getFallbackSyncTargetMachine(db = getDatabase()) {
+  return getPrimaryMachine(db);
+}
+function getPrimaryMachineStartupWarning(db = getDatabase()) {
+  if (getPrimaryMachine(db))
+    return null;
+  const candidate = getPrimaryMachineCandidate(db);
+  if (!candidate) {
+    return "No primary machine configured. Fallback sync target is unset because no machines are registered yet.";
+  }
+  return `No primary machine configured. Fallback sync target is unset. Candidate: ${candidate.name} (${candidate.id.slice(0, 8)} / ${candidate.hostname}). Confirm it with set_primary_machine.`;
+}
+function touchMachine(id, db = getDatabase()) {
+  db.run("UPDATE machines SET last_seen_at = ? WHERE id = ?", [now(), id]);
+}
+function getCurrentMachineId(db = getDatabase()) {
+  const host = hostname();
+  const m = db.query("SELECT id FROM machines WHERE hostname = ?").get(host);
+  if (m) {
+    touchMachine(m.id, db);
+    return m.id;
+  }
+  return registerMachine(undefined, db).id;
+}
 // src/lib/search.ts
 init_database();
 
@@ -12470,6 +12638,23 @@ function ensureDir2(dir) {
     mkdirSync5(dir, { recursive: true });
   }
 }
+// src/lib/machine-visibility.ts
+function resolveVisibleMachineId(machineId, db) {
+  if (machineId !== undefined) {
+    return machineId;
+  }
+  try {
+    return getCurrentMachineId(db);
+  } catch {
+    return null;
+  }
+}
+function visibleToMachineFilter(machineId, db) {
+  return {
+    visible_to_machine_id: resolveVisibleMachineId(machineId, db)
+  };
+}
+
 // src/db/tool-events.ts
 init_database();
 
@@ -12485,12 +12670,14 @@ class MemoryInjector {
     const minImportance = options.min_importance || this.config.injection.min_importance;
     const categories = options.categories || this.config.injection.categories;
     const db = options.db;
+    const visibleMachineId = resolveVisibleMachineId(options.machine_id, db);
     const allMemories = [];
     const globalMems = listMemories({
       scope: "global",
       category: categories,
       min_importance: minImportance,
       status: "active",
+      ...visibleToMachineFilter(visibleMachineId),
       limit: 100
     }, db);
     allMemories.push(...globalMems);
@@ -12501,6 +12688,7 @@ class MemoryInjector {
         min_importance: minImportance,
         status: "active",
         project_id: options.project_id,
+        ...visibleToMachineFilter(visibleMachineId),
         limit: 100
       }, db);
       allMemories.push(...sharedMems);
@@ -12512,6 +12700,7 @@ class MemoryInjector {
         min_importance: minImportance,
         status: "active",
         agent_id: options.agent_id,
+        ...visibleToMachineFilter(visibleMachineId),
         limit: 100
       }, db);
       allMemories.push(...privateMems);
@@ -12523,6 +12712,7 @@ class MemoryInjector {
         ...options.session_id ? { session_id: options.session_id } : {},
         ...options.agent_id ? { agent_id: options.agent_id } : {},
         ...options.project_id ? { project_id: options.project_id } : {},
+        ...visibleToMachineFilter(visibleMachineId),
         limit: 100
       }, db);
       allMemories.push(...workingMems);
@@ -12630,12 +12820,14 @@ ${sections.join(`
     const minImportance = options.min_importance || this.config.injection.min_importance;
     const categories = options.categories || this.config.injection.categories;
     const db = options.db;
+    const visibleMachineId = resolveVisibleMachineId(options.machine_id, db);
     const allMemories = [];
     const globalMems = listMemories({
       scope: "global",
       category: categories,
       min_importance: minImportance,
       status: "active",
+      ...visibleToMachineFilter(visibleMachineId),
       limit: 100
     }, db);
     allMemories.push(...globalMems);
@@ -12646,6 +12838,7 @@ ${sections.join(`
         min_importance: minImportance,
         status: "active",
         project_id: options.project_id,
+        ...visibleToMachineFilter(visibleMachineId),
         limit: 100
       }, db);
       allMemories.push(...sharedMems);
@@ -12657,6 +12850,7 @@ ${sections.join(`
         min_importance: minImportance,
         status: "active",
         agent_id: options.agent_id,
+        ...visibleToMachineFilter(visibleMachineId),
         limit: 100
       }, db);
       allMemories.push(...privateMems);
@@ -12668,6 +12862,7 @@ ${sections.join(`
         ...options.session_id ? { session_id: options.session_id } : {},
         ...options.agent_id ? { agent_id: options.agent_id } : {},
         ...options.project_id ? { project_id: options.project_id } : {},
+        ...visibleToMachineFilter(visibleMachineId),
         limit: 100
       }, db);
       allMemories.push(...workingMems);
@@ -13975,9 +14170,11 @@ export {
   unlinkEntityFromMemory,
   unfocus,
   touchMemory,
+  touchMachine,
   touchAgent,
   syncMemories,
   shortUuid,
+  setPrimaryMachine,
   setFocus,
   setActiveProfile,
   setActiveModel,
@@ -13986,11 +14183,13 @@ export {
   resolveProjectId,
   resolvePartialId,
   resetDatabase,
+  renameMachine,
   releaseResourceLocks,
   releaseMemoryWriteLock,
   releaseLock,
   releaseAllAgentLocks,
   registerProject,
+  registerMachine,
   registerAgent,
   redactSecrets,
   providerRegistry,
@@ -14005,6 +14204,7 @@ export {
   listProjects,
   listProfiles,
   listMemories,
+  listMachines,
   listEntities,
   listAgentsByProject,
   listAgents,
@@ -14014,12 +14214,17 @@ export {
   getRelation,
   getRelatedEntities,
   getProject,
+  getPrimaryMachineStartupWarning,
+  getPrimaryMachineCandidate,
+  getPrimaryMachine,
   getMemoryVersions,
   getMemoryByKey,
   getMemory,
   getMemoriesForEntity,
   getMemoriesByKey,
+  getMachine,
   getFocus,
+  getFallbackSyncTargetMachine,
   getEntityMemoryLinks,
   getEntityGraph,
   getEntityByName,
@@ -14028,6 +14233,7 @@ export {
   getDedupStats,
   getDbPath2 as getDbPath,
   getDatabase,
+  getCurrentMachineId,
   getAutoMemoryStats,
   getAgent,
   getActiveProfile,
@@ -14040,6 +14246,7 @@ export {
   deleteRelation,
   deleteProfile,
   deleteMemory,
+  deleteMachine,
   deleteEntity,
   defaultSyncAgents,
   dedup,