#30 Configure kotlinOptions{} after project evaluation to respect provided outputPath for report

Author: PatilShreyas

gradle-plugin/src/main/kotlin/dev/shreyaspatil/composeCompilerMetricsGenerator/plugin/ReportGenPlugin.kt

@@ -27,9 +27,11 @@ import com.android.build.api.dsl.CommonExtension
 import com.android.build.api.variant.AndroidComponentsExtension
 import dev.shreyaspatil.composeCompilerMetricsGenerator.plugin.task.createComposeCompilerReportGenTaskForVariant
 import dev.shreyaspatil.composeCompilerMetricsGenerator.plugin.task.executingComposeCompilerReportGenerationGradleTask
-import dev.shreyaspatil.composeCompilerMetricsGenerator.plugin.utils.kotlinOptions
 import org.gradle.api.Plugin
 import org.gradle.api.Project
+import org.gradle.kotlin.dsl.getByType
+import org.jetbrains.kotlin.gradle.dsl.KotlinAndroidProjectExtension
+import org.jetbrains.kotlin.gradle.dsl.KotlinJvmOptions
 
 @Suppress("UnstableApiUsage")
 class ReportGenPlugin : Plugin<Project> {
@@ -53,33 +55,38 @@ class ReportGenPlugin : Plugin<Project> {
             if (isComposeEnabled != true) {
                 error("Jetpack Compose is not found enabled in this module '$name'")
             }
-        }
 
-        // When this method returns true it means gradle task for generating report is executing otherwise
-        // normal compilation task is executing.
-        val isFromReportGenGradleTask = target.executingComposeCompilerReportGenerationGradleTask()
-        if (isFromReportGenGradleTask) {
-            commonExtension?.configureKotlinOptionsForComposeCompilerReport(reportExt)
+            // When this method returns true it means gradle task for generating report is executing otherwise
+            // normal compilation task is executing.
+            val isFromReportGenGradleTask = target.executingComposeCompilerReportGenerationGradleTask()
+            if (isFromReportGenGradleTask) {
+                val kotlinAndroidExt = extensions.getByType<KotlinAndroidProjectExtension>()
+                kotlinAndroidExt.target {
+                    // Exclude for test variants, no use!
+                    compilations.filter { !it.name.endsWith("Test") }.forEach {
+                        it.kotlinOptions {
+                            configureKotlinOptionsForComposeCompilerReport(this@afterEvaluate)
+                        }
+                    }
+                }
+            }
         }
     }
 
-    private fun CommonExtension<*, *, *, *>.configureKotlinOptionsForComposeCompilerReport(
-        reportExtension: ComposeCompilerReportExtension,
-    ) {
-        kotlinOptions {
-            val outputPath = reportExtension.composeRawMetricsOutputDirectory.absolutePath
-            if (reportExtension.enableReport.get()) {
-                freeCompilerArgs += listOf(
-                    "-P",
-                    "plugin:androidx.compose.compiler.plugins.kotlin:reportsDestination=$outputPath"
-                )
-            }
-            if (reportExtension.enableMetrics.get()) {
-                freeCompilerArgs += listOf(
-                    "-P",
-                    "plugin:androidx.compose.compiler.plugins.kotlin:metricsDestination=$outputPath"
-                )
-            }
+    private fun KotlinJvmOptions.configureKotlinOptionsForComposeCompilerReport(project: Project) {
+        val reportExtension = project.extensions.getByType<ComposeCompilerReportExtension>()
+        val outputPath = reportExtension.composeRawMetricsOutputDirectory.absolutePath
+        if (reportExtension.enableReport.get()) {
+            freeCompilerArgs += listOf(
+                "-P",
+                "plugin:androidx.compose.compiler.plugins.kotlin:reportsDestination=$outputPath"
+            )
+        }
+        if (reportExtension.enableMetrics.get()) {
+            freeCompilerArgs += listOf(
+                "-P",
+                "plugin:androidx.compose.compiler.plugins.kotlin:metricsDestination=$outputPath"
+            )
         }
     }
 }