JS: support [suspend] fun main([args: Array<String>]) (KT-26628, KT-26572 fixed)

js/js.translator/src/org/jetbrains/kotlin/js/translate/general/Translation.java

@@ -22,6 +22,7 @@ import org.jetbrains.kotlin.builtins.KotlinBuiltIns;
 import org.jetbrains.kotlin.descriptors.DeclarationDescriptor;
 import org.jetbrains.kotlin.descriptors.FunctionDescriptor;
 import org.jetbrains.kotlin.descriptors.ModuleDescriptor;
+import org.jetbrains.kotlin.descriptors.PropertyDescriptor;
 import org.jetbrains.kotlin.idea.MainFunctionDetector;
 import org.jetbrains.kotlin.js.backend.ast.*;
 import org.jetbrains.kotlin.js.backend.ast.metadata.MetadataProperties;
@@ -41,17 +42,22 @@ import org.jetbrains.kotlin.js.translate.context.TranslationContext;
 import org.jetbrains.kotlin.js.translate.declaration.FileDeclarationVisitor;
 import org.jetbrains.kotlin.js.translate.expression.ExpressionVisitor;
 import org.jetbrains.kotlin.js.translate.expression.PatternTranslator;
+import org.jetbrains.kotlin.js.translate.reference.ReferenceTranslator;
 import org.jetbrains.kotlin.js.translate.test.JSTestGenerator;
 import org.jetbrains.kotlin.js.translate.utils.*;
 import org.jetbrains.kotlin.js.translate.utils.mutator.AssignToExpressionMutator;
+import org.jetbrains.kotlin.name.FqNameUnsafe;
+import org.jetbrains.kotlin.name.Name;
 import org.jetbrains.kotlin.psi.KtDeclaration;
 import org.jetbrains.kotlin.psi.KtExpression;
 import org.jetbrains.kotlin.psi.KtFile;
 import org.jetbrains.kotlin.psi.KtSimpleNameExpression;
 import org.jetbrains.kotlin.resolve.BindingTrace;
+import org.jetbrains.kotlin.resolve.DescriptorUtils;
 import org.jetbrains.kotlin.resolve.bindingContextUtil.BindingContextUtilsKt;
 import org.jetbrains.kotlin.resolve.constants.*;
 import org.jetbrains.kotlin.resolve.constants.evaluate.ConstantExpressionEvaluator;
+import org.jetbrains.kotlin.resolve.scopes.MemberScope;
 import org.jetbrains.kotlin.serialization.js.ast.JsAstDeserializer;
 import org.jetbrains.kotlin.types.KotlinType;
 import org.jetbrains.kotlin.types.TypeUtils;
@@ -468,8 +474,24 @@ public final class Translation {
         if (functionDescriptor == null) {
             return null;
         }
-        JsArrayLiteral argument = new JsArrayLiteral(toStringLiteralList(arguments));
-        JsExpression call = CallTranslator.INSTANCE.buildCall(context, functionDescriptor, Collections.singletonList(argument), null);
+
+        int parameterCount = functionDescriptor.getValueParameters().size();
+        assert parameterCount <= 1;
+
+        List<JsExpression> args = new ArrayList<>();
+        if (parameterCount != 0) {
+            args.add(new JsArrayLiteral(toStringLiteralList(arguments)));
+        }
+
+        if (functionDescriptor.isSuspend()) {
+            MemberScope scope = moduleDescriptor.getPackage(new FqNameUnsafe("kotlin.coroutines.js.internal").toSafe()).getMemberScope();
+            PropertyDescriptor emptyContinuation = DescriptorUtils.getPropertyByName(scope, Name.identifier("EmptyContinuation"));
+
+            args.add(ReferenceTranslator.translateAsValueReference(emptyContinuation, context));
+            args.add(new JsBooleanLiteral(false));
+        }
+
+        JsExpression call = CallTranslator.INSTANCE.buildCall(context, functionDescriptor, args, null);
         context.addTopLevelStatement(call.makeStmt());
         return staticContext.getFragment();
     }

js/js.translator/testData/box/main/noArgs.kt

@@ -0,0 +1,11 @@
+// EXPECTED_REACHABLE_NODES: 1281
+// IGNORE_BACKEND: JS_IR
+// CALL_MAIN
+
+var ok: String = "fail"
+
+fun main() {
+    ok = "OK"
+}
+
+fun box() = ok

js/js.translator/testData/box/main/simple.kt

@@ -0,0 +1,14 @@
+// EXPECTED_REACHABLE_NODES: 1281
+// IGNORE_BACKEND: JS_IR
+// CALL_MAIN
+
+var ok: String = "fail"
+
+fun main(args: Array<String>) {
+    assertEquals(1, args.size)
+    assertEquals("testArg", args[0])
+
+    ok = "OK"
+}
+
+fun box() = ok

js/js.translator/testData/box/main/suspendMain.kt

@@ -0,0 +1,28 @@
+// EXPECTED_REACHABLE_NODES: 1296
+// IGNORE_BACKEND: JS_IR
+// CALL_MAIN
+
+import kotlin.coroutines.*
+
+var ok: String = "fail"
+
+var callback: () -> Unit = {}
+
+suspend fun main(args: Array<String>) {
+    assertEquals(1, args.size)
+    assertEquals("testArg", args[0])
+
+    suspendCoroutine<Unit> { cont ->
+        callback = {
+            cont.resume(Unit)
+        }
+    }
+
+    ok = "OK"
+}
+
+fun box(): String {
+    assertEquals("fail", ok)
+    callback()
+    return ok
+}

js/js.translator/testData/box/main/suspendMainNoArgs.kt

@@ -0,0 +1,25 @@
+// EXPECTED_REACHABLE_NODES: 1296
+// IGNORE_BACKEND: JS_IR
+// CALL_MAIN
+
+import kotlin.coroutines.*
+
+var ok: String = "fail"
+
+var callback: () -> Unit = {}
+
+suspend fun main() {
+    suspendCoroutine<Unit> { cont ->
+        callback = {
+            cont.resume(Unit)
+        }
+    }
+
+    ok = "OK"
+}
+
+fun box(): String {
+    assertEquals("fail", ok)
+    callback()
+    return ok
+}

js/js.translator/testData/box/main/suspendMainThrows.kt

@@ -0,0 +1,30 @@
+// EXPECTED_REACHABLE_NODES: 1299
+// IGNORE_BACKEND: JS_IR
+// CALL_MAIN
+
+import kotlin.coroutines.*
+
+var callback: () -> Unit = {}
+
+val exception = Exception()
+
+suspend fun main() {
+
+    suspendCoroutine<Unit> { cont ->
+        callback = {
+            cont.resume(Unit)
+        }
+    }
+
+    throw exception
+}
+
+fun box(): String {
+    try {
+        callback()
+    } catch (e: Exception) {
+        assertTrue(e === exception)
+    }
+
+    return "OK"
+}