Return correct result when filter returns array

Fix: #215 Signed-off-by: Maxim Sukharev <[email protected]>
bblfsh · Jun 7, 2019 · becc671 · becc671
1 parent 9ac15b7
commit becc671
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
 	}