[FIR] Support of importing optional annotation classes (JS/Native), ^KT-51757 Fixed

compiler/fir/java/src/org/jetbrains/kotlin/fir/java/deserialization/JvmClassFileBasedSymbolProvider.kt

@@ -27,7 +27,7 @@ import org.jetbrains.kotlin.metadata.jvm.deserialization.JvmProtoBufUtil
 import org.jetbrains.kotlin.name.ClassId
 import org.jetbrains.kotlin.name.FqName
 import org.jetbrains.kotlin.resolve.jvm.JvmClassName
-import org.jetbrains.kotlin.serialization.deserialization.IncompatibleVersionErrorData
+import org.jetbrains.kotlin.serialization.deserialization.*
 import java.nio.file.Path
 import java.nio.file.Paths
 
@@ -132,7 +132,7 @@ class JvmClassFileBasedSymbolProvider(
             nameResolver,
             classProto,
             JvmBinaryAnnotationDeserializer(session, kotlinClass, kotlinClassFinder, result.byteContent),
-            kotlinClass.containingLibrary.toPath(),
+            moduleDataProvider.getModuleData(kotlinClass.containingLibrary?.toPath()),
             KotlinJvmBinarySourceElement(kotlinClass),
             classPostProcessor = { loadAnnotationsFromClassFile(result, it) }
         )

compiler/fir/java/src/org/jetbrains/kotlin/fir/java/deserialization/OptionalAnnotationClassesProvider.kt

@@ -0,0 +1,73 @@
+/*
+ * Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.fir.java.deserialization
+
+import org.jetbrains.kotlin.fir.FirSession
+import org.jetbrains.kotlin.fir.declarations.FirDeclarationOrigin
+import org.jetbrains.kotlin.fir.deserialization.AbstractFirDeserializedSymbolProvider
+import org.jetbrains.kotlin.fir.deserialization.FirDeserializationContext
+import org.jetbrains.kotlin.fir.deserialization.ModuleDataProvider
+import org.jetbrains.kotlin.fir.deserialization.PackagePartsCacheData
+import org.jetbrains.kotlin.fir.scopes.FirKotlinScopeProvider
+import org.jetbrains.kotlin.load.kotlin.PackagePartProvider
+import org.jetbrains.kotlin.metadata.ProtoBuf
+import org.jetbrains.kotlin.metadata.jvm.JvmProtoBuf
+import org.jetbrains.kotlin.metadata.jvm.deserialization.JvmFlags
+import org.jetbrains.kotlin.name.ClassId
+import org.jetbrains.kotlin.name.FqName
+import org.jetbrains.kotlin.serialization.deserialization.ClassData
+import org.jetbrains.kotlin.serialization.deserialization.getClassId
+
+class OptionalAnnotationClassesProvider(
+    session: FirSession,
+    moduleDataProvider: ModuleDataProvider,
+    kotlinScopeProvider: FirKotlinScopeProvider,
+    packagePartProvider: PackagePartProvider,
+    defaultDeserializationOrigin: FirDeclarationOrigin = FirDeclarationOrigin.Library
+) : AbstractFirDeserializedSymbolProvider(session, moduleDataProvider, kotlinScopeProvider, defaultDeserializationOrigin) {
+    private val optionalAnnotationClasses: Map<ClassId, ClassData>
+    private val optionalAnnotationPackages: Set<FqName>
+
+    init {
+        val optionalAnnotationClasses = mutableMapOf<ClassId, ClassData>()
+        val optionalAnnotationPackages = mutableSetOf<FqName>()
+
+        for (klass in packagePartProvider.getAllOptionalAnnotationClasses()) {
+            val classId = klass.nameResolver.getClassId(klass.classProto.fqName)
+            optionalAnnotationClasses[classId] = klass
+            optionalAnnotationPackages.add(classId.packageFqName)
+        }
+
+        this.optionalAnnotationClasses = optionalAnnotationClasses
+        this.optionalAnnotationPackages = optionalAnnotationPackages
+    }
+
+    override fun computePackagePartsInfos(packageFqName: FqName): List<PackagePartsCacheData> {
+        return emptyList()
+    }
+
+    override fun extractClassMetadata(
+        classId: ClassId,
+        parentContext: FirDeserializationContext?
+    ): ClassMetadataFindResult? {
+        val optionalAnnotationClass = optionalAnnotationClasses[classId] ?: return null
+
+        return ClassMetadataFindResult.Metadata(
+            optionalAnnotationClass.nameResolver,
+            optionalAnnotationClass.classProto,
+            null,
+            moduleDataProvider.allModuleData.last(),
+            null,
+            classPostProcessor = null
+        )
+    }
+
+    override fun isNewPlaceForBodyGeneration(classProto: ProtoBuf.Class): Boolean {
+        return JvmFlags.IS_COMPILED_IN_JVM_DEFAULT_MODE.get(classProto.getExtension(JvmProtoBuf.jvmClassFlags))
+    }
+
+    override fun getPackage(fqName: FqName): FqName? = if (optionalAnnotationPackages.contains(fqName)) fqName else null
+}