From fbd1c9bca378c215619e9d84d562ffc64456f1c5 Mon Sep 17 00:00:00 2001
From: Daniel Wedul <github@wedul.com>
Date: Wed, 15 May 2024 14:05:43 -0600
Subject: [PATCH] [1760]: Create a hidden tree command that will list all
 possible commands available (even hiddden ones).

---
 cmd/provenanced/cmd/docgen.go | 86 +++++++++++++++++++++++++++++++++++
 cmd/provenanced/cmd/root.go   |  1 +
 2 files changed, 87 insertions(+)

diff --git a/cmd/provenanced/cmd/docgen.go b/cmd/provenanced/cmd/docgen.go
index f59b6e1ea1..4bd4889ffa 100644
--- a/cmd/provenanced/cmd/docgen.go
+++ b/cmd/provenanced/cmd/docgen.go
@@ -3,6 +3,7 @@ package cmd
 import (
 	"fmt"
 	"os"
+	"strings"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/cobra/doc"
@@ -101,3 +102,88 @@ func exists(dir string) bool {
 	_, err := os.Stat(dir)
 	return err == nil
 }
+
+// GetTreeCmd gets the tree command that will output all the commands available.
+func GetTreeCmd() *cobra.Command {
+	aliasesFlag := "aliases"
+	cmd := &cobra.Command{
+		Use:    "tree [sub-command]",
+		Short:  "Get a tree of the commands optionally limited to a specific sub-command",
+		Hidden: true,
+		Args:   cobra.ArbitraryArgs,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) > 0 && args[0] != "provenanced" {
+				args = append([]string{"provenanced"}, args...)
+			}
+
+			incAliases, _ := cmd.Flags().GetBool(aliasesFlag)
+			cmds := getAllCommands(cmd.Root(), "", args, incAliases)
+			cmd.Printf(strings.Join(cmds, "\n") + "\n")
+			return nil
+		},
+	}
+	cmd.Flags().Bool(aliasesFlag, false, "Include command aliases in the output")
+
+	return cmd
+}
+
+// hasNameOrAlias returns true if the provided command as a name or an alias equal to the provided arg.
+func hasNameOrAlias(cmd *cobra.Command, arg string) bool {
+	if cmd.Name() == arg {
+		return true
+	}
+	for _, alias := range cmd.Aliases {
+		if alias == arg {
+			return true
+		}
+	}
+	return false
+}
+
+// getNameAndAliases gets a list of the command's name and aliases (without any duplicates).
+func getNameAndAliases(cmd *cobra.Command) []string {
+	rv := make([]string, 1, 1+len(cmd.Aliases))
+	rv[0] = cmd.Name()
+	for _, alias := range cmd.Aliases {
+		known := false
+		for _, prev := range rv {
+			if prev == alias {
+				known = true
+			}
+		}
+		if !known {
+			rv = append(rv, alias)
+		}
+	}
+	return rv
+}
+
+// getAllCommands gets all of the possible commands under the given command.
+func getAllCommands(cmd *cobra.Command, prev string, limitTo []string, incAliases bool) []string {
+	if len(limitTo) > 0 {
+		if !hasNameOrAlias(cmd, limitTo[0]) {
+			return nil
+		}
+		limitTo = limitTo[1:]
+	}
+
+	name := cmd.Name()
+	if incAliases {
+		names := getNameAndAliases(cmd)
+		if len(names) > 1 {
+			name = "[" + strings.Join(names, " ") + "]"
+		}
+	}
+	cur := strings.TrimSpace(prev + " " + name)
+
+	subCmds := cmd.Commands()
+	if len(subCmds) == 0 {
+		return []string{cur}
+	}
+
+	var rv []string
+	for _, subCmd := range subCmds {
+		rv = append(rv, getAllCommands(subCmd, cur, limitTo, incAliases)...)
+	}
+	return rv
+}
diff --git a/cmd/provenanced/cmd/root.go b/cmd/provenanced/cmd/root.go
index 5910743788..9f829bb840 100644
--- a/cmd/provenanced/cmd/root.go
+++ b/cmd/provenanced/cmd/root.go
@@ -185,6 +185,7 @@ func initRootCmd(rootCmd *cobra.Command, encodingConfig params.EncodingConfig, b
 		snapshot.Cmd(newApp),
 		GetPreUpgradeCmd(),
 		GetDocGenCmd(),
+		GetTreeCmd(),
 		pruning.Cmd(newApp, app.DefaultNodeHome),
 	)