@grec0/memory-bank-mcp 0.1.21 → 0.1.23

package/dist/tools/searchMemory.js

@@ -4,6 +4,7 @@
  */
 import { AgentBoard } from "../common/agentBoard.js";
 import { sessionLogger } from "../common/sessionLogger.js";
+import { sessionState } from "../common/sessionState.js";
 /**
  * Searches the Memory Bank for relevant code
  */
@@ -36,11 +37,12 @@ export async function searchMemory(params, indexManager, workspaceRoot) {
             filterByFile: params.filterByFile,
             filterByLanguage: params.filterByLanguage,
         });
-        // Session Logging
-        if (params.agentId && workspaceRoot) {
+        // Session Logging via Session State
+        const activeAgentId = sessionState.getCurrentAgentId();
+        if (activeAgentId && workspaceRoot) {
             try {
                 const board = new AgentBoard(workspaceRoot, params.projectId);
-                const sessionId = await board.getSessionId(params.agentId);
+                const sessionId = await board.getSessionId(activeAgentId);
                 if (sessionId) {
                     await sessionLogger.logSessionEvent(params.projectId, sessionId, {
                         timestamp: new Date().toISOString(),
@@ -54,7 +56,7 @@ export async function searchMemory(params, indexManager, workspaceRoot) {
                             resultCount: results.length,
                             topResults: results.slice(0, 5).map(r => ({ path: r.filePath, score: r.score }))
                         }
-                    });
+                    }, activeAgentId);
                 }
             }
             catch (logError) {

package/dist/tools/writeFile.js

@@ -4,6 +4,9 @@
  */
 import * as fs from "fs";
 import * as path from "path";
+import { AgentBoard } from "../common/agentBoard.js";
+import { sessionLogger } from "../common/sessionLogger.js";
+import { sessionState } from "../common/sessionState.js";
 /**
  * Writes a file and optionally reindexes it
  */
@@ -13,6 +16,40 @@ export async function writeFile(params, indexManager, workspaceRoot) {
         const filePath = path.isAbsolute(params.path)
             ? params.path
             : path.join(workspaceRoot, params.path);
+        // Auto-Locking & Logging via Session State
+        const activeAgentId = sessionState.getCurrentAgentId();
+        if (activeAgentId) {
+            const board = new AgentBoard(workspaceRoot, params.projectId);
+            // 1. Check/Claim Lock
+            // Normalize path for locking (relative to workspace)
+            const relativePath = path.relative(workspaceRoot, filePath).replace(/\\/g, '/');
+            const canWrite = await board.claimResource(activeAgentId, relativePath);
+            if (!canWrite) {
+                return {
+                    success: false,
+                    filePath: params.path,
+                    message: `File is locked by another agent. Write rejected.`
+                };
+            }
+            // 2. Log Session Event
+            try {
+                const sessionId = await board.getSessionId(activeAgentId);
+                if (sessionId) {
+                    await sessionLogger.logSessionEvent(params.projectId, sessionId, {
+                        timestamp: new Date().toISOString(),
+                        type: 'decision',
+                        data: {
+                            action: 'write_file',
+                            path: params.path,
+                            byteSize: Buffer.byteLength(params.content, "utf-8")
+                        }
+                    });
+                }
+            }
+            catch (e) {
+                console.error("Failed to log session event:", e);
+            }
+        }
         // Ensure directory exists
         const dir = path.dirname(filePath);
         if (!fs.existsSync(dir)) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@grec0/memory-bank-mcp",
-  "version": "0.1.21",
+  "version": "0.1.23",
   "description": "MCP server for semantic code indexing with Memory Bank - AI-powered codebase understanding",
   "license": "MIT",
   "author": "@grec0",
@@ -38,7 +38,7 @@
     "@babel/parser": "^7.23.0",
     "@babel/traverse": "^7.23.0",
     "@lancedb/lancedb": "^0.9.0",
-    "@modelcontextprotocol/sdk": "1.6.1",
+    "@modelcontextprotocol/sdk": "1.25.2",
     "@types/node": "^22",
     "gpt-tokenizer": "3.4.0",
     "ignore": "^5.3.0",
@@ -46,12 +46,14 @@
     "tree-sitter-wasms": "0.1.13",
     "web-tree-sitter": "0.26.3",
     "zod": "^3.22.4",
-    "zod-to-json-schema": "^3.23.5"
+    "zod-to-json-schema": "^3.23.5",
+    "better-sqlite3": "^11.7.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.7.0",
     "@types/babel__parser": "^7.1.1",
     "@types/babel__traverse": "^7.20.4",
+    "@types/better-sqlite3": "^7.6.12",
     "@types/jest": "^29.5.14",
     "jest": "^29.7.0",
     "shx": "^0.3.4",