Example So File Usage

from tree_sitter import Language

from albero import Token, TreeSitterHighlighter, get_mapping, lang_from_so

# NOTE: This will only work on MacOS as the so file was compiled for it
py_lang: Language = lang_from_so("./examples/languages-darwin.so", "python")
mapping: dict[str, str] = get_mapping("python")


highlighter = TreeSitterHighlighter()

highlighter.add_language("python", py_lang, mapping)
highlighter.add_file(
    "test", "python"
)  # Trying to add a file with a language not in the system gives an AlberoException
highlighter.update_file(
    "test", "def test(): ..."
)  # Trying to update a file not in the system gives an AlberoException

tokens: list[Token] = highlighter.get_highlights(
    "test"
)  # Trying to use a file not in the system gives an AlberoException

highlighter.remove_file("test")
highlighter.remove_language(
    "python"
)  # Can be done first but it will already auto-close any files that use it

print(tokens)

See the file example file here.