61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package flag
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/containous/traefik/v2/pkg/config/parser"
|
|
)
|
|
|
|
func getFlagTypes(element interface{}) map[string]reflect.Kind {
|
|
ref := map[string]reflect.Kind{}
|
|
|
|
if element == nil {
|
|
return ref
|
|
}
|
|
|
|
tp := reflect.TypeOf(element).Elem()
|
|
|
|
addFlagType(ref, "", tp)
|
|
|
|
return ref
|
|
}
|
|
|
|
func addFlagType(ref map[string]reflect.Kind, name string, typ reflect.Type) {
|
|
switch typ.Kind() {
|
|
case reflect.Bool, reflect.Slice:
|
|
ref[name] = typ.Kind()
|
|
|
|
case reflect.Map:
|
|
addFlagType(ref, getName(name, parser.MapNamePlaceholder), typ.Elem())
|
|
|
|
case reflect.Ptr:
|
|
if typ.Elem().Kind() == reflect.Struct {
|
|
ref[name] = typ.Kind()
|
|
}
|
|
addFlagType(ref, name, typ.Elem())
|
|
|
|
case reflect.Struct:
|
|
for j := 0; j < typ.NumField(); j++ {
|
|
subField := typ.Field(j)
|
|
|
|
if !parser.IsExported(subField) {
|
|
continue
|
|
}
|
|
|
|
if subField.Anonymous {
|
|
addFlagType(ref, getName(name), subField.Type)
|
|
} else {
|
|
addFlagType(ref, getName(name, subField.Name), subField.Type)
|
|
}
|
|
}
|
|
|
|
default:
|
|
// noop
|
|
}
|
|
}
|
|
|
|
func getName(names ...string) string {
|
|
return strings.TrimPrefix(strings.ToLower(strings.Join(names, ".")), ".")
|
|
}
|