Add completions for Gentoo ebuilds in dependency variables

Author: robert7k

README.md

@@ -85,7 +85,7 @@ doesn't support:
   - [x] package names
     - [ ] `build.sh`
     - [x] `PKGBUILD`: by [pyalpm](https://github.com/ornitorrincos/pyalpm)
-    - [ ] `ebuild`
+    - [x] `ebuild`: by [portage](https://wiki.gentoo.org/wiki/Portage)
 - [ ] [Code Action](https://microsoft.github.io/language-server-protocol/specifications/specification-current#textDocument_codeAction)
   - [ ] `PKGBUILD`
     - [ ] generate a template by the name of directory containing `PKGBUILD`, the

pyproject.toml

@@ -82,6 +82,9 @@ file = "requirements/misc.txt"
 
 [tool.setuptools.dynamic.optional-dependencies.pkgbuild]
 file = "requirements/pkgbuild.txt"
+
+[tool.setuptools.dynamic.optional-dependencies.ebuild]
+file = "requirements/ebuild.txt"
 # end: scripts/update-pyproject.toml.pl
 
 [tool.setuptools_scm]

src/termux_language_server/assets/jinja2/ebuild.md.j2

@@ -0,0 +1,7 @@
+{{ pkg.description }}
+
+- version: **{{ pkg.version }}**
+- slot: **{{ pkg.slot }}**
+- homepage: <{{ pkg.homepage }}>
+- license: *{{ pkg.license }}*
+- keywords: {{ pkg.keywords }}

src/termux_language_server/assets/queries/package.scm

@@ -4,6 +4,6 @@
    value: (array (word) @package)
    )
  (#match?
-  @variable.name "^(depends|optdepends|makedepends|conflicts|provides)$"
+  @variable.name "^(depends|optdepends|makedepends|conflicts|provides|DEPEND|RDEPEND|BDEPEND|IDEPEND|PDEPEND)$"
   )
  )

src/termux_language_server/packages/__init__.py

@@ -12,7 +12,14 @@
         "conflicts",
         "provides",
         "replaces",
-    }
+    },
+    "ebuild": {
+        "DEPEND",
+        "RDEPEND",
+        "BDEPEND",
+        "IDEPEND",
+        "PDEPEND",
+    },
 }
 
 
@@ -27,6 +34,8 @@ def search_package_document(name: str, filetype: FILETYPE) -> str:
     """
     if filetype == "PKGBUILD":
         from .pkgbuild import get_package_document
+    elif filetype == "ebuild":
+        from .ebuild import get_package_document
     else:
         raise NotImplementedError
     return get_package_document(name)
@@ -43,6 +52,8 @@ def search_package_names(name: str, filetype: FILETYPE) -> dict[str, str]:
     """
     if filetype == "PKGBUILD":
         from .pkgbuild import get_package_names
+    elif filetype == "ebuild":
+        from .ebuild import get_package_names
     else:
         raise NotImplementedError
     return get_package_names(name)

src/termux_language_server/server.py

@@ -27,7 +27,9 @@
     Hover,
     MarkupContent,
     MarkupKind,
+    Position,
     PublishDiagnosticsParams,
+    Range,
     TextDocumentPositionParams,
     TextEdit,
 )
@@ -48,6 +50,64 @@
 from .utils import get_filetype, get_schema, parser
 
 
+RE_PKG_START = re.compile(r"[A-Za-z_0-9/\-\.>=<!~*]*$")
+RE_PKG_END = re.compile(r"^[A-Za-z_0-9/\-\.>=<!~*]*")
+RE_EMPTY = re.compile(r"^")
+
+
+def _is_in_dep_string(tree, position, filetype):
+    r"""Check if cursor is inside a dependency variable's string.
+
+    :param tree:
+    :param position:
+    :type position: Position
+    :param filetype:
+    :type filetype: str
+    :rtype: bool
+    """
+    point = (position.line, position.character)
+    node = tree.root_node.descendant_for_point_range(point, point)
+    return (
+        node is not None
+        and node.type == "string"
+        and node.parent is not None
+        and node.parent.children[0].text is not None
+        and node.parent.children[0].text.decode()
+        in PACKAGE_VARIABLES.get(filetype, set())
+    )
+
+
+def _package_completions(prefix, filetype, position):
+    r"""Build completion list for package names.
+
+    :param prefix:
+    :type prefix: str
+    :param filetype:
+    :type filetype: str
+    :param position:
+    :type position: Position
+    :rtype: CompletionList
+    """
+    edit_range = Range(
+        start=Position(position.line, position.character - len(prefix)),
+        end=position,
+    )
+    return CompletionList(
+        False,
+        [
+            CompletionItem(
+                k,
+                kind=CompletionItemKind.Module,
+                documentation=MarkupContent(MarkupKind.Markdown, v)
+                if v
+                else None,
+                text_edit=TextEdit(range=edit_range, new_text=k),
+            )
+            for k, v in search_package_names(prefix, filetype).items()
+        ],
+    )
+
+
 class TermuxLanguageServer(LanguageServer):
     r"""Termux language server."""
 
@@ -190,13 +250,20 @@ def hover(params: TextDocumentPositionParams) -> Hover | None:
                 }
             ):
                 if (
-                    parent.type == "array"
+                    parent.type in {"array", "string"}
                     and parent.parent is not None
                     and parent.parent.children[0].text is not None
                     and parent.parent.children[0].text.decode()
                     in PACKAGE_VARIABLES.get(filetype, set())
                 ):
-                    result = search_package_document(text, filetype)
+                    result = search_package_document(
+                        document.word_at_position(
+                            params.position, RE_PKG_START, RE_PKG_END
+                        )
+                        if parent.type == "string"
+                        else text,
+                        filetype,
+                    )
                     if result is None:
                         return None
                     return Hover(
@@ -239,34 +306,40 @@ def completions(params: CompletionParams) -> CompletionList:
             uni = PositionFinder(params.position, right_equal=True).find(
                 document.uri, self.trees[document.uri]
             )
-            if uni is None:
-                return CompletionList(False, [])
+            if uni is None or uni.node.type == '"':
+                if _is_in_dep_string(
+                    self.trees[document.uri],
+                    params.position,
+                    filetype,
+                ):
+                    return _package_completions(
+                        document.word_at_position(
+                            params.position, RE_PKG_START, RE_EMPTY
+                        ),
+                        filetype,
+                        params.position,
+                    )
+                if uni is None:
+                    return CompletionList(False, [])
             parent = uni.node.parent
             if parent is None:
                 return CompletionList(False, [])
             text = uni.text
             if (
-                parent.type == "array"
+                parent.type in {"array", "string"}
                 and parent.parent is not None
                 and parent.parent.children[0].text is not None
                 and parent.parent.children[0].text.decode()
                 in PACKAGE_VARIABLES.get(filetype, set())
             ):
-                return CompletionList(
-                    False,
-                    [
-                        CompletionItem(
-                            k,
-                            kind=CompletionItemKind.Module,
-                            documentation=MarkupContent(
-                                MarkupKind.Markdown, v
-                            ),
-                            insert_text=k,
-                        )
-                        for k, v in search_package_names(
-                            text, filetype
-                        ).items()
-                    ],
+                return _package_completions(
+                    document.word_at_position(
+                        params.position, RE_PKG_START, RE_EMPTY
+                    )
+                    if parent.type == "string"
+                    else text,
+                    filetype,
+                    params.position,
                 )
             schema = get_schema(filetype)
             if (