semantic highlighting of new WHILE and EXCEPT options for RF 5.0

Author: d-biehl

CHANGELOG.md

@@ -8,6 +8,8 @@ All notable changes to the "robotcode" extension will be documented in this file
   - see [README](https://github.com/d-biehl/robotcode#using-pre-release-version)
 - rework handling VSCode test items to ensure all defined tests can be executed, also when they are ambiguous
   - see [#37](https://github.com/d-biehl/robotcode/issues/37)
+- semantic highlighting of new WHILE and EXCEPT options for RF 5.0
+-
 
 
 ##  0.7.0

robotcode/language_server/robotframework/parts/semantic_tokens.py

@@ -366,6 +366,26 @@ async def generate_sem_sub_tokens(
                 yield SemTokenInfo.from_token(token, sem_type, sem_mod, col_offset + kw_index, len(kw))
             elif token.type == RobotToken.NAME and isinstance(node, (LibraryImport, ResourceImport, VariablesImport)):
                 yield SemTokenInfo.from_token(token, RobotSemTokenTypes.NAMESPACE, sem_mod, col_offset, length)
+            elif get_robot_version() >= (5, 0) and token.type == RobotToken.OPTION:
+                from robot.parsing.model.statements import ExceptHeader, WhileHeader
+
+                if (
+                    isinstance(node, ExceptHeader)
+                    and token.value.startswith("type=")
+                    or isinstance(node, WhileHeader)
+                    and token.value.startswith("limit=")
+                ):
+                    if col_offset is None:
+                        col_offset = token.col_offset
+
+                    name, value = token.value.split("=", 1)
+                    yield SemTokenInfo.from_token(token, RobotSemTokenTypes.VARIABLE, sem_mod, col_offset, len(name))
+                    yield SemTokenInfo.from_token(
+                        token, SemanticTokenTypes.OPERATOR, sem_mod, col_offset + len(name), 1
+                    )
+                    yield SemTokenInfo.from_token(token, sem_type, sem_mod, col_offset + len(name) + 1, len(value))
+                else:
+                    yield SemTokenInfo.from_token(token, sem_type, sem_mod, col_offset, length)
             elif (
                 token.type in RobotToken.SETTING_TOKENS
                 and token.value