Enhance entity checks to traverse superclasses for entity and immutable entity annotations

Author: xterao

src/main/kotlin/org/domaframework/doma/intellij/common/util/TypeUtil.kt

@@ -76,21 +76,30 @@ object TypeUtil {
         type: PsiType?,
         project: Project,
     ): Boolean {
-        val clazz = type?.let { project.getJavaClazz(type) }
-        return clazz?.isEntity() == true
+        var clazz = type?.let { project.getJavaClazz(type) }
+        while (clazz != null) {
+            if (clazz.isEntity()) {
+                return true
+            }
+            clazz = clazz.superClass
+        }
+        return false
     }
 
     fun isImmutableEntity(
         project: Project,
         canonicalText: String,
     ): Boolean {
-        val returnTypeClass = project.getJavaClazz(canonicalText)
-        val entity =
-            returnTypeClass?.getClassAnnotation(DomaClassName.ENTITY.className) ?: return false
-        return entity.let { entity ->
-            AnnotationUtil.getBooleanAttributeValue(entity, "immutable") == true
-        } == true ||
-            returnTypeClass.isRecord == true
+        var returnTypeClass = project.getJavaClazz(canonicalText)
+        while (returnTypeClass != null) {
+            val entity = returnTypeClass.getClassAnnotation(DomaClassName.ENTITY.className)
+            if (entity != null) {
+                return AnnotationUtil.getBooleanAttributeValue(entity, "immutable") == true ||
+                    returnTypeClass.isRecord
+            }
+            returnTypeClass = returnTypeClass.superClass
+        }
+        return false
     }
 
     /**