package commands import ( "fmt" "io" "reflect" "strconv" ) type optMap map[string]interface{} // Request represents a call to a command from a consumer type Request interface { Path() []string Option(name string) (interface{}, bool) SetOption(name string, val interface{}) Arguments() []string Stream() io.Reader SetStream(io.Reader) ConvertOptions(options map[string]Option) error } type request struct { path []string options optMap arguments []string in io.Reader } // Path returns the command path of this request func (r *request) Path() []string { return r.path } // Option returns the value of the option for given name. func (r *request) Option(name string) (interface{}, bool) { val, err := r.options[name] return val, err } // SetOption sets the value of the option for given name. func (r *request) SetOption(name string, val interface{}) { r.options[name] = val } // Arguments returns the arguments slice func (r *request) Arguments() []string { return r.arguments } // Stream returns the input stream Reader func (r *request) Stream() io.Reader { return r.in } // SetStream sets the value of the input stream Reader func (r *request) SetStream(in io.Reader) { r.in = in } type converter func(string) (interface{}, error) var converters = map[reflect.Kind]converter{ Bool: func(v string) (interface{}, error) { if v == "" { return true, nil } return strconv.ParseBool(v) }, Int: func(v string) (interface{}, error) { return strconv.ParseInt(v, 0, 32) }, Uint: func(v string) (interface{}, error) { return strconv.ParseInt(v, 0, 32) }, Float: func(v string) (interface{}, error) { return strconv.ParseFloat(v, 64) }, } func (r *request) ConvertOptions(options map[string]Option) error { converted := make(map[string]interface{}) for k, v := range r.options { opt, ok := options[k] if !ok { return fmt.Errorf("Unrecognized option: '%s'", k) } kind := reflect.TypeOf(v).Kind() var value interface{} if kind != opt.Type { if kind == String { convert := converters[opt.Type] val, err := convert(v.(string)) if err != nil { return fmt.Errorf("Could not convert string value '%s' to type '%s'", v, opt.Type.String()) } value = val } else { return fmt.Errorf("Option '%s' should be type '%s', but got type '%s'", k, opt.Type.String(), kind.String()) } } else { value = v } for _, name := range opt.Names { if _, ok := r.options[name]; name != k && ok { return fmt.Errorf("Duplicate command options were provided ('%s' and '%s')", k, name) } converted[name] = value } } r.options = converted return nil } // NewEmptyRequest initializes an empty request func NewEmptyRequest() Request { return NewRequest(nil, nil, nil, nil) } // NewRequest returns a request initialized with given arguments func NewRequest(path []string, opts optMap, args []string, in io.Reader) Request { if path == nil { path = make([]string, 0) } if opts == nil { opts = make(map[string]interface{}) } if args == nil { args = make([]string, 0) } return &request{path, opts, args, in} }