support for traversing AST tree with parens and objectcomps

jamesonjlee · sparkprime · commit a1964b49f189 · 2018-06-06T02:06:10.000-04:00
diff --git a/parser/context.go b/parser/context.go
@@ -263,7 +263,7 @@ func specialChildren(node ast.Node) []ast.Node {
 	case *ast.ArrayComp:
 		return []ast.Node{node.Body}
 	case *ast.ObjectComp:
-
+		return inObjectFieldsChildren(node.Fields)
 	case *ast.Self:
 		return nil
 	case *ast.SuperIndex:
@@ -274,6 +274,8 @@ func specialChildren(node ast.Node) []ast.Node {
 		return nil
 	case *ast.Var:
 		return nil
+	case *ast.Parens:
+		return nil
 	}
 	panic(fmt.Sprintf("specialChildren: Unknown node %#v", node))
 }
