处理有扩展类的时候继承找不到成员

Author: CppCXY

EmmyLua-Common/src/main/ext/com/tang/intellij/lua/stubs/index/LuaClassMemberIndex.kt

@@ -22,6 +22,7 @@ import com.tang.intellij.lua.comment.psi.LuaDocTagField
 import com.tang.intellij.lua.psi.LuaClassMember
 import com.tang.intellij.lua.psi.LuaClassMethod
 import com.tang.intellij.lua.psi.LuaTableField
+import com.tang.intellij.lua.psi.search.LuaShortNamesManager
 import com.tang.intellij.lua.search.SearchContext
 import com.tang.intellij.lua.stubs.StubKeys
 import com.tang.intellij.lua.ty.ITyClass
@@ -41,6 +42,25 @@ class LuaClassMemberIndex : StubIndex<Int, LuaClassMember>() {
             return ContainerUtil.process(all, processor)
         }
 
+//        fun processNonIndex(
+//            className: String,
+//            fieldName: String,
+//            context: SearchContext,
+//            processor: Processor<LuaClassMember>
+//        ) {
+//            val classDef = LuaShortNamesManager.getInstance(context.project).findClass(className, context)
+//            if (classDef != null) {
+//                val type = classDef.type
+//                if (type is)
+//                // from alias
+//                type.lazyInit(context)
+//                val member = LuaShortNamesManager.getInstance(context.project).findMember(type, fieldName, context)
+//                if (member != null) {
+//                    processor.process(member)
+//                }
+//            }
+//        }
+
         fun process(
             className: String,
             fieldName: String,
@@ -66,9 +86,12 @@ class LuaClassMemberIndex : StubIndex<Int, LuaClassMember>() {
 
                     var founded = false
                     TyClass.processSuperClass(type, context) { superType ->
-                        if(process(superType.className, fieldName, context, processor, false)){
-                            founded = true
-                        }
+                        LuaShortNamesManager.getInstance(context.project)
+                            .processAllMembers(superType, fieldName, context, Processor {
+                                processor.process(it)
+                                founded = true
+                                true
+                            })
                         true
                     }
                     return founded
@@ -101,9 +124,9 @@ class LuaClassMemberIndex : StubIndex<Int, LuaClassMember>() {
                         return false
 
                     val superClassName = type.superClassName
-                    if(superClassName != null) {
+                    if (superClassName != null) {
                         val superClass = LuaClassIndex.find(superClassName, context)
-                        if(superClass is TyClass && !superClass.isInterface){
+                        if (superClass is TyClass && !superClass.isInterface) {
                             return process(superClassName, fieldName, context, processor, true)
                         }
                     }
@@ -122,10 +145,12 @@ class LuaClassMemberIndex : StubIndex<Int, LuaClassMember>() {
                         docField = it
                         false
                     }
+
                     is LuaTableField -> {
                         tableField = it
                         true
                     }
+
                     else -> {
                         if (perfect == null)
                             perfect = it
@@ -142,16 +167,18 @@ class LuaClassMemberIndex : StubIndex<Int, LuaClassMember>() {
             var perfect: LuaClassMember? = null
             var docField: LuaDocTagField? = null
             var tableField: LuaTableField? = null
-            processOrigin(type.className, fieldName, context,  {
+            processOrigin(type.className, fieldName, context, {
                 when (it) {
                     is LuaDocTagField -> {
                         docField = it
                         false
                     }
+
                     is LuaTableField -> {
                         tableField = it
                         true
                     }
+
                     else -> {
                         if (perfect == null)
                             perfect = it

EmmyLua-Common/src/main/java/com/tang/intellij/lua/ty/TyClass.kt

@@ -177,6 +177,7 @@ abstract class TyClass(
 
                 return member?.guessType(context)
             }
+
             LuaLiteralKind.String -> {
                 var member = LuaClassMemberIndex.find(this, element.text, context)
                 if (member == null) {
@@ -265,7 +266,7 @@ abstract class TyClass(
 
     override fun getClassCallType(context: SearchContext): ITyFunction? {
         val ty = findMemberType("ctor", context)
-        if(ty is ITyFunction){
+        if (ty is ITyFunction) {
             return ty
         }
         return null
@@ -421,8 +422,7 @@ class TyPsiDocClass(tagClass: LuaDocTagClass) : TyClass(tagClass.name) {
         }
         if (tagClass.`interface` != null) {
             isInterface = true
-        }
-        else if(tagClass.enum != null){
+        } else if (tagClass.enum != null) {
             isEnum = true
         }