agent-file-stash 0.2.5 → 0.4.0

package/README.md

@@ -112,6 +112,22 @@ filestash status:
   Tokens saved (total):   ~53,851
 ```
 
+When each project keeps its own stash (for example with `FILESTASH_DIR=.vscode/file-stash`), sum all of them with `--all`. It scans the given directory (default: home) for `.file-stash/` and `file-stash/` folders:
+
+```bash
+npx agent-file-stash status --all ~/Projects
+```
+
+```
+filestash status (3 databases under /Users/me/Projects):
+  ~   252,340 tokens    112 files  /Users/me/Projects/api
+  ~    45,157 tokens    128 files  /Users/me/Projects/web
+  ~     7,043 tokens     18 files  /Users/me/Projects/legacy
+  ~   304,540 tokens    258 files  TOTAL
+```
+
+Savings come from re-reads within a session (unchanged files and diffs). A new session always receives full content, since the file is not in its context yet.
+
 #### `help`
 
 ```bash

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;
@@ -456,8 +497,8 @@ function createStash(config) {
 }
 
 // packages/cli/src/index.ts
-import { resolve as resolve4, join as join2 } from "node:path";
-import { existsSync as existsSync3, readFileSync as readFileSync3, writeFileSync } from "node:fs";
+import { resolve as resolve4, join as join3 } from "node:path";
+import { existsSync as existsSync4, readFileSync as readFileSync3, writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 
 // packages/cli/src/mcp.ts
@@ -669,6 +710,33 @@ Use this to verify filestash is working and see how many tokens it has saved.`
   process.on("SIGTERM", shutdown);
 }
 
+// packages/cli/src/scan.ts
+import { existsSync as existsSync3, readdirSync } from "node:fs";
+import { join as join2 } from "node:path";
+var STASH_DIR_NAMES = /* @__PURE__ */ new Set([".file-stash", "file-stash"]);
+var SKIPPED_DIRS = /* @__PURE__ */ new Set(["node_modules", ".git", "build", "dist", "target", "bin", "obj"]);
+var MAX_SCAN_DEPTH = 8;
+function findStashDatabases(root, depth = 0) {
+  if (depth > MAX_SCAN_DEPTH) return [];
+  let entries;
+  try {
+    entries = readdirSync(root, { withFileTypes: true });
+  } catch {
+    return [];
+  }
+  const found = [];
+  for (const entry of entries) {
+    if (!entry.isDirectory() || SKIPPED_DIRS.has(entry.name)) continue;
+    const dir = join2(root, entry.name);
+    if (STASH_DIR_NAMES.has(entry.name) && existsSync3(join2(dir, "stash.db"))) {
+      found.push(join2(dir, "stash.db"));
+      continue;
+    }
+    found.push(...findStashDatabases(dir, depth + 1));
+  }
+  return found;
+}
+
 // packages/cli/src/index.ts
 var _origEmitWarning = process.emitWarning;
 process.emitWarning = function(msg, ...args) {
@@ -678,8 +746,8 @@ process.emitWarning = function(msg, ...args) {
 var CLI_STATUS_SESSION = "cli-status";
 async function runStatus() {
   const stashDir = resolve4(process.env.FILESTASH_DIR ?? ".file-stash");
-  const dbPath = join2(stashDir, "stash.db");
-  if (!existsSync3(dbPath)) {
+  const dbPath = join3(stashDir, "stash.db");
+  if (!existsSync4(dbPath)) {
     console.log("No filestash database found. Run 'filestash serve' to start stashing.");
     process.exit(0);
   }
@@ -691,6 +759,36 @@ async function runStatus() {
   console.log(`  Tokens saved (total):   ~${stats.tokensSaved.toLocaleString()}`);
   await stash.close();
 }
+async function runStatusAll(root) {
+  const databases = findStashDatabases(resolve4(root));
+  if (databases.length === 0) {
+    console.log(`No filestash databases found under ${resolve4(root)}.`);
+    return;
+  }
+  const rows = [];
+  for (const dbPath of databases) {
+    const { stash } = createStash({ dbPath, sessionId: CLI_STATUS_SESSION });
+    try {
+      await stash.init();
+      const stats = await stash.getStats();
+      rows.push({ project: resolve4(dbPath, "../..").replace(/\/\.vscode$/, ""), files: stats.filesTracked, tokens: stats.tokensSaved });
+    } catch (e) {
+      console.error(`  skipped ${dbPath}: ${e instanceof Error ? e.message : String(e)}`);
+    } finally {
+      await stash.close();
+    }
+  }
+  rows.sort((a, b) => b.tokens - a.tokens);
+  const totalTokens = rows.reduce((sum, r) => sum + r.tokens, 0);
+  const totalFiles = rows.reduce((sum, r) => sum + r.files, 0);
+  console.log(`filestash status (${rows.length} databases under ${resolve4(root)}):`);
+  for (const r of rows) {
+    console.log(`  ~${r.tokens.toLocaleString().padStart(10)} tokens  ${String(r.files).padStart(5)} files  ${r.project}`);
+  }
+  console.log(`  ~${totalTokens.toLocaleString().padStart(10)} tokens  ${String(totalFiles).padStart(5)} files  TOTAL`);
+  console.log(`
+Savings count re-reads within a session (unchanged files and diffs); a new session always receives full content.`);
+}
 async function runInit() {
   const home = homedir();
   const mcpServersEntry = {
@@ -701,33 +799,33 @@ async function runInit() {
     type: "local",
     command: ["npx", "agent-file-stash", "serve"]
   };
-  const xdgConfig = process.env.XDG_CONFIG_HOME || join2(home, ".config");
+  const xdgConfig = process.env.XDG_CONFIG_HOME || join3(home, ".config");
   const targets = [
     {
       name: "Claude Code",
-      path: join2(home, ".claude.json"),
+      path: join3(home, ".claude.json"),
       key: "mcpServers",
       entry: mcpServersEntry
     },
     {
       name: "Cursor",
-      path: join2(home, ".cursor", "mcp.json"),
+      path: join3(home, ".cursor", "mcp.json"),
       key: "mcpServers",
       entry: mcpServersEntry
     },
     {
       name: "OpenCode",
-      path: join2(xdgConfig, "opencode", "opencode.json"),
+      path: join3(xdgConfig, "opencode", "opencode.json"),
       key: "mcp",
       entry: opencodeMcpEntry
     }
   ];
   let configured = 0;
   for (const target of targets) {
-    const dir = join2(target.path, "..");
-    if (!existsSync3(dir)) continue;
+    const dir = join3(target.path, "..");
+    if (!existsSync4(dir)) continue;
     let config = {};
-    if (existsSync3(target.path)) {
+    if (existsSync4(target.path)) {
       try {
         config = JSON.parse(readFileSync3(target.path, "utf-8"));
       } catch {
@@ -768,6 +866,8 @@ Usage:
   agent-file-stash init      Auto-configure for your editor
   agent-file-stash serve     Start the MCP server (default)
   agent-file-stash status    Show stash statistics
+  agent-file-stash status --all [dir]
+                             Sum statistics of every stash under dir (default: home)
   agent-file-stash help      Show this help message
 
 Environment:
@@ -776,6 +876,8 @@ Environment:
 var command = process.argv[2];
 if (!command || command === "serve") {
   await startMcpServer();
+} else if (command === "status" && process.argv[3] === "--all") {
+  await runStatusAll(process.argv[4] ?? homedir());
 } else if (command === "status") {
   await runStatus();
 } else if (command === "init") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-file-stash",
-  "version": "0.2.5",
+  "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"