From 8cc05adcf427f8730718398a2aba71430536f539 Mon Sep 17 00:00:00 2001
From: Raoul Wols <r@primef.actor>
Date: Thu, 26 Oct 2023 07:06:42 +0200
Subject: [PATCH] Reply with MarkupContent in completion documentation (#87)

---
 cmake_language_server/server.py | 41 ++++++++++++++++++++-------------
 tests/test_server.py            |  5 ++--
 2 files changed, 28 insertions(+), 18 deletions(-)

diff --git a/cmake_language_server/server.py b/cmake_language_server/server.py
index c54fa77..5f25176 100644
--- a/cmake_language_server/server.py
+++ b/cmake_language_server/server.py
@@ -92,7 +92,7 @@ class CMakeLanguageServer(LanguageServer):
                     CompletionItem(
                         label=x,
                         kind=CompletionItemKind.Function,
-                        documentation=self._api.get_command_doc(x),
+                        documentation=self._get_command_doc(x),
                         insert_text=x,
                     )
                     for x in commands
@@ -104,7 +104,7 @@ class CMakeLanguageServer(LanguageServer):
                     CompletionItem(
                         label=x,
                         kind=CompletionItemKind.Variable,
-                        documentation=self._api.get_variable_doc(x),
+                        documentation=self._get_variable_doc(x),
                         insert_text=x,
                     )
                     for x in variables
@@ -129,7 +129,7 @@ class CMakeLanguageServer(LanguageServer):
                             CompletionItem(
                                 label=x,
                                 kind=CompletionItemKind.Module,
-                                documentation=self._api.get_module_doc(x, False),
+                                documentation=self._get_module_doc(x, False),
                                 insert_text=x,
                             )
                             for x in modules
@@ -140,7 +140,7 @@ class CMakeLanguageServer(LanguageServer):
                             CompletionItem(
                                 label=x,
                                 kind=CompletionItemKind.Module,
-                                documentation=self._api.get_module_doc(x, True),
+                                documentation=self._get_module_doc(x, True),
                                 insert_text=x,
                             )
                             for x in modules
@@ -175,27 +175,21 @@ class CMakeLanguageServer(LanguageServer):
 
         @self.feature(TEXT_DOCUMENT_HOVER)
         def hover(params: TextDocumentPositionParams) -> Optional[Hover]:
-            assert self._api is not None
-            api = self._api
-
             word = self._cursor_word(params.text_document.uri, params.position, True)
             if not word:
                 return None
 
-            candidates: List[Callable[[str], Optional[str]]] = [
-                lambda x: api.get_command_doc(x.lower()),
-                lambda x: api.get_variable_doc(x),
-                lambda x: api.get_module_doc(x, False),
-                lambda x: api.get_module_doc(x, True),
+            candidates: List[Callable[[str], Optional[MarkupContent]]] = [
+                lambda x: self._get_command_doc(x.lower()),
+                lambda x: self._get_variable_doc(x),
+                lambda x: self._get_module_doc(x, False),
+                lambda x: self._get_module_doc(x, True),
             ]
             for c in candidates:
                 doc = c(word[0])
                 if doc is None:
                     continue
-                return Hover(
-                    contents=MarkupContent(kind=MarkupKind.Markdown, value=doc),
-                    range=word[1],
-                )
+                return Hover(contents=doc, range=word[1])
             return None
 
         @self.thread()
@@ -241,6 +235,21 @@ class CMakeLanguageServer(LanguageServer):
                 return word
         return None
 
+    def _get_command_doc(self, command: str) -> Optional[MarkupContent]:
+        assert self._api is not None
+        docs = self._api.get_command_doc(command)
+        return None if docs is None else MarkupContent(MarkupKind.Markdown, docs)
+
+    def _get_variable_doc(self, variable: str) -> Optional[MarkupContent]:
+        assert self._api is not None
+        docs = self._api.get_variable_doc(variable)
+        return None if docs is None else MarkupContent(MarkupKind.Markdown, docs)
+
+    def _get_module_doc(self, module: str, package: bool) -> Optional[MarkupContent]:
+        assert self._api is not None
+        docs = self._api.get_module_doc(module, package)
+        return None if docs is None else MarkupContent(MarkupKind.Markdown, docs)
+
 
 def main() -> None:
     from argparse import ArgumentParser
diff --git a/tests/test_server.py b/tests/test_server.py
index 1d960d0..25b13f6 100644
--- a/tests/test_server.py
+++ b/tests/test_server.py
@@ -19,6 +19,7 @@ from lsprotocol.types import (
     FormattingOptions,
     HoverParams,
     InitializeParams,
+    MarkupContent,
     Position,
     TextDocumentIdentifier,
     TextDocumentItem,
@@ -105,8 +106,8 @@ def test_completions(
     response = _test_completion(client_server, datadir, "projec", context)
     item = next(filter(lambda x: x.label == "project", response.items), None)
     assert item is not None
-    assert isinstance(item.documentation, str)
-    assert "<PROJECT-NAME>" in item.documentation
+    assert isinstance(item.documentation, MarkupContent)
+    assert "<PROJECT-NAME>" in item.documentation.value
 
 
 @pytest.mark.parametrize(