Do AST better maybe?

argh/parser2_test.go

@@ -59,19 +59,14 @@ func TestParser2(t *testing.T) {
 					},
 				},
 			},
-			/*
 			expAST: []argh.Node{
 				&argh.Command{
 					Name:   "pies",
 					Values: map[string]string{},
-						Nodes: []argh.Node{
-							&argh.CompoundShortFlag{
 					Nodes: []argh.Node{
 						&argh.Flag{Name: "e"},
 						&argh.Flag{Name: "a"},
 						&argh.Flag{Name: "t"},
-								},
-							},
 						&argh.Flag{Name: "wat"},
 						&argh.Command{
 							Name: "hello",
@@ -83,7 +78,6 @@ func TestParser2(t *testing.T) {
 					},
 				},
 			},
-			*/
 		},
 		{
 			name: "bare",
@@ -432,12 +426,13 @@ func TestParser2(t *testing.T) {
 			},
 		},
 	} {
+		if tc.expPT != nil {
+			t.Run(tc.name+" parse tree", func(ct *testing.T) {
 				if tc.skip {
-			continue
+					ct.SkipNow()
+					return
 				}
 
-		if tc.expPT != nil {
-			t.Run(tc.name+" parse tree", func(ct *testing.T) {
 				pt, err := argh.ParseArgs2(tc.args, tc.cfg)
 				if err != nil {
 					assert.ErrorIs(ct, err, tc.expErr)
@@ -452,6 +447,11 @@ func TestParser2(t *testing.T) {
 
 		if tc.expAST != nil {
 			t.Run(tc.name+" ast", func(ct *testing.T) {
+				if tc.skip {
+					ct.SkipNow()
+					return
+				}
+
 				pt, err := argh.ParseArgs2(tc.args, tc.cfg)
 				if err != nil {
 					ct.Logf("err=%+#v", err)

argh/querier.go

@@ -1,10 +1,7 @@
 package argh
 
-import "fmt"
-
 type Querier interface {
 	Program() (Command, bool)
-	TypedAST() []TypedNode
 	AST() []Node
 }
 
@@ -25,50 +22,34 @@ func (dq *defaultQuerier) Program() (Command, bool) {
 	return v, ok
 }
 
-func (dq *defaultQuerier) TypedAST() []TypedNode {
-	ret := []TypedNode{}
-
-	for _, node := range dq.nodes {
-		if _, ok := node.(ArgDelimiter); ok {
-			continue
-		}
-
-		if _, ok := node.(StopFlag); ok {
-			continue
-		}
-
-		ret = append(
-			ret,
-			TypedNode{
-				Type: fmt.Sprintf("%T", node),
-				Node: node,
-			},
-		)
-	}
-
-	return ret
-}
-
 func (dq *defaultQuerier) AST() []Node {
 	ret := []Node{}
 
-	for _, node := range dq.nodes {
+	for i, node := range dq.nodes {
-		if _, ok := node.(ArgDelimiter); ok {
+		tracef("AST i=%d node type=%T", i, node)
+
+		if _, ok := node.(*ArgDelimiter); ok {
 			continue
 		}
 
-		if _, ok := node.(StopFlag); ok {
+		if _, ok := node.(*StopFlag); ok {
 			continue
 		}
 
-		if v, ok := node.(CompoundShortFlag); ok {
+		if v, ok := node.(*CompoundShortFlag); ok {
 			ret = append(ret, NewQuerier(v.Nodes).AST()...)
 
 			continue
 		}
 
-		if v, ok := node.(Command); ok {
+		if v, ok := node.(*Command); ok {
-			ret = append(ret, NewQuerier(v.Nodes).AST()...)
+			ret = append(
+				ret,
+				&Command{
+					Name:   v.Name,
+					Values: v.Values,
+					Nodes:  NewQuerier(v.Nodes).AST(),
+				})
 
 			continue
 		}