✨ Support document_link

requirements.txt

@@ -2,7 +2,7 @@
 
 platformdirs
 pygls
-# get_completion_list_by_enum()
-tree-sitter-lsp >= 0.0.9
+# UNI.get_uri()
+tree-sitter-lsp >= 0.0.11
 tree-sitter-zathurarc >= 0.0.3
 webcolors

src/zathura_language_server/assets/queries/import.scm

@@ -0,0 +1 @@
+(path) @path

src/zathura_language_server/finders.py

@@ -2,10 +2,11 @@
 ===========
 """
 
+import os
 from dataclasses import dataclass
 
 from lsprotocol.types import DiagnosticSeverity
-from tree_sitter_lsp.finders import ErrorFinder, SchemaFinder
+from tree_sitter_lsp.finders import ErrorFinder, QueryFinder, SchemaFinder
 from tree_sitter_zathurarc import language
 
 from .schema import ZathurarcTrie
@@ -34,6 +35,36 @@ def __init__(
         super().__init__(language, message, severity)
 
 
+@dataclass(init=False)
+class ImportZathurarcFinder(QueryFinder):
+    r"""ImportZathurarcFinder."""
+
+    def __init__(
+        self,
+        message: str = "{{uni.get_text()}}: error",
+        severity: DiagnosticSeverity = DiagnosticSeverity.Information,
+    ):
+        r"""Init.
+
+        :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",
+                ),
+                "import.scm",
+            )
+        ) as f:
+            text = f.read()
+        query = language.query(text)
+        super().__init__(query, message, severity)
+
+
 @dataclass(init=False)
 class ZathurarcFinder(SchemaFinder):
     r"""Zathurarcfinder."""

src/zathura_language_server/server.py

@@ -8,12 +8,15 @@
     TEXT_DOCUMENT_COMPLETION,
     TEXT_DOCUMENT_DID_CHANGE,
     TEXT_DOCUMENT_DID_OPEN,
+    TEXT_DOCUMENT_DOCUMENT_LINK,
     TEXT_DOCUMENT_HOVER,
     CompletionItem,
     CompletionItemKind,
     CompletionList,
     CompletionParams,
     DidChangeTextDocumentParams,
+    DocumentLink,
+    DocumentLinkParams,
     Hover,
     MarkupContent,
     MarkupKind,
@@ -28,7 +31,7 @@
 from tree_sitter_lsp.finders import PositionFinder
 from tree_sitter_zathurarc import parser
 
-from .finders import DIAGNOSTICS_FINDER_CLASSES
+from .finders import DIAGNOSTICS_FINDER_CLASSES, ImportZathurarcFinder
 from .utils import get_schema
 
 
@@ -64,6 +67,19 @@ def did_change(params: DidChangeTextDocumentParams) -> None:
             )
             self.publish_diagnostics(params.text_document.uri, diagnostics)
 
+        @self.feature(TEXT_DOCUMENT_DOCUMENT_LINK)
+        def document_link(params: DocumentLinkParams) -> list[DocumentLink]:
+            r"""Get document links.
+
+            :param params:
+            :type params: DocumentLinkParams
+            :rtype: list[DocumentLink]
+            """
+            document = self.workspace.get_document(params.text_document.uri)
+            return ImportZathurarcFinder().get_document_links(
+                document.uri, self.trees[document.uri]
+            )
+
         @self.feature(TEXT_DOCUMENT_HOVER)
         def hover(params: TextDocumentPositionParams) -> Hover | None:
             r"""Hover.

tests/test_utils.py

@@ -1,5 +1,9 @@
 r"""Test utils."""
 
+import os
+
+from tree_sitter_zathurarc import parser
+from zathura_language_server.finders import ImportZathurarcFinder
 from zathura_language_server.utils import get_schema
 
 
@@ -19,3 +23,12 @@ def test_get_schema() -> None:
             .get("description", "")
             .splitlines()
         )
+
+    @staticmethod
+    def test_ImportZathurarcFinder() -> None:
+        with open(
+            os.path.join(os.path.dirname(__file__), "zathurarc"), "rb"
+        ) as f:
+            text = f.read()
+        tree = parser.parse(text)
+        assert ImportZathurarcFinder().get_document_links("", tree)