Skip to content

Prevent K2 warning from showing for KSP users #98

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ZacSweers merged 1 commit into from
May 1, 2026
Merged

Prevent K2 warning from showing for KSP users #98

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
91 changes: 61 additions & 30 deletions gradle-plugin/src/main/java/com/squareup/anvil/plugin/AnvilPlugin.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,12 @@ import org.gradle.api.logging.Logger
import org.gradle.api.provider.Provider
import org.gradle.api.tasks.TaskContainer
import org.gradle.api.tasks.TaskProvider
import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
import org.jetbrains.kotlin.gradle.dsl.KotlinProjectExtension
import org.jetbrains.kotlin.gradle.dsl.KotlinSingleTargetExtension
import org.jetbrains.kotlin.gradle.dsl.KotlinVersion.KOTLIN_1_9
import org.jetbrains.kotlin.gradle.dsl.KotlinVersion.KOTLIN_2_0
import org.jetbrains.kotlin.gradle.dsl.kotlinExtension
import org.jetbrains.kotlin.gradle.internal.KaptGenerateStubsTask
import org.jetbrains.kotlin.gradle.plugin.FilesSubpluginOption
import org.jetbrains.kotlin.gradle.plugin.KaptExtension
Expand Down Expand Up @@ -98,12 +102,60 @@ internal open class AnvilPlugin : KotlinCompilerPluginSupportPlugin {
getConfiguration(target, "main").extendsFrom(commonConfiguration)
}
}

target.afterEvaluate {
target.kotlinExtension.targets
.flatMap { it.compilations }
.forEach { kotlinCompilation ->
val variant = getVariant(kotlinCompilation)
val variantFilter = variant.variantFilter

if (!variantFilter.generateDaggerFactories &&
variantFilter.generateDaggerFactoriesOnly
) {
throw GradleException(
"You cannot set generateDaggerFactories to false and generateDaggerFactoriesOnly " +
"to true at the same time for variant ${variant.name}.",
)
}

if (!variantFilter.generateDaggerFactoriesOnly) {
target.dependencies.add(
kotlinCompilation.compileOnlyConfigurationName,
"$GROUP:annotations:$VERSION",
)
kotlinCompilation.substituteDependencies(
target,
"com.squareup.anvil:annotations",
"$GROUP:annotations",
)
}
if (variantFilter.addOptionalAnnotations) {
target.dependencies.add(
kotlinCompilation.compileOnlyConfigurationName,
"$GROUP:annotations-optional:$VERSION",
)
kotlinCompilation.substituteDependencies(
target,
"com.squareup.anvil:annotations-optional",
"$GROUP:annotations-optional",
)
}
}
}
}

override fun isApplicable(kotlinCompilation: KotlinCompilation<*>): Boolean {
return when (kotlinCompilation.platformType) {
// If the variant is ignored, then don't apply the compiler plugin.
androidJvm, jvm -> !getVariant(kotlinCompilation).variantFilter.ignore
androidJvm, jvm -> {
val variantFilter = getVariant(kotlinCompilation).variantFilter
if (variantFilter.useKspBackend || variantFilter.useKspComponentMergingBackend) {
false
} else {
// If the variant is ignored, then don't apply the compiler plugin.
!variantFilter.ignore
}
}
else -> false
}
}
Expand All @@ -125,15 +177,6 @@ internal open class AnvilPlugin : KotlinCompilerPluginSupportPlugin {
val variant = getVariant(kotlinCompilation)
val project = variant.project

if (!variant.variantFilter.generateDaggerFactories &&
variant.variantFilter.generateDaggerFactoriesOnly
) {
throw GradleException(
"You cannot set generateDaggerFactories to false and generateDaggerFactoriesOnly " +
"to true at the same time for variant ${variant.name}.",
)
}

// Make the kotlin compiler classpath extend our configurations to pick up our extra
// generators.
project.configurations.getByName(variant.compilerPluginClasspathName)
Expand All @@ -143,25 +186,6 @@ internal open class AnvilPlugin : KotlinCompilerPluginSupportPlugin {

if (!variant.variantFilter.generateDaggerFactoriesOnly) {
disableCorrectErrorTypes(variant)

kotlinCompilation.dependencies {
compileOnly("$GROUP:annotations:$VERSION")
}
kotlinCompilation.substituteDependencies(
project,
"com.squareup.anvil:annotations",
"$GROUP:annotations",
)
}
if (variant.variantFilter.addOptionalAnnotations) {
kotlinCompilation.dependencies {
compileOnly("$GROUP:annotations-optional:$VERSION")
}
kotlinCompilation.substituteDependencies(
project,
"com.squareup.anvil:annotations-optional",
"$GROUP:annotations-optional",
)
}

// Notice that we use the name of the variant as a directory name. Generated code
Expand Down Expand Up @@ -457,6 +481,13 @@ private val agpPlugins = listOf(
"com.android.dynamic-feature",
)

private val KotlinProjectExtension.targets: Iterable<KotlinTarget>
get() = when (this) {
is KotlinSingleTargetExtension<*> -> listOf(this.target)
is KotlinMultiplatformExtension -> targets
else -> error("Unexpected 'kotlin' extension $this")
}

private const val KAPT_PLUGIN_ID = "org.jetbrains.kotlin.kapt"

internal fun String.capitalize(): String = replaceFirstChar(Char::uppercaseChar)
Expand Down
Loading