| // Copyright 2015 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package obj |
| |
| import ( |
| "flag" |
| "fmt" |
| "os" |
| "strconv" |
| ) |
| |
| func Flagfn2(string, string, func(string, string)) { panic("flag") } |
| |
| func Flagcount(name, usage string, val *int) { |
| flag.Var((*count)(val), name, usage) |
| } |
| |
| func Flagint32(name, usage string, val *int32) { |
| flag.Var((*int32Value)(val), name, usage) |
| } |
| |
| func Flagint64(name, usage string, val *int64) { |
| flag.Int64Var(val, name, *val, usage) |
| } |
| |
| func Flagstr(name, usage string, val *string) { |
| flag.StringVar(val, name, *val, usage) |
| } |
| |
| func Flagfn0(name, usage string, f func()) { |
| flag.Var(fn0(f), name, usage) |
| } |
| |
| func Flagfn1(name, usage string, f func(string)) { |
| flag.Var(fn1(f), name, usage) |
| } |
| |
| func Flagprint(fd int) { |
| if fd == 1 { |
| flag.CommandLine.SetOutput(os.Stdout) |
| } |
| flag.PrintDefaults() |
| } |
| |
| func Flagparse(usage func()) { |
| flag.Usage = usage |
| flag.Parse() |
| } |
| |
| // count is a flag.Value that is like a flag.Bool and a flag.Int. |
| // If used as -name, it increments the count, but -name=x sets the count. |
| // Used for verbose flag -v. |
| type count int |
| |
| func (c *count) String() string { |
| return fmt.Sprint(int(*c)) |
| } |
| |
| func (c *count) Set(s string) error { |
| switch s { |
| case "true": |
| *c++ |
| case "false": |
| *c = 0 |
| default: |
| n, err := strconv.Atoi(s) |
| if err != nil { |
| return fmt.Errorf("invalid count %q", s) |
| } |
| *c = count(n) |
| } |
| return nil |
| } |
| |
| func (c *count) IsBoolFlag() bool { |
| return true |
| } |
| |
| type int32Value int32 |
| |
| func newIntValue(val int32, p *int32) *int32Value { |
| *p = val |
| return (*int32Value)(p) |
| } |
| |
| func (i *int32Value) Set(s string) error { |
| v, err := strconv.ParseInt(s, 0, 64) |
| *i = int32Value(v) |
| return err |
| } |
| |
| func (i *int32Value) Get() interface{} { return int32(*i) } |
| |
| func (i *int32Value) String() string { return fmt.Sprint(*i) } |
| |
| type fn0 func() |
| |
| func (f fn0) Set(s string) error { |
| f() |
| return nil |
| } |
| |
| func (f fn0) Get() interface{} { return nil } |
| |
| func (f fn0) String() string { return "" } |
| |
| func (f fn0) IsBoolFlag() bool { |
| return true |
| } |
| |
| type fn1 func(string) |
| |
| func (f fn1) Set(s string) error { |
| f(s) |
| return nil |
| } |
| |
| func (f fn1) String() string { return "" } |