From 8d2ac7c4837b6a0d168cce4b72d937a0ef11a8b7 Mon Sep 17 00:00:00 2001
From: Alexey Sedunov <alexey.sedunov@jetbrains.com>
Date: Mon, 16 Dec 2013 14:46:47 +0400
Subject: [PATCH] Implement hierarchical search

---
 .../declarationSearchUtil.kt                  | 30 ++++++++++++++++---
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/idea/src/org/jetbrains/jet/plugin/search/declarationsSearch/declarationSearchUtil.kt b/idea/src/org/jetbrains/jet/plugin/search/declarationsSearch/declarationSearchUtil.kt
index 8cb4bf56d90..b2e0cc9461e 100644
--- a/idea/src/org/jetbrains/jet/plugin/search/declarationsSearch/declarationSearchUtil.kt
+++ b/idea/src/org/jetbrains/jet/plugin/search/declarationsSearch/declarationSearchUtil.kt
@@ -62,11 +62,14 @@ public class HierarchySearchRequest<T: PsiElement> (
         override val originalElement: T,
         override val searchScope: SearchScope,
         val searchDeeply: Boolean = true
-) : SearchRequestWithElement<T>
+) : SearchRequestWithElement<T> {
+    fun copy<U: PsiElement>(newOriginalElement: U): HierarchySearchRequest<U> =
+            HierarchySearchRequest(newOriginalElement, searchScope, searchDeeply)
+}
 
-abstract class HierarchyTraverser<T> {
-    protected abstract fun nextElements(current: T): Iterable<T>
-    protected abstract fun shouldDescend(element: T): Boolean
+trait HierarchyTraverser<T> {
+    protected fun nextElements(current: T): Iterable<T>
+    protected fun shouldDescend(element: T): Boolean
 
     fun forEach(initialElement: T, body: (T) -> Unit) {
         val stack = Stack<T>()
@@ -99,3 +102,22 @@ fun <T: PsiElement> Processor<T>.consumeHierarchy(request: SearchRequestWithElem
         }
     }
 }
+
+abstract class HierarchySearch<T: PsiElement>(
+        protected val traverser: HierarchyTraverser<T>
+): DeclarationsSearch<T, HierarchySearchRequest<T>>() {
+    protected open fun doSearchAll(request: HierarchySearchRequest<T>, consumer: Processor<T>) {
+        consumer.consumeHierarchy(request, traverser)
+    }
+
+    protected abstract fun doSearchDirect(request: HierarchySearchRequest<T>, consumer: Processor<T>)
+
+    protected override fun doSearch(request: HierarchySearchRequest<T>, consumer: Processor<T>) {
+        if (request.searchDeeply) {
+            doSearchAll(request, consumer)
+        }
+        else {
+            doSearchDirect(request, consumer)
+        }
+    }
+}
\ No newline at end of file