Add a function to load a project from multiple sources (#285)

UnitTestBot · Dec 26, 2024 · fd7df17 · fd7df17
1 parent 61ad160
commit fd7df17
Showing 1 changed file with 19 additions and 12 deletions.
diff --git a/jacodb-ets/src/main/kotlin/org/jacodb/ets/utils/LoadEtsFile.kt b/jacodb-ets/src/main/kotlin/org/jacodb/ets/utils/LoadEtsFile.kt
@@ -126,20 +126,27 @@ fun loadEtsProjectFromIR(
     projectFilesPath: Path,
     sdkFilesPath: Path?,
 ): EtsScene {
-    val walker = { irFolder: Path ->
-        irFolder.walk()
-            .filter { it.extension == "json" }
-            .map {
-                it.inputStream().use { stream ->
-                    val etsFileDto = EtsFileDto.loadFromJson(stream)
-                    convertToEtsFile(etsFileDto)
-                }
-            }
-            .toList()
-    }
-
     val projectFiles = walker(projectFilesPath)
     val sdkFiles = sdkFilesPath?.let { walker(it) }.orEmpty()
 
     return EtsScene(projectFiles, sdkFiles)
 }
+
+fun loadEtsProjectFromMultipleIR(input: List<Path>, sdkPaths: List<Path>): EtsScene {
+    val projectFiles = input.flatMap(walker)
+    val sdkFiles = sdkPaths.flatMap(walker)
+
+    return EtsScene(projectFiles, sdkFiles)
+}
+
+private val walker = { irFolder: Path ->
+    irFolder.walk()
+        .filter { it.extension == "json" }
+        .map {
+            it.inputStream().use { stream ->
+                val etsFileDto = EtsFileDto.loadFromJson(stream)
+                convertToEtsFile(etsFileDto)
+            }
+        }
+        .toList()
+}