app.softwork.serviceloader.ServiceLoaderPlugin.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gradle-plugin Show documentation
A Gradle plugin to generate and validate service loaders
The newest version!
package app.softwork.serviceloader

import org.gradle.api.Project
import org.gradle.api.provider.Provider
import org.gradle.kotlin.dsl.listProperty
import org.jetbrains.kotlin.gradle.dsl.KotlinJvmProjectExtension
import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
import org.jetbrains.kotlin.gradle.plugin.*

public class ServiceLoaderPlugin : KotlinCompilerPluginSupportPlugin {
    override fun apply(target: Project) {
        super.apply(target)

        target.plugins.withId("org.jetbrains.kotlin.multiplatform") {
            val kotlin = target.extensions.getByType(KotlinMultiplatformExtension::class.java)
            kotlin.sourceSets.configureEach {
                dependencies {
                    implementation(runtimeDependency())
                }
            }
        }
        target.plugins.withId("org.jetbrains.kotlin.jvm") {
            val kotlin = target.extensions.getByType(KotlinJvmProjectExtension::class.java)
            kotlin.sourceSets.configureEach {
                dependencies {
                    implementation(runtimeDependency())
                }
            }
        }
    }

    private fun runtimeDependency() = "app.softwork.serviceloader:ksp-annotation:$VERSION"

    override fun isApplicable(kotlinCompilation: KotlinCompilation<*>): Boolean =
        kotlinCompilation.platformType == KotlinPlatformType.jvm

    override fun getCompilerPluginId(): String = "app.softwork.serviceloader"

    override fun getPluginArtifact(): SubpluginArtifact = SubpluginArtifact(
        groupId = "app.softwork.serviceloader",
        artifactId = "kotlin-plugin",
        version = VERSION,
    )

    override fun applyToCompilation(kotlinCompilation: KotlinCompilation<*>): Provider> {
        val project = kotlinCompilation.project
        val kotlinSourceSet = kotlinCompilation.defaultSourceSet

        val outputDir = project.layout.buildDirectory.dir(
            "generated/serviceloader/${kotlinSourceSet.name}/resources"
        )

        val compileTaskProvider = kotlinCompilation.compileTaskProvider
        compileTaskProvider.configure {
            outputs.dir(outputDir)
        }

        val outputDirWithTask = project.files(outputDir) {
            builtBy(compileTaskProvider)
        }

        kotlinSourceSet.resources.srcDir(outputDirWithTask)

        val options = project.objects.listProperty()
        options.add(
            FilesSubpluginOption(
                key = "outputDir",
                files = listOf(outputDir.get().asFile),
            )
        )
        return options
    }
}