From c8ae9a70902f67ce592bad912f3e94ea2a221b92 Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Thu, 17 Jun 2021 15:46:50 +0300
Subject: [PATCH] [imltogradle] Speedup files traversing by ignoring
 directories

---
 .../jetbrains/kotlin/generators/imltogradle/Main.kt    | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/generators/ide-iml-to-gradle-generator/src/org/jetbrains/kotlin/generators/imltogradle/Main.kt b/generators/ide-iml-to-gradle-generator/src/org/jetbrains/kotlin/generators/imltogradle/Main.kt
index 7c79364ce8d..7479e4b8693 100644
--- a/generators/ide-iml-to-gradle-generator/src/org/jetbrains/kotlin/generators/imltogradle/Main.kt
+++ b/generators/ide-iml-to-gradle-generator/src/org/jetbrains/kotlin/generators/imltogradle/Main.kt
@@ -18,6 +18,7 @@ import org.jetbrains.jps.model.module.*
 import org.jetbrains.kotlin.generators.imltogradle.GradleDependencyNotation.IntellijDepGradleDependencyNotation
 import java.io.File
 import java.util.*
+import kotlin.system.measureNanoTime
 
 private lateinit var intellijModuleNameToGradleDependencyNotationsMapping: Map<String, List<GradleDependencyNotation>>
 private val KOTLIN_REPO_ROOT = File(".").canonicalFile
@@ -50,11 +51,14 @@ val jsonUrlPrefixes = mapOf(
 fun main() {
     val ijCommunityModuleNameToJpsModuleMapping = INTELLIJ_REPO_ROOT.loadJpsProject().modules.associateBy { it.name }
 
-    val imlFiles = INTELLIJ_REPO_ROOT.walk()
-        .onEnter { it != INTELLIJ_REPO_ROOT.resolve("out") }
+    val skipDirNames = setOf("src", "out", "org", "com", "testData")
+    val imlFiles = INTELLIJ_REPO_ROOT
+        .walk()
+        .onEnter { dir -> dir.name !in skipDirNames }
         .filter {
             it.isFile && it.extension == "iml" &&
-                    (it.name.startsWith("kotlin.") || it.nameWithoutExtension in intellijModulesForWhichGenerateBuildGradle)
+                    (it.name.startsWith("kotlin.") ||
+                            it.nameWithoutExtension in intellijModulesForWhichGenerateBuildGradle)
         }
         .toList()