👽 Update APIs of tree-sitter

Freed-Wu · Mar 27, 2024 · ddcc0df · ddcc0df
1 parent 7505b48
commit ddcc0df
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/src/tmux_language_server/__main__.py b/src/tmux_language_server/__main__.py
@@ -76,10 +76,10 @@ def main() -> None:
     if args.generate_schema or args.check or args.convert:
         from lsp_tree_sitter.diagnose import check
         from lsp_tree_sitter.utils import pprint
-        from tree_sitter_tmux import parser
 
         from .finders import DIAGNOSTICS_FINDER_CLASSES
         from .schema import TmuxTrie
+        from .utils import parser
 
         if args.generate_schema:
             from .misc import get_schema

diff --git a/src/tmux_language_server/server.py b/src/tmux_language_server/server.py
@@ -26,10 +26,9 @@
     TextDocumentPositionParams,
 )
 from pygls.server import LanguageServer
-from tree_sitter_tmux import parser
 
 from .finders import DIAGNOSTICS_FINDER_CLASSES, ImportTmuxFinder
-from .utils import get_schema
+from .utils import get_schema, parser
 
 
 class TmuxLanguageServer(LanguageServer):

diff --git a/src/tmux_language_server/utils.py b/src/tmux_language_server/utils.py
@@ -6,12 +6,16 @@
 import os
 from typing import Any
 
-from tree_sitter import Query
+from tree_sitter import Language, Parser, Query
+from tree_sitter_tmux import language as get_language_ptr
 
 from . import FILETYPE
 
 SCHEMAS = {}
 QUERIES = {}
+language = Language(get_language_ptr(), "tmux")
+parser = Parser()
+parser.set_language(language)
 
 
 def get_query(name: str, filetype: FILETYPE = "tmux") -> Query:
@@ -33,8 +37,6 @@ def get_query(name: str, filetype: FILETYPE = "tmux") -> Query:
             )
         ) as f:
             text = f.read()
-        from tree_sitter_tmux import language
-
         QUERIES[name] = language.query(text)
     return QUERIES[name]