forked from square/dagger
-
Notifications
You must be signed in to change notification settings - Fork 2k
/
AggregatedRootIrValidator.kt
89 lines (84 loc) · 3.61 KB
/
AggregatedRootIrValidator.kt
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
/*
* Copyright (C) 2021 The Dagger Authors.
*
* 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 dagger.hilt.processor.internal.root.ir
import kotlin.jvm.Throws
// Validates roots being processed.
object AggregatedRootIrValidator {
@JvmStatic
@Throws(InvalidRootsException::class)
fun rootsToProcess(
isCrossCompilationRootValidationDisabled: Boolean,
processedRoots: Set<ProcessedRootSentinelIr>,
aggregatedRoots: Set<AggregatedRootIr>
): Set<AggregatedRootIr> {
val processedRootNames = processedRoots.flatMap { it.roots }.toSet()
val rootsToProcess =
aggregatedRoots.filterNot { processedRootNames.contains(it.root.canonicalName()) }.sortedBy {
it.root.canonicalName()
}
val testRootsToProcess = rootsToProcess.filter { it.isTestRoot }
val appRootsToProcess = rootsToProcess - testRootsToProcess
fun Collection<AggregatedRootIr>.rootsToString() = map { it.root }.joinToString()
if (appRootsToProcess.size > 1) {
throw InvalidRootsException(
"Cannot process multiple app roots in the same compilation unit: " +
appRootsToProcess.rootsToString()
)
}
if (testRootsToProcess.isNotEmpty() && appRootsToProcess.isNotEmpty()) {
throw InvalidRootsException(
"""
Cannot process test roots and app roots in the same compilation unit:
App root in this compilation unit: ${appRootsToProcess.rootsToString()}
Test roots in this compilation unit: ${testRootsToProcess.rootsToString()}
""".trimIndent()
)
}
// Perform validation across roots previous compilation units.
if (!isCrossCompilationRootValidationDisabled) {
val alreadyProcessedTestRoots =
aggregatedRoots.filter {
it.isTestRoot && processedRootNames.contains(it.root.canonicalName())
}
// TODO(b/185742783): Add an explanation or link to docs to explain why we're forbidding this.
if (alreadyProcessedTestRoots.isNotEmpty() && rootsToProcess.isNotEmpty()) {
throw InvalidRootsException(
"""
Cannot process new roots when there are test roots from a previous compilation unit:
Test roots from previous compilation unit: ${alreadyProcessedTestRoots.rootsToString()}
All roots from this compilation unit: ${rootsToProcess.rootsToString()}
""".trimIndent()
)
}
val alreadyProcessedAppRoots =
aggregatedRoots.filter {
!it.isTestRoot && processedRootNames.contains(it.root.canonicalName())
}
if (alreadyProcessedAppRoots.isNotEmpty() && appRootsToProcess.isNotEmpty()) {
throw InvalidRootsException(
"""
Cannot process new app roots when there are app roots from a previous compilation unit:
App roots in previous compilation unit: ${alreadyProcessedAppRoots.rootsToString()}
App roots in this compilation unit: ${appRootsToProcess.rootsToString()}
""".trimIndent()
)
}
}
return rootsToProcess.toSet()
}
}
// An exception thrown when root validation fails.
class InvalidRootsException(msg: String) : Exception(msg)