Merge pull request #216 from smacker/search_array

Return correct result when filter returns array
bblfsh · Jun 11, 2019 · c7adde0 · c7adde0
2 parents 9ac15b7 + becc671
commit c7adde0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/server/server.go b/server/server.go
@@ -125,7 +125,15 @@ func (s *Server) handleParse(ctx *gin.Context) {
 		}
 		results := nodes.Array{}
 		for iter.Next() {
-			results = append(results, iter.Node().(nodes.Node))
+			n := iter.Node()
+			if nodes.KindOf(n) == nodes.KindArray {
+				for _, child := range n.(nodes.Array) {
+					results = append(results, child)
+				}
+			} else {
+				results = append(results, n.(nodes.Node))
+			}
+
 		}
 		resp = results
 	}