Analyse environment variables

Author: d-biehl

robotcode/language_server/robotframework/diagnostics/analyzer.py

@@ -3,6 +3,7 @@
 import ast
 import asyncio
 import itertools
+import os
 import re
 from collections import defaultdict
 from dataclasses import dataclass
@@ -35,7 +36,11 @@
 )
 from ..utils.async_ast import AsyncVisitor
 from ..utils.version import get_robot_version
-from .entities import VariableDefinition, VariableNotFoundDefinition
+from .entities import (
+    EnvironmentVariableDefinition,
+    VariableDefinition,
+    VariableNotFoundDefinition,
+)
 from .library_doc import KeywordDoc, is_embedded_keyword
 from .namespace import DIAGNOSTICS_SOURCE_NAME, KeywordFinder, Namespace
 
@@ -194,6 +199,17 @@ async def visit(self, node: ast.AST) -> None:
                                     code="VariableNotFound",
                                 )
                             else:
+                                if isinstance(var, EnvironmentVariableDefinition) and var.default_value is None:
+                                    env_name = var.name[2:-1]
+                                    if os.environ.get(env_name, None) is None:
+                                        await self.append_diagnostics(
+                                            range=range_from_token(var_token),
+                                            message=f"Environment variable '{var.name}' not found.",
+                                            severity=severity,
+                                            source=DIAGNOSTICS_SOURCE_NAME,
+                                            code="EnvirommentVariableNotFound",
+                                        )
+
                                 if self.namespace.document is not None:
                                     var_range = range_from_token(var_token)
                                     if var.name_range != var_range:

robotcode/language_server/robotframework/diagnostics/entities.py

@@ -264,6 +264,8 @@ class EnvironmentVariableDefinition(VariableDefinition):
     type: VariableDefinitionType = VariableDefinitionType.ENVIRONMENT_VARIABLE
     resolvable: bool = True
 
+    default_value: Any = field(default=None, compare=False)
+
     @single_call
     def __hash__(self) -> int:
         return hash((type(self), self.name, self.type))

robotcode/language_server/robotframework/diagnostics/namespace.py

@@ -976,7 +976,10 @@ async def find_variable(
         await self.ensure_initialized()
 
         if name[:2] == "%{" and name[-1] == "}":
-            return EnvironmentVariableDefinition(0, 0, 0, 0, "", name, None)
+            var_name, _, default_value = name[2:-1].partition("=")
+            return EnvironmentVariableDefinition(
+                0, 0, 0, 0, "", f"%{{{var_name}}}", None, default_value=default_value or None
+            )
 
         try:
             matcher = VariableMatcher(name)