software.amazon.lastmile.kotlin.inject.anvil.KotlinInjectExtensionSymbolProcessorProvider.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of compiler Show documentation
Extensions for kotlin-inject to make dependency injection easier with a similar feature set as Anvil.
The newest version!
package software.amazon.lastmile.kotlin.inject.anvil

import com.google.auto.service.AutoService
import com.google.devtools.ksp.processing.SymbolProcessor
import com.google.devtools.ksp.processing.SymbolProcessorEnvironment
import com.google.devtools.ksp.processing.SymbolProcessorProvider
import software.amazon.lastmile.kotlin.inject.anvil.processor.ContributesBindingProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.ContributesSubcomponentFactoryProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.ContributesSubcomponentProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.ContributesToProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.CreateComponentProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.GenerateKotlinInjectComponentProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.MergeComponentProcessor
import software.amazon.lastmile.kotlin.inject.anvil.processor.extend.ContributingAnnotationProcessor

/**
 * Entry point for KSP to pick up our [SymbolProcessor].
 */
@AutoService(SymbolProcessorProvider::class)
@Suppress("unused")
class KotlinInjectExtensionSymbolProcessorProvider : SymbolProcessorProvider {
    override fun create(environment: SymbolProcessorEnvironment): SymbolProcessor {
        fun MutableSet.addIfEnabled(symbolProcessor: SymbolProcessor) {
            val disabled = environment.options[symbolProcessor::class.qualifiedName] == "disabled"
            if (!disabled) {
                add(symbolProcessor)
            } else {
                environment.logger
                    .info("Disabled kotlin-inject-anvil processor ${symbolProcessor::class}")
            }
        }

        val symbolProcessors = buildSet {
            addIfEnabled(
                ContributesToProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                ),
            )
            addIfEnabled(
                ContributesBindingProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                ),
            )
            addIfEnabled(
                ContributesSubcomponentFactoryProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                ),
            )
            addIfEnabled(
                MergeComponentProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                    contributesSubcomponentProcessor = ContributesSubcomponentProcessor(
                        codeGenerator = environment.codeGenerator,
                        logger = environment.logger,
                    ),
                    options = environment.options,
                ),
            )
            addIfEnabled(
                ContributingAnnotationProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                ),
            )
            addIfEnabled(
                GenerateKotlinInjectComponentProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                ),
            )
            addIfEnabled(
                CreateComponentProcessor(
                    codeGenerator = environment.codeGenerator,
                    logger = environment.logger,
                ),
            )
        }

        return CompositeSymbolProcessor(symbolProcessors)
    }
}