diff --git a/src/Bicep.LangServer/SemanticTokenVisitor.cs b/src/Bicep.LangServer/SemanticTokenVisitor.cs index afb1df9f325..81381a7ba8d 100644 --- 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 index ba9cbb2aae5..ea0cab36b33 100644 --- 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); + } } }