[K2/JS] Use declaration session for looking up containing declaration

There was a couple of helper functions,
in particular hasAnnotationOrInsideAnnotatedClass,
with looked up containing class or file of declaration,
using call-site session.

This is incorrect because it can find a file of actual declaration,
instead of expect one, with unpredictable result.

^KT-67978 Fixed

Author: kunyavskiy

compiler/fir/checkers/checkers.js/src/org/jetbrains/kotlin/fir/analysis/js/checkers/FirJsHelpers.kt

@@ -122,14 +122,14 @@ fun FirBasedSymbol<*>.isExportedObject(session: FirSession): Boolean {
     return when {
         hasAnnotationOrInsideAnnotatedClass(JsStandardClassIds.Annotations.JsExportIgnore, session) -> false
         hasAnnotationOrInsideAnnotatedClass(JsStandardClassIds.Annotations.JsExport, session) -> true
-        else -> getContainingFile(session)?.hasAnnotation(JsStandardClassIds.Annotations.JsExport, session) == true
+        else -> getContainingFile()?.hasAnnotation(JsStandardClassIds.Annotations.JsExport, session) == true
     }
 }
 
-internal fun FirBasedSymbol<*>.getContainingFile(session: FirSession): FirFile? {
+internal fun FirBasedSymbol<*>.getContainingFile(): FirFile? {
     return when (this) {
-        is FirCallableSymbol<*> -> session.firProvider.getFirCallableContainerFile(this)
-        is FirClassLikeSymbol<*> -> session.firProvider.getFirClassifierContainerFileIfAny(this)
+        is FirCallableSymbol<*> -> moduleData.session.firProvider.getFirCallableContainerFile(this)
+        is FirClassLikeSymbol<*> -> moduleData.session.firProvider.getFirClassifierContainerFileIfAny(this)
         else -> return null
     }
 }

compiler/fir/checkers/checkers.js/src/org/jetbrains/kotlin/fir/analysis/js/checkers/FirJsModuleCheckUtils.kt

@@ -37,7 +37,7 @@ internal fun checkJsModuleUsage(
 
     val calleeSession = callee.moduleData.session
     val calleeRoot = getRootClassLikeSymbolOrSelf(callee, calleeSession)
-    val calleeContainingFile = calleeRoot.getContainingFile(calleeSession)
+    val calleeContainingFile = calleeRoot.getContainingFile()
 
     val callToModule = calleeRoot.getAnnotationStringParameter(JsStandardClassIds.Annotations.JsModule, calleeSession) != null ||
             calleeContainingFile?.symbol?.getAnnotationStringParameter(JsStandardClassIds.Annotations.JsModule, calleeSession) != null

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/checkers/FirHelpers.kt

@@ -791,7 +791,7 @@ val CheckerContext.isTopLevel get() = containingDeclarations.lastOrNull().let {
 
 fun FirBasedSymbol<*>.hasAnnotationOrInsideAnnotatedClass(classId: ClassId, session: FirSession): Boolean {
     if (hasAnnotation(classId, session)) return true
-    val container = getContainingClassSymbol(session) ?: return false
+    val container = getContainingClassSymbol(moduleData.session) ?: return false
     return container.hasAnnotationOrInsideAnnotatedClass(classId, session)
 }
 

compiler/testData/diagnostics/tests/multiplatform/expectsWithJsExport.fir.kt

@@ -13,26 +13,26 @@ expect class <!WRONG_EXPORTED_DECLARATION!>WithExportOnExpect<!> {
 }
 
 expect class WithExportOnActual {
-    <!WRONG_EXPORTED_DECLARATION!>fun foo()<!>
-    <!WRONG_EXPORTED_DECLARATION, WRONG_EXPORTED_DECLARATION!>val bar: Int<!>
+    fun foo()
+    val bar: Int
 }
 
 expect class WithExportTypealiasOnActual {
-    <!WRONG_EXPORTED_DECLARATION!>fun foo()<!>
-    <!WRONG_EXPORTED_DECLARATION, WRONG_EXPORTED_DECLARATION!>val bar: Int<!>
+    fun foo()
+    val bar: Int
 }
 
-expect class <!WRONG_EXPORTED_DECLARATION!>WithFileExportOnActual<!> {
+expect class WithFileExportOnActual {
     fun foo()
     val bar: Int
 }
 
 // FILE: common2.kt
 @file:Export
 
-expect class WithExportOnExpectFile {
-    fun foo()
-    val bar: Int
+expect class <!WRONG_EXPORTED_DECLARATION!>WithExportOnExpectFile<!> {
+    <!WRONG_EXPORTED_DECLARATION!>fun foo()<!>
+    <!WRONG_EXPORTED_DECLARATION, WRONG_EXPORTED_DECLARATION!>val bar: Int<!>
 }
 
 // MODULE: m1-js()()(m1-common)