Refactor file type check logic and improve subproject path retrieval

Author: xterao

src/main/kotlin/org/domaframework/doma/intellij/common/FileTypeCheck.kt

@@ -46,14 +46,6 @@ fun isJavaOrKotlinFileType(daoFile: PsiFile): Boolean {
     }
 }
 
-fun isJavaOrKotlinFileType(file: VirtualFile): Boolean {
-    val fileType = file.fileType
-    return when (fileType.name) {
-        "JAVA", "Kotlin", "CLASS" -> true
-        else -> false
-    }
-}
-
 /*
  * Determine whether the open file is an SQL template file extension
  */
@@ -85,16 +77,27 @@ fun searchDaoFile(
 ): VirtualFile? {
     val projectRootPath = contentRoot?.path ?: return null
     if (projectRootPath.endsWith(SRC_MAIN_PATH)) {
+        println("Find Dao file path in Project")
         return contentRoot.findFileByRelativePath(relativeDaoFilePath)
     }
 
     if (projectRootPath.length > originFilePath.length) {
         return null
     }
 
-    val subProject =
-        originFilePath.substring(projectRootPath.length, originFilePath.indexOf(SRC_MAIN_PATH))
-    return contentRoot
-        .findFileByRelativePath(subProject)
-        ?.findFileByRelativePath(relativeDaoFilePath)
+    // TODO Dynamically build the source directory path and retrieve subproject info
+    // by inspecting file metadata instead of using string manipulation.
+    println("Find Dao file path in Sub Project")
+    val index = originFilePath.indexOf(SRC_MAIN_PATH)
+    val projectRootPathBefore = projectRootPath.substringBefore(SRC_MAIN_PATH)
+    if (index < 0 || projectRootPath.length < index) return null
+    val subProjectName =
+        originFilePath.substring(projectRootPathBefore.length, index)
+
+    val daoFile =
+        contentRoot
+            .findFileByRelativePath(subProjectName)
+            ?.findFileByRelativePath(relativeDaoFilePath)
+
+    return daoFile
 }