AA: collect .kts source files too
diff --git a/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/impl/KtModuleUtils.kt b/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/impl/KtModuleUtils.kt
index bb2f4b4..7a5ab47 100644
--- a/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/impl/KtModuleUtils.kt
+++ b/analysis/analysis-api-standalone/src/org/jetbrains/kotlin/analysis/project/structure/impl/KtModuleUtils.kt
@@ -28,6 +28,7 @@
import org.jetbrains.kotlin.config.JVMConfigurationKeys
import org.jetbrains.kotlin.idea.KotlinFileType
import org.jetbrains.kotlin.js.resolve.JsPlatformAnalyzerServices
+import org.jetbrains.kotlin.parsing.KotlinParserDefinition
import org.jetbrains.kotlin.wasm.resolve.WasmPlatformAnalyzerServices
import org.jetbrains.kotlin.platform.TargetPlatform
import org.jetbrains.kotlin.platform.isCommon
@@ -86,7 +87,7 @@
* Collect source file path from the given [root] store them in [result].
*
* E.g., for `project/app/src` as a [root], this will walk the file tree and
- * collect all `.kt` and `.java` files under that folder.
+ * collect all `.kt`, `.kts`, and `.java` files under that folder.
*
* Note that this util gracefully skips [IOException] during file tree traversal.
*/
@@ -110,7 +111,10 @@
if (!Files.isRegularFile(file) || !Files.isReadable(file))
return FileVisitResult.CONTINUE
val ext = getFileExtension(file.fileName.toString())
- if (ext == KotlinFileType.EXTENSION || ext == JavaFileType.DEFAULT_EXTENSION) {
+ if (ext == KotlinFileType.EXTENSION ||
+ ext == KotlinParserDefinition.STD_SCRIPT_SUFFIX ||
+ ext == JavaFileType.DEFAULT_EXTENSION
+ ) {
result.add(file.toString())
}
return FileVisitResult.CONTINUE
