codemunch-pro 0.1.0__tar.gz

codemunch_pro-0.1.0/.gitignore

@@ -0,0 +1,11 @@
+__pycache__/
+*.pyc
+*.pyo
+.venv/
+dist/
+build/
+*.egg-info/
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+*.db

codemunch_pro-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Jai Dunlop
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

codemunch_pro-0.1.0/PKG-INFO

@@ -0,0 +1,142 @@
+Metadata-Version: 2.4
+Name: codemunch-pro
+Version: 0.1.0
+Summary: Intelligent code indexing MCP server. Tree-sitter AST extraction, hybrid search (FTS5 + vector), call graphs, 10 languages, incremental indexing.
+Project-URL: Homepage, https://github.com/BigJai/codemunch-pro
+Project-URL: Repository, https://github.com/BigJai/codemunch-pro
+Project-URL: Issues, https://github.com/BigJai/codemunch-pro/issues
+Author-email: Jai Dunlop <jaidunlop85@gmail.com>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ast,call-graph,code-indexing,code-search,developer-tools,mcp,mcp-server,tree-sitter
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Text Processing :: Indexing
+Requires-Python: >=3.10
+Requires-Dist: fastembed>=0.6.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: mcp[cli]>=1.6.0
+Requires-Dist: pathspec>=0.12.0
+Requires-Dist: sqlite-vec>=0.1.0
+Requires-Dist: tree-sitter-language-pack>=0.7.0
+Description-Content-Type: text/markdown
+
+# CodeMunch Pro
+
+Intelligent code indexing MCP server. Tree-sitter AST extraction, hybrid search (FTS5 + vector), call graphs, 10 languages, incremental indexing.
+
+**Save 99% of tokens** — get exact function source via byte-offset seek instead of reading entire files.
+
+## Install
+
+```bash
+pip install codemunch-pro
+```
+
+## Quick Start
+
+### Claude Desktop / Cline
+
+Add to your MCP client config:
+
+```json
+{
+  "mcpServers": {
+    "codemunch-pro": {
+      "command": "codemunch-pro"
+    }
+  }
+}
+```
+
+### HTTP Server
+
+```bash
+codemunch-pro --transport streamable-http --port 5002
+```
+
+## 13 MCP Tools
+
+| Tool | Description |
+|------|-------------|
+| `index_folder` | Index a local directory (incremental, SHA-256 based) |
+| `index_repo` | Index a GitHub/GitLab repo (v1.1) |
+| `list_repos` | List all indexed repositories with stats |
+| `invalidate_cache` | Force re-index a repository |
+| `file_tree` | Get directory tree with file counts |
+| `file_outline` | List symbols in a single file |
+| `repo_outline` | List all symbols in repo (summary) |
+| `get_symbol` | Get full source of one symbol (O(1) byte seek) |
+| `get_symbols` | Batch get multiple symbols |
+| `search_symbols` | Hybrid search (FTS5 + vector RRF) |
+| `search_text` | Full-text search in file contents |
+| `get_callees` | What does this function call? |
+| `get_callers` | Who calls this function? |
+
+## 10 Languages
+
+Python, JavaScript, TypeScript, Go, Rust, Java, C, C++, C#, Ruby
+
+All via [tree-sitter-language-pack](https://pypi.org/project/tree-sitter-language-pack/) — zero compilation, pre-built binaries.
+
+## Key Features
+
+### O(1) Symbol Retrieval
+Every symbol stores its byte offset and length. `get_symbol` seeks directly to the function source — no reading entire files. A 200-byte function from a 40KB file = **99.5% token savings**.
+
+### Incremental Indexing
+Files are hashed (SHA-256). Only changed files are re-parsed. Re-indexing a 10K file repo after changing one file takes milliseconds.
+
+### Hybrid Search (FTS5 + Vector)
+Combines BM25 keyword matching with semantic vector similarity using Reciprocal Rank Fusion. Search "authentication middleware" and find `auth_middleware`, `verify_token`, and `login_handler`.
+
+### Call Graphs
+Traces function calls through the AST. `get_callees("main")` shows what `main` calls. `get_callers("authenticate")` shows who calls `authenticate`. Supports depth traversal.
+
+### Full-Text Content Search
+Search raw file contents — string literals, TODO comments, config values, error messages. Not just symbol names.
+
+## How It Works
+
+1. **Parse** — tree-sitter builds an AST for each source file
+2. **Extract** — Walk AST to find functions, classes, methods, types, interfaces
+3. **Store** — SQLite database per repo with FTS5 virtual tables
+4. **Embed** — FastEmbed (ONNX, CPU-only) generates 384-dim vectors for semantic search
+5. **Graph** — Call expressions extracted from function bodies, edges stored and resolved
+6. **Serve** — FastMCP exposes 13 tools via stdio or HTTP
+
+## Architecture
+
+```
+~/.codemunch-pro/
+├── myproject_a1b2c3d4e5f6.db    # Per-repo SQLite database
+├── otherproject_7890abcdef.db
+└── ...
+
+Each DB contains:
+├── files          # Indexed files with SHA-256 hashes
+├── symbols        # Functions, classes, methods, types
+├── symbols_fts    # FTS5 full-text search index
+├── symbols_vec    # sqlite-vec 384-dim vector index
+├── call_edges     # Call graph (caller → callee)
+└── file_content_fts  # Raw file content search
+```
+
+## Use Cases
+
+- **AI Coding Agents**: Give your agent surgical access to codebases without burning context
+- **Code Review**: Find all callers of a function before changing its signature
+- **Onboarding**: Search symbols semantically — "where is error handling?" finds relevant code
+- **Refactoring**: Map call graphs before moving functions between modules
+- **Documentation**: Extract all public APIs with signatures and docstrings
+
+## License
+
+MIT

codemunch_pro-0.1.0/README.md

@@ -0,0 +1,112 @@
+# CodeMunch Pro
+
+Intelligent code indexing MCP server. Tree-sitter AST extraction, hybrid search (FTS5 + vector), call graphs, 10 languages, incremental indexing.
+
+**Save 99% of tokens** — get exact function source via byte-offset seek instead of reading entire files.
+
+## Install
+
+```bash
+pip install codemunch-pro
+```
+
+## Quick Start
+
+### Claude Desktop / Cline
+
+Add to your MCP client config:
+
+```json
+{
+  "mcpServers": {
+    "codemunch-pro": {
+      "command": "codemunch-pro"
+    }
+  }
+}
+```
+
+### HTTP Server
+
+```bash
+codemunch-pro --transport streamable-http --port 5002
+```
+
+## 13 MCP Tools
+
+| Tool | Description |
+|------|-------------|
+| `index_folder` | Index a local directory (incremental, SHA-256 based) |
+| `index_repo` | Index a GitHub/GitLab repo (v1.1) |
+| `list_repos` | List all indexed repositories with stats |
+| `invalidate_cache` | Force re-index a repository |
+| `file_tree` | Get directory tree with file counts |
+| `file_outline` | List symbols in a single file |
+| `repo_outline` | List all symbols in repo (summary) |
+| `get_symbol` | Get full source of one symbol (O(1) byte seek) |
+| `get_symbols` | Batch get multiple symbols |
+| `search_symbols` | Hybrid search (FTS5 + vector RRF) |
+| `search_text` | Full-text search in file contents |
+| `get_callees` | What does this function call? |
+| `get_callers` | Who calls this function? |
+
+## 10 Languages
+
+Python, JavaScript, TypeScript, Go, Rust, Java, C, C++, C#, Ruby
+
+All via [tree-sitter-language-pack](https://pypi.org/project/tree-sitter-language-pack/) — zero compilation, pre-built binaries.
+
+## Key Features
+
+### O(1) Symbol Retrieval
+Every symbol stores its byte offset and length. `get_symbol` seeks directly to the function source — no reading entire files. A 200-byte function from a 40KB file = **99.5% token savings**.
+
+### Incremental Indexing
+Files are hashed (SHA-256). Only changed files are re-parsed. Re-indexing a 10K file repo after changing one file takes milliseconds.
+
+### Hybrid Search (FTS5 + Vector)
+Combines BM25 keyword matching with semantic vector similarity using Reciprocal Rank Fusion. Search "authentication middleware" and find `auth_middleware`, `verify_token`, and `login_handler`.
+
+### Call Graphs
+Traces function calls through the AST. `get_callees("main")` shows what `main` calls. `get_callers("authenticate")` shows who calls `authenticate`. Supports depth traversal.
+
+### Full-Text Content Search
+Search raw file contents — string literals, TODO comments, config values, error messages. Not just symbol names.
+
+## How It Works
+
+1. **Parse** — tree-sitter builds an AST for each source file
+2. **Extract** — Walk AST to find functions, classes, methods, types, interfaces
+3. **Store** — SQLite database per repo with FTS5 virtual tables
+4. **Embed** — FastEmbed (ONNX, CPU-only) generates 384-dim vectors for semantic search
+5. **Graph** — Call expressions extracted from function bodies, edges stored and resolved
+6. **Serve** — FastMCP exposes 13 tools via stdio or HTTP
+
+## Architecture
+
+```
+~/.codemunch-pro/
+├── myproject_a1b2c3d4e5f6.db    # Per-repo SQLite database
+├── otherproject_7890abcdef.db
+└── ...
+
+Each DB contains:
+├── files          # Indexed files with SHA-256 hashes
+├── symbols        # Functions, classes, methods, types
+├── symbols_fts    # FTS5 full-text search index
+├── symbols_vec    # sqlite-vec 384-dim vector index
+├── call_edges     # Call graph (caller → callee)
+└── file_content_fts  # Raw file content search
+```
+
+## Use Cases
+
+- **AI Coding Agents**: Give your agent surgical access to codebases without burning context
+- **Code Review**: Find all callers of a function before changing its signature
+- **Onboarding**: Search symbols semantically — "where is error handling?" finds relevant code
+- **Refactoring**: Map call graphs before moving functions between modules
+- **Documentation**: Extract all public APIs with signatures and docstrings
+
+## License
+
+MIT

codemunch_pro-0.1.0/pyproject.toml

@@ -0,0 +1,59 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "codemunch-pro"
+version = "0.1.0"
+description = "Intelligent code indexing MCP server. Tree-sitter AST extraction, hybrid search (FTS5 + vector), call graphs, 10 languages, incremental indexing."
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [
+  { name = "Jai Dunlop", email = "jaidunlop85@gmail.com" },
+]
+keywords = [
+  "mcp",
+  "mcp-server",
+  "code-indexing",
+  "tree-sitter",
+  "code-search",
+  "call-graph",
+  "ast",
+  "developer-tools",
+]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Intended Audience :: Developers",
+  "License :: OSI Approved :: MIT License",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Topic :: Software Development :: Libraries",
+  "Topic :: Text Processing :: Indexing",
+]
+dependencies = [
+  "mcp[cli]>=1.6.0",
+  "tree-sitter-language-pack>=0.7.0",
+  "httpx>=0.27.0",
+  "fastembed>=0.6.0",
+  "sqlite-vec>=0.1.0",
+  "pathspec>=0.12.0",
+]
+
+[project.scripts]
+codemunch-pro = "codemunch_pro.__main__:main"
+
+[project.urls]
+Homepage = "https://github.com/BigJai/codemunch-pro"
+Repository = "https://github.com/BigJai/codemunch-pro"
+Issues = "https://github.com/BigJai/codemunch-pro/issues"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/codemunch_pro"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+asyncio_mode = "auto"

codemunch_pro-0.1.0/server.json

@@ -0,0 +1,148 @@
+{
+  "$schema": "https://cdn.mcpregistry.io/specs/v2025-12-11/server-schema.json",
+  "name": "CodeMunch Pro",
+  "description": "Intelligent code indexing MCP server. Tree-sitter AST extraction, hybrid search (FTS5 + vector), call graphs, 10 languages, incremental indexing. Save 99% of tokens with O(1) byte-offset symbol retrieval.",
+  "repository": {
+    "url": "https://github.com/BigJai/codemunch-pro",
+    "source": "github",
+    "id": "BigJai/codemunch-pro"
+  },
+  "version_detail": {
+    "version": "0.1.0",
+    "release_date": "2026-03-02",
+    "is_latest": true
+  },
+  "packages": [
+    {
+      "registry_name": "pypi",
+      "name": "codemunch-pro",
+      "version": "0.1.0",
+      "runtime": "python",
+      "runtime_version": ">=3.10",
+      "package_arguments": [],
+      "environment_variables": []
+    }
+  ],
+  "remotes": [],
+  "tools": [
+    {
+      "name": "index_folder",
+      "description": "Index a local directory using tree-sitter AST parsing. Supports 10 languages with incremental indexing.",
+      "annotations": {
+        "title": "Index Folder",
+        "readOnlyHint": false,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "index_repo",
+      "description": "Index a GitHub or GitLab repository by cloning it locally.",
+      "annotations": {
+        "title": "Index Repository",
+        "readOnlyHint": false,
+        "openWorldHint": true
+      }
+    },
+    {
+      "name": "list_repos",
+      "description": "List all indexed repositories with stats.",
+      "annotations": {
+        "title": "List Repositories",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "invalidate_cache",
+      "description": "Force re-index a repository by clearing its cache.",
+      "annotations": {
+        "title": "Invalidate Cache",
+        "readOnlyHint": false,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "file_tree",
+      "description": "Get directory tree of an indexed repository with file counts.",
+      "annotations": {
+        "title": "File Tree",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "file_outline",
+      "description": "List all symbols in a single file ordered by line number.",
+      "annotations": {
+        "title": "File Outline",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "repo_outline",
+      "description": "List all symbols in the repository (summary view).",
+      "annotations": {
+        "title": "Repository Outline",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "get_symbol",
+      "description": "Get full source code of a symbol via O(1) byte-offset seek. Saves 99% tokens vs reading entire files.",
+      "annotations": {
+        "title": "Get Symbol",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "get_symbols",
+      "description": "Batch get source code for multiple symbols at once.",
+      "annotations": {
+        "title": "Get Symbols (Batch)",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "search_symbols",
+      "description": "Hybrid search across symbols using FTS5 keyword + vector semantic similarity with Reciprocal Rank Fusion.",
+      "annotations": {
+        "title": "Search Symbols",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "search_text",
+      "description": "Full-text search in file contents. Find string literals, comments, config values.",
+      "annotations": {
+        "title": "Search Text",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "get_callees",
+      "description": "Get what a function calls (outgoing call graph edges) with depth traversal.",
+      "annotations": {
+        "title": "Get Callees",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    },
+    {
+      "name": "get_callers",
+      "description": "Get who calls a function (incoming call graph edges) with depth traversal.",
+      "annotations": {
+        "title": "Get Callers",
+        "readOnlyHint": true,
+        "openWorldHint": false
+      }
+    }
+  ],
+  "prompts": [],
+  "resources": []
+}

codemunch_pro-0.1.0/src/codemunch_pro/__init__.py

@@ -0,0 +1,7 @@
+"""CodeMunch Pro — Intelligent Code Indexing MCP Server.
+
+Tree-sitter AST extraction, hybrid search (FTS5 + vector),
+call graphs, 10 languages, incremental indexing.
+"""
+
+__version__ = '0.1.0'

codemunch_pro-0.1.0/src/codemunch_pro/__main__.py

@@ -0,0 +1,45 @@
+"""CLI entry point for CodeMunch Pro."""
+
+import argparse
+import sys
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        prog='codemunch-pro',
+        description='Intelligent code indexing MCP server',
+    )
+    parser.add_argument(
+        '--version', action='version', version=f'%(prog)s 0.1.0',
+    )
+    parser.add_argument(
+        '--transport',
+        choices=['stdio', 'streamable-http'],
+        default='stdio',
+        help='MCP transport (default: stdio)',
+    )
+    parser.add_argument(
+        '--port',
+        type=int,
+        default=5002,
+        help='Port for streamable-http transport (default: 5002)',
+    )
+
+    args = parser.parse_args()
+
+    from codemunch_pro.server import create_server
+
+    mcp = create_server(transport=args.transport, port=args.port)
+
+    if args.transport == 'stdio':
+        mcp.run(transport='stdio')
+    else:
+        mcp.run(
+            transport='streamable-http',
+            host='0.0.0.0',
+            port=args.port,
+        )
+
+
+if __name__ == '__main__':
+    main()

codemunch_pro-0.1.0/src/codemunch_pro/embedder/__init__.py

@@ -0,0 +1,5 @@
+"""Embedder module — FastEmbed ONNX embeddings."""
+
+from codemunch_pro.embedder.embed import Embedder
+
+__all__ = ['Embedder']

codemunch_pro-0.1.0/src/codemunch_pro/embedder/embed.py

@@ -0,0 +1,71 @@
+"""FastEmbed ONNX embeddings for semantic code search."""
+
+from __future__ import annotations
+
+import logging
+from typing import Sequence
+
+logger = logging.getLogger(__name__)
+
+MODEL_NAME = 'BAAI/bge-small-en-v1.5'
+EMBEDDING_DIM = 384
+
+
+class Embedder:
+    """Lazy-loading FastEmbed wrapper for code symbol embeddings."""
+
+    def __init__(self, model_name: str = MODEL_NAME):
+        self.model_name = model_name
+        self._model = None
+
+    @property
+    def model(self):
+        """Lazy-load the embedding model on first use."""
+        if self._model is None:
+            from fastembed import TextEmbedding
+            logger.info('Loading embedding model: %s', self.model_name)
+            self._model = TextEmbedding(model_name=self.model_name)
+        return self._model
+
+    def embed(self, texts: Sequence[str]) -> list[list[float]]:
+        """Embed a batch of texts.
+
+        Args:
+            texts: Sequence of strings to embed.
+
+        Returns:
+            List of embedding vectors (each is a list of floats).
+        """
+        if not texts:
+            return []
+        return [vec.tolist() for vec in self.model.embed(list(texts))]
+
+    def embed_one(self, text: str) -> list[float]:
+        """Embed a single text string."""
+        results = self.embed([text])
+        return results[0] if results else []
+
+    @staticmethod
+    def format_symbol_text(
+        name: str,
+        kind: str,
+        signature: str = '',
+        docstring: str = '',
+        language: str = '',
+    ) -> str:
+        """Format a symbol's metadata into a text string for embedding.
+
+        Creates a semantic representation that captures what the symbol does,
+        not just its name.
+        """
+        parts = []
+        if language:
+            parts.append(f'{language}')
+        parts.append(f'{kind}: {name}')
+        if signature:
+            parts.append(signature)
+        if docstring:
+            # Truncate long docstrings
+            doc = docstring[:200]
+            parts.append(doc)
+        return ' | '.join(parts)

codemunch_pro-0.1.0/src/codemunch_pro/parser/__init__.py

@@ -0,0 +1,13 @@
+"""Parser module — tree-sitter AST extraction."""
+
+from codemunch_pro.parser.symbols import Symbol, CallEdge
+from codemunch_pro.parser.languages import LANGUAGES, get_language_for_file
+from codemunch_pro.parser.extractor import extract_symbols
+
+__all__ = [
+    'Symbol',
+    'CallEdge',
+    'LANGUAGES',
+    'get_language_for_file',
+    'extract_symbols',
+]

codemunch_pro-0.1.0/src/codemunch_pro/parser/call_graph.py

@@ -0,0 +1,29 @@
+"""Call graph utilities — callee/caller traversal with depth control."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+
+
+@dataclass
+class CallGraphNode:
+    """A node in the call graph traversal result."""
+
+    qualified_name: str
+    kind: str
+    file_path: str
+    line: int
+    depth: int
+    calls: list[str] = field(default_factory=list)
+    called_by: list[str] = field(default_factory=list)
+
+    def to_dict(self) -> dict:
+        return {
+            'qualified_name': self.qualified_name,
+            'kind': self.kind,
+            'file_path': self.file_path,
+            'line': self.line,
+            'depth': self.depth,
+            'calls': self.calls,
+            'called_by': self.called_by,
+        }