feat(completion): add as keyword completion (#158)

Fixes #150

Author: i582

server/src/completion/CompletionContext.ts

@@ -29,6 +29,7 @@ export class CompletionContext {
     isMessageContext: boolean = false
     isBouncedMessage: boolean = false
     isInitOfName: boolean = false
+    afterFieldType: boolean = false
 
     contextTy: Ty | null = null
 
@@ -115,6 +116,13 @@ export class CompletionContext {
             if (type) {
                 this.contextTy = TypeInferer.inferType(new Expression(type, this.element.file))
             }
+
+            const anchor = parent.childForFieldName("_completion_anchor")
+            if (anchor && element.node.equals(anchor)) {
+                this.afterFieldType = true
+                this.isExpression = false
+                this.isStatement = false
+            }
         }
 
         if (element.node.type === "type_identifier") {
@@ -263,6 +271,7 @@ export class CompletionContext {
             !this.inNameOfFieldInit &&
             !this.inTraitList &&
             !this.inParameter &&
+            !this.afterFieldType &&
             !this.isInitOfName
         )
     }

server/src/completion/providers/AsKeywordCompletionProvider.ts

@@ -0,0 +1,23 @@
+import {CompletionProvider} from "@server/completion/CompletionProvider"
+import {CompletionItemKind, InsertTextFormat} from "vscode-languageserver-types"
+import {CompletionContext} from "@server/completion/CompletionContext"
+import {CompletionResult, CompletionWeight} from "@server/completion/WeightedCompletionItem"
+
+export class AsKeywordCompletionProvider implements CompletionProvider {
+    isAvailable(ctx: CompletionContext): boolean {
+        return ctx.afterFieldType
+    }
+
+    addCompletion(_ctx: CompletionContext, result: CompletionResult): void {
+        result.add({
+            label: "as",
+            labelDetails: {
+                detail: " type",
+            },
+            kind: CompletionItemKind.Keyword,
+            insertText: "as $0",
+            insertTextFormat: InsertTextFormat.Snippet,
+            weight: CompletionWeight.CONTEXT_ELEMENT,
+        })
+    }
+}

server/src/completion/providers/ReferenceCompletionProvider.ts

@@ -20,6 +20,7 @@ export class ReferenceCompletionProvider implements CompletionProvider {
             !ctx.topLevelInStructOrMessage &&
             !ctx.topLevel &&
             !ctx.inTlbSerialization &&
+            !ctx.afterFieldType &&
             !ctx.inParameter
         )
     }

server/src/completion/providers/TlbSerializationCompletionProvider.ts

@@ -37,12 +37,14 @@ export class TlbSerializationCompletionProvider implements CompletionProvider {
         const type = field.typeNode()?.type()
         if (!type) return
 
+        const semicolonPart = !ctx.beforeSemicolon ? ";" : ""
+
         const typeName = type.name()
         if (typeName === "Cell" || typeName === "Slice" || typeName === "Builder") {
             result.add({
                 label: "remaining",
                 kind: CompletionItemKind.Keyword,
-                insertText: "remaining",
+                insertText: `remaining${semicolonPart}`,
                 weight: CompletionWeight.CONTEXT_ELEMENT,
             })
             return
@@ -54,7 +56,9 @@ export class TlbSerializationCompletionProvider implements CompletionProvider {
                     label: type,
                     kind: CompletionItemKind.Keyword,
                     insertTextFormat: InsertTextFormat.Snippet,
-                    insertText: type.includes("{X}") ? type.replace("{X}", "$1") : type,
+                    insertText: type.includes("{X}")
+                        ? type.replace("{X}", "$1")
+                        : type + semicolonPart,
                     weight: CompletionWeight.CONTEXT_ELEMENT,
                 })
             }

server/src/e2e/suite/testcases/completion/tlb-serialization.test

@@ -72,3 +72,36 @@ struct Transfer {
 }
 ------------------------------------------------------------------------
 13 remaining
+
+========================================================================
+TL-B serialization as
+========================================================================
+primitive Builder;
+
+struct Transfer {
+    value: Builder <caret>;
+}
+------------------------------------------------------------------------
+13 as type
+
+========================================================================
+TL-B serialization as 2
+========================================================================
+primitive Builder;
+
+struct Transfer {
+    value: Builder <caret>
+}
+------------------------------------------------------------------------
+13 as type
+
+========================================================================
+TL-B serialization as 3
+========================================================================
+primitive Builder;
+
+contract Transfer {
+    value: Builder <caret>
+}
+------------------------------------------------------------------------
+13 as type

server/src/server.ts

@@ -98,6 +98,7 @@ import {AddImport} from "@server/intentions/AddImport"
 import {NotImportedSymbolInspection} from "@server/inspections/NotImportedSymbolInspection"
 import {FillAllStructInit, FillRequiredStructInit} from "@server/intentions/FillAllStructInit"
 import {generateInitDoc, generateReceiverDoc} from "@server/documentation/receivers_documentation"
+import {AsKeywordCompletionProvider} from "@server/completion/providers/AsKeywordCompletionProvider"
 
 /**
  * Whenever LS is initialized.
@@ -688,6 +689,7 @@ connection.onInitialize(async (params: lsp.InitializeParams): Promise<lsp.Initia
             const providers: CompletionProvider[] = [
                 new SnippetsCompletionProvider(),
                 new KeywordsCompletionProvider(),
+                new AsKeywordCompletionProvider(),
                 new MapTypeCompletionProvider(),
                 new BouncedTypeCompletionProvider(),
                 new ContractDeclCompletionProvider(),

tree-sitter-tact/grammar.js

@@ -299,6 +299,7 @@ module.exports = grammar({
       seq(
         ":",
         field("type", $._type),
+        field("_completion_anchor", optional($.identifier)),
         field("tlb", optional($.tlb_serialization)),
         optional(seq("=", field("value", $._expression))),
       ),