123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315 |
- // mauflag - An extendable command-line argument parser for Golang
- // Copyright (C) 2016 Maunium
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU General Public License for more details.
- // You should have received a copy of the GNU General Public License
- // along with this program. If not, see <http://www.gnu.org/licenses/>.
- package mauflag
- import (
- "errors"
- "strconv"
- "strings"
- )
- // Value represents the value a flag has
- type Value interface {
- // Set or add the data in the given string to this value
- Set(string) error
- // Name should return the type name of this value or (if this value is an array,
- // a map or something of that sort) the type name of one object in this value
- Name() string
- }
- type boolValue bool
- func (val *boolValue) Name() string {
- return "boolean"
- }
- func (val *boolValue) Set(newVal string) error {
- b, err := strconv.ParseBool(newVal)
- if err == nil {
- *val = boolValue(b)
- }
- return err
- }
- type stringValue string
- func (val *stringValue) Name() string {
- return "string"
- }
- func (val *stringValue) Set(newVal string) error {
- *val = stringValue(newVal)
- return nil
- }
- type ssMapValue map[string]string
- func (val *ssMapValue) Name() string {
- return "string-string pair"
- }
- func (val *ssMapValue) Set(newVal string) error {
- index := strings.IndexRune(newVal, '=')
- if index == -1 {
- index = strings.IndexRune(newVal, ',')
- if index == -1 {
- index = strings.IndexRune(newVal, ':')
- if index == -1 {
- return errors.New("No pair separator found")
- }
- }
- }
- (*val)[newVal[:index]] = newVal[:index+1]
- return nil
- }
- type stringArrayValue []string
- func (val *stringArrayValue) Name() string {
- return "string"
- }
- func (val *stringArrayValue) Set(newVal string) error {
- *val = append(*val, newVal)
- return nil
- }
- type intArrayValue []int
- func (val *intArrayValue) Name() string {
- return "integer"
- }
- func (val *intArrayValue) Set(newVal string) error {
- i, err := strconv.Atoi(newVal)
- if err != nil {
- return err
- }
- *val = append(*val, i)
- return nil
- }
- type int64ArrayValue []int64
- func (val *int64ArrayValue) Name() string {
- return "64-bit integer"
- }
- func (val *int64ArrayValue) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 64)
- if err != nil {
- return err
- }
- *val = append(*val, i)
- return nil
- }
- type intValue int
- func (val *intValue) Name() string {
- return "integer"
- }
- func (val *intValue) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 32)
- if err == nil {
- *val = intValue(i)
- }
- return err
- }
- type uintValue uint
- func (val *uintValue) Name() string {
- return "unsigned integer"
- }
- func (val *uintValue) Set(newVal string) error {
- i, err := strconv.ParseUint(newVal, 10, 32)
- if err == nil {
- *val = uintValue(i)
- }
- return err
- }
- type int8Value int8
- func (val *int8Value) Name() string {
- return "8-bit integer"
- }
- func (val *int8Value) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 8)
- if err == nil {
- *val = int8Value(i)
- }
- return err
- }
- type uint8Value uint8
- func (val *uint8Value) Name() string {
- return "unsigned 8-bit integer"
- }
- func (val *uint8Value) Set(newVal string) error {
- i, err := strconv.ParseUint(newVal, 10, 8)
- if err == nil {
- *val = uint8Value(i)
- }
- return err
- }
- type byteValue byte
- func (val *byteValue) Name() string {
- return "rune (unsigned 8-bit integer)"
- }
- func (val *byteValue) Set(newVal string) error {
- i, err := strconv.ParseUint(newVal, 10, 8)
- if err == nil {
- *val = byteValue(i)
- }
- return err
- }
- type int16Value int16
- func (val *int16Value) Name() string {
- return "16-bit integer"
- }
- func (val *int16Value) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 16)
- if err == nil {
- *val = int16Value(i)
- }
- return err
- }
- type uint16Value uint16
- func (val *uint16Value) Name() string {
- return "unsigned 16-bit integer"
- }
- func (val *uint16Value) Set(newVal string) error {
- i, err := strconv.ParseUint(newVal, 10, 16)
- if err == nil {
- *val = uint16Value(i)
- }
- return err
- }
- type int32Value int32
- func (val *int32Value) Name() string {
- return "32-bit integer"
- }
- func (val *int32Value) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 32)
- if err == nil {
- *val = int32Value(i)
- }
- return err
- }
- type runeValue rune
- func (val *runeValue) Name() string {
- return "rune (signed 32-bit integer)"
- }
- func (val *runeValue) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 32)
- if err == nil {
- *val = runeValue(i)
- }
- return err
- }
- type uint32Value uint32
- func (val *uint32Value) Name() string {
- return "unsigned 32-bit integer"
- }
- func (val *uint32Value) Set(newVal string) error {
- i, err := strconv.ParseUint(newVal, 10, 32)
- if err == nil {
- *val = uint32Value(i)
- }
- return err
- }
- type int64Value int64
- func (val *int64Value) Name() string {
- return "64-bit integer"
- }
- func (val *int64Value) Set(newVal string) error {
- i, err := strconv.ParseInt(newVal, 10, 64)
- if err == nil {
- *val = int64Value(i)
- }
- return err
- }
- type uint64Value uint64
- func (val *uint64Value) Name() string {
- return "unsigned 64-bit integer"
- }
- func (val *uint64Value) Set(newVal string) error {
- i, err := strconv.ParseUint(newVal, 10, 64)
- if err == nil {
- *val = uint64Value(i)
- }
- return err
- }
- type float32Value float32
- func (val *float32Value) Name() string {
- return "32-bit float"
- }
- func (val *float32Value) Set(newVal string) error {
- i, err := strconv.ParseFloat(newVal, 32)
- if err == nil {
- *val = float32Value(i)
- }
- return err
- }
- type float64Value float64
- func (val *float64Value) Name() string {
- return "64-bit float"
- }
- func (val *float64Value) Set(newVal string) error {
- i, err := strconv.ParseFloat(newVal, 64)
- if err == nil {
- *val = float64Value(i)
- }
- return err
- }
|