[Wasm] Add K1 compiler diagnostics for js(code) calls (KT-56955)

This function is handled as intrinsics and supported in limited context

wasm/wasm.frontend/src/org/jetbrains/kotlin/wasm/resolve/WasmPlatformConfigurator.kt

@@ -15,10 +15,7 @@ import org.jetbrains.kotlin.resolve.PlatformConfiguratorBase
 import org.jetbrains.kotlin.resolve.calls.checkers.LateinitIntrinsicApplicabilityChecker
 import org.jetbrains.kotlin.resolve.checkers.ExpectedActualDeclarationChecker
 import org.jetbrains.kotlin.wasm.analyze.WasmDiagnosticSuppressor
-import org.jetbrains.kotlin.wasm.resolve.diagnostics.WasmExternalDeclarationChecker
-import org.jetbrains.kotlin.wasm.resolve.diagnostics.WasmExternalInheritanceChecker
-import org.jetbrains.kotlin.wasm.resolve.diagnostics.WasmImportAnnotationChecker
-import org.jetbrains.kotlin.wasm.resolve.diagnostics.WasmJsFunAnnotationChecker
+import org.jetbrains.kotlin.wasm.resolve.diagnostics.*
 
 // TODO: Review the list of used K/JS checkers.
 //       Refactor useful checkers into common module.
@@ -41,7 +38,7 @@ object WasmPlatformConfigurator : PlatformConfiguratorBase(
 ) {
     override fun configureModuleComponents(container: StorageComponentContainer) {
         container.useInstance(NameSuggestion())
-        container.useImpl<JsCallChecker>()
+        container.useImpl<WasmJsCallChecker>()
         container.useImpl<JsNameClashChecker>()
         container.useImpl<JsNameCharsChecker>()
         container.useInstance(JsModuleClassLiteralChecker)

wasm/wasm.frontend/src/org/jetbrains/kotlin/wasm/resolve/diagnostics/DefaultErrorMessagesWasm.kt

@@ -5,6 +5,7 @@
 
 package org.jetbrains.kotlin.wasm.resolve.diagnostics
 
+import org.jetbrains.kotlin.diagnostics.rendering.CommonRenderers
 import org.jetbrains.kotlin.diagnostics.rendering.DefaultErrorMessages
 import org.jetbrains.kotlin.diagnostics.rendering.DiagnosticFactoryToRendererMap
 import org.jetbrains.kotlin.diagnostics.rendering.Renderers
@@ -25,6 +26,21 @@ private val DIAGNOSTIC_FACTORY_TO_RENDERER by lazy {
         put(ErrorsWasm.WASM_IMPORT_UNSUPPORTED_RETURN_TYPE, "Unsupported @WasmImport return type {0}", Renderers.RENDER_TYPE)
 
         put(ErrorsWasm.WRONG_JS_FUN_TARGET, "Only top-level external functions can be implemented using @JsFun")
+
+        put(
+            ErrorsWasm.JSCODE_WRONG_CONTEXT,
+            "Calls to js(code) should be a single expression inside a top-level function body or a property initializer in Kotlin/Wasm"
+        )
+        put(
+            ErrorsWasm.JSCODE_UNSUPPORTED_FUNCTION_KIND,
+            "Calls to js(code) are not supported in {0} in Kotlin/Wasm",
+            CommonRenderers.STRING
+        )
+        put(
+            ErrorsWasm.JSCODE_INVALID_PARAMETER_NAME,
+            "Parameters passed to js(code) should have a valid JavaScript name"
+        )
+
     }
 }
 

wasm/wasm.frontend/src/org/jetbrains/kotlin/wasm/resolve/diagnostics/ErrorsWasm.java

@@ -28,6 +28,10 @@ public interface ErrorsWasm {
 
     DiagnosticFactory0<PsiElement> WRONG_JS_FUN_TARGET = DiagnosticFactory0.create(ERROR);
 
+    DiagnosticFactory0<PsiElement> JSCODE_WRONG_CONTEXT = DiagnosticFactory0.create(ERROR);
+    DiagnosticFactory1<PsiElement, String> JSCODE_UNSUPPORTED_FUNCTION_KIND = DiagnosticFactory1.create(ERROR);
+    DiagnosticFactory0<PsiElement> JSCODE_INVALID_PARAMETER_NAME = DiagnosticFactory0.create(ERROR);
+
     @SuppressWarnings("UnusedDeclaration")
     Object _initializer = new Object() {
         {

wasm/wasm.frontend/src/org/jetbrains/kotlin/wasm/resolve/diagnostics/WasmJsCodeCallsChecker.kt

@@ -0,0 +1,97 @@
+/*
+ * Copyright 2010-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.wasm.resolve.diagnostics
+
+import com.intellij.psi.PsiElement
+import org.jetbrains.kotlin.descriptors.FunctionDescriptor
+import org.jetbrains.kotlin.descriptors.PropertyDescriptor
+import org.jetbrains.kotlin.descriptors.isTopLevelInPackage
+import org.jetbrains.kotlin.js.common.isValidES5Identifier
+import org.jetbrains.kotlin.js.resolve.diagnostics.ErrorsJs
+import org.jetbrains.kotlin.js.resolve.diagnostics.JsCallChecker.Companion.extractStringValue
+import org.jetbrains.kotlin.js.resolve.diagnostics.JsCallChecker.Companion.isJsCall
+import org.jetbrains.kotlin.js.resolve.diagnostics.findPsi
+import org.jetbrains.kotlin.psi.KtCallExpression
+import org.jetbrains.kotlin.resolve.TemporaryBindingTrace
+import org.jetbrains.kotlin.resolve.calls.checkers.CallChecker
+import org.jetbrains.kotlin.resolve.calls.checkers.CallCheckerContext
+import org.jetbrains.kotlin.resolve.calls.model.ResolvedCall
+import org.jetbrains.kotlin.resolve.constants.evaluate.ConstantExpressionEvaluator
+import org.jetbrains.kotlin.types.TypeUtils
+import org.jetbrains.kotlin.wasm.util.hasValidJsCodeBody
+
+class WasmJsCallChecker(
+    private val constantExpressionEvaluator: ConstantExpressionEvaluator
+) : CallChecker {
+
+    override fun check(resolvedCall: ResolvedCall<*>, reportOn: PsiElement, context: CallCheckerContext) {
+        if (context.isAnnotationContext || !resolvedCall.isJsCall()) return
+
+        val containingDeclaration = context.scope.ownerDescriptor
+        if (
+            !(containingDeclaration is FunctionDescriptor || containingDeclaration is PropertyDescriptor) ||
+            !containingDeclaration.isTopLevelInPackage()
+        ) {
+            context.trace.report(ErrorsWasm.JSCODE_WRONG_CONTEXT.on(reportOn))
+        } else {
+            when (containingDeclaration) {
+                is FunctionDescriptor -> {
+                    if (!containingDeclaration.hasValidJsCodeBody(context.trace.bindingContext)) {
+                        context.trace.report(ErrorsWasm.JSCODE_WRONG_CONTEXT.on(reportOn))
+                    } else {
+                        if (containingDeclaration.isSuspend) {
+                            context.trace.report(ErrorsWasm.JSCODE_UNSUPPORTED_FUNCTION_KIND.on(reportOn, "suspend function"))
+                        }
+                        if (containingDeclaration.isInline) {
+                            context.trace.report(ErrorsWasm.JSCODE_UNSUPPORTED_FUNCTION_KIND.on(reportOn, "inline function"))
+                        }
+                        if (containingDeclaration.extensionReceiverParameter != null) {
+                            context.trace.report(ErrorsWasm.JSCODE_UNSUPPORTED_FUNCTION_KIND.on(reportOn, "function with extension receiver"))
+                        }
+                        for (parameter in containingDeclaration.valueParameters) {
+                            if (parameter.name.identifierOrNullIfSpecial?.isValidES5Identifier() != true) {
+                                context.trace.report(ErrorsWasm.JSCODE_INVALID_PARAMETER_NAME.on(parameter.findPsi() ?: reportOn))
+                            }
+                        }
+                    }
+                }
+                is PropertyDescriptor -> {
+                    if (!containingDeclaration.hasValidJsCodeBody(context.trace.bindingContext)) {
+                        context.trace.report(ErrorsWasm.JSCODE_WRONG_CONTEXT.on(reportOn))
+                    }
+                }
+            }
+        }
+
+        val expression = resolvedCall.call.callElement
+        if (expression !is KtCallExpression) return
+
+        val arguments = expression.valueArgumentList?.arguments
+        val argument = arguments?.firstOrNull()?.getArgumentExpression() ?: return
+
+        val trace = TemporaryBindingTrace.create(context.trace, "WasmJsCallChecker")
+        val evaluationResult = constantExpressionEvaluator.evaluateExpression(argument, trace, TypeUtils.NO_EXPECTED_TYPE)
+        val code = extractStringValue(evaluationResult)
+
+        if (code == null) {
+            context.trace.report(ErrorsJs.JSCODE_ARGUMENT_SHOULD_BE_CONSTANT.on(argument))
+            return
+        }
+
+        trace.commit()
+    }
+}

wasm/wasm.frontend/src/org/jetbrains/kotlin/wasm/util/jsCodeUtils.kt

@@ -6,15 +6,22 @@
 package org.jetbrains.kotlin.wasm.util
 
 import org.jetbrains.kotlin.descriptors.FunctionDescriptor
+import org.jetbrains.kotlin.descriptors.PropertyDescriptor
 import org.jetbrains.kotlin.js.resolve.diagnostics.JsCallChecker.Companion.isJsCall
 import org.jetbrains.kotlin.psi.*
 import org.jetbrains.kotlin.resolve.BindingContext
 import org.jetbrains.kotlin.resolve.calls.util.getResolvedCall
 import org.jetbrains.kotlin.resolve.source.getPsi
 
+fun PropertyDescriptor.hasValidJsCodeBody(bindingContext: BindingContext): Boolean {
+    val property = source.getPsi() as? KtProperty ?: return false
+    val initializer = property.initializer ?: return false
+    return initializer.isJsCall(bindingContext)
+}
+
 fun FunctionDescriptor.hasValidJsCodeBody(bindingContext: BindingContext): Boolean {
-    val psi = source.getPsi() as? KtNamedFunction ?: return false
-    return psi.hasValidJsCodeBody(bindingContext)
+    val function = source.getPsi() as? KtNamedFunction ?: return false
+    return function.hasValidJsCodeBody(bindingContext)
 }
 
 private fun KtDeclarationWithBody.hasValidJsCodeBody(bindingContext: BindingContext): Boolean {