Color themes

huntercfreeman · May 22, 2023 · 7b2b180 · 7b2b180
1 parent 8e5755f
commit 7b2b180
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/BlazorStudio.ClassLib/CodeAnalysis/C/Facts/CLanguageFacts.Keywords.cs b/BlazorStudio.ClassLib/CodeAnalysis/C/Facts/CLanguageFacts.Keywords.cs
@@ -74,5 +74,20 @@ public class Keywords
             VOLATILE_KEYWORD,
             WHILE_KEYWORD,
         }.ToImmutableArray();
+
+        public static readonly ImmutableArray<string> CONTROL_KEYWORDS = new[]
+        {
+            BREAK_KEYWORD,
+            CASE_KEYWORD,
+            CONTINUE_KEYWORD,
+            DO_KEYWORD,
+            ELSE_KEYWORD,
+            FOR_KEYWORD,
+            GOTO_KEYWORD,
+            IF_KEYWORD,
+            RETURN_KEYWORD,
+            SWITCH_KEYWORD,
+            WHILE_KEYWORD,
+        }.ToImmutableArray();
     }
 }
diff --git a/BlazorStudio.ClassLib/CodeAnalysis/C/Syntax/LexSession.cs b/BlazorStudio.ClassLib/CodeAnalysis/C/Syntax/LexSession.cs
@@ -250,9 +250,14 @@ private ISyntaxToken ConsumeIdentifierOrKeyword()
 
         if (CLanguageFacts.Keywords.ALL.Contains(textValue))
         {
+            var decorationByte = (byte)GenericDecorationKind.Keyword;
+
+            if (CLanguageFacts.Keywords.CONTROL_KEYWORDS.Contains(textValue))
+                decorationByte = (byte)GenericDecorationKind.KeywordControl;
+
             textSpan = textSpan with
             {
-                DecorationByte = (byte)GenericDecorationKind.Keyword,
+                DecorationByte = decorationByte,
             };
 
             return new KeywordToken(textSpan);