map_input_source.go
No OneTemporary
Actions

Size

6 KB

Subscribers

None

map_input_source.go
View Options

	package altsrc

	import (
	"fmt"
	"reflect"
	"strings"
	"time"

	"gopkg.in/urfave/cli.v1"
	)

	// MapInputSource implements InputSourceContext to return
	// data from the map that is loaded.
	type MapInputSource struct {
	valueMap map[interface{}]interface{}
	}

	// nestedVal checks if the name has '.' delimiters.
	// If so, it tries to traverse the tree by the '.' delimited sections to find
	// a nested value for the key.
	func nestedVal(name string, tree map[interface{}]interface{}) (interface{}, bool) {
	if sections := strings.Split(name, "."); len(sections) > 1 {
	node := tree
	for _, section := range sections[:len(sections)-1] {
	if child, ok := node[section]; !ok {
	return nil, false
	} else {
	if ctype, ok := child.(map[interface{}]interface{}); !ok {
	return nil, false
	} else {
	node = ctype
	}
	}
	}
	if val, ok := node[sections[len(sections)-1]]; ok {
	return val, true
	}
	}
	return nil, false
	}

	// Int returns an int from the map if it exists otherwise returns 0
	func (fsm *MapInputSource) Int(name string) (int, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(int)
	if !isType {
	return 0, incorrectTypeForFlagError(name, "int", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(int)
	if !isType {
	return 0, incorrectTypeForFlagError(name, "int", nestedGenericValue)
	}
	return otherValue, nil
	}

	return 0, nil
	}

	// Duration returns a duration from the map if it exists otherwise returns 0
	func (fsm *MapInputSource) Duration(name string) (time.Duration, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(time.Duration)
	if !isType {
	return 0, incorrectTypeForFlagError(name, "duration", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(time.Duration)
	if !isType {
	return 0, incorrectTypeForFlagError(name, "duration", nestedGenericValue)
	}
	return otherValue, nil
	}

	return 0, nil
	}

	// Float64 returns an float64 from the map if it exists otherwise returns 0
	func (fsm *MapInputSource) Float64(name string) (float64, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(float64)
	if !isType {
	return 0, incorrectTypeForFlagError(name, "float64", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(float64)
	if !isType {
	return 0, incorrectTypeForFlagError(name, "float64", nestedGenericValue)
	}
	return otherValue, nil
	}

	return 0, nil
	}

	// String returns a string from the map if it exists otherwise returns an empty string
	func (fsm *MapInputSource) String(name string) (string, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(string)
	if !isType {
	return "", incorrectTypeForFlagError(name, "string", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(string)
	if !isType {
	return "", incorrectTypeForFlagError(name, "string", nestedGenericValue)
	}
	return otherValue, nil
	}

	return "", nil
	}

	// StringSlice returns an []string from the map if it exists otherwise returns nil
	func (fsm *MapInputSource) StringSlice(name string) ([]string, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.([]string)
	if !isType {
	return nil, incorrectTypeForFlagError(name, "[]string", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.([]string)
	if !isType {
	return nil, incorrectTypeForFlagError(name, "[]string", nestedGenericValue)
	}
	return otherValue, nil
	}

	return nil, nil
	}

	// IntSlice returns an []int from the map if it exists otherwise returns nil
	func (fsm *MapInputSource) IntSlice(name string) ([]int, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.([]int)
	if !isType {
	return nil, incorrectTypeForFlagError(name, "[]int", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.([]int)
	if !isType {
	return nil, incorrectTypeForFlagError(name, "[]int", nestedGenericValue)
	}
	return otherValue, nil
	}

	return nil, nil
	}

	// Generic returns an cli.Generic from the map if it exists otherwise returns nil
	func (fsm *MapInputSource) Generic(name string) (cli.Generic, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(cli.Generic)
	if !isType {
	return nil, incorrectTypeForFlagError(name, "cli.Generic", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(cli.Generic)
	if !isType {
	return nil, incorrectTypeForFlagError(name, "cli.Generic", nestedGenericValue)
	}
	return otherValue, nil
	}

	return nil, nil
	}

	// Bool returns an bool from the map otherwise returns false
	func (fsm *MapInputSource) Bool(name string) (bool, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(bool)
	if !isType {
	return false, incorrectTypeForFlagError(name, "bool", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(bool)
	if !isType {
	return false, incorrectTypeForFlagError(name, "bool", nestedGenericValue)
	}
	return otherValue, nil
	}

	return false, nil
	}

	// BoolT returns an bool from the map otherwise returns true
	func (fsm *MapInputSource) BoolT(name string) (bool, error) {
	otherGenericValue, exists := fsm.valueMap[name]
	if exists {
	otherValue, isType := otherGenericValue.(bool)
	if !isType {
	return true, incorrectTypeForFlagError(name, "bool", otherGenericValue)
	}
	return otherValue, nil
	}
	nestedGenericValue, exists := nestedVal(name, fsm.valueMap)
	if exists {
	otherValue, isType := nestedGenericValue.(bool)
	if !isType {
	return true, incorrectTypeForFlagError(name, "bool", nestedGenericValue)
	}
	return otherValue, nil
	}

	return true, nil
	}

	func incorrectTypeForFlagError(name, expectedTypeName string, value interface{}) error {
	valueType := reflect.TypeOf(value)
	valueTypeName := ""
	if valueType != nil {
	valueTypeName = valueType.Name()
	}

	return fmt.Errorf("Mismatched type for flag '%s'. Expected '%s' but actual is '%s'", name, expectedTypeName, valueTypeName)
	}

File Metadata

Mime Type: text/plain
Expires: Sat, Jan 24, 8:48 PM (1 d, 14 h)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3604197

map_input_source.goNo OneTemporaryActions

map_input_source.goView Options

File Metadata

Event Timeline

map_input_source.go
No OneTemporary
Actions

map_input_source.go
View Options