codebasesearch 0.1.19 → 0.1.21

package/mcp.js

@@ -134,17 +134,19 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       };
     }
 
+    const repoPath = typeof repositoryPath === 'string' ? repositoryPath : null;
     const text =
       result.resultsCount === 0
         ? `No results found for query: "${query}"`
         : `Found ${result.resultsCount} result${result.resultsCount !== 1 ? 's' : ''} for query: "${query}"\n\n${result.results
-            .map(
-              (r) =>
-                `${r.rank}. ${r.absolutePath}:${r.lines} (score: ${r.score}%)\n${r.snippet
-                  .split('\n')
-                  .map((line) => `   ${line}`)
-                  .join('\n')}`
-            )
+            .map((r) => {
+              const pathPart = r.relativePath || r.absolutePath;
+              const lineCount = r.totalLines ? ` [${r.totalLines}L]` : '';
+              const ctx = r.enclosingContext ? ` (in: ${r.enclosingContext})` : '';
+              const header = `${r.rank}. ${pathPart}${lineCount}:${r.lines}${ctx} (score: ${r.score}%)`;
+              const body = r.snippet.split('\n').map((line) => `   ${line}`).join('\n');
+              return `${header}\n${body}`;
+            })
             .join('\n\n')}`;
 
     return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codebasesearch",
-  "version": "0.1.19",
+  "version": "0.1.21",
   "description": "Ultra-simple code search tool with Jina embeddings, LanceDB, and MCP protocol support",
   "type": "module",
   "bin": {

package/src/search-worker.js

@@ -1,10 +1,34 @@
 import { parentPort } from 'worker_threads';
 import { resolve, relative } from 'path';
-import { existsSync } from 'fs';
+import { existsSync, readFileSync } from 'fs';
 import { loadIgnorePatterns } from './ignore-parser.js';
 import { scanRepository } from './scanner.js';
 import { buildTextIndex, searchText } from './text-search.js';
 
+function findEnclosingContext(content, lineStart) {
+  const lines = content.split('\n');
+  const targetLine = Math.min(lineStart - 1, lines.length - 1);
+  const skip = new Set(['if', 'for', 'while', 'switch', 'catch', 'else']);
+  for (let i = targetLine; i >= 0; i--) {
+    const line = lines[i];
+    const m = line.match(/(?:^|\s)(?:async\s+)?(?:function\s+(\w+)|class\s+(\w+)|(?:const|let|var)\s+(\w+)\s*=\s*(?:async\s*)?\(|(?:static\s+)?(?:async\s+)?(\w+)\s*\([^)]*\)\s*\{)/);
+    if (m) {
+      const name = m[1] || m[2] || m[3] || m[4];
+      if (name && !skip.has(name)) return name;
+    }
+  }
+  return null;
+}
+
+function getFileTotalLines(absoluteFilePath) {
+  try {
+    const content = readFileSync(absoluteFilePath, 'utf8');
+    return content.split('\n').length;
+  } catch {
+    return null;
+  }
+}
+
 let indexCache = new Map();
 
 async function initializeIndex(repositoryPath) {
@@ -55,12 +79,18 @@ async function performSearch(repositoryPath, query) {
       resultsCount: results.length,
       results: results.slice(0, 10).map((result, idx) => {
         const absoluteFilePath = resolve(absolutePath, result.file_path);
+        const totalLines = getFileTotalLines(absoluteFilePath);
+        const enclosingContext = findEnclosingContext(result.content, result.line_start);
+        const relPath = relative(absolutePath, absoluteFilePath);
         return {
           rank: idx + 1,
           absolutePath: absoluteFilePath,
+          relativePath: relPath,
           lines: `${result.line_start}-${result.line_end}`,
+          totalLines,
+          enclosingContext,
           score: (result.score * 100).toFixed(1),
-          snippet: result.content.split('\n').slice(0, 3).join('\n'),
+          snippet: result.content.split('\n').slice(0, 30).join('\n'),
         };
       }),
     };