@stackmemoryai/stackmemory 1.2.1 → 1.2.4

package/dist/src/integrations/mcp/handlers/team-handlers.js

@@ -0,0 +1,211 @@
+import { fileURLToPath as __fileURLToPath } from 'url';
+import { dirname as __pathDirname } from 'path';
+const __filename = __fileURLToPath(import.meta.url);
+const __dirname = __pathDirname(__filename);
+import { logger } from "../../../core/monitoring/logger.js";
+class TeamHandlers {
+  constructor(deps) {
+    this.deps = deps;
+  }
+  /**
+   * Get context from other agents working on the same project.
+   * Returns recent frames and shared anchors from other sessions.
+   */
+  async handleTeamContextGet(args) {
+    try {
+      const limit = args.limit ?? 10;
+      const types = args.types;
+      const since = args.since;
+      const projectId = this.getProjectId();
+      const runId = this.getRunId();
+      const frames = await this.deps.dbAdapter.getRecentFramesExcludingRun(
+        projectId,
+        runId,
+        { limit, types, since }
+      );
+      const sharedAnchors = await this.deps.dbAdapter.getSharedAnchors(
+        projectId,
+        { limit, since }
+      );
+      if (frames.length === 0 && sharedAnchors.length === 0) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: "No context from other agents found in this project."
+            }
+          ]
+        };
+      }
+      const frameSummaries = frames.map((f) => ({
+        frame_id: f.frame_id,
+        run_id: f.run_id,
+        name: f.name,
+        type: f.type,
+        state: f.state,
+        created_at: f.created_at,
+        digest_text: f.digest_text || null,
+        anchors: f.anchors.map((a) => ({
+          type: a.type,
+          text: a.text,
+          priority: a.priority
+        }))
+      }));
+      const sharedSummaries = sharedAnchors.map((a) => ({
+        type: a.type,
+        text: a.text,
+        priority: a.priority,
+        frame_name: a.frame_name,
+        run_id: a.run_id,
+        shared_by: a.metadata?.sharedBy
+      }));
+      const text = `Team Context (${frames.length} frames, ${sharedAnchors.length} shared anchors):
+
+` + frameSummaries.map(
+        (f) => `[${f.run_id?.slice(0, 8)}] ${f.name} (${f.type}, ${f.state})` + (f.digest_text ? `
+  ${f.digest_text}` : "") + (f.anchors.length > 0 ? "\n  Anchors: " + f.anchors.map((a) => `${a.type}: ${a.text}`).join("; ") : "")
+      ).join("\n") + (sharedSummaries.length > 0 ? "\n\nShared Anchors:\n" + sharedSummaries.map((a) => `  [${a.type}] ${a.text} (priority: ${a.priority})`).join("\n") : "");
+      return {
+        content: [{ type: "text", text }],
+        metadata: { frames: frameSummaries, sharedAnchors: sharedSummaries }
+      };
+    } catch (error) {
+      logger.error(
+        "Error getting team context",
+        error instanceof Error ? error : new Error(String(error))
+      );
+      throw error;
+    }
+  }
+  /**
+   * Share a piece of context with other agents.
+   * Creates a high-priority anchor tagged with shared=true.
+   */
+  async handleTeamContextShare(args) {
+    try {
+      const { content, type = "FACT", priority = 8 } = args;
+      if (!content) {
+        throw new Error("content is required");
+      }
+      const validTypes = [
+        "FACT",
+        "DECISION",
+        "CONSTRAINT",
+        "INTERFACE_CONTRACT",
+        "TODO",
+        "RISK"
+      ];
+      if (!validTypes.includes(type)) {
+        throw new Error(
+          `Invalid type. Must be one of: ${validTypes.join(", ")}`
+        );
+      }
+      let frameId = this.deps.frameManager.getCurrentFrameId();
+      if (!frameId) {
+        frameId = this.deps.frameManager.createFrame({
+          type: "tool_scope",
+          name: "team_context_share"
+        });
+      }
+      const runId = this.getRunId();
+      this.deps.frameManager.addAnchor(type, content, priority, {
+        shared: true,
+        sharedBy: runId,
+        sharedAt: Date.now()
+      });
+      logger.info("Shared team context", { type, priority });
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Shared ${type}: ${content}`
+          }
+        ]
+      };
+    } catch (error) {
+      logger.error(
+        "Error sharing team context",
+        error instanceof Error ? error : new Error(String(error))
+      );
+      throw error;
+    }
+  }
+  /**
+   * Search across all agents' context in the project.
+   * Uses FTS5 search without run_id filtering.
+   */
+  async handleTeamSearch(args) {
+    try {
+      const { query, limit = 20, include_events = false } = args;
+      if (!query) {
+        throw new Error("query is required");
+      }
+      const projectId = this.getProjectId();
+      const results = await this.deps.dbAdapter.search({
+        query,
+        projectId,
+        limit
+      });
+      if (results.length === 0) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: `No results found for "${query}" across team sessions.`
+            }
+          ]
+        };
+      }
+      const frameResults = results.map((f) => ({
+        frame_id: f.frame_id,
+        run_id: f.run_id,
+        name: f.name,
+        type: f.type,
+        state: f.state,
+        score: f.score,
+        digest_text: f.digest_text || null,
+        created_at: f.created_at,
+        events: []
+      }));
+      if (include_events) {
+        for (const fr of frameResults) {
+          const events = await this.deps.dbAdapter.getFrameEvents(fr.frame_id, {
+            limit: 5
+          });
+          fr.events = events.map((e) => ({
+            event_type: e.event_type,
+            payload: e.payload,
+            ts: e.ts
+          }));
+        }
+      }
+      const text = `Team Search: ${results.length} results for "${query}":
+
+` + frameResults.map(
+        (f) => `[${f.run_id?.slice(0, 8)}] ${f.name} (score: ${f.score?.toFixed(3)})` + (f.digest_text ? `
+  ${f.digest_text}` : "")
+      ).join("\n");
+      return {
+        content: [{ type: "text", text }],
+        metadata: { results: frameResults, total: results.length }
+      };
+    } catch (error) {
+      logger.error(
+        "Error in team search",
+        error instanceof Error ? error : new Error(String(error))
+      );
+      throw error;
+    }
+  }
+  /** Extract projectId from the dbAdapter (protected field on DatabaseAdapter) */
+  getProjectId() {
+    return this.deps.dbAdapter.projectId;
+  }
+  /** Extract current runId from frameManager (private field) */
+  getRunId() {
+    return this.deps.frameManager.currentRunId;
+  }
+}
+export {
+  TeamHandlers
+};

package/dist/src/integrations/mcp/handlers/trace-handlers.js

@@ -73,7 +73,10 @@ ${summaryText}`
       }
       return result;
     } catch (error) {
-      logger.error("Error getting traces", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Error getting traces",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }
@@ -129,9 +132,7 @@ ${summaryText}`
       }
       if (include_recommendations && analysis.recommendations) {
         analysisText += "\n\nRecommendations:\n";
-        analysisText += analysis.recommendations.map(
-          (rec, i) => `${i + 1}. ${rec}`
-        ).join("\n");
+        analysisText += analysis.recommendations.map((rec, i) => `${i + 1}. ${rec}`).join("\n");
       }
       return {
         content: [
@@ -147,7 +148,10 @@ ${summaryText}`
         }
       };
     } catch (error) {
-      logger.error("Error analyzing traces", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Error analyzing traces",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }
@@ -184,7 +188,10 @@ Navigated to: ${url}`
         }
       };
     } catch (error) {
-      logger.error("Error starting browser debug session", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Error starting browser debug session",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }
@@ -218,7 +225,10 @@ Navigated to: ${url}`
         }
       };
     } catch (error) {
-      logger.error("Error taking screenshot", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Error taking screenshot",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }
@@ -251,7 +261,10 @@ Result: ${JSON.stringify(result, null, 2)}`
         }
       };
     } catch (error) {
-      logger.error("Error executing script", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Error executing script",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }
@@ -279,7 +292,10 @@ Result: ${JSON.stringify(result, null, 2)}`
         }
       };
     } catch (error) {
-      logger.error("Error stopping browser debug session", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Error stopping browser debug session",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }

package/dist/src/integrations/mcp/index.js

@@ -2,7 +2,7 @@ import { fileURLToPath as __fileURLToPath } from 'url';
 import { dirname as __pathDirname } from 'path';
 const __filename = __fileURLToPath(import.meta.url);
 const __dirname = __pathDirname(__filename);
-import { RefactoredStackMemoryMCP } from "./refactored-server.js";
+import { StackMemoryMCP } from "./refactored-server.js";
 import { MCPHandlerFactory } from "./handlers/index.js";
 import { ContextHandlers } from "./handlers/context-handlers.js";
 import { TaskHandlers } from "./handlers/task-handlers.js";
@@ -16,7 +16,7 @@ export {
   LinearHandlers,
   MCPHandlerFactory,
   MCPToolDefinitions,
-  RefactoredStackMemoryMCP as StackMemoryMCP,
+  StackMemoryMCP,
   TaskHandlers,
   TraceHandlers
 };

package/dist/src/integrations/mcp/refactored-server.js

@@ -11,13 +11,14 @@ import { existsSync, mkdirSync } from "fs";
 import { join, dirname } from "path";
 import { execSync } from "child_process";
 import { v4 as uuidv4 } from "uuid";
-import { RefactoredFrameManager } from "../../core/context/refactored-frame-manager.js";
+import { FrameManager } from "../../core/context/frame-manager.js";
 import { LinearTaskManager } from "../../features/tasks/linear-task-manager.js";
 import { LinearAuthManager } from "../linear/auth.js";
 import { LinearSyncEngine, DEFAULT_SYNC_CONFIG } from "../linear/sync.js";
 import { BrowserMCPIntegration } from "../../features/browser/browser-mcp.js";
 import { TraceDetector } from "../../core/trace/trace-detector.js";
 import { LLMContextRetrieval } from "../../core/retrieval/index.js";
+import { SQLiteAdapter } from "../../core/database/sqlite-adapter.js";
 import { ConfigManager } from "../../core/config/config-manager.js";
 import { logger } from "../../core/monitoring/logger.js";
 import { MCPHandlerFactory } from "./handlers/index.js";
@@ -34,7 +35,7 @@ function getEnv(key, defaultValue) {
 function getOptionalEnv(key) {
   return process.env[key];
 }
-class RefactoredStackMemoryMCP {
+class StackMemoryMCP {
   server;
   db;
   projectRoot;
@@ -47,6 +48,7 @@ class RefactoredStackMemoryMCP {
   browserMCP;
   traceDetector;
   contextRetrieval;
+  dbAdapter;
   configManager;
   toolScoringMiddleware;
   // Handler factory
@@ -70,6 +72,7 @@ class RefactoredStackMemoryMCP {
     }
     const dbPath = join(dbDir, "context.db");
     this.db = new Database(dbPath);
+    this.dbAdapter = new SQLiteAdapter(this.projectId, { dbPath });
     logger.info("Database initialized", { dbPath });
   }
   /**
@@ -78,7 +81,7 @@ class RefactoredStackMemoryMCP {
   initializeComponents(config) {
     const configPath = join(this.projectRoot, ".stackmemory", "config.yaml");
     this.configManager = new ConfigManager(configPath);
-    this.frameManager = new RefactoredFrameManager(this.db, this.projectId);
+    this.frameManager = new FrameManager(this.db, this.projectId);
     this.taskStore = new LinearTaskManager(this.projectRoot, this.db);
     this.linearAuthManager = new LinearAuthManager(this.projectRoot);
     this.linearSync = new LinearSyncEngine(
@@ -140,7 +143,8 @@ class RefactoredStackMemoryMCP {
       linearAuthManager: this.linearAuthManager,
       linearSync: this.linearSync,
       traceDetector: this.traceDetector,
-      browserMCP: this.browserMCP
+      browserMCP: this.browserMCP,
+      dbAdapter: this.dbAdapter
     };
     this.handlerFactory = new MCPHandlerFactory(dependencies);
     this.toolDefinitions = new MCPToolDefinitions();
@@ -275,6 +279,8 @@ class RefactoredStackMemoryMCP {
    */
   async start() {
     try {
+      await this.dbAdapter.connect();
+      await this.dbAdapter.initializeSchema();
       await this.frameManager.initialize();
       const transport = new StdioServerTransport();
       await this.server.connect(transport);
@@ -285,7 +291,10 @@ class RefactoredStackMemoryMCP {
       });
       this.setupCleanup();
     } catch (error) {
-      logger.error("Failed to start MCP server", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Failed to start MCP server",
+        error instanceof Error ? error : new Error(String(error))
+      );
       throw error;
     }
   }
@@ -299,19 +308,28 @@ class RefactoredStackMemoryMCP {
         if (this.browserMCP) {
           await this.browserMCP.cleanup();
         }
+        if (this.dbAdapter) {
+          await this.dbAdapter.disconnect();
+        }
         if (this.db) {
           this.db.close();
         }
         logger.info("MCP server shutdown complete");
       } catch (error) {
-        logger.error("Error during cleanup", error instanceof Error ? error : new Error(String(error)));
+        logger.error(
+          "Error during cleanup",
+          error instanceof Error ? error : new Error(String(error))
+        );
       }
       process.exit(0);
     };
     process.on("SIGINT", cleanup);
     process.on("SIGTERM", cleanup);
     process.on("uncaughtException", (error) => {
-      logger.error("Uncaught exception", error instanceof Error ? error : new Error(String(error)));
+      logger.error(
+        "Uncaught exception",
+        error instanceof Error ? error : new Error(String(error))
+      );
       cleanup();
     });
   }
@@ -355,10 +373,13 @@ async function main() {
       enableTracing: process.env["DISABLE_TRACING"] !== "true",
       enableBrowser: process.env["DISABLE_BROWSER"] !== "true"
     };
-    const server = new RefactoredStackMemoryMCP(config);
+    const server = new StackMemoryMCP(config);
     await server.start();
   } catch (error) {
-    logger.error("Failed to start server", error instanceof Error ? error : new Error(String(error)));
+    logger.error(
+      "Failed to start server",
+      error instanceof Error ? error : new Error(String(error))
+    );
     process.exit(1);
   }
 }
@@ -369,5 +390,5 @@ if (import.meta.url === `file://${process.argv[1]}`) {
   });
 }
 export {
-  RefactoredStackMemoryMCP
+  StackMemoryMCP
 };

package/dist/src/integrations/mcp/server.js

@@ -30,6 +30,7 @@ import { TraceDetector } from "../../core/trace/trace-detector.js";
 import { LLMContextRetrieval } from "../../core/retrieval/index.js";
 import { DiscoveryHandlers } from "./handlers/discovery-handlers.js";
 import { DiffMemHandlers } from "./handlers/diffmem-handlers.js";
+import { GreptileHandlers } from "./handlers/greptile-handlers.js";
 import { GraphitiClient } from "../graphiti/client.js";
 import { fuzzyEdit } from "../../utils/fuzzy-edit.js";
 import { v4 as uuidv4 } from "uuid";
@@ -64,6 +65,7 @@ class LocalStackMemoryMCP {
   contextRetrieval;
   discoveryHandlers;
   diffMemHandlers;
+  greptileHandlers;
   providerHandlers = null;
   graphitiClient = null;
   pendingPlans = /* @__PURE__ */ new Map();
@@ -138,6 +140,7 @@ class LocalStackMemoryMCP {
       projectRoot: this.projectRoot
     });
     this.diffMemHandlers = new DiffMemHandlers();
+    this.greptileHandlers = new GreptileHandlers();
     this.initProviderHandlers();
     if (process.env.GRAPHITI_ENDPOINT) {
       this.graphitiClient = new GraphitiClient({
@@ -1129,6 +1132,8 @@ ${summary}...`, 0.8);
                 }
               }
             ] : [],
+            // Greptile tools (only active when GREPTILE_API_KEY is set)
+            ...process.env.GREPTILE_API_KEY ? this.greptileHandlers.getToolDefinitions() : [],
             // Provider tools (only active when STACKMEMORY_MULTI_PROVIDER=true)
             ...isFeatureEnabled("multiProvider") ? [
               {
@@ -1377,6 +1382,28 @@ ${summary}...`, 0.8);
             case "diffmem_status":
               result = await this.diffMemHandlers.handleStatus();
               break;
+            // Greptile handlers
+            case "greptile_pr_comments":
+              result = await this.greptileHandlers.handleListPRComments(args);
+              break;
+            case "greptile_pr_details":
+              result = await this.greptileHandlers.handleGetMergeRequest(args);
+              break;
+            case "greptile_list_prs":
+              result = await this.greptileHandlers.handleListPullRequests(args);
+              break;
+            case "greptile_trigger_review":
+              result = await this.greptileHandlers.handleTriggerCodeReview(args);
+              break;
+            case "greptile_search_patterns":
+              result = await this.greptileHandlers.handleSearchPatterns(args);
+              break;
+            case "greptile_create_pattern":
+              result = await this.greptileHandlers.handleCreatePattern(args);
+              break;
+            case "greptile_status":
+              result = await this.greptileHandlers.handleStatus();
+              break;
             case "sm_edit":
               result = await this.handleSmEdit(args);
               break;