Add support for custom functions and refactor related settings

Author: xterao

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

@@ -0,0 +1,29 @@
+/*
+ * Copyright Doma Tools Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.domaframework.doma.intellij.common
+
+// TODO Dynamically build the source directory path and retrieve subproject info
+// by inspecting file metadata instead of using string manipulation.
+object CommonPathParameterHelper {
+    val SRC_MAIN_PATH: String
+        get() = "/src/main"
+
+    val RESOURCES_PATH: String
+        get() = "resources"
+
+    val RESOURCES_META_INF_PATH: String
+        get() = "META-INF"
+}

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

@@ -18,7 +18,7 @@ package org.domaframework.doma.intellij.common
 import com.intellij.openapi.fileTypes.FileTypeManager
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiFile
-import org.domaframework.doma.intellij.common.CommonPathParameter.Companion.SRC_MAIN_PATH
+import org.domaframework.doma.intellij.common.CommonPathParameterHelper.SRC_MAIN_PATH
 
 /**
  * Get extension by file type identifier
@@ -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
  */

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

@@ -18,7 +18,7 @@ package org.domaframework.doma.intellij.common
 import com.intellij.openapi.vfs.StandardFileSystems
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiFile
-import org.domaframework.doma.intellij.common.CommonPathParameter.Companion.RESOURCES_META_INF_PATH
+import org.domaframework.doma.intellij.common.CommonPathParameterHelper.RESOURCES_META_INF_PATH
 
 fun getJarRoot(
     virtualFile: VirtualFile,

src/main/kotlin/org/domaframework/doma/intellij/common/dao/DaoMethodUtil.kt

@@ -27,8 +27,8 @@ import com.intellij.psi.PsiManager
 import com.intellij.psi.PsiMethod
 import com.intellij.psi.search.GlobalSearchScope
 import com.intellij.psi.util.PsiTreeUtil
-import org.domaframework.doma.intellij.common.CommonPathParameter.Companion.RESOURCES_META_INF_PATH
-import org.domaframework.doma.intellij.common.CommonPathParameter.Companion.RESOURCES_PATH
+import org.domaframework.doma.intellij.common.CommonPathParameterHelper.RESOURCES_META_INF_PATH
+import org.domaframework.doma.intellij.common.CommonPathParameterHelper.RESOURCES_PATH
 import org.domaframework.doma.intellij.common.getExtension
 import org.domaframework.doma.intellij.common.getJarRoot
 import org.domaframework.doma.intellij.common.getMethodDaoFilePath

src/main/kotlin/org/domaframework/doma/intellij/common/helper/ActiveProjectHelper.kt

@@ -0,0 +1,37 @@
+/*
+ * Copyright Doma Tools Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.domaframework.doma.intellij.common.helper
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.project.ProjectManager
+import com.intellij.openapi.wm.IdeFocusManager
+
+object ActiveProjectHelper {
+    val activeProject: Project?
+        get() {
+            val openProjects: Array<out Project> = ProjectManager.getInstance().openProjects
+            var activeProject: Project? = null
+
+            for (project in openProjects) {
+                if (IdeFocusManager.getInstance(project).focusOwner != null) {
+                    activeProject = project
+                    break
+                }
+            }
+
+            return activeProject
+        }
+}

src/main/kotlin/org/domaframework/doma/intellij/common/helper/ExpressionFunctionsHelper.kt

@@ -0,0 +1,57 @@
+/*
+ * Copyright Doma Tools Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.domaframework.doma.intellij.common.helper
+
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiClass
+import org.domaframework.doma.intellij.extension.getJavaClazz
+import org.domaframework.doma.intellij.extension.psi.psiClassType
+
+class ExpressionFunctionsHelper {
+    companion object {
+        var expressionFunction: PsiClass? = null
+
+        fun setExpressionFunctionsInterface(project: Project): PsiClass? {
+            val expressionFunctionsClass =
+                project.getJavaClazz("org.seasar.doma.expr.ExpressionFunctions")
+            if (expressionFunctionsClass != null) {
+                expressionFunction = expressionFunctionsClass
+            }
+            return expressionFunction
+        }
+
+        fun isInheritor(expressionClazz: PsiClass?): Boolean {
+            val project = expressionClazz?.project
+
+            expressionFunction?.let { functionInterface ->
+                expressionClazz?.let {
+                    if (expressionClazz.isInheritor(functionInterface, true)) return true
+
+                    val parentType =
+                        expressionClazz.superTypes.firstOrNull()?.canonicalText
+                            ?: expressionClazz.psiClassType.canonicalText
+                    return project?.let {
+                        expressionClazz.psiClassType.canonicalText
+                        project
+                            .getJavaClazz(parentType)
+                            ?.isInheritor(functionInterface, true) == true
+                    } == true
+                }
+            }
+            return false
+        }
+    }
+}

src/main/kotlin/org/domaframework/doma/intellij/common/psi/PsiDaoMethod.kt

@@ -32,7 +32,7 @@ import com.intellij.psi.PsiManager
 import com.intellij.psi.PsiMethod
 import com.intellij.psi.PsiNameValuePair
 import com.intellij.util.IncorrectOperationException
-import org.domaframework.doma.intellij.common.CommonPathParameter.Companion.RESOURCES_PATH
+import org.domaframework.doma.intellij.common.CommonPathParameterHelper.RESOURCES_PATH
 import org.domaframework.doma.intellij.common.dao.formatSqlPathFromDaoPath
 import org.domaframework.doma.intellij.common.getExtension
 import org.domaframework.doma.intellij.extension.findFile

src/main/kotlin/org/domaframework/doma/intellij/extension/ModuleExtensions.kt

@@ -21,7 +21,7 @@ import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.JavaPsiFacade
 import com.intellij.psi.PsiClass
 import com.intellij.psi.search.GlobalSearchScope
-import org.domaframework.doma.intellij.common.CommonPathParameter.Companion.RESOURCES_META_INF_PATH
+import org.domaframework.doma.intellij.common.CommonPathParameterHelper.RESOURCES_META_INF_PATH
 import org.domaframework.doma.intellij.common.dao.formatSqlPathFromDaoPath
 import org.jetbrains.kotlin.idea.util.sourceRoots
 

src/main/kotlin/org/domaframework/doma/intellij/extension/ProjectExtensions.kt

@@ -45,4 +45,8 @@ fun Project.getJavaClazz(fqdn: String): PsiClass? {
         .getInstance(this)
         .findClasses(fqdn, scope)
         .firstOrNull()
+        ?: JavaPsiFacade.getInstance(this).findClass(
+            fqdn,
+            GlobalSearchScope.allScope(this),
+        )
 }

src/main/kotlin/org/domaframework/doma/intellij/extension/psi/PsiMethodExtension.kt

@@ -15,7 +15,6 @@
  */
 package org.domaframework.doma.intellij.extension.psi
 
-import com.intellij.codeInsight.AnnotationUtil
 import com.intellij.psi.PsiMethod
 import com.intellij.psi.PsiParameter
 
@@ -25,13 +24,3 @@ val PsiMethod.methodParameters: List<PsiParameter>
     get() = this.parameterList.parameters.toList()
 
 fun PsiMethod.searchParameter(searchName: String): List<PsiParameter> = this.methodParameters.filter { it.name.startsWith(searchName) }
-
-@OptIn(ExperimentalStdlibApi::class)
-fun PsiMethod.getDomaAnnotationType(): DomaAnnotationType {
-    DomaAnnotationType.entries.forEach {
-        if (AnnotationUtil.findAnnotation(this, it.fqdn) != null) {
-            return it
-        }
-    }
-    return DomaAnnotationType.Unknown
-}