add IF/FOR as keywords to completion

Author: d-biehl

robotcode/language_server/robotframework/diagnostics/library_doc.py

@@ -69,8 +69,8 @@
 RUN_KEYWORDS = [*RUN_KEYWORD_NAMES, *RUN_KEYWORD_WITH_CONDITION_NAMES, RUN_KEYWORDS_NAME, RUN_KEYWORD_IF_NAME]
 
 BUILTIN_LIBRARY_NAME = "BuiltIn"
-
-DEFAULT_LIBRARIES = (BUILTIN_LIBRARY_NAME, "Reserved", "Easter")
+RESERVED_LIBRARY_NAME = "Reserved"
+DEFAULT_LIBRARIES = (BUILTIN_LIBRARY_NAME, RESERVED_LIBRARY_NAME, "Easter")
 ROBOT_LIBRARY_PACKAGE = "robot.libraries"
 
 ALLOWED_LIBRARY_FILE_EXTENSIONS = [".py"]

robotcode/language_server/robotframework/parts/completion.py

@@ -119,6 +119,19 @@ async def resolve(self, sender: Any, completion_item: CompletionItem) -> Complet
     "IF": [r"IF  \${${1}}", "$0", "END", ""],
 }
 
+RESERVERD_KEYWORDS = [
+    "FOR",
+    # "WHILE",
+    # "BREAK",
+    # "CONTINUE",
+    "END",
+    "IF",
+    "ELSE",
+    "ELIF",
+    "ELSE IF",
+    # "RETURN",
+]
+
 
 class CompletionCollector(ModelHelperMixin):
     _logger = LoggingDescriptor()
@@ -532,6 +545,16 @@ def enumerate_indexes(s: str, c: str) -> Iterator[int]:
             )
             result.append(c)
 
+        for k in RESERVERD_KEYWORDS:
+            c = CompletionItem(
+                label=k,
+                kind=CompletionItemKind.KEYWORD,
+                sort_text=f"040_{k}",
+                insert_text_format=InsertTextFormat.PLAINTEXT,
+                text_edit=TextEdit(range=r, new_text=k) if r is not None else None,
+            )
+            result.append(c)
+
         return result
 
     async def complete_default(