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 package org.jetbrains.kotlin.allopen.gradle
import org.gradle.api.Plugin
import org.gradle.api.Project 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.gradle.tooling.provider.model.ToolingModelBuilderRegistry
import org.jetbrains.kotlin.allopen.gradle.model.builder.AllOpenModelBuilder import org.jetbrains.kotlin.allopen.gradle.model.builder.AllOpenModelBuilder
import org.jetbrains.kotlin.gradle.plugin.* import org.jetbrains.kotlin.gradle.plugin.*
import javax.inject.Inject 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 { companion object {
fun isEnabled(project: Project) = project.plugins.findPlugin(AllOpenGradleSubplugin::class.java) != null
fun getAllOpenExtension(project: Project): AllOpenExtension { fun getAllOpenExtension(project: Project): AllOpenExtension {
return project.extensions.getByType(AllOpenExtension::class.java) 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) { override fun apply(project: Project) {
project.extensions.create("allOpen", AllOpenExtension::class.java) project.extensions.create("allOpen", AllOpenExtension::class.java)
registry.register(AllOpenModelBuilder()) registry.register(AllOpenModelBuilder())
} }
}
class AllOpenKotlinGradleSubplugin : KotlinGradleSubplugin<AbstractCompile> { override fun isApplicable(kotlinCompilation: KotlinCompilation<*>): Boolean = true
companion object {
const val ALLOPEN_ARTIFACT_NAME = "kotlin-allopen"
private val ANNOTATION_ARG_NAME = "annotation" override fun applyToCompilation(
private val PRESET_ARG_NAME = "preset" 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( return project.provider {
project: Project, val options = mutableListOf<SubpluginOption>()
kotlinCompile: AbstractCompile,
javaCompile: AbstractCompile?,
variantData: Any?,
androidProjectHandler: Any?,
kotlinCompilation: KotlinCompilation<*>?
): List<SubpluginOption> {
if (!AllOpenGradleSubplugin.isEnabled(project)) return emptyList()
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
options += SubpluginOption(ANNOTATION_ARG_NAME, anno)
} }
for (preset in allOpenExtension.myPresets) {
options += SubpluginOption(PRESET_ARG_NAME, preset)
}
return options
} }
override fun getCompilerPluginId() = "org.jetbrains.kotlin.allopen" override fun getCompilerPluginId() = "org.jetbrains.kotlin.allopen"
@@ -1 +0,0 @@
org.jetbrains.kotlin.allopen.gradle.AllOpenKotlinGradleSubplugin