From 992fb409c22d2ae880643433df180ad81ec5ed73 Mon Sep 17 00:00:00 2001
From: Anthony Martin <38542602+anthony-c-martin@users.noreply.github.com>
Date: Thu, 19 Oct 2023 07:04:53 -0400
Subject: [PATCH] Publish semantic tokens for object type keys

---
 src/Bicep.LangServer/SemanticTokenVisitor.cs       | 14 ++++++++++++++
 .../LanguageHelpers/SemanticTokenVisitor.cs        | 14 ++++++++++++++
 2 files changed, 28 insertions(+)

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);
+        }
     }
 }