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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
/*
* 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.testing
import java.io.File
/**
* Simpler wrapper that represents a module in a test project.
*
* It has helper methods to add plugins, dependencies or sources to the project.
*
* It is loosely structured, which means it won't check its integrity. (e.g. you can add android
* sources w/o adding android plugin). It is meant to serve as a convenient way to configure
* modules.
*/
class TestModule(
val moduleRoot: File,
plugins: List<PluginDeclaration> = emptyList(),
dependencies: List<DependencyDeclaration> = emptyList()
) {
val plugins = LinkedHashSet(plugins)
val dependencies = LinkedHashSet(dependencies)
val buildFileAdditions = LinkedHashSet<String>()
val name = moduleRoot.name
init {
moduleRoot.mkdirs()
}
/**
* Adds the given source file to the main source set.
*/
fun addSource(name: String, contents: String) {
val srcDir = when {
name.endsWith(".kt") -> kotlinSourceDir
name.endsWith(".java") -> javaSourceDir
else -> error("must provide java or kotlin file")
}
srcDir.resolve(name).writeText(contents)
}
/**
* Adds the given source file to the test source set.
*/
fun addTestSource(name: String, contents: String) {
val srcDir = when {
name.endsWith(".kt") -> kotlinTestSourceDir
name.endsWith(".java") -> javaTestSourceDir
else -> error("must provide java or kotlin file")
}
srcDir.resolve(name).writeText(contents)
}
/**
* Adds the given source file to the AndroidTest source set.
*/
fun addAndroidTestSource(name: String, contents: String) {
val srcDir = when {
name.endsWith(".kt") -> kotlinAndroidTestSourceDir
name.endsWith(".java") -> javaAndroidTestSourceDir
else -> error("must provide java or kotlin file")
}
srcDir.resolve(name).writeText(contents)
}
/**
* Adds the given source file to the given KotlinSourceSet in a multiplatform project.
*/
fun addMultiplatformSource(sourceSet: String, name: String, contents: String) {
require(name.endsWith(".kt")) { "multiplatform source extension needs to be *.kt." }
val srcDir = multiplatformKotlinSourceDir(sourceSet)
srcDir.resolve(name).writeText(contents)
}
private fun multiplatformKotlinSourceDir(sourceSet: String) = moduleRoot.resolve("src/$sourceSet/kotlin").also {
it.mkdirs()
}
private val kotlinSourceDir
get() = moduleRoot.resolve("src/main/kotlin").also {
it.mkdirs()
}
private val javaSourceDir
get() = moduleRoot.resolve("src/main/java").also {
it.mkdirs()
}
private val kotlinTestSourceDir
get() = moduleRoot.resolve("src/test/kotlin").also {
it.mkdirs()
}
private val javaTestSourceDir
get() = moduleRoot.resolve("src/test/java").also {
it.mkdirs()
}
private val kotlinAndroidTestSourceDir
get() = moduleRoot.resolve("src/androidTest/kotlin").also {
it.mkdirs()
}
private val javaAndroidTestSourceDir
get() = moduleRoot.resolve("src/androidTest/java").also {
it.mkdirs()
}
private val servicesDir
get() = moduleRoot.resolve("src/main/resources/META-INF/services/").also {
it.mkdirs()
}
val kspServicesFile
get() = servicesDir.resolve("com.google.devtools.ksp.processing.SymbolProcessorProvider")
private val buildFile
get() = moduleRoot.resolve("build.gradle.kts")
/**
* Writes the build file.
*/
fun writeBuildFile() {
val contents = buildString {
appendln("plugins {")
plugins.forEach { plugin ->
appendln(plugin.toCode().prependIndent(" "))
}
appendln("}")
appendln("dependencies {")
dependencies.forEach { dependency ->
appendln(dependency.toCode().prependIndent(" "))
}
appendln("}")
buildFileAdditions.forEach {
appendln(it)
}
}
buildFile.writeText(contents)
}
}
|
