From c5fe74cf3a38f857dfccde35b1be58ac4ea5eb7e Mon Sep 17 00:00:00 2001
From: Vasily Levchenko <vasily.v.levchenko@gmail.com>
Date: Thu, 29 Sep 2016 17:42:35 +0300
Subject: [PATCH] NativeInteropPlugin: scan in every declared sourceSet in
 project for def file

---
 .../org/jetbrains/kotlin/NativeInteropPlugin.groovy       | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/buildSrc/src/main/groovy/org/jetbrains/kotlin/NativeInteropPlugin.groovy b/buildSrc/src/main/groovy/org/jetbrains/kotlin/NativeInteropPlugin.groovy
index e191b535f29..6066f13e798 100644
--- a/buildSrc/src/main/groovy/org/jetbrains/kotlin/NativeInteropPlugin.groovy
+++ b/buildSrc/src/main/groovy/org/jetbrains/kotlin/NativeInteropPlugin.groovy
@@ -38,9 +38,11 @@ class NativeInteropPlugin implements Plugin<Project> {
         prj.afterEvaluate {
             prj.tasks.getByName(genStubsTaskName) {
                 // TODO: handle other source sets
-                prj.sourceSets.main.kotlin.srcDirs.each { srcDir ->
-                    inputs.files prj.fileTree(srcDir.path).include('**/*.def')
-                    args srcDir
+                prj.sourceSets.each {
+                    it.kotlin.srcDirs.each { srcDir ->
+                      inputs.files prj.fileTree(srcDir.path).include('**/*.def')
+                      args srcDir
+                    }
                 }
             }
         }