• Home
• dev.mokkery
• mokkery-plugin
• 2.6.1
• source code
• MokkeryClassGenerationExtension.kt

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

dev.mokkery.plugin.jvm.MokkeryClassGenerationExtension.kt Maven / Gradle / Ivy

package dev.mokkery.plugin.jvm

import dev.mokkery.plugin.core.Mokkery
import org.jetbrains.kotlin.backend.jvm.extensions.ClassGenerator
import org.jetbrains.kotlin.backend.jvm.extensions.ClassGeneratorExtension
import org.jetbrains.kotlin.config.CompilerConfiguration
import org.jetbrains.kotlin.config.messageCollector
import org.jetbrains.kotlin.ir.declarations.IrClass

class MokkeryClassGenerationExtension(
    configuration: CompilerConfiguration
) : ClassGeneratorExtension {

    private val collector = configuration.messageCollector

    override fun generateClass(generator: ClassGenerator, declaration: IrClass?): ClassGenerator {
        if (declaration == null) return generator
        if (declaration.origin != Mokkery.Origin) return generator
        return JvmParamAssertionSkippingGenerator(generator, collector)
    }
}