Resolve issue where Java files cannot be selected separately (#2590)

See #2590
Kotlin · Aug 3, 2022 · 47c87aa · 47c87aa
1 parent 2372302
commit 47c87aa
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/plugins/base/src/main/kotlin/translators/psi/DefaultPsiToDocumentableTranslator.kt b/plugins/base/src/main/kotlin/translators/psi/DefaultPsiToDocumentableTranslator.kt
@@ -80,13 +80,23 @@ class DefaultPsiToDocumentableTranslator(
                 ?: listOf()
             val localFileSystem = VirtualFileManager.getInstance().getFileSystem("file")
 
-            val psiFiles = sourceRoots.parallelMap { sourceRoot ->
+            val psiFiles = mutableListOf<PsiJavaFile>()
+            psiFiles.addAll(sourceRoots.parallelMap { sourceRoot ->
                 sourceRoot.absoluteFile.walkTopDown().mapNotNull {
                     localFileSystem.findFileByPath(it.path)?.let { vFile ->
                         PsiManager.getInstance(environment.project).findFile(vFile) as? PsiJavaFile
                     }
                 }.toList()
-            }.flatten()
+            }.flatten())
+            environment.configuration.javaSourceRoots.forEach {
+                if (it.endsWith(".java")){
+                    localFileSystem.findFileByPath(it)?.let{vFile ->
+                        (PsiManager.getInstance(environment.project).findFile(vFile) as? PsiJavaFile)?.let { psiJavaFile ->
+                            psiFiles.add(psiJavaFile)
+                        }
+                    }
+                }
+            }
 
             val docParser =
                 DokkaPsiParser(