feat: semantic highlighting of block lambda results (#1011)

### Summary of Changes Add missing semantic highlighting for block lambda results.
Safe-DS · Apr 8, 2024 · 228733c · 228733c
1 parent 2c72cee
commit 228733c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/packages/safe-ds-lang/src/language/lsp/safe-ds-semantic-token-provider.ts b/packages/safe-ds-lang/src/language/lsp/safe-ds-semantic-token-provider.ts
@@ -7,6 +7,7 @@ import {
     isSdsAnnotationCall,
     isSdsArgument,
     isSdsAttribute,
+    isSdsBlockLambdaResult,
     isSdsClass,
     isSdsDeclaration,
     isSdsEnum,
@@ -150,6 +151,12 @@ export class SafeDsSemanticTokenProvider extends AbstractSemanticTokenProvider {
                 type: SemanticTokenTypes.property,
                 modifier,
             };
+        } else if (isSdsBlockLambdaResult(node)) {
+            return {
+                // For lack of a better option, we use the token type for parameters here
+                type: SemanticTokenTypes.parameter,
+                modifier: additionalModifiers,
+            };
         } else if (isSdsClass(node)) {
             const isBuiltinClass = this.builtinClasses.isBuiltinClass(node);
             return {

diff --git a/packages/safe-ds-lang/tests/language/lsp/safe-ds-semantic-token-provider.test.ts b/packages/safe-ds-lang/tests/language/lsp/safe-ds-semantic-token-provider.test.ts
@@ -40,6 +40,17 @@ describe('SafeDsSemanticTokenProvider', async () => {
             `,
             expectedTokenTypes: [SemanticTokenTypes.property, SemanticTokenTypes.property],
         },
+        {
+            testName: 'block lambda result declaration',
+            code: `
+                pipeline myPipeline {
+                    () {
+                        yield <|result|> = 1;
+                    };
+                }
+            `,
+            expectedTokenTypes: [SemanticTokenTypes.parameter],
+        },
         {
             testName: 'class declaration',
             code: 'class <|C|>',