fix(completion): add parameters to contract completion inside initOf (#258)

Fixes #254

Author: i582

server/src/completion/ReferenceCompletionProcessor.ts

@@ -141,7 +141,8 @@ export class ReferenceCompletionProcessor implements ScopeProcessor {
                 data: additionalData,
             })
         } else if (node instanceof Contract) {
-            const suffix = this.ctx.isInitOfName ? "()" : ""
+            const parameters = node.initFunction()?.parametersPresentation() ?? "()"
+            const suffix = this.ctx.isInitOfName ? parameters : ""
             const initFunction = node.initFunction()
             const needParens = this.ctx.isInitOfName && !this.ctx.beforeParen
             const hasParameters = initFunction !== null && initFunction.parameters().length > 0

server/src/e2e/suite/testcases/completion/initOf.test

@@ -54,3 +54,29 @@ fun test() {
 ------------------------------------------------------------------------
 3  Bar()
 3  Foo()
+
+========================================================================
+initOf contract name completion with parens and init function
+========================================================================
+contract Foo {
+    init(age: Int, value: String) {}
+}
+
+contract Bar {
+    init(age: Int) {}
+}
+
+contract Baz {
+    init() {}
+}
+
+trait Some {}
+trait Some2 {}
+
+fun test() {
+    initOf <caret>()
+}
+------------------------------------------------------------------------
+3  Bar(age: Int)
+3  Baz()
+3  Foo(age: Int, value: String)

server/src/psi/Decls.ts

@@ -162,6 +162,12 @@ export class InitFunction extends Node {
         return `init${parametersNode.text}`
     }
 
+    public parametersPresentation(): string {
+        const parametersNode = this.node.childForFieldName("parameters")
+        if (!parametersNode) return "()"
+        return parametersNode.text
+    }
+
     public endParen(): SyntaxNode | null {
         const parametersNode = this.node.childForFieldName("parameters")
         if (!parametersNode) return null