From 4cdd9c06b32e76c9c72969d4af56da4b05e9a53f Mon Sep 17 00:00:00 2001
From: rsteube <rsteube@users.noreply.github.com>
Date: Sun, 1 Oct 2023 19:11:46 +0200
Subject: [PATCH] traverse: collect context information about arguments

---
 traverse.go | 46 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/traverse.go b/traverse.go
index 900b63844..c224dfc24 100644
--- a/traverse.go
+++ b/traverse.go
@@ -8,8 +8,18 @@ import (
 	"github.com/rsteube/carapace/internal/pflagfork"
 	"github.com/rsteube/carapace/pkg/style"
 	"github.com/spf13/cobra"
+	"gopkg.in/yaml.v3"
 )
 
+type inArgsX []inArgX
+type inArgX struct {
+	Value       string  `yaml:",omitempty"`
+	Name        string  `yaml:",omitempty"`
+	Description string  `yaml:",omitempty"`
+	Type        string  `yaml:",omitempty"`
+	OptArg      *inArgX `yaml:",omitempty"`
+}
+
 func traverse(c *cobra.Command, args []string) (Action, Context) {
 	LOG.Printf("traverse called for %#v with args %#v\n", c.Name(), args)
 	storage.preRun(c, args)
@@ -19,6 +29,7 @@ func traverse(c *cobra.Command, args []string) (Action, Context) {
 		c.FParseErrWhitelist.UnknownFlags = true
 	}
 
+	inArgsX := make(inArgsX, 0)
 	inArgs := []string{}        // args consumed by current command
 	inPositionals := []string{} // positionals consumed by current command
 	var inFlag *pflagfork.Flag  // last encountered flag that still expects arguments
@@ -32,6 +43,10 @@ loop:
 		// flag argument
 		case inFlag != nil && inFlag.Consumes(arg):
 			LOG.Printf("arg %#v is a flag argument\n", arg)
+			inArgsX = append(inArgsX, inArgX{
+				Value: arg,
+				Type:  "flag argument",
+			})
 			inArgs = append(inArgs, arg)
 			inFlag.Args = append(inFlag.Args, arg)
 
@@ -43,6 +58,10 @@ loop:
 		// dash
 		case arg == "--":
 			LOG.Printf("arg %#v is dash\n", arg)
+			inArgsX = append(inArgsX, inArgX{
+				Type:  "dash",
+				Value: arg,
+			})
 			inArgs = append(inArgs, context.Args[i:]...)
 			break loop
 
@@ -54,6 +73,17 @@ loop:
 
 			if inFlag == nil {
 				LOG.Printf("flag %#v is unknown", arg)
+				inArgsX = append(inArgsX, inArgX{
+					Type:  "unknown flag",
+					Value: arg,
+				})
+			} else {
+				inArgsX = append(inArgsX, inArgX{
+					Name:        inFlag.Name,
+					Description: inFlag.Usage,
+					Type:        "flag", // TODO flagtype
+					Value:       arg,
+				})
 			}
 			continue
 
@@ -73,16 +103,30 @@ loop:
 				context.Args = c.Flags().Args()
 			}
 
-			return traverse(subcommand(c, arg), args[i+1:])
+			subCmd := subcommand(c, arg)
+			inArgsX = append(inArgsX, inArgX{
+				Name:        subCmd.Name(),
+				Description: subCmd.Short,
+				Type:        "subcommand",
+				Value:       arg,
+			})
+			return traverse(subCmd, args[i+1:])
 
 		// positional
 		default:
 			LOG.Printf("arg %#v is a positional\n", arg)
+			inArgsX = append(inArgsX, inArgX{
+				Type:  "positional argument",
+				Value: arg,
+			})
 			inArgs = append(inArgs, arg)
 			inPositionals = append(inPositionals, arg)
 		}
 	}
 
+	m, _ := yaml.Marshal(inArgsX)
+	LOG.Println(string(m))
+
 	toParse := inArgs
 	if inFlag != nil && len(inFlag.Args) == 0 && inFlag.Consumes("") {
 		LOG.Printf("removing arg %#v since it is a flag missing its argument\n", toParse[len(toParse)-1])