roam-code 8.0.1__tar.gz → 8.1.1__tar.gz

{roam_code-8.0.1 → roam_code-8.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: roam-code
-Version: 8.0.1
+Version: 8.1.1
 Summary: Instant codebase comprehension for AI coding agents
 Author: CosmoHac
 License-Expression: MIT
@@ -1023,7 +1023,7 @@ Delete `.roam/` from your project root to clean up local data.
 git clone https://github.com/Cranot/roam-code.git
 cd roam-code
 pip install -e ".[dev]"   # includes pytest, ruff
-pytest tests/              # 1664 tests, Python 3.9-3.13
+pytest tests/              # 1691 tests, Python 3.9-3.13
 
 # Or use Make targets:
 make dev      # install with dev extras
@@ -1079,7 +1079,7 @@ roam-code/
 │   └── output/
 │       ├── formatter.py               # Token-efficient formatting
 │       └── sarif.py                   # SARIF 2.1.0 output
-└── tests/                             # 1664 tests across 28 test files
+└── tests/                             # 1691 tests across 29 test files
 ```
 
 </details>
@@ -1120,7 +1120,7 @@ Optional: [fastmcp](https://github.com/jlowin/fastmcp) (MCP server)
 git clone https://github.com/Cranot/roam-code.git
 cd roam-code
 pip install -e .
-pytest tests/   # All 1664 tests must pass
+pytest tests/   # All 1691 tests must pass
 ```
 
 Good first contributions: add a [Tier 1 language](src/roam/languages/) (see `go_lang.py` or `php_lang.py` as templates), improve reference resolution, add benchmark repos, extend SARIF converters, add MCP tools.

{roam_code-8.0.1 → roam_code-8.1.1}/README.md

@@ -988,7 +988,7 @@ Delete `.roam/` from your project root to clean up local data.
 git clone https://github.com/Cranot/roam-code.git
 cd roam-code
 pip install -e ".[dev]"   # includes pytest, ruff
-pytest tests/              # 1664 tests, Python 3.9-3.13
+pytest tests/              # 1691 tests, Python 3.9-3.13
 
 # Or use Make targets:
 make dev      # install with dev extras
@@ -1044,7 +1044,7 @@ roam-code/
 │   └── output/
 │       ├── formatter.py               # Token-efficient formatting
 │       └── sarif.py                   # SARIF 2.1.0 output
-└── tests/                             # 1664 tests across 28 test files
+└── tests/                             # 1691 tests across 29 test files
 ```
 
 </details>
@@ -1085,7 +1085,7 @@ Optional: [fastmcp](https://github.com/jlowin/fastmcp) (MCP server)
 git clone https://github.com/Cranot/roam-code.git
 cd roam-code
 pip install -e .
-pytest tests/   # All 1664 tests must pass
+pytest tests/   # All 1691 tests must pass
 ```
 
 Good first contributions: add a [Tier 1 language](src/roam/languages/) (see `go_lang.py` or `php_lang.py` as templates), improve reference resolution, add benchmark repos, extend SARIF converters, add MCP tools.

{roam_code-8.0.1 → roam_code-8.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "roam-code"
-version = "8.0.1"
+version = "8.1.1"
 description = "Instant codebase comprehension for AI coding agents"
 readme = "README.md"
 requires-python = ">=3.9"
@@ -55,6 +55,9 @@ where = ["src"]
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 addopts = "-q --tb=short"
+markers = [
+    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
+]
 
 [tool.ruff]
 target-version = "py39"

{roam_code-8.0.1 → roam_code-8.1.1}/src/roam/commands/cmd_complexity.py

@@ -5,6 +5,8 @@ functions/methods by cognitive complexity to identify the hardest-to-
 understand code in the project.
 """
 
+from __future__ import annotations
+
 import click
 
 from roam.db.connection import open_db
@@ -12,6 +14,15 @@ from roam.commands.resolve import ensure_index
 from roam.output.formatter import loc, abbrev_kind, to_json, json_envelope
 
 
+def _safe_metric(row, key, default=0.0):
+    """Safely access a metric column that may not exist in older DBs."""
+    try:
+        v = row[key]
+        return v if v is not None else default
+    except (KeyError, IndexError):
+        return default
+
+
 def _severity(score: float) -> str:
     """Map cognitive complexity score to a severity label."""
     if score >= 25:
@@ -144,11 +155,11 @@ def complexity(ctx, target, limit, threshold, by_file, bumpy_road):
                         "return_count": r["return_count"],
                         "bool_op_count": r["bool_op_count"],
                         "callback_depth": r["callback_depth"],
-                        "cyclomatic_density": r["cyclomatic_density"] or 0,
-                        "halstead_volume": r["halstead_volume"] or 0,
-                        "halstead_difficulty": r["halstead_difficulty"] or 0,
-                        "halstead_effort": r["halstead_effort"] or 0,
-                        "halstead_bugs": r["halstead_bugs"] or 0,
+                        "cyclomatic_density": _safe_metric(r, "cyclomatic_density"),
+                        "halstead_volume": _safe_metric(r, "halstead_volume"),
+                        "halstead_difficulty": _safe_metric(r, "halstead_difficulty"),
+                        "halstead_effort": _safe_metric(r, "halstead_effort"),
+                        "halstead_bugs": _safe_metric(r, "halstead_bugs"),
                         "severity": _severity(r["cognitive_complexity"]),
                     }
                     for r in rows
@@ -181,11 +192,11 @@ def complexity(ctx, target, limit, threshold, by_file, bumpy_road):
                 factors.append(f"params={r['param_count']}")
             if r["return_count"] >= 4:
                 factors.append(f"ret={r['return_count']}")
-            cd = r["cyclomatic_density"]
-            if cd and cd > 0.15:
+            cd = _safe_metric(r, "cyclomatic_density")
+            if cd > 0.15:
                 factors.append(f"density={cd:.2f}")
-            hv = r["halstead_volume"]
-            if hv and hv > 500:
+            hv = _safe_metric(r, "halstead_volume")
+            if hv > 500:
                 factors.append(f"H.vol={hv:.0f}")
 
             factor_str = f" ({', '.join(factors)})" if factors else ""

{roam_code-8.0.1 → roam_code-8.1.1}/src/roam/commands/cmd_dead.py

@@ -40,6 +40,20 @@ _API_PREFIXES = ("get", "use", "create", "validate", "fetch", "update",
                  "delete", "find", "check", "make", "build", "parse")
 
 
+_ABC_METHOD_NAMES = frozenset({
+    "language_name", "file_extensions", "extract_symbols", "extract_references",
+    "get_docstring", "get_signature", "node_text",
+    "detect", "supported_bridges",
+    "resolve_cross_language", "get_bridge_edges",
+})
+
+
+def _is_test_path(file_path):
+    """Check if a file is a test file (discovered by pytest, not imported)."""
+    base = os.path.basename(file_path).lower()
+    return base.startswith("test_") or base.endswith("_test.py")
+
+
 def _dead_action(r, file_imported):
     """Compute actionable verdict and confidence % for a dead symbol.
 
@@ -62,6 +76,18 @@ def _dead_action(r, file_imported):
     except (KeyError, IndexError):
         kind = ""
 
+    # Test file symbols — discovered by pytest, never imported directly
+    if _is_test_path(r["file_path"]):
+        return "INTENTIONAL", 10
+
+    # CLI command functions — loaded dynamically via LazyGroup/importlib
+    if base.startswith("cmd_") and kind == "function":
+        return "INTENTIONAL", 20
+
+    # ABC method overrides — called polymorphically, not by direct import
+    if kind == "method" and name in _ABC_METHOD_NAMES:
+        return "INTENTIONAL", 10
+
     # Entry point / lifecycle hooks (check original case for camelCase hooks)
     if name in _ENTRY_NAMES or name_lower in _ENTRY_NAMES:
         return "INTENTIONAL", 60
@@ -251,6 +277,8 @@ def _analyze_dead(conn):
     Returns (high, low, imported_files) where high/low are lists of Row objects.
     """
     rows = conn.execute(UNREFERENCED_EXPORTS).fetchall()
+    # Exclude test files — their symbols are discovered by pytest, not imported
+    rows = [r for r in rows if not _is_test_path(r["file_path"])]
     if not rows:
         return [], [], set()
 

{roam_code-8.0.1 → roam_code-8.1.1}/src/roam/commands/cmd_health.py

@@ -52,13 +52,21 @@ _FRAMEWORK_NAMES = frozenset({
 _UTILITY_PATH_PATTERNS = (
     "composables/", "utils/", "services/", "lib/", "helpers/",
     "shared/", "config/", "core/", "hooks/", "stores/",
+    "output/", "db/", "common/", "internal/", "infra/",
+)
+
+_UTILITY_FILE_PATTERNS = (
+    "resolve.py", "helpers.py", "common.py", "base.py",
 )
 
 
 def _is_utility_path(file_path):
-    """Check if a file is in a utility/infrastructure directory."""
+    """Check if a file is in a utility/infrastructure directory or is a known utility file."""
     p = file_path.replace("\\", "/").lower()
-    return any(pat in p for pat in _UTILITY_PATH_PATTERNS)
+    if any(pat in p for pat in _UTILITY_PATH_PATTERNS):
+        return True
+    basename = p.rsplit("/", 1)[-1] if "/" in p else p
+    return basename in _UTILITY_FILE_PATTERNS
 
 
 def _percentile(sorted_values, pct):

{roam_code-8.0.1 → roam_code-8.1.1}/src/roam/languages/python_lang.py

@@ -2,6 +2,12 @@
 from __future__ import annotations
 from .base import LanguageExtractor
 
+# Builtin type names that don't create real reference edges
+_BUILTIN_TYPES = frozenset({
+    "int", "str", "float", "bool", "bytes", "None",
+    "list", "dict", "set", "tuple", "type", "object",
+})
+
 
 class PythonExtractor(LanguageExtractor):
     """Full Python symbol and reference extractor."""
@@ -170,6 +176,15 @@ class PythonExtractor(LanguageExtractor):
             parent_name=parent_name,
         ))
 
+        # Extract instance attributes from __init__ body (self.x = ...)
+        if name == "__init__" and parent_name:
+            body = node.child_by_field_name("body")
+            if body:
+                self_name = self._detect_self_name(node)
+                self._extract_init_attributes(
+                    body, source, symbols, parent_name, dunder_all, self_name,
+                )
+
     def _extract_class(self, node, source, file_path, symbols, parent_name, dunder_all, decorator_node=None):
         name_node = node.child_by_field_name("name")
         if name_node is None:
@@ -320,6 +335,88 @@ class PythonExtractor(LanguageExtractor):
                 return text
         return None
 
+    def _detect_self_name(self, func_node) -> bytes:
+        """Detect the self/cls parameter name from a method's first argument (Pyan-inspired)."""
+        params = func_node.child_by_field_name("parameters")
+        if params:
+            for child in params.children:
+                if child.type == "identifier":
+                    return child.text
+                # typed_parameter: (self: Self) or typed_default_parameter
+                if child.type in ("typed_parameter", "typed_default_parameter"):
+                    name_node = child.child_by_field_name("name")
+                    if name_node and name_node.type == "identifier":
+                        return name_node.text
+        return b"self"
+
+    def _extract_init_attributes(self, body_node, source, symbols, parent_name,
+                                 dunder_all, self_name):
+        """Extract instance attributes from __init__ body (self.x = value)."""
+        seen = set()
+        # Collect existing class-level property names to avoid duplicates
+        for sym in symbols:
+            if sym.get("parent_name") == parent_name and sym.get("kind") == "property":
+                seen.add(sym["name"])
+
+        self._collect_self_assignments(body_node, source, symbols, parent_name,
+                                       dunder_all, self_name, seen)
+
+    def _collect_self_assignments(self, node, source, symbols, parent_name,
+                                  dunder_all, self_name, seen):
+        """Walk __init__ body collecting self.x = ... assignments."""
+        for child in node.children:
+            if child.type == "expression_statement":
+                for sub in child.children:
+                    if sub.type == "assignment":
+                        self._try_extract_self_attr(
+                            sub, source, symbols, parent_name, dunder_all,
+                            self_name, seen,
+                        )
+            elif child.type == "assignment":
+                self._try_extract_self_attr(
+                    child, source, symbols, parent_name, dunder_all,
+                    self_name, seen,
+                )
+            # Recurse into conditional/try/with blocks (common in __init__)
+            elif child.type in ("if_statement", "try_statement", "with_statement",
+                                "for_statement", "block"):
+                self._collect_self_assignments(
+                    child, source, symbols, parent_name, dunder_all,
+                    self_name, seen,
+                )
+
+    def _try_extract_self_attr(self, assign_node, source, symbols, parent_name,
+                                dunder_all, self_name, seen):
+        """If assign_node is `self.x = value`, extract x as a property symbol."""
+        left = assign_node.child_by_field_name("left")
+        if left is None or left.type != "attribute":
+            return
+        obj = left.child_by_field_name("object")
+        if obj is None or obj.type != "identifier" or obj.text != self_name:
+            return
+        attr_node = left.child_by_field_name("attribute")
+        if attr_node is None:
+            return
+        attr_name = self.node_text(attr_node, source)
+        if attr_name in seen:
+            return
+        seen.add(attr_name)
+
+        right = assign_node.child_by_field_name("right")
+        default_value = self._extract_literal_value(right, source) if right else None
+
+        qualified = f"{parent_name}.{attr_name}"
+        symbols.append(self._make_symbol(
+            name=attr_name,
+            kind="property",
+            line_start=assign_node.start_point[0] + 1,
+            line_end=assign_node.end_point[0] + 1,
+            qualified_name=qualified,
+            visibility=self._visibility(attr_name),
+            parent_name=parent_name,
+            default_value=default_value,
+        ))
+
     def _is_exported(self, name: str, dunder_all: set[str] | None) -> bool:
         if dunder_all is not None:
             return name in dunder_all
@@ -333,6 +430,15 @@ class PythonExtractor(LanguageExtractor):
                 self._extract_from_import(child, source, refs, scope_name)
             elif child.type == "call":
                 self._extract_call(child, source, refs, scope_name)
+            elif child.type == "decorated_definition":
+                self._extract_decorator_refs(child, source, refs, scope_name)
+                self._walk_refs(child, source, file_path, refs, scope_name)
+            elif child.type in ("assignment", "expression_statement"):
+                # Walk type annotations on assignments: x: Path = ..., self.x: int = ...
+                # Note: type aliases (PathList = List[Path]) are NOT handled here —
+                # the RHS is a subscript expression, not a type annotation field.
+                self._extract_assignment_type_refs(child, source, refs, scope_name)
+                self._walk_refs(child, source, file_path, refs, scope_name)
             else:
                 # Recurse, updating scope for classes/functions
                 new_scope = scope_name
@@ -341,8 +447,110 @@ class PythonExtractor(LanguageExtractor):
                     if n:
                         fname = self.node_text(n, source)
                         new_scope = f"{scope_name}.{fname}" if scope_name else fname
+                    # Extract type annotation refs from function parameters and return
+                    if child.type == "function_definition":
+                        self._extract_type_refs(child, source, refs, new_scope)
                 self._walk_refs(child, source, file_path, refs, new_scope)
 
+    def _extract_decorator_refs(self, decorated_node, source, refs, scope_name):
+        """Extract references from decorators (e.g., @cache, @app.route)."""
+        for child in decorated_node.children:
+            if child.type == "decorator":
+                # The decorator content is after the '@'
+                for sub in child.children:
+                    if sub.type == "identifier":
+                        name = self.node_text(sub, source)
+                        refs.append(self._make_reference(
+                            target_name=name,
+                            kind="call",
+                            line=sub.start_point[0] + 1,
+                            source_name=scope_name,
+                        ))
+                    elif sub.type == "attribute":
+                        name = self.node_text(sub, source)
+                        refs.append(self._make_reference(
+                            target_name=name,
+                            kind="call",
+                            line=sub.start_point[0] + 1,
+                            source_name=scope_name,
+                        ))
+                    elif sub.type == "call":
+                        self._extract_call(sub, source, refs, scope_name)
+
+    def _extract_type_refs(self, func_node, source, refs, scope_name):
+        """Extract references from type annotations in function signatures."""
+        # Parameter type annotations
+        params = func_node.child_by_field_name("parameters")
+        if params:
+            for param in params.children:
+                type_node = param.child_by_field_name("type")
+                if type_node:
+                    self._walk_type_node(type_node, source, refs, scope_name)
+
+        # Return type annotation
+        ret = func_node.child_by_field_name("return_type")
+        if ret:
+            self._walk_type_node(ret, source, refs, scope_name)
+
+    def _extract_assignment_type_refs(self, node, source, refs, scope_name):
+        """Extract type_ref edges from annotated assignments (class fields, module vars)."""
+        targets = [node] if node.type == "assignment" else []
+        if node.type == "expression_statement":
+            for sub in node.children:
+                if sub.type == "assignment":
+                    targets.append(sub)
+        for assign in targets:
+            type_node = assign.child_by_field_name("type")
+            if type_node:
+                self._walk_type_node(type_node, source, refs, scope_name)
+
+    def _walk_type_node(self, node, source, refs, scope_name):
+        """Walk a type annotation node and extract type references."""
+        if node.type == "identifier":
+            name = self.node_text(node, source)
+            if name not in _BUILTIN_TYPES:
+                refs.append(self._make_reference(
+                    target_name=name,
+                    kind="type_ref",
+                    line=node.start_point[0] + 1,
+                    source_name=scope_name,
+                ))
+        elif node.type == "attribute":
+            name = self.node_text(node, source)
+            refs.append(self._make_reference(
+                target_name=name,
+                kind="type_ref",
+                line=node.start_point[0] + 1,
+                source_name=scope_name,
+            ))
+        elif node.type == "string":
+            # Forward reference: Optional["Config"] -> extract "Config"
+            for child in node.children:
+                if child.type == "string_content":
+                    name = self.node_text(child, source).strip()
+                    # Simple identifier forward ref: "Config"
+                    if name.isidentifier() and name not in _BUILTIN_TYPES:
+                        refs.append(self._make_reference(
+                            target_name=name,
+                            kind="type_ref",
+                            line=child.start_point[0] + 1,
+                            source_name=scope_name,
+                        ))
+                    # Dotted forward ref: "module.ClassName"
+                    elif "." in name and all(
+                        p.isidentifier() for p in name.split(".")
+                    ):
+                        refs.append(self._make_reference(
+                            target_name=name,
+                            kind="type_ref",
+                            line=child.start_point[0] + 1,
+                            source_name=scope_name,
+                        ))
+        else:
+            # Recurse into generic types like List[Item], Optional[str], etc.
+            for child in node.children:
+                self._walk_type_node(child, source, refs, scope_name)
+
     def _extract_import(self, node, source, refs, scope_name):
         # import x, import x.y, import x as y
         for child in node.children:

{roam_code-8.0.1 → roam_code-8.1.1}/src/roam_code.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: roam-code
-Version: 8.0.1
+Version: 8.1.1
 Summary: Instant codebase comprehension for AI coding agents
 Author: CosmoHac
 License-Expression: MIT
@@ -1023,7 +1023,7 @@ Delete `.roam/` from your project root to clean up local data.
 git clone https://github.com/Cranot/roam-code.git
 cd roam-code
 pip install -e ".[dev]"   # includes pytest, ruff
-pytest tests/              # 1664 tests, Python 3.9-3.13
+pytest tests/              # 1691 tests, Python 3.9-3.13
 
 # Or use Make targets:
 make dev      # install with dev extras
@@ -1079,7 +1079,7 @@ roam-code/
 │   └── output/
 │       ├── formatter.py               # Token-efficient formatting
 │       └── sarif.py                   # SARIF 2.1.0 output
-└── tests/                             # 1664 tests across 28 test files
+└── tests/                             # 1691 tests across 29 test files
 ```
 
 </details>
@@ -1120,7 +1120,7 @@ Optional: [fastmcp](https://github.com/jlowin/fastmcp) (MCP server)
 git clone https://github.com/Cranot/roam-code.git
 cd roam-code
 pip install -e .
-pytest tests/   # All 1664 tests must pass
+pytest tests/   # All 1691 tests must pass
 ```
 
 Good first contributions: add a [Tier 1 language](src/roam/languages/) (see `go_lang.py` or `php_lang.py` as templates), improve reference resolution, add benchmark repos, extend SARIF converters, add MCP tools.

{roam_code-8.0.1 → roam_code-8.1.1}/src/roam_code.egg-info/SOURCES.txt

@@ -145,6 +145,7 @@ tests/test_json_contracts.py
 tests/test_languages.py
 tests/test_performance.py
 tests/test_pr_risk_author.py
+tests/test_python_extractor_v2.py
 tests/test_resolve.py
 tests/test_salesforce.py
 tests/test_smoke.py