🚸 Support AUR

Author: Freed-Wu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "termux-language-server"
-version = "0.1.8"
+version = "0.1.9"
 description = "language server for PKGBUILD/build.sh/ebuild"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -31,6 +31,7 @@ pkgbuild = [
   # "git+https://gitlab.archlinux.org/pacman/namcap",
   "pyalpm>=0.10.0",
   "jinja2>=3",
+  "marisa-trie>=1",
   "platformdirs>=4",
 ]
 

src/termux_language_server/__init__.py

@@ -2,4 +2,4 @@
 `importlib.metadata.version() <https://docs.python.org/3/library/importlib.metadata.html#distribution-versions>`_.
 """
 
-__version__ = "0.1.8"
+__version__ = "0.1.9"

src/termux_language_server/queries/packages.scm

@@ -10,6 +10,13 @@
     (raw_string) @package.PKGBUILD)
   (#any-of? @_name "depends" "makedepends" "optdepends" "conflicts" "provides" "replaces"))
 
+(variable_assignment
+  name: (variable_name) @_name
+  value: (array
+    (string
+      (string_content) @package.PKGBUILD))
+  (#any-of? @_name "depends" "makedepends" "optdepends" "conflicts" "provides" "replaces"))
+
 (variable_assignment
   name: (variable_name) @_name
   value: (string

src/termux_language_server/searcher/pacman.py

@@ -2,12 +2,14 @@
 ==========
 """
 
+import os
 from dataclasses import dataclass, field
 from pathlib import Path
 
 from jinja2 import Template
 from lsp_tree_sitter.completer import PackageSearcher
-from platformdirs import user_config_path
+from marisa_trie import Trie
+from platformdirs import user_cache_dir, user_config_path
 from pyalpm import DB, Handle, Package
 from tree_sitter import Node
 
@@ -19,6 +21,15 @@ def get_template(name: str = "PKGBUILD.md.jinja") -> Template:
     return Template(path.read_text())
 
 
+def get_trie() -> Trie | None:
+    path = os.path.join(user_cache_dir("paru"), "packages.aur")
+    if not os.path.exists(path):
+        return
+    with open(path) as f:
+        lines = f.readlines()
+    return Trie(lines)
+
+
 @dataclass
 class PacmanSearcher(PackageSearcher):
     label: str = "package.PKGBUILD"
@@ -30,11 +41,11 @@ class PacmanSearcher(PackageSearcher):
         "provides",
         "replaces",
     )
-    url_template: str = "https://archlinux.org/packages/{}"
     template: Template = field(default_factory=get_template)
     db: DB = field(
         default_factory=lambda: Handle(".", "/var/lib/pacman").get_localdb()
     )
+    trie: Trie | None = field(default_factory=get_trie)
 
     def __call__(self, node: Node | None) -> bool:
         node = node.parent if node and node.type == "string_content" else node
@@ -52,26 +63,41 @@ def get_pkgs(self, name: str) -> list[Package]:
         return pkgs
 
     def has_package(self, name: str) -> bool:
-        return self.get_pkgs(name) != []
+        if self.get_pkgs(name):
+            return True
+        if self.trie:
+            return name in self.trie.keys(name)
+        return False
 
     def get_package_url(self, name: str) -> str:
-        return self.url_template.format(name)
+        if self.get_pkgs(name):
+            return f"https://archlinux.org/packages/{name}"
+        return f"https://aur.archlinux.org/packages/{name}"
 
     def get_package_version(self, name: str) -> str:
+        if not self.get_pkgs(name):
+            return ""
         pkg = self.get_pkgs(name)[0]
         version = pkg.version
         if pkg.name != name:
             version = pkg.name + " " + version
         return version
 
     def get_package_names(self, name: str) -> dict[str, str]:
-        return {
+        names = {
             pkg.name: self.template.render(pkg=pkg)
             for pkg in self.db.search(name)
             if pkg.name.startswith(name)
         }
+        if self.trie:
+            for pkg in self.trie.keys(name):
+                if pkg not in names:
+                    names[pkg] = ""
+        return names
 
     def get_package_document(self, name: str) -> str:
+        if not self.get_pkgs(name):
+            return ""
         docs = []
         for pkg in self.get_pkgs(name):
             docs += [self.template.render(pkg=pkg)]