gemini-helper-friend 2.0.0 → 2.0.2

package/dist/tools/task-store.js

@@ -0,0 +1,191 @@
+/**
+ * SQLite-based persistent task store
+ * Replaces in-memory Map to support async task persistence across MCP sessions
+ */
+import Database from 'better-sqlite3';
+import { join } from 'path';
+import { fileURLToPath } from 'url';
+import { dirname } from 'path';
+import { existsSync, mkdirSync } from 'fs';
+import { TASK_ID_MIN, TASK_ID_MAX, TASK_EXPIRY_MS, CLEANUP_INTERVAL_MS, DATA_DIR_NAME, DB_FILE_NAME, } from './constants.js';
+import { cleanupOrphanedTempFiles } from './cleanup.js';
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+// Database path - store in project root/.mcp-data directory
+const DATA_DIR = join(__dirname, '../..', DATA_DIR_NAME);
+const DB_PATH = join(DATA_DIR, DB_FILE_NAME);
+// Ensure data directory exists
+if (!existsSync(DATA_DIR)) {
+    mkdirSync(DATA_DIR, { recursive: true });
+}
+// Initialize database
+const db = new Database(DB_PATH);
+// Enable WAL mode for better concurrency
+db.pragma('journal_mode = WAL');
+// Create tasks table
+db.exec(`
+  CREATE TABLE IF NOT EXISTS tasks (
+    id INTEGER PRIMARY KEY,
+    status TEXT NOT NULL,
+    task_type TEXT NOT NULL,
+    progress INTEGER NOT NULL DEFAULT 0,
+    result TEXT,
+    error TEXT,
+    started_at TEXT NOT NULL,
+    completed_at TEXT,
+    updated_at TEXT NOT NULL,
+    pid INTEGER
+  );
+
+  CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status);
+  CREATE INDEX IF NOT EXISTS idx_tasks_started_at ON tasks(started_at);
+`);
+// Migration: Add pid column if it doesn't exist (for existing databases)
+try {
+    db.exec(`ALTER TABLE tasks ADD COLUMN pid INTEGER;`);
+    console.error('[TaskStore] Migration: Added pid column to existing database');
+}
+catch (error) {
+    // Column already exists or other error - safe to ignore
+    if (!error.message?.includes('duplicate column')) {
+        console.error('[TaskStore] Migration note:', error.message);
+    }
+}
+// Prepared statements for better performance
+const insertTaskStmt = db.prepare(`
+  INSERT INTO tasks (id, status, task_type, progress, started_at, updated_at)
+  VALUES (?, ?, ?, ?, ?, ?)
+`);
+const updateTaskStmt = db.prepare(`
+  UPDATE tasks 
+  SET status = ?, progress = ?, result = ?, error = ?, completed_at = ?, updated_at = ?, pid = ?
+  WHERE id = ?
+`);
+const getTaskStmt = db.prepare(`
+  SELECT * FROM tasks WHERE id = ?
+`);
+const getCompletedTasksStmt = db.prepare(`
+  SELECT * FROM tasks WHERE status IN ('completed', 'failed')
+  ORDER BY completed_at DESC
+`);
+const deleteTaskStmt = db.prepare(`
+  DELETE FROM tasks WHERE id = ?
+`);
+const deleteExpiredTasksStmt = db.prepare(`
+  DELETE FROM tasks 
+  WHERE datetime(started_at, '+1 hour') < datetime('now')
+`);
+// ID counter - get max ID from database on startup
+let taskIdCounter = TASK_ID_MIN;
+const maxIdRow = db.prepare('SELECT MAX(id) as max_id FROM tasks').get();
+if (maxIdRow.max_id && maxIdRow.max_id >= taskIdCounter) {
+    taskIdCounter = maxIdRow.max_id + 1;
+}
+/**
+ * Generate unique 5-digit task ID
+ * Wraps around from TASK_ID_MAX back to TASK_ID_MIN
+ */
+export function generateTaskId() {
+    const id = taskIdCounter;
+    taskIdCounter++;
+    if (taskIdCounter > TASK_ID_MAX) {
+        taskIdCounter = TASK_ID_MIN;
+    }
+    return id;
+}
+/**
+ * Create a new task in pending state
+ */
+export function createTask(taskType) {
+    const id = generateTaskId();
+    const now = new Date().toISOString();
+    const task = {
+        id,
+        status: 'pending',
+        task_type: taskType,
+        progress: 0,
+        started_at: now,
+        updated_at: now,
+    };
+    insertTaskStmt.run(id, task.status, task.task_type, task.progress, task.started_at, task.updated_at);
+    // Schedule cleanup after expiry (note: this only works if process stays alive)
+    setTimeout(() => {
+        deleteTaskStmt.run(id);
+    }, TASK_EXPIRY_MS);
+    return task;
+}
+/**
+ * Update task status
+ */
+export function updateTask(id, updates) {
+    const task = getTask(id);
+    if (!task)
+        return null;
+    // Merge updates
+    const updatedTask = { ...task, ...updates, updated_at: new Date().toISOString() };
+    updateTaskStmt.run(updatedTask.status, updatedTask.progress, updatedTask.result || null, updatedTask.error || null, updatedTask.completed_at || null, updatedTask.updated_at, updatedTask.pid || null, id);
+    return updatedTask;
+}
+/**
+ * Get task by ID
+ */
+export function getTask(id) {
+    const row = getTaskStmt.get(id);
+    if (!row)
+        return null;
+    return {
+        id: row.id,
+        status: row.status,
+        task_type: row.task_type,
+        progress: row.progress,
+        result: row.result || undefined,
+        error: row.error || undefined,
+        started_at: row.started_at,
+        completed_at: row.completed_at || undefined,
+        updated_at: row.updated_at,
+        pid: row.pid || undefined,
+    };
+}
+/**
+ * Get all completed or failed tasks
+ */
+export function getCompletedTasks() {
+    const rows = getCompletedTasksStmt.all();
+    return rows.map(row => ({
+        id: row.id,
+        status: row.status,
+        task_type: row.task_type,
+        progress: row.progress,
+        result: row.result || undefined,
+        error: row.error || undefined,
+        started_at: row.started_at,
+        completed_at: row.completed_at || undefined,
+        updated_at: row.updated_at,
+        pid: row.pid || undefined,
+    }));
+}
+/**
+ * Delete expired tasks (older than 1 hour)
+ * Call this periodically to clean up old tasks
+ */
+export function cleanupExpiredTasks() {
+    const result = deleteExpiredTasksStmt.run();
+    return result.changes;
+}
+/**
+ * Close database connection (call on shutdown)
+ */
+export function closeDatabase() {
+    db.close();
+}
+// Cleanup expired tasks and orphaned temp files on startup
+cleanupExpiredTasks();
+cleanupOrphanedTempFiles();
+// Periodic cleanup at configured interval
+setInterval(() => {
+    const deleted = cleanupExpiredTasks();
+    if (deleted > 0) {
+        console.error(`[TaskStore] Cleaned up ${deleted} expired tasks`);
+    }
+}, CLEANUP_INTERVAL_MS);
+//# sourceMappingURL=task-store.js.map

package/dist/tools/task-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-store.js","sourceRoot":"","sources":["../../src/tools/task-store.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC/B,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC3C,OAAO,EACL,WAAW,EACX,WAAW,EACX,cAAc,EACd,mBAAmB,EACnB,aAAa,EACb,YAAY,GAEb,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAC;AAExD,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAkCtC,4DAA4D;AAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;AACzD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AAE7C,+BAA+B;AAC/B,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;IAC1B,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,sBAAsB;AACtB,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC;AAEjC,yCAAyC;AACzC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;AAEhC,qBAAqB;AACrB,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;CAgBP,CAAC,CAAC;AAEH,yEAAyE;AACzE,IAAI,CAAC;IACH,EAAE,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;IACrD,OAAO,CAAC,KAAK,CAAC,8DAA8D,CAAC,CAAC;AAChF,CAAC;AAAC,OAAO,KAAU,EAAE,CAAC;IACpB,wDAAwD;IACxD,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QACjD,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAC9D,CAAC;AACH,CAAC;AAED,6CAA6C;AAC7C,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC;;;CAGjC,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC;;;;CAIjC,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC;;CAE9B,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,EAAE,CAAC,OAAO,CAAC;;;CAGxC,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC;;CAEjC,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,EAAE,CAAC,OAAO,CAAC;;;CAGzC,CAAC,CAAC;AAEH,mDAAmD;AACnD,IAAI,aAAa,GAAG,WAAW,CAAC;AAChC,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC,GAAG,EAA+B,CAAC;AACtG,IAAI,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,MAAM,IAAI,aAAa,EAAE,CAAC;IACxD,aAAa,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;AACtC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,cAAc;IAC5B,MAAM,EAAE,GAAG,aAAa,CAAC;IACzB,aAAa,EAAE,CAAC;IAChB,IAAI,aAAa,GAAG,WAAW,EAAE,CAAC;QAChC,aAAa,GAAG,WAAW,CAAC;IAC9B,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,QAAkB;IAC3C,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAErC,MAAM,IAAI,GAAc;QACtB,EAAE;QACF,MAAM,EAAE,SAAS;QACjB,SAAS,EAAE,QAAQ;QACnB,QAAQ,EAAE,CAAC;QACX,UAAU,EAAE,GAAG;QACf,UAAU,EAAE,GAAG;KAChB,CAAC;IAEF,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IAErG,+EAA+E;IAC/E,UAAU,CAAC,GAAG,EAAE;QACd,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC,EAAE,cAAc,CAAC,CAAC;IAEnB,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,EAAU,EAAE,OAA2B;IAChE,MAAM,IAAI,GAAG,OAAO,CAAC,EAAE,CAAC,CAAC;IACzB,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IAEvB,gBAAgB;IAChB,MAAM,WAAW,GAAG,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;IAElF,cAAc,CAAC,GAAG,CAChB,WAAW,CAAC,MAAM,EAClB,WAAW,CAAC,QAAQ,EACpB,WAAW,CAAC,MAAM,IAAI,IAAI,EAC1B,WAAW,CAAC,KAAK,IAAI,IAAI,EACzB,WAAW,CAAC,YAAY,IAAI,IAAI,EAChC,WAAW,CAAC,UAAU,EACtB,WAAW,CAAC,GAAG,IAAI,IAAI,EACvB,EAAE,CACH,CAAC;IAEF,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,OAAO,CAAC,EAAU;IAChC,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAwB,CAAC;IACvD,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IAEtB,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,MAAM,EAAE,GAAG,CAAC,MAAoB;QAChC,SAAS,EAAE,GAAG,CAAC,SAAqB;QACpC,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,SAAS;QAC/B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,SAAS;QAC7B,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,SAAS;QAC3C,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,GAAG,EAAE,GAAG,CAAC,GAAG,IAAI,SAAS;KAC1B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB;IAC/B,MAAM,IAAI,GAAG,qBAAqB,CAAC,GAAG,EAAe,CAAC;IAEtD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtB,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,MAAM,EAAE,GAAG,CAAC,MAAoB;QAChC,SAAS,EAAE,GAAG,CAAC,SAAqB;QACpC,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,SAAS;QAC/B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,SAAS;QAC7B,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,SAAS;QAC3C,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,GAAG,EAAE,GAAG,CAAC,GAAG,IAAI,SAAS;KAC1B,CAAC,CAAC,CAAC;AACN,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB;IACjC,MAAM,MAAM,GAAG,sBAAsB,CAAC,GAAG,EAAE,CAAC;IAC5C,OAAO,MAAM,CAAC,OAAO,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa;IAC3B,EAAE,CAAC,KAAK,EAAE,CAAC;AACb,CAAC;AAED,2DAA2D;AAC3D,mBAAmB,EAAE,CAAC;AACtB,wBAAwB,EAAE,CAAC;AAE3B,0CAA0C;AAC1C,WAAW,CAAC,GAAG,EAAE;IACf,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;IACtC,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,0BAA0B,OAAO,gBAAgB,CAAC,CAAC;IACnE,CAAC;AACH,CAAC,EAAE,mBAAmB,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gemini-helper-friend",
-  "version": "2.0.0",
+  "version": "2.0.2",
   "description": "Your autonomous AI helper that works on your codebase using Gemini CLI",
   "type": "module",
   "main": "dist/index.js",
@@ -48,6 +48,8 @@
   ],
   "dependencies": {
     "@modelcontextprotocol/sdk": "^0.5.0",
+    "@types/better-sqlite3": "^7.6.13",
+    "better-sqlite3": "^12.5.0",
     "yaml": "^2.3.4",
     "zod": "^3.25.76",
     "zod-to-json-schema": "^3.24.6"