2013-07-18 00:00:02 +00:00
|
|
|
package cli
|
|
|
|
|
|
|
|
import (
|
2013-11-20 08:05:18 +00:00
|
|
|
"errors"
|
2013-11-21 01:25:13 +00:00
|
|
|
"flag"
|
2014-12-02 17:44:55 +00:00
|
|
|
"fmt"
|
2017-03-31 07:24:15 +00:00
|
|
|
"os"
|
2016-07-31 21:14:46 +00:00
|
|
|
"reflect"
|
2013-11-18 23:37:59 +00:00
|
|
|
"strings"
|
2016-09-12 03:32:43 +00:00
|
|
|
"syscall"
|
2013-07-18 00:00:02 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// Context is a type that is passed through to
|
|
|
|
// each Handler action in a cli application. Context
|
|
|
|
// can be used to retrieve context-specific Args and
|
|
|
|
// parsed command-line options.
|
|
|
|
type Context struct {
|
2016-07-31 21:14:46 +00:00
|
|
|
App *App
|
|
|
|
Command Command
|
2016-11-14 16:35:22 +00:00
|
|
|
shellComplete bool
|
2016-07-31 21:14:46 +00:00
|
|
|
flagSet *flag.FlagSet
|
|
|
|
setFlags map[string]bool
|
|
|
|
parentContext *Context
|
2013-07-18 00:00:02 +00:00
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// NewContext creates a new context. For use in when invoking an App or Command action.
|
2015-05-18 15:39:48 +00:00
|
|
|
func NewContext(app *App, set *flag.FlagSet, parentCtx *Context) *Context {
|
2016-11-04 20:56:28 +00:00
|
|
|
c := &Context{App: app, flagSet: set, parentContext: parentCtx}
|
2013-12-03 13:42:09 +00:00
|
|
|
|
2016-11-04 20:56:28 +00:00
|
|
|
if parentCtx != nil {
|
2016-11-14 16:35:22 +00:00
|
|
|
c.shellComplete = parentCtx.shellComplete
|
2016-11-04 20:56:28 +00:00
|
|
|
}
|
2013-07-19 01:00:52 +00:00
|
|
|
|
2016-11-04 20:56:28 +00:00
|
|
|
return c
|
2014-03-06 01:24:22 +00:00
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// NumFlags returns the number of flags set
|
2015-02-26 21:51:01 +00:00
|
|
|
func (c *Context) NumFlags() int {
|
|
|
|
return c.flagSet.NFlag()
|
2013-07-19 01:00:52 +00:00
|
|
|
}
|
|
|
|
|
2015-06-08 20:21:26 +00:00
|
|
|
// Set sets a context flag to a value.
|
|
|
|
func (c *Context) Set(name, value string) error {
|
2017-01-11 12:38:54 +00:00
|
|
|
c.setFlags = nil
|
2015-06-08 20:21:26 +00:00
|
|
|
return c.flagSet.Set(name, value)
|
2013-09-24 01:41:31 +00:00
|
|
|
}
|
|
|
|
|
2016-04-30 15:46:47 +00:00
|
|
|
// GlobalSet sets a context flag to a value on the global flagset
|
|
|
|
func (c *Context) GlobalSet(name, value string) error {
|
2017-01-11 12:38:54 +00:00
|
|
|
globalContext(c).setFlags = nil
|
2016-04-30 15:46:47 +00:00
|
|
|
return globalContext(c).flagSet.Set(name, value)
|
2013-09-24 19:36:01 +00:00
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// IsSet determines if the flag was actually set
|
2013-12-18 18:09:16 +00:00
|
|
|
func (c *Context) IsSet(name string) bool {
|
|
|
|
if c.setFlags == nil {
|
|
|
|
c.setFlags = make(map[string]bool)
|
2014-04-15 14:16:47 +00:00
|
|
|
|
2013-12-18 18:09:16 +00:00
|
|
|
c.flagSet.Visit(func(f *flag.Flag) {
|
|
|
|
c.setFlags[f.Name] = true
|
|
|
|
})
|
2013-07-19 01:00:52 +00:00
|
|
|
|
2016-07-31 21:14:46 +00:00
|
|
|
c.flagSet.VisitAll(func(f *flag.Flag) {
|
|
|
|
if _, ok := c.setFlags[f.Name]; ok {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.setFlags[f.Name] = false
|
|
|
|
})
|
2014-08-02 21:32:32 +00:00
|
|
|
|
2016-07-31 21:14:46 +00:00
|
|
|
// XXX hack to support IsSet for flags with EnvVar
|
|
|
|
//
|
|
|
|
// There isn't an easy way to do this with the current implementation since
|
|
|
|
// whether a flag was set via an environment variable is very difficult to
|
|
|
|
// determine here. Instead, we intend to introduce a backwards incompatible
|
|
|
|
// change in version 2 to add `IsSet` to the Flag interface to push the
|
|
|
|
// responsibility closer to where the information required to determine
|
|
|
|
// whether a flag is set by non-standard means such as environment
|
2017-10-06 05:28:18 +00:00
|
|
|
// variables is available.
|
2016-07-31 21:14:46 +00:00
|
|
|
//
|
|
|
|
// See https://github.com/urfave/cli/issues/294 for additional discussion
|
|
|
|
flags := c.Command.Flags
|
|
|
|
if c.Command.Name == "" { // cannot == Command{} since it contains slice types
|
|
|
|
if c.App != nil {
|
|
|
|
flags = c.App.Flags
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for _, f := range flags {
|
|
|
|
eachName(f.GetName(), func(name string) {
|
|
|
|
if isSet, ok := c.setFlags[name]; isSet || !ok {
|
|
|
|
return
|
|
|
|
}
|
2013-07-19 01:00:52 +00:00
|
|
|
|
2016-09-09 04:10:46 +00:00
|
|
|
val := reflect.ValueOf(f)
|
|
|
|
if val.Kind() == reflect.Ptr {
|
|
|
|
val = val.Elem()
|
|
|
|
}
|
2013-07-19 01:00:52 +00:00
|
|
|
|
2017-03-31 07:24:15 +00:00
|
|
|
filePathValue := val.FieldByName("FilePath")
|
|
|
|
if filePathValue.IsValid() {
|
|
|
|
eachName(filePathValue.String(), func(filePath string) {
|
|
|
|
if _, err := os.Stat(filePath); err == nil {
|
|
|
|
c.setFlags[name] = true
|
|
|
|
return
|
|
|
|
}
|
|
|
|
})
|
2016-08-01 03:09:57 +00:00
|
|
|
}
|
2013-09-24 19:36:01 +00:00
|
|
|
|
2017-03-31 07:24:15 +00:00
|
|
|
envVarValue := val.FieldByName("EnvVar")
|
|
|
|
if envVarValue.IsValid() {
|
|
|
|
eachName(envVarValue.String(), func(envVar string) {
|
|
|
|
envVar = strings.TrimSpace(envVar)
|
|
|
|
if _, ok := syscall.Getenv(envVar); ok {
|
|
|
|
c.setFlags[name] = true
|
|
|
|
return
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
2016-07-31 21:14:46 +00:00
|
|
|
})
|
|
|
|
}
|
2013-12-18 18:09:16 +00:00
|
|
|
}
|
2013-09-24 19:36:01 +00:00
|
|
|
|
2016-07-31 21:14:46 +00:00
|
|
|
return c.setFlags[name]
|
2014-04-15 14:16:47 +00:00
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// GlobalIsSet determines if the global flag was actually set
|
2014-11-11 20:48:07 +00:00
|
|
|
func (c *Context) GlobalIsSet(name string) bool {
|
2016-07-31 21:14:46 +00:00
|
|
|
ctx := c
|
|
|
|
if ctx.parentContext != nil {
|
|
|
|
ctx = ctx.parentContext
|
2013-12-18 18:09:16 +00:00
|
|
|
}
|
|
|
|
|
2016-07-31 21:14:46 +00:00
|
|
|
for ; ctx != nil; ctx = ctx.parentContext {
|
|
|
|
if ctx.IsSet(name) {
|
|
|
|
return true
|
2015-05-18 15:39:48 +00:00
|
|
|
}
|
2014-11-11 20:48:07 +00:00
|
|
|
}
|
2016-07-31 21:14:46 +00:00
|
|
|
return false
|
2014-11-11 20:48:07 +00:00
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// FlagNames returns a slice of flag names used in this context.
|
2014-06-14 03:09:28 +00:00
|
|
|
func (c *Context) FlagNames() (names []string) {
|
|
|
|
for _, flag := range c.Command.Flags {
|
2015-12-15 17:29:03 +00:00
|
|
|
name := strings.Split(flag.GetName(), ",")[0]
|
2014-06-14 03:09:28 +00:00
|
|
|
if name == "help" {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
names = append(names, name)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// GlobalFlagNames returns a slice of global flag names used by the app.
|
2014-10-02 22:16:29 +00:00
|
|
|
func (c *Context) GlobalFlagNames() (names []string) {
|
|
|
|
for _, flag := range c.App.Flags {
|
2015-12-15 17:29:03 +00:00
|
|
|
name := strings.Split(flag.GetName(), ",")[0]
|
2014-10-02 22:16:29 +00:00
|
|
|
if name == "help" || name == "version" {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
names = append(names, name)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Parent returns the parent context, if any
|
2015-06-19 21:06:04 +00:00
|
|
|
func (c *Context) Parent() *Context {
|
|
|
|
return c.parentContext
|
|
|
|
}
|
|
|
|
|
2016-09-17 23:54:29 +00:00
|
|
|
// value returns the value of the flag coressponding to `name`
|
|
|
|
func (c *Context) value(name string) interface{} {
|
|
|
|
return c.flagSet.Lookup(name).Value.(flag.Getter).Get()
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Args contains apps console arguments
|
2013-11-28 15:53:50 +00:00
|
|
|
type Args []string
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Args returns the command line arguments associated with the context.
|
2013-11-24 13:40:21 +00:00
|
|
|
func (c *Context) Args() Args {
|
|
|
|
args := Args(c.flagSet.Args())
|
|
|
|
return args
|
2013-07-19 01:00:52 +00:00
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// NArg returns the number of the command line arguments.
|
2016-02-21 13:57:11 +00:00
|
|
|
func (c *Context) NArg() int {
|
|
|
|
return len(c.Args())
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Get returns the nth argument, or else a blank string
|
2013-11-28 15:53:50 +00:00
|
|
|
func (a Args) Get(n int) string {
|
|
|
|
if len(a) > n {
|
|
|
|
return a[n]
|
|
|
|
}
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// First returns the first argument, or else a blank string
|
2013-11-28 15:53:50 +00:00
|
|
|
func (a Args) First() string {
|
|
|
|
return a.Get(0)
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Tail returns the rest of the arguments (not the first one)
|
2013-11-28 15:53:50 +00:00
|
|
|
// or else an empty string slice
|
|
|
|
func (a Args) Tail() []string {
|
|
|
|
if len(a) >= 2 {
|
|
|
|
return []string(a)[1:]
|
|
|
|
}
|
|
|
|
return []string{}
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Present checks if there are any arguments present
|
2013-11-28 15:53:50 +00:00
|
|
|
func (a Args) Present() bool {
|
|
|
|
return len(a) != 0
|
|
|
|
}
|
|
|
|
|
2016-05-09 14:12:59 +00:00
|
|
|
// Swap swaps arguments at the given indexes
|
2014-07-13 14:18:45 +00:00
|
|
|
func (a Args) Swap(from, to int) error {
|
|
|
|
if from >= len(a) || to >= len(a) {
|
|
|
|
return errors.New("index out of range")
|
|
|
|
}
|
|
|
|
a[from], a[to] = a[to], a[from]
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2016-04-30 15:46:47 +00:00
|
|
|
func globalContext(ctx *Context) *Context {
|
|
|
|
if ctx == nil {
|
|
|
|
return nil
|
2013-07-18 00:00:02 +00:00
|
|
|
}
|
2013-07-24 14:35:45 +00:00
|
|
|
|
2016-04-30 15:46:47 +00:00
|
|
|
for {
|
|
|
|
if ctx.parentContext == nil {
|
|
|
|
return ctx
|
2014-08-02 21:32:32 +00:00
|
|
|
}
|
2016-04-30 15:46:47 +00:00
|
|
|
ctx = ctx.parentContext
|
2014-08-02 21:32:32 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-06-29 21:20:27 +00:00
|
|
|
func lookupGlobalFlagSet(name string, ctx *Context) *flag.FlagSet {
|
|
|
|
if ctx.parentContext != nil {
|
|
|
|
ctx = ctx.parentContext
|
2013-12-03 13:42:09 +00:00
|
|
|
}
|
2015-06-29 21:20:27 +00:00
|
|
|
for ; ctx != nil; ctx = ctx.parentContext {
|
2015-05-18 15:39:48 +00:00
|
|
|
if f := ctx.flagSet.Lookup(name); f != nil {
|
|
|
|
return ctx.flagSet
|
2013-07-19 01:00:52 +00:00
|
|
|
}
|
|
|
|
}
|
2015-05-18 15:39:48 +00:00
|
|
|
return nil
|
2014-03-06 01:24:22 +00:00
|
|
|
}
|
|
|
|
|
2014-03-19 02:26:57 +00:00
|
|
|
func copyFlag(name string, ff *flag.Flag, set *flag.FlagSet) {
|
|
|
|
switch ff.Value.(type) {
|
|
|
|
case *StringSlice:
|
|
|
|
default:
|
|
|
|
set.Set(name, ff.Value.String())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-11-20 08:05:18 +00:00
|
|
|
func normalizeFlags(flags []Flag, set *flag.FlagSet) error {
|
|
|
|
visited := make(map[string]bool)
|
|
|
|
set.Visit(func(f *flag.Flag) {
|
|
|
|
visited[f.Name] = true
|
|
|
|
})
|
2013-11-18 23:35:23 +00:00
|
|
|
for _, f := range flags {
|
2015-12-15 17:29:03 +00:00
|
|
|
parts := strings.Split(f.GetName(), ",")
|
2013-11-18 23:35:23 +00:00
|
|
|
if len(parts) == 1 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
var ff *flag.Flag
|
|
|
|
for _, name := range parts {
|
2013-11-20 08:05:18 +00:00
|
|
|
name = strings.Trim(name, " ")
|
|
|
|
if visited[name] {
|
|
|
|
if ff != nil {
|
|
|
|
return errors.New("Cannot use two forms of the same flag: " + name + " " + ff.Name)
|
|
|
|
}
|
|
|
|
ff = set.Lookup(name)
|
2013-11-18 23:35:23 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if ff == nil {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
for _, name := range parts {
|
2013-11-20 08:05:18 +00:00
|
|
|
name = strings.Trim(name, " ")
|
2014-03-19 02:26:57 +00:00
|
|
|
if !visited[name] {
|
|
|
|
copyFlag(name, ff, set)
|
|
|
|
}
|
2013-11-18 23:35:23 +00:00
|
|
|
}
|
|
|
|
}
|
2013-11-20 08:05:18 +00:00
|
|
|
return nil
|
2013-11-18 23:35:23 +00:00
|
|
|
}
|
2014-12-02 17:44:55 +00:00
|
|
|
|
|
|
|
func checkRequiredFlags(flags []Flag, set *flag.FlagSet) error {
|
2015-03-02 20:06:42 +00:00
|
|
|
// If the help flag is included then none of the other flags are required.
|
|
|
|
for _, f := range flags {
|
2019-07-12 03:34:17 +00:00
|
|
|
if f.GetName() == "help" {
|
2015-03-02 20:06:42 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-12-02 17:44:55 +00:00
|
|
|
visited := make(map[string]bool)
|
|
|
|
set.Visit(func(f *flag.Flag) {
|
|
|
|
visited[f.Name] = true
|
|
|
|
})
|
|
|
|
|
|
|
|
for _, f := range flags {
|
|
|
|
if f.IsRequired() {
|
2019-07-12 03:34:17 +00:00
|
|
|
key := strings.Split(f.GetName(), ",")[0]
|
2014-12-02 17:44:55 +00:00
|
|
|
if !visited[key] {
|
2019-07-12 03:34:17 +00:00
|
|
|
return fmt.Errorf("Required flag %s not set", f.GetName())
|
2014-12-02 17:44:55 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|