kiro-memory 1.5.0 → 1.6.0

package/plugin/dist/services/sqlite/Database.js

@@ -70,7 +70,7 @@ var BunQueryCompat = class {
 // src/shared/paths.ts
 import { join as join2, dirname, basename } from "path";
 import { homedir as homedir2 } from "os";
-import { mkdirSync as mkdirSync2 } from "fs";
+import { existsSync as existsSync2, mkdirSync as mkdirSync2 } from "fs";
 import { fileURLToPath } from "url";
 
 // src/utils/logger.ts
@@ -300,7 +300,9 @@ function getDirname() {
   return dirname(fileURLToPath(import.meta.url));
 }
 var _dirname = getDirname();
-var DATA_DIR = process.env.KIRO_MEMORY_DATA_DIR || process.env.CONTEXTKIT_DATA_DIR || join2(homedir2(), ".contextkit");
+var _legacyDir = join2(homedir2(), ".contextkit");
+var _defaultDir = existsSync2(_legacyDir) ? _legacyDir : join2(homedir2(), ".kiro-memory");
+var DATA_DIR = process.env.KIRO_MEMORY_DATA_DIR || process.env.CONTEXTKIT_DATA_DIR || _defaultDir;
 var KIRO_CONFIG_DIR = process.env.KIRO_CONFIG_DIR || join2(homedir2(), ".kiro");
 var PLUGIN_ROOT = join2(KIRO_CONFIG_DIR, "plugins", "kiro-memory");
 var ARCHIVES_DIR = join2(DATA_DIR, "archives");
@@ -309,7 +311,8 @@ var TRASH_DIR = join2(DATA_DIR, "trash");
 var BACKUPS_DIR = join2(DATA_DIR, "backups");
 var MODES_DIR = join2(DATA_DIR, "modes");
 var USER_SETTINGS_PATH = join2(DATA_DIR, "settings.json");
-var DB_PATH = join2(DATA_DIR, "contextkit.db");
+var _legacyDb = join2(DATA_DIR, "contextkit.db");
+var DB_PATH = existsSync2(_legacyDb) ? _legacyDb : join2(DATA_DIR, "kiro-memory.db");
 var VECTOR_DB_DIR = join2(DATA_DIR, "vector-db");
 var OBSERVER_SESSIONS_DIR = join2(DATA_DIR, "observer-sessions");
 var KIRO_SETTINGS_PATH = join2(KIRO_CONFIG_DIR, "settings.json");
@@ -324,7 +327,11 @@ var SQLITE_CACHE_SIZE_PAGES = 1e4;
 var dbInstance = null;
 var KiroMemoryDatabase = class {
   db;
-  constructor(dbPath = DB_PATH) {
+  /**
+   * @param dbPath - Percorso al file SQLite (default: DB_PATH)
+   * @param skipMigrations - Se true, salta il migration runner (per hook ad alta frequenza)
+   */
+  constructor(dbPath = DB_PATH, skipMigrations = false) {
     if (dbPath !== ":memory:") {
       ensureDir(DATA_DIR);
     }
@@ -335,8 +342,18 @@ var KiroMemoryDatabase = class {
     this.db.run("PRAGMA temp_store = memory");
     this.db.run(`PRAGMA mmap_size = ${SQLITE_MMAP_SIZE_BYTES}`);
     this.db.run(`PRAGMA cache_size = ${SQLITE_CACHE_SIZE_PAGES}`);
-    const migrationRunner = new MigrationRunner(this.db);
-    migrationRunner.runAllMigrations();
+    if (!skipMigrations) {
+      const migrationRunner = new MigrationRunner(this.db);
+      migrationRunner.runAllMigrations();
+    }
+  }
+  /**
+   * Esegue una funzione all'interno di una transazione atomica.
+   * Se fn() lancia un errore, la transazione viene annullata automaticamente.
+   */
+  withTransaction(fn) {
+    const transaction = this.db.transaction(fn);
+    return transaction(this.db);
   }
   /**
    * Close the database connection
@@ -634,6 +651,7 @@ async function initializeDatabase() {
   return await manager.initialize();
 }
 export {
+  KiroMemoryDatabase as ContextKitDatabase,
   Database,
   DatabaseManager,
   KiroMemoryDatabase,

package/plugin/dist/services/sqlite/Search.js

@@ -1,15 +1,21 @@
 import { createRequire } from 'module';const require = createRequire(import.meta.url);
 
 // src/services/sqlite/Search.ts
+function sanitizeFTS5Query(query) {
+  const terms = query.replace(/[""]/g, "").split(/\s+/).filter((t) => t.length > 0).map((t) => `"${t}"`);
+  return terms.join(" ");
+}
 function searchObservationsFTS(db, query, filters = {}) {
   const limit = filters.limit || 50;
   try {
+    const safeQuery = sanitizeFTS5Query(query);
+    if (!safeQuery) return searchObservationsLIKE(db, query, filters);
     let sql = `
       SELECT o.* FROM observations o
       JOIN observations_fts fts ON o.id = fts.rowid
       WHERE observations_fts MATCH ?
     `;
-    const params = [query];
+    const params = [safeQuery];
     if (filters.project) {
       sql += " AND o.project = ?";
       params.push(filters.project);

package/plugin/dist/services/sqlite/index.js

@@ -70,7 +70,7 @@ var BunQueryCompat = class {
 // src/shared/paths.ts
 import { join as join2, dirname, basename } from "path";
 import { homedir as homedir2 } from "os";
-import { mkdirSync as mkdirSync2 } from "fs";
+import { existsSync as existsSync2, mkdirSync as mkdirSync2 } from "fs";
 import { fileURLToPath } from "url";
 
 // src/utils/logger.ts
@@ -300,7 +300,9 @@ function getDirname() {
   return dirname(fileURLToPath(import.meta.url));
 }
 var _dirname = getDirname();
-var DATA_DIR = process.env.KIRO_MEMORY_DATA_DIR || process.env.CONTEXTKIT_DATA_DIR || join2(homedir2(), ".contextkit");
+var _legacyDir = join2(homedir2(), ".contextkit");
+var _defaultDir = existsSync2(_legacyDir) ? _legacyDir : join2(homedir2(), ".kiro-memory");
+var DATA_DIR = process.env.KIRO_MEMORY_DATA_DIR || process.env.CONTEXTKIT_DATA_DIR || _defaultDir;
 var KIRO_CONFIG_DIR = process.env.KIRO_CONFIG_DIR || join2(homedir2(), ".kiro");
 var PLUGIN_ROOT = join2(KIRO_CONFIG_DIR, "plugins", "kiro-memory");
 var ARCHIVES_DIR = join2(DATA_DIR, "archives");
@@ -309,7 +311,8 @@ var TRASH_DIR = join2(DATA_DIR, "trash");
 var BACKUPS_DIR = join2(DATA_DIR, "backups");
 var MODES_DIR = join2(DATA_DIR, "modes");
 var USER_SETTINGS_PATH = join2(DATA_DIR, "settings.json");
-var DB_PATH = join2(DATA_DIR, "contextkit.db");
+var _legacyDb = join2(DATA_DIR, "contextkit.db");
+var DB_PATH = existsSync2(_legacyDb) ? _legacyDb : join2(DATA_DIR, "kiro-memory.db");
 var VECTOR_DB_DIR = join2(DATA_DIR, "vector-db");
 var OBSERVER_SESSIONS_DIR = join2(DATA_DIR, "observer-sessions");
 var KIRO_SETTINGS_PATH = join2(KIRO_CONFIG_DIR, "settings.json");
@@ -324,7 +327,11 @@ var SQLITE_CACHE_SIZE_PAGES = 1e4;
 var dbInstance = null;
 var KiroMemoryDatabase = class {
   db;
-  constructor(dbPath = DB_PATH) {
+  /**
+   * @param dbPath - Percorso al file SQLite (default: DB_PATH)
+   * @param skipMigrations - Se true, salta il migration runner (per hook ad alta frequenza)
+   */
+  constructor(dbPath = DB_PATH, skipMigrations = false) {
     if (dbPath !== ":memory:") {
       ensureDir(DATA_DIR);
     }
@@ -335,8 +342,18 @@ var KiroMemoryDatabase = class {
     this.db.run("PRAGMA temp_store = memory");
     this.db.run(`PRAGMA mmap_size = ${SQLITE_MMAP_SIZE_BYTES}`);
     this.db.run(`PRAGMA cache_size = ${SQLITE_CACHE_SIZE_PAGES}`);
-    const migrationRunner = new MigrationRunner(this.db);
-    migrationRunner.runAllMigrations();
+    if (!skipMigrations) {
+      const migrationRunner = new MigrationRunner(this.db);
+      migrationRunner.runAllMigrations();
+    }
+  }
+  /**
+   * Esegue una funzione all'interno di una transazione atomica.
+   * Se fn() lancia un errore, la transazione viene annullata automaticamente.
+   */
+  withTransaction(fn) {
+    const transaction = this.db.transaction(fn);
+    return transaction(this.db);
   }
   /**
    * Close the database connection
@@ -797,15 +814,21 @@ function deletePrompt(db, id) {
 }
 
 // src/services/sqlite/Search.ts
+function sanitizeFTS5Query(query) {
+  const terms = query.replace(/[""]/g, "").split(/\s+/).filter((t) => t.length > 0).map((t) => `"${t}"`);
+  return terms.join(" ");
+}
 function searchObservationsFTS(db, query, filters = {}) {
   const limit = filters.limit || 50;
   try {
+    const safeQuery = sanitizeFTS5Query(query);
+    if (!safeQuery) return searchObservationsLIKE(db, query, filters);
     let sql = `
       SELECT o.* FROM observations o
       JOIN observations_fts fts ON o.id = fts.rowid
       WHERE observations_fts MATCH ?
     `;
-    const params = [query];
+    const params = [safeQuery];
     if (filters.project) {
       sql += " AND o.project = ?";
       params.push(filters.project);

package/plugin/dist/shared/paths.js

@@ -3,7 +3,7 @@ import { createRequire } from 'module';const require = createRequire(import.meta
 // src/shared/paths.ts
 import { join as join2, dirname, basename } from "path";
 import { homedir as homedir2 } from "os";
-import { mkdirSync as mkdirSync2 } from "fs";
+import { existsSync as existsSync2, mkdirSync as mkdirSync2 } from "fs";
 import { execSync } from "child_process";
 import { fileURLToPath } from "url";
 
@@ -234,7 +234,9 @@ function getDirname() {
   return dirname(fileURLToPath(import.meta.url));
 }
 var _dirname = getDirname();
-var DATA_DIR = process.env.KIRO_MEMORY_DATA_DIR || process.env.CONTEXTKIT_DATA_DIR || join2(homedir2(), ".contextkit");
+var _legacyDir = join2(homedir2(), ".contextkit");
+var _defaultDir = existsSync2(_legacyDir) ? _legacyDir : join2(homedir2(), ".kiro-memory");
+var DATA_DIR = process.env.KIRO_MEMORY_DATA_DIR || process.env.CONTEXTKIT_DATA_DIR || _defaultDir;
 var KIRO_CONFIG_DIR = process.env.KIRO_CONFIG_DIR || join2(homedir2(), ".kiro");
 var PLUGIN_ROOT = join2(KIRO_CONFIG_DIR, "plugins", "kiro-memory");
 var ARCHIVES_DIR = join2(DATA_DIR, "archives");
@@ -243,7 +245,8 @@ var TRASH_DIR = join2(DATA_DIR, "trash");
 var BACKUPS_DIR = join2(DATA_DIR, "backups");
 var MODES_DIR = join2(DATA_DIR, "modes");
 var USER_SETTINGS_PATH = join2(DATA_DIR, "settings.json");
-var DB_PATH = join2(DATA_DIR, "contextkit.db");
+var _legacyDb = join2(DATA_DIR, "contextkit.db");
+var DB_PATH = existsSync2(_legacyDb) ? _legacyDb : join2(DATA_DIR, "kiro-memory.db");
 var VECTOR_DB_DIR = join2(DATA_DIR, "vector-db");
 var OBSERVER_SESSIONS_DIR = join2(DATA_DIR, "observer-sessions");
 var KIRO_SETTINGS_PATH = join2(KIRO_CONFIG_DIR, "settings.json");

package/plugin/dist/viewer.js

@@ -23975,11 +23975,22 @@ function useSSE() {
         console.error("Failed to fetch projects:", err);
       }
     };
+    const fetchAll = () => {
+      fetchObservations();
+      fetchSummaries();
+      fetchPrompts();
+      fetchProjects();
+    };
+    let wasConnected = false;
     const connect = () => {
       if (!mountedRef.current) return;
       eventSource = new EventSource("/events");
       eventSource.onopen = () => {
         if (!mountedRef.current) return;
+        if (wasConnected) {
+          fetchAll();
+        }
+        wasConnected = true;
         retryCount = 0;
         setState((prev) => ({ ...prev, isConnected: true }));
       };
@@ -24003,10 +24014,7 @@ function useSSE() {
         fetchPrompts();
       });
     };
-    fetchObservations();
-    fetchSummaries();
-    fetchPrompts();
-    fetchProjects();
+    fetchAll();
     connect();
     return () => {
       mountedRef.current = false;