Core classes for declaration search

2013-11-22 16:34:48 +04:00
parent f35c77e731
commit d261fe8726
3 changed files with 114 additions and 1 deletions
@@ -0,0 +1,8 @@
+<root>
+  <item name='com.intellij.util.EmptyQuery com.intellij.util.Query&lt;T&gt; getEmptyQuery()'>
+    <annotation name='org.jetbrains.annotations.NotNull'/>
+  </item>
+  <item name='com.intellij.util.QueryFactory com.intellij.util.Query&lt;Result&gt; createUniqueResultsQuery(Parameters)'>
+    <annotation name='org.jetbrains.annotations.NotNull'/>
+  </item>
+</root>
@@ -35,6 +35,8 @@ import org.jetbrains.jet.lang.psi.JetBlockExpression
 import org.jetbrains.jet.lang.psi.JetPsiUtil
 import org.jetbrains.jet.lang.psi.JetPsiFactory
 import kotlin.test.assertTrue
+import com.intellij.psi.search.SearchScope
+import com.intellij.psi.search.PsiSearchScopeUtil

 fun PsiElement.getParentByTypeAndPredicate<T: PsiElement>(
        parentClass : Class<T>, strict : Boolean = false, predicate: (T) -> Boolean
@@ -147,4 +149,6 @@ fun <T: JetClassOrObject> StubBasedPsiElementBase<out PsiJetClassOrObjectStub<T>
    }

    return result
-}
+}
+
+fun SearchScope.contains(element: PsiElement): Boolean = PsiSearchScopeUtil.isInScope(this, element)
@@ -0,0 +1,101 @@
+/*
+ * Copyright 2010-2013 JetBrains s.r.o.
+ *
+ * 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 org.jetbrains.jet.plugin.search.declarationsSearch
+
+import com.intellij.psi.search.SearchScope
+import com.intellij.util.Query
+import com.intellij.util.QueryFactory
+import com.intellij.util.Processor
+import com.intellij.openapi.application.QueryExecutorBase
+import com.intellij.openapi.progress.ProgressIndicatorProvider
+import com.intellij.psi.PsiElement
+import com.intellij.openapi.project.Project
+import com.intellij.util.EmptyQuery
+import java.util.Stack
+import java.util.HashSet
+import com.intellij.openapi.progress.ProgressIndicatorProvider
+import com.intellij.psi.search.GlobalSearchScope
+import org.jetbrains.jet.lang.psi.psiUtil.contains
+
+trait DeclarationSearchRequest<in T> {
+    val project: Project
+    val searchScope: SearchScope
+}
+
+public trait SearchRequestWithElement<T : PsiElement> : DeclarationSearchRequest<T> {
+    val originalElement: T
+    override val project: Project get() = originalElement.getProject()
+}
+
+abstract class DeclarationsSearch<T: PsiElement, R: DeclarationSearchRequest<T>>: QueryFactory<T, R>() {
+    {
+        registerExecutor(
+                object : QueryExecutorBase<T, R>(true) {
+                    override fun processQuery(queryParameters: R, consumer: Processor<T>) {
+                        doSearch(queryParameters, consumer)
+                    }
+                }
+        )
+    }
+
+    protected abstract fun doSearch(request: R, consumer: Processor<T>)
+    protected open fun isApplicable(request: R): Boolean = true
+
+    fun search(request: R): Query<T> = if (isApplicable(request)) createUniqueResultsQuery(request) else EmptyQuery.getEmptyQuery<T>()
+}
+
+public class HierarchySearchRequest<T: PsiElement> (
+        override val originalElement: T,
+        override val searchScope: SearchScope,
+        val searchDeeply: Boolean = true
+) : SearchRequestWithElement<T>
+
+abstract class HierarchyTraverser<T> {
+    protected abstract fun nextElements(current: T): Iterable<T>
+    protected abstract fun shouldDescend(element: T): Boolean
+
+    fun forEach(initialElement: T, body: (T) -> Unit) {
+        val stack = Stack<T>()
+        val processed = HashSet<T>()
+
+        stack.push(initialElement)
+        while (!stack.empty) {
+            ProgressIndicatorProvider.checkCanceled()
+
+            val current = stack.pop()!!
+            if (!processed.add(current)) continue
+
+            for (next in nextElements(current)) {
+                ProgressIndicatorProvider.checkCanceled()
+
+                body(next)
+
+                if (shouldDescend(next)) {
+                    stack.push(next)
+                }
+            }
+        }
+    }
+}
+
+fun <T: PsiElement> Processor<T>.consumeHierarchy(request: SearchRequestWithElement<T>, traverser: HierarchyTraverser<T>) {
+    traverser.forEach(request.originalElement) { element ->
+        if (element in request.searchScope) {
+            process(element)
+        }
+    }
+}