🧑‍💻 Add get_uri()

neomutt · Feb 14, 2024 · d7f7446 · d7f7446
1 parent 040cfeb
commit d7f7446
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/tree_sitter_lsp/__init__.py b/src/tree_sitter_lsp/__init__.py
@@ -90,6 +90,13 @@ def get_path(self) -> str:
         """
         return self.uri2path(self.uri)
 
+    def get_uri(self) -> str:
+        r"""Get uri.
+
+        :rtype: str
+        """
+        return os.path.join(os.path.dirname(self.uri), self.get_text())
+
     @staticmethod
     def uri2path(uri: str) -> str:
         r"""Uri2path.
@@ -133,7 +140,9 @@ def get_text_edit(self, new_text: str) -> TextEdit:
         """
         return TextEdit(self.get_range(), new_text)
 
-    def get_document_link(self, target: str, **kwargs) -> DocumentLink:
+    def get_document_link(
+        self, target: str = "{{uni.get_uri()}}", **kwargs
+    ) -> DocumentLink:
         r"""Get document link.
 
         :param target:
@@ -423,7 +432,7 @@ def get_text_edits(self, uri: str, tree: Tree) -> list[TextEdit]:
         return []
 
     def get_document_links(
-        self, uri: str, tree: Tree, template: str
+        self, uri: str, tree: Tree, template: str = "{{uni.get_uri()}}"
     ) -> list[DocumentLink]:
         r"""Get document links.