Add an option to configure AA type inference for single file

UnitTestBot · Dec 26, 2024 · b6d907c · b6d907c
1 parent fd7df17
commit b6d907c
Showing 1 changed file with 5 additions and 2 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
@@ -94,11 +94,14 @@ fun generateSdkIR(sdkPath: Path): Path = generateEtsIR(
     useArkAnalyzerTypeInference = 0,
 )
 
-fun loadEtsFileAutoConvert(projectPath: Path): EtsFile {
+fun loadEtsFileAutoConvert(
+    projectPath: Path,
+    useArkAnalyzerTypeInference: Int? = 1,
+): EtsFile {
     val irFilePath = generateEtsIR(
         projectPath,
         isProject = false,
-        useArkAnalyzerTypeInference = 1,
+        useArkAnalyzerTypeInference = useArkAnalyzerTypeInference,
     )
     irFilePath.inputStream().use { stream ->
         val etsFileDto = EtsFileDto.loadFromJson(stream)