lscope 0.1.0__tar.gz → 0.1.2__tar.gz

{lscope-0.1.0 → lscope-0.1.2}/PKG-INFO

@@ -1,15 +1,14 @@
 Metadata-Version: 2.4
 Name: lscope
-Version: 0.1.0
+Version: 0.1.2
 Summary: ast-grep using ladybug
 Requires-Python: >=3.14
 Description-Content-Type: text/markdown
 Requires-Dist: icebug>=12.9
-Requires-Dist: ladybug>=0.17.1
-Requires-Dist: polars>=1.41.2
+Requires-Dist: ladybug>=0.18.0
 Requires-Dist: pyarrow>=24.0.0
 Requires-Dist: tqdm>=4.68.3
-Requires-Dist: tree-sitter>=0.25.2
+Requires-Dist: tree-sitter==0.25.2
 Requires-Dist: tree-sitter-c-sharp>=0.23.1
 Requires-Dist: tree-sitter-cpp>=0.23.4
 Requires-Dist: tree-sitter-go>=0.23.4

lscope-0.1.2/lscope/__init__.py

@@ -0,0 +1 @@
+# lscope — semantic code-graph extractor (tree-sitter → Ladybug)

lscope-0.1.2/lscope/__main__.py

@@ -0,0 +1,10 @@
+"""Allow running lscope as ``python -m lscope``."""
+import sys
+import traceback
+from lscope.main import main
+
+try:
+    sys.exit(main())
+except Exception:
+    traceback.print_exc()
+    sys.exit(1)

{lscope-0.1.0 → lscope-0.1.2/lscope}/main.py

@@ -15,6 +15,7 @@ non-sensical flag combinations are rejected up-front with a clear message.
 """
 
 import argparse
+import importlib.resources
 import os
 import sys
 import threading
@@ -259,7 +260,12 @@ _REQUIRED_REL_CONNECTIONS = (
 
 
 def _default_schema_path() -> str:
-    return os.path.join(os.path.dirname(os.path.abspath(__file__)), "schema.cypher")
+    try:
+        # When installed as a package, schema.cypher is bundled as package data.
+        return str(importlib.resources.files("lscope").joinpath("schema.cypher"))
+    except (ModuleNotFoundError, TypeError):
+        # Fallback when running directly (e.g. python lscope/main.py)
+        return os.path.join(os.path.dirname(os.path.abspath(__file__)), "schema.cypher")
 
 
 def _load_schema_text(schema_path: str | None) -> str | None:
@@ -293,8 +299,8 @@ def apply_schema(conn, schema_text: str | None) -> None:
 
 def ensure_schema(conn, schema_path: str | None = None) -> None:
     """Create the semantic schema when the database has no graph tables."""
-    tables = conn.execute("CALL show_tables() RETURN name").get_as_pl()
-    existing = set(tables["name"].to_list()) if tables.height else set()
+    tables = conn.execute("CALL show_tables() RETURN name").get_as_arrow()
+    existing = set(tables.column("name").to_pylist()) if tables.num_rows else set()
     if _REQUIRED_SCHEMA_TABLES <= existing:
         for source, target in _REQUIRED_REL_CONNECTIONS:
             if source in existing and target in existing:
@@ -754,7 +760,7 @@ def ingest_analyses_parallel(
 # --------------------------------------------------------------------------- #
 def _run_query(conn, query: str, params: dict | None = None):
     qr = conn.execute(query, params or {})
-    return qr.get_as_pl()
+    return qr.get_as_arrow()
 
 
 def find_functions(conn, pattern: str, languages: list[str] | None = None):
@@ -777,7 +783,7 @@ def find_functions(conn, pattern: str, languages: list[str] | None = None):
                 {"pattern": pattern},
             )
         )
-    return pl.concat(frames, how="vertical")
+    return pa.concat_tables(frames)
 
 
 def find_callers(conn, func_name: str):
@@ -803,7 +809,7 @@ def find_callers(conn, func_name: str):
                     {"func_name": func_name},
                 )
             )
-    return pl.concat(frames, how="vertical")
+    return pa.concat_tables(frames)
 
 
 # --------------------------------------------------------------------------- #
@@ -1036,16 +1042,23 @@ def main(argv: list[str] | None = None) -> int:
     args = parser.parse_args(argv)
     action = _resolve_actions(args)
 
-    if action == "index":
-        return run_index(args)
-    if action == "find-functions":
-        return run_find_functions(args)
-    if action == "find-callers":
-        return run_find_callers(args)
-    if action == "schema-only":
-        return run_schema_only(args)
-    parser.error(f"Unhandled action {action!r}")
-    return 2  # unreachable
+    try:
+        if action == "index":
+            return run_index(args)
+        if action == "find-functions":
+            return run_find_functions(args)
+        if action == "find-callers":
+            return run_find_callers(args)
+        if action == "schema-only":
+            return run_schema_only(args)
+        parser.error(f"Unhandled action {action!r}")
+        return 2  # unreachable
+    except SystemExit:
+        raise
+    except Exception:
+        import traceback
+        traceback.print_exc()
+        return 1
 
 
 if __name__ == "__main__":

lscope-0.1.2/lscope/schema.cypher

@@ -0,0 +1,329 @@
+CALL disable_default_hash_index=true;
+CALL debug_enable_multi_writes=true;
+
+// Semantic code-graph schema.
+//
+// Every relationship is stored in the CodeRelation relationship-table group.
+// The `type` property contains one of:
+//   CONTAINS, DEFINES, CALLS, IMPORTS, EXTENDS, IMPLEMENTS, HAS_METHOD,
+//   HAS_PROPERTY, ACCESSES, METHOD_OVERRIDES, METHOD_IMPLEMENTS, MEMBER_OF,
+//   STEP_IN_PROCESS, HANDLES_ROUTE, FETCHES, HANDLES_TOOL, ENTRY_POINT_OF.
+//
+// For ACCESSES, `reason` records the access mode (for example "read" or
+// "write"). For STEP_IN_PROCESS, `step` records the zero-based execution order.
+
+CREATE NODE TABLE File (
+    id          STRING,
+    name        STRING,
+    path        STRING,
+    filePath    STRING,
+    language    STRING,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Folder (
+    id          STRING,
+    name        STRING,
+    path        STRING,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Function (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    signature       STRING,
+    parameterCount  INT32,
+    returnType      STRING,
+    visibility      STRING,
+    isAsync         BOOLEAN,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Class (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    visibility      STRING,
+    isAbstract      BOOLEAN,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Interface (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    visibility      STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Method (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    signature       STRING,
+    parameterCount  INT32,
+    returnType      STRING,
+    visibility      STRING,
+    isAsync         BOOLEAN,
+    isStatic        BOOLEAN,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Property (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    declaredType    STRING,
+    visibility      STRING,
+    isStatic        BOOLEAN,
+    isMutable       BOOLEAN,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE CodeElement (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    kind            STRING,
+    filePath        STRING,
+    language        STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Community (
+    id              STRING,
+    heuristicLabel  STRING,
+    cohesion        DOUBLE,
+    symbolCount     INT32,
+    keywords        STRING[],
+    description     STRING,
+    enrichedBy      STRING,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Process (
+    id              STRING,
+    heuristicLabel  STRING,
+    processType     STRING,
+    stepCount       INT32,
+    communities     STRING[],
+    entryPointId    STRING,
+    terminalId      STRING,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Route (
+    id              STRING,
+    name            STRING,
+    path            STRING,
+    httpMethod      STRING,
+    filePath        STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Tool (
+    id              STRING,
+    name            STRING,
+    description     STRING,
+    toolType        STRING,
+    filePath        STRING,
+    PRIMARY KEY (id)
+);
+
+// Language-specific semantic node types.
+CREATE NODE TABLE Struct (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    visibility      STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Enum (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    visibility      STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Trait (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    visibility      STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE NODE TABLE Impl (
+    id              STRING,
+    name            STRING,
+    qualifiedName   STRING,
+    filePath        STRING,
+    language        STRING,
+    startLine       INT32,
+    endLine         INT32,
+    PRIMARY KEY (id)
+);
+
+CREATE REL TABLE GROUP CodeRelation (
+    // Filesystem containment and definitions.
+    FROM Folder TO Folder,
+    FROM Folder TO File,
+    FROM File TO Function,
+    FROM File TO Method,
+    FROM File TO Class,
+    FROM File TO Interface,
+    FROM File TO Property,
+    FROM File TO CodeElement,
+    FROM File TO Route,
+    FROM File TO Tool,
+    FROM File TO Struct,
+    FROM File TO Enum,
+    FROM File TO Trait,
+    FROM File TO Impl,
+
+    // Nested declarations and members.
+    FROM Class TO Class,
+    FROM Class TO Struct,
+    FROM Class TO Impl,
+    FROM Class TO Method,
+    FROM Class TO Function,
+    FROM Class TO Property,
+    FROM Interface TO Class,
+    FROM Interface TO Struct,
+    FROM Interface TO Trait,
+    FROM Interface TO Impl,
+    FROM Interface TO Method,
+    FROM Interface TO Property,
+    FROM Struct TO Class,
+    FROM Struct TO Struct,
+    FROM Struct TO Impl,
+    FROM Struct TO Method,
+    FROM Struct TO Property,
+    FROM Enum TO Method,
+    FROM Enum TO Property,
+    FROM Trait TO Class,
+    FROM Trait TO Struct,
+    FROM Trait TO Interface,
+    FROM Trait TO Impl,
+    FROM Trait TO Method,
+    FROM Trait TO Property,
+    FROM Impl TO Method,
+    FROM Impl TO Property,
+    FROM Impl TO Class,
+    FROM Impl TO Struct,
+    FROM Impl TO Enum,
+    FROM Impl TO Trait,
+    FROM Impl TO Interface,
+    FROM Impl TO Impl,
+    FROM Function TO Class,
+    FROM Function TO Interface,
+    FROM Function TO Struct,
+    FROM Function TO Trait,
+    FROM Function TO Impl,
+    FROM Method TO Class,
+    FROM Method TO Interface,
+    FROM Method TO Struct,
+    FROM Method TO Trait,
+    FROM Method TO Impl,
+
+    // Imports, inheritance, implementation, calls, and property access.
+    FROM File TO File,
+    FROM Function TO Function,
+    FROM Function TO Method,
+    FROM Function TO Property,
+    FROM Method TO Function,
+    FROM Method TO Method,
+    FROM Method TO Property,
+    FROM Class TO Interface,
+    FROM Class TO Trait,
+    FROM Struct TO Interface,
+    FROM Struct TO Trait,
+    FROM Enum TO Interface,
+    FROM Enum TO Trait,
+    FROM Interface TO Interface,
+    FROM Trait TO Trait,
+
+    // Communities and process traces.
+    FROM File TO Community,
+    FROM Function TO Community,
+    FROM Class TO Community,
+    FROM Interface TO Community,
+    FROM Method TO Community,
+    FROM Property TO Community,
+    FROM CodeElement TO Community,
+    FROM Route TO Community,
+    FROM Tool TO Community,
+    FROM Struct TO Community,
+    FROM Enum TO Community,
+    FROM Trait TO Community,
+    FROM Impl TO Community,
+    FROM File TO Process,
+    FROM Function TO Process,
+    FROM Class TO Process,
+    FROM Interface TO Process,
+    FROM Method TO Process,
+    FROM Property TO Process,
+    FROM CodeElement TO Process,
+    FROM Route TO Process,
+    FROM Tool TO Process,
+    FROM Struct TO Process,
+    FROM Enum TO Process,
+    FROM Trait TO Process,
+    FROM Impl TO Process,
+
+    // Framework routes, external fetches, and tool handlers.
+    FROM Function TO Route,
+    FROM Method TO Route,
+    FROM Class TO Route,
+    FROM Route TO Route,
+    FROM Function TO Tool,
+    FROM Method TO Tool,
+    FROM Class TO Tool,
+    FROM Tool TO Function,
+    FROM Tool TO Method,
+    FROM Tool TO Class,
+
+    type        STRING,
+    confidence  DOUBLE,
+    reason      STRING,
+    step        INT32
+);

{lscope-0.1.0 → lscope-0.1.2}/lscope.egg-info/PKG-INFO

@@ -1,15 +1,14 @@
 Metadata-Version: 2.4
 Name: lscope
-Version: 0.1.0
+Version: 0.1.2
 Summary: ast-grep using ladybug
 Requires-Python: >=3.14
 Description-Content-Type: text/markdown
 Requires-Dist: icebug>=12.9
-Requires-Dist: ladybug>=0.17.1
-Requires-Dist: polars>=1.41.2
+Requires-Dist: ladybug>=0.18.0
 Requires-Dist: pyarrow>=24.0.0
 Requires-Dist: tqdm>=4.68.3
-Requires-Dist: tree-sitter>=0.25.2
+Requires-Dist: tree-sitter==0.25.2
 Requires-Dist: tree-sitter-c-sharp>=0.23.1
 Requires-Dist: tree-sitter-cpp>=0.23.4
 Requires-Dist: tree-sitter-go>=0.23.4

{lscope-0.1.0 → lscope-0.1.2}/lscope.egg-info/SOURCES.txt

@@ -1,6 +1,9 @@
 README.md
-main.py
 pyproject.toml
+lscope/__init__.py
+lscope/__main__.py
+lscope/main.py
+lscope/schema.cypher
 lscope.egg-info/PKG-INFO
 lscope.egg-info/SOURCES.txt
 lscope.egg-info/dependency_links.txt

lscope-0.1.2/lscope.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+lscope = lscope.main:main

{lscope-0.1.0 → lscope-0.1.2}/lscope.egg-info/requires.txt

@@ -1,9 +1,8 @@
 icebug>=12.9
-ladybug>=0.17.1
-polars>=1.41.2
+ladybug>=0.18.0
 pyarrow>=24.0.0
 tqdm>=4.68.3
-tree-sitter>=0.25.2
+tree-sitter==0.25.2
 tree-sitter-c-sharp>=0.23.1
 tree-sitter-cpp>=0.23.4
 tree-sitter-go>=0.23.4

lscope-0.1.2/lscope.egg-info/top_level.txt

@@ -0,0 +1 @@
+lscope

{lscope-0.1.0 → lscope-0.1.2}/pyproject.toml

@@ -1,16 +1,15 @@
 [project]
 name = "lscope"
-version = "0.1.0"
+version = "0.1.2"
 description = "ast-grep using ladybug"
 readme = "README.md"
 requires-python = ">=3.14"
 dependencies = [
     "icebug>=12.9",
-    "ladybug>=0.17.1",
-    "polars>=1.41.2",
+    "ladybug>=0.18.0",
     "pyarrow>=24.0.0",
     "tqdm>=4.68.3",
-    "tree-sitter>=0.25.2",
+    "tree-sitter==0.25.2",
     "tree-sitter-c-sharp>=0.23.1",
     "tree-sitter-cpp>=0.23.4",
     "tree-sitter-go>=0.23.4",
@@ -24,10 +23,11 @@ dependencies = [
 ]
 
 [project.scripts]
-lscope = "main:main"
+lscope = "lscope.main:main"
 
 [tool.setuptools]
-py-modules = ["main"]
+packages = ["lscope"]
+package-data = {"lscope" = ["*.cypher"]}
 
 [dependency-groups]
 dev = [

{lscope-0.1.0 → lscope-0.1.2}/test/test_main.py

@@ -1,6 +1,12 @@
 import ladybug
 
-from main import LanguageRegistry, analyze_file, ensure_schema, ingest_analysis
+from lscope.main import (
+    LanguageRegistry,
+    _ingest_chunk_edges,
+    analyze_file,
+    ensure_schema,
+    ingest_analyses_parallel,
+)
 
 
 NESTED_TYPES = """\
@@ -32,7 +38,9 @@ def test_ingests_classes_nested_in_functions_and_methods(tmp_path):
         ensure_schema(conn)
         analysis = _analyze_nested_types()
 
-        assert ingest_analysis(conn, analysis) == 6
+        total_nodes, def_edges = ingest_analyses_parallel(db, [analysis], 1)
+        _ingest_chunk_edges(conn, def_edges)
+        assert total_nodes == 6
         rows = conn.execute(
             """
             MATCH (owner)-[r:CodeRelation]->(nested:Class)
@@ -40,8 +48,8 @@ def test_ingests_classes_nested_in_functions_and_methods(tmp_path):
             RETURN label(owner) AS owner_label, nested.name AS nested_name
             ORDER BY nested_name
             """
-        ).get_as_pl()
-        assert rows.to_dicts() == [
+        ).get_as_arrow()
+        assert [dict(zip(rows.column_names, row)) for row in zip(*(rows.column(c).to_pylist() for c in rows.column_names))] == [
             {"owner_label": "Function", "nested_name": "Inner"},
             {"owner_label": "Method", "nested_name": "Local"},
         ]
@@ -62,7 +70,11 @@ def test_ensure_schema_migrates_existing_relation_group(tmp_path):
 
         ensure_schema(conn)
 
-        assert ingest_analysis(conn, _analyze_nested_types()) == 6
+        total_nodes, def_edges = ingest_analyses_parallel(
+            db, [_analyze_nested_types()], 1
+        )
+        _ingest_chunk_edges(conn, def_edges)
+        assert total_nodes == 6
     finally:
         conn.close()
         db.close()

lscope-0.1.0/lscope.egg-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-lscope = main:main

lscope-0.1.0/lscope.egg-info/top_level.txt

@@ -1 +0,0 @@
-main