opencode-codebase-index 0.10.0 → 0.11.0

package/dist/cli.js

@@ -8,7 +8,11 @@ var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __commonJS = (cb, mod) => function __require() {
-  return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
+  try {
+    return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
+  } catch (e) {
+    throw mod = 0, e;
+  }
 };
 var __copyProps = (to, from, except, desc) => {
   if (from && typeof from === "object" || typeof from === "function") {
@@ -2499,6 +2503,8 @@ var BaseEmbeddingProvider = class {
     this.credentials = credentials;
     this.modelInfo = modelInfo;
   }
+  credentials;
+  modelInfo;
   async embedQuery(query) {
     const result = await this.embedBatch([query]);
     return {
@@ -4238,17 +4244,17 @@ var Database = class {
     if (edges.length === 0) return;
     this.inner.upsertCallEdgesBatch(edges);
   }
-  getCallers(targetName, branch) {
+  getCallers(targetName, branch, callTypeFilter) {
     this.throwIfClosed();
-    return this.inner.getCallers(targetName, branch);
+    return this.inner.getCallers(targetName, branch, callTypeFilter ?? null);
   }
-  getCallersWithContext(targetName, branch) {
+  getCallersWithContext(targetName, branch, callTypeFilter) {
     this.throwIfClosed();
-    return this.inner.getCallersWithContext(targetName, branch);
+    return this.inner.getCallersWithContext(targetName, branch, callTypeFilter ?? null);
   }
-  getCallees(symbolId, branch) {
+  getCallees(symbolId, branch, callTypeFilter) {
     this.throwIfClosed();
-    return this.inner.getCallees(symbolId, branch);
+    return this.inner.getCallees(symbolId, branch, callTypeFilter ?? null);
   }
   deleteCallEdgesByFile(filePath) {
     this.throwIfClosed();
@@ -4258,6 +4264,10 @@ var Database = class {
     this.throwIfClosed();
     this.inner.resolveCallEdge(edgeId, toSymbolId);
   }
+  findShortestPath(fromName, toName, branch, maxDepth) {
+    this.throwIfClosed();
+    return this.inner.findShortestPath(fromName, toName, branch, maxDepth ?? null);
+  }
   addSymbolsToBranch(branch, symbolIds) {
     this.throwIfClosed();
     this.inner.addSymbolsToBranch(branch, symbolIds);
@@ -7056,6 +7066,7 @@ var Indexer = class {
           targetName: site.calleeName,
           toSymbolId: void 0,
           callType: site.callType,
+          confidence: site.confidence,
           line: site.line,
           col: site.column,
           isResolved: false
@@ -8175,12 +8186,12 @@ var Indexer = class {
       })
     );
   }
-  async getCallers(targetName) {
+  async getCallers(targetName, callTypeFilter) {
     const { database } = await this.ensureInitialized();
     const seen = /* @__PURE__ */ new Set();
     const results = [];
     for (const branchKey of this.getBranchCatalogKeys()) {
-      for (const edge of database.getCallersWithContext(targetName, branchKey)) {
+      for (const edge of database.getCallersWithContext(targetName, branchKey, callTypeFilter)) {
         if (!seen.has(edge.id)) {
           seen.add(edge.id);
           results.push(edge);
@@ -8189,12 +8200,12 @@ var Indexer = class {
     }
     return results;
   }
-  async getCallees(symbolId) {
+  async getCallees(symbolId, callTypeFilter) {
     const { database } = await this.ensureInitialized();
     const seen = /* @__PURE__ */ new Set();
     const results = [];
     for (const branchKey of this.getBranchCatalogKeys()) {
-      for (const edge of database.getCallees(symbolId, branchKey)) {
+      for (const edge of database.getCallees(symbolId, branchKey, callTypeFilter)) {
         if (!seen.has(edge.id)) {
           seen.add(edge.id);
           results.push(edge);
@@ -8203,6 +8214,17 @@ var Indexer = class {
     }
     return results;
   }
+  async findCallPath(fromName, toName, maxDepth) {
+    const { database } = await this.ensureInitialized();
+    let shortest = [];
+    for (const branchKey of this.getBranchCatalogKeys()) {
+      const path18 = database.findShortestPath(fromName, toName, branchKey, maxDepth);
+      if (path18.length > 0 && (shortest.length === 0 || path18.length < shortest.length)) {
+        shortest = path18;
+      }
+    }
+    return shortest;
+  }
   async close() {
     await this.database?.close();
     this.database = null;
@@ -9978,11 +10000,12 @@ ${formatted.join("\n\n")}` }] };
   );
   server.tool(
     "call_graph",
-    "Query the call graph to find callers or callees of a function/method. Use to understand code flow and dependencies.",
+    "Query the call graph to find callers or callees of a function/method. Use to understand code flow and dependencies. Supports relationship types: Call, MethodCall, Constructor, Import, Inherits, Implements.",
     {
       name: z2.string().describe("Function or method name to query"),
       direction: z2.enum(["callers", "callees"]).default("callers").describe("Direction: 'callers' finds who calls this function, 'callees' finds what this function calls"),
-      symbolId: z2.string().optional().describe("Symbol ID (required for 'callees' direction)")
+      symbolId: z2.string().optional().describe("Symbol ID (required for 'callees' direction)"),
+      relationshipType: z2.enum(["Call", "MethodCall", "Constructor", "Import", "Inherits", "Implements"]).optional().describe("Filter by relationship type. Omit to show all.")
     },
     async (args) => {
       await runtime.ensureInitialized();
@@ -9991,26 +10014,52 @@ ${formatted.join("\n\n")}` }] };
         if (!args.symbolId) {
           return { content: [{ type: "text", text: "Error: 'symbolId' is required when direction is 'callees'." }] };
         }
-        const callees = await indexer.getCallees(args.symbolId);
+        const callees = await indexer.getCallees(args.symbolId, args.relationshipType);
         if (callees.length === 0) {
-          return { content: [{ type: "text", text: `No callees found for symbol ${args.symbolId}.` }] };
+          return { content: [{ type: "text", text: `No callees found for symbol ${args.symbolId}${args.relationshipType ? ` with type ${args.relationshipType}` : ""}.` }] };
         }
-        const formatted2 = callees.map(
-          (e, i) => `[${i + 1}] \u2192 ${e.targetName} (${e.callType}) at line ${e.line}${e.isResolved ? ` [resolved: ${e.toSymbolId}]` : " [unresolved]"}`
-        );
+        const formatted2 = callees.map((e, i) => {
+          const conf = e.confidence !== "Direct" ? ` [${e.confidence.toLowerCase()}]` : "";
+          return `[${i + 1}] \u2192 ${e.targetName} (${e.callType})${conf} at line ${e.line}${e.isResolved ? ` [resolved: ${e.toSymbolId}]` : " [unresolved]"}`;
+        });
         return { content: [{ type: "text", text: `Callees (${callees.length}):
 
 ${formatted2.join("\n")}` }] };
       }
-      const callers = await indexer.getCallers(args.name);
+      const callers = await indexer.getCallers(args.name, args.relationshipType);
       if (callers.length === 0) {
-        return { content: [{ type: "text", text: `No callers found for "${args.name}".` }] };
+        return { content: [{ type: "text", text: `No callers found for "${args.name}"${args.relationshipType ? ` with type ${args.relationshipType}` : ""}.` }] };
       }
-      const formatted = callers.map(
-        (e, i) => `[${i + 1}] \u2190 from ${e.fromSymbolName ?? "<unknown>"} in ${e.fromSymbolFilePath ?? "<unknown file>"} [${e.fromSymbolId}] (${e.callType}) at line ${e.line}${e.isResolved ? " [resolved]" : " [unresolved]"}`
-      );
+      const formatted = callers.map((e, i) => {
+        const conf = e.confidence !== "Direct" ? ` [${e.confidence.toLowerCase()}]` : "";
+        return `[${i + 1}] \u2190 from ${e.fromSymbolName ?? "<unknown>"} in ${e.fromSymbolFilePath ?? "<unknown file>"} [${e.fromSymbolId}] (${e.callType})${conf} at line ${e.line}${e.isResolved ? " [resolved]" : " [unresolved]"}`;
+      });
       return { content: [{ type: "text", text: `"${args.name}" is called by ${callers.length} function(s):
 
+${formatted.join("\n")}` }] };
+    }
+  );
+  server.tool(
+    "call_graph_path",
+    "Find the shortest connection path between two symbols in the call graph. Returns the chain of calls connecting them.",
+    {
+      from: z2.string().describe("Source function/method name (starting point)"),
+      to: z2.string().describe("Target function/method name (destination)"),
+      maxDepth: z2.number().optional().default(10).describe("Maximum traversal depth (default: 10)")
+    },
+    async (args) => {
+      await runtime.ensureInitialized();
+      const indexer = runtime.getIndexer();
+      const path18 = await indexer.findCallPath(args.from, args.to, args.maxDepth);
+      if (path18.length === 0) {
+        return { content: [{ type: "text", text: `No path found between "${args.from}" and "${args.to}". They may be in disconnected components, or the call graph index needs updating.` }] };
+      }
+      const formatted = path18.map((hop, i) => {
+        const prefix = i === 0 ? "[start]" : `--${hop.callType}-->`;
+        const location = hop.filePath ? ` (${hop.filePath}:${hop.line})` : "";
+        return `${prefix} ${hop.symbolName}${location}`;
+      });
+      return { content: [{ type: "text", text: `Path (${path18.length} hops):
 ${formatted.join("\n")}` }] };
     }
   );