[FIR2IR] Find defaultValue from expect functions when 'allowNonCachedDeclarations'

... To support code fragments (such as in the debuggers
'evaluate expression'.

See: IDEA-358599

Author: sellmair

analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/components/compilerFacility/FirIdeNormalAnalysisLibrarySourceModuleCompilerFacilityTestGenerated.java

@@ -0,0 +1,13 @@
+MODULE_FRAGMENT
+  FILE fqName:<root> fileName:fragment.kt
+    CLASS CLASS name:CodeFragment modality:FINAL visibility:public superTypes:[kotlin.Any]
+      $this: VALUE_PARAMETER INSTANCE_RECEIVER name:<this> type:<root>.CodeFragment
+      CONSTRUCTOR visibility:public <> () returnType:<root>.CodeFragment [primary]
+        BLOCK_BODY
+          DELEGATING_CONSTRUCTOR_CALL 'public constructor <init> () [primary] declared in kotlin.Any'
+      FUN name:run visibility:public modality:FINAL <> () returnType:kotlin.Int
+        EXPRESSION_BODY
+          BLOCK type=kotlin.Int origin=null
+            CALL 'public final fun plus (other: kotlin.Int): kotlin.Int [operator] declared in kotlin.Int' type=kotlin.Int origin=PLUS
+              $this: CALL 'public final fun foo (default: kotlin.Int): kotlin.Int declared in <root>.JvmKt' type=kotlin.Int origin=null
+              other: CALL 'public final fun bar (): kotlin.Int declared in <root>.JvmKt' type=kotlin.Int origin=null

analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/components/compilerFacility/FirIdeNormalAnalysisSourceModuleCompilerFacilityTestGenerated.java

@@ -0,0 +1,30 @@
+// LANGUAGE: +MultiPlatformProjects
+
+// MODULE: common
+// TARGET_PLATFORM: Common
+// FILE: common.kt
+
+
+expect fun foo(default: Int = 42): Int
+
+expect fun bar(): Int
+
+fun test() {
+    <caret_context>val x = 0
+}
+
+// MODULE: jvm()()(common)
+// TARGET_PLATFORM: JVM
+// FILE: jvm.kt
+actual fun foo(default: Int) = default
+
+actual fun bar() = 0
+
+// MODULE: main
+// MODULE_KIND: CodeFragment
+// CONTEXT_MODULE: common
+// ANALYSIS_CONTEXT_MODULE: jvm
+
+// FILE: fragment.kt
+// CODE_FRAGMENT_KIND: EXPRESSION
+foo() + bar()

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/expectFunctionWithDefaultParam.ir.txt

@@ -0,0 +1,5 @@
+public final class CodeFragment {
+    // source: 'fragment.kt'
+    public method <init>(): void
+    public final static method run(): int
+}

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/expectFunctionWithDefaultParam.kt

@@ -36,7 +36,17 @@ object AnalysisApiTestDirectives : SimpleDirectivesContainer() {
     )
 
     val CONTEXT_MODULE by stringDirective(
-        description = "Specifies the module name which should be treated as a context module for the current one",
+        description = "Specifies the module name used to find the 'context psi element' for this module",
+        applicability = DirectiveApplicability.Module
+    )
+
+    /*
+    Note: the 'contextElement' can be different from the 'contextModule'.
+    E.g., consider a multiplatform project where the contextElement is in 'commonMain', but the contextModule can be
+    configured as 'jvmMain'
+    */
+    val ANALYSIS_CONTEXT_MODULE by stringDirective(
+        description = "Specifies the module name which should be treated as a context module for the current one (can overwrite 'CONTEXT_MODULE')",
         applicability = DirectiveApplicability.Module
     )
 }

analysis/analysis-api/testData/components/compilerFacility/compilation/codeFragments/expectFunctionWithDefaultParam.txt

@@ -6,9 +6,11 @@
 package org.jetbrains.kotlin.analysis.test.framework.projectStructure
 
 import com.intellij.openapi.project.Project
+import org.jetbrains.kotlin.analysis.api.KaImplementationDetail
 import org.jetbrains.kotlin.analysis.api.projectStructure.KaDanglingFileResolutionMode
 import org.jetbrains.kotlin.analysis.api.platform.projectStructure.KaDanglingFileModuleImpl
 import org.jetbrains.kotlin.analysis.api.platform.projectStructure.forcedSpecialModule
+import org.jetbrains.kotlin.analysis.api.projectStructure.analysisContextModule
 import org.jetbrains.kotlin.analysis.test.framework.services.TestForeignValue
 import org.jetbrains.kotlin.analysis.test.framework.services.TestForeignValueProviderService
 import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
@@ -20,6 +22,7 @@ import org.jetbrains.kotlin.psi.KtElement
 import org.jetbrains.kotlin.psi.KtExpressionCodeFragment
 import org.jetbrains.kotlin.psi.KtFile
 import org.jetbrains.kotlin.psi.KtTypeCodeFragment
+import org.jetbrains.kotlin.psi.analysisContext
 import org.jetbrains.kotlin.psi.psiUtil.getParentOfType
 import org.jetbrains.kotlin.psi.psiUtil.isIdentifier
 import org.jetbrains.kotlin.test.directives.model.DirectiveApplicability
@@ -28,6 +31,7 @@ import org.jetbrains.kotlin.test.directives.model.singleOrZeroValue
 import org.jetbrains.kotlin.test.model.TestFile
 import org.jetbrains.kotlin.test.model.TestModule
 import org.jetbrains.kotlin.test.services.TestServices
+import org.jetbrains.kotlin.test.services.moduleStructure
 import org.jetbrains.kotlin.test.services.sourceFileProvider
 import org.jetbrains.org.objectweb.asm.Type
 import java.nio.file.Path

analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/AnalysisApiTestDirectives.kt

@@ -11,10 +11,6 @@ import com.intellij.psi.PsiManager
 import com.intellij.psi.search.GlobalSearchScope
 import org.jetbrains.kotlin.analysis.api.projectStructure.*
 import org.jetbrains.kotlin.analysis.api.standalone.base.projectStructure.StandaloneProjectFactory
-import org.jetbrains.kotlin.analysis.api.projectStructure.KaDanglingFileModule
-import org.jetbrains.kotlin.analysis.api.projectStructure.KaLibraryModule
-import org.jetbrains.kotlin.analysis.api.projectStructure.KaModule
-import org.jetbrains.kotlin.analysis.api.projectStructure.KaNotUnderContentRootModule
 import org.jetbrains.kotlin.analysis.test.framework.AnalysisApiTestDirectives
 import org.jetbrains.kotlin.analysis.test.framework.services.environmentManager
 import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
@@ -46,7 +42,7 @@ object TestModuleStructureFactory {
     fun createProjectStructureByTestStructure(
         moduleStructure: TestModuleStructure,
         testServices: TestServices,
-        project: Project
+        project: Project,
     ): KtTestModuleStructure {
         val modules = createModules(moduleStructure, testServices, project)
 
@@ -81,14 +77,17 @@ object TestModuleStructureFactory {
             val contextModuleName = testModule.directives.singleOrZeroValue(AnalysisApiTestDirectives.CONTEXT_MODULE)
             val contextModule = contextModuleName?.let(existingModules::getValue)
 
+            val analysisContextModuleName = testModule.directives.singleOrZeroValue(AnalysisApiTestDirectives.ANALYSIS_CONTEXT_MODULE)
+            val analysisContextModule = analysisContextModuleName?.let(existingModules::getValue)
+
             val dependencyBinaryRoots = testModule.regularDependencies.flatMap { dependency ->
                 val libraryModule = existingModules.getValue(dependency.moduleName).ktModule as? KaLibraryModule
                 libraryModule?.binaryRoots.orEmpty()
             }
 
             val ktTestModule = testServices
                 .getKtModuleFactoryForTestModule(testModule)
-                .createModule(testModule, contextModule, dependencyBinaryRoots, testServices, project)
+                .createModule(testModule, analysisContextModule ?: contextModule, dependencyBinaryRoots, testServices, project)
 
             existingModules[testModule.name] = ktTestModule
             result.add(ktTestModule)
@@ -146,7 +145,7 @@ object TestModuleStructureFactory {
         testModule: TestModule,
         testServices: TestServices,
         ktModule: KtModuleWithModifiableDependencies,
-        libraryCache: (paths: Set<Path>, factory: () -> KaLibraryModule) -> KaLibraryModule
+        libraryCache: (paths: Set<Path>, factory: () -> KaLibraryModule) -> KaLibraryModule,
     ) {
         val project = ktModule.project
 

analysis/analysis-test-framework/tests/org/jetbrains/kotlin/analysis/test/framework/projectStructure/KtCodeFragmentTestModuleFactory.kt

@@ -751,7 +751,28 @@ class Fir2IrCallableDeclarationsGenerator(private val c: Fir2IrComponents) : Fir
                 isNoinline = valueParameter.isNoinline,
                 isHidden = false,
             ).apply {
-                val defaultValue = valueParameter.defaultValue
+                val defaultValue = valueParameter.defaultValue ?: run tryFindInExpect@{
+                    /*
+                    Scenarios, such as evaluating expressions in the debugger, will 'allow non-cached declarations'.
+                    In this case some symbols, which can be considered 'dependencies' will not produce IR.
+                    Therefore, the 'FunctionDefaultParametersActualizer' will not run.
+
+                    In such cases, we're trying to look up default values from such 'dependency' classes manually
+                     */
+                    if (!c.configuration.allowNonCachedDeclarations) return@tryFindInExpect null
+
+                    val actualFunction = valueParameter.containingFunctionSymbol
+                    if (!actualFunction.isActual) return@tryFindInExpect null
+
+                    val expectFunction = actualFunction.getSingleMatchedExpectForActualOrNull()
+                        ?: return@tryFindInExpect null
+
+                    val expectValueParam = expectFunction.valueParameterSymbols.find { it.name == name }
+                        ?: return@tryFindInExpect null
+
+                    expectValueParam.fir.defaultValue
+                }
+
                 if (!skipDefaultParameter && defaultValue != null) {
                     this.defaultValue = when {
                         forcedDefaultValueConversion && defaultValue !is FirExpressionStub ->