implement hover and goto for assign tokens

Author: d-biehl

robotcode/language_server/robotframework/parts/goto.py

@@ -25,6 +25,7 @@
     get_tokens_at_position,
     range_from_token,
     range_from_token_or_node,
+    tokenize_variables,
 )
 
 if TYPE_CHECKING:
@@ -103,9 +104,11 @@ async def _definition_default(
 
         tokens = get_tokens_at_position(node, position)
 
-        for t in tokens:
+        for token in tokens:
             try:
-                for sub_token in filter(lambda s: s.type == RobotToken.VARIABLE, t.tokenize_variables()):
+                for sub_token in filter(
+                    lambda s: s.type == RobotToken.VARIABLE, tokenize_variables(token, ignore_errors=True)
+                ):
                     range = range_from_token(sub_token)
 
                     if position.is_in_range(range):

robotcode/language_server/robotframework/parts/hover.py

@@ -24,6 +24,7 @@
     get_tokens_at_position,
     range_from_token,
     range_from_token_or_node,
+    tokenize_variables,
 )
 
 if TYPE_CHECKING:
@@ -91,9 +92,11 @@ async def _hover_default(self, nodes: List[ast.AST], document: TextDocument, pos
 
         tokens = get_tokens_at_position(node, position)
 
-        for t in tokens:
+        for token in tokens:
             try:
-                for sub_token in filter(lambda s: s.type == RobotToken.VARIABLE, t.tokenize_variables()):
+                for sub_token in filter(
+                    lambda s: s.type == RobotToken.VARIABLE, tokenize_variables(token, ignore_errors=True)
+                ):
                     range = range_from_token(sub_token)
 
                     if position.is_in_range(range):

robotcode/language_server/robotframework/utils/ast.py

@@ -200,7 +200,7 @@ def tokenize_variables(
     from robot.api.parsing import Token as RobotToken
     from robot.variables import VariableIterator
 
-    if token.type not in {*RobotToken.ALLOW_VARIABLES, RobotToken.KEYWORD}:
+    if token.type not in {*RobotToken.ALLOW_VARIABLES, RobotToken.KEYWORD, RobotToken.ASSIGN}:
         return _tokenize_no_variables(token)
     variables = VariableIterator(token.value, identifiers=identifiers, ignore_errors=ignore_errors)
     if not variables: