⚡ Use tree-sitter-query

requirements/languages.txt

@@ -0,0 +1,3 @@
+#!/usr/bin/env -S pip install -r
+
+tree-sitter-languages

src/tree_sitter_lsp/assets/queries/error.scm

@@ -0,0 +1,2 @@
+(ERROR) @error
+; ex: filetype=query

src/tree_sitter_lsp/finders.py

@@ -17,7 +17,7 @@
     Range,
     TextEdit,
 )
-from tree_sitter import Node, Tree
+from tree_sitter import Language, Node, Tree
 from tree_sitter.binding import Query
 
 from . import UNI, Finder
@@ -44,26 +44,6 @@ def __call__(self, uni: UNI) -> bool:
         )
 
 
-@dataclass
-class ErrorFinder(Finder):
-    r"""Errorfinder."""
-
-    message: str = "{{uni.get_text()}}: error"
-    severity: DiagnosticSeverity = DiagnosticSeverity.Error
-
-    def __call__(self, uni: UNI) -> bool:
-        r"""Call.
-
-        :param uni:
-        :type uni: UNI
-        :rtype: bool
-        """
-        node = uni.node
-        return node.has_error and not (
-            any(child.has_error for child in node.children)
-        )
-
-
 @dataclass
 class NotFileFinder(Finder):
     r"""NotFilefinder."""
@@ -619,3 +599,78 @@ def captures2uni(self, capture: tuple[Node, str], uri: str) -> UNI:
         :rtype: UNI
         """
         return UNI(uri, capture[0])
+
+
+@dataclass(init=False)
+class ErrorFinder(QueryFinder):
+    r"""Errorfinder."""
+
+    def __init__(
+        self,
+        language: Language | None = None,
+        message: str = "{{uni.get_text()}}: error",
+        severity: DiagnosticSeverity = DiagnosticSeverity.Error,
+    ):
+        r"""Init.
+
+        :param language:
+        :type language: Language | None
+        :param message:
+        :type message: str
+        :param severity:
+        :type severity: DiagnosticSeverity
+        """
+        with open(
+            os.path.join(
+                os.path.join(
+                    os.path.join(os.path.dirname(__file__), "assets"),
+                    "queries",
+                ),
+                "error.scm",
+            )
+        ) as f:
+            text = f.read()
+        if language is not None:
+            query = language.query(text)
+            super().__init__(query, message, severity)
+        else:
+            Finder.__init__(self, message, severity)
+            self.find_all = Finder().find_all
+
+    def __call__(self, uni: UNI) -> bool:
+        r"""Call.
+
+        :param uni:
+        :type uni: UNI
+        :rtype: bool
+        """
+        node = uni.node
+        return node.has_error and not (
+            any(child.has_error for child in node.children)
+        )
+
+
+@dataclass(init=False)
+class ErrorQueryFinder(ErrorFinder):
+    r"""Errorqueryfinder."""
+
+    def __init__(
+        self,
+        filetype: str,
+        message: str = "{{uni.get_text()}}: error",
+        severity: DiagnosticSeverity = DiagnosticSeverity.Error,
+    ) -> None:
+        r"""Init.
+
+        :param filetype:
+        :type filetype: str
+        :param message:
+        :type message: str
+        :param severity:
+        :type severity: DiagnosticSeverity
+        :rtype: None
+        """
+        from tree_sitter_languages import get_language
+
+        language = get_language(filetype)
+        super().__init__(language, message, severity)

src/tree_sitter_lsp/utils.py

@@ -45,9 +45,9 @@ def get_finders(
     :rtype: tuple[list[Finder], list[type[Finder]]]
     """
     if classes is None:
-        from .finders import ErrorFinder, MissingFinder
+        from .finders import ErrorFinder
 
-        classes = [ErrorFinder, MissingFinder]
+        classes = [ErrorFinder]
 
     finders = []
     finder_classes = []