Support Immutable inheritors vision

Author: Enaium

231/src/main/kotlin/cn/enaium/jimmer/buddy/utitlity/navigation.kt

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2025 Enaium
+ *
+ * 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
+ *
+ *     http://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 cn.enaium.jimmer.buddy.utitlity
+
+import com.intellij.codeInsight.navigation.NavigationUtil.getPsiElementPopup
+import com.intellij.openapi.ui.popup.JBPopup
+import com.intellij.psi.PsiElement
+
+/**
+ * @author Enaium
+ */
+fun getPsiElementPopup(elements: List<PsiElement>, title: String): JBPopup {
+    return getPsiElementPopup(elements.toTypedArray(), title)
+}

242/src/main/kotlin/cn/enaium/jimmer/buddy/utitlity/navigation.kt

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2025 Enaium
+ *
+ * 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
+ *
+ *     http://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 cn.enaium.jimmer.buddy.utitlity
+
+import com.intellij.codeInsight.navigation.getPsiElementPopup
+import com.intellij.openapi.ui.popup.JBPopup
+import com.intellij.psi.PsiElement
+
+/**
+ * @author Enaium
+ */
+fun getPsiElementPopup(elements: List<PsiElement>, title: String): JBPopup {
+    return getPsiElementPopup(elements.toTypedArray(), title)
+}

README.md

@@ -20,6 +20,7 @@ A plugin that adds first-class support for Project Jimmer
 - DTO class member go to DTO file.
 - Execute SQL to clipboard.
 - DTO type count and navigation.
+- Immutable inheritor count and navigation.
 
 ### First-class DTO development
 

gradle.properties

@@ -1,4 +1,4 @@
 kotlin.code.style=official
-version=1.6.0
+version=1.7.0
 kotlin.daemon.jvmargs=-Xmx3G
 org.gradle.jvmargs=-Xmx3G

shared/main/kotlin/cn/enaium/jimmer/buddy/extensions/insight/DtoTypeTargetCodeVisionProvider.kt

@@ -18,10 +18,9 @@ package cn.enaium.jimmer.buddy.extensions.insight
 
 import cn.enaium.jimmer.buddy.extensions.dto.psi.DtoPsiDtoType
 import cn.enaium.jimmer.buddy.extensions.dto.psi.DtoPsiRoot
-import com.intellij.codeInsight.navigation.NavigationUtil.getPsiElementPopup
+import cn.enaium.jimmer.buddy.utitlity.getPsiElementPopup
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
-import com.intellij.openapi.ui.popup.JBPopup
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
 import com.intellij.psi.search.FilenameIndex
@@ -60,7 +59,7 @@ class DtoTypeTargetCodeVisionProvider : ImmutableCodeVisionProvider() {
         val project = editor.project ?: return
         val qualifiedName = element.toUElementOfType<UClass>()?.qualifiedName ?: return
         getPsiElementPopup(
-            findDtoTypes(project, qualifiedName).toTypedArray(),
+            findDtoTypes(project, qualifiedName),
             "Choose DTO Type"
         ).showInBestPositionFor(editor)
     }

shared/main/kotlin/cn/enaium/jimmer/buddy/extensions/insight/ImmutableInheritorCodeVisionProvider.kt

@@ -0,0 +1,71 @@
+/*
+ * Copyright 2025 Enaium
+ *
+ * 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
+ *
+ *     http://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 cn.enaium.jimmer.buddy.extensions.insight
+
+import cn.enaium.jimmer.buddy.utility.isImmutable
+import cn.enaium.jimmer.buddy.utitlity.getPsiElementPopup
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiClass
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.search.searches.ClassInheritorsSearch
+import org.babyfish.jimmer.sql.MappedSuperclass
+import org.jetbrains.kotlin.idea.base.util.allScope
+import org.jetbrains.uast.UClass
+import org.jetbrains.uast.toUElementOfType
+import java.awt.event.MouseEvent
+
+/**
+ * @author Enaium
+ */
+class ImmutableInheritorCodeVisionProvider : ImmutableCodeVisionProvider() {
+
+    override val id: String
+        get() = "JimmerBuddy.immutable.inheritor"
+    override val name: String
+        get() = "Immutable Inheritor"
+
+    override fun getHint(element: PsiElement, file: PsiFile): String? {
+        val uClass = element.toUElementOfType<UClass>() ?: return null
+        uClass.uAnnotations.find { it.qualifiedName == MappedSuperclass::class.qualifiedName } ?: return null
+
+        val count = findInheritors(uClass.javaPsi).size
+
+        return if (count > 0) {
+            "$count Inheritors"
+        } else {
+            null
+        }
+    }
+
+    override fun handleClick(
+        editor: Editor,
+        element: PsiElement,
+        event: MouseEvent?
+    ) {
+        val uClass = element.toUElementOfType<UClass>() ?: return
+        getPsiElementPopup(
+            findInheritors(uClass.javaPsi),
+            "Choose Inheritors"
+        ).showInBestPositionFor(editor)
+    }
+
+    private fun findInheritors(psiClass: PsiClass): List<PsiClass> {
+        return ClassInheritorsSearch.search(psiClass, psiClass.project.allScope(), false)
+            .filter { it.isImmutable() }
+    }
+}

shared/main/resources/META-INF/plugin.xml

@@ -152,6 +152,8 @@
                                order="first"/>
         <codeInsight.daemonBoundCodeVisionProvider
                 implementation="cn.enaium.jimmer.buddy.extensions.insight.DtoTypeTargetCodeVisionProvider"/>
+        <codeInsight.daemonBoundCodeVisionProvider
+                implementation="cn.enaium.jimmer.buddy.extensions.insight.ImmutableInheritorCodeVisionProvider"/>
     </extensions>
     <extensions defaultExtensionNs="org.jetbrains.kotlin">
         <supportsKotlinPluginMode supportsK1="true" supportsK2="${supportsK2}"/>