LuaShortNamesManager

Author: tangzx

EmmyLua-Common/src/main/ext/com/tang/intellij/lua/psi/search/CompositeLuaShortNamesManager.kt

@@ -0,0 +1,96 @@
+/*
+ * Copyright (c) 2017. tangzx([email protected])
+ *
+ * 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 com.tang.intellij.lua.psi.search
+
+import com.intellij.openapi.project.Project
+import com.intellij.util.Processor
+import com.tang.intellij.lua.psi.LuaClass
+import com.tang.intellij.lua.psi.LuaClassMember
+import com.tang.intellij.lua.psi.LuaTypeAlias
+import com.tang.intellij.lua.search.SearchContext
+import com.tang.intellij.lua.ty.ITyClass
+
+class CompositeLuaShortNamesManager : LuaShortNamesManager() {
+    private val list = EP_NAME.extensions
+
+    override fun findClass(name: String, context: SearchContext): LuaClass? {
+        for (ep in list) {
+            val c = ep.findClass(name, context)
+            if (c != null)
+                return c
+        }
+        return null
+    }
+
+    override fun findMember(type: ITyClass, fieldName: String, context: SearchContext): LuaClassMember? {
+        for (manager in list) {
+            val ret = manager.findMember(type, fieldName, context)
+            if (ret != null) return ret
+        }
+        return null
+    }
+
+    override fun processAllClassNames(project: Project, processor: Processor<String>): Boolean {
+        for (ep in list) {
+            if (!ep.processAllClassNames(project, processor))
+                return false
+        }
+        return true
+    }
+
+    override fun processClassesWithName(name: String, context: SearchContext, processor: Processor<LuaClass>): Boolean {
+        for (ep in list) {
+            if (!ep.processClassesWithName(name, context, processor))
+                return false
+        }
+        return true
+    }
+
+    override fun getClassMembers(clazzName: String, context: SearchContext): Collection<LuaClassMember> {
+        val collection = mutableListOf<LuaClassMember>()
+        for (manager in list) {
+            val col = manager.getClassMembers(clazzName, context)
+            collection.addAll(col)
+        }
+        return collection
+    }
+
+    override fun processAllMembers(type: ITyClass, fieldName: String, context: SearchContext, processor: Processor<LuaClassMember>): Boolean {
+        for (manager in list) {
+            if (!manager.processAllMembers(type, fieldName, context, processor))
+                return false
+        }
+        return true
+    }
+
+    override fun findAlias(name: String, context: SearchContext): LuaTypeAlias? {
+        for (manager in list) {
+            val alias = manager.findAlias(name, context)
+            if (alias != null)
+                return alias
+        }
+        return null
+    }
+
+    override fun processAllAlias(project: Project, processor: Processor<String>): Boolean {
+        for (manager in list) {
+            if (!manager.processAllAlias(project, processor))
+                return false
+        }
+        return true
+    }
+}

EmmyLua-Common/src/main/ext/com/tang/intellij/lua/psi/search/LuaShortNamesManager.kt

@@ -36,7 +36,7 @@ abstract class LuaShortNamesManager {
         fun getInstance(project: Project): LuaShortNamesManager {
             var manager = project.getUserData(KEY)
             if (manager == null) {
-                manager = LuaShortNamesManagerImpl()
+                manager = CompositeLuaShortNamesManager()
                 project.putUserData(KEY, manager)
             }
             return manager