feat(tlb): better highlighting (#741)

Author: i582

server/src/languages/tlb/completion/providers/BuiltinTypesCompletionProvider.ts

@@ -26,7 +26,7 @@ export class BuiltinTypesCompletionProvider implements CompletionProvider {
         ["Int", "257 bits"],
         ["UInt", "256 bits"],
         ["Bits", "1023 bits"],
-        ["bits", "1023 bits"],
+        ["bits", "X bits"],
         ["uint", ""],
         ["uint8", ""],
         ["uint16", ""],

server/src/languages/tlb/semantic-tokens/index.ts

@@ -24,7 +24,11 @@ export function provideTlbSemanticTokens(file: TlbFile): SemanticTokens {
 
     RecursiveVisitor.visit(file.rootNode, (node): boolean => {
         switch (node.type) {
-            case "##": {
+            case "#":
+            case "##":
+            case "#<":
+            case "#<=":
+            case "builtin_field": {
                 pushToken(node, SemanticTokenTypes.macro)
                 break
             }
@@ -41,6 +45,11 @@ export function provideTlbSemanticTokens(file: TlbFile): SemanticTokens {
                 break
             }
             case "type_identifier": {
+                if (isBuiltinType(node.text)) {
+                    pushToken(node, SemanticTokenTypes.macro)
+                    break
+                }
+
                 const parent = node.parent
                 if (!parent) break
 
@@ -91,13 +100,22 @@ export function provideTlbSemanticTokens(file: TlbFile): SemanticTokens {
             case "combinator": {
                 break
             }
-            case "builtin_field": {
-                pushToken(node, SemanticTokenTypes.property)
-                break
-            }
         }
         return true
     })
 
     return builder.build()
 }
+
+function isBuiltinType(name: string): boolean {
+    return (
+        name === "Any" ||
+        name === "Cell" ||
+        name === "Int" ||
+        name === "UInt" ||
+        name === "Bits" ||
+        name.startsWith("bits") ||
+        name.startsWith("uint") ||
+        name.startsWith("int")
+    )
+}