com.jeppeman.mockposable.compiler.MockposablePlugin.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockposable-compiler Show documentation

A tool that enables stubbing and verification of @Composable-annotated functions

The newest version!

package com.jeppeman.mockposable.compiler

import com.google.auto.service.AutoService
import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
import org.jetbrains.kotlin.cli.common.messages.CompilerMessageSeverity
import org.jetbrains.kotlin.cli.common.messages.CompilerMessageSourceLocation
import org.jetbrains.kotlin.cli.common.messages.MessageCollector
import org.jetbrains.kotlin.cli.common.messages.toLogger
import org.jetbrains.kotlin.com.intellij.mock.MockProject
import org.jetbrains.kotlin.com.intellij.openapi.extensions.LoadingOrder
import org.jetbrains.kotlin.compiler.plugin.ComponentRegistrar
import org.jetbrains.kotlin.compiler.plugin.ExperimentalCompilerApi
import org.jetbrains.kotlin.config.CompilerConfiguration

@OptIn(ExperimentalCompilerApi::class)
@Suppress("unused") // Invoked by kotlinc
@AutoService(ComponentRegistrar::class)
class MockposablePlugin : ComponentRegistrar {
    override val supportsK2: Boolean
        get() = true

    override fun registerProjectComponents(
        project: MockProject,
        configuration: CompilerConfiguration
    ) {
        val plugins = configuration.get(KEY_MOCK_PLUGINS, "").split(";")
        val messageCollector = MockposableMessageCollector(
            configuration.get(
                CLIConfigurationKeys.MESSAGE_COLLECTOR_KEY,
                MessageCollector.NONE
            )
        )

        for (extension in plugins) {
            // No extra transformations needed for compose-ui
            if (extension == "compose-ui") continue

            project.registerIrLast(
                when (extension) {
                    "mockito" -> MockitoIrGenerationExtension(messageCollector)
                    "mockk" -> MockKIrGenerationExtension(messageCollector)
                    else -> throw IllegalArgumentException("Unsupported plugin extension: $extension")
                }
            )
        }
    }
}

private fun MockProject.registerIrLast(extension: IrGenerationExtension) {
    extensionArea.getExtensionPoint(IrGenerationExtension.extensionPointName)
        .registerExtension(extension, LoadingOrder.LAST, this)
}

private class MockposableMessageCollector(
    private val proxy: MessageCollector
) : MessageCollector by proxy {
    override fun report(
        severity: CompilerMessageSeverity,
        message: String,
        location: CompilerMessageSourceLocation?
    ) {
        proxy.report(severity, "$LOG_TAG $message", location)
    }
}

private const val LOG_TAG = "[Mockposable]"