summaryrefslogtreecommitdiff
path: root/vendor/github.com/pborman/getopt/counter.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/pborman/getopt/counter.go')
-rw-r--r--vendor/github.com/pborman/getopt/counter.go81
1 files changed, 81 insertions, 0 deletions
diff --git a/vendor/github.com/pborman/getopt/counter.go b/vendor/github.com/pborman/getopt/counter.go
new file mode 100644
index 0000000..8b11ec1
--- /dev/null
+++ b/vendor/github.com/pborman/getopt/counter.go
@@ -0,0 +1,81 @@
+// Copyright 2013 Google Inc. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package getopt
+
+import (
+ "fmt"
+ "strconv"
+)
+
+type counterValue int
+
+func (b *counterValue) Set(value string, opt Option) error {
+ if value == "" {
+ *b++
+ } else {
+ v, err := strconv.ParseInt(value, 0, strconv.IntSize)
+ if err != nil {
+ if e, ok := err.(*strconv.NumError); ok {
+ switch e.Err {
+ case strconv.ErrRange:
+ err = fmt.Errorf("value out of range: %s", value)
+ case strconv.ErrSyntax:
+ err = fmt.Errorf("not a valid number: %s", value)
+ }
+ }
+ return err
+ }
+ *b = counterValue(v)
+ }
+ return nil
+}
+
+func (b *counterValue) String() string {
+ return strconv.Itoa(int(*b))
+}
+
+// Counter creates a counting flag stored as an int. Each time the option
+// is seen while parsing the value is incremented. The value of the counter
+// may be explicitly set by using the long form:
+//
+// --counter=5
+// --c=5
+//
+// Further instances of the option will increment from the set value.
+func Counter(name rune, helpvalue ...string) *int {
+ return CommandLine.Counter(name, helpvalue...)
+}
+
+func (s *Set) Counter(name rune, helpvalue ...string) *int {
+ var p int
+ s.CounterVarLong(&p, "", name, helpvalue...)
+ return &p
+}
+
+func CounterLong(name string, short rune, helpvalue ...string) *int {
+ return CommandLine.CounterLong(name, short, helpvalue...)
+}
+
+func (s *Set) CounterLong(name string, short rune, helpvalue ...string) *int {
+ var p int
+ s.CounterVarLong(&p, name, short, helpvalue...)
+ return &p
+}
+
+func CounterVar(p *int, name rune, helpvalue ...string) Option {
+ return CommandLine.CounterVar(p, name, helpvalue...)
+}
+
+func (s *Set) CounterVar(p *int, name rune, helpvalue ...string) Option {
+ return s.CounterVarLong(p, "", name, helpvalue...)
+}
+
+func CounterVarLong(p *int, name string, short rune, helpvalue ...string) Option {
+ return CommandLine.CounterVarLong(p, name, short, helpvalue...)
+}
+
+func (s *Set) CounterVarLong(p *int, name string, short rune, helpvalue ...string) Option {
+ return s.VarLong((*counterValue)(p), name, short, helpvalue...).SetFlag()
+}