codebrain 0.1.1__tar.gz → 0.3.0__tar.gz

{codebrain-0.1.1 → codebrain-0.3.0}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2025 CodeBrain Contributors
+Copyright (c) 2026 CodeBrain Contributors
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

{codebrain-0.1.1 → codebrain-0.3.0}/PKG-INFO

@@ -1,9 +1,30 @@
 Metadata-Version: 2.4
 Name: codebrain
-Version: 0.1.1
+Version: 0.3.0
 Summary: Know what breaks before you break it. Structural knowledge graph for codebases — impact analysis, dead code detection, health scores. No LLM required.
 Author: CodeBrain Contributors
-License: MIT
+License: MIT License
+        
+        Copyright (c) 2026 CodeBrain Contributors
+        
+        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.
+        
 Project-URL: Homepage, https://github.com/monk0062006/CodeBrain
 Project-URL: Repository, https://github.com/monk0062006/CodeBrain
 Project-URL: Issues, https://github.com/monk0062006/CodeBrain/issues
@@ -28,6 +49,7 @@ Requires-Dist: click>=8.1
 Requires-Dist: watchdog>=3.0
 Requires-Dist: mcp<1.20,>=1.0
 Requires-Dist: jinja2>=3.1
+Requires-Dist: psutil>=5.9
 Provides-Extra: api
 Requires-Dist: fastapi>=0.110; extra == "api"
 Requires-Dist: uvicorn>=0.27; extra == "api"
@@ -59,14 +81,26 @@ cd your-project
 brain init
 ```
 
-That's it. Now you can:
+You'll immediately see:
+
+```
+Indexing my-project (93 files) ...
+Done in 1.2s — 1861 nodes, 8837 edges
+
+=== Instant Findings ===
+Health: 60/100 (B)
+Riskiest symbols: Flask (34 files), App (35 files), Scaffold (28 files)
+Dead code: 11 candidates (none high-confidence)
+```
+
+Then explore:
 
 ```bash
 brain impact my_function    # What breaks if I change this?
+brain callers my_function   # Who calls this?
 brain health                # Codebase health score (0-100)
 brain hotspots              # Riskiest symbols in the codebase
 brain deadcode              # Unused functions with confidence levels
-brain coupling              # Which files are too tightly coupled?
 brain zoom                  # Google Maps-style navigation of your architecture
 ```
 
@@ -173,7 +207,7 @@ Module: main.py — 429 symbols, 15853 lines. Isolated (no dependents).
 ### Claude Code (MCP)
 
 ```bash
-brain agent    # auto-generates CLAUDE.md + MCP config
+brain setup    # auto-generates CLAUDE.md + MCP config
 ```
 
 Or add manually to your MCP config:
@@ -214,6 +248,7 @@ The `propose_change` MCP tool validates changes **before** the AI writes them. W
 | `brain status` | Show index statistics |
 | `brain search <query>` | Find symbols by name |
 | `brain impact <name>` | What breaks if this changes? |
+| `brain callers <name>` | Who calls this function? (reverse trace) |
 | `brain trace <name>` | Forward call chain from a symbol |
 | `brain deps <name>` | Dependencies of a symbol or file |
 | `brain deadcode` | Find unused functions/classes |
@@ -265,7 +300,13 @@ Or use environment variables:
 | Python | AST (stdlib) | Full — functions, classes, imports, calls, dataflow |
 | TypeScript/JSX | tree-sitter | Full — requires `pip install codebrain[ts]` |
 | JavaScript/JSX | tree-sitter | Full — requires `pip install codebrain[ts]` |
-| TypeScript (fallback) | regex | Experimental — used when tree-sitter not installed |
+| Java | tree-sitter | Full — Spring Boot aware (DI, endpoints, JPA) |
+| Go | tree-sitter | Full — interfaces, goroutines, struct methods |
+| Rust | tree-sitter | Full — traits, impls, modules, derive macros |
+| C# | tree-sitter | Full — async/await, LINQ, attributes |
+| Kotlin | tree-sitter | Full — coroutines, extensions, data classes |
+| Dart/Flutter | regex | Good — classes, mixins, extensions |
+| Vue SFC | regex | Good — components, props, events |
 
 ### Adding Languages
 
@@ -275,9 +316,9 @@ Implement `BaseParser` and register via entry points:
 from codebrain.parser.base import BaseParser
 from codebrain.parser.models import ParsedFile
 
-class RustParser(BaseParser):
+class MyParser(BaseParser):
     def extensions(self) -> frozenset[str]:
-        return frozenset({".rs"})
+        return frozenset({".xyz"})
 
     def parse(self, path, repo_root) -> ParsedFile:
         ...  # parse and return nodes + edges

{codebrain-0.1.1 → codebrain-0.3.0}/README.md

@@ -10,14 +10,26 @@ cd your-project
 brain init
 ```
 
-That's it. Now you can:
+You'll immediately see:
+
+```
+Indexing my-project (93 files) ...
+Done in 1.2s — 1861 nodes, 8837 edges
+
+=== Instant Findings ===
+Health: 60/100 (B)
+Riskiest symbols: Flask (34 files), App (35 files), Scaffold (28 files)
+Dead code: 11 candidates (none high-confidence)
+```
+
+Then explore:
 
 ```bash
 brain impact my_function    # What breaks if I change this?
+brain callers my_function   # Who calls this?
 brain health                # Codebase health score (0-100)
 brain hotspots              # Riskiest symbols in the codebase
 brain deadcode              # Unused functions with confidence levels
-brain coupling              # Which files are too tightly coupled?
 brain zoom                  # Google Maps-style navigation of your architecture
 ```
 
@@ -124,7 +136,7 @@ Module: main.py — 429 symbols, 15853 lines. Isolated (no dependents).
 ### Claude Code (MCP)
 
 ```bash
-brain agent    # auto-generates CLAUDE.md + MCP config
+brain setup    # auto-generates CLAUDE.md + MCP config
 ```
 
 Or add manually to your MCP config:
@@ -165,6 +177,7 @@ The `propose_change` MCP tool validates changes **before** the AI writes them. W
 | `brain status` | Show index statistics |
 | `brain search <query>` | Find symbols by name |
 | `brain impact <name>` | What breaks if this changes? |
+| `brain callers <name>` | Who calls this function? (reverse trace) |
 | `brain trace <name>` | Forward call chain from a symbol |
 | `brain deps <name>` | Dependencies of a symbol or file |
 | `brain deadcode` | Find unused functions/classes |
@@ -216,7 +229,13 @@ Or use environment variables:
 | Python | AST (stdlib) | Full — functions, classes, imports, calls, dataflow |
 | TypeScript/JSX | tree-sitter | Full — requires `pip install codebrain[ts]` |
 | JavaScript/JSX | tree-sitter | Full — requires `pip install codebrain[ts]` |
-| TypeScript (fallback) | regex | Experimental — used when tree-sitter not installed |
+| Java | tree-sitter | Full — Spring Boot aware (DI, endpoints, JPA) |
+| Go | tree-sitter | Full — interfaces, goroutines, struct methods |
+| Rust | tree-sitter | Full — traits, impls, modules, derive macros |
+| C# | tree-sitter | Full — async/await, LINQ, attributes |
+| Kotlin | tree-sitter | Full — coroutines, extensions, data classes |
+| Dart/Flutter | regex | Good — classes, mixins, extensions |
+| Vue SFC | regex | Good — components, props, events |
 
 ### Adding Languages
 
@@ -226,9 +245,9 @@ Implement `BaseParser` and register via entry points:
 from codebrain.parser.base import BaseParser
 from codebrain.parser.models import ParsedFile
 
-class RustParser(BaseParser):
+class MyParser(BaseParser):
     def extensions(self) -> frozenset[str]:
-        return frozenset({".rs"})
+        return frozenset({".xyz"})
 
     def parse(self, path, repo_root) -> ParsedFile:
         ...  # parse and return nodes + edges

codebrain-0.3.0/codebrain/__init__.py

@@ -0,0 +1,10 @@
+"""CodeBrain — persistent structural knowledge graph for codebases."""
+
+from importlib.metadata import PackageNotFoundError, version as _pkg_version
+
+try:
+    __version__ = _pkg_version("codebrain")
+except PackageNotFoundError:
+    # Source checkout without installation (e.g. running from a worktree).
+    # Keep in sync with [project.version] in pyproject.toml.
+    __version__ = "0.3.0"

codebrain-0.3.0/codebrain/actions/__init__.py

@@ -0,0 +1,16 @@
+"""Autonomous actions engine — CodeBrain acts, not just analyzes."""
+
+from codebrain.actions.test_gen import TestGenerator, GeneratedTest
+from codebrain.actions.reviewer import CodeReviewer, CodeReview, ReviewFinding
+from codebrain.actions.refactor import Refactorer, RefactorSuggestion, RefactorResult
+
+__all__ = [
+    "TestGenerator",
+    "GeneratedTest",
+    "CodeReviewer",
+    "CodeReview",
+    "ReviewFinding",
+    "Refactorer",
+    "RefactorSuggestion",
+    "RefactorResult",
+]

codebrain-0.3.0/codebrain/actions/base.py

@@ -0,0 +1,139 @@
+"""Shared infrastructure for autonomous actions."""
+
+from __future__ import annotations
+
+import ast
+import re
+from pathlib import Path
+
+from codebrain.context import ContextGenerator
+from codebrain.graph.query import QueryEngine
+from codebrain.graph.store import GraphStore
+from codebrain.rewriter import LLMClient
+from codebrain.utils import is_test_file
+
+
+class ActionEngine:
+    """Base class for all autonomous actions."""
+
+    def __init__(self, store: GraphStore, repo_root: Path, llm: LLMClient):
+        self.store = store
+        self.repo_root = repo_root
+        self.llm = llm
+        self.context = ContextGenerator(store)
+        self.engine = QueryEngine(store)
+
+    def _get_symbol_context(self, name: str) -> list[dict]:
+        """Full structural context for a symbol."""
+        return self.context.for_symbol(name)
+
+    def _get_file_source(self, file_path: str) -> str:
+        """Read actual source code from disk."""
+        full = self.repo_root / file_path
+        if not full.exists():
+            return ""
+        try:
+            return full.read_text(encoding="utf-8", errors="replace")
+        except Exception:
+            return ""
+
+    def _get_function_source(self, file_path: str, name: str, line_start: int) -> str:
+        """Extract a single function/class source from a file."""
+        source = self._get_file_source(file_path)
+        if not source:
+            return ""
+        try:
+            tree = ast.parse(source)
+        except SyntaxError:
+            return ""
+        for node in ast.walk(tree):
+            if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef, ast.ClassDef)):
+                if node.name == name:
+                    lines = source.splitlines()
+                    end = getattr(node, "end_lineno", None) or (node.lineno + 20)
+                    return "\n".join(lines[node.lineno - 1 : end])
+        # Fallback: return from line_start
+        if line_start > 0:
+            lines = source.splitlines()
+            return "\n".join(lines[line_start - 1 : line_start + 30])
+        return ""
+
+    def _get_existing_tests(self, file_path: str) -> str:
+        """Find existing test files and return a sample of test patterns."""
+        # Look for test files in tests/ directory
+        tests_dir = self.repo_root / "tests"
+        if not tests_dir.exists():
+            tests_dir = self.repo_root / "test"
+        if not tests_dir.exists():
+            return "# No existing test directory found"
+
+        samples = []
+        for tf in sorted(tests_dir.iterdir()):
+            if tf.name.startswith("test_") and tf.suffix == ".py":
+                try:
+                    content = tf.read_text(encoding="utf-8", errors="replace")
+                    # Extract first test function as pattern example
+                    for match in re.finditer(
+                        r"((?:@\w+.*\n)*def test_\w+\(.*?\):.*?)(?=\ndef |\nclass |\Z)",
+                        content,
+                        re.DOTALL,
+                    ):
+                        sample = match.group(1).strip()
+                        if len(sample) < 500:
+                            samples.append(f"# From {tf.name}:\n{sample}")
+                            break
+                except Exception:
+                    continue
+            if len(samples) >= 3:
+                break
+
+        return "\n\n".join(samples) if samples else "# No test examples found"
+
+    def _validate_python(self, code: str) -> list[str]:
+        """Check Python syntax and basic correctness."""
+        errors = []
+        try:
+            ast.parse(code)
+        except SyntaxError as e:
+            errors.append(f"Syntax error: {e.msg} at line {e.lineno}")
+        return errors
+
+    def _extract_code(self, response: str) -> str:
+        """Strip markdown fences from LLM output."""
+        pattern = r"```(?:python|java|typescript|javascript)?\s*\n(.*?)```"
+        matches = re.findall(pattern, response, re.DOTALL)
+        if matches:
+            return max(matches, key=len).strip()
+        # If no fences, return as-is but trim non-code lines at start
+        lines = response.strip().split("\n")
+        for i, line in enumerate(lines):
+            s = line.strip()
+            if s.startswith(("import ", "from ", "def ", "class ", "#", '"""', "@", "async ")):
+                return "\n".join(lines[i:]).strip()
+        return response.strip()
+
+    def _interpret_fingerprint(self, fingerprint: str) -> str:
+        """Convert structural fingerprint to actionable guidance."""
+        if not fingerprint:
+            return "- No structural fingerprint available."
+        guidance = []
+        if "returns:" in fingerprint:
+            returns = fingerprint.split("returns:")[1].split("|")[0].strip()
+            guidance.append(f"- Returns: {returns}. Test each return path.")
+        if "raises:" in fingerprint:
+            exc = fingerprint.split("raises:")[1].split("|")[0].strip()
+            guidance.append(f"- Raises: {exc}. Test exception paths.")
+        if "mutates:" in fingerprint:
+            mut = fingerprint.split("mutates:")[1].split("|")[0].strip()
+            guidance.append(f"- Mutates: {mut}. Verify state changes.")
+        if "io:" in fingerprint:
+            io = fingerprint.split("io:")[1].split("|")[0].strip()
+            guidance.append(f"- I/O: {io}. Mock these in tests.")
+        if "yields" in fingerprint:
+            guidance.append("- Generator function. Test iteration and empty case.")
+        if "async" in fingerprint:
+            guidance.append("- Async function. Use @pytest.mark.asyncio.")
+        if "control:" in fingerprint:
+            ctrl = fingerprint.split("control:")[1].split("|")[0].strip()
+            guidance.append(f"- Control flow: {ctrl}. Test each branch.")
+        return "\n".join(guidance) if guidance else "- No special patterns detected."