node-walk 0.1.0__py3-none-any.whl

node_walk/storage/sqlite_store.py

@@ -0,0 +1,285 @@
+"""
+SQLite implementation of GraphStore.
+
+All SQL is contained here. The rest of the codebase only depends on
+the abstract GraphStore interface from node_walk.storage.base.
+"""
+
+from __future__ import annotations
+
+import json
+import sqlite3
+from pathlib import Path
+
+from node_walk.ir.enums import Language, RelationshipType, ResolutionStatus, SymbolKind
+from node_walk.ir.models import (
+    AnalysisResult,
+    FileInfo,
+    Relationship,
+    SourceLocation,
+    Symbol,
+)
+from node_walk.storage import schema as _schema
+from node_walk.storage.base import GraphStore
+
+
+class SQLiteGraphStore(GraphStore):
+    """
+    SQLite-backed graph store.
+
+    Usage::
+
+        store = SQLiteGraphStore(".node_walk/graph.db")
+        store.store_results(analysis_results)
+    """
+
+    def __init__(self, db_path: str | Path) -> None:
+        self._db_path = Path(db_path)
+        self._db_path.parent.mkdir(parents=True, exist_ok=True)
+        self._conn = sqlite3.connect(str(self._db_path), check_same_thread=False)
+        self._conn.row_factory = sqlite3.Row
+        _schema.initialize(self._conn)
+
+    def close(self) -> None:
+        self._conn.close()
+
+    # ------------------------------------------------------------------
+    # Write operations
+    # ------------------------------------------------------------------
+
+    def store_result(self, result: AnalysisResult) -> None:
+        self.store_results([result])
+
+    def store_results(self, results: list[AnalysisResult]) -> None:
+        """Bulk-insert all files, symbols, and relationships in a single transaction."""
+        with self._conn:
+            for res in results:
+                self._upsert_file(res.file)
+                for sym in res.symbols:
+                    self._insert_symbol(sym)
+                for rel in res.relationships:
+                    self._insert_relationship(rel)
+
+    def clear(self) -> None:
+        with self._conn:
+            self._conn.execute("DELETE FROM relationships")
+            self._conn.execute("DELETE FROM symbols")
+            self._conn.execute("DELETE FROM files")
+
+    def update_relationship(
+        self, rel_id: str, target_id: str, resolution: ResolutionStatus
+    ) -> None:
+        with self._conn:
+            self._conn.execute(
+                "UPDATE relationships SET target_id = ?, resolution = ? WHERE id = ?",
+                (target_id, resolution.value, rel_id),
+            )
+
+    # ------------------------------------------------------------------
+    # Read operations
+    # ------------------------------------------------------------------
+
+    def get_file(self, file_id: str) -> FileInfo | None:
+        row = self._conn.execute(
+            "SELECT * FROM files WHERE id = ?", (file_id,)
+        ).fetchone()
+        return self._row_to_file(row) if row else None
+
+    def get_all_files(self) -> list[FileInfo]:
+        rows = self._conn.execute("SELECT * FROM files ORDER BY path").fetchall()
+        return [self._row_to_file(r) for r in rows]
+
+    def get_symbol(self, symbol_id: str) -> Symbol | None:
+        row = self._conn.execute(
+            "SELECT * FROM symbols WHERE id = ?", (symbol_id,)
+        ).fetchone()
+        return self._row_to_symbol(row) if row else None
+
+    def find_symbols_by_name(self, name: str, exact: bool = False) -> list[Symbol]:
+        if exact:
+            rows = self._conn.execute(
+                "SELECT * FROM symbols WHERE name = ? ORDER BY qualified_name",
+                (name,),
+            ).fetchall()
+        else:
+            rows = self._conn.execute(
+                "SELECT * FROM symbols WHERE name LIKE ? ORDER BY qualified_name",
+                (f"%{name}%",),
+            ).fetchall()
+        return [self._row_to_symbol(r) for r in rows]
+
+    def find_symbols_by_qualified_name(self, qname: str) -> list[Symbol]:
+        rows = self._conn.execute(
+            "SELECT * FROM symbols WHERE qualified_name = ? OR qualified_name LIKE ?",
+            (qname, f"%{qname}%"),
+        ).fetchall()
+        return [self._row_to_symbol(r) for r in rows]
+
+    def get_all_symbols(self) -> list[Symbol]:
+        rows = self._conn.execute(
+            "SELECT * FROM symbols ORDER BY qualified_name"
+        ).fetchall()
+        return [self._row_to_symbol(r) for r in rows]
+
+    def get_relationships_from(
+        self, symbol_id: str, rel_type: RelationshipType | None = None
+    ) -> list[Relationship]:
+        if rel_type:
+            rows = self._conn.execute(
+                "SELECT * FROM relationships WHERE source_id = ? AND type = ?",
+                (symbol_id, rel_type.value),
+            ).fetchall()
+        else:
+            rows = self._conn.execute(
+                "SELECT * FROM relationships WHERE source_id = ?", (symbol_id,)
+            ).fetchall()
+        return [self._row_to_rel(r) for r in rows]
+
+    def get_relationships_to(
+        self, symbol_id: str, rel_type: RelationshipType | None = None
+    ) -> list[Relationship]:
+        if rel_type:
+            rows = self._conn.execute(
+                "SELECT * FROM relationships WHERE target_id = ? AND type = ?",
+                (symbol_id, rel_type.value),
+            ).fetchall()
+        else:
+            rows = self._conn.execute(
+                "SELECT * FROM relationships WHERE target_id = ?", (symbol_id,)
+            ).fetchall()
+        return [self._row_to_rel(r) for r in rows]
+
+    def get_all_unresolved_relationships(self) -> list[Relationship]:
+        rows = self._conn.execute(
+            "SELECT * FROM relationships WHERE target_id = '' OR resolution = 'unresolved'"
+        ).fetchall()
+        return [self._row_to_rel(r) for r in rows]
+
+    def stats(self) -> dict[str, int]:
+        files = self._conn.execute("SELECT COUNT(*) FROM files").fetchone()[0]
+        symbols = self._conn.execute("SELECT COUNT(*) FROM symbols").fetchone()[0]
+        rels = self._conn.execute("SELECT COUNT(*) FROM relationships").fetchone()[0]
+        unresolved = self._conn.execute(
+            "SELECT COUNT(*) FROM relationships WHERE target_id = ''"
+        ).fetchone()[0]
+
+        kind_rows = self._conn.execute(
+            "SELECT kind, COUNT(*) as cnt FROM symbols GROUP BY kind"
+        ).fetchall()
+        kind_counts = {f"symbols_{r['kind'].lower()}": r["cnt"] for r in kind_rows}
+
+        rel_rows = self._conn.execute(
+            "SELECT type, COUNT(*) as cnt FROM relationships GROUP BY type"
+        ).fetchall()
+        rel_counts = {f"rel_{r['type'].lower()}": r["cnt"] for r in rel_rows}
+
+        return {
+            "files": files,
+            "symbols": symbols,
+            "relationships": rels,
+            "unresolved_relationships": unresolved,
+            **kind_counts,
+            **rel_counts,
+        }
+
+    # ------------------------------------------------------------------
+    # Private helpers — insert
+    # ------------------------------------------------------------------
+
+    def _upsert_file(self, file: FileInfo) -> None:
+        self._conn.execute(
+            """
+            INSERT INTO files (id, path, language, content_hash, size_bytes)
+            VALUES (?, ?, ?, ?, ?)
+            ON CONFLICT(path) DO UPDATE SET
+                language     = excluded.language,
+                content_hash = excluded.content_hash,
+                size_bytes   = excluded.size_bytes,
+                indexed_at   = datetime('now')
+            """,
+            (file.id, file.path, file.language.value, file.content_hash, file.size_bytes),
+        )
+
+    def _insert_symbol(self, sym: Symbol) -> None:
+        self._conn.execute(
+            """
+            INSERT OR IGNORE INTO symbols
+                (id, name, qualified_name, kind, language, file_id,
+                 start_line, end_line, signature, parent_id, docstring, is_async)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            """,
+            (
+                sym.id, sym.name, sym.qualified_name, sym.kind.value,
+                sym.language.value, sym.file_id, sym.start_line, sym.end_line,
+                sym.signature, sym.parent_id, sym.docstring, int(sym.is_async),
+            ),
+        )
+
+    def _insert_relationship(self, rel: Relationship) -> None:
+        loc = rel.source_location
+        self._conn.execute(
+            """
+            INSERT OR IGNORE INTO relationships
+                (id, source_id, target_id, type, source_file_id, source_line,
+                 source_col, resolution, metadata_json)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+            """,
+            (
+                rel.id, rel.source_id, rel.target_id, rel.type.value,
+                loc.file_id if loc else None,
+                loc.line if loc else None,
+                loc.col if loc else None,
+                rel.resolution.value,
+                json.dumps(rel.metadata),
+            ),
+        )
+
+    # ------------------------------------------------------------------
+    # Private helpers — row → model conversion
+    # ------------------------------------------------------------------
+
+    @staticmethod
+    def _row_to_file(row: sqlite3.Row) -> FileInfo:
+        return FileInfo(
+            id=row["id"],
+            path=row["path"],
+            language=Language(row["language"]),
+            content_hash=row["content_hash"],
+            size_bytes=row["size_bytes"],
+        )
+
+    @staticmethod
+    def _row_to_symbol(row: sqlite3.Row) -> Symbol:
+        return Symbol(
+            id=row["id"],
+            name=row["name"],
+            qualified_name=row["qualified_name"],
+            kind=SymbolKind(row["kind"]),
+            language=Language(row["language"]),
+            file_id=row["file_id"],
+            start_line=row["start_line"],
+            end_line=row["end_line"],
+            signature=row["signature"] or "",
+            parent_id=row["parent_id"],
+            docstring=row["docstring"] or "",
+            is_async=bool(row["is_async"]),
+        )
+
+    @staticmethod
+    def _row_to_rel(row: sqlite3.Row) -> Relationship:
+        loc: SourceLocation | None = None
+        if row["source_file_id"] and row["source_line"]:
+            loc = SourceLocation(
+                file_id=row["source_file_id"],
+                line=row["source_line"],
+                col=row["source_col"] or 0,
+            )
+        return Relationship(
+            id=row["id"],
+            source_id=row["source_id"],
+            target_id=row["target_id"] or "",
+            type=RelationshipType(row["type"]),
+            source_location=loc,
+            resolution=ResolutionStatus(row["resolution"]),
+            metadata=json.loads(row["metadata_json"]),
+        )

node_walk-0.1.0.dist-info/METADATA

@@ -0,0 +1,107 @@
+Metadata-Version: 2.5
+Name: node-walk
+Version: 0.1.0
+Summary: Semantic code intelligence — local-first graph of your codebase for humans and LLMs
+Author: CodeGraph Contributors
+License: MIT
+License-File: LICENSE
+Keywords: code-analysis,developer-tools,llm,semantic-graph
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.11
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: tree-sitter-python>=0.23.0
+Requires-Dist: tree-sitter>=0.23.0
+Requires-Dist: typer>=0.12.0
+Provides-Extra: dev
+Requires-Dist: pytest-cov>=5.0.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# CodeGraph
+
+**Semantic code intelligence for humans and LLMs.**
+
+Local-first · Lightweight · Python first · SQLite backed
+
+---
+
+## Quickstart
+
+### 1. Install & Setup
+```bash
+# Clone and setup the environment
+git clone <repo-url> CodeGraph
+cd CodeGraph
+python -m venv .venv
+.venv\Scripts\activate     # Windows (or source .venv/bin/activate on Unix)
+pip install -e ".[dev]"
+```
+
+### 2. Indexing and Querying
+All CLI commands discover the graph database by searching for a `.node_walk/` directory in the current directory or walking up parent directories.
+
+**To index any directory:**
+```bash
+# Activate CodeGraph venv, then go to the repository you want to index
+cd /path/to/your/project
+node_walk index .
+```
+
+**Run queries within that project's directory:**
+```bash
+# Find a symbol
+node_walk find UserService
+
+# See its definition
+node_walk definition UserService
+
+# Find callers (shows interactive picker if multiple symbols match)
+node_walk callers UserService.create_user
+
+# Trace outgoing dependencies (out-edges)
+node_walk trace UserService.create_user --depth 5
+
+# Blast radius (in-edges - what breaks if this changes?)
+node_walk blast-radius UserService.create_user
+
+# View source code of a symbol
+node_walk source UserService.create_user
+
+# Graph statistics
+node_walk stats
+
+# Export to JSON
+node_walk export --output graph.json
+```
+
+
+## Architecture
+
+```
+Repository → Language Analysis → Code IR → SQLite Graph → Query Engine → CLI
+```
+
+- **Language analysis**: Tree-sitter based Python parser
+- **Code IR**: Pydantic v2 models — Symbol, Relationship, FileInfo
+- **Storage**: SQLite with 10 indexes, WAL mode
+- **Query engine**: Pure SQLite including recursive CTEs for graph traversal
+- **CLI**: Typer + Rich
+
+## Development
+
+```bash
+python -m venv .venv
+.venv\Scripts\activate     # Windows
+pip install -e ".[dev]"
+pytest tests/ -v
+```
+
+## Graph data
+
+Stored in `.node_walk/graph.db` inside the indexed repository.
+Safe to delete and re-index at any time.

node_walk-0.1.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+node_walk/__init__.py,sha256=lyJBhaMT5oq2AaZrboYhoK_55Mnk5C4zIUhMXP6LcgA,91
+node_walk/indexer.py,sha256=RJuMeb2uMDUp4iShzOm9EaaRloFF0uwvWGbKrc9htws,6666
+node_walk/analysis/__init__.py,sha256=mb15--gtv-wpXn4jHq8MH6Rgx9ZlifBdMgWvxz1bAgE,394
+node_walk/analysis/base.py,sha256=4mYsZA7Vr6CEMIQtAzfE7kqjPg14bFWizoON_b9EI7Y,6440
+node_walk/analysis/python_analyzer.py,sha256=2gfwkdrszY4GilV48SXpGQmsEBmDauHu33ppg6JjUoA,492
+node_walk/analysis/python/__init__.py,sha256=IwtnjMD5df-Av70dT1MfCYvLWKU4ea3kz7jlGfbrr04,230
+node_walk/analysis/python/analyzer.py,sha256=D230OhlRpImwuWVsM8JDgQP4Hb3R3QI4RCunM83qKeI,1435
+node_walk/analysis/python/scope.py,sha256=8J62GeH1Ouy77xdJnj9Wo3XHyw92BI5g7yBJvg3Fbe0,1320
+node_walk/analysis/python/visitor.py,sha256=0bMzgXUl5l9aCSyoFgpIEQL8QojLlSarTAHWI7WqaR4,21734
+node_walk/cli/__init__.py,sha256=EuI73cd5fRn3eHKATNRerJCyomrPvDfvss_6PiResFA,50
+node_walk/cli/main.py,sha256=C0sGKa7iLKYHm6imrGDq7xfnuvRHP8DrTiOsEkBeJR4,19718
+node_walk/ir/__init__.py,sha256=0oQQSIpYwTRWHuJXv1IvDAeUwwh1PphRIhnpglqGpPM,741
+node_walk/ir/enums.py,sha256=BvgDkxm7YKm6hAQjVVdST9HwZkYM-oJkBDEK1mWUduc,1764
+node_walk/ir/models.py,sha256=ai1Up6Ke5vSnkLJkp1oye970tEmrS1DU5Ewj69lEIGs,3300
+node_walk/query/__init__.py,sha256=XpqXvMPtbn_i382jxIKOYXQRPm7sus6YZv1ZCkf4ZJo,50
+node_walk/query/engine.py,sha256=RJlY-zgGdcFrI6PZlM06yyTxJ9_2Y7HtcEMd4EjypR0,13887
+node_walk/storage/__init__.py,sha256=2P4IgasCF48CXlewNakplrEZTx9G3jvrYbtE_N1Yqd8,383
+node_walk/storage/base.py,sha256=FfErayseFMHuZynKoTnmGDNIloSIBiEjHDYx9LlwSdM,2419
+node_walk/storage/repository.py,sha256=7t9GoJCe5aGeQX3lR3HjSOI8IYxwu0NusDIV-RFFykk,471
+node_walk/storage/schema.py,sha256=l86Ojd3Biswzda_1N9flGXrZNL_gtHYnfjCfOTO6w0s,4461
+node_walk/storage/sqlite_store.py,sha256=RK1-dN4PIM5iwh-2NhCkbMNlMQL_CA0tG2qmQtdaJVo,10599
+node_walk-0.1.0.dist-info/METADATA,sha256=aH3NFkim2D7t2R3PVSiSuScYfoAme1Eqqb9J5OApTkE,2839
+node_walk-0.1.0.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+node_walk-0.1.0.dist-info/entry_points.txt,sha256=GfEbFx9MiZz2i7__MlP93hHCPZT3Xw1rO7dniuZgoag,122
+node_walk-0.1.0.dist-info/licenses/LICENSE,sha256=JALI9baeqOBYv5lOMAqMVUDxpuqHN0fex7dRuUfetO8,1069
+node_walk-0.1.0.dist-info/RECORD,,

node_walk-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.0
+Root-Is-Purelib: true
+Tag: py3-none-any

node_walk-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,4 @@
+[console_scripts]
+node-walk = node_walk.cli.main:app
+node_walk = node_walk.cli.main:app
+nodewalk = node_walk.cli.main:app

node_walk-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Shriram Naik
+
+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.