Update Jacoco.kt

build-logic/convention/src/main/kotlin/com/google/samples/apps/nowinandroid/Jacoco.kt

@@ -88,13 +88,11 @@ internal fun Project.configureJacoco(
                     html.required = true
                 }
 
-                // TODO: This is missing files in src/debug/, src/prod, src/demo, src/demoDebug...
+                // Collect all java and kotlin source directories from the variant's source sets
-                sourceDirectories.setFrom(
+                val sourceDirs = variant.sources
-                    files(
+                    .map { it.java.srcDirs + it.kotlin.srcDirs }
-                        "$projectDir/src/main/java",
+                    .flatten()
-                        "$projectDir/src/main/kotlin",
+                sourceDirectories.setFrom(sourceDirs)
-                    ),
-                )
 
                 executionData.setFrom(
                     project.fileTree("$buildDir/outputs/unit_test_code_coverage/${variant.name}UnitTest")