Publish semantic tokens for object type keys

Azure · Oct 19, 2023 · 992fb40 · 992fb40
1 parent 865577d
commit 992fb40
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/Bicep.LangServer/SemanticTokenVisitor.cs b/src/Bicep.LangServer/SemanticTokenVisitor.cs
@@ -397,5 +397,19 @@ public override void VisitAssertDeclarationSyntax(AssertDeclarationSyntax syntax
             AddTokenType(syntax.Name, SemanticTokenType.Variable);
             base.VisitAssertDeclarationSyntax(syntax);
         }
+
+        public override void VisitObjectTypePropertySyntax(ObjectTypePropertySyntax syntax)
+        {
+            if (syntax.Key is StringSyntax @string)
+            {
+                Visit(@string);
+            }
+            else
+            {
+                AddTokenType(syntax.Key, SemanticTokenType.TypeParameter);
+            }
+            Visit(syntax.Colon);
+            Visit(syntax.Value);
+        }
     }
 }
diff --git a/src/Bicep.Wasm/LanguageHelpers/SemanticTokenVisitor.cs b/src/Bicep.Wasm/LanguageHelpers/SemanticTokenVisitor.cs
@@ -365,5 +365,19 @@ public override void VisitAssertDeclarationSyntax(AssertDeclarationSyntax syntax
             AddTokenType(syntax.Name, SemanticTokenType.Variable);
             base.VisitAssertDeclarationSyntax(syntax);
         }
+
+        public override void VisitObjectTypePropertySyntax(ObjectTypePropertySyntax syntax)
+        {
+            if (syntax.Key is StringSyntax @string)
+            {
+                Visit(@string);
+            }
+            else
+            {
+                AddTokenType(syntax.Key, SemanticTokenType.TypeParameter);
+            }
+            Visit(syntax.Colon);
+            Visit(syntax.Value);
+        }
     }
 }