agent-file-stash 0.3.0 → 0.4.0

package/dist/cli.mjs

@@ -160,6 +160,11 @@ CREATE TABLE IF NOT EXISTS session_reads (
   PRIMARY KEY (session_id, path)
 );
 
+CREATE TABLE IF NOT EXISTS sessions (
+  session_id  TEXT PRIMARY KEY,
+  pid         INTEGER NOT NULL
+);
+
 CREATE TABLE IF NOT EXISTS stats (
   key   TEXT PRIMARY KEY,
   value INTEGER NOT NULL DEFAULT 0
@@ -181,6 +186,15 @@ var HASH_LENGTH = 16;
 function contentHash(content) {
   return createHash("sha256").update(content).digest("hex").slice(0, HASH_LENGTH);
 }
+function isProcessAlive(pid) {
+  if (!Number.isInteger(pid) || pid <= 0) return false;
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch (err) {
+    return err.code === "EPERM";
+  }
+}
 function queryOne(db, sql, params = []) {
   const rows = db.prepare(sql).all(...params);
   return rows[0];
@@ -199,9 +213,35 @@ var StashStore = class {
     const { DatabaseSync } = await import("node:sqlite");
     this.db = new DatabaseSync(this.dbPath);
     this.db.exec("PRAGMA journal_mode=WAL");
+    this.db.exec("PRAGMA busy_timeout=5000");
     this.db.exec(SCHEMA);
+    this.registerSession(this.db);
+    this.pruneClosedSessions(this.db);
     this.initialized = true;
   }
+  registerSession(db) {
+    db.prepare("INSERT OR REPLACE INTO sessions (session_id, pid) VALUES (?, ?)").run(this.sessionId, process.pid);
+  }
+  pruneClosedSessions(db) {
+    db.exec("BEGIN IMMEDIATE");
+    try {
+      const sessions = db.prepare("SELECT session_id, pid FROM sessions").all();
+      for (const s of sessions) {
+        if (s.session_id !== this.sessionId && !isProcessAlive(s.pid)) {
+          db.prepare("DELETE FROM sessions WHERE session_id = ?").run(s.session_id);
+        }
+      }
+      db.exec("DELETE FROM session_reads WHERE session_id NOT IN (SELECT session_id FROM sessions)");
+      db.exec("DELETE FROM session_stats WHERE session_id NOT IN (SELECT session_id FROM sessions)");
+      db.exec(
+        "DELETE FROM file_versions WHERE NOT EXISTS (SELECT 1 FROM session_reads r WHERE r.path = file_versions.path AND r.hash = file_versions.hash)"
+      );
+      db.exec("COMMIT");
+    } catch (err) {
+      db.exec("ROLLBACK");
+      throw err;
+    }
+  }
   getDb() {
     if (!this.db) throw new Error("StashStore not initialized. Call init() first.");
     return this.db;
@@ -373,6 +413,7 @@ var StashStore = class {
   }
   async close() {
     if (this.db) {
+      this.db.prepare("DELETE FROM sessions WHERE session_id = ?").run(this.sessionId);
       this.db.close();
       this.db = null;
       this.initialized = false;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-file-stash",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "mcpName": "io.github.atilio-ts/agent-file-stash",
   "description": "File stash with diff tracking for AI coding agents. Drop-in replacement for file reads that saves tokens.",
   "type": "module",
@@ -55,7 +55,7 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "https://github.com/atilio-ts/agent-file-stash"
+    "url": "git+https://github.com/atilio-ts/agent-file-stash.git"
   },
   "homepage": "https://github.com/atilio-ts/agent-file-stash",
   "packageManager": "pnpm@10.29.3"