Highlight vars nicely.

thesoftwarephilosopher · Aug 22, 2024 · c47f45c · c47f45c
1 parent 15aad6a
commit c47f45c
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/site/token-provider.ts b/site/token-provider.ts
@@ -9,13 +9,16 @@
  * 2. `export/etc` are highlighted as control-flow keywords
  * 3. Class fields are highlighted as variables
  * 4. Highlight function/class def names properly
+ * 5. Highlight var/let/const names properly
  * 
  */
 
 export const rules = [
   // { token: "identifier.ts", foreground: "9CDCFE" },
   { token: "variable.property.ts", foreground: "9CDCFE" },
   // { token: "function.ts", foreground: "DCDCAA" },
+  { token: "constant.ts", foreground: "4FC1FF" },
+  { token: "variable.ts", foreground: "9CDCFE" },
   { token: "method.ts", foreground: "DCDCAA" },
   // { token: "delimiter.ts", foreground: "569CD6" },
 ];
@@ -81,14 +84,22 @@ export const tokenProvider = {
         { token: 'keyword' },
         {
           cases: {
-            '$1~function\\s+': { token: 'method' },
-            '$1~class\\s+': { token: 'type.identifier' },
-            '': { token: 'string.sql' },
+            '$1~function\\s+': 'method',
+            '$1~class\\s+': 'type.identifier',
+            '@default': 'identifier',
           }
         },
         { token: '@rematch' },
       ]],
 
+      // highlight var/const/let defs nicely
+      [/((?:const|let|var)\s+)(#?[\w$]+)/, ['keyword', {
+        cases: {
+          '$1~const\\s+': 'constant',
+          '@default': 'variable',
+        }
+      }]],
+
       // identifiers and keywords
       [/#?[a-z_$][\w$]*/, {
         cases: {