getopt: init module

getopt/int.go

@@ -0,0 +1,61 @@
+package getopt
+
+import (
+	"fmt"
+)
+
+type intVal int
+
+func (i *intVal) String() string {
+	return fmt.Sprint(*i)
+}
+
+func (i *intVal) Set(val string) error {
+	_, err := fmt.Sscanf(val, "%d", i)
+	return err
+}
+
+// IntVar defines a int flag with specified name, default value,
+// and usage string. The argument p points to a int variable in
+// which to store the value of the flag.
+func (set *FlagSet) IntVar(p *int, name string, value int, usage string) {
+	*p = value
+	set.Var((*intVal)(p), name, usage)
+}
+
+// Int defines an int flag with specified name, default value, and usage
+// string. The return value is the address of an int variable that
+// stores the value of the flag.
+func (set *FlagSet) Int(name string, value int, usage string) *int {
+	p := new(int)
+	set.IntVar(p, name, value, usage)
+	return p
+}
+
+type int64Val int64
+
+func (i *int64Val) String() string {
+	return fmt.Sprint(int64(*i))
+}
+
+func (i *int64Val) Set(val string) error {
+	_, err := fmt.Sscanf(val, "%d", i)
+	return err
+}
+
+// Int64Var defines an int64 flag with specified name, default value,
+// and usage string. The argument p points to an int64 variable in which
+// to store the value of the flag.
+func (set *FlagSet) Int64Var(p *int64, name string, value int64, usage string) {
+	*p = value
+	set.Var((*int64Val)(p), name, usage)
+}
+
+// Int64 defines an int64 flag with specified name, default value, and
+// usage string. The return value is the address of an int64 variable
+// that stores the value of the flag.
+func (set *FlagSet) Int64(name string, value int64, usage string) *int64 {
+	var i int64
+	set.Int64Var(&i, name, value, usage)
+	return &i
+}