JMS #39: Removed all calls to os.Exit().

This commit is contained in:
Jeremy Saenz 2013-11-01 06:45:19 -07:00
parent e6e641143c
commit b25b7a883c
3 changed files with 26 additions and 15 deletions

12
app.go
View File

@ -57,16 +57,20 @@ func (a *App) Run(arguments []string) error {
return err return err
} }
checkHelp(context) if checkHelp(context) {
checkVersion(context) return nil
}
if checkVersion(context) {
return nil
}
args := context.Args() args := context.Args()
if len(args) > 0 { if len(args) > 0 {
name := args[0] name := args[0]
c := a.Command(name) c := a.Command(name)
if c != nil { if c != nil {
c.Run(context) return c.Run(context)
return nil
} }
} }

View File

@ -3,7 +3,6 @@ package cli
import ( import (
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"os"
"strings" "strings"
) )
@ -24,7 +23,7 @@ type Command struct {
} }
// Invokes the command given the context, parses ctx.Args() to generate command-specific flags // Invokes the command given the context, parses ctx.Args() to generate command-specific flags
func (c Command) Run(ctx *Context) { func (c Command) Run(ctx *Context) error {
// append help to flags // append help to flags
c.Flags = append( c.Flags = append(
c.Flags, c.Flags,
@ -55,12 +54,15 @@ func (c Command) Run(ctx *Context) {
fmt.Println("Incorrect Usage.\n") fmt.Println("Incorrect Usage.\n")
ShowCommandHelp(ctx, c.Name) ShowCommandHelp(ctx, c.Name)
fmt.Println("") fmt.Println("")
os.Exit(1) return err
} }
context := NewContext(ctx.App, set, ctx.globalSet) context := NewContext(ctx.App, set, ctx.globalSet)
checkCommandHelp(context, c.Name) if checkCommandHelp(context, c.Name) {
return nil
}
c.Action(context) c.Action(context)
return nil
} }
// Returns true if Command.Name or Command.ShortName matches given name // Returns true if Command.Name or Command.ShortName matches given name

19
help.go
View File

@ -73,7 +73,6 @@ func ShowCommandHelp(c *Context, command string) {
} }
fmt.Printf("No help topic for '%v'\n", command) fmt.Printf("No help topic for '%v'\n", command)
os.Exit(1)
} }
// Prints the version number of the App // Prints the version number of the App
@ -88,23 +87,29 @@ func printHelp(templ string, data interface{}) {
w.Flush() w.Flush()
} }
func checkVersion(c *Context) { func checkVersion(c *Context) bool {
if c.GlobalBool("version") { if c.GlobalBool("version") {
ShowVersion(c) ShowVersion(c)
os.Exit(0) return true
} }
return false
} }
func checkHelp(c *Context) { func checkHelp(c *Context) bool {
if c.GlobalBool("h") || c.GlobalBool("help") { if c.GlobalBool("h") || c.GlobalBool("help") {
ShowAppHelp(c) ShowAppHelp(c)
os.Exit(0) return true
} }
return false
} }
func checkCommandHelp(c *Context, name string) { func checkCommandHelp(c *Context, name string) bool {
if c.Bool("h") || c.Bool("help") { if c.Bool("h") || c.Bool("help") {
ShowCommandHelp(c, name) ShowCommandHelp(c, name)
os.Exit(0) return true
} }
return false
} }