KotlinExtractorComponentRegistrar.kt

// For ComponentRegistrar
@file:Suppress("DEPRECATION")

package com.github.codeql

import com.intellij.mock.MockProject
import com.intellij.openapi.extensions.LoadingOrder
import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
import org.jetbrains.kotlin.config.CompilerConfiguration

class KotlinExtractorComponentRegistrar : Kotlin2ComponentRegistrar() {
    override fun registerProjectComponents(
        project: MockProject,
        configuration: CompilerConfiguration
    ) {
        val invocationTrapFile = configuration[KEY_INVOCATION_TRAP_FILE]
        if (invocationTrapFile == null) {
            throw Exception("Required argument for TRAP invocation file not given")
        }
        // Register with LoadingOrder.LAST to ensure the extractor runs after other
        // IR generation plugins (like kotlinx.serialization) have generated their code.
        val extensionPoint = project.extensionArea.getExtensionPoint(IrGenerationExtension.extensionPointName)
        extensionPoint.registerExtension(
            KotlinExtractorExtension(
                invocationTrapFile,
                configuration[KEY_CHECK_TRAP_IDENTICAL] ?: false,
                configuration[KEY_COMPILATION_STARTTIME],
                configuration[KEY_EXIT_AFTER_EXTRACTION] ?: false
            ),
            LoadingOrder.LAST,
            project
        )
    }
}