🐛 Add dependency to all Kotlin compile tasks

Author: smidf

plugin/src/main/kotlin/cz/eman/swagger/codegen/SwaggerCodeGenPlugin.kt

@@ -32,7 +32,6 @@ open class SwaggerCodeGenPlugin : Plugin<Project> {
         val configsExt = project.extensions.create(SWAGGER_TASK, SwaggerCodeGenConfig::class.java)
         autoHookJava(project, configsExt)
         autoHookKotlin(project, configsExt)
-        autoHookKotlin(project, configsExt, true)
     }
 
     /**
@@ -69,41 +68,31 @@ open class SwaggerCodeGenPlugin : Plugin<Project> {
     }
 
     /**
-     * Auto-hooks openApi generation to project that contain compile task [COMPILE_KOTLIN] or [COMPILE_KOTLIN_JVM].
-     * Finds the task and crates java generator task for it with dependency.
+     * Auto-hooks openApi generation to project that contain Kotlin compile task.
+     * Finds the task and creates java generator task for it with dependency.
      *
      * @param project used to search for compile tasks
      * @param configsExt swagger gen config
      */
-    private fun autoHookKotlin(
-        project: Project,
-        configsExt: SwaggerCodeGenConfig,
-        isMpp: Boolean = false
-    ) {
-        var hooked = false
+    private fun autoHookKotlin(project: Project, configsExt: SwaggerCodeGenConfig) {
         project.afterEvaluate {
             if (configsExt.autoHook) {
-                val compileTaskName = if (isMpp) {
-                    COMPILE_KOTLIN_JVM
-                } else {
-                    COMPILE_KOTLIN
+                val generators = configsExt.configs.mapNotNull { taskConfig ->
+                    createGenerator(
+                        project,
+                        LANGUAGE_KOTLIN,
+                        configsExt,
+                        taskConfig
+                    )
                 }
-                project.getTasksByName(compileTaskName, false).firstOrNull()?.let { compileKotlin ->
-                    configsExt.configs.forEach { taskConfig ->
-                        createGenerator(
-                            project,
-                            LANGUAGE_KOTLIN,
-                            configsExt,
-                            taskConfig
-                        )?.let { task ->
-                            compileKotlin.dependsOn(task)
-                        }
-                    }
-                    hooked = true
+
+                val compileTasks = project.kotlinCompileTasks()
+                logger.info("Kotlin auto-hooked: ${compileTasks.isNotEmpty()}")
+                compileTasks.all { task ->
+                    task.dependsOn(generators)
                 }
             }
         }
-        logger.info("Kotlin (isMpp: $isMpp) auto-hooked: $hooked")
     }
 
     /**

plugin/src/main/kotlin/cz/eman/swagger/codegen/Utils.kt

@@ -0,0 +1,8 @@
+package cz.eman.swagger.codegen
+
+import org.gradle.api.Project
+import org.gradle.api.tasks.TaskCollection
+import org.jetbrains.kotlin.gradle.dsl.KotlinCompile
+
+internal fun Project.kotlinCompileTasks(): TaskCollection<KotlinCompile<*>> =
+    tasks.withType(KotlinCompile::class.java)

sample/kotlin-client/build.gradle.kts

@@ -1,5 +1,4 @@
 import cz.eman.swagger.codegen.SwaggerCodeGenConfig
-import cz.eman.swagger.codegen.SwaggerCodeGenTask
 import cz.eman.swagger.codegen.SwaggerCodeGenTaskConfig
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 
@@ -93,7 +92,3 @@ tasks.withType<KotlinCompile> {
         jvmTarget = javaVersion.toString()
     }
 }
-
-tasks.withType<SwaggerCodeGenTask> {
-    project.tasks.findByName("kaptGenerateStubsKotlin")?.dependsOn(this)
-}