codegraph-ir 0.1.0__py3-none-any.whl

cgir/config.py

@@ -0,0 +1,22 @@
+"""Runtime configuration."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from pydantic import BaseModel, Field
+
+
+class CGIRConfig(BaseModel):
+    repo_path: Path
+    out_dir: Path = Field(default_factory=lambda: Path(".cgir"))
+    target_languages: list[str] = Field(default_factory=lambda: ["python"])
+    source_backend: str = "tree-sitter"
+
+    @classmethod
+    def for_scan(cls, repo_path: Path, out_dir: Path | None = None) -> CGIRConfig:
+        repo_path = repo_path.resolve()
+        return cls(
+            repo_path=repo_path,
+            out_dir=(out_dir or repo_path / ".cgir").resolve(),
+        )

cgir/export/__init__.py

@@ -0,0 +1,5 @@
+"""Export passes: json (working), graphml (P2 stub), neo4j (P2 stub)."""
+
+from cgir.export.json_export import write_index
+
+__all__ = ["write_index"]

cgir/export/graphml.py

@@ -0,0 +1,52 @@
+"""GraphML export — opens the RepoGraph in Gephi, yEd, or Cytoscape.
+
+GraphML attribute values must be scalars, so list/dict attrs are
+JSON-encoded strings and ``None`` attrs are dropped. ``kind`` carries the
+Node/Edge enum value.
+"""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+import networkx as nx
+
+from cgir.ir.graph import RepoGraph
+
+
+def write(out_dir: Path, graph: RepoGraph) -> Path:
+    """Write ``<out_dir>/repo_graph.graphml`` and return its path."""
+    out_dir.mkdir(parents=True, exist_ok=True)
+    flat: nx.MultiDiGraph = nx.MultiDiGraph()
+    for node in graph.nodes():
+        attrs: dict[str, str | int | float | bool] = {
+            "kind": node.kind.value,
+            "name": node.name,
+        }
+        if node.path is not None:
+            attrs["path"] = node.path
+        if node.start_line is not None:
+            attrs["start_line"] = node.start_line
+        if node.end_line is not None:
+            attrs["end_line"] = node.end_line
+        for key, value in node.attrs.items():
+            scalar = _to_scalar(value)
+            if scalar is not None:
+                attrs[key] = scalar
+        flat.add_node(node.id, **attrs)
+    for node in graph.nodes():
+        for edge in graph.out_edges(node.id):
+            flat.add_edge(edge.src, edge.dst, kind=edge.kind.value)
+
+    path = out_dir / "repo_graph.graphml"
+    nx.write_graphml(flat, path)
+    return path
+
+
+def _to_scalar(value: object) -> str | int | float | bool | None:
+    if value is None:
+        return None
+    if isinstance(value, str | int | float | bool):
+        return value
+    return json.dumps(value)