Adjust command hiding to use similar convention as hidden flags

plus breaking out "setup" portion of `App.Run` into its own method, cleaning up
some bits of the help templates, and allowing for runtime opt-in of displaying
template errors to stderr.
main
Dan Buch 8 years ago
parent f397b1618c
commit cc481d6b0e
No known key found for this signature in database
GPG Key ID: FAEF12936DD3E3EC

@ -84,6 +84,8 @@ type App struct {
Writer io.Writer Writer io.Writer
// Other custom info // Other custom info
Metadata map[string]interface{} Metadata map[string]interface{}
didSetup bool
} }
// Tries to find out when this binary was compiled. // Tries to find out when this binary was compiled.
@ -111,8 +113,16 @@ func NewApp() *App {
} }
} }
// Entry point to the cli app. Parses the arguments slice and routes to the proper flag/args combination // Setup runs initialization code to ensure all data structures are ready for
func (a *App) Run(arguments []string) (err error) { // `Run` or inspection prior to `Run`. It is internally called by `Run`, but
// will return early if setup has already happened.
func (a *App) Setup() {
if a.didSetup {
return
}
a.didSetup = true
if a.Author != "" || a.Email != "" { if a.Author != "" || a.Email != "" {
a.Authors = append(a.Authors, Author{Name: a.Author, Email: a.Email}) a.Authors = append(a.Authors, Author{Name: a.Author, Email: a.Email})
} }
@ -148,6 +158,11 @@ func (a *App) Run(arguments []string) (err error) {
if !a.HideVersion { if !a.HideVersion {
a.appendFlag(VersionFlag) a.appendFlag(VersionFlag)
} }
}
// Entry point to the cli app. Parses the arguments slice and routes to the proper flag/args combination
func (a *App) Run(arguments []string) (err error) {
a.Setup()
// parse flags // parse flags
set := flagSet(a.Name, a.Flags) set := flagSet(a.Name, a.Flags)
@ -357,11 +372,41 @@ func (a *App) Command(name string) *Command {
return nil return nil
} }
// Returnes the array containing all the categories with the commands they contain // Categories returns a slice containing all the categories with the commands they contain
func (a *App) Categories() CommandCategories { func (a *App) Categories() CommandCategories {
return a.categories return a.categories
} }
// VisibleCategories returns a slice of categories and commands that are
// Hidden=false
func (a *App) VisibleCategories() []*CommandCategory {
ret := []*CommandCategory{}
for _, category := range a.categories {
if visible := func() *CommandCategory {
for _, command := range category.Commands {
if !command.Hidden {
return category
}
}
return nil
}(); visible != nil {
ret = append(ret, visible)
}
}
return ret
}
// VisibleCommands returns a slice of the Commands with Hidden=false
func (a *App) VisibleCommands() []Command {
ret := []Command{}
for _, command := range a.Commands {
if !command.Hidden {
ret = append(ret, command)
}
}
return ret
}
// VisibleFlags returns a slice of the Flags with Hidden=false // VisibleFlags returns a slice of the Flags with Hidden=false
func (a *App) VisibleFlags() []Flag { func (a *App) VisibleFlags() []Flag {
return visibleFlags(a.Flags) return visibleFlags(a.Flags)

@ -1124,8 +1124,8 @@ func TestApp_Run_Categories(t *testing.T) {
output := buf.String() output := buf.String()
t.Logf("output: %q\n", buf.Bytes()) t.Logf("output: %q\n", buf.Bytes())
if !strings.Contains(output, "1:\n command1") { if !strings.Contains(output, "1:\n command1") {
t.Errorf("want buffer to include category %q, did not: \n%q", "1:\n command1", output) t.Errorf("want buffer to include category %q, did not: \n%q", "1:\n command1", output)
} }
} }

@ -28,3 +28,14 @@ func (c CommandCategories) AddCommand(category string, command Command) CommandC
} }
return append(c, &CommandCategory{Name: category, Commands: []Command{command}}) return append(c, &CommandCategory{Name: category, Commands: []Command{command}})
} }
// VisibleCommands returns a slice of the Commands with Hidden=false
func (c *CommandCategory) VisibleCommands() []Command {
ret := []Command{}
for _, command := range c.Commands {
if !command.Hidden {
ret = append(ret, command)
}
}
return ret
}

@ -34,7 +34,7 @@ func TestHandleExitCoder_ExitCoder(t *testing.T) {
defer func() { OsExiter = os.Exit }() defer func() { OsExiter = os.Exit }()
HandleExitCoder(NewExitError("galactic perimiter breach", 9)) HandleExitCoder(NewExitError("galactic perimeter breach", 9))
expect(t, exitCode, 9) expect(t, exitCode, 9)
expect(t, called, true) expect(t, called, true)
@ -51,7 +51,7 @@ func TestHandleExitCoder_MultiErrorWithExitCoder(t *testing.T) {
defer func() { OsExiter = os.Exit }() defer func() { OsExiter = os.Exit }()
exitErr := NewExitError("galactic perimiter breach", 9) exitErr := NewExitError("galactic perimeter breach", 9)
err := NewMultiError(errors.New("wowsa"), errors.New("egad"), exitErr) err := NewMultiError(errors.New("wowsa"), errors.New("egad"), exitErr)
HandleExitCoder(err) HandleExitCoder(err)

@ -3,6 +3,7 @@ package cli
import ( import (
"fmt" "fmt"
"io" "io"
"os"
"strings" "strings"
"text/tabwriter" "text/tabwriter"
"text/template" "text/template"
@ -21,15 +22,15 @@ VERSION:
{{.Version}} {{.Version}}
{{end}}{{end}}{{if len .Authors}} {{end}}{{end}}{{if len .Authors}}
AUTHOR(S): AUTHOR(S):
{{range .Authors}}{{ . }}{{end}} {{range .Authors}}{{.}}{{end}}
{{end}}{{if .Commands}} {{end}}{{if .VisibleCommands}}
COMMANDS:{{range .Categories}}{{if .Name}} COMMANDS:{{range .VisibleCategories}}{{if .Name}}
{{.Name}}{{ ":" }}{{end}}{{range .Commands}} {{.Name}}:{{end}}{{range .VisibleCommands}}
{{if not .Hidden}}{{.Name}}{{with .ShortName}}, {{.}}{{end}}{{ "\t" }}{{.Usage}}{{end}}{{end}} {{.Name}}{{with .ShortName}}, {{.}}{{end}}{{"\t"}}{{.Usage}}{{end}}
{{end}}{{end}}{{if .VisibleFlags}} {{end}}{{end}}{{if .VisibleFlags}}
GLOBAL OPTIONS: GLOBAL OPTIONS:
{{range .VisibleFlags}}{{.}} {{range .VisibleFlags}}{{.}}
{{end}}{{end}}{{if .Copyright }} {{end}}{{end}}{{if .Copyright}}
COPYRIGHT: COPYRIGHT:
{{.Copyright}} {{.Copyright}}
{{end}} {{end}}
@ -52,7 +53,7 @@ DESCRIPTION:
OPTIONS: OPTIONS:
{{range .VisibleFlags}}{{.}} {{range .VisibleFlags}}{{.}}
{{end}}{{ end }} {{end}}{{end}}
` `
// The text template for the subcommand help topic. // The text template for the subcommand help topic.
@ -64,9 +65,9 @@ var SubcommandHelpTemplate = `NAME:
USAGE: USAGE:
{{.HelpName}} command{{if .VisibleFlags}} [command options]{{end}} {{if .ArgsUsage}}{{.ArgsUsage}}{{else}}[arguments...]{{end}} {{.HelpName}} command{{if .VisibleFlags}} [command options]{{end}} {{if .ArgsUsage}}{{.ArgsUsage}}{{else}}[arguments...]{{end}}
COMMANDS:{{range .Categories}}{{if .Name}} COMMANDS:{{range .VisibleCategories}}{{if .Name}}
{{.Name}}{{ ":" }}{{end}}{{range .Commands}} {{.Name}}:{{end}}{{range .VisibleCommands}}
{{if not .Hidden}}{{.Name}}{{with .ShortName}}, {{.}}{{end}}{{ "\t" }}{{.Usage}}{{end}}{{end}} {{.Name}}{{with .ShortName}}, {{.}}{{end}}{{"\t"}}{{.Usage}}{{end}}
{{end}}{{if .VisibleFlags}} {{end}}{{if .VisibleFlags}}
OPTIONS: OPTIONS:
{{range .VisibleFlags}}{{.}} {{range .VisibleFlags}}{{.}}
@ -191,7 +192,10 @@ func printHelp(out io.Writer, templ string, data interface{}) {
err := t.Execute(w, data) err := t.Execute(w, data)
if err != nil { if err != nil {
// If the writer is closed, t.Execute will fail, and there's nothing // If the writer is closed, t.Execute will fail, and there's nothing
// we can do to recover. We could send this to os.Stderr if we need. // we can do to recover.
if os.Getenv("CLI_TEMPLATE_ERROR_DEBUG") != "" {
fmt.Fprintf(os.Stderr, "CLI TEMPLATE ERROR: %#v\n", err)
}
return return
} }
w.Flush() w.Flush()

@ -136,10 +136,10 @@ func TestShowAppHelp_HiddenCommand(t *testing.T) {
app.Run([]string{"app", "--help"}) app.Run([]string{"app", "--help"})
if strings.Contains(output.String(), "secretfrob") { if strings.Contains(output.String(), "secretfrob") {
t.Fatalf("expected output to exclude \"secretfrob\"; got: %q", output.String()) t.Errorf("expected output to exclude \"secretfrob\"; got: %q", output.String())
} }
if !strings.Contains(output.String(), "frobbly") { if !strings.Contains(output.String(), "frobbly") {
t.Fatalf("expected output to include \"frobbly\"; got: %q", output.String()) t.Errorf("expected output to include \"frobbly\"; got: %q", output.String())
} }
} }

Loading…
Cancel
Save