edhindex 1.0.0 → 1.0.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.
Files changed (199) hide show
  1. package/README.md +101 -0
  2. package/dist/cli/commands/config_cmd.d.ts +2 -0
  3. package/dist/cli/commands/config_cmd.d.ts.map +1 -0
  4. package/dist/cli/commands/config_cmd.js +53 -0
  5. package/dist/cli/commands/config_cmd.js.map +1 -0
  6. package/dist/cli/commands/doctor_cmd.d.ts +2 -0
  7. package/dist/cli/commands/doctor_cmd.d.ts.map +1 -0
  8. package/dist/cli/commands/doctor_cmd.js +32 -0
  9. package/dist/cli/commands/doctor_cmd.js.map +1 -0
  10. package/dist/cli/commands/graph.d.ts +2 -0
  11. package/dist/cli/commands/graph.d.ts.map +1 -0
  12. package/dist/cli/commands/graph.js +26 -0
  13. package/dist/cli/commands/graph.js.map +1 -0
  14. package/dist/cli/commands/index_cmd.d.ts +2 -0
  15. package/dist/cli/commands/index_cmd.d.ts.map +1 -0
  16. package/dist/cli/commands/index_cmd.js +110 -0
  17. package/dist/cli/commands/index_cmd.js.map +1 -0
  18. package/dist/cli/commands/init.d.ts +2 -0
  19. package/dist/cli/commands/init.d.ts.map +1 -0
  20. package/dist/cli/commands/init.js +53 -0
  21. package/dist/cli/commands/init.js.map +1 -0
  22. package/dist/cli/commands/knowledge-graph.d.ts +8 -0
  23. package/dist/cli/commands/knowledge-graph.d.ts.map +1 -0
  24. package/dist/cli/commands/knowledge-graph.js +76 -0
  25. package/dist/cli/commands/knowledge-graph.js.map +1 -0
  26. package/dist/cli/commands/models.d.ts +2 -0
  27. package/dist/cli/commands/models.d.ts.map +1 -0
  28. package/dist/cli/commands/models.js +17 -0
  29. package/dist/cli/commands/models.js.map +1 -0
  30. package/dist/cli/commands/reset.d.ts +2 -0
  31. package/dist/cli/commands/reset.d.ts.map +1 -0
  32. package/dist/cli/commands/reset.js +35 -0
  33. package/dist/cli/commands/reset.js.map +1 -0
  34. package/dist/cli/commands/search.d.ts +2 -0
  35. package/dist/cli/commands/search.d.ts.map +1 -0
  36. package/dist/cli/commands/search.js +40 -0
  37. package/dist/cli/commands/search.js.map +1 -0
  38. package/dist/cli/commands/start.d.ts +2 -0
  39. package/dist/cli/commands/start.d.ts.map +1 -0
  40. package/dist/cli/commands/start.js +265 -0
  41. package/dist/cli/commands/start.js.map +1 -0
  42. package/dist/cli/commands/status.d.ts +2 -0
  43. package/dist/cli/commands/status.d.ts.map +1 -0
  44. package/dist/cli/commands/status.js +47 -0
  45. package/dist/cli/commands/status.js.map +1 -0
  46. package/dist/cli/index.d.ts +3 -0
  47. package/dist/cli/index.d.ts.map +1 -0
  48. package/dist/cli/index.js +121 -0
  49. package/dist/cli/index.js.map +1 -0
  50. package/dist/config/index.d.ts +28 -0
  51. package/dist/config/index.d.ts.map +1 -0
  52. package/dist/config/index.js +65 -0
  53. package/dist/config/index.js.map +1 -0
  54. package/dist/doctor/index.d.ts +8 -0
  55. package/dist/doctor/index.d.ts.map +1 -0
  56. package/dist/doctor/index.js +117 -0
  57. package/dist/doctor/index.js.map +1 -0
  58. package/dist/embeddings/provider.d.ts +9 -0
  59. package/dist/embeddings/provider.d.ts.map +1 -0
  60. package/dist/embeddings/provider.js +2 -0
  61. package/dist/embeddings/provider.js.map +1 -0
  62. package/dist/embeddings/transformers.d.ts +26 -0
  63. package/dist/embeddings/transformers.d.ts.map +1 -0
  64. package/dist/embeddings/transformers.js +120 -0
  65. package/dist/embeddings/transformers.js.map +1 -0
  66. package/dist/graph/build.d.ts +13 -0
  67. package/dist/graph/build.d.ts.map +1 -0
  68. package/dist/graph/build.js +33 -0
  69. package/dist/graph/build.js.map +1 -0
  70. package/dist/graph/builder.d.ts +24 -0
  71. package/dist/graph/builder.d.ts.map +1 -0
  72. package/dist/graph/builder.js +231 -0
  73. package/dist/graph/builder.js.map +1 -0
  74. package/dist/graph/render.d.ts +3 -0
  75. package/dist/graph/render.d.ts.map +1 -0
  76. package/dist/graph/render.js +60 -0
  77. package/dist/graph/render.js.map +1 -0
  78. package/dist/graph/serve.d.ts +6 -0
  79. package/dist/graph/serve.d.ts.map +1 -0
  80. package/dist/graph/serve.js +389 -0
  81. package/dist/graph/serve.js.map +1 -0
  82. package/dist/graph/service.d.ts +29 -0
  83. package/dist/graph/service.d.ts.map +1 -0
  84. package/dist/graph/service.js +73 -0
  85. package/dist/graph/service.js.map +1 -0
  86. package/dist/graph/storage.d.ts +33 -0
  87. package/dist/graph/storage.d.ts.map +1 -0
  88. package/dist/graph/storage.js +164 -0
  89. package/dist/graph/storage.js.map +1 -0
  90. package/dist/graph/svg.d.ts +27 -0
  91. package/dist/graph/svg.d.ts.map +1 -0
  92. package/dist/graph/svg.js +212 -0
  93. package/dist/graph/svg.js.map +1 -0
  94. package/dist/graph/types.d.ts +34 -0
  95. package/dist/graph/types.d.ts.map +1 -0
  96. package/dist/graph/types.js +9 -0
  97. package/dist/graph/types.js.map +1 -0
  98. package/dist/indexer/chunker.d.ts +18 -0
  99. package/dist/indexer/chunker.d.ts.map +1 -0
  100. package/dist/indexer/chunker.js +139 -0
  101. package/dist/indexer/chunker.js.map +1 -0
  102. package/dist/indexer/file-utils.d.ts +17 -0
  103. package/dist/indexer/file-utils.d.ts.map +1 -0
  104. package/dist/indexer/file-utils.js +123 -0
  105. package/dist/indexer/file-utils.js.map +1 -0
  106. package/dist/indexer/ignore.d.ts +5 -0
  107. package/dist/indexer/ignore.d.ts.map +1 -0
  108. package/dist/indexer/ignore.js +96 -0
  109. package/dist/indexer/ignore.js.map +1 -0
  110. package/dist/indexer/incremental.d.ts +16 -0
  111. package/dist/indexer/incremental.d.ts.map +1 -0
  112. package/dist/indexer/incremental.js +55 -0
  113. package/dist/indexer/incremental.js.map +1 -0
  114. package/dist/indexer/indexer.d.ts +36 -0
  115. package/dist/indexer/indexer.d.ts.map +1 -0
  116. package/dist/indexer/indexer.js +121 -0
  117. package/dist/indexer/indexer.js.map +1 -0
  118. package/dist/indexer/parser.d.ts +30 -0
  119. package/dist/indexer/parser.d.ts.map +1 -0
  120. package/dist/indexer/parser.js +225 -0
  121. package/dist/indexer/parser.js.map +1 -0
  122. package/dist/logging.d.ts +15 -0
  123. package/dist/logging.d.ts.map +1 -0
  124. package/dist/logging.js +42 -0
  125. package/dist/logging.js.map +1 -0
  126. package/dist/mcp/server.d.ts +4 -0
  127. package/dist/mcp/server.d.ts.map +1 -0
  128. package/dist/mcp/server.js +226 -0
  129. package/dist/mcp/server.js.map +1 -0
  130. package/dist/mcp-config.d.ts +4 -0
  131. package/dist/mcp-config.d.ts.map +1 -0
  132. package/dist/mcp-config.js +175 -0
  133. package/dist/mcp-config.js.map +1 -0
  134. package/dist/progress.d.ts +12 -0
  135. package/dist/progress.d.ts.map +1 -0
  136. package/dist/progress.js +16 -0
  137. package/dist/progress.js.map +1 -0
  138. package/dist/retrieval/bm25.d.ts +6 -0
  139. package/dist/retrieval/bm25.d.ts.map +1 -0
  140. package/dist/retrieval/bm25.js +20 -0
  141. package/dist/retrieval/bm25.js.map +1 -0
  142. package/dist/retrieval/reranker.d.ts +11 -0
  143. package/dist/retrieval/reranker.d.ts.map +1 -0
  144. package/dist/retrieval/reranker.js +46 -0
  145. package/dist/retrieval/reranker.js.map +1 -0
  146. package/dist/retrieval/search.d.ts +35 -0
  147. package/dist/retrieval/search.d.ts.map +1 -0
  148. package/dist/retrieval/search.js +95 -0
  149. package/dist/retrieval/search.js.map +1 -0
  150. package/dist/security.d.ts +5 -0
  151. package/dist/security.d.ts.map +1 -0
  152. package/dist/security.js +42 -0
  153. package/dist/security.js.map +1 -0
  154. package/dist/storage/fts.d.ts +30 -0
  155. package/dist/storage/fts.d.ts.map +1 -0
  156. package/dist/storage/fts.js +107 -0
  157. package/dist/storage/fts.js.map +1 -0
  158. package/dist/storage/metadata.d.ts +46 -0
  159. package/dist/storage/metadata.d.ts.map +1 -0
  160. package/dist/storage/metadata.js +133 -0
  161. package/dist/storage/metadata.js.map +1 -0
  162. package/dist/vector/lancedb.d.ts +25 -0
  163. package/dist/vector/lancedb.d.ts.map +1 -0
  164. package/dist/vector/lancedb.js +93 -0
  165. package/dist/vector/lancedb.js.map +1 -0
  166. package/dist/vector/store.d.ts +16 -0
  167. package/dist/vector/store.d.ts.map +1 -0
  168. package/dist/vector/store.js +2 -0
  169. package/dist/vector/store.js.map +1 -0
  170. package/dist/version.d.ts +12 -0
  171. package/dist/version.d.ts.map +1 -0
  172. package/dist/version.js +46 -0
  173. package/dist/version.js.map +1 -0
  174. package/dist/watcher/index.d.ts +15 -0
  175. package/dist/watcher/index.d.ts.map +1 -0
  176. package/dist/watcher/index.js +65 -0
  177. package/dist/watcher/index.js.map +1 -0
  178. package/package.json +9 -3
  179. package/src/cli/commands/graph.ts +30 -0
  180. package/src/cli/commands/init.ts +41 -3
  181. package/src/cli/commands/knowledge-graph.ts +85 -0
  182. package/src/cli/commands/reset.ts +40 -0
  183. package/src/cli/commands/start.ts +54 -21
  184. package/src/cli/index.ts +33 -0
  185. package/src/config/index.ts +13 -0
  186. package/src/graph/build.ts +45 -0
  187. package/src/graph/builder.ts +243 -0
  188. package/src/graph/render.ts +67 -0
  189. package/src/graph/serve.ts +399 -0
  190. package/src/graph/service.ts +84 -0
  191. package/src/graph/storage.ts +181 -0
  192. package/src/graph/svg.ts +241 -0
  193. package/src/graph/types.ts +68 -0
  194. package/src/mcp/server.ts +212 -85
  195. package/src/mcp-config.ts +185 -0
  196. package/src/storage/metadata.ts +9 -1
  197. package/.edhindexignore +0 -66
  198. package/BUILD_BRIEF.md +0 -228
  199. package/tsconfig.json +0 -21
@@ -0,0 +1,226 @@
1
+ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
2
+ import { ListToolsRequestSchema, CallToolRequestSchema, ListResourcesRequestSchema, ReadResourceRequestSchema, McpError, ErrorCode, } from '@modelcontextprotocol/sdk/types.js';
3
+ import { GraphService } from '../graph/service.js';
4
+ import { renderGraphSVG } from '../graph/render.js';
5
+ import { logger } from '../logging.js';
6
+ export function createMCPServer(searchEngine, rootPath) {
7
+ let graphSvc = null;
8
+ function getGraph() {
9
+ if (!graphSvc) {
10
+ try {
11
+ graphSvc = new GraphService(rootPath);
12
+ }
13
+ catch {
14
+ throw new McpError(ErrorCode.InternalError, 'No graph available — run `edhindex start` to build the index');
15
+ }
16
+ }
17
+ return graphSvc;
18
+ }
19
+ const server = new Server({ name: 'edhindex', version: '1.0.0' }, { capabilities: { tools: {}, resources: {} } });
20
+ server.setRequestHandler(ListToolsRequestSchema, async () => ({
21
+ tools: [
22
+ {
23
+ name: 'search_codebase',
24
+ description: `Search the indexed codebase using hybrid (keyword + semantic + reranked) search. Workspace: ${rootPath}. Searches file contents, symbols, and code structure.`,
25
+ inputSchema: {
26
+ type: 'object',
27
+ properties: {
28
+ query: { type: 'string', description: 'Search query' },
29
+ maxResults: { type: 'number', description: 'Max results (default 10)', default: 10 },
30
+ fileFilter: { type: 'string', description: 'File path filter (glob)' },
31
+ languageFilter: { type: 'string', description: 'Language filter (ts, js, py, go)' },
32
+ },
33
+ required: ['query'],
34
+ },
35
+ },
36
+ {
37
+ name: 'get_graph',
38
+ description: 'Get the full knowledge graph — all nodes and edges representing the indexed codebase structure. Returns nodes (files, folders, classes, functions, etc.) and their relationships (imports, exports, contains, defines).',
39
+ inputSchema: {
40
+ type: 'object',
41
+ properties: {
42
+ limit: { type: 'number', description: 'Max nodes to return (default all)' },
43
+ offset: { type: 'number', description: 'Pagination offset' },
44
+ type: { type: 'string', description: 'Filter by node type (class, function, file, folder, etc.)' },
45
+ },
46
+ },
47
+ },
48
+ {
49
+ name: 'get_node',
50
+ description: 'Get details for a specific node by ID, including all neighbors (connected nodes and edges).',
51
+ inputSchema: {
52
+ type: 'object',
53
+ properties: {
54
+ id: { type: 'string', description: 'Node ID (e.g. "kg:sym:src/app.ts:App")' },
55
+ file: { type: 'string', description: 'Alternative: find node by file path' },
56
+ symbol: { type: 'string', description: 'Symbol name (used with file)' },
57
+ },
58
+ },
59
+ },
60
+ {
61
+ name: 'search_graph',
62
+ description: 'Search nodes in the knowledge graph by name, symbol, or file path. Returns matching nodes with their connections.',
63
+ inputSchema: {
64
+ type: 'object',
65
+ properties: {
66
+ query: { type: 'string', description: 'Search query to match against node labels and file paths' },
67
+ type: { type: 'string', description: 'Filter by node type (class, function, file, etc.)' },
68
+ },
69
+ required: ['query'],
70
+ },
71
+ },
72
+ {
73
+ name: 'get_graph_stats',
74
+ description: 'Get statistics about the knowledge graph — total node count, edge count, breakdown by type.',
75
+ inputSchema: { type: 'object', properties: {} },
76
+ },
77
+ ],
78
+ }));
79
+ server.setRequestHandler(CallToolRequestSchema, async (request) => {
80
+ const name = request.params.name;
81
+ const args = request.params.arguments || {};
82
+ switch (name) {
83
+ case 'search_codebase': return handleSearch(args, searchEngine, rootPath);
84
+ case 'get_graph': return handleGetGraph(args, getGraph());
85
+ case 'get_node': return handleGetNode(args, getGraph());
86
+ case 'search_graph': return handleSearchGraph(args, getGraph());
87
+ case 'get_graph_stats': return handleGraphStats(getGraph());
88
+ default:
89
+ throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`);
90
+ }
91
+ });
92
+ server.setRequestHandler(ListResourcesRequestSchema, async () => ({
93
+ resources: [
94
+ {
95
+ uri: `edhindex://${encodeURIComponent(rootPath)}/graph`,
96
+ name: 'Knowledge Graph',
97
+ description: 'Full knowledge graph of the codebase — nodes and edges',
98
+ mimeType: 'application/json',
99
+ },
100
+ ],
101
+ }));
102
+ server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
103
+ const uri = request.params.uri;
104
+ if (uri.includes('/graph')) {
105
+ const svc = getGraph();
106
+ const data = svc.getGraphData();
107
+ return {
108
+ contents: [{ uri, mimeType: 'application/json', text: JSON.stringify(data, null, 2) }],
109
+ };
110
+ }
111
+ throw new McpError(ErrorCode.InvalidRequest, `Unknown resource: ${uri}`);
112
+ });
113
+ server.onerror = (error) => logger.error('MCP Server error:', error);
114
+ return server;
115
+ }
116
+ async function handleSearch(args, searchEngine, rootPath) {
117
+ const query = String(args.query || '').trim();
118
+ if (!query)
119
+ throw new McpError(ErrorCode.InvalidParams, 'query is required');
120
+ const maxResults = Math.min(Math.max(Number(args.maxResults) || 10, 1), 100);
121
+ const fileFilter = args.fileFilter ? String(args.fileFilter) : undefined;
122
+ const languageFilter = args.languageFilter ? String(args.languageFilter) : undefined;
123
+ if (fileFilter) {
124
+ const { sanitizePath } = await import('../security.js');
125
+ if (!sanitizePath(rootPath, fileFilter)) {
126
+ throw new McpError(ErrorCode.InvalidParams, 'File filter is outside the workspace');
127
+ }
128
+ }
129
+ try {
130
+ const results = await searchEngine.search({ query, maxResults, fileFilter, languageFilter });
131
+ return {
132
+ content: [
133
+ {
134
+ type: 'text',
135
+ text: formatResults(results),
136
+ },
137
+ ],
138
+ };
139
+ }
140
+ catch (e) {
141
+ logger.error('Search failed:', e);
142
+ throw new McpError(ErrorCode.InternalError, `Search failed: ${e.message}`);
143
+ }
144
+ }
145
+ function handleGetGraph(args, svc) {
146
+ const data = svc.getGraphData(args.limit, args.offset);
147
+ const type = args.type;
148
+ const nodes = type ? data.nodes.filter(n => n.type === type) : data.nodes;
149
+ const nodeIds = new Set(nodes.map(n => n.id));
150
+ const edges = data.edges.filter(e => nodeIds.has(e.source) && nodeIds.has(e.target));
151
+ const svg = renderGraphSVG(nodes, edges);
152
+ return {
153
+ content: [
154
+ { type: 'text', text: `Knowledge Graph: ${nodes.length} nodes, ${edges.length} edges\n\nGraph visualization (SVG):` },
155
+ { type: 'text', text: svg },
156
+ { type: 'text', text: `\n\nFull data:\n${JSON.stringify({ nodes, edges, totalNodes: svc.getStats().nodeCount, totalEdges: svc.getStats().edgeCount }, null, 2)}` },
157
+ ],
158
+ };
159
+ }
160
+ function handleGetNode(args, svc) {
161
+ let node = null;
162
+ if (args.id) {
163
+ node = svc.getNode(args.id);
164
+ }
165
+ else if (args.file) {
166
+ node = svc.getNodeByPath(args.file, args.symbol);
167
+ }
168
+ if (!node)
169
+ throw new McpError(ErrorCode.InvalidParams, 'Node not found');
170
+ const neighbors = svc.getNeighbors(node.id);
171
+ const children = svc.getChildren(node.id);
172
+ return {
173
+ content: [
174
+ {
175
+ type: 'text',
176
+ text: JSON.stringify({ node, neighbors, children }, null, 2),
177
+ },
178
+ ],
179
+ };
180
+ }
181
+ function handleSearchGraph(args, svc) {
182
+ const query = String(args.query || '').trim();
183
+ if (!query)
184
+ throw new McpError(ErrorCode.InvalidParams, 'query is required');
185
+ let results = svc.searchNodes(query);
186
+ if (args.type)
187
+ results = results.filter(n => n.type === args.type);
188
+ const enriched = results.slice(0, 30).map(n => {
189
+ const neighbors = svc.getNeighbors(n.id);
190
+ return { node: n, connectionCount: neighbors.length };
191
+ });
192
+ return {
193
+ content: [
194
+ {
195
+ type: 'text',
196
+ text: JSON.stringify({ results: enriched, total: results.length }, null, 2),
197
+ },
198
+ ],
199
+ };
200
+ }
201
+ function handleGraphStats(svc) {
202
+ const stats = svc.getStats();
203
+ return {
204
+ content: [
205
+ {
206
+ type: 'text',
207
+ text: JSON.stringify(stats, null, 2),
208
+ },
209
+ ],
210
+ };
211
+ }
212
+ function formatResults(results) {
213
+ if (results.length === 0)
214
+ return 'No results found.';
215
+ const lines = [`Found ${results.length} result(s):\n`];
216
+ for (let i = 0; i < results.length; i++) {
217
+ const r = results[i];
218
+ lines.push(`[${i + 1}] ${r.file}:${r.startLine}-${r.endLine}`);
219
+ lines.push(` Language: ${r.language} | Type: ${r.matchType} | Score: ${r.score.toFixed(3)}`);
220
+ if (r.symbol)
221
+ lines.push(` Symbol: ${r.symbol}${r.kind ? ` (${r.kind})` : ''}`);
222
+ lines.push(` ${'─'.repeat(60)}`);
223
+ }
224
+ return lines.join('\n');
225
+ }
226
+ //# sourceMappingURL=server.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EACL,sBAAsB,EACtB,qBAAqB,EACrB,0BAA0B,EAC1B,yBAAyB,EACzB,QAAQ,EACR,SAAS,GACV,MAAM,oCAAoC,CAAC;AAE5C,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,MAAM,UAAU,eAAe,CAAC,YAA0B,EAAE,QAAgB;IAC1E,IAAI,QAAQ,GAAwB,IAAI,CAAC;IAEzC,SAAS,QAAQ;QACf,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,QAAQ,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;YACxC,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,8DAA8D,CAAC,CAAC;YAC9G,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EACtC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAC/C,CAAC;IAEF,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC5D,KAAK,EAAE;YACL;gBACE,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,+FAA+F,QAAQ,wDAAwD;gBAC5K,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE;wBACtD,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0BAA0B,EAAE,OAAO,EAAE,EAAE,EAAE;wBACpF,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yBAAyB,EAAE;wBACtE,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kCAAkC,EAAE;qBACpF;oBACD,QAAQ,EAAE,CAAC,OAAO,CAAC;iBACpB;aACF;YACD;gBACE,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,yNAAyN;gBACtO,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mCAAmC,EAAE;wBAC3E,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mBAAmB,EAAE;wBAC5D,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2DAA2D,EAAE;qBACnG;iBACF;aACF;YACD;gBACE,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,6FAA6F;gBAC1G,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wCAAwC,EAAE;wBAC7E,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,qCAAqC,EAAE;wBAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8BAA8B,EAAE;qBACxE;iBACF;aACF;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,mHAAmH;gBAChI,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0DAA0D,EAAE;wBAClG,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mDAAmD,EAAE;qBAC3F;oBACD,QAAQ,EAAE,CAAC,OAAO,CAAC;iBACpB;aACF;YACD;gBACE,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,6FAA6F;gBAC1G,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;aAChD;SACF;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAY,EAAE,EAAE;QACrE,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;QACjC,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;QAE5C,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,iBAAiB,CAAC,CAAC,OAAO,YAAY,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;YAC1E,KAAK,WAAW,CAAC,CAAC,OAAO,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC1D,KAAK,UAAU,CAAC,CAAC,OAAO,aAAa,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YACxD,KAAK,cAAc,CAAC,CAAC,OAAO,iBAAiB,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChE,KAAK,iBAAiB,CAAC,CAAC,OAAO,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC5D;gBACE,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAChE,SAAS,EAAE;YACT;gBACE,GAAG,EAAE,cAAc,kBAAkB,CAAC,QAAQ,CAAC,QAAQ;gBACvD,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,wDAAwD;gBACrE,QAAQ,EAAE,kBAAkB;aAC7B;SACF;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,yBAAyB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QACpE,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC;QAC/B,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,QAAQ,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;YAChC,OAAO;gBACL,QAAQ,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;aACvF,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,qBAAqB,GAAG,EAAE,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,OAAO,GAAG,CAAC,KAAU,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;IAE1E,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAAS,EAAE,YAA0B,EAAE,QAAgB;IACjF,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9C,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;IAE7E,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC7E,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACzE,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAErF,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACxD,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,sCAAsC,CAAC,CAAC;QACtF,CAAC;IACH,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC;QAC7F,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC;iBAC7B;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7E,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,IAAS,EAAE,GAAiB;IAClD,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;IAC1E,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAErF,MAAM,GAAG,GAAG,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAEzC,OAAO;QACL,OAAO,EAAE;YACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oBAAoB,KAAK,CAAC,MAAM,WAAW,KAAK,CAAC,MAAM,sCAAsC,EAAE;YACrH,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE;YAC3B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,UAAU,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;SACnK;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,IAAS,EAAE,GAAiB;IACjD,IAAI,IAAI,GAAG,IAAI,CAAC;IAChB,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;QACZ,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC;SAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACrB,IAAI,GAAG,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC;IACD,IAAI,CAAC,IAAI;QAAE,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;IAEzE,MAAM,SAAS,GAAG,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAE1C,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;aAC7D;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAS,EAAE,GAAiB;IACrD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9C,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;IAE7E,IAAI,OAAO,GAAG,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC,IAAI,IAAI,CAAC,IAAI;QAAE,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,CAAC;IAEnE,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;QAC5C,MAAM,SAAS,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACzC,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,eAAe,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;aAC5E;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAiB;IACzC,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC7B,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;aACrC;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,OAAuB;IAC5C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,mBAAmB,CAAC;IACrD,MAAM,KAAK,GAAa,CAAC,SAAS,OAAO,CAAC,MAAM,eAAe,CAAC,CAAC;IACjE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/D,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,QAAQ,cAAc,CAAC,CAAC,SAAS,eAAe,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpG,IAAI,CAAC,CAAC,MAAM;YAAE,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACnF,KAAK,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}
@@ -0,0 +1,4 @@
1
+ export type MCPClient = 'opencode' | 'cline' | 'roo-code' | 'claude-code' | 'cursor' | 'windsurf' | 'github-copilot' | 'continue' | 'none';
2
+ export declare function writeMCPConfigs(rootPath: string, clients: MCPClient[]): string[];
3
+ export declare function promptAndWriteMCPConfigs(rootPath: string): Promise<void>;
4
+ //# sourceMappingURL=mcp-config.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mcp-config.d.ts","sourceRoot":"","sources":["../src/mcp-config.ts"],"names":[],"mappings":"AAKA,MAAM,MAAM,SAAS,GACjB,UAAU,GACV,OAAO,GACP,UAAU,GACV,aAAa,GACb,QAAQ,GACR,UAAU,GACV,gBAAgB,GAChB,UAAU,GACV,MAAM,CAAC;AA0JX,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,MAAM,EAAE,CAOhF;AAED,wBAAsB,wBAAwB,CAAC,QAAQ,EAAE,MAAM,iBAO9D"}
@@ -0,0 +1,175 @@
1
+ import { writeFileSync, readFileSync, existsSync, mkdirSync } from 'node:fs';
2
+ import { join } from 'node:path';
3
+ import { createInterface } from 'node:readline';
4
+ import chalk from 'chalk';
5
+ const CLIENT_OPTIONS = [
6
+ { key: 'opencode', label: 'OpenCode', description: 'opencode.json at project root' },
7
+ { key: 'cline', label: 'Cline', description: '.cline/mcp.json at project root' },
8
+ { key: 'roo-code', label: 'Roo Code', description: '.roo/mcp.json at project root' },
9
+ { key: 'claude-code', label: 'Claude Code', description: 'claude.jsonc at project root' },
10
+ { key: 'cursor', label: 'Cursor', description: '.cursor/mcp.json at project root' },
11
+ { key: 'windsurf', label: 'Windsurf', description: '.windsurf/mcp_config.json at project root' },
12
+ { key: 'github-copilot', label: 'GitHub Copilot', description: '.vscode/settings.json (VS Code workspace)' },
13
+ { key: 'continue', label: 'Continue', description: '.continue/config.json at project root' },
14
+ ];
15
+ function isInteractive() {
16
+ return process.stdin.isTTY && process.stdout.isTTY;
17
+ }
18
+ function askClients() {
19
+ return new Promise((resolve) => {
20
+ if (!isInteractive()) {
21
+ resolve([]);
22
+ return;
23
+ }
24
+ const rl = createInterface({ input: process.stdin, output: process.stdout });
25
+ const lines = CLIENT_OPTIONS.map((opt, i) => ` ${i + 1}. ${opt.label} — ${opt.description}`);
26
+ const prompt = [
27
+ `\n${chalk.bold('MCP Client Configuration')}`,
28
+ 'Which MCP clients do you want to configure?',
29
+ 'Enter numbers separated by commas (e.g. 1,3,5) or press Enter to skip:',
30
+ '',
31
+ ...lines,
32
+ '',
33
+ ].join('\n');
34
+ rl.question(prompt, (answer) => {
35
+ rl.close();
36
+ const trimmed = answer.trim();
37
+ if (!trimmed) {
38
+ resolve([]);
39
+ return;
40
+ }
41
+ const indices = trimmed
42
+ .split(/[, ]+/)
43
+ .map((s) => parseInt(s, 10))
44
+ .filter((n) => !isNaN(n) && n >= 1 && n <= CLIENT_OPTIONS.length);
45
+ const selected = [...new Set(indices.map((i) => CLIENT_OPTIONS[i - 1].key))];
46
+ resolve(selected);
47
+ });
48
+ });
49
+ }
50
+ function writeSingleConfig(rootPath, client) {
51
+ const mcpServer = {
52
+ mcpServers: {
53
+ edhindex: {
54
+ command: 'edhindex',
55
+ args: ['start'],
56
+ },
57
+ },
58
+ };
59
+ const mcpServerCopilot = {
60
+ edhindex: {
61
+ command: 'edhindex',
62
+ args: ['start'],
63
+ },
64
+ };
65
+ switch (client) {
66
+ case 'opencode': {
67
+ const fp = join(rootPath, 'opencode.json');
68
+ const cfg = { $schema: 'https://opencode.ai/config.json' };
69
+ if (existsSync(fp)) {
70
+ try {
71
+ Object.assign(cfg, JSON.parse(readFileSync(fp, 'utf-8')));
72
+ }
73
+ catch { /* merge */ }
74
+ }
75
+ cfg.mcp = {
76
+ edhindex: { type: 'local', command: ['edhindex', 'start'], enabled: true, timeout: 120000 },
77
+ };
78
+ writeFileSync(fp, JSON.stringify(cfg, null, 2) + '\n', 'utf-8');
79
+ return fp;
80
+ }
81
+ case 'cline': {
82
+ const dir = join(rootPath, '.cline');
83
+ if (!existsSync(dir))
84
+ mkdirSync(dir, { recursive: true });
85
+ const fp = join(dir, 'mcp.json');
86
+ writeFileSync(fp, JSON.stringify(mcpServer, null, 2) + '\n', 'utf-8');
87
+ return fp;
88
+ }
89
+ case 'roo-code': {
90
+ const dir = join(rootPath, '.roo');
91
+ if (!existsSync(dir))
92
+ mkdirSync(dir, { recursive: true });
93
+ const fp = join(dir, 'mcp.json');
94
+ writeFileSync(fp, JSON.stringify(mcpServer, null, 2) + '\n', 'utf-8');
95
+ return fp;
96
+ }
97
+ case 'claude-code': {
98
+ const fp = join(rootPath, 'claude.jsonc');
99
+ writeFileSync(fp, JSON.stringify(mcpServer, null, 2) + '\n', 'utf-8');
100
+ return fp;
101
+ }
102
+ case 'cursor': {
103
+ const dir = join(rootPath, '.cursor');
104
+ if (!existsSync(dir))
105
+ mkdirSync(dir, { recursive: true });
106
+ const fp = join(dir, 'mcp.json');
107
+ writeFileSync(fp, JSON.stringify(mcpServer, null, 2) + '\n', 'utf-8');
108
+ return fp;
109
+ }
110
+ case 'windsurf': {
111
+ const dir = join(rootPath, '.windsurf');
112
+ if (!existsSync(dir))
113
+ mkdirSync(dir, { recursive: true });
114
+ const fp = join(dir, 'mcp_config.json');
115
+ writeFileSync(fp, JSON.stringify(mcpServer, null, 2) + '\n', 'utf-8');
116
+ return fp;
117
+ }
118
+ case 'github-copilot': {
119
+ const dir = join(rootPath, '.vscode');
120
+ if (!existsSync(dir))
121
+ mkdirSync(dir, { recursive: true });
122
+ const fp = join(dir, 'settings.json');
123
+ const settings = {};
124
+ if (existsSync(fp)) {
125
+ try {
126
+ Object.assign(settings, JSON.parse(readFileSync(fp, 'utf-8')));
127
+ }
128
+ catch { /* start fresh if corrupt */ }
129
+ }
130
+ settings['github.copilot.chat.mcpServers'] = mcpServerCopilot;
131
+ writeFileSync(fp, JSON.stringify(settings, null, 2) + '\n', 'utf-8');
132
+ return fp;
133
+ }
134
+ case 'continue': {
135
+ const dir = join(rootPath, '.continue');
136
+ if (!existsSync(dir))
137
+ mkdirSync(dir, { recursive: true });
138
+ const fp = join(dir, 'config.json');
139
+ const cfg = {};
140
+ if (existsSync(fp)) {
141
+ try {
142
+ Object.assign(cfg, JSON.parse(readFileSync(fp, 'utf-8')));
143
+ }
144
+ catch { /* start fresh if corrupt */ }
145
+ }
146
+ const existing = cfg.mcpServers || [];
147
+ if (!existing.find((s) => s.name === 'edhindex')) {
148
+ cfg.mcpServers = [...existing, { name: 'edhindex', command: 'edhindex', args: ['start'] }];
149
+ }
150
+ writeFileSync(fp, JSON.stringify(cfg, null, 2) + '\n', 'utf-8');
151
+ return fp;
152
+ }
153
+ default:
154
+ throw new Error(`Unknown MCP client: ${client}`);
155
+ }
156
+ }
157
+ export function writeMCPConfigs(rootPath, clients) {
158
+ const paths = [];
159
+ for (const client of clients) {
160
+ if (client === 'none')
161
+ continue;
162
+ paths.push(writeSingleConfig(rootPath, client));
163
+ }
164
+ return paths;
165
+ }
166
+ export async function promptAndWriteMCPConfigs(rootPath) {
167
+ const clients = await askClients();
168
+ if (clients.length === 0)
169
+ return;
170
+ const paths = writeMCPConfigs(rootPath, clients);
171
+ for (const fp of paths) {
172
+ console.log(chalk.green(`MCP config written: ${fp}`));
173
+ }
174
+ }
175
+ //# sourceMappingURL=mcp-config.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mcp-config.js","sourceRoot":"","sources":["../src/mcp-config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,KAAK,MAAM,OAAO,CAAC;AAmB1B,MAAM,cAAc,GAAmB;IACrC,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,+BAA+B,EAAE;IACpF,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,iCAAiC,EAAE;IAChF,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,+BAA+B,EAAE;IACpF,EAAE,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,8BAA8B,EAAE;IACzF,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,kCAAkC,EAAE;IACnF,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,2CAA2C,EAAE;IAChG,EAAE,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,2CAA2C,EAAE;IAC5G,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,uCAAuC,EAAE;CAC7F,CAAC;AAEF,SAAS,aAAa;IACpB,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;AACrD,CAAC;AAED,SAAS,UAAU;IACjB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,EAAE,CAAC,CAAC;YACZ,OAAO;QACT,CAAC;QACD,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7E,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,KAAK,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;QAC9F,MAAM,MAAM,GAAG;YACb,KAAK,KAAK,CAAC,IAAI,CAAC,0BAA0B,CAAC,EAAE;YAC7C,6CAA6C;YAC7C,wEAAwE;YACxE,EAAE;YACF,GAAG,KAAK;YACR,EAAE;SACH,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,EAAE;YAC7B,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,CAAC,EAAE,CAAC,CAAC;gBACZ,OAAO;YACT,CAAC;YACD,MAAM,OAAO,GAAG,OAAO;iBACpB,KAAK,CAAC,OAAO,CAAC;iBACd,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;iBAC3B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;YACpE,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC7E,OAAO,CAAC,QAAQ,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB,CAAC,QAAgB,EAAE,MAAiB;IAC5D,MAAM,SAAS,GAAG;QAChB,UAAU,EAAE;YACV,QAAQ,EAAE;gBACR,OAAO,EAAE,UAAU;gBACnB,IAAI,EAAE,CAAC,OAAO,CAAC;aAChB;SACF;KACF,CAAC;IACF,MAAM,gBAAgB,GAAG;QACvB,QAAQ,EAAE;YACR,OAAO,EAAE,UAAU;YACnB,IAAI,EAAE,CAAC,OAAO,CAAC;SAChB;KACF,CAAC;IAEF,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;YAC3C,MAAM,GAAG,GAA4B,EAAE,OAAO,EAAE,iCAAiC,EAAE,CAAC;YACpF,IAAI,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnB,IAAI,CAAC;oBAAC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;gBAAC,CAAC;gBAAC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC;YAC1F,CAAC;YACD,GAAG,CAAC,GAAG,GAAG;gBACR,QAAQ,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE;aAC5F,CAAC;YACF,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YAChE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YACjC,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YACtE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YACnC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YACjC,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YACtE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;YAC1C,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YACtE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YACtC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YACjC,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YACtE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;YACxC,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YACtE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,gBAAgB,CAAC,CAAC,CAAC;YACtB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YACtC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;YACtC,MAAM,QAAQ,GAA4B,EAAE,CAAC;YAC7C,IAAI,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;gBACjE,CAAC;gBAAC,MAAM,CAAC,CAAC,4BAA4B,CAAC,CAAC;YAC1C,CAAC;YACD,QAAQ,CAAC,gCAAgC,CAAC,GAAG,gBAAgB,CAAC;YAC9D,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YACrE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;YACpC,MAAM,GAAG,GAA4B,EAAE,CAAC;YACxC,IAAI,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC5D,CAAC;gBAAC,MAAM,CAAC,CAAC,4BAA4B,CAAC,CAAC;YAC1C,CAAC;YACD,MAAM,QAAQ,GAAI,GAAG,CAAC,UAAwC,IAAI,EAAE,CAAC;YACrE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAA0B,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,EAAE,CAAC;gBAC1E,GAAG,CAAC,UAAU,GAAG,CAAC,GAAG,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7F,CAAC;YACD,aAAa,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;YAChE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD;YACE,MAAM,IAAI,KAAK,CAAC,uBAAuB,MAAM,EAAE,CAAC,CAAC;IACrD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,OAAoB;IACpE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,MAAM,KAAK,MAAM;YAAE,SAAS;QAChC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,QAAgB;IAC7D,MAAM,OAAO,GAAG,MAAM,UAAU,EAAE,CAAC;IACnC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IACjC,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACjD,KAAK,MAAM,EAAE,IAAI,KAAK,EAAE,CAAC;QACvB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC,CAAC;IACxD,CAAC;AACH,CAAC"}
@@ -0,0 +1,12 @@
1
+ export type ProgressPhase = 'scanning' | 'embedding' | 'building_fts' | 'ready';
2
+ export interface ProgressEvent {
3
+ phase: ProgressPhase;
4
+ current?: number;
5
+ total?: number;
6
+ percent?: number;
7
+ message?: string;
8
+ }
9
+ export type ProgressCallback = (event: ProgressEvent) => void;
10
+ export declare function onProgress(cb: ProgressCallback): () => void;
11
+ export declare function emitProgress(event: ProgressEvent): void;
12
+ //# sourceMappingURL=progress.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"progress.d.ts","sourceRoot":"","sources":["../src/progress.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,aAAa,GAAG,UAAU,GAAG,WAAW,GAAG,cAAc,GAAG,OAAO,CAAC;AAEhF,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,aAAa,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,MAAM,gBAAgB,GAAG,CAAC,KAAK,EAAE,aAAa,KAAK,IAAI,CAAC;AAI9D,wBAAgB,UAAU,CAAC,EAAE,EAAE,gBAAgB,GAAG,MAAM,IAAI,CAG3D;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,aAAa,QAQhD"}
@@ -0,0 +1,16 @@
1
+ const listeners = new Set();
2
+ export function onProgress(cb) {
3
+ listeners.add(cb);
4
+ return () => listeners.delete(cb);
5
+ }
6
+ export function emitProgress(event) {
7
+ for (const cb of listeners) {
8
+ try {
9
+ cb(event);
10
+ }
11
+ catch {
12
+ // swallow
13
+ }
14
+ }
15
+ }
16
+ //# sourceMappingURL=progress.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"progress.js","sourceRoot":"","sources":["../src/progress.ts"],"names":[],"mappings":"AAYA,MAAM,SAAS,GAA0B,IAAI,GAAG,EAAE,CAAC;AAEnD,MAAM,UAAU,UAAU,CAAC,EAAoB;IAC7C,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClB,OAAO,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAoB;IAC/C,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,EAAE,CAAC,KAAK,CAAC,CAAC;QACZ,CAAC;QAAC,MAAM,CAAC;YACP,UAAU;QACZ,CAAC;IACH,CAAC;AACH,CAAC"}
@@ -0,0 +1,6 @@
1
+ import { FTSStore } from '../storage/fts.js';
2
+ export declare function searchBM25(fts: FTSStore, query: string, topK: number): Array<{
3
+ id: string;
4
+ score: number;
5
+ }>;
6
+ //# sourceMappingURL=bm25.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"bm25.d.ts","sourceRoot":"","sources":["../../src/retrieval/bm25.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAa,MAAM,mBAAmB,CAAC;AAExD,wBAAgB,UAAU,CAAC,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,KAAK,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAmB3G"}
@@ -0,0 +1,20 @@
1
+ export function searchBM25(fts, query, topK) {
2
+ const results = fts.searchWithRank(query, topK * 2);
3
+ if (results.length === 0)
4
+ return [];
5
+ if (results.length === 1) {
6
+ return [{ id: results[0].id, score: 1.0 }];
7
+ }
8
+ const negRanks = results.map(r => -r.rank);
9
+ const minNeg = Math.min(...negRanks);
10
+ const maxNeg = Math.max(...negRanks);
11
+ const range = maxNeg - minNeg;
12
+ if (range === 0) {
13
+ return results.map(r => ({ id: r.id, score: 1.0 }));
14
+ }
15
+ return results.map((r, i) => ({
16
+ id: r.id,
17
+ score: (negRanks[i] - minNeg) / range,
18
+ }));
19
+ }
20
+ //# sourceMappingURL=bm25.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"bm25.js","sourceRoot":"","sources":["../../src/retrieval/bm25.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,UAAU,CAAC,GAAa,EAAE,KAAa,EAAE,IAAY;IACnE,MAAM,OAAO,GAAG,GAAG,CAAC,cAAc,CAAC,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;IACpD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEpC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IACrC,MAAM,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;IAC9B,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QAChB,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5B,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,KAAK,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,GAAG,KAAK;KACtC,CAAC,CAAC,CAAC;AACN,CAAC"}
@@ -0,0 +1,11 @@
1
+ import { TransformersReranker } from '../embeddings/transformers.js';
2
+ import { MetadataStore, ChunkMetadata } from '../storage/metadata.js';
3
+ export interface RerankedResult {
4
+ id: string;
5
+ score: number;
6
+ metadata: ChunkMetadata | null;
7
+ }
8
+ export declare function rerankResults(query: string, candidates: Array<{
9
+ id: string;
10
+ }>, metadataStore: MetadataStore, reranker: TransformersReranker | null, topK: number, useRerank: boolean): Promise<RerankedResult[]>;
11
+ //# sourceMappingURL=reranker.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"reranker.d.ts","sourceRoot":"","sources":["../../src/retrieval/reranker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAGtE,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,aAAa,GAAG,IAAI,CAAC;CAChC;AAED,wBAAsB,aAAa,CACjC,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,KAAK,CAAC;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,CAAC,EACjC,aAAa,EAAE,aAAa,EAC5B,QAAQ,EAAE,oBAAoB,GAAG,IAAI,EACrC,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,OAAO,GACjB,OAAO,CAAC,cAAc,EAAE,CAAC,CA8C3B"}
@@ -0,0 +1,46 @@
1
+ import { logger } from '../logging.js';
2
+ export async function rerankResults(query, candidates, metadataStore, reranker, topK, useRerank) {
3
+ if (candidates.length === 0)
4
+ return [];
5
+ if (!useRerank || !reranker) {
6
+ return candidates.slice(0, topK).map(c => ({
7
+ id: c.id,
8
+ score: 1,
9
+ metadata: metadataStore.getChunk(c.id) || null,
10
+ }));
11
+ }
12
+ const texts = [];
13
+ const validCandidates = [];
14
+ for (const c of candidates) {
15
+ const meta = metadataStore.getChunk(c.id);
16
+ if (meta) {
17
+ texts.push(meta.imports + ' ' + meta.exports + ' ' + (meta.symbol || ''));
18
+ validCandidates.push(c);
19
+ }
20
+ }
21
+ if (texts.length === 0) {
22
+ return candidates.slice(0, topK).map(c => ({
23
+ id: c.id,
24
+ score: 1,
25
+ metadata: metadataStore.getChunk(c.id) || null,
26
+ }));
27
+ }
28
+ try {
29
+ const scores = await reranker.rerank(query, texts);
30
+ const results = scores.slice(0, topK).map(s => ({
31
+ id: validCandidates[s.index].id,
32
+ score: s.score,
33
+ metadata: metadataStore.getChunk(validCandidates[s.index].id) || null,
34
+ }));
35
+ return results;
36
+ }
37
+ catch (e) {
38
+ logger.debug('Reranking failed, falling back to unranked:', e);
39
+ return validCandidates.slice(0, topK).map(c => ({
40
+ id: c.id,
41
+ score: 1,
42
+ metadata: metadataStore.getChunk(c.id) || null,
43
+ }));
44
+ }
45
+ }
46
+ //# sourceMappingURL=reranker.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"reranker.js","sourceRoot":"","sources":["../../src/retrieval/reranker.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAQvC,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,KAAa,EACb,UAAiC,EACjC,aAA4B,EAC5B,QAAqC,EACrC,IAAY,EACZ,SAAkB;IAElB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvC,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC5B,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACzC,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,KAAK,EAAE,CAAC;YACR,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,IAAI;SAC/C,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,eAAe,GAA0B,EAAE,CAAC;IAElD,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC;YAC1E,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACzC,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,KAAK,EAAE,CAAC;YACR,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,IAAI;SAC/C,CAAC,CAAC,CAAC;IACN,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACnD,MAAM,OAAO,GAAqB,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAChE,EAAE,EAAE,eAAe,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE;YAC/B,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,IAAI,IAAI;SACtE,CAAC,CAAC,CAAC;QACJ,OAAO,OAAO,CAAC;IACjB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,6CAA6C,EAAE,CAAC,CAAC,CAAC;QAC/D,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC9C,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,KAAK,EAAE,CAAC;YACR,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,IAAI;SAC/C,CAAC,CAAC,CAAC;IACN,CAAC;AACH,CAAC"}
@@ -0,0 +1,35 @@
1
+ import { FTSStore } from '../storage/fts.js';
2
+ import { MetadataStore } from '../storage/metadata.js';
3
+ import { VectorStore } from '../vector/store.js';
4
+ import { EmbeddingProvider } from '../embeddings/provider.js';
5
+ import { TransformersReranker } from '../embeddings/transformers.js';
6
+ export interface SearchOptions {
7
+ query: string;
8
+ maxResults?: number;
9
+ rerank?: boolean;
10
+ fileFilter?: string;
11
+ languageFilter?: string;
12
+ }
13
+ export interface SearchResult {
14
+ id: string;
15
+ file: string;
16
+ language: string;
17
+ symbol: string;
18
+ kind: string | null;
19
+ startLine: number;
20
+ endLine: number;
21
+ content: string;
22
+ score: number;
23
+ matchType: 'keyword' | 'vector' | 'hybrid';
24
+ }
25
+ export declare class SearchEngine {
26
+ private fts;
27
+ private metadata;
28
+ private vectors;
29
+ private embedder;
30
+ private reranker;
31
+ private useRerank;
32
+ constructor(fts: FTSStore, metadata: MetadataStore, vectors: VectorStore, embedder: EmbeddingProvider, reranker: TransformersReranker | null, useRerank: boolean);
33
+ search(opts: SearchOptions): Promise<SearchResult[]>;
34
+ }
35
+ //# sourceMappingURL=search.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../src/retrieval/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAiB,MAAM,wBAAwB,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AAKrE,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,SAAS,GAAG,QAAQ,GAAG,QAAQ,CAAC;CAC5C;AAKD,qBAAa,YAAY;IACvB,OAAO,CAAC,GAAG,CAAW;IACtB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,QAAQ,CAA8B;IAC9C,OAAO,CAAC,SAAS,CAAU;gBAGzB,GAAG,EAAE,QAAQ,EACb,QAAQ,EAAE,aAAa,EACvB,OAAO,EAAE,WAAW,EACpB,QAAQ,EAAE,iBAAiB,EAC3B,QAAQ,EAAE,oBAAoB,GAAG,IAAI,EACrC,SAAS,EAAE,OAAO;IAUd,MAAM,CAAC,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;CAsF3D"}