Merge pull request #11 from Helweg/feature/call-graph

feat: add call graph extraction and query support

Author: Helweg

AGENTS.md

@@ -4,7 +4,7 @@
 
 Semantic codebase indexing plugin for OpenCode. Hybrid TypeScript/Rust architecture:
 - **TypeScript** (`src/`): Plugin logic, embedding providers, OpenCode tools
-- **Rust** (`native/`): Tree-sitter parsing, usearch vectors, SQLite storage, BM25 inverted index
+- **Rust** (`native/`): Tree-sitter parsing, usearch vectors, SQLite storage, BM25 inverted index, call graph extraction
 
 ## Build/Test/Lint
 
@@ -52,10 +52,11 @@ native/src/
 ├── parser.rs             # Tree-sitter parsing (14 languages: TS, JS, Python, Rust, Go, Java, C#, Ruby, Bash, C, C++, JSON, TOML, YAML)
 ├── chunker.rs            # Semantic chunking with overlap
 ├── store.rs              # usearch vector store (F16 quantization)
-├── db.rs                 # SQLite: embeddings, chunks, branch catalog
+├── db.rs                 # SQLite: embeddings, chunks, branch catalog, symbols, call edges
+├── call_extractor.rs     # Tree-sitter query-based call extraction (TS/JS, Python, Go, Rust)
 ├── inverted_index.rs     # BM25 keyword search
 ├── hasher.rs             # xxhash content hashing
-└── types.rs              # Shared types
+└── types.rs              # Shared types (Language enum with from_string)
 
 tests/                    # Vitest tests (30s timeout for native ops)
 commands/                 # Slash command definitions (/search, /find, /index)
@@ -75,6 +76,8 @@ skill/                    # OpenCode skill guidance
 | Add slash command | `commands/` + register in `src/index.ts` config() |
 
 | Add/modify MCP tool | `src/mcp-server.ts` (createMcpServer) |
+| Modify call graph extraction | `native/src/call_extractor.rs` + query files in `native/queries/` |
+| Add call graph language | `native/queries/<lang>-calls.scm` + update `call_extractor.rs` |
 ## CODE MAP
 
 ### TypeScript Exports (`src/index.ts`)
@@ -89,12 +92,13 @@ skill/                    # OpenCode skill guidance
 | `index_health_check` | Tool | GC orphaned embeddings/chunks |
 | `index_metrics` | Tool | Get performance metrics (requires debug.enabled + debug.metrics) |
 | `index_logs` | Tool | Get debug logs (requires debug.enabled) |
+| `call_graph` | Tool | Query call graph for callers/callees of functions |
 
 
 ### MCP Server Exports (`src/mcp-server.ts`)
 | Symbol | Type | Purpose |
 |--------|------|---------|
-| `createMcpServer` | fn | Creates MCP Server with 8 tools + 4 prompts, lazy Indexer init |
+| `createMcpServer` | fn | Creates MCP Server with 9 tools + 4 prompts, lazy Indexer init |
 
 ### CLI Entry (`src/cli.ts`)
 | Symbol | Type | Purpose |
@@ -218,6 +222,7 @@ afterEach(() => { fs.rmSync(tempDir, { recursive: true, force: true }); });
 | `logger.test.ts` | Logger utility, metrics collection |
 
 | `mcp-server.test.ts` | MCP server: tool/prompt registration, execution via InMemoryTransport |
+| `call-graph.test.ts` | Call extraction, storage, resolution, branch awareness, integration |
 ### Benchmarks
 ```bash
 npx tsx benchmarks/run.ts   # Performance testing for native operations

CHANGELOG.md

@@ -20,6 +20,15 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+- **Call graph extraction and query**: Tree-sitter query-based extraction of function calls, method calls, constructors, and imports across 5 languages (TypeScript/JavaScript, Python, Go, Rust)
+- **`call_graph` tool**: Query callers or callees of any function/method with branch-aware filtering
+- **DB schema v2**: `symbols`, `call_edges`, and `branch_symbols` tables with full CRUD, GC, and batch operations
+- **Same-file call resolution**: Automatically resolves call edges to symbols defined in the same file during indexing
+
+### Fixed
+- **Missing `call_graph` export**: The `call_graph` tool was not exported from the plugin entry point — now available to OpenCode users
+
 ## [0.5.0] - 2026-02-23
 
 ### Added

README.md

@@ -82,7 +82,7 @@ Use the same semantic search from any MCP-compatible client. Index once, search
    npx opencode-codebase-index-mcp                            # uses current directory
    ```
 
-The MCP server exposes all 8 tools (`codebase_search`, `codebase_peek`, `find_similar`, `index_codebase`, `index_status`, `index_health_check`, `index_metrics`, `index_logs`) and 4 prompts (`search`, `find`, `index`, `status`).
+The MCP server exposes all 9 tools (`codebase_search`, `codebase_peek`, `find_similar`, `call_graph`, `index_codebase`, `index_status`, `index_health_check`, `index_metrics`, `index_logs`) and 4 prompts (`search`, `find`, `index`, `status`).
 
 The MCP dependencies (`@modelcontextprotocol/sdk`, `zod`) are optional peer dependencies — they're only needed if you use the MCP server.
 
@@ -112,6 +112,7 @@ src/api/checkout.ts:89      (Route handler for /pay)
 | Don't know the function name | `codebase_search` | Semantic search finds by meaning |
 | Exploring unfamiliar codebase | `codebase_search` | Discovers related code across files |
 | Just need to find locations | `codebase_peek` | Returns metadata only, saves ~90% tokens |
+| Understand code flow | `call_graph` | Find callers/callees of any function |
 | Know exact identifier | `grep` | Faster, finds all occurrences |
 | Need ALL matches | `grep` | Semantic returns top N only |
 | Mixed discovery + precision | `/find` (hybrid) | Best of both worlds |
@@ -211,7 +212,7 @@ When you switch branches, code changes but embeddings for unchanged content rema
 
 ```
 .opencode/index/
-├── codebase.db           # SQLite: embeddings, chunks, branch catalog
+├── codebase.db           # SQLite: embeddings, chunks, branch catalog, symbols, call edges
 ├── vectors.usearch       # Vector index (uSearch)
 ├── inverted-index.json   # BM25 keyword index
 └── file-hashes.json      # File change detection
@@ -267,6 +268,12 @@ Returns collected metrics about indexing and search performance. Requires `debug
 Returns recent debug logs with optional filtering.
 - **Parameters**: `category` (optional: `search`, `embedding`, `cache`, `gc`, `branch`), `level` (optional: `error`, `warn`, `info`, `debug`), `limit` (default: 50).
 
+### `call_graph`
+Query the call graph to find callers or callees of a function/method. Automatically built during indexing for TypeScript, JavaScript, Python, Go, and Rust.
+- **Use for**: Understanding code flow, tracing dependencies, impact analysis.
+- **Parameters**: `name` (function name), `direction` (`callers` or `callees`), `symbolId` (required for `callees`, returned by previous queries).
+- **Example**: Find who calls `validateToken` → `call_graph(name="validateToken", direction="callers")`
+
 ## 🎮 Slash Commands
 
 The plugin automatically registers these slash commands:
@@ -585,8 +592,9 @@ CI will automatically run tests and type checking on your PR.
 The Rust native module handles performance-critical operations:
 - **tree-sitter**: Language-aware code parsing with JSDoc/docstring extraction
 - **usearch**: High-performance vector similarity search with F16 quantization
-- **SQLite**: Persistent storage for embeddings, chunks, and branch catalog
+- **SQLite**: Persistent storage for embeddings, chunks, branch catalog, symbols, and call edges
 - **BM25 inverted index**: Fast keyword search for hybrid retrieval
+- **Call graph extraction**: Tree-sitter query-based extraction of function calls, method calls, constructors, and imports (TypeScript/JavaScript, Python, Go, Rust)
 - **xxhash**: Fast content hashing for change detection
 
 Rebuild with: `npm run build:native` (requires Rust toolchain)

commands/call-graph.md

@@ -0,0 +1,24 @@
+---
+description: Trace callers or callees using the call graph
+---
+
+Trace function dependencies using the `call_graph` tool.
+
+User input: $ARGUMENTS
+
+Interpret input as follows:
+- Default to `direction="callers"` unless input asks for callees/calls/makes calls.
+- `name=<function>` or plain text function name sets `name`.
+- `symbolId=<id>` is required for `direction="callees"`.
+
+Execution flow:
+1. If direction is `callers`, call `call_graph` with `{ name, direction: "callers" }`.
+2. If direction is `callees` and `symbolId` is present, call `call_graph` with `{ name, direction: "callees", symbolId }`.
+3. If direction is `callees` and `symbolId` is missing, first call `call_graph` with `direction="callers"` to get symbol IDs, then ask the user to choose one if multiple are returned.
+
+Examples:
+- `/call-graph Database` → callers for `Database`
+- `/call-graph callers name=Indexer` → callers for `Indexer`
+- `/call-graph callees name=Database symbolId=sym_abc123` → callees for selected symbol
+
+If output says no callers found, suggest running `/index force` first.

native/Cargo.lock

@@ -37,6 +37,7 @@ walkdir = "2.5"
 ignore = "0.4"
 thiserror = "1.0"
 anyhow = "1.0"
+streaming-iterator = "0.1"
 # On Linux/macOS: use default features (simsimd + fp16lib) for SIMD-accelerated vector ops
 # On Windows: disable simsimd — MSVC lacks _mm512_reduce_add_ph intrinsic (usearch#325)
 [target.'cfg(not(target_os = "windows"))'.dependencies]

native/Cargo.toml

@@ -0,0 +1,16 @@
+; =============================================================
+; Tree-sitter query for extracting function calls from Go
+; =============================================================
+
+; Direct function calls: foo(), bar(1, 2)
+(call_expression
+  function: (identifier) @callee.name) @call
+
+; Method/package calls: obj.Method(), fmt.Println()
+(call_expression
+  function: (selector_expression
+    field: (field_identifier) @callee.name)) @call
+
+; Import: import "fmt"
+(import_spec
+  path: (interpreted_string_literal) @import.name) @import

native/queries/go-calls.scm

@@ -0,0 +1,56 @@
+; =============================================================
+; Tree-sitter query file for extracting function calls from TS/JS
+; Captures are named with @ prefix and used to extract node text
+; =============================================================
+
+; -------------------------------------------------------------
+; Direct function calls: foo(), bar(1, 2)
+; Captures the function identifier being called
+; -------------------------------------------------------------
+(call_expression
+  function: (identifier) @callee.name) @call
+
+; -------------------------------------------------------------
+; Method calls: obj.method(), this.foo(), array.map()
+; Captures the property (method name) being called
+; -------------------------------------------------------------
+(call_expression
+  function: (member_expression
+    property: (property_identifier) @callee.name)) @call
+
+; -------------------------------------------------------------
+; Constructor calls: new Foo(), new Bar(args)
+; Captures the class/constructor name
+; -------------------------------------------------------------
+(new_expression
+  constructor: (identifier) @callee.name) @constructor
+
+; -------------------------------------------------------------
+; ES6 named imports: import { foo, bar as baz } from 'module'
+; Captures each imported name and the source module
+; -------------------------------------------------------------
+(import_statement
+  (import_clause
+    (named_imports
+      (import_specifier
+        name: (identifier) @import.name)))
+  source: (string) @import.source) @import
+
+; -------------------------------------------------------------
+; Default imports: import React from 'react'
+; Captures the default import name and source
+; -------------------------------------------------------------
+(import_statement
+  (import_clause
+    (identifier) @import.default)
+  source: (string) @import.source) @import
+
+; -------------------------------------------------------------
+; Namespace imports: import * as utils from './utils'
+; Captures the namespace alias and source
+; -------------------------------------------------------------
+(import_statement
+  (import_clause
+    (namespace_import
+      (identifier) @import.namespace))
+  source: (string) @import.source) @import

native/queries/javascript-calls.scm

@@ -0,0 +1,25 @@
+; =============================================================
+; Tree-sitter query for extracting function calls from Python
+; =============================================================
+
+; Direct function calls: foo(), bar(1, 2)
+(call
+  function: (identifier) @callee.name) @call
+
+; Method calls: obj.method(), self.foo()
+(call
+  function: (attribute
+    attribute: (identifier) @callee.name)) @call
+
+; Constructor calls (same as function calls in Python, but capitalized by convention)
+; Handled by the Call type — caller can check capitalization
+
+; Import: import foo
+(import_statement
+  name: (dotted_name
+    (identifier) @import.name)) @import
+
+; From import: from module import foo, bar
+(import_from_statement
+  name: (dotted_name
+    (identifier) @import.name)) @import

native/queries/python-calls.scm

@@ -0,0 +1,32 @@
+; =============================================================
+; Tree-sitter query for extracting function calls from Rust
+; =============================================================
+
+; Direct function calls: foo(), bar(1, 2)
+(call_expression
+  function: (identifier) @callee.name) @call
+
+; Method calls: obj.method(), self.foo()
+(call_expression
+  function: (field_expression
+    field: (field_identifier) @callee.name)) @call
+
+; Path calls: std::fs::read(), Vec::new()
+(call_expression
+  function: (scoped_identifier
+    name: (identifier) @callee.name)) @call
+
+; Macro calls: println!(), vec![]
+(macro_invocation
+  macro: (identifier) @callee.name) @call
+
+; Use imports: use std::fs;
+(use_declaration
+  argument: (scoped_identifier
+    name: (identifier) @import.name)) @import
+
+; Use list imports: use std::{foo, bar};
+(use_declaration
+  argument: (scoped_use_list
+    list: (use_list
+      (identifier) @import.name))) @import