feat: add language enum

Author: nohehf

.github/workflows/CI.yml

@@ -35,7 +35,7 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.11'
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
@@ -62,7 +62,7 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.11'
           architecture: ${{ matrix.platform.target }}
       - name: Build wheels
         uses: PyO3/maturin-action@v1
@@ -89,7 +89,7 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.11'
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:

Makefile

@@ -0,0 +1,8 @@
+.PHONY: develop
+develop:
+	maturin develop
+
+
+test: develop
+    ## TODO: Add actual tests with pytest
+	python tests/test.py

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "stack-graphs-python-bindings"
-requires-python = ">=3.8"
+requires-python = ">=3.11"
 classifiers = [
     "Programming Language :: Rust",
     "Programming Language :: Python :: Implementation :: CPython",

src/classes.rs

@@ -1,6 +1,7 @@
 use std::fmt::Display;
 
 use pyo3::prelude::*;
+
 use tree_sitter_stack_graphs::cli::util::{SourcePosition, SourceSpan};
 
 #[pyclass]

stack_graphs_python.pyi

@@ -1,4 +1,10 @@
-def index(paths: list[str], db_path: str) -> None: ...
+from enum import Enum
+
+class Language(Enum):
+    Python = 0
+    JavaScript = 1
+    TypeScript = 2
+    Java = 3
 
 class Position:
     path: str
@@ -7,4 +13,5 @@ class Position:
 
     def __init__(self, path: str, line: int, column: int) -> None: ...
 
+def index(paths: list[str], db_path: str, language: Language) -> None: ...
 def query_definition(reference: Position, db_path: str) -> list[Position]: ...

tests/test.py

@@ -1,5 +1,5 @@
 import os
-from stack_graphs_python import index, query_definition, Position
+from stack_graphs_python import index, query_definition, Position, Language
 
 # index ./js_sample directory
 
@@ -10,7 +10,7 @@
 print("Indexing directory: ", dir)
 print("Database path: ", db)
 
-index([dir], db)
+index([dir], db, language=Language.Python)
 
 source_reference: Position = Position(path=dir + "/index.js", line=2, column=12)
 
@@ -25,3 +25,8 @@
     print("Line: ", result.line)
     print("Column: ", result.column)
     print("\n")
+
+assert len(results) == 2
+assert results[0].path.endswith("index.js")
+assert results[0].line == 0
+assert results[0].column == 9