Rework AllOpenSubplugin for the new subplugins API

This commit is contained in:
Sergey Igushkin
2020-06-06 00:25:47 +03:00
parent e8a303650c
commit f2bc391bdd
2 changed files with 23 additions and 33 deletions
@@ -16,62 +16,53 @@
package org.jetbrains.kotlin.allopen.gradle
import org.gradle.api.Plugin
import org.gradle.api.Project
import org.gradle.api.tasks.compile.AbstractCompile
import org.gradle.api.provider.Provider
import org.gradle.tooling.provider.model.ToolingModelBuilderRegistry
import org.jetbrains.kotlin.allopen.gradle.model.builder.AllOpenModelBuilder
import org.jetbrains.kotlin.gradle.plugin.*
import javax.inject.Inject
class AllOpenGradleSubplugin @Inject internal constructor(private val registry: ToolingModelBuilderRegistry) : Plugin<Project> {
class AllOpenGradleSubplugin @Inject internal constructor(private val registry: ToolingModelBuilderRegistry) :
KotlinCompilerPluginSupportPlugin {
companion object {
fun isEnabled(project: Project) = project.plugins.findPlugin(AllOpenGradleSubplugin::class.java) != null
fun getAllOpenExtension(project: Project): AllOpenExtension {
return project.extensions.getByType(AllOpenExtension::class.java)
}
private const val ALLOPEN_ARTIFACT_NAME = "kotlin-allopen"
private val ANNOTATION_ARG_NAME = "annotation"
private val PRESET_ARG_NAME = "preset"
}
override fun apply(project: Project) {
project.extensions.create("allOpen", AllOpenExtension::class.java)
registry.register(AllOpenModelBuilder())
}
}
class AllOpenKotlinGradleSubplugin : KotlinGradleSubplugin<AbstractCompile> {
companion object {
const val ALLOPEN_ARTIFACT_NAME = "kotlin-allopen"
override fun isApplicable(kotlinCompilation: KotlinCompilation<*>): Boolean = true
private val ANNOTATION_ARG_NAME = "annotation"
private val PRESET_ARG_NAME = "preset"
}
override fun applyToCompilation(
kotlinCompilation: KotlinCompilation<*>
): Provider<List<SubpluginOption>> {
val project = kotlinCompilation.target.project
override fun isApplicable(project: Project, task: AbstractCompile) = AllOpenGradleSubplugin.isEnabled(project)
val allOpenExtension = project.extensions.getByType(AllOpenExtension::class.java)
override fun apply(
project: Project,
kotlinCompile: AbstractCompile,
javaCompile: AbstractCompile?,
variantData: Any?,
androidProjectHandler: Any?,
kotlinCompilation: KotlinCompilation<*>?
): List<SubpluginOption> {
if (!AllOpenGradleSubplugin.isEnabled(project)) return emptyList()
return project.provider {
val options = mutableListOf<SubpluginOption>()
val allOpenExtension = project.extensions.findByType(AllOpenExtension::class.java) ?: return emptyList()
for (anno in allOpenExtension.myAnnotations) {
options += SubpluginOption(ANNOTATION_ARG_NAME, anno)
}
val options = mutableListOf<SubpluginOption>()
for (preset in allOpenExtension.myPresets) {
options += SubpluginOption(PRESET_ARG_NAME, preset)
}
for (anno in allOpenExtension.myAnnotations) {
options += SubpluginOption(ANNOTATION_ARG_NAME, anno)
options
}
for (preset in allOpenExtension.myPresets) {
options += SubpluginOption(PRESET_ARG_NAME, preset)
}
return options
}
override fun getCompilerPluginId() = "org.jetbrains.kotlin.allopen"
@@ -1 +0,0 @@
org.jetbrains.kotlin.allopen.gradle.AllOpenKotlinGradleSubplugin