org.jetbrains.kotlin.assignment.plugin.gradle.AssignmentSubplugin.kt Maven / Gradle / Ivy
/*
* Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
*/
package org.jetbrains.kotlin.assignment.plugin.gradle
import org.gradle.api.Project
import org.gradle.api.provider.Provider
import org.gradle.tooling.provider.model.ToolingModelBuilderRegistry
import org.jetbrains.kotlin.assignment.plugin.gradle.model.builder.AssignmentModelBuilder
import org.jetbrains.kotlin.gradle.plugin.*
import javax.inject.Inject
class AssignmentSubplugin
@Inject internal constructor(
private val registry: ToolingModelBuilderRegistry
) : KotlinCompilerPluginSupportPlugin {
companion object {
const val COMPILER_PLUGIN_ARTIFACT_NAME = "kotlin-assignment-compiler-plugin-embeddable"
const val COMPILER_PLUGIN_ID = "org.jetbrains.kotlin.assignment"
private const val ANNOTATION_ARG_NAME = "annotation"
}
override fun apply(target: Project) {
target.extensions.create("assignment", AssignmentExtension::class.java)
registry.register(AssignmentModelBuilder())
}
override fun isApplicable(kotlinCompilation: KotlinCompilation<*>): Boolean = true
override fun applyToCompilation(kotlinCompilation: KotlinCompilation<*>): Provider> {
val project = kotlinCompilation.target.project
val extension = project.extensions.findByType(AssignmentExtension::class.java)
?: return project.provider { emptyList() }
return project.provider {
val options = mutableListOf()
for (anno in extension.myAnnotations) {
options += SubpluginOption(ANNOTATION_ARG_NAME, anno)
}
options
}
}
override fun getCompilerPluginId(): String = COMPILER_PLUGIN_ID
override fun getPluginArtifact(): SubpluginArtifact = JetBrainsSubpluginArtifact(artifactId = COMPILER_PLUGIN_ARTIFACT_NAME)
}