Fix HiltGradlePlugin not working in KMP's androidMain directory

RELNOTES=N/A PiperOrigin-RevId: 583124622
google · Nov 16, 2023 · f0c2510 · f0c2510
1 parent 2b4346a
commit f0c2510
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/.../dagger/hilt/android/plugin/main/src/main/kotlin/dagger/hilt/android/plugin/util/Tasks.kt b/.../dagger/hilt/android/plugin/main/src/main/kotlin/dagger/hilt/android/plugin/util/Tasks.kt
@@ -49,7 +49,9 @@ internal fun addKaptTaskProcessorOptions(
     """.trimIndent()
   }
   project.tasks.withType(KaptTask::class.java) { task ->
-    if (task.name == "kapt${component.name.capitalize()}Kotlin") {
+    if (task.name == "kapt${component.name.capitalize()}Kotlin" ||
+        // Task names in shared/src/AndroidMain in KMP projects has a platform suffix.
+        task.name == "kapt${component.name.capitalize()}KotlinAndroid") {
       val argProvider = produceArgProvider.invoke(task)
       // TODO: Update once KT-58009 is fixed.
       try {
@@ -83,7 +85,9 @@ internal fun addKspTaskProcessorOptions(
     """.trimIndent()
   }
   project.tasks.withType(KspTaskJvm::class.java) { task ->
-    if (task.name == "ksp${component.name.capitalize()}Kotlin") {
+    if (task.name == "ksp${component.name.capitalize()}Kotlin" ||
+        // Task names in shared/src/AndroidMain in KMP projects has a platform suffix.
+        task.name == "ksp${component.name.capitalize()}KotlinAndroid") {
       task.commandLineArgumentProviders.add(produceArgProvider.invoke(task))
     }
   }
@@ -102,4 +106,5 @@ internal fun Task.isKspTask(): Boolean = try {
   kspTaskClass.isAssignableFrom(this::class.java)
 } catch (ex: ClassNotFoundException) {
   false
-}
+}
+