coderay 1.0.6__tar.gz → 1.0.7__tar.gz

{coderay-1.0.6/src/coderay.egg-info → coderay-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coderay
-Version: 1.0.6
+Version: 1.0.7
 Summary: X-ray your codebase — semantic search, code graphs, file skeletons, and MCP server
 Author-email: Bogdan Copocean <bogdancopocean@gmail.com>
 License-Expression: MIT

{coderay-1.0.6 → coderay-1.0.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "coderay"
-version = "1.0.6"
+version = "1.0.7"
 description = "X-ray your codebase — semantic search, code graphs, file skeletons, and MCP server"
 readme = "README.md"
 license = "MIT"

{coderay-1.0.6 → coderay-1.0.7}/src/coderay/core/models.py

@@ -68,3 +68,22 @@ class GraphEdge:
     source: str
     target: str
     kind: EdgeKind
+
+
+@dataclass(frozen=True)
+class ImpactResult:
+    """Result of an impact-radius query with resolution diagnostics."""
+
+    resolved_node: str | None
+    nodes: list[GraphNode]
+    hint: str | None = None
+
+    def to_dict(self) -> dict:
+        """Serialize to a JSON-compatible dict."""
+        d: dict = {
+            "resolved_node": self.resolved_node,
+            "results": [n.to_dict() for n in self.nodes],
+        }
+        if self.hint:
+            d["hint"] = self.hint
+        return d

{coderay-1.0.6 → coderay-1.0.7}/src/coderay/graph/code_graph.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import logging
-from collections import defaultdict
+from collections import defaultdict, deque
 from typing import Any
 
 import networkx as nx
 
-from coderay.core.models import EdgeKind, GraphEdge, GraphNode, NodeKind
+from coderay.core.models import EdgeKind, GraphEdge, GraphNode, ImpactResult, NodeKind
 from coderay.parsing.languages import (
     get_init_filenames,
     get_resolution_suffixes,
@@ -280,29 +280,76 @@ class CodeGraph:
     # Public query methods
     # ------------------------------------------------------------------
 
-    def get_impact_radius(self, symbol: str, depth: int = 2) -> list[GraphNode]:
+    _IMPACT_EDGE_KINDS = frozenset(
+        {EdgeKind.CALLS, EdgeKind.IMPORTS, EdgeKind.INHERITS}
+    )
+
+    def get_impact_radius(
+        self, symbol: str, depth: int = 2
+    ) -> ImpactResult:
         """Find all nodes that could be affected if ``symbol`` changes.
 
+        Traverses predecessors via CALLS, IMPORTS, and INHERITS edges
+        only.  DEFINES edges (containment) are excluded because a
+        module defining a symbol is not "affected" by changes to it.
+
         Args:
-            depth: Number of reverse-BFS hops. Higher values may return
-                very large sets.
+            symbol: Fully qualified node ID or resolvable name.
+            depth: Number of reverse-BFS hops.
         """
         resolved = self.resolve_symbol(symbol) or symbol
         if resolved not in self._g:
-            return []
+            return self._not_found_result(symbol)
+
         visited: set[str] = set()
-        frontier = {resolved}
-        for _ in range(depth):
-            next_frontier: set[str] = set()
-            for nid in frontier:
-                if nid not in self._g:
-                    continue
+        queue: deque[tuple[str, int]] = deque(
+            (pred, 1)
+            for pred in self._g.predecessors(resolved)
+            if self._is_impact_edge(pred, resolved)
+        )
+        while queue:
+            nid, hop = queue.popleft()
+            if nid in visited:
+                continue
+            visited.add(nid)
+            if hop < depth and nid in self._g:
                 for pred in self._g.predecessors(nid):
-                    if pred not in visited:
-                        visited.add(pred)
-                        next_frontier.add(pred)
-            frontier = next_frontier
-        return [self.get_node(nid) for nid in visited if self.get_node(nid) is not None]
+                    if (
+                        pred not in visited
+                        and self._is_impact_edge(pred, nid)
+                    ):
+                        queue.append((pred, hop + 1))
+
+        nodes = [
+            self.get_node(nid)
+            for nid in visited
+            if self.get_node(nid) is not None
+        ]
+        return ImpactResult(resolved_node=resolved, nodes=nodes)
+
+    def _is_impact_edge(self, source: str, target: str) -> bool:
+        """Check whether the edge is a dependency (not containment)."""
+        kind = self._g.edges[source, target].get("kind")
+        return kind in self._IMPACT_EDGE_KINDS
+
+    def _not_found_result(self, symbol: str) -> ImpactResult:
+        """Build an ImpactResult with diagnostic hints for a missing node."""
+        file_part = (
+            symbol.split("::")[0] if "::" in symbol else None
+        )
+        available = (
+            sorted(self._file_index.get(file_part, set()))
+            if file_part
+            else []
+        )
+        hint = f"Node '{symbol}' not in the graph."
+        if available:
+            hint += (
+                f" Available nodes in {file_part}: {available}"
+            )
+        return ImpactResult(
+            resolved_node=None, nodes=[], hint=hint
+        )
 
     @property
     def node_count(self) -> int:

{coderay-1.0.6 → coderay-1.0.7}/src/coderay/graph/extractor.py

@@ -380,12 +380,11 @@ class GraphTreeSitterParser(BaseTreeSitterParser):
                 qualified_name=qualified,
             )
         )
+        definer = self._module_id
+        if scope_stack:
+            definer = f"{self.file_path}::{'.'.join(scope_stack)}"
         self._edges.append(
-            GraphEdge(
-                source=self._module_id,
-                target=node_id,
-                kind=EdgeKind.DEFINES,
-            )
+            GraphEdge(source=definer, target=node_id, kind=EdgeKind.DEFINES)
         )
 
         # Register in FileContext so calls resolve to the full node_id.
@@ -422,12 +421,11 @@ class GraphTreeSitterParser(BaseTreeSitterParser):
                 qualified_name=qualified,
             )
         )
+        definer = self._module_id
+        if scope_stack:
+            definer = f"{self.file_path}::{'.'.join(scope_stack)}"
         self._edges.append(
-            GraphEdge(
-                source=self._module_id,
-                target=node_id,
-                kind=EdgeKind.DEFINES,
-            )
+            GraphEdge(source=definer, target=node_id, kind=EdgeKind.DEFINES)
         )
 
         # Extract base classes from the superclass list and resolve them

{coderay-1.0.6 → coderay-1.0.7}/src/coderay/mcp_server/server.py

@@ -132,6 +132,13 @@ def get_file_skeleton(
         str,
         Field(description="Absolute or relative path to the file"),
     ],
+    include_imports: Annotated[
+        bool,
+        Field(
+            description="Include import statements in the skeleton. "
+            "Set to false to reduce noise when you only need signatures.",
+        ),
+    ] = True,
 ) -> str:
     """Get the API surface of a file (signatures, no bodies)."""
     from coderay.skeleton.extractor import extract_skeleton
@@ -140,7 +147,7 @@ def get_file_skeleton(
     if not p.is_file():
         raise FileNotFoundError(f"File not found: {file_path}")
     content = p.read_text(encoding="utf-8", errors="replace")
-    return extract_skeleton(p, content)
+    return extract_skeleton(p, content, include_imports=include_imports)
 
 
 @mcp.tool(
@@ -176,10 +183,7 @@ def get_impact_radius(
             "No graph found. Ask the user to run 'coderay build' "
             "in their terminal, then retry."
         )
-    impact = graph.get_impact_radius(node_id, depth=max_depth)
-    return {
-        "results": [n.to_dict() for n in impact],
-    }
+    return graph.get_impact_radius(node_id, depth=max_depth).to_dict()
 
 
 @mcp.resource(

{coderay-1.0.6 → coderay-1.0.7}/src/coderay/retrieval/models.py

@@ -14,6 +14,7 @@ class SearchResult:
     end_line: int
     symbol: str
     content: str
+    score: float = 0.0
     truncated: bool = False
 
     @classmethod
@@ -46,6 +47,7 @@ class SearchResult:
             end_line=row["end_line"],
             symbol=row["symbol"],
             content=content,
+            score=float(row.get("score", 0.0)),
             truncated=truncated,
         )
 
@@ -57,6 +59,7 @@ class SearchResult:
             "end_line": self.end_line,
             "symbol": self.symbol,
             "content": self.content,
+            "score": self.score,
         }
         if self.truncated:
             d["truncated"] = True

{coderay-1.0.6 → coderay-1.0.7}/src/coderay/skeleton/extractor.py

@@ -13,13 +13,22 @@ ELLIPSIS = "..."
 def extract_skeleton(
     path: str | Path,
     content: str,
+    *,
+    include_imports: bool = True,
 ) -> str:
-    """Extract the skeleton of a source file (signatures, no bodies)."""
+    """Extract the skeleton of a source file (signatures, no bodies).
+
+    Args:
+        path: Source file path (used for language detection).
+        content: Full file content.
+        include_imports: When False, import statements are omitted from
+            the output to reduce noise.
+    """
     ctx = parse_file(path, content)
     if ctx is None:
         return content
 
-    parser = SkeletonTreeSitterParser(ctx)
+    parser = SkeletonTreeSitterParser(ctx, include_imports=include_imports)
     try:
         lines = parser.collect_lines()
     except Exception:  # pragma: no cover - defensive fallback
@@ -31,6 +40,10 @@ def extract_skeleton(
 class SkeletonTreeSitterParser(BaseTreeSitterParser):
     """Tree-sitter based skeleton extractor for source files."""
 
+    def __init__(self, context, *, include_imports: bool = True) -> None:
+        super().__init__(context)
+        self._include_imports = include_imports
+
     def collect_lines(self) -> list[str]:
         """Return the skeleton of a file as a list of lines."""
         tree = self.get_tree()
@@ -111,7 +124,9 @@ class SkeletonTreeSitterParser(BaseTreeSitterParser):
             return
 
         if ntype in lang_cfg.import_types:
-            lines.append(indent + self.node_text(node).strip())
+            if self._include_imports:
+                lines.append(indent + self.node_text(node).strip())
+            return
 
         if ntype in lang_cfg.decorator_scope_types:
             for child in node.named_children:

{coderay-1.0.6 → coderay-1.0.7/src/coderay.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coderay
-Version: 1.0.6
+Version: 1.0.7
 Summary: X-ray your codebase — semantic search, code graphs, file skeletons, and MCP server
 Author-email: Bogdan Copocean <bogdancopocean@gmail.com>
 License-Expression: MIT