From 1bd178986143e1a81f2aba73abccceaa0e837c77 Mon Sep 17 00:00:00 2001
From: Ian Lynagh <igfoo@github.com>
Date: Thu, 22 Aug 2024 17:29:07 +0100
Subject: [PATCH] KE2: Get source file list from arguments

---
 java/kotlin-extractor2/src/main/kotlin/KotlinExtractor.kt | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/java/kotlin-extractor2/src/main/kotlin/KotlinExtractor.kt b/java/kotlin-extractor2/src/main/kotlin/KotlinExtractor.kt
index 305f30d4a82..870f470a61d 100644
--- a/java/kotlin-extractor2/src/main/kotlin/KotlinExtractor.kt
+++ b/java/kotlin-extractor2/src/main/kotlin/KotlinExtractor.kt
@@ -11,12 +11,16 @@ import org.jetbrains.kotlin.analysis.api.standalone.buildStandaloneAnalysisAPISe
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtLibraryModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtSdkModule
 import org.jetbrains.kotlin.analysis.project.structure.builder.buildKtSourceModule
+import org.jetbrains.kotlin.cli.common.arguments.parseCommandLineArguments
+import org.jetbrains.kotlin.cli.common.arguments.K2JVMCompilerArguments
 import org.jetbrains.kotlin.platform.jvm.JvmPlatforms
 import org.jetbrains.kotlin.psi.*
 
 @OptIn(KaAnalysisApiInternals::class)
-fun main() {
+fun main(args : Array<String>) {
     lateinit var sourceModule: KaSourceModule
+    val k2args : K2JVMCompilerArguments = parseCommandLineArguments(args.toList())
+
     val session = buildStandaloneAnalysisAPISession {
         registerProjectService(KotlinLifetimeTokenProvider::class.java, KotlinAlwaysAccessibleLifetimeTokenProvider())
 
@@ -32,7 +36,7 @@ fun main() {
             )
             sourceModule = addModule(
                 buildKtSourceModule {
-                    addSourceRoot(Paths.get("testSrc"))
+                    addSourceRoots(k2args.freeArgs.map { Paths.get(it) })
                     addRegularDependency(sdk)
                     platform = JvmPlatforms.defaultJvmPlatform
                     moduleName = "<source>"