diff --git a/server/src/tools/semantic_tokens.py b/server/src/tools/semantic_tokens.py index f701605..33a561f 100644 --- a/server/src/tools/semantic_tokens.py +++ b/server/src/tools/semantic_tokens.py @@ -108,6 +108,12 @@ class _Highlighter(Visitor): if name == "global": line, col = routine.contents_pos self._tokens.append((((line - 1, col - 1), len(name), "keyword", []))) + first_arg = command.args[0] + token_info = self._get_token_info(first_arg) + if token_info: + (line, col), length = token_info + mods = [TokenModifier.declaration, TokenModifier.write] + self._tokens.append(((line, col), length, "variable", mods)) if routine.contents == "puts": line, col = routine.contents_pos