feat(completion): add completion for top-level function definitions

Fixes #78

Author: i582

src/main/kotlin/org/tonstudio/tact/lang/completion/contributors/TactCompletionContributor.kt

@@ -8,6 +8,7 @@ import org.tonstudio.tact.lang.completion.TactCompletionPatterns.cachedReference
 import org.tonstudio.tact.lang.completion.TactCompletionPatterns.onAsmInstruction
 import org.tonstudio.tact.lang.completion.TactCompletionPatterns.onContractTraitTopLevel
 import org.tonstudio.tact.lang.completion.TactCompletionPatterns.onStatement
+import org.tonstudio.tact.lang.completion.TactCompletionPatterns.onTopLevel
 import org.tonstudio.tact.lang.completion.TactCompletionPatterns.referenceExpression
 import org.tonstudio.tact.lang.completion.providers.*
 import org.tonstudio.tact.lang.completion.sort.withTactSorter
@@ -18,6 +19,8 @@ class TactCompletionContributor : CompletionContributor() {
         extend(CompletionType.BASIC, cachedReferenceExpression(), ReferenceCompletionProvider)
         extend(CompletionType.BASIC, referenceExpression(), TlbTypesCompletionProvider)
 
+        extend(CompletionType.BASIC, onTopLevel(), FunctionDefinitionCompletionProvider)
+
         extend(CompletionType.BASIC, onContractTraitTopLevel(), ReceiversCompletionProvider)
         extend(CompletionType.BASIC, onContractTraitTopLevel(), ContractInitCompletionProvider)
         extend(CompletionType.BASIC, onContractTraitTopLevel(), ContractTraitMembersCompletionProvider)

src/main/kotlin/org/tonstudio/tact/lang/completion/contributors/TactKeywordsCompletionContributor.kt

@@ -28,14 +28,6 @@ class TactKeywordsCompletionContributor : CompletionContributor() {
         extend(CompletionType.BASIC, onTopLevel(), ConstCompletionProvider)
         extend(CompletionType.BASIC, onTopLevel(), ImportsCompletionProvider())
 
-        extend(
-            CompletionType.BASIC,
-            onTopLevel(),
-            ContextKeywordsCompletionProvider(
-                "fun",
-                needSpace = true,
-            )
-        )
         extend(
             CompletionType.BASIC,
             onTopLevel(),
@@ -174,17 +166,6 @@ class TactKeywordsCompletionContributor : CompletionContributor() {
         }
     }
 
-    open class ContextKeywordsCompletionProvider(
-        vararg keywords: String,
-        needSpace: Boolean = false,
-    ) : KeywordsCompletionProvider(
-        *keywords,
-        needSpace = needSpace,
-        properties = TactLookupElementProperties(
-            isContextElement = true,
-        ),
-    )
-
     open class KeywordsCompletionProvider(
         private vararg val keywords: String,
         private val needSpace: Boolean = false,

src/main/kotlin/org/tonstudio/tact/lang/completion/providers/FunctionDefinitionCompletionProvider.kt

@@ -0,0 +1,53 @@
+package org.tonstudio.tact.lang.completion.providers
+
+import com.intellij.codeInsight.completion.CompletionParameters
+import com.intellij.codeInsight.completion.CompletionProvider
+import com.intellij.codeInsight.completion.CompletionResultSet
+import com.intellij.codeInsight.lookup.LookupElementBuilder
+import com.intellij.codeInsight.template.impl.ConstantNode
+import com.intellij.util.ProcessingContext
+import org.tonstudio.tact.lang.completion.TactCompletionUtil
+
+object FunctionDefinitionCompletionProvider : CompletionProvider<CompletionParameters>() {
+    override fun addCompletions(parameters: CompletionParameters, context: ProcessingContext, result: CompletionResultSet) {
+        val modifiers = listOf("", "inline ", "extends ", "asm ")
+        modifiers.forEach { modifier ->
+            val element = LookupElementBuilder.create("${modifier}fun")
+                .bold()
+                .withTailText(" <name>() {}", true)
+                .withInsertHandler(
+                    TactCompletionUtil.TemplateStringInsertHandler(
+                        " \$name$(\$params$)\$return$ {\$END$}",
+                        true,
+                        "name" to ConstantNode("foo"),
+                        "params" to ConstantNode(""),
+                        "return" to ConstantNode(""),
+                    )
+                )
+
+            result.addElement(element)
+        }
+
+        val extendsPrefixes = listOf("", "asm ")
+        val extendsModifiers = listOf("", "mutates ")
+        extendsPrefixes.forEach { prefix ->
+            extendsModifiers.forEach { modifier ->
+                val extendsElement = LookupElementBuilder.create("${prefix}extends ${modifier}fun")
+                    .bold()
+                    .withTailText(" <name>(self: Type) {}", true)
+                    .withInsertHandler(
+                        TactCompletionUtil.TemplateStringInsertHandler(
+                            " \$name$(self: \$type$\$params$)\$return$ {\$END$}",
+                            true,
+                            "name" to ConstantNode("add"),
+                            "type" to ConstantNode("Int"),
+                            "params" to ConstantNode(""),
+                            "return" to ConstantNode(""),
+                        )
+                    )
+
+                result.addElement(extendsElement)
+            }
+        }
+    }
+}