From 3e7f87db14ddd7f4b20e85550602595ec91e3042 Mon Sep 17 00:00:00 2001
From: skanehira <sho19921005@gmail.com>
Date: Sat, 2 Nov 2019 01:43:59 +0900
Subject: [PATCH] fix a problem with searching

---
 gui/gui.go | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/gui/gui.go b/gui/gui.go
index 08d0f08..01757c7 100644
--- a/gui/gui.go
+++ b/gui/gui.go
@@ -130,7 +130,7 @@ func (g *Gui) Search() {
 			g.Tree.SetRoot(&root)
 			if text != "" {
 				root := g.Tree.GetRoot()
-				root.SetChildren(g.walk(root, text))
+				root.SetChildren(g.walk(root.GetChildren(), text))
 			}
 		})
 		input.SetLabel("word").SetLabelWidth(5).SetDoneFunc(func(key tcell.Key) {
@@ -143,19 +143,19 @@ func (g *Gui) Search() {
 	}
 }
 
-func (g *Gui) walk(node *tview.TreeNode, text string) []*tview.TreeNode {
-	var nodes []*tview.TreeNode
-	if strings.Index(strings.ToLower(node.GetText()), text) != -1 {
-		nodes = append(nodes, node)
-		return nodes
-	}
+func (g *Gui) walk(nodes []*tview.TreeNode, text string) []*tview.TreeNode {
+	var newNodes []*tview.TreeNode
 
-	for _, node := range node.GetChildren() {
-		nodes = append(nodes, g.walk(node, text)...)
+	for _, child := range nodes {
+		log.Println(child.GetText())
+		if strings.Index(strings.ToLower(child.GetText()), text) != -1 {
+			newNodes = append(newNodes, child)
+		} else {
+			newNodes = append(newNodes, g.walk(child.GetChildren(), text)...)
+		}
 	}
 
-	return nodes
-
+	return newNodes
 }
 
 func (g *Gui) SaveJSON() {