Compare commits
5 Commits
Author | SHA1 | Date |
---|---|---|
|
8235b935a3 | |
|
b72816f193 | |
|
8ecad2c6af | |
|
8b32c8d316 | |
|
6cc03a124c |
|
@ -0,0 +1,19 @@
|
||||||
|
all: goimports vet
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f go.*
|
||||||
|
|
||||||
|
redomod:
|
||||||
|
rm -f go.*
|
||||||
|
GO111MODULE= go mod init
|
||||||
|
GO111MODULE= go mod tidy
|
||||||
|
|
||||||
|
test:
|
||||||
|
go test
|
||||||
|
|
||||||
|
vet:
|
||||||
|
@GO111MODULE=off go vet
|
||||||
|
@echo this go binary package builds okay
|
||||||
|
|
||||||
|
goimports:
|
||||||
|
goimports -w *.go
|
2
go.mod
2
go.mod
|
@ -1,4 +1,4 @@
|
||||||
module github.com/alexflint/go-arg
|
module go.wit.com/dev/alexflint/arg
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/alexflint/go-scalar v1.2.0
|
github.com/alexflint/go-scalar v1.2.0
|
||||||
|
|
35
parse.go
35
parse.go
|
@ -80,9 +80,26 @@ var ErrVersion = errors.New("version requested by user")
|
||||||
var mustParseExit = os.Exit
|
var mustParseExit = os.Exit
|
||||||
var mustParseOut io.Writer = os.Stdout
|
var mustParseOut io.Writer = os.Stdout
|
||||||
|
|
||||||
|
/*
|
||||||
|
This allows you to have common arg values defined in a GO package
|
||||||
|
|
||||||
|
package 'foo'
|
||||||
|
function init() {
|
||||||
|
args.Register(&argsFoo)
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
// This stores the args sent from the GO packages
|
||||||
|
var register []interface{}
|
||||||
|
|
||||||
|
func Register(dest ...interface{}) {
|
||||||
|
register = append(register, dest...)
|
||||||
|
}
|
||||||
|
|
||||||
// MustParse processes command line arguments and exits upon failure
|
// MustParse processes command line arguments and exits upon failure
|
||||||
func MustParse(dest ...interface{}) *Parser {
|
func MustParse(dest ...interface{}) *Parser {
|
||||||
return mustParse(Config{Exit: mustParseExit, Out: mustParseOut}, dest...)
|
register = append(register, dest...)
|
||||||
|
return mustParse(Config{Exit: mustParseExit, Out: mustParseOut}, register...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// mustParse is a helper that facilitates testing
|
// mustParse is a helper that facilitates testing
|
||||||
|
@ -107,8 +124,24 @@ func Parse(dest ...interface{}) error {
|
||||||
return p.Parse(flags())
|
return p.Parse(flags())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// pass in a "pretend" os.Args. Used for bash autocomplete
|
||||||
|
func ParseFlags(flags []string, dest ...interface{}) (*Parser, error) {
|
||||||
|
p, err := NewParser(Config{}, dest...)
|
||||||
|
if err != nil {
|
||||||
|
return p, err
|
||||||
|
}
|
||||||
|
overrideFlags = append(overrideFlags, flags...)
|
||||||
|
err = p.Parse(flags)
|
||||||
|
return p, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var overrideFlags []string
|
||||||
|
|
||||||
// flags gets all command line arguments other than the first (program name)
|
// flags gets all command line arguments other than the first (program name)
|
||||||
func flags() []string {
|
func flags() []string {
|
||||||
|
if len(overrideFlags) > 0 {
|
||||||
|
return overrideFlags
|
||||||
|
}
|
||||||
if len(os.Args) == 0 { // os.Args could be empty
|
if len(os.Args) == 0 { // os.Args could be empty
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue