codegraph-ir 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

cgir/__init__.py

@@ -3,4 +3,4 @@
 See ``Code-IR.md`` at the repo root for the product specification.
 """
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"

cgir/analyses/cfg.py

@@ -79,7 +79,13 @@ def build(graph: RepoGraph, repo_path: Path, adapter: LanguageAdapter | None = N
         body = file_adapter.function_body(func_ts)
         if body is None:
             continue
-        builder = _CFGBuilder(graph=graph, owner=func, source=source, adapter=file_adapter)
+        builder = _CFGBuilder(
+            graph=graph,
+            owner=func,
+            source=source,
+            adapter=file_adapter,
+            global_names=file_adapter.global_declared_names(func_ts, source),
+        )
         builder.build_block(body, predecessors=[func.id], controller=None)
 
 
@@ -89,6 +95,9 @@ class _CFGBuilder:
     owner: Node
     source: bytes
     adapter: LanguageAdapter
+    # names declared global/nonlocal in this function: writes to them mutate
+    # outer state and are recorded as `mutates`, not local `writes`.
+    global_names: set[str] = field(default_factory=set)
     _counter: int = field(default=0, init=False)
 
     def build_block(
@@ -102,6 +111,11 @@ class _CFGBuilder:
 
     def build_stmt(self, ts_node: TSNode, preds: list[str], controller: str | None) -> list[str]:
         desc = self.adapter.describe_statement(ts_node, self.source)
+        if isinstance(desc, AssignDesc) and self.global_names:
+            outer = [w for w in desc.writes if w in self.global_names]
+            if outer:
+                desc.writes = [w for w in desc.writes if w not in self.global_names]
+                desc.mutates = list(desc.mutates) + outer
         if isinstance(desc, BranchDesc):
             return self._build_branch(ts_node, desc, preds, controller)
         if isinstance(desc, LoopDesc):

cgir/languages/base.py

@@ -264,6 +264,14 @@ class LanguageAdapter(ABC):
     def describe_statement(self, node: TSNode, source: bytes) -> StatementDesc:
         """Classify one statement and extract its parts (see descriptors)."""
 
+    def global_declared_names(self, func_node: TSNode, source: bytes) -> set[str]:
+        """Names this function declares as outer-scope (`global`/`nonlocal`).
+
+        Assignments to these names mutate state *outside* the function, so
+        the CFG builder records them as ``mutates`` rather than local
+        ``writes``. Default: none (TS/Go have no such declaration form)."""
+        return set()
+
     # --- phase 3: ingest extraction ------------------------------------------
 
     @abstractmethod

cgir/languages/go.py

@@ -295,8 +295,11 @@ class GoAdapter(LanguageAdapter):
         first_decl = next(
             (c for c in root.named_children if c.type not in {"comment", "package_clause"}), None
         )
+        pinnable = {"function_declaration", "method_declaration", "type_declaration"}
         module_pins = pin_index.module_pins(
-            first_decl.start_point[0] if first_decl is not None else None
+            first_decl.start_point[0]
+            if first_decl is not None and first_decl.type in pinnable
+            else None
         )
         if module_pins:
             for decl in decls:

cgir/languages/python.py

@@ -205,6 +205,28 @@ class PythonAdapter(LanguageAdapter):
     def function_body(self, func_node: TSNode) -> TSNode | None:
         return func_node.child_by_field_name("body")
 
+    def global_declared_names(self, func_node: TSNode, source: bytes) -> set[str]:
+        """Names bound by `global`/`nonlocal` in *this* function's own body.
+
+        Nested function bodies are skipped — their declarations apply to
+        their own CFG walk, not this one."""
+        names: set[str] = set()
+        body = func_node.child_by_field_name("body")
+        if body is None:
+            return names
+        stack: list[TSNode] = [body]
+        while stack:
+            node = stack.pop()
+            if node.type in {"function_definition", "class_definition"}:
+                continue  # a nested scope's declarations are not ours
+            if node.type in {"global_statement", "nonlocal_statement"}:
+                for child in node.named_children:
+                    if child.type == "identifier":
+                        names.add(_text(child, source))
+                continue
+            stack.extend(node.named_children)
+        return names
+
     def block_statements(self, block: TSNode) -> list[TSNode]:
         return [c for c in block.named_children if c.type != "comment"]
 
@@ -335,8 +357,14 @@ class PythonAdapter(LanguageAdapter):
         for child in root.children:
             decls.extend(self._top_level_decl(child, source, module_name, pin_index))
         first_decl = next((c for c in root.children if c.type != "comment"), None)
+        # A header block only belongs to the first statement when that
+        # statement is *pinnable* (a definition) — touching an import keeps
+        # the pins module-level.
+        pinnable = {"function_definition", "class_definition", "decorated_definition"}
         module_pins = pin_index.module_pins(
-            first_decl.start_point[0] if first_decl is not None else None
+            first_decl.start_point[0]
+            if first_decl is not None and first_decl.type in pinnable
+            else None
         )
         if module_pins:
             for decl in decls:

cgir/languages/typescript.py

@@ -391,8 +391,20 @@ class TypeScriptAdapter(LanguageAdapter):
         for child in root.children:
             decls.extend(self._top_level(child, source, rel_path, pin_index))
         first_decl = next((c for c in root.children if c.type != "comment"), None)
+        # Only a pinnable definition claims an adjacent header block; an
+        # import keeps the pins module-level.
+        pinnable = {
+            "function_declaration",
+            "class_declaration",
+            "abstract_class_declaration",
+            "export_statement",
+            "lexical_declaration",
+            "variable_declaration",
+        }
         module_pins = pin_index.module_pins(
-            first_decl.start_point[0] if first_decl is not None else None
+            first_decl.start_point[0]
+            if first_decl is not None and first_decl.type in pinnable
+            else None
         )
         if module_pins:
             for decl in decls:

{codegraph_ir-0.2.0.dist-info → codegraph_ir-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codegraph-ir
-Version: 0.2.0
+Version: 0.2.1
 Summary: The deterministic contract layer for AI-modified codebases — effects, purity, contracts, and blast radius, with zero LLM calls.
 Project-URL: Homepage, https://github.com/asonkiya/llm-semantic-compilers
 Project-URL: Repository, https://github.com/asonkiya/llm-semantic-compilers

{codegraph_ir-0.2.0.dist-info → codegraph_ir-0.2.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-cgir/__init__.py,sha256=V7neyGEmb34avZRvMxCGk1Cedjlxo-0OVTTeRevz6eA,153
+cgir/__init__.py,sha256=P6fJ6w2O8KkY-PZHk_scyugcQ4VC0s3lzSZfng_Yi4M,153
 cgir/cli.py,sha256=HOz6lY7kOz0uJKh_x_alCylOFkUNv6RL-hEDEB5pvJc,29609
 cgir/config.py,sha256=-PV5vhm_cKdRu5eYZ88GLW1e8FazgczbZ1CWal9rHSk,630
 cgir/hooks.py,sha256=Pg1KQvLmU01aLy0yhWYKt7vLn6k7TxyV1v3KKlMwYoE,7059
@@ -9,7 +9,7 @@ cgir/verify.py,sha256=ks0apBRShE5a0XXoxq6F_X10uygxPRyfEkc9Lg5PzBU,6314
 cgir/watch.py,sha256=yNrVltZ9GMlkRnxH_OAVz21o4Wy8Q55QfYIG5te1ryI,5867
 cgir/analyses/__init__.py,sha256=zXyy1sTQ1_vr6ngVp-wHXQ_tvfPMMoqo6Y8VDUzaPJs,298
 cgir/analyses/call_graph.py,sha256=lT_hiaAnNomtkzWcxiXubHLqbhVKzVIwi7LTpgaXNyI,4432
-cgir/analyses/cfg.py,sha256=22qXPTIxR_gfRe7P0SNivfs6haXdPb_vbaiSN-WkkMw,12866
+cgir/analyses/cfg.py,sha256=zoHl1aek0hpBRrAvoG2HAVVrlq5DLrKPAwzTQJWuM2w,13504
 cgir/analyses/effects.py,sha256=0wjkoK4z3vTEAU1YQg7Du54lo7qbL8M4gDe7GGiMmBg,4051
 cgir/analyses/entrypoints.py,sha256=ZDAbfXpmu1fBwWj7moBEgo9UeEH6NzQJUwAR_I51YkI,2203
 cgir/analyses/param_flow.py,sha256=CGLmylORfT_GG_xk8eNx9RvH8TGVSQ3scSGJlf25a5k,2623
@@ -32,12 +32,12 @@ cgir/ir/edges.py,sha256=pDIFgpg4UVx9lcTi8Y7aoAxGYT5yEU6aEqywq-KNa4I,700
 cgir/ir/graph.py,sha256=T18-WS_wi_nDfjd7P7_xjZylkUPrR0JTtpwpJqXBnKs,4592
 cgir/ir/nodes.py,sha256=P5dZ4GqQZwsxANg5e_jYc5uN3R6hkUYqjP7O6n-XDsY,858
 cgir/languages/__init__.py,sha256=4CQavFpRbmfdw9MjIh8nKOv0o3md8E8YC-ECiuUGXdY,779
-cgir/languages/base.py,sha256=mNR-aMM57D9PFaABVUvE9l5EFKq6QZ13yvQuAXeAo-w,9682
+cgir/languages/base.py,sha256=IaiKOGeucf_HziaSw5q8-ojYOBe47N6Dl2U1Z3FsEc8,10087
 cgir/languages/cache.py,sha256=_wcNLb5OmVBSuIXEIxEvVOT6GA5FzhXxWKOsC1sGCm0,2878
-cgir/languages/go.py,sha256=SOWOIMpkoLIsGpzMIn_IgwMhS51Ke6cxl-HPv5fhQ4Y,19818
-cgir/languages/python.py,sha256=tyjzbNtMuz5BiR4uObCxEIguDu64G9ys_I2VX-4LbUw,43392
+cgir/languages/go.py,sha256=U6n03hK_seaR5847xP-R1rCPCbJhGxUgcAYQBRuM6cQ,19960
+cgir/languages/python.py,sha256=tzgdCB_BQm5jbEMnfcoU4URyt6Ul-sUakd6Y-cDqJTU,44688
 cgir/languages/registry.py,sha256=N0ViSe5O3tP-n-FljrdP-TxwSAtdz6N-S5k2SgF628U,895
-cgir/languages/typescript.py,sha256=R6xH_engTOBevT2pkPqyh6-7Kkn8VSxbdZq9g2EAHWg,35293
+cgir/languages/typescript.py,sha256=MhtqQ4wsGzjjVT1FBYU6B-CWxbTgRLhNMitpKQJajmk,35713
 cgir/regenerate/__init__.py,sha256=tES3u6yn28zcxOaKUGli5iqCc0j5W501SS-zxvsy_a8,227
 cgir/regenerate/prompt_pack.py,sha256=wJs9x5Jn7ap09HHF4ULJ2yHiZ56YYhLaQ6uXZEjkDHc,584
 cgir/regenerate/regenerator.py,sha256=DBpufeJItrhBkxt4v45Ltnp8BnV1gAUshxwKa-U8fzA,3517
@@ -58,8 +58,8 @@ cgir/sources/joern_source.py,sha256=TeI5Uu7E3vDLLnCDVuHu1U_k1RW7lpdKAwZGzb5OuEc,
 cgir/sources/tree_sitter_source.py,sha256=LEgPihpMdwLJ4j3cB4Jsx3SXJfutTftUx1Ewzdbme8Q,9574
 cgir/trace/__init__.py,sha256=vMwb47f70-zpsupyVCC_cq6lEIe2k2Kn2W8Z87-30Hg,153
 cgir/trace/trace_map.py,sha256=x5k46_5CZYPe7PxXmFSKZrbhlg1rXK0_OBHsawE5xWM,2536
-codegraph_ir-0.2.0.dist-info/METADATA,sha256=g-5IH6JpjDC3uG5w6Jbm_bdrOuUL-Iw3PLUV23qsYrM,6707
-codegraph_ir-0.2.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-codegraph_ir-0.2.0.dist-info/entry_points.txt,sha256=-y9moAY3BkNj-4tRQW8JM_xLjioEIsgQFKRECvVZpj0,38
-codegraph_ir-0.2.0.dist-info/licenses/LICENSE,sha256=ZKZDuzSRXDx9ssBTaEKsx97f7j7w5L3LIvns0fVbMLo,1072
-codegraph_ir-0.2.0.dist-info/RECORD,,
+codegraph_ir-0.2.1.dist-info/METADATA,sha256=Xn_FrYrThQuo2Q5CnpofPEkvcL8Pp3ADKu64ifGAwac,6707
+codegraph_ir-0.2.1.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+codegraph_ir-0.2.1.dist-info/entry_points.txt,sha256=-y9moAY3BkNj-4tRQW8JM_xLjioEIsgQFKRECvVZpj0,38
+codegraph_ir-0.2.1.dist-info/licenses/LICENSE,sha256=ZKZDuzSRXDx9ssBTaEKsx97f7j7w5L3LIvns0fVbMLo,1072
+codegraph_ir-0.2.1.dist-info/RECORD,,