azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added use of nameFilter to one of the implementations of DeclarationProvider

Browse Source
This commit is contained in:
Valentin Kipyatkov
2014-10-31 22:09:15 +03:00
parent ed2a50ffcb
commit 059deb3c88
1 changed files with 24 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/idea-analysis/src/org/jetbrains/jet/plugin/stubindex/resolve/StubBasedPackageMemberDeclarationProvider.kt
+24 -9
View File
@@ -31,6 +31,7 @@ import org.jetbrains.jet.lang.resolve.lazy.data.JetClassLikeInfo
import org.jetbrains.jet.lang.resolve.lazy.data.JetClassInfoUtil
import org.jetbrains.jet.lang.resolve.lazy.ResolveSessionUtils
import org.jetbrains.jet.lang.resolve.scopes.JetScope
import java.util.ArrayList
public class StubBasedPackageMemberDeclarationProvider(
private val fqName: FqName,
@@ -39,10 +40,30 @@ public class StubBasedPackageMemberDeclarationProvider(
) : PackageMemberDeclarationProvider {
override fun getDeclarations(kindFilter: (JetScope.DescriptorKind) -> Boolean, nameFilter: (Name) -> Boolean): List<JetDeclaration> {
return TOP_LEVEL_DECLARATION_INDICES.flatMap { index ->
val fqNames = index.getAllKeys(project).toSet().map { FqName(it) }.filter { !it.isRoot() && it.parent() == fqName }
fqNames.flatMap { index.get(it.asString(), project, searchScope) }
val result = ArrayList<JetDeclaration>()
if (kindFilter(JetScope.DescriptorKind.CLASSIFIER)) {
result.addDeclarations(JetFullClassNameIndex.getInstance(), nameFilter)
}
if (kindFilter(JetScope.DescriptorKind.NON_EXTENSION_FUNCTION) || kindFilter(JetScope.DescriptorKind.EXTENSION_FUNCTION)) {
result.addDeclarations(JetTopLevelFunctionsFqnNameIndex.getInstance(), nameFilter)
}
if (kindFilter(JetScope.DescriptorKind.NON_EXTENSION_PROPERTY) || kindFilter(JetScope.DescriptorKind.EXTENSION_PROPERTY)) {
result.addDeclarations(JetTopLevelPropertiesFqnNameIndex.getInstance(), nameFilter)
}
return result
}
private fun MutableCollection<JetDeclaration>.addDeclarations(index: StringStubIndexExtension<out JetNamedDeclaration>,
nameFilter: (Name) -> Boolean) {
index.getAllKeys(project)
.stream()
.map { FqName(it) }
.filter { !it.isRoot() && it.parent() == fqName && nameFilter(it.shortName()) }
.flatMapTo(this) { index[it.asString(), project, searchScope].stream() }
}
override fun getClassOrObjectDeclarations(name: Name): Collection<JetClassLikeInfo> {
@@ -70,9 +91,3 @@ public class StubBasedPackageMemberDeclarationProvider(
return fqName.child(ResolveSessionUtils.safeNameForLazyResolve(name)).asString()
}
}
private val TOP_LEVEL_DECLARATION_INDICES: List<StringStubIndexExtension<out JetNamedDeclaration>> = listOf(
JetFullClassNameIndex.getInstance(),
JetTopLevelFunctionsFqnNameIndex.getInstance(),
JetTopLevelPropertiesFqnNameIndex.getInstance()
)