@@ -23,6 +23,14 @@ class ScaleDimensPlugin : Plugin<Project> {
2323 val resSourceDirectoriesBySourceSet = mutableMapOf<String , Set <File >>()
2424
2525 androidComponents.apply {
26+
27+ finalizeDsl { common ->
28+ common.sourceSets
29+ .map { sourceSet -> sourceSet.name to (sourceSet.res as AndroidSourceDirectorySet ).srcDirs }
30+ .forEach { resSourceDirectoriesBySourceSet[it.first] = it.second }
31+
32+ }
33+
2634 onVariants { variant ->
2735 val relevantSourcesSets = setOfNotNull(
2836 " main" ,
@@ -43,24 +51,16 @@ class ScaleDimensPlugin : Plugin<Project> {
4351 }
4452 .flatten()
4553
46- variant.sources.res?.let {
47- val addSourceTaskProvider =
48- project.tasks.register<ScaleDimensTask >(" scaleDimens${variant.name} " ) {
49- group = " scale-dimens"
50- this .resourceDirectories.set(resSourceDirectories)
51- this .extension.set(extension)
52- }
53- it.addGeneratedSourceDirectory(
54- addSourceTaskProvider,
55- ScaleDimensTask ::outputFolder
56- )
57- }
58- }
59- finalizeDsl { common ->
60- common.sourceSets
61- .map { sourceSet -> sourceSet.name to (sourceSet.res as AndroidSourceDirectorySet ).srcDirs }
62- .forEach { resSourceDirectoriesBySourceSet[it.first] = it.second }
63-
54+ val addSourceTaskProvider =
55+ project.tasks.register<ScaleDimensTask >(" scaleDimens${variant.name} " ) {
56+ group = " scale-dimens"
57+ this .resourceDirectories.set(resSourceDirectories)
58+ this .extension.set(extension)
59+ }
60+ variant.sources.res?.addGeneratedSourceDirectory(
61+ addSourceTaskProvider,
62+ ScaleDimensTask ::outputFolder
63+ )
6464 }
6565 }
6666 }
0 commit comments