[FIR] Add extension point to metadata serialization

This is a temporary solution for KT-64312
Introduced extension point should be removed (KT-64695)
This commit is contained in:
Dmitriy Novozhilov
2024-01-03 12:19:42 +02:00
committed by Space Team
parent 717d1c774b
commit fb3963dead
3 changed files with 86 additions and 0 deletions
@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.fir.analysis.extensions.FirAdditionalCheckersExtensi
import org.jetbrains.kotlin.fir.backend.Fir2IrScriptConfiguratorExtension
import org.jetbrains.kotlin.fir.builder.FirScriptConfiguratorExtension
import org.jetbrains.kotlin.fir.resolve.FirSamConversionTransformerExtension
import org.jetbrains.kotlin.fir.serialization.FirMetadataSerializerPlugin
import kotlin.reflect.KClass
abstract class FirExtensionRegistrar : FirExtensionRegistrarAdapter() {
@@ -34,6 +35,8 @@ abstract class FirExtensionRegistrar : FirExtensionRegistrarAdapter() {
FirScriptConfiguratorExtension::class,
Fir2IrScriptConfiguratorExtension::class,
FirFunctionTypeKindExtension::class,
@OptIn(FirExtensionApiInternals::class)
FirMetadataSerializerPlugin::class,
)
internal val ALLOWED_EXTENSIONS_FOR_LIBRARY_SESSION = listOf(
@@ -107,6 +110,12 @@ abstract class FirExtensionRegistrar : FirExtensionRegistrarAdapter() {
registerExtension(FirFunctionTypeKindExtension::class, this)
}
@FirExtensionApiInternals
@JvmName("plusMetadataSerializerPlugin")
operator fun (FirMetadataSerializerPlugin.Factory).unaryPlus() {
registerExtension(FirMetadataSerializerPlugin::class, this)
}
// ------------------ reference methods ------------------
@JvmName("plusStatusTransformerExtension")
@@ -169,6 +178,12 @@ abstract class FirExtensionRegistrar : FirExtensionRegistrarAdapter() {
FirFunctionTypeKindExtension.Factory { this.invoke(it) }.unaryPlus()
}
@FirExtensionApiInternals
@JvmName("plusMetadataSerializerPlugin")
operator fun ((FirSession) -> FirMetadataSerializerPlugin).unaryPlus() {
FirMetadataSerializerPlugin.Factory { this.invoke(it) }.unaryPlus()
}
// ------------------ utilities ------------------
@JvmName("bindLeft")