@brainbank/mcp 0.2.0 → 0.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +88 -42
- package/dist/mcp-server.js +301 -315
- package/dist/mcp-server.js.map +1 -1
- package/package.json +15 -3
- package/src/mcp-server.ts +265 -0
- package/src/workspace-factory.ts +66 -0
- package/src/workspace-pool.ts +224 -0
package/dist/mcp-server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/mcp-server.ts"],"sourcesContent":["#!/usr/bin/env node\n\n/**\n * BrainBank — MCP Server\n * \n * Exposes BrainBank as an MCP server via stdio transport.\n * Works with Google Antigravity, Claude Desktop, and any MCP-compatible client.\n * \n * Usage in Antigravity mcp_config.json:\n * {\n * \"mcpServers\": {\n * \"brainbank\": {\n * \"command\": \"npx\",\n * \"args\": [\"tsx\", \"/path/to/brainbank/packages/mcp/src/mcp-server.ts\"],\n * \"env\": { \"BRAINBANK_REPO\": \"/path/to/your/repo\" }\n * }\n * }\n * }\n * \n * Tools (6):\n * brainbank_search — Unified search (hybrid, vector, or keyword mode)\n * brainbank_context — Formatted knowledge context for a task\n * brainbank_index — Trigger code/git/docs indexing\n * brainbank_stats — Index statistics\n * brainbank_history — Git history for a specific file\n * brainbank_collection — KV collection operations (add, search, trim)\n */\n\nimport { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';\nimport { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';\nimport { z } from 'zod/v3';\nimport * as fs from 'node:fs';\nimport * as path from 'node:path';\nimport { BrainBank } from 'brainbank';\nimport { code } from 'brainbank/code';\nimport { git } from 'brainbank/git';\nimport { docs } from 'brainbank/docs';\n\n// ── Configuration from env ──────────────────────────\n\n/**\n * Detect repo root by walking up from startDir until we find `.git/`.\n * Returns startDir itself if no `.git/` is found (mono-repo or non-git project).\n */\nfunction findRepoRoot(startDir: string): string {\n let dir = path.resolve(startDir);\n while (true) {\n if (fs.existsSync(path.join(dir, '.git'))) return dir;\n const parent = path.dirname(dir);\n if (parent === dir) break; // filesystem root\n dir = parent;\n }\n return path.resolve(startDir); // fallback: use startDir as-is\n}\n\nconst defaultRepoPath = process.env.BRAINBANK_REPO || undefined;\n\n// ── Reranker (default: none, set BRAINBANK_RERANKER=qwen3 to enable) ──\n\nasync function createReranker() {\n const rerankerEnv = process.env.BRAINBANK_RERANKER ?? 'none';\n if (rerankerEnv === 'none') return undefined;\n if (rerankerEnv === 'qwen3') {\n const { Qwen3Reranker } = await import('@brainbank/reranker');\n return new Qwen3Reranker();\n }\n return undefined;\n}\n\n// ── Embedding Provider (default: local) ──\n\nasync function createEmbeddingProvider() {\n const embeddingEnv = process.env.BRAINBANK_EMBEDDING ?? 'local';\n if (embeddingEnv === 'openai') {\n const { OpenAIEmbedding } = await import('brainbank');\n return new OpenAIEmbedding();\n }\n if (embeddingEnv === 'perplexity') {\n const { PerplexityEmbedding } = await import('brainbank');\n return new PerplexityEmbedding();\n }\n if (embeddingEnv === 'perplexity-context') {\n const { PerplexityContextEmbedding } = await import('brainbank');\n return new PerplexityContextEmbedding();\n }\n return undefined; // BrainBank defaults to local WASM\n}\n\n// ── Multi-Workspace BrainBank Pool ─────────────────────\n\nconst MAX_POOL_SIZE = 10;\n\ninterface PoolEntry {\n brain: BrainBank;\n lastAccess: number;\n}\n\nconst _pool = new Map<string, PoolEntry>();\nlet _sharedReranker: any = undefined;\nlet _sharedEmbedding: any = undefined;\nlet _sharedReady = false;\n\nasync function ensureShared() {\n if (_sharedReady) return;\n _sharedReranker = await createReranker();\n _sharedEmbedding = await createEmbeddingProvider();\n _sharedReady = true;\n}\n\nasync function getBrainBank(targetRepo?: string): Promise<BrainBank> {\n const rp = targetRepo ?? defaultRepoPath;\n if (!rp) {\n throw new Error(\n 'No repository specified. Pass the `repo` parameter with the workspace path, ' +\n 'or set BRAINBANK_REPO environment variable.'\n );\n }\n const resolved = rp.replace(/\\/+$/, '');\n\n if (_pool.has(resolved)) {\n const entry = _pool.get(resolved)!;\n try {\n const codeStats = entry.brain.indexer('code')?.stats?.();\n if (codeStats && codeStats.hnswSize === 0) {\n const dbPath = path.join(resolved, '.brainbank', 'brainbank.db');\n const dbSize = fs.existsSync(dbPath) ? fs.statSync(dbPath).size : 0;\n if (dbSize > 100_000) {\n evictPool(resolved);\n } else {\n entry.lastAccess = Date.now();\n return entry.brain;\n }\n } else {\n entry.lastAccess = Date.now();\n return entry.brain;\n }\n } catch {\n entry.lastAccess = Date.now();\n return entry.brain;\n }\n }\n\n await ensureShared();\n\n if (_pool.size >= MAX_POOL_SIZE) {\n let oldest: string | undefined;\n let oldestTime = Infinity;\n for (const [key, entry] of _pool) {\n if (entry.lastAccess < oldestTime) {\n oldestTime = entry.lastAccess;\n oldest = key;\n }\n }\n if (oldest) evictPool(oldest);\n }\n\n const brain = await _createBrain(resolved);\n _pool.set(resolved, { brain, lastAccess: Date.now() });\n return brain;\n}\n\nasync function _createBrain(resolved: string): Promise<BrainBank> {\n const opts: Record<string, any> = { repoPath: resolved, reranker: _sharedReranker };\n if (_sharedEmbedding) {\n opts.embeddingProvider = _sharedEmbedding;\n opts.embeddingDims = _sharedEmbedding.dims;\n }\n const brain = new BrainBank(opts)\n .use(code({ repoPath: resolved }))\n .use(git({ repoPath: resolved }))\n .use(docs());\n\n try {\n await brain.initialize();\n } catch (err: any) {\n if (err?.message?.includes('Invalid the given array length')) {\n const dbPath = path.join(resolved, '.brainbank', 'brainbank.db');\n try { fs.unlinkSync(dbPath); } catch {}\n try { fs.unlinkSync(dbPath + '-wal'); } catch {}\n try { fs.unlinkSync(dbPath + '-shm'); } catch {}\n\n const fresh = new BrainBank(opts)\n .use(code({ repoPath: resolved }))\n .use(git({ repoPath: resolved }))\n .use(docs());\n await fresh.initialize();\n return fresh;\n }\n throw err;\n }\n\n return brain;\n}\n\nfunction evictPool(resolved: string) {\n const entry = _pool.get(resolved);\n if (entry) {\n try { entry.brain.close(); } catch {}\n _pool.delete(resolved);\n }\n}\n\n// ── MCP Server Setup ────────────────────────────────\n\nconst server = new McpServer({\n name: 'brainbank',\n version: '0.2.0',\n});\n\n// ── Tool: brainbank_search ──────────────────────────\n// Replaces: brainbank_search, brainbank_hybrid_search, brainbank_keyword_search\n\nserver.registerTool(\n 'brainbank_search',\n {\n title: 'BrainBank Search',\n description:\n 'Search indexed code and git commits. Supports three modes:\\n' +\n '- hybrid (default): vector + BM25 fused with RRF — best quality\\n' +\n '- vector: semantic similarity only\\n' +\n '- keyword: instant BM25 for exact terms, function names, error messages',\n inputSchema: z.object({\n query: z.string().describe('Search query — works with both keywords and natural language'),\n mode: z.enum(['hybrid', 'vector', 'keyword']).optional().default('hybrid').describe('Search strategy'),\n codeK: z.number().optional().default(8).describe('Max code results'),\n gitK: z.number().optional().default(5).describe('Max git results'),\n minScore: z.number().optional().default(0.25).describe('Minimum similarity score (0-1), only for vector mode'),\n collections: z.record(z.string(), z.number()).optional().describe(\n 'Max results per source. Reserved: \"code\", \"git\", \"docs\". Any other key = KV collection.'\n ),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ query, mode, codeK, gitK, minScore, collections, repo }) => {\n const brainbank = await getBrainBank(repo);\n\n let results;\n if (mode === 'keyword') {\n results = await brainbank.searchBM25(query, { codeK, gitK });\n } else if (mode === 'vector') {\n results = await brainbank.search(query, { codeK, gitK, minScore });\n } else {\n results = await brainbank.hybridSearch(query, { codeK, gitK, collections });\n }\n\n if (results.length === 0) {\n return { content: [{ type: 'text', text: 'No results found.' }] };\n }\n\n const modeLabel = mode === 'keyword' ? 'Keyword (BM25)' : mode === 'vector' ? 'Vector' : 'Hybrid (Vector + BM25 → RRF)';\n return { content: [{ type: 'text', text: formatResults(results, modeLabel) }] };\n },\n);\n\n// ── Tool: brainbank_context ─────────────────────────\n\nserver.registerTool(\n 'brainbank_context',\n {\n title: 'BrainBank Context',\n description: 'Get a formatted knowledge context block for a task. Returns relevant code, git history, and co-edit patterns as markdown.',\n inputSchema: z.object({\n task: z.string().describe('Description of the task you need context for'),\n affectedFiles: z.array(z.string()).optional().default([]).describe('Files you plan to modify (improves co-edit suggestions)'),\n codeResults: z.number().optional().default(6).describe('Max code results'),\n gitResults: z.number().optional().default(5).describe('Max git commit results'),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ task, affectedFiles, codeResults, gitResults, repo }) => {\n const brainbank = await getBrainBank(repo);\n const context = await brainbank.getContext(task, {\n affectedFiles,\n codeResults,\n gitResults,\n });\n\n return { content: [{ type: 'text', text: context }] };\n },\n);\n\n// ── Tool: brainbank_index ───────────────────────────\n\nserver.registerTool(\n 'brainbank_index',\n {\n title: 'BrainBank Index',\n description: 'Index (or re-index) code, git history, and docs. Incremental — only changed files are processed.',\n inputSchema: z.object({\n modules: z.array(z.enum(['code', 'git', 'docs'])).optional().describe('Which modules to index (default: all)'),\n docsPath: z.string().optional().describe('Path to a docs folder to register and index'),\n forceReindex: z.boolean().optional().default(false).describe('Force re-index of all files'),\n gitDepth: z.number().optional().default(500).describe('Number of git commits to index'),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ modules, docsPath, forceReindex, gitDepth, repo }) => {\n const brainbank = await getBrainBank(repo);\n\n if (docsPath) {\n const absPath = path.resolve(docsPath);\n const collName = path.basename(absPath);\n try {\n brainbank.addCollection({\n name: collName,\n path: absPath,\n pattern: '**/*.md',\n ignore: ['deprecated/**', 'node_modules/**'],\n });\n } catch {\n // docs module not loaded\n }\n }\n\n const result = await brainbank.index({ modules, forceReindex, gitDepth });\n\n const lines = [\n '## Indexing Complete',\n '',\n `**Code**: ${result.code?.indexed ?? 0} files indexed, ${result.code?.skipped ?? 0} skipped, ${result.code?.chunks ?? 0} chunks`,\n `**Git**: ${result.git?.indexed ?? 0} commits indexed, ${result.git?.skipped ?? 0} skipped`,\n ];\n\n if (result.docs) {\n for (const [name, stat] of Object.entries(result.docs)) {\n lines.push(`**Docs [${name}]**: ${stat.indexed} indexed, ${stat.skipped} skipped, ${stat.chunks} chunks`);\n }\n }\n\n const stats = brainbank.stats();\n lines.push('');\n lines.push(`**Totals**: ${stats.code?.chunks ?? 0} code chunks, ${stats.git?.commits ?? 0} commits, ${stats.documents?.documents ?? 0} docs`);\n\n return { content: [{ type: 'text', text: lines.join('\\n') }] };\n },\n);\n\n// ── Tool: brainbank_stats ───────────────────────────\n\nserver.registerTool(\n 'brainbank_stats',\n {\n title: 'BrainBank Stats',\n description: 'Get index statistics: file count, code chunks, git commits, HNSW sizes, KV collections.',\n inputSchema: z.object({\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ repo }) => {\n const brainbank = await getBrainBank(repo);\n const s = brainbank.stats();\n\n const lines = ['## BrainBank Stats', ''];\n\n if (s.code) {\n lines.push(`**Code**: ${s.code.files} files, ${s.code.chunks} chunks, ${s.code.hnswSize} vectors`);\n }\n if (s.git) {\n lines.push(`**Git**: ${s.git.commits} commits, ${s.git.filesTracked} files, ${s.git.coEdits} co-edit pairs`);\n }\n if (s.documents) {\n lines.push(`**Docs**: ${s.documents.collections} collections, ${s.documents.documents} documents`);\n }\n\n const kvNames = brainbank.listCollectionNames();\n if (kvNames.length > 0) {\n lines.push('');\n lines.push('**KV Collections**:');\n for (const name of kvNames) {\n const coll = brainbank.collection(name);\n lines.push(`- ${name}: ${coll.count()} items`);\n }\n }\n\n return { content: [{ type: 'text', text: lines.join('\\n') }] };\n },\n);\n\n// ── Tool: brainbank_history ─────────────────────────\n\nserver.registerTool(\n 'brainbank_history',\n {\n title: 'BrainBank File History',\n description: 'Get git commit history for a file. Shows changes, authors, and line counts.',\n inputSchema: z.object({\n filePath: z.string().describe('File path (relative or partial, e.g. \"auth.ts\")'),\n limit: z.number().optional().default(20).describe('Max commits to return'),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ filePath, limit, repo }) => {\n const brainbank = await getBrainBank(repo);\n const history = await brainbank.fileHistory(filePath, limit);\n\n if (history.length === 0) {\n return { content: [{ type: 'text', text: `No git history found for \"${filePath}\"` }] };\n }\n\n const lines = [`## Git History: ${filePath}`, ''];\n for (const h of history as any[]) {\n lines.push(`**[${h.short_hash}]** ${h.message} *(${h.author}, +${h.additions}/-${h.deletions})*`);\n }\n\n return { content: [{ type: 'text', text: lines.join('\\n') }] };\n },\n);\n\n// ── Tool: brainbank_collection ──────────────────────\n// Replaces: brainbank_collection_add, brainbank_collection_search, brainbank_collection_trim\n\nserver.registerTool(\n 'brainbank_collection',\n {\n title: 'BrainBank Collection',\n description:\n 'Operate on KV collections (auto-created). Actions:\\n' +\n '- add: store content with optional metadata\\n' +\n '- search: hybrid vector + keyword search\\n' +\n '- trim: keep only N most recent items',\n inputSchema: z.object({\n action: z.enum(['add', 'search', 'trim']).describe('Operation to perform'),\n collection: z.string().describe('Collection name (e.g. \"errors\", \"decisions\")'),\n content: z.string().optional().describe('Content to store (required for add)'),\n query: z.string().optional().describe('Search query (required for search)'),\n metadata: z.record(z.any()).optional().default({}).describe('Metadata for add'),\n k: z.number().optional().default(5).describe('Max results for search'),\n keep: z.number().optional().describe('Items to keep for trim'),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ action, collection, content, query, metadata, k, keep, repo }) => {\n const brainbank = await getBrainBank(repo);\n const coll = brainbank.collection(collection);\n\n if (action === 'add') {\n if (!content) throw new Error('BrainBank: content is required for add action.');\n const id = await coll.add(content, metadata);\n return {\n content: [{ type: 'text', text: `✓ Item #${id} added to '${collection}' (${coll.count()} total)` }],\n };\n }\n\n if (action === 'search') {\n if (!query) throw new Error('BrainBank: query is required for search action.');\n const results = await coll.search(query, { k });\n\n if (results.length === 0) {\n return { content: [{ type: 'text', text: `No results in '${collection}' for \"${query}\"` }] };\n }\n\n const lines = [`## Collection: ${collection}`, ''];\n for (const r of results) {\n const score = Math.round((r.score ?? 0) * 100);\n lines.push(`[${score}%] ${r.content}`);\n if (Object.keys(r.metadata).length > 0) {\n lines.push(` ${JSON.stringify(r.metadata)}`);\n }\n lines.push('');\n }\n return { content: [{ type: 'text', text: lines.join('\\n') }] };\n }\n\n if (action === 'trim') {\n if (keep == null) throw new Error('BrainBank: keep is required for trim action.');\n const result = await coll.trim({ keep });\n return {\n content: [{ type: 'text', text: `✓ Trimmed ${result.removed} items from '${collection}' (kept ${keep})` }],\n };\n }\n\n throw new Error(`BrainBank: Unknown action \"${action}\".`);\n },\n);\n\n// ── Shared result formatter ─────────────────────────\n\nfunction formatResults(results: any[], mode: string): string {\n const lines: string[] = [`## ${mode}`, ''];\n for (const r of results) {\n const score = Math.round(r.score * 100);\n if (r.type === 'code') {\n const m = r.metadata;\n lines.push(`[CODE ${score}%] ${r.filePath} — ${m.name || m.chunkType} (L${m.startLine}-${m.endLine})`);\n lines.push(r.content);\n lines.push('');\n } else if (r.type === 'commit') {\n const m = r.metadata;\n lines.push(`[COMMIT ${score}%] ${m.shortHash} — ${r.content} (${m.author})`);\n if (m.files?.length) lines.push(` Files: ${m.files.join(', ')}`);\n lines.push('');\n } else if (r.type === 'document') {\n const ctx = r.context ? ` — ${r.context}` : '';\n lines.push(`[DOC ${score}%] ${r.filePath} [${r.metadata.collection}]${ctx}`);\n lines.push(r.content);\n lines.push('');\n } else if (r.type === 'collection') {\n const col = r.metadata?.collection ?? 'unknown';\n lines.push(`[COLLECTION ${score}%] [${col}]`);\n lines.push(r.content);\n lines.push('');\n }\n }\n return lines.join('\\n');\n}\n\n// ── Start Server ────────────────────────────────────\n\nasync function main() {\n const transport = new StdioServerTransport();\n await server.connect(transport);\n}\n\nmain().catch(err => {\n console.error(`BrainBank MCP Server Error: ${err.message}`);\n process.exit(1);\n});\n"],"mappings":";;;;;AA4BA,SAAS,iBAAiB;AAC1B,SAAS,4BAA4B;AACrC,SAAS,SAAS;AAClB,YAAY,QAAQ;AACpB,YAAY,UAAU;AACtB,SAAS,iBAAiB;AAC1B,SAAS,YAAY;AACrB,SAAS,WAAW;AACpB,SAAS,YAAY;AAmBrB,IAAM,kBAAkB,QAAQ,IAAI,kBAAkB;AAItD,eAAe,iBAAiB;AAC5B,QAAM,cAAc,QAAQ,IAAI,sBAAsB;AACtD,MAAI,gBAAgB,OAAQ,QAAO;AACnC,MAAI,gBAAgB,SAAS;AACzB,UAAM,EAAE,cAAc,IAAI,MAAM,OAAO,qBAAqB;AAC5D,WAAO,IAAI,cAAc;AAAA,EAC7B;AACA,SAAO;AACX;AARe;AAYf,eAAe,0BAA0B;AACrC,QAAM,eAAe,QAAQ,IAAI,uBAAuB;AACxD,MAAI,iBAAiB,UAAU;AAC3B,UAAM,EAAE,gBAAgB,IAAI,MAAM,OAAO,WAAW;AACpD,WAAO,IAAI,gBAAgB;AAAA,EAC/B;AACA,MAAI,iBAAiB,cAAc;AAC/B,UAAM,EAAE,oBAAoB,IAAI,MAAM,OAAO,WAAW;AACxD,WAAO,IAAI,oBAAoB;AAAA,EACnC;AACA,MAAI,iBAAiB,sBAAsB;AACvC,UAAM,EAAE,2BAA2B,IAAI,MAAM,OAAO,WAAW;AAC/D,WAAO,IAAI,2BAA2B;AAAA,EAC1C;AACA,SAAO;AACX;AAfe;AAmBf,IAAM,gBAAgB;AAOtB,IAAM,QAAQ,oBAAI,IAAuB;AACzC,IAAI,kBAAuB;AAC3B,IAAI,mBAAwB;AAC5B,IAAI,eAAe;AAEnB,eAAe,eAAe;AAC1B,MAAI,aAAc;AAClB,oBAAkB,MAAM,eAAe;AACvC,qBAAmB,MAAM,wBAAwB;AACjD,iBAAe;AACnB;AALe;AAOf,eAAe,aAAa,YAAyC;AACjE,QAAM,KAAK,cAAc;AACzB,MAAI,CAAC,IAAI;AACL,UAAM,IAAI;AAAA,MACN;AAAA,IAEJ;AAAA,EACJ;AACA,QAAM,WAAW,GAAG,QAAQ,QAAQ,EAAE;AAEtC,MAAI,MAAM,IAAI,QAAQ,GAAG;AACrB,UAAM,QAAQ,MAAM,IAAI,QAAQ;AAChC,QAAI;AACA,YAAM,YAAY,MAAM,MAAM,QAAQ,MAAM,GAAG,QAAQ;AACvD,UAAI,aAAa,UAAU,aAAa,GAAG;AACvC,cAAM,SAAc,UAAK,UAAU,cAAc,cAAc;AAC/D,cAAM,SAAY,cAAW,MAAM,IAAO,YAAS,MAAM,EAAE,OAAO;AAClE,YAAI,SAAS,KAAS;AAClB,oBAAU,QAAQ;AAAA,QACtB,OAAO;AACH,gBAAM,aAAa,KAAK,IAAI;AAC5B,iBAAO,MAAM;AAAA,QACjB;AAAA,MACJ,OAAO;AACH,cAAM,aAAa,KAAK,IAAI;AAC5B,eAAO,MAAM;AAAA,MACjB;AAAA,IACJ,QAAQ;AACJ,YAAM,aAAa,KAAK,IAAI;AAC5B,aAAO,MAAM;AAAA,IACjB;AAAA,EACJ;AAEA,QAAM,aAAa;AAEnB,MAAI,MAAM,QAAQ,eAAe;AAC7B,QAAI;AACJ,QAAI,aAAa;AACjB,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO;AAC9B,UAAI,MAAM,aAAa,YAAY;AAC/B,qBAAa,MAAM;AACnB,iBAAS;AAAA,MACb;AAAA,IACJ;AACA,QAAI,OAAQ,WAAU,MAAM;AAAA,EAChC;AAEA,QAAM,QAAQ,MAAM,aAAa,QAAQ;AACzC,QAAM,IAAI,UAAU,EAAE,OAAO,YAAY,KAAK,IAAI,EAAE,CAAC;AACrD,SAAO;AACX;AAlDe;AAoDf,eAAe,aAAa,UAAsC;AAC9D,QAAM,OAA4B,EAAE,UAAU,UAAU,UAAU,gBAAgB;AAClF,MAAI,kBAAkB;AAClB,SAAK,oBAAoB;AACzB,SAAK,gBAAgB,iBAAiB;AAAA,EAC1C;AACA,QAAM,QAAQ,IAAI,UAAU,IAAI,EAC3B,IAAI,KAAK,EAAE,UAAU,SAAS,CAAC,CAAC,EAChC,IAAI,IAAI,EAAE,UAAU,SAAS,CAAC,CAAC,EAC/B,IAAI,KAAK,CAAC;AAEf,MAAI;AACA,UAAM,MAAM,WAAW;AAAA,EAC3B,SAAS,KAAU;AACf,QAAI,KAAK,SAAS,SAAS,gCAAgC,GAAG;AAC1D,YAAM,SAAc,UAAK,UAAU,cAAc,cAAc;AAC/D,UAAI;AAAE,QAAG,cAAW,MAAM;AAAA,MAAG,QAAQ;AAAA,MAAC;AACtC,UAAI;AAAE,QAAG,cAAW,SAAS,MAAM;AAAA,MAAG,QAAQ;AAAA,MAAC;AAC/C,UAAI;AAAE,QAAG,cAAW,SAAS,MAAM;AAAA,MAAG,QAAQ;AAAA,MAAC;AAE/C,YAAM,QAAQ,IAAI,UAAU,IAAI,EAC3B,IAAI,KAAK,EAAE,UAAU,SAAS,CAAC,CAAC,EAChC,IAAI,IAAI,EAAE,UAAU,SAAS,CAAC,CAAC,EAC/B,IAAI,KAAK,CAAC;AACf,YAAM,MAAM,WAAW;AACvB,aAAO;AAAA,IACX;AACA,UAAM;AAAA,EACV;AAEA,SAAO;AACX;AA/Be;AAiCf,SAAS,UAAU,UAAkB;AACjC,QAAM,QAAQ,MAAM,IAAI,QAAQ;AAChC,MAAI,OAAO;AACP,QAAI;AAAE,YAAM,MAAM,MAAM;AAAA,IAAG,QAAQ;AAAA,IAAC;AACpC,UAAM,OAAO,QAAQ;AAAA,EACzB;AACJ;AANS;AAUT,IAAM,SAAS,IAAI,UAAU;AAAA,EACzB,MAAM;AAAA,EACN,SAAS;AACb,CAAC;AAKD,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aACI;AAAA,IAIJ,aAAa,EAAE,OAAO;AAAA,MAClB,OAAO,EAAE,OAAO,EAAE,SAAS,mEAA8D;AAAA,MACzF,MAAM,EAAE,KAAK,CAAC,UAAU,UAAU,SAAS,CAAC,EAAE,SAAS,EAAE,QAAQ,QAAQ,EAAE,SAAS,iBAAiB;AAAA,MACrG,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,SAAS,kBAAkB;AAAA,MACnE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,SAAS,iBAAiB;AAAA,MACjE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,IAAI,EAAE,SAAS,sDAAsD;AAAA,MAC7G,aAAa,EAAE,OAAO,EAAE,OAAO,GAAG,EAAE,OAAO,CAAC,EAAE,SAAS,EAAE;AAAA,QACrD;AAAA,MACJ;AAAA,MACA,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,OAAO,MAAM,OAAO,MAAM,UAAU,aAAa,KAAK,MAAM;AACjE,UAAM,YAAY,MAAM,aAAa,IAAI;AAEzC,QAAI;AACJ,QAAI,SAAS,WAAW;AACpB,gBAAU,MAAM,UAAU,WAAW,OAAO,EAAE,OAAO,KAAK,CAAC;AAAA,IAC/D,WAAW,SAAS,UAAU;AAC1B,gBAAU,MAAM,UAAU,OAAO,OAAO,EAAE,OAAO,MAAM,SAAS,CAAC;AAAA,IACrE,OAAO;AACH,gBAAU,MAAM,UAAU,aAAa,OAAO,EAAE,OAAO,MAAM,YAAY,CAAC;AAAA,IAC9E;AAEA,QAAI,QAAQ,WAAW,GAAG;AACtB,aAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,oBAAoB,CAAC,EAAE;AAAA,IACpE;AAEA,UAAM,YAAY,SAAS,YAAY,mBAAmB,SAAS,WAAW,WAAW;AACzF,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,cAAc,SAAS,SAAS,EAAE,CAAC,EAAE;AAAA,EAClF;AACJ;AAIA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aAAa;AAAA,IACb,aAAa,EAAE,OAAO;AAAA,MAClB,MAAM,EAAE,OAAO,EAAE,SAAS,8CAA8C;AAAA,MACxE,eAAe,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,EAAE,SAAS,yDAAyD;AAAA,MAC5H,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,SAAS,kBAAkB;AAAA,MACzE,YAAY,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,SAAS,wBAAwB;AAAA,MAC9E,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,MAAM,eAAe,aAAa,YAAY,KAAK,MAAM;AAC9D,UAAM,YAAY,MAAM,aAAa,IAAI;AACzC,UAAM,UAAU,MAAM,UAAU,WAAW,MAAM;AAAA,MAC7C;AAAA,MACA;AAAA,MACA;AAAA,IACJ,CAAC;AAED,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,CAAC,EAAE;AAAA,EACxD;AACJ;AAIA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aAAa;AAAA,IACb,aAAa,EAAE,OAAO;AAAA,MAClB,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,QAAQ,OAAO,MAAM,CAAC,CAAC,EAAE,SAAS,EAAE,SAAS,uCAAuC;AAAA,MAC7G,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,6CAA6C;AAAA,MACtF,cAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,KAAK,EAAE,SAAS,6BAA6B;AAAA,MAC1F,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,GAAG,EAAE,SAAS,gCAAgC;AAAA,MACtF,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,SAAS,UAAU,cAAc,UAAU,KAAK,MAAM;AAC3D,UAAM,YAAY,MAAM,aAAa,IAAI;AAEzC,QAAI,UAAU;AACV,YAAM,UAAe,aAAQ,QAAQ;AACrC,YAAM,WAAgB,cAAS,OAAO;AACtC,UAAI;AACA,kBAAU,cAAc;AAAA,UACpB,MAAM;AAAA,UACN,MAAM;AAAA,UACN,SAAS;AAAA,UACT,QAAQ,CAAC,iBAAiB,iBAAiB;AAAA,QAC/C,CAAC;AAAA,MACL,QAAQ;AAAA,MAER;AAAA,IACJ;AAEA,UAAM,SAAS,MAAM,UAAU,MAAM,EAAE,SAAS,cAAc,SAAS,CAAC;AAExE,UAAM,QAAQ;AAAA,MACV;AAAA,MACA;AAAA,MACA,aAAa,OAAO,MAAM,WAAW,CAAC,mBAAmB,OAAO,MAAM,WAAW,CAAC,aAAa,OAAO,MAAM,UAAU,CAAC;AAAA,MACvH,YAAY,OAAO,KAAK,WAAW,CAAC,qBAAqB,OAAO,KAAK,WAAW,CAAC;AAAA,IACrF;AAEA,QAAI,OAAO,MAAM;AACb,iBAAW,CAAC,MAAM,IAAI,KAAK,OAAO,QAAQ,OAAO,IAAI,GAAG;AACpD,cAAM,KAAK,WAAW,IAAI,QAAQ,KAAK,OAAO,aAAa,KAAK,OAAO,aAAa,KAAK,MAAM,SAAS;AAAA,MAC5G;AAAA,IACJ;AAEA,UAAM,QAAQ,UAAU,MAAM;AAC9B,UAAM,KAAK,EAAE;AACb,UAAM,KAAK,eAAe,MAAM,MAAM,UAAU,CAAC,iBAAiB,MAAM,KAAK,WAAW,CAAC,aAAa,MAAM,WAAW,aAAa,CAAC,OAAO;AAE5I,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,KAAK,IAAI,EAAE,CAAC,EAAE;AAAA,EACjE;AACJ;AAIA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aAAa;AAAA,IACb,aAAa,EAAE,OAAO;AAAA,MAClB,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,KAAK,MAAM;AAChB,UAAM,YAAY,MAAM,aAAa,IAAI;AACzC,UAAM,IAAI,UAAU,MAAM;AAE1B,UAAM,QAAQ,CAAC,sBAAsB,EAAE;AAEvC,QAAI,EAAE,MAAM;AACR,YAAM,KAAK,aAAa,EAAE,KAAK,KAAK,WAAW,EAAE,KAAK,MAAM,YAAY,EAAE,KAAK,QAAQ,UAAU;AAAA,IACrG;AACA,QAAI,EAAE,KAAK;AACP,YAAM,KAAK,YAAY,EAAE,IAAI,OAAO,aAAa,EAAE,IAAI,YAAY,WAAW,EAAE,IAAI,OAAO,gBAAgB;AAAA,IAC/G;AACA,QAAI,EAAE,WAAW;AACb,YAAM,KAAK,aAAa,EAAE,UAAU,WAAW,iBAAiB,EAAE,UAAU,SAAS,YAAY;AAAA,IACrG;AAEA,UAAM,UAAU,UAAU,oBAAoB;AAC9C,QAAI,QAAQ,SAAS,GAAG;AACpB,YAAM,KAAK,EAAE;AACb,YAAM,KAAK,qBAAqB;AAChC,iBAAW,QAAQ,SAAS;AACxB,cAAM,OAAO,UAAU,WAAW,IAAI;AACtC,cAAM,KAAK,KAAK,IAAI,KAAK,KAAK,MAAM,CAAC,QAAQ;AAAA,MACjD;AAAA,IACJ;AAEA,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,KAAK,IAAI,EAAE,CAAC,EAAE;AAAA,EACjE;AACJ;AAIA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aAAa;AAAA,IACb,aAAa,EAAE,OAAO;AAAA,MAClB,UAAU,EAAE,OAAO,EAAE,SAAS,iDAAiD;AAAA,MAC/E,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,SAAS,uBAAuB;AAAA,MACzE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,UAAU,OAAO,KAAK,MAAM;AACjC,UAAM,YAAY,MAAM,aAAa,IAAI;AACzC,UAAM,UAAU,MAAM,UAAU,YAAY,UAAU,KAAK;AAE3D,QAAI,QAAQ,WAAW,GAAG;AACtB,aAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,6BAA6B,QAAQ,IAAI,CAAC,EAAE;AAAA,IACzF;AAEA,UAAM,QAAQ,CAAC,mBAAmB,QAAQ,IAAI,EAAE;AAChD,eAAW,KAAK,SAAkB;AAC9B,YAAM,KAAK,MAAM,EAAE,UAAU,OAAO,EAAE,OAAO,MAAM,EAAE,MAAM,MAAM,EAAE,SAAS,KAAK,EAAE,SAAS,IAAI;AAAA,IACpG;AAEA,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,KAAK,IAAI,EAAE,CAAC,EAAE;AAAA,EACjE;AACJ;AAKA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aACI;AAAA,IAIJ,aAAa,EAAE,OAAO;AAAA,MAClB,QAAQ,EAAE,KAAK,CAAC,OAAO,UAAU,MAAM,CAAC,EAAE,SAAS,sBAAsB;AAAA,MACzE,YAAY,EAAE,OAAO,EAAE,SAAS,8CAA8C;AAAA,MAC9E,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,qCAAqC;AAAA,MAC7E,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,oCAAoC;AAAA,MAC1E,UAAU,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,EAAE,SAAS,kBAAkB;AAAA,MAC9E,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,SAAS,wBAAwB;AAAA,MACrE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,wBAAwB;AAAA,MAC7D,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,QAAQ,YAAY,SAAS,OAAO,UAAU,GAAG,MAAM,KAAK,MAAM;AACvE,UAAM,YAAY,MAAM,aAAa,IAAI;AACzC,UAAM,OAAO,UAAU,WAAW,UAAU;AAE5C,QAAI,WAAW,OAAO;AAClB,UAAI,CAAC,QAAS,OAAM,IAAI,MAAM,gDAAgD;AAC9E,YAAM,KAAK,MAAM,KAAK,IAAI,SAAS,QAAQ;AAC3C,aAAO;AAAA,QACH,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,gBAAW,EAAE,cAAc,UAAU,MAAM,KAAK,MAAM,CAAC,UAAU,CAAC;AAAA,MACtG;AAAA,IACJ;AAEA,QAAI,WAAW,UAAU;AACrB,UAAI,CAAC,MAAO,OAAM,IAAI,MAAM,iDAAiD;AAC7E,YAAM,UAAU,MAAM,KAAK,OAAO,OAAO,EAAE,EAAE,CAAC;AAE9C,UAAI,QAAQ,WAAW,GAAG;AACtB,eAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,kBAAkB,UAAU,UAAU,KAAK,IAAI,CAAC,EAAE;AAAA,MAC/F;AAEA,YAAM,QAAQ,CAAC,kBAAkB,UAAU,IAAI,EAAE;AACjD,iBAAW,KAAK,SAAS;AACrB,cAAM,QAAQ,KAAK,OAAO,EAAE,SAAS,KAAK,GAAG;AAC7C,cAAM,KAAK,IAAI,KAAK,MAAM,EAAE,OAAO,EAAE;AACrC,YAAI,OAAO,KAAK,EAAE,QAAQ,EAAE,SAAS,GAAG;AACpC,gBAAM,KAAK,KAAK,KAAK,UAAU,EAAE,QAAQ,CAAC,EAAE;AAAA,QAChD;AACA,cAAM,KAAK,EAAE;AAAA,MACjB;AACA,aAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,KAAK,IAAI,EAAE,CAAC,EAAE;AAAA,IACjE;AAEA,QAAI,WAAW,QAAQ;AACnB,UAAI,QAAQ,KAAM,OAAM,IAAI,MAAM,8CAA8C;AAChF,YAAM,SAAS,MAAM,KAAK,KAAK,EAAE,KAAK,CAAC;AACvC,aAAO;AAAA,QACH,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,kBAAa,OAAO,OAAO,gBAAgB,UAAU,WAAW,IAAI,IAAI,CAAC;AAAA,MAC7G;AAAA,IACJ;AAEA,UAAM,IAAI,MAAM,8BAA8B,MAAM,IAAI;AAAA,EAC5D;AACJ;AAIA,SAAS,cAAc,SAAgB,MAAsB;AACzD,QAAM,QAAkB,CAAC,MAAM,IAAI,IAAI,EAAE;AACzC,aAAW,KAAK,SAAS;AACrB,UAAM,QAAQ,KAAK,MAAM,EAAE,QAAQ,GAAG;AACtC,QAAI,EAAE,SAAS,QAAQ;AACnB,YAAM,IAAI,EAAE;AACZ,YAAM,KAAK,SAAS,KAAK,MAAM,EAAE,QAAQ,WAAM,EAAE,QAAQ,EAAE,SAAS,MAAM,EAAE,SAAS,IAAI,EAAE,OAAO,GAAG;AACrG,YAAM,KAAK,EAAE,OAAO;AACpB,YAAM,KAAK,EAAE;AAAA,IACjB,WAAW,EAAE,SAAS,UAAU;AAC5B,YAAM,IAAI,EAAE;AACZ,YAAM,KAAK,WAAW,KAAK,MAAM,EAAE,SAAS,WAAM,EAAE,OAAO,KAAK,EAAE,MAAM,GAAG;AAC3E,UAAI,EAAE,OAAO,OAAQ,OAAM,KAAK,YAAY,EAAE,MAAM,KAAK,IAAI,CAAC,EAAE;AAChE,YAAM,KAAK,EAAE;AAAA,IACjB,WAAW,EAAE,SAAS,YAAY;AAC9B,YAAM,MAAM,EAAE,UAAU,WAAM,EAAE,OAAO,KAAK;AAC5C,YAAM,KAAK,QAAQ,KAAK,MAAM,EAAE,QAAQ,KAAK,EAAE,SAAS,UAAU,IAAI,GAAG,EAAE;AAC3E,YAAM,KAAK,EAAE,OAAO;AACpB,YAAM,KAAK,EAAE;AAAA,IACjB,WAAW,EAAE,SAAS,cAAc;AAChC,YAAM,MAAM,EAAE,UAAU,cAAc;AACtC,YAAM,KAAK,eAAe,KAAK,OAAO,GAAG,GAAG;AAC5C,YAAM,KAAK,EAAE,OAAO;AACpB,YAAM,KAAK,EAAE;AAAA,IACjB;AAAA,EACJ;AACA,SAAO,MAAM,KAAK,IAAI;AAC1B;AA3BS;AA+BT,eAAe,OAAO;AAClB,QAAM,YAAY,IAAI,qBAAqB;AAC3C,QAAM,OAAO,QAAQ,SAAS;AAClC;AAHe;AAKf,KAAK,EAAE,MAAM,SAAO;AAChB,UAAQ,MAAM,+BAA+B,IAAI,OAAO,EAAE;AAC1D,UAAQ,KAAK,CAAC;AAClB,CAAC;","names":[]}
|
|
1
|
+
{"version":3,"sources":["../src/mcp-server.ts","../src/workspace-pool.ts","../src/workspace-factory.ts"],"sourcesContent":["#!/usr/bin/env node\n\n/**\n * BrainBank — MCP Server\n * \n * Exposes BrainBank as an MCP server via stdio transport.\n * Works with Google Antigravity, Claude Desktop, and any MCP-compatible client.\n * \n * Usage in mcp_config.json:\n * {\n * \"mcpServers\": {\n * \"brainbank\": {\n * \"command\": \"npx\",\n * \"args\": [\"-y\", \"@brainbank/mcp\"]\n * }\n * }\n * }\n * \n * Tools (2):\n * brainbank_context — Workflow Trace: search + call tree + called-by annotations\n * brainbank_index — Re-index (requires .brainbank/config.json)\n */\n\nimport { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';\nimport { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';\nimport { z } from 'zod/v3';\nimport { existsSync } from 'node:fs';\nimport { WorkspacePool } from './workspace-pool.js';\nimport { createWorkspaceBrain, resolveRepoPath } from './workspace-factory.js';\n\n\n\n// ── Multi-Workspace BrainBank Pool ─────────────────────\n\nconst pool = new WorkspacePool({\n factory: createWorkspaceBrain,\n maxMemoryMB: parseInt(process.env.BRAINBANK_MAX_MEMORY_MB ?? '2048', 10),\n ttlMinutes: parseInt(process.env.BRAINBANK_TTL_MINUTES ?? '30', 10),\n onError: (repo, err) => {\n const msg = err instanceof Error ? err.message : String(err);\n console.error(`BrainBank pool error [${repo}]: ${msg}`);\n },\n});\n\n/** Resolve repo and get a BrainBank from the pool. */\nasync function getBrainBank(targetRepo?: string) {\n return pool.get(resolveRepoPath(targetRepo));\n}\n\n// ── MCP Server Setup ────────────────────────────────\n\nconst server = new McpServer({\n name: 'brainbank',\n version: '0.4.0',\n});\n\n// ── Tool: brainbank_context ─────────────────────────\n\nserver.registerTool(\n 'brainbank_context',\n {\n title: 'BrainBank Context',\n description:\n 'Get a formatted knowledge context block for a task. Returns a Workflow Trace: ' +\n 'search hits + full call tree with `called by` annotations, topologically ordered. ' +\n 'All source code included — no trimming, no truncation.',\n inputSchema: z.object({\n task: z.string().describe('Description of the task you need context for'),\n affectedFiles: z.array(z.string()).optional().default([]).describe('Files you plan to modify (improves co-edit suggestions)'),\n codeResults: z.number().optional().default(20).describe('Max code results'),\n gitResults: z.number().optional().default(5).describe('Max git commit results'),\n docsResults: z.number().optional().describe('Max document results (omit to skip docs)'),\n sources: z.record(z.number()).optional().describe('Per-source result limits, overrides codeResults/gitResults/docsResults (e.g. { code: 10, git: 0, docs: 5 })'),\n path: z.string().optional().describe('Filter results to files under this path prefix (e.g. src/services/)'),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n // BrainBankQL context fields\n lines: z.boolean().optional().describe('Prefix each code line with its source line number (e.g. 127| code)'),\n symbols: z.boolean().optional().describe('Append symbol index (all functions, classes, interfaces) for matched files'),\n compact: z.boolean().optional().describe('Show only function/class signatures, skip bodies'),\n callTree: z.union([z.boolean(), z.object({ depth: z.number() })]).optional().describe('Include call tree expansion. Pass { depth: N } to control depth'),\n imports: z.boolean().optional().describe('Include dependency/import summary section'),\n expander: z.boolean().optional().describe('Enable LLM-powered context expansion to discover related chunks not found by search'),\n }),\n },\n async ({ task, affectedFiles, codeResults, gitResults, docsResults, sources, path, repo, lines, symbols, compact, callTree, imports, expander }) => {\n const repoPath = resolveRepoPath(repo);\n const brainbank = await getBrainBank(repo);\n\n // Build sources from explicit params, then let `sources` override\n const base: Record<string, number> = { code: codeResults, git: gitResults };\n if (docsResults !== undefined) base.docs = docsResults;\n const resolvedSources = sources ? { ...base, ...sources } : base;\n\n // Build fields from explicit params (only include defined values)\n const fields: Record<string, unknown> = {};\n if (lines !== undefined) fields.lines = lines;\n if (symbols !== undefined) fields.symbols = symbols;\n if (compact !== undefined) fields.compact = compact;\n if (callTree !== undefined) fields.callTree = callTree;\n if (imports !== undefined) fields.imports = imports;\n if (expander !== undefined) fields.expander = expander;\n\n const context = await brainbank.getContext(task, {\n affectedFiles,\n sources: resolvedSources,\n pathPrefix: path,\n source: 'mcp',\n fields: Object.keys(fields).length > 0 ? fields : undefined,\n });\n\n return { content: [{ type: 'text' as const, text: context }] };\n },\n);\n\n// ── Tool: brainbank_index ───────────────────────────\n\nserver.registerTool(\n 'brainbank_index',\n {\n title: 'BrainBank Index',\n description:\n 'Re-index code, git history, and docs. Requires .brainbank/config.json to exist. ' +\n 'Incremental — only changed files are processed.',\n inputSchema: z.object({\n forceReindex: z.boolean().optional().default(false).describe('Force re-index of all files'),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n }),\n },\n async ({ forceReindex, repo }) => {\n const repoPath = resolveRepoPath(repo);\n\n // Require config.json — force users to set up structure first\n if (!existsSync(`${repoPath}/.brainbank/config.json`)) {\n return {\n content: [{\n type: 'text' as const,\n text:\n `BrainBank: No .brainbank/config.json found at ${repoPath}.\\n\\n` +\n `## How to set up\\n\\n` +\n `Create \\`${repoPath}/.brainbank/config.json\\` with:\\n\\n` +\n '```json\\n' +\n '{\\n' +\n ' \"plugins\": [\"code\"],\\n' +\n ' \"code\": {\\n' +\n ' \"embedding\": \"perplexity-context\",\\n' +\n ' \"ignore\": [\\n' +\n ' \"node_modules/**\", \"dist/**\", \"build/**\",\\n' +\n ' \".next/**\", \"coverage/**\", \"__pycache__/**\",\\n' +\n ' \"**/*.min.js\", \"**/*.min.css\",\\n' +\n ' \"tests/**\", \"test/**\",\\n' +\n ' \"**/test_*.py\", \"**/*_test.py\",\\n' +\n ' \"**/*.test.ts\", \"**/*.spec.ts\"\\n' +\n ' ]\\n' +\n ' },\\n' +\n ' \"embedding\": \"perplexity-context\"\\n' +\n '}\\n' +\n '```\\n\\n' +\n `**Embedding options:** \\`local\\` (free, offline), \\`openai\\`, \\`perplexity\\`, \\`perplexity-context\\` (best quality)\\n` +\n `**Plugins available:** \\`code\\`, \\`git\\`, \\`docs\\`\\n\\n` +\n `Then run:\\n` +\n '```bash\\nbrainbank index . --force --yes\\n```',\n }],\n };\n }\n\n const brainbank = await getBrainBank(repo);\n const result = await brainbank.index({ forceReindex });\n\n const lines = ['## Indexing Complete', ''];\n\n const codeResult = result.code as { indexed?: number; skipped?: number; chunks?: number } | undefined;\n const gitResult = result.git as { indexed?: number; skipped?: number } | undefined;\n\n lines.push(`**Code**: ${codeResult?.indexed ?? 0} files indexed, ${codeResult?.skipped ?? 0} skipped, ${codeResult?.chunks ?? 0} chunks`);\n lines.push(`**Git**: ${gitResult?.indexed ?? 0} commits indexed, ${gitResult?.skipped ?? 0} skipped`);\n\n const docsResult = result.docs as Record<string, { indexed: number; skipped: number; chunks: number }> | undefined;\n if (docsResult) {\n for (const [name, stat] of Object.entries(docsResult)) {\n lines.push(`**Docs [${name}]**: ${stat.indexed} indexed, ${stat.skipped} skipped, ${stat.chunks} chunks`);\n }\n }\n\n const stats = brainbank.stats();\n const codeStats = stats.code as { chunks?: number } | undefined;\n const gitStats = stats.git as { commits?: number } | undefined;\n const docStats = stats.documents as { documents?: number } | undefined;\n lines.push('');\n lines.push(`**Totals**: ${codeStats?.chunks ?? 0} code chunks, ${gitStats?.commits ?? 0} commits, ${docStats?.documents ?? 0} docs`);\n\n return { content: [{ type: 'text' as const, text: lines.join('\\n') }] };\n },\n);\n\n// ── Tool: brainbank_files ───────────────────────────\n\nserver.registerTool(\n 'brainbank_files',\n {\n title: 'BrainBank Files',\n description:\n 'Fetch full file contents from the index. Use AFTER brainbank_context ' +\n 'to view complete files identified by search. No semantic search runs — ' +\n 'this is a direct file viewer.\\n\\n' +\n 'Supports:\\n' +\n '- Exact paths: \"src/auth/login.ts\"\\n' +\n '- Directories: \"src/graph/\" (trailing / = all files under path)\\n' +\n '- Glob patterns: \"src/**/*.service.ts\"\\n' +\n '- Fuzzy basename: \"plugin.ts\" (matches src/plugin.ts when exact fails)',\n inputSchema: z.object({\n files: z.array(z.string()).describe(\n 'File paths to fetch. Exact paths, directories (trailing /), ' +\n 'glob patterns (e.g. src/**/*.ts), or fuzzy basenames.',\n ),\n repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),\n lines: z.boolean().optional().describe('Prefix each line with source line number'),\n }),\n },\n async ({ files, repo, lines }) => {\n const brainbank = await getBrainBank(repo);\n const results = brainbank.resolveFiles(files);\n\n if (results.length === 0) {\n return { content: [{ type: 'text' as const, text: 'No matching files found in the index.' }] };\n }\n\n // Format: markdown with file headers + fenced code blocks\n const parts: string[] = [];\n for (const r of results) {\n const meta = r.metadata as Record<string, unknown>;\n const lang = (meta.language as string) ?? '';\n const startLine = (meta.startLine as number) ?? 1;\n\n parts.push(`## ${r.filePath}\\n`);\n parts.push('```' + lang);\n\n if (lines) {\n const codeLines = r.content.split('\\n');\n const pad = String(startLine + codeLines.length - 1).length;\n parts.push(codeLines.map((l, i) =>\n `${String(startLine + i).padStart(pad)}| ${l}`,\n ).join('\\n'));\n } else {\n parts.push(r.content);\n }\n\n parts.push('```\\n');\n }\n\n return { content: [{ type: 'text' as const, text: parts.join('\\n') }] };\n },\n);\n\n// ── Start Server ────────────────────────────────────\n\nasync function main() {\n const transport = new StdioServerTransport();\n await server.connect(transport);\n}\n\nmain().catch(err => {\n const message = err instanceof Error ? err.message : String(err);\n console.error(`BrainBank MCP Server Error: ${message}`);\n process.exit(1);\n});\n","/**\n * WorkspacePool — BrainBank instance lifecycle manager.\n *\n * Manages cached BrainBank instances per workspace with:\n * - Memory-pressure eviction (oldest idle first)\n * - TTL eviction for inactive workspaces\n * - Active-operation tracking (prevents mid-query eviction)\n * - Hot-reload of stale HNSW indices\n */\n\nimport type { BrainBank } from 'brainbank';\n\n/** Pool configuration. */\nexport interface PoolOptions {\n /** Max total estimated memory in MB. Default: 2048. */\n maxMemoryMB?: number;\n /** Minutes of inactivity before eviction. Default: 30. */\n ttlMinutes?: number;\n /** Factory function to create a BrainBank for a repo path. */\n factory: (repoPath: string) => Promise<BrainBank>;\n /** Called when a workspace is evicted. */\n onEvict?: (repoPath: string) => void;\n /** Called when an error occurs during pool operations. */\n onError?: (repoPath: string, err: unknown) => void;\n}\n\n/** Internal pool entry. */\ninterface PoolEntry {\n brain: BrainBank;\n repoPath: string;\n lastAccess: number;\n createdAt: number;\n activeOps: number;\n}\n\n/** Public pool statistics. */\nexport interface PoolStats {\n size: number;\n totalMemoryMB: number;\n entries: PoolEntryStats[];\n}\n\n/** Per-entry statistics. */\nexport interface PoolEntryStats {\n repoPath: string;\n lastAccessAgo: string;\n memoryMB: number;\n activeOps: number;\n}\n\nconst DEFAULT_MAX_MEMORY_MB = 2048;\nconst DEFAULT_TTL_MINUTES = 30;\nconst EVICTION_INTERVAL_MS = 60_000;\n\n/** Format milliseconds as a human-readable \"ago\" string. */\nfunction formatAgo(ms: number): string {\n if (ms < 60_000) return `${Math.round(ms / 1000)}s ago`;\n if (ms < 3_600_000) return `${Math.round(ms / 60_000)}m ago`;\n return `${Math.round(ms / 3_600_000)}h ago`;\n}\n\nexport class WorkspacePool {\n private _pool = new Map<string, PoolEntry>();\n private _timer: ReturnType<typeof setInterval> | null = null;\n private _maxMemoryBytes: number;\n private _ttlMs: number;\n private _factory: (repoPath: string) => Promise<BrainBank>;\n private _onEvict?: (repoPath: string) => void;\n private _onError?: (repoPath: string, err: unknown) => void;\n\n constructor(options: PoolOptions) {\n this._maxMemoryBytes = (options.maxMemoryMB ?? DEFAULT_MAX_MEMORY_MB) * 1024 * 1024;\n this._ttlMs = (options.ttlMinutes ?? DEFAULT_TTL_MINUTES) * 60 * 1000;\n this._factory = options.factory;\n this._onEvict = options.onEvict;\n this._onError = options.onError;\n\n this._timer = setInterval(() => this._evictStale(), EVICTION_INTERVAL_MS);\n // Don't hold the process open for the timer\n if (this._timer.unref) this._timer.unref();\n }\n\n /** Number of cached workspaces. */\n get size(): number {\n return this._pool.size;\n }\n\n /**\n * Get a BrainBank for the given repo path.\n * Returns a cached instance (with hot-reload) or creates a new one.\n */\n async get(repoPath: string): Promise<BrainBank> {\n const key = repoPath.replace(/\\/+$/, '');\n\n const existing = this._pool.get(key);\n if (existing) {\n existing.lastAccess = Date.now();\n try { await existing.brain.ensureFresh(); } catch { /* stale is better than nothing */ }\n return existing.brain;\n }\n\n this._evictByMemoryPressure();\n\n const brain = await this._factory(key);\n this._pool.set(key, {\n brain,\n repoPath: key,\n lastAccess: Date.now(),\n createdAt: Date.now(),\n activeOps: 0,\n });\n\n return brain;\n }\n\n /**\n * Execute an operation with active-op tracking.\n * Prevents the workspace from being evicted while the operation runs.\n */\n async withBrain<T>(repoPath: string, fn: (brain: BrainBank) => Promise<T>): Promise<T> {\n const brain = await this.get(repoPath);\n const key = repoPath.replace(/\\/+$/, '');\n const entry = this._pool.get(key);\n if (entry) entry.activeOps++;\n\n try {\n return await fn(brain);\n } finally {\n if (entry) {\n entry.activeOps--;\n entry.lastAccess = Date.now();\n }\n }\n }\n\n /** Manually evict a specific workspace. */\n evict(repoPath: string): void {\n const key = repoPath.replace(/\\/+$/, '');\n this._evictEntry(key);\n }\n\n /** Get pool statistics. */\n stats(): PoolStats {\n const now = Date.now();\n let totalMemory = 0;\n const entries: PoolEntryStats[] = [];\n\n for (const entry of this._pool.values()) {\n const memBytes = entry.brain.memoryHint();\n const memMB = Math.round(memBytes / 1024 / 1024 * 100) / 100;\n totalMemory += memBytes;\n\n entries.push({\n repoPath: entry.repoPath,\n lastAccessAgo: formatAgo(now - entry.lastAccess),\n memoryMB: memMB,\n activeOps: entry.activeOps,\n });\n }\n\n return {\n size: this._pool.size,\n totalMemoryMB: Math.round(totalMemory / 1024 / 1024 * 100) / 100,\n entries,\n };\n }\n\n /** Close all entries and stop the eviction timer. */\n close(): void {\n if (this._timer) {\n clearInterval(this._timer);\n this._timer = null;\n }\n for (const key of [...this._pool.keys()]) {\n this._evictEntry(key);\n }\n }\n\n /** Evict workspaces that haven't been accessed within the TTL. */\n private _evictStale(): void {\n const cutoff = Date.now() - this._ttlMs;\n for (const [key, entry] of this._pool) {\n if (entry.lastAccess < cutoff && entry.activeOps === 0) {\n this._evictEntry(key);\n }\n }\n }\n\n /** Evict oldest idle entries until total memory is under the limit. */\n private _evictByMemoryPressure(): void {\n let totalMemory = 0;\n for (const entry of this._pool.values()) {\n totalMemory += entry.brain.memoryHint();\n }\n\n if (totalMemory < this._maxMemoryBytes) return;\n\n // Sort by lastAccess ascending (oldest first), filter idle\n const candidates = [...this._pool.entries()]\n .filter(([, e]) => e.activeOps === 0)\n .sort(([, a], [, b]) => a.lastAccess - b.lastAccess);\n\n for (const [key, entry] of candidates) {\n if (totalMemory < this._maxMemoryBytes) break;\n totalMemory -= entry.brain.memoryHint();\n this._evictEntry(key);\n }\n }\n\n /** Evict a single entry by key. */\n private _evictEntry(key: string): void {\n const entry = this._pool.get(key);\n if (!entry) return;\n\n try {\n entry.brain.close();\n } catch (err: unknown) {\n this._onError?.(key, err);\n }\n\n this._pool.delete(key);\n this._onEvict?.(key);\n }\n}\n","/**\n * WorkspaceFactory — creates BrainBank instances via the core factory.\n *\n * Delegates to `createBrain()` from the `brainbank` package, passing\n * a portable `BrainContext`. No plugin hardcoding — the factory handles\n * plugin discovery from config and installed packages.\n */\n\nimport type { BrainBank, BrainContext } from 'brainbank';\n\nimport * as fs from 'node:fs';\nimport * as path from 'node:path';\n\n/**\n * Detect repo root by walking up from startDir until we find `.git/`.\n * Returns startDir itself if no `.git/` is found.\n */\nexport function findRepoRoot(startDir: string): string {\n let dir = path.resolve(startDir);\n while (true) {\n if (fs.existsSync(path.join(dir, '.git'))) return dir;\n const parent = path.dirname(dir);\n if (parent === dir) break;\n dir = parent;\n }\n return path.resolve(startDir);\n}\n\n/** Resolve the effective repo path from an optional target, env, or cwd. */\nexport function resolveRepoPath(targetRepo?: string): string {\n const rp = targetRepo\n ?? process.env.BRAINBANK_REPO\n ?? findRepoRoot(process.cwd());\n return rp.replace(/\\/+$/, '');\n}\n\n/**\n * Create a BrainBank instance for a workspace.\n * Uses the core factory which handles:\n * - Config loading from .brainbank/config.json\n * - Dynamic plugin discovery and registration\n * - Embedding/reranker provider setup\n * - Folder plugin auto-discovery\n */\nexport async function createWorkspaceBrain(repoPath: string): Promise<BrainBank> {\n const { createBrain, resetFactoryCache } = await import('brainbank') as typeof import('brainbank');\n resetFactoryCache();\n\n const context: BrainContext = {\n repoPath,\n env: process.env as Record<string, string | undefined>,\n };\n\n // Silence stdout during initialization — the core factory emits ANSI-colored\n // console.log messages (plugin loading, multi-repo detection) that corrupt\n // the MCP JSON-RPC stdio transport. Redirect console.log → stderr temporarily.\n const origLog = console.log;\n console.log = (...args: unknown[]) => console.error(...args);\n try {\n const brain = await createBrain(context);\n await brain.initialize();\n return brain;\n } finally {\n console.log = origLog;\n }\n}\n"],"mappings":";;;;;AAuBA,SAAS,iBAAiB;AAC1B,SAAS,4BAA4B;AACrC,SAAS,SAAS;AAClB,SAAS,cAAAA,mBAAkB;;;ACwB3B,IAAM,wBAAwB;AAC9B,IAAM,sBAAsB;AAC5B,IAAM,uBAAuB;AAG7B,SAAS,UAAU,IAAoB;AACnC,MAAI,KAAK,IAAQ,QAAO,GAAG,KAAK,MAAM,KAAK,GAAI,CAAC;AAChD,MAAI,KAAK,KAAW,QAAO,GAAG,KAAK,MAAM,KAAK,GAAM,CAAC;AACrD,SAAO,GAAG,KAAK,MAAM,KAAK,IAAS,CAAC;AACxC;AAJS;AAMF,IAAM,gBAAN,MAAoB;AAAA,EA7D3B,OA6D2B;AAAA;AAAA;AAAA,EACf,QAAQ,oBAAI,IAAuB;AAAA,EACnC,SAAgD;AAAA,EAChD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,SAAsB;AAC9B,SAAK,mBAAmB,QAAQ,eAAe,yBAAyB,OAAO;AAC/E,SAAK,UAAU,QAAQ,cAAc,uBAAuB,KAAK;AACjE,SAAK,WAAW,QAAQ;AACxB,SAAK,WAAW,QAAQ;AACxB,SAAK,WAAW,QAAQ;AAExB,SAAK,SAAS,YAAY,MAAM,KAAK,YAAY,GAAG,oBAAoB;AAExE,QAAI,KAAK,OAAO,MAAO,MAAK,OAAO,MAAM;AAAA,EAC7C;AAAA;AAAA,EAGA,IAAI,OAAe;AACf,WAAO,KAAK,MAAM;AAAA,EACtB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,IAAI,UAAsC;AAC5C,UAAM,MAAM,SAAS,QAAQ,QAAQ,EAAE;AAEvC,UAAM,WAAW,KAAK,MAAM,IAAI,GAAG;AACnC,QAAI,UAAU;AACV,eAAS,aAAa,KAAK,IAAI;AAC/B,UAAI;AAAE,cAAM,SAAS,MAAM,YAAY;AAAA,MAAG,QAAQ;AAAA,MAAqC;AACvF,aAAO,SAAS;AAAA,IACpB;AAEA,SAAK,uBAAuB;AAE5B,UAAM,QAAQ,MAAM,KAAK,SAAS,GAAG;AACrC,SAAK,MAAM,IAAI,KAAK;AAAA,MAChB;AAAA,MACA,UAAU;AAAA,MACV,YAAY,KAAK,IAAI;AAAA,MACrB,WAAW,KAAK,IAAI;AAAA,MACpB,WAAW;AAAA,IACf,CAAC;AAED,WAAO;AAAA,EACX;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,UAAa,UAAkB,IAAkD;AACnF,UAAM,QAAQ,MAAM,KAAK,IAAI,QAAQ;AACrC,UAAM,MAAM,SAAS,QAAQ,QAAQ,EAAE;AACvC,UAAM,QAAQ,KAAK,MAAM,IAAI,GAAG;AAChC,QAAI,MAAO,OAAM;AAEjB,QAAI;AACA,aAAO,MAAM,GAAG,KAAK;AAAA,IACzB,UAAE;AACE,UAAI,OAAO;AACP,cAAM;AACN,cAAM,aAAa,KAAK,IAAI;AAAA,MAChC;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAM,UAAwB;AAC1B,UAAM,MAAM,SAAS,QAAQ,QAAQ,EAAE;AACvC,SAAK,YAAY,GAAG;AAAA,EACxB;AAAA;AAAA,EAGA,QAAmB;AACf,UAAM,MAAM,KAAK,IAAI;AACrB,QAAI,cAAc;AAClB,UAAM,UAA4B,CAAC;AAEnC,eAAW,SAAS,KAAK,MAAM,OAAO,GAAG;AACrC,YAAM,WAAW,MAAM,MAAM,WAAW;AACxC,YAAM,QAAQ,KAAK,MAAM,WAAW,OAAO,OAAO,GAAG,IAAI;AACzD,qBAAe;AAEf,cAAQ,KAAK;AAAA,QACT,UAAU,MAAM;AAAA,QAChB,eAAe,UAAU,MAAM,MAAM,UAAU;AAAA,QAC/C,UAAU;AAAA,QACV,WAAW,MAAM;AAAA,MACrB,CAAC;AAAA,IACL;AAEA,WAAO;AAAA,MACH,MAAM,KAAK,MAAM;AAAA,MACjB,eAAe,KAAK,MAAM,cAAc,OAAO,OAAO,GAAG,IAAI;AAAA,MAC7D;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,QAAc;AACV,QAAI,KAAK,QAAQ;AACb,oBAAc,KAAK,MAAM;AACzB,WAAK,SAAS;AAAA,IAClB;AACA,eAAW,OAAO,CAAC,GAAG,KAAK,MAAM,KAAK,CAAC,GAAG;AACtC,WAAK,YAAY,GAAG;AAAA,IACxB;AAAA,EACJ;AAAA;AAAA,EAGQ,cAAoB;AACxB,UAAM,SAAS,KAAK,IAAI,IAAI,KAAK;AACjC,eAAW,CAAC,KAAK,KAAK,KAAK,KAAK,OAAO;AACnC,UAAI,MAAM,aAAa,UAAU,MAAM,cAAc,GAAG;AACpD,aAAK,YAAY,GAAG;AAAA,MACxB;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGQ,yBAA+B;AACnC,QAAI,cAAc;AAClB,eAAW,SAAS,KAAK,MAAM,OAAO,GAAG;AACrC,qBAAe,MAAM,MAAM,WAAW;AAAA,IAC1C;AAEA,QAAI,cAAc,KAAK,gBAAiB;AAGxC,UAAM,aAAa,CAAC,GAAG,KAAK,MAAM,QAAQ,CAAC,EACtC,OAAO,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,CAAC,EACnC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU;AAEvD,eAAW,CAAC,KAAK,KAAK,KAAK,YAAY;AACnC,UAAI,cAAc,KAAK,gBAAiB;AACxC,qBAAe,MAAM,MAAM,WAAW;AACtC,WAAK,YAAY,GAAG;AAAA,IACxB;AAAA,EACJ;AAAA;AAAA,EAGQ,YAAY,KAAmB;AACnC,UAAM,QAAQ,KAAK,MAAM,IAAI,GAAG;AAChC,QAAI,CAAC,MAAO;AAEZ,QAAI;AACA,YAAM,MAAM,MAAM;AAAA,IACtB,SAAS,KAAc;AACnB,WAAK,WAAW,KAAK,GAAG;AAAA,IAC5B;AAEA,SAAK,MAAM,OAAO,GAAG;AACrB,SAAK,WAAW,GAAG;AAAA,EACvB;AACJ;;;ACrNA,YAAY,QAAQ;AACpB,YAAY,UAAU;AAMf,SAAS,aAAa,UAA0B;AACnD,MAAI,MAAW,aAAQ,QAAQ;AAC/B,SAAO,MAAM;AACT,QAAO,cAAgB,UAAK,KAAK,MAAM,CAAC,EAAG,QAAO;AAClD,UAAM,SAAc,aAAQ,GAAG;AAC/B,QAAI,WAAW,IAAK;AACpB,UAAM;AAAA,EACV;AACA,SAAY,aAAQ,QAAQ;AAChC;AATgB;AAYT,SAAS,gBAAgB,YAA6B;AACzD,QAAM,KAAK,cACJ,QAAQ,IAAI,kBACZ,aAAa,QAAQ,IAAI,CAAC;AACjC,SAAO,GAAG,QAAQ,QAAQ,EAAE;AAChC;AALgB;AAehB,eAAsB,qBAAqB,UAAsC;AAC7E,QAAM,EAAE,aAAa,kBAAkB,IAAI,MAAM,OAAO,WAAW;AACnE,oBAAkB;AAElB,QAAM,UAAwB;AAAA,IAC1B;AAAA,IACA,KAAK,QAAQ;AAAA,EACjB;AAKA,QAAM,UAAU,QAAQ;AACxB,UAAQ,MAAM,IAAI,SAAoB,QAAQ,MAAM,GAAG,IAAI;AAC3D,MAAI;AACA,UAAM,QAAQ,MAAM,YAAY,OAAO;AACvC,UAAM,MAAM,WAAW;AACvB,WAAO;AAAA,EACX,UAAE;AACE,YAAQ,MAAM;AAAA,EAClB;AACJ;AArBsB;;;AFVtB,IAAM,OAAO,IAAI,cAAc;AAAA,EAC3B,SAAS;AAAA,EACT,aAAa,SAAS,QAAQ,IAAI,2BAA2B,QAAQ,EAAE;AAAA,EACvE,YAAY,SAAS,QAAQ,IAAI,yBAAyB,MAAM,EAAE;AAAA,EAClE,SAAS,wBAAC,MAAM,QAAQ;AACpB,UAAM,MAAM,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAC3D,YAAQ,MAAM,yBAAyB,IAAI,MAAM,GAAG,EAAE;AAAA,EAC1D,GAHS;AAIb,CAAC;AAGD,eAAe,aAAa,YAAqB;AAC7C,SAAO,KAAK,IAAI,gBAAgB,UAAU,CAAC;AAC/C;AAFe;AAMf,IAAM,SAAS,IAAI,UAAU;AAAA,EACzB,MAAM;AAAA,EACN,SAAS;AACb,CAAC;AAID,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aACI;AAAA,IAGJ,aAAa,EAAE,OAAO;AAAA,MAClB,MAAM,EAAE,OAAO,EAAE,SAAS,8CAA8C;AAAA,MACxE,eAAe,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,EAAE,SAAS,yDAAyD;AAAA,MAC5H,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,SAAS,kBAAkB;AAAA,MAC1E,YAAY,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,SAAS,wBAAwB;AAAA,MAC9E,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,0CAA0C;AAAA,MACtF,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,6GAA6G;AAAA,MAC/J,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,qEAAqE;AAAA,MAC1G,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA;AAAA,MAEhF,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,oEAAoE;AAAA,MAC3G,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,4EAA4E;AAAA,MACrH,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,kDAAkD;AAAA,MAC3F,UAAU,EAAE,MAAM,CAAC,EAAE,QAAQ,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,SAAS,iEAAiE;AAAA,MACvJ,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,MACpF,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,qFAAqF;AAAA,IACnI,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,MAAM,eAAe,aAAa,YAAY,aAAa,SAAS,MAAAC,OAAM,MAAM,OAAO,SAAS,SAAS,UAAU,SAAS,SAAS,MAAM;AAChJ,UAAM,WAAW,gBAAgB,IAAI;AACrC,UAAM,YAAY,MAAM,aAAa,IAAI;AAGzC,UAAM,OAA+B,EAAE,MAAM,aAAa,KAAK,WAAW;AAC1E,QAAI,gBAAgB,OAAW,MAAK,OAAO;AAC3C,UAAM,kBAAkB,UAAU,EAAE,GAAG,MAAM,GAAG,QAAQ,IAAI;AAG5D,UAAM,SAAkC,CAAC;AACzC,QAAI,UAAU,OAAW,QAAO,QAAQ;AACxC,QAAI,YAAY,OAAW,QAAO,UAAU;AAC5C,QAAI,YAAY,OAAW,QAAO,UAAU;AAC5C,QAAI,aAAa,OAAW,QAAO,WAAW;AAC9C,QAAI,YAAY,OAAW,QAAO,UAAU;AAC5C,QAAI,aAAa,OAAW,QAAO,WAAW;AAE9C,UAAM,UAAU,MAAM,UAAU,WAAW,MAAM;AAAA,MAC7C;AAAA,MACA,SAAS;AAAA,MACT,YAAYA;AAAA,MACZ,QAAQ;AAAA,MACR,QAAQ,OAAO,KAAK,MAAM,EAAE,SAAS,IAAI,SAAS;AAAA,IACtD,CAAC;AAED,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,CAAC,EAAE;AAAA,EACjE;AACJ;AAIA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aACI;AAAA,IAEJ,aAAa,EAAE,OAAO;AAAA,MAClB,cAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,KAAK,EAAE,SAAS,6BAA6B;AAAA,MAC1F,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,IACpF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,cAAc,KAAK,MAAM;AAC9B,UAAM,WAAW,gBAAgB,IAAI;AAGrC,QAAI,CAACC,YAAW,GAAG,QAAQ,yBAAyB,GAAG;AACnD,aAAO;AAAA,QACH,SAAS,CAAC;AAAA,UACN,MAAM;AAAA,UACN,MACI,iDAAiD,QAAQ;AAAA;AAAA;AAAA;AAAA,WAE7C,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAsB5B,CAAC;AAAA,MACL;AAAA,IACJ;AAEA,UAAM,YAAY,MAAM,aAAa,IAAI;AACzC,UAAM,SAAS,MAAM,UAAU,MAAM,EAAE,aAAa,CAAC;AAErD,UAAM,QAAQ,CAAC,wBAAwB,EAAE;AAEzC,UAAM,aAAa,OAAO;AAC1B,UAAM,YAAY,OAAO;AAEzB,UAAM,KAAK,aAAa,YAAY,WAAW,CAAC,mBAAmB,YAAY,WAAW,CAAC,aAAa,YAAY,UAAU,CAAC,SAAS;AACxI,UAAM,KAAK,YAAY,WAAW,WAAW,CAAC,qBAAqB,WAAW,WAAW,CAAC,UAAU;AAEpG,UAAM,aAAa,OAAO;AAC1B,QAAI,YAAY;AACZ,iBAAW,CAAC,MAAM,IAAI,KAAK,OAAO,QAAQ,UAAU,GAAG;AACnD,cAAM,KAAK,WAAW,IAAI,QAAQ,KAAK,OAAO,aAAa,KAAK,OAAO,aAAa,KAAK,MAAM,SAAS;AAAA,MAC5G;AAAA,IACJ;AAEA,UAAM,QAAQ,UAAU,MAAM;AAC9B,UAAM,YAAY,MAAM;AACxB,UAAM,WAAW,MAAM;AACvB,UAAM,WAAW,MAAM;AACvB,UAAM,KAAK,EAAE;AACb,UAAM,KAAK,eAAe,WAAW,UAAU,CAAC,iBAAiB,UAAU,WAAW,CAAC,aAAa,UAAU,aAAa,CAAC,OAAO;AAEnI,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,MAAM,KAAK,IAAI,EAAE,CAAC,EAAE;AAAA,EAC1E;AACJ;AAIA,OAAO;AAAA,EACH;AAAA,EACA;AAAA,IACI,OAAO;AAAA,IACP,aACI;AAAA,IAQJ,aAAa,EAAE,OAAO;AAAA,MAClB,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;AAAA,QACvB;AAAA,MAEJ;AAAA,MACA,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,2CAA2C;AAAA,MAChF,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,0CAA0C;AAAA,IACrF,CAAC;AAAA,EACL;AAAA,EACA,OAAO,EAAE,OAAO,MAAM,MAAM,MAAM;AAC9B,UAAM,YAAY,MAAM,aAAa,IAAI;AACzC,UAAM,UAAU,UAAU,aAAa,KAAK;AAE5C,QAAI,QAAQ,WAAW,GAAG;AACtB,aAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,wCAAwC,CAAC,EAAE;AAAA,IACjG;AAGA,UAAM,QAAkB,CAAC;AACzB,eAAW,KAAK,SAAS;AACrB,YAAM,OAAO,EAAE;AACf,YAAM,OAAQ,KAAK,YAAuB;AAC1C,YAAM,YAAa,KAAK,aAAwB;AAEhD,YAAM,KAAK,MAAM,EAAE,QAAQ;AAAA,CAAI;AAC/B,YAAM,KAAK,QAAQ,IAAI;AAEvB,UAAI,OAAO;AACP,cAAM,YAAY,EAAE,QAAQ,MAAM,IAAI;AACtC,cAAM,MAAM,OAAO,YAAY,UAAU,SAAS,CAAC,EAAE;AACrD,cAAM,KAAK,UAAU;AAAA,UAAI,CAAC,GAAG,MACzB,GAAG,OAAO,YAAY,CAAC,EAAE,SAAS,GAAG,CAAC,KAAK,CAAC;AAAA,QAChD,EAAE,KAAK,IAAI,CAAC;AAAA,MAChB,OAAO;AACH,cAAM,KAAK,EAAE,OAAO;AAAA,MACxB;AAEA,YAAM,KAAK,OAAO;AAAA,IACtB;AAEA,WAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAiB,MAAM,MAAM,KAAK,IAAI,EAAE,CAAC,EAAE;AAAA,EAC1E;AACJ;AAIA,eAAe,OAAO;AAClB,QAAM,YAAY,IAAI,qBAAqB;AAC3C,QAAM,OAAO,QAAQ,SAAS;AAClC;AAHe;AAKf,KAAK,EAAE,MAAM,SAAO;AAChB,QAAM,UAAU,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;AAC/D,UAAQ,MAAM,+BAA+B,OAAO,EAAE;AACtD,UAAQ,KAAK,CAAC;AAClB,CAAC;","names":["existsSync","path","existsSync"]}
|
package/package.json
CHANGED
|
@@ -1,21 +1,33 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@brainbank/mcp",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.3.1",
|
|
4
4
|
"description": "MCP (Model Context Protocol) server for BrainBank",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/mcp-server.js",
|
|
7
7
|
"bin": "dist/mcp-server.js",
|
|
8
8
|
"files": [
|
|
9
|
-
"dist/"
|
|
9
|
+
"dist/",
|
|
10
|
+
"src/"
|
|
10
11
|
],
|
|
11
12
|
"scripts": {
|
|
12
|
-
"build": "tsup"
|
|
13
|
+
"build": "tsup",
|
|
14
|
+
"prepublishOnly": "npm run build"
|
|
13
15
|
},
|
|
14
16
|
"dependencies": {
|
|
15
17
|
"brainbank": ">=0.2.2",
|
|
16
18
|
"@modelcontextprotocol/sdk": "^1.27.1",
|
|
17
19
|
"zod": "^4.3.6"
|
|
18
20
|
},
|
|
21
|
+
"peerDependencies": {
|
|
22
|
+
"@brainbank/code": ">=0.1.0",
|
|
23
|
+
"@brainbank/git": ">=0.1.0",
|
|
24
|
+
"@brainbank/docs": ">=0.1.0"
|
|
25
|
+
},
|
|
26
|
+
"peerDependenciesMeta": {
|
|
27
|
+
"@brainbank/code": { "optional": true },
|
|
28
|
+
"@brainbank/git": { "optional": true },
|
|
29
|
+
"@brainbank/docs": { "optional": true }
|
|
30
|
+
},
|
|
19
31
|
"repository": {
|
|
20
32
|
"type": "git",
|
|
21
33
|
"url": "git+https://github.com/pinecall/brainbank.git",
|
|
@@ -0,0 +1,265 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
|
|
3
|
+
/**
|
|
4
|
+
* BrainBank — MCP Server
|
|
5
|
+
*
|
|
6
|
+
* Exposes BrainBank as an MCP server via stdio transport.
|
|
7
|
+
* Works with Google Antigravity, Claude Desktop, and any MCP-compatible client.
|
|
8
|
+
*
|
|
9
|
+
* Usage in mcp_config.json:
|
|
10
|
+
* {
|
|
11
|
+
* "mcpServers": {
|
|
12
|
+
* "brainbank": {
|
|
13
|
+
* "command": "npx",
|
|
14
|
+
* "args": ["-y", "@brainbank/mcp"]
|
|
15
|
+
* }
|
|
16
|
+
* }
|
|
17
|
+
* }
|
|
18
|
+
*
|
|
19
|
+
* Tools (2):
|
|
20
|
+
* brainbank_context — Workflow Trace: search + call tree + called-by annotations
|
|
21
|
+
* brainbank_index — Re-index (requires .brainbank/config.json)
|
|
22
|
+
*/
|
|
23
|
+
|
|
24
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
25
|
+
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
26
|
+
import { z } from 'zod/v3';
|
|
27
|
+
import { existsSync } from 'node:fs';
|
|
28
|
+
import { WorkspacePool } from './workspace-pool.js';
|
|
29
|
+
import { createWorkspaceBrain, resolveRepoPath } from './workspace-factory.js';
|
|
30
|
+
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
// ── Multi-Workspace BrainBank Pool ─────────────────────
|
|
34
|
+
|
|
35
|
+
const pool = new WorkspacePool({
|
|
36
|
+
factory: createWorkspaceBrain,
|
|
37
|
+
maxMemoryMB: parseInt(process.env.BRAINBANK_MAX_MEMORY_MB ?? '2048', 10),
|
|
38
|
+
ttlMinutes: parseInt(process.env.BRAINBANK_TTL_MINUTES ?? '30', 10),
|
|
39
|
+
onError: (repo, err) => {
|
|
40
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
41
|
+
console.error(`BrainBank pool error [${repo}]: ${msg}`);
|
|
42
|
+
},
|
|
43
|
+
});
|
|
44
|
+
|
|
45
|
+
/** Resolve repo and get a BrainBank from the pool. */
|
|
46
|
+
async function getBrainBank(targetRepo?: string) {
|
|
47
|
+
return pool.get(resolveRepoPath(targetRepo));
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
// ── MCP Server Setup ────────────────────────────────
|
|
51
|
+
|
|
52
|
+
const server = new McpServer({
|
|
53
|
+
name: 'brainbank',
|
|
54
|
+
version: '0.4.0',
|
|
55
|
+
});
|
|
56
|
+
|
|
57
|
+
// ── Tool: brainbank_context ─────────────────────────
|
|
58
|
+
|
|
59
|
+
server.registerTool(
|
|
60
|
+
'brainbank_context',
|
|
61
|
+
{
|
|
62
|
+
title: 'BrainBank Context',
|
|
63
|
+
description:
|
|
64
|
+
'Get a formatted knowledge context block for a task. Returns a Workflow Trace: ' +
|
|
65
|
+
'search hits + full call tree with `called by` annotations, topologically ordered. ' +
|
|
66
|
+
'All source code included — no trimming, no truncation.',
|
|
67
|
+
inputSchema: z.object({
|
|
68
|
+
task: z.string().describe('Description of the task you need context for'),
|
|
69
|
+
affectedFiles: z.array(z.string()).optional().default([]).describe('Files you plan to modify (improves co-edit suggestions)'),
|
|
70
|
+
codeResults: z.number().optional().default(20).describe('Max code results'),
|
|
71
|
+
gitResults: z.number().optional().default(5).describe('Max git commit results'),
|
|
72
|
+
docsResults: z.number().optional().describe('Max document results (omit to skip docs)'),
|
|
73
|
+
sources: z.record(z.number()).optional().describe('Per-source result limits, overrides codeResults/gitResults/docsResults (e.g. { code: 10, git: 0, docs: 5 })'),
|
|
74
|
+
path: z.string().optional().describe('Filter results to files under this path prefix (e.g. src/services/)'),
|
|
75
|
+
repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),
|
|
76
|
+
// BrainBankQL context fields
|
|
77
|
+
lines: z.boolean().optional().describe('Prefix each code line with its source line number (e.g. 127| code)'),
|
|
78
|
+
symbols: z.boolean().optional().describe('Append symbol index (all functions, classes, interfaces) for matched files'),
|
|
79
|
+
compact: z.boolean().optional().describe('Show only function/class signatures, skip bodies'),
|
|
80
|
+
callTree: z.union([z.boolean(), z.object({ depth: z.number() })]).optional().describe('Include call tree expansion. Pass { depth: N } to control depth'),
|
|
81
|
+
imports: z.boolean().optional().describe('Include dependency/import summary section'),
|
|
82
|
+
expander: z.boolean().optional().describe('Enable LLM-powered context expansion to discover related chunks not found by search'),
|
|
83
|
+
}),
|
|
84
|
+
},
|
|
85
|
+
async ({ task, affectedFiles, codeResults, gitResults, docsResults, sources, path, repo, lines, symbols, compact, callTree, imports, expander }) => {
|
|
86
|
+
const repoPath = resolveRepoPath(repo);
|
|
87
|
+
const brainbank = await getBrainBank(repo);
|
|
88
|
+
|
|
89
|
+
// Build sources from explicit params, then let `sources` override
|
|
90
|
+
const base: Record<string, number> = { code: codeResults, git: gitResults };
|
|
91
|
+
if (docsResults !== undefined) base.docs = docsResults;
|
|
92
|
+
const resolvedSources = sources ? { ...base, ...sources } : base;
|
|
93
|
+
|
|
94
|
+
// Build fields from explicit params (only include defined values)
|
|
95
|
+
const fields: Record<string, unknown> = {};
|
|
96
|
+
if (lines !== undefined) fields.lines = lines;
|
|
97
|
+
if (symbols !== undefined) fields.symbols = symbols;
|
|
98
|
+
if (compact !== undefined) fields.compact = compact;
|
|
99
|
+
if (callTree !== undefined) fields.callTree = callTree;
|
|
100
|
+
if (imports !== undefined) fields.imports = imports;
|
|
101
|
+
if (expander !== undefined) fields.expander = expander;
|
|
102
|
+
|
|
103
|
+
const context = await brainbank.getContext(task, {
|
|
104
|
+
affectedFiles,
|
|
105
|
+
sources: resolvedSources,
|
|
106
|
+
pathPrefix: path,
|
|
107
|
+
source: 'mcp',
|
|
108
|
+
fields: Object.keys(fields).length > 0 ? fields : undefined,
|
|
109
|
+
});
|
|
110
|
+
|
|
111
|
+
return { content: [{ type: 'text' as const, text: context }] };
|
|
112
|
+
},
|
|
113
|
+
);
|
|
114
|
+
|
|
115
|
+
// ── Tool: brainbank_index ───────────────────────────
|
|
116
|
+
|
|
117
|
+
server.registerTool(
|
|
118
|
+
'brainbank_index',
|
|
119
|
+
{
|
|
120
|
+
title: 'BrainBank Index',
|
|
121
|
+
description:
|
|
122
|
+
'Re-index code, git history, and docs. Requires .brainbank/config.json to exist. ' +
|
|
123
|
+
'Incremental — only changed files are processed.',
|
|
124
|
+
inputSchema: z.object({
|
|
125
|
+
forceReindex: z.boolean().optional().default(false).describe('Force re-index of all files'),
|
|
126
|
+
repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),
|
|
127
|
+
}),
|
|
128
|
+
},
|
|
129
|
+
async ({ forceReindex, repo }) => {
|
|
130
|
+
const repoPath = resolveRepoPath(repo);
|
|
131
|
+
|
|
132
|
+
// Require config.json — force users to set up structure first
|
|
133
|
+
if (!existsSync(`${repoPath}/.brainbank/config.json`)) {
|
|
134
|
+
return {
|
|
135
|
+
content: [{
|
|
136
|
+
type: 'text' as const,
|
|
137
|
+
text:
|
|
138
|
+
`BrainBank: No .brainbank/config.json found at ${repoPath}.\n\n` +
|
|
139
|
+
`## How to set up\n\n` +
|
|
140
|
+
`Create \`${repoPath}/.brainbank/config.json\` with:\n\n` +
|
|
141
|
+
'```json\n' +
|
|
142
|
+
'{\n' +
|
|
143
|
+
' "plugins": ["code"],\n' +
|
|
144
|
+
' "code": {\n' +
|
|
145
|
+
' "embedding": "perplexity-context",\n' +
|
|
146
|
+
' "ignore": [\n' +
|
|
147
|
+
' "node_modules/**", "dist/**", "build/**",\n' +
|
|
148
|
+
' ".next/**", "coverage/**", "__pycache__/**",\n' +
|
|
149
|
+
' "**/*.min.js", "**/*.min.css",\n' +
|
|
150
|
+
' "tests/**", "test/**",\n' +
|
|
151
|
+
' "**/test_*.py", "**/*_test.py",\n' +
|
|
152
|
+
' "**/*.test.ts", "**/*.spec.ts"\n' +
|
|
153
|
+
' ]\n' +
|
|
154
|
+
' },\n' +
|
|
155
|
+
' "embedding": "perplexity-context"\n' +
|
|
156
|
+
'}\n' +
|
|
157
|
+
'```\n\n' +
|
|
158
|
+
`**Embedding options:** \`local\` (free, offline), \`openai\`, \`perplexity\`, \`perplexity-context\` (best quality)\n` +
|
|
159
|
+
`**Plugins available:** \`code\`, \`git\`, \`docs\`\n\n` +
|
|
160
|
+
`Then run:\n` +
|
|
161
|
+
'```bash\nbrainbank index . --force --yes\n```',
|
|
162
|
+
}],
|
|
163
|
+
};
|
|
164
|
+
}
|
|
165
|
+
|
|
166
|
+
const brainbank = await getBrainBank(repo);
|
|
167
|
+
const result = await brainbank.index({ forceReindex });
|
|
168
|
+
|
|
169
|
+
const lines = ['## Indexing Complete', ''];
|
|
170
|
+
|
|
171
|
+
const codeResult = result.code as { indexed?: number; skipped?: number; chunks?: number } | undefined;
|
|
172
|
+
const gitResult = result.git as { indexed?: number; skipped?: number } | undefined;
|
|
173
|
+
|
|
174
|
+
lines.push(`**Code**: ${codeResult?.indexed ?? 0} files indexed, ${codeResult?.skipped ?? 0} skipped, ${codeResult?.chunks ?? 0} chunks`);
|
|
175
|
+
lines.push(`**Git**: ${gitResult?.indexed ?? 0} commits indexed, ${gitResult?.skipped ?? 0} skipped`);
|
|
176
|
+
|
|
177
|
+
const docsResult = result.docs as Record<string, { indexed: number; skipped: number; chunks: number }> | undefined;
|
|
178
|
+
if (docsResult) {
|
|
179
|
+
for (const [name, stat] of Object.entries(docsResult)) {
|
|
180
|
+
lines.push(`**Docs [${name}]**: ${stat.indexed} indexed, ${stat.skipped} skipped, ${stat.chunks} chunks`);
|
|
181
|
+
}
|
|
182
|
+
}
|
|
183
|
+
|
|
184
|
+
const stats = brainbank.stats();
|
|
185
|
+
const codeStats = stats.code as { chunks?: number } | undefined;
|
|
186
|
+
const gitStats = stats.git as { commits?: number } | undefined;
|
|
187
|
+
const docStats = stats.documents as { documents?: number } | undefined;
|
|
188
|
+
lines.push('');
|
|
189
|
+
lines.push(`**Totals**: ${codeStats?.chunks ?? 0} code chunks, ${gitStats?.commits ?? 0} commits, ${docStats?.documents ?? 0} docs`);
|
|
190
|
+
|
|
191
|
+
return { content: [{ type: 'text' as const, text: lines.join('\n') }] };
|
|
192
|
+
},
|
|
193
|
+
);
|
|
194
|
+
|
|
195
|
+
// ── Tool: brainbank_files ───────────────────────────
|
|
196
|
+
|
|
197
|
+
server.registerTool(
|
|
198
|
+
'brainbank_files',
|
|
199
|
+
{
|
|
200
|
+
title: 'BrainBank Files',
|
|
201
|
+
description:
|
|
202
|
+
'Fetch full file contents from the index. Use AFTER brainbank_context ' +
|
|
203
|
+
'to view complete files identified by search. No semantic search runs — ' +
|
|
204
|
+
'this is a direct file viewer.\n\n' +
|
|
205
|
+
'Supports:\n' +
|
|
206
|
+
'- Exact paths: "src/auth/login.ts"\n' +
|
|
207
|
+
'- Directories: "src/graph/" (trailing / = all files under path)\n' +
|
|
208
|
+
'- Glob patterns: "src/**/*.service.ts"\n' +
|
|
209
|
+
'- Fuzzy basename: "plugin.ts" (matches src/plugin.ts when exact fails)',
|
|
210
|
+
inputSchema: z.object({
|
|
211
|
+
files: z.array(z.string()).describe(
|
|
212
|
+
'File paths to fetch. Exact paths, directories (trailing /), ' +
|
|
213
|
+
'glob patterns (e.g. src/**/*.ts), or fuzzy basenames.',
|
|
214
|
+
),
|
|
215
|
+
repo: z.string().optional().describe('Repository path (default: BRAINBANK_REPO)'),
|
|
216
|
+
lines: z.boolean().optional().describe('Prefix each line with source line number'),
|
|
217
|
+
}),
|
|
218
|
+
},
|
|
219
|
+
async ({ files, repo, lines }) => {
|
|
220
|
+
const brainbank = await getBrainBank(repo);
|
|
221
|
+
const results = brainbank.resolveFiles(files);
|
|
222
|
+
|
|
223
|
+
if (results.length === 0) {
|
|
224
|
+
return { content: [{ type: 'text' as const, text: 'No matching files found in the index.' }] };
|
|
225
|
+
}
|
|
226
|
+
|
|
227
|
+
// Format: markdown with file headers + fenced code blocks
|
|
228
|
+
const parts: string[] = [];
|
|
229
|
+
for (const r of results) {
|
|
230
|
+
const meta = r.metadata as Record<string, unknown>;
|
|
231
|
+
const lang = (meta.language as string) ?? '';
|
|
232
|
+
const startLine = (meta.startLine as number) ?? 1;
|
|
233
|
+
|
|
234
|
+
parts.push(`## ${r.filePath}\n`);
|
|
235
|
+
parts.push('```' + lang);
|
|
236
|
+
|
|
237
|
+
if (lines) {
|
|
238
|
+
const codeLines = r.content.split('\n');
|
|
239
|
+
const pad = String(startLine + codeLines.length - 1).length;
|
|
240
|
+
parts.push(codeLines.map((l, i) =>
|
|
241
|
+
`${String(startLine + i).padStart(pad)}| ${l}`,
|
|
242
|
+
).join('\n'));
|
|
243
|
+
} else {
|
|
244
|
+
parts.push(r.content);
|
|
245
|
+
}
|
|
246
|
+
|
|
247
|
+
parts.push('```\n');
|
|
248
|
+
}
|
|
249
|
+
|
|
250
|
+
return { content: [{ type: 'text' as const, text: parts.join('\n') }] };
|
|
251
|
+
},
|
|
252
|
+
);
|
|
253
|
+
|
|
254
|
+
// ── Start Server ────────────────────────────────────
|
|
255
|
+
|
|
256
|
+
async function main() {
|
|
257
|
+
const transport = new StdioServerTransport();
|
|
258
|
+
await server.connect(transport);
|
|
259
|
+
}
|
|
260
|
+
|
|
261
|
+
main().catch(err => {
|
|
262
|
+
const message = err instanceof Error ? err.message : String(err);
|
|
263
|
+
console.error(`BrainBank MCP Server Error: ${message}`);
|
|
264
|
+
process.exit(1);
|
|
265
|
+
});
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* WorkspaceFactory — creates BrainBank instances via the core factory.
|
|
3
|
+
*
|
|
4
|
+
* Delegates to `createBrain()` from the `brainbank` package, passing
|
|
5
|
+
* a portable `BrainContext`. No plugin hardcoding — the factory handles
|
|
6
|
+
* plugin discovery from config and installed packages.
|
|
7
|
+
*/
|
|
8
|
+
|
|
9
|
+
import type { BrainBank, BrainContext } from 'brainbank';
|
|
10
|
+
|
|
11
|
+
import * as fs from 'node:fs';
|
|
12
|
+
import * as path from 'node:path';
|
|
13
|
+
|
|
14
|
+
/**
|
|
15
|
+
* Detect repo root by walking up from startDir until we find `.git/`.
|
|
16
|
+
* Returns startDir itself if no `.git/` is found.
|
|
17
|
+
*/
|
|
18
|
+
export function findRepoRoot(startDir: string): string {
|
|
19
|
+
let dir = path.resolve(startDir);
|
|
20
|
+
while (true) {
|
|
21
|
+
if (fs.existsSync(path.join(dir, '.git'))) return dir;
|
|
22
|
+
const parent = path.dirname(dir);
|
|
23
|
+
if (parent === dir) break;
|
|
24
|
+
dir = parent;
|
|
25
|
+
}
|
|
26
|
+
return path.resolve(startDir);
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
/** Resolve the effective repo path from an optional target, env, or cwd. */
|
|
30
|
+
export function resolveRepoPath(targetRepo?: string): string {
|
|
31
|
+
const rp = targetRepo
|
|
32
|
+
?? process.env.BRAINBANK_REPO
|
|
33
|
+
?? findRepoRoot(process.cwd());
|
|
34
|
+
return rp.replace(/\/+$/, '');
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
/**
|
|
38
|
+
* Create a BrainBank instance for a workspace.
|
|
39
|
+
* Uses the core factory which handles:
|
|
40
|
+
* - Config loading from .brainbank/config.json
|
|
41
|
+
* - Dynamic plugin discovery and registration
|
|
42
|
+
* - Embedding/reranker provider setup
|
|
43
|
+
* - Folder plugin auto-discovery
|
|
44
|
+
*/
|
|
45
|
+
export async function createWorkspaceBrain(repoPath: string): Promise<BrainBank> {
|
|
46
|
+
const { createBrain, resetFactoryCache } = await import('brainbank') as typeof import('brainbank');
|
|
47
|
+
resetFactoryCache();
|
|
48
|
+
|
|
49
|
+
const context: BrainContext = {
|
|
50
|
+
repoPath,
|
|
51
|
+
env: process.env as Record<string, string | undefined>,
|
|
52
|
+
};
|
|
53
|
+
|
|
54
|
+
// Silence stdout during initialization — the core factory emits ANSI-colored
|
|
55
|
+
// console.log messages (plugin loading, multi-repo detection) that corrupt
|
|
56
|
+
// the MCP JSON-RPC stdio transport. Redirect console.log → stderr temporarily.
|
|
57
|
+
const origLog = console.log;
|
|
58
|
+
console.log = (...args: unknown[]) => console.error(...args);
|
|
59
|
+
try {
|
|
60
|
+
const brain = await createBrain(context);
|
|
61
|
+
await brain.initialize();
|
|
62
|
+
return brain;
|
|
63
|
+
} finally {
|
|
64
|
+
console.log = origLog;
|
|
65
|
+
}
|
|
66
|
+
}
|