From b3bff241398f0545fe1870b62371594c495e5825 Mon Sep 17 00:00:00 2001
From: Maxim Bachinsky <maxim.bachinsky@touchin.ru>
Date: Thu, 20 Aug 2020 15:22:41 +0300
Subject: [PATCH] fixed problem with source sets for java code

---
 .../src/main/java/apigen/ApiGeneratorAndroidPlugin.kt      | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/gradle/plugins/src/main/java/apigen/ApiGeneratorAndroidPlugin.kt b/gradle/plugins/src/main/java/apigen/ApiGeneratorAndroidPlugin.kt
index 0442f43..d30c4f5 100644
--- a/gradle/plugins/src/main/java/apigen/ApiGeneratorAndroidPlugin.kt
+++ b/gradle/plugins/src/main/java/apigen/ApiGeneratorAndroidPlugin.kt
@@ -3,6 +3,7 @@ package apigen
 import com.android.build.gradle.AppExtension
 import com.android.build.gradle.LibraryExtension
 import org.gradle.api.Project
+import org.gradle.api.tasks.compile.JavaCompile
 import org.gradle.kotlin.dsl.findByType
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 
@@ -34,6 +35,12 @@ class ApiGeneratorAndroidPlugin : ApiGeneratorPlugin() {
                         .forEach {
                             it.source(outputDir)
                         }
+
+                tasks
+                        .filterIsInstance<JavaCompile>()
+                        .forEach {
+                            it.source(outputDir)
+                        }
             }
         }
     }