1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
/*
* Copyright 2021 Google LLC
* Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.google.devtools.ksp.gradle
import com.android.build.api.dsl.CommonExtension
import com.android.build.gradle.BaseExtension
import com.android.build.gradle.api.AndroidSourceSet
import org.gradle.api.Named
import org.gradle.api.Project
import org.gradle.api.file.FileCollection
import org.gradle.api.tasks.TaskProvider
import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinJvmAndroidCompilation
import java.io.File
/**
* This helper class handles communication with the android plugin.
* It is isolated in a separate class to avoid adding dependency on the android plugin.
* Instead, we add a compileOnly dependency to the Android Plugin, which means we can still function
* without the Android plugin. The downside is that we need to ensure never to access Android
* plugin APIs directly without checking its existence (we have tests covering that case).
*/
@Suppress("UnstableApiUsage") // some android APIs are unsable.
object AndroidPluginIntegration {
private val agpPluginIds = listOf("com.android.application", "com.android.library", "com.android.dynamic-feature")
fun findSourceSets(project: Project, onSourceSet: (String) -> Unit) {
agpPluginIds.forEach { agpPluginId ->
project.pluginManager.withPlugin(agpPluginId) {
// for android apps, we need a configuration per source set
decorateAndroidExtension(project, onSourceSet)
}
}
}
private fun decorateAndroidExtension(project: Project, onSourceSet: (String) -> Unit) {
val sourceSets = when (val androidExt = project.extensions.getByName("android")) {
is BaseExtension -> androidExt.sourceSets
is CommonExtension<*, *, *, *> -> androidExt.sourceSets
else -> throw RuntimeException("Unsupported Android Gradle plugin version.")
}
sourceSets.configureEach {
onSourceSet(it.name)
}
}
fun getCompilationSourceSets(kotlinCompilation: KotlinJvmAndroidCompilation): List<String> {
return kotlinCompilation.androidVariant
.sourceSets
.filterIsInstance(AndroidSourceSet::class.java)
.map { it.name }
}
fun registerGeneratedJavaSources(
project: Project,
kotlinCompilation: KotlinJvmAndroidCompilation,
kspTaskProvider: TaskProvider<KspTaskJvm>,
javaOutputDir: File,
classOutputDir: File,
resourcesOutputDir: FileCollection,
) {
val kspJavaOutput = project.fileTree(javaOutputDir).builtBy(kspTaskProvider)
val kspClassOutput = project.fileTree(classOutputDir).builtBy(kspTaskProvider)
kspJavaOutput.include("**/*.java")
kspClassOutput.include("**/*.class")
kotlinCompilation.androidVariant.registerExternalAptJavaOutput(kspJavaOutput)
kotlinCompilation.androidVariant.registerPreJavacGeneratedBytecode(kspClassOutput)
kotlinCompilation.androidVariant.registerPostJavacGeneratedBytecode(resourcesOutputDir)
}
}
|