cast/caste.go

644 lines
14 KiB
Go
Raw Normal View History

2014-04-05 13:21:52 +08:00
// Copyright © 2014 Steve Francia <spf@spf13.com>.
//
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.
package cast
import (
"fmt"
2014-12-09 21:17:55 +08:00
"html/template"
"reflect"
2014-04-05 13:21:52 +08:00
"strconv"
2015-02-19 10:44:31 +08:00
"strings"
2014-04-05 13:21:52 +08:00
"time"
)
2017-03-03 23:41:53 +08:00
// ToTimeE casts an interface to a time.Time type.
func ToTimeE(i interface{}) (tim time.Time, err error) {
i = indirect(i)
2016-12-25 01:25:03 +08:00
switch v := i.(type) {
2014-04-05 13:21:52 +08:00
case time.Time:
2016-12-25 01:25:03 +08:00
return v, nil
2014-04-05 13:21:52 +08:00
case string:
2016-12-25 01:25:03 +08:00
d, e := StringToDate(v)
2014-04-05 13:21:52 +08:00
if e == nil {
return d, nil
2014-04-05 13:21:52 +08:00
}
return time.Time{}, fmt.Errorf("Could not parse Date/Time format: %v\n", e)
2016-12-25 01:25:03 +08:00
case int:
return time.Unix(int64(v), 0), nil
case int32:
return time.Unix(int64(v), 0), nil
case int64:
return time.Unix(v, 0), nil
2014-04-05 13:21:52 +08:00
default:
return time.Time{}, fmt.Errorf("Unable to Cast %#v to Time\n", i)
2014-04-05 13:21:52 +08:00
}
}
2017-03-03 23:41:53 +08:00
// ToDurationE casts an interface to a time.Duration type.
2015-02-19 10:56:24 +08:00
func ToDurationE(i interface{}) (d time.Duration, err error) {
i = indirect(i)
switch s := i.(type) {
case time.Duration:
return s, nil
case int64, int32, int16, int8, int:
d = time.Duration(ToInt64(s))
return
case float32, float64:
d = time.Duration(ToFloat64(s))
return
2015-02-19 10:56:24 +08:00
case string:
if strings.ContainsAny(s, "nsuµmh") {
d, err = time.ParseDuration(s)
} else {
d, err = time.ParseDuration(s + "ns")
}
2015-02-19 10:56:24 +08:00
return
default:
err = fmt.Errorf("Unable to Cast %#v to Duration\n", i)
return
}
}
2017-03-03 23:41:53 +08:00
// ToBoolE casts an interface to a bool type.
func ToBoolE(i interface{}) (bool, error) {
i = indirect(i)
2014-04-05 13:21:52 +08:00
switch b := i.(type) {
case bool:
return b, nil
2014-04-05 13:21:52 +08:00
case nil:
return false, nil
2014-04-05 13:21:52 +08:00
case int:
if i.(int) != 0 {
return true, nil
2014-04-05 13:21:52 +08:00
}
return false, nil
case string:
return strconv.ParseBool(i.(string))
2014-04-05 13:21:52 +08:00
default:
return false, fmt.Errorf("Unable to Cast %#v to bool", i)
2014-04-05 13:21:52 +08:00
}
}
2017-03-03 23:41:53 +08:00
// ToFloat64E casts an interface to a float64 type.
func ToFloat64E(i interface{}) (float64, error) {
i = indirect(i)
2014-04-05 13:21:52 +08:00
switch s := i.(type) {
case float64:
return s, nil
2014-04-05 13:21:52 +08:00
case float32:
return float64(s), nil
2014-04-07 23:43:25 +08:00
case int64:
return float64(s), nil
2014-04-07 23:43:25 +08:00
case int32:
return float64(s), nil
2014-04-07 23:43:25 +08:00
case int16:
return float64(s), nil
2014-04-07 23:43:25 +08:00
case int8:
return float64(s), nil
2014-04-07 23:43:25 +08:00
case int:
return float64(s), nil
2014-04-05 13:21:52 +08:00
case string:
v, err := strconv.ParseFloat(s, 64)
if err == nil {
return float64(v), nil
2014-04-05 13:21:52 +08:00
}
return 0.0, fmt.Errorf("Unable to Cast %#v to float", i)
2014-04-05 13:21:52 +08:00
default:
return 0.0, fmt.Errorf("Unable to Cast %#v to float", i)
2014-04-05 13:21:52 +08:00
}
}
2017-03-03 23:41:53 +08:00
// ToInt64E casts an interface to an int64 type.
2016-01-29 02:41:34 +08:00
func ToInt64E(i interface{}) (int64, error) {
i = indirect(i)
switch s := i.(type) {
case int64:
return s, nil
case int:
return int64(s), nil
case int32:
return int64(s), nil
case int16:
return int64(s), nil
case int8:
return int64(s), nil
case string:
v, err := strconv.ParseInt(s, 0, 0)
if err == nil {
return v, nil
}
return 0, fmt.Errorf("Unable to Cast %#v to int64", i)
case float64:
return int64(s), nil
case bool:
if bool(s) {
return int64(1), nil
}
return int64(0), nil
case nil:
return int64(0), nil
default:
return int64(0), fmt.Errorf("Unable to Cast %#v to int64", i)
}
}
2017-03-03 23:41:53 +08:00
// ToInt32E casts an interface to an int32 type.
2017-02-10 15:00:46 +08:00
func ToInt32E(i interface{}) (int32, error) {
i = indirect(i)
switch s := i.(type) {
case int64:
return int32(s), nil
case int:
return int32(s), nil
case int32:
return s, nil
case int16:
return int32(s), nil
case int8:
return int32(s), nil
case string:
v, err := strconv.ParseInt(s, 0, 0)
if err == nil {
return int32(v), nil
}
return 0, fmt.Errorf("Unable to Cast %#v to int32", i)
case float64:
return int32(s), nil
case bool:
if bool(s) {
return int32(1), nil
}
return int32(0), nil
case nil:
return int32(0), nil
default:
return int32(0), fmt.Errorf("Unable to Cast %#v to int32", i)
}
}
2017-03-03 23:41:53 +08:00
// ToInt16E casts an interface to an int16 type.
2017-02-10 15:00:46 +08:00
func ToInt16E(i interface{}) (int16, error) {
i = indirect(i)
switch s := i.(type) {
case int64:
return int16(s), nil
case int:
return int16(s), nil
case int32:
return int16(s), nil
case int16:
return s, nil
case int8:
return int16(s), nil
case string:
v, err := strconv.ParseInt(s, 0, 0)
if err == nil {
return int16(v), nil
}
return 0, fmt.Errorf("Unable to Cast %#v to int16", i)
case float64:
return int16(s), nil
case bool:
if bool(s) {
return int16(1), nil
}
return int16(0), nil
case nil:
return int16(0), nil
default:
return int16(0), fmt.Errorf("Unable to Cast %#v to int16", i)
}
}
2017-03-03 23:41:53 +08:00
// ToInt8E casts an interface to an int8 type.
2017-02-10 15:00:46 +08:00
func ToInt8E(i interface{}) (int8, error) {
i = indirect(i)
switch s := i.(type) {
case int64:
return int8(s), nil
case int:
return int8(s), nil
case int32:
return int8(s), nil
case int16:
return int8(s), nil
case int8:
return s, nil
case string:
v, err := strconv.ParseInt(s, 0, 0)
if err == nil {
return int8(v), nil
}
return 0, fmt.Errorf("Unable to Cast %#v to int8", i)
case float64:
return int8(s), nil
case bool:
if bool(s) {
return int8(1), nil
}
return int8(0), nil
case nil:
return int8(0), nil
default:
return int8(0), fmt.Errorf("Unable to Cast %#v to int8", i)
}
}
2017-03-03 23:41:53 +08:00
// ToIntE casts an interface to an int type.
func ToIntE(i interface{}) (int, error) {
i = indirect(i)
2014-04-05 13:21:52 +08:00
switch s := i.(type) {
case int:
return s, nil
2014-04-05 13:21:52 +08:00
case int64:
return int(s), nil
2014-04-05 13:21:52 +08:00
case int32:
return int(s), nil
2014-04-05 13:21:52 +08:00
case int16:
return int(s), nil
2014-04-05 13:21:52 +08:00
case int8:
return int(s), nil
2014-04-05 13:21:52 +08:00
case string:
v, err := strconv.ParseInt(s, 0, 0)
if err == nil {
return int(v), nil
2014-04-05 13:21:52 +08:00
}
return 0, fmt.Errorf("Unable to Cast %#v to int", i)
case float64:
return int(s), nil
2014-04-05 13:21:52 +08:00
case bool:
if bool(s) {
return 1, nil
2014-04-05 13:21:52 +08:00
}
return 0, nil
case nil:
return 0, nil
2014-04-05 13:21:52 +08:00
default:
return 0, fmt.Errorf("Unable to Cast %#v to int", i)
2014-04-05 13:21:52 +08:00
}
}
// From html/template/content.go
// Copyright 2011 The Go Authors. All rights reserved.
// indirect returns the value, after dereferencing as many times
// as necessary to reach the base type (or nil).
func indirect(a interface{}) interface{} {
if a == nil {
return nil
}
if t := reflect.TypeOf(a); t.Kind() != reflect.Ptr {
// Avoid creating a reflect.Value if it's not a pointer.
return a
}
v := reflect.ValueOf(a)
for v.Kind() == reflect.Ptr && !v.IsNil() {
v = v.Elem()
}
return v.Interface()
}
// From html/template/content.go
// Copyright 2011 The Go Authors. All rights reserved.
// indirectToStringerOrError returns the value, after dereferencing as many times
// as necessary to reach the base type (or nil) or an implementation of fmt.Stringer
// or error,
func indirectToStringerOrError(a interface{}) interface{} {
if a == nil {
return nil
}
var errorType = reflect.TypeOf((*error)(nil)).Elem()
var fmtStringerType = reflect.TypeOf((*fmt.Stringer)(nil)).Elem()
v := reflect.ValueOf(a)
for !v.Type().Implements(fmtStringerType) && !v.Type().Implements(errorType) && v.Kind() == reflect.Ptr && !v.IsNil() {
v = v.Elem()
}
return v.Interface()
}
2017-03-03 23:41:53 +08:00
// ToStringE casts an interface to a string type.
func ToStringE(i interface{}) (string, error) {
i = indirectToStringerOrError(i)
2014-04-05 13:21:52 +08:00
switch s := i.(type) {
case string:
return s, nil
2015-11-26 14:42:19 +08:00
case bool:
return strconv.FormatBool(s), nil
2014-04-05 13:21:52 +08:00
case float64:
2017-03-04 00:27:06 +08:00
return strconv.FormatFloat(s, 'f', -1, 64), nil
case int64:
2017-03-04 00:27:06 +08:00
return strconv.FormatInt(s, 10), nil
2014-04-05 13:21:52 +08:00
case int:
2017-03-04 00:27:06 +08:00
return strconv.Itoa(s), nil
case []byte:
return string(s), nil
2014-12-09 21:17:55 +08:00
case template.HTML:
return string(s), nil
2015-06-08 00:35:28 +08:00
case template.URL:
return string(s), nil
case template.JS:
return string(s), nil
case template.CSS:
return string(s), nil
case template.HTMLAttr:
return string(s), nil
2014-04-05 13:21:52 +08:00
case nil:
return "", nil
case fmt.Stringer:
return s.String(), nil
case error:
return s.Error(), nil
2014-04-05 13:21:52 +08:00
default:
return "", fmt.Errorf("Unable to Cast %#v to string", i)
2014-04-05 13:21:52 +08:00
}
}
2017-03-03 23:41:53 +08:00
// ToStringMapStringE casts an interface to a map[string]string type.
func ToStringMapStringE(i interface{}) (map[string]string, error) {
var m = map[string]string{}
switch v := i.(type) {
case map[string]string:
return v, nil
case map[string]interface{}:
for k, val := range v {
m[ToString(k)] = ToString(val)
}
return m, nil
case map[interface{}]string:
for k, val := range v {
m[ToString(k)] = ToString(val)
}
return m, nil
case map[interface{}]interface{}:
for k, val := range v {
m[ToString(k)] = ToString(val)
}
return m, nil
default:
return m, fmt.Errorf("Unable to Cast %#v to map[string]string", i)
}
}
2017-03-03 23:41:53 +08:00
// ToStringMapStringSliceE casts an interface to a map[string][]string type.
2015-07-31 04:13:54 +08:00
func ToStringMapStringSliceE(i interface{}) (map[string][]string, error) {
var m = map[string][]string{}
switch v := i.(type) {
case map[string][]string:
return v, nil
case map[string][]interface{}:
2015-07-31 04:13:54 +08:00
for k, val := range v {
m[ToString(k)] = ToStringSlice(val)
}
return m, nil
case map[string]string:
for k, val := range v {
m[ToString(k)] = []string{val}
}
case map[string]interface{}:
for k, val := range v {
switch vt := val.(type) {
case []interface{}:
m[ToString(k)] = ToStringSlice(vt)
case []string:
m[ToString(k)] = vt
default:
m[ToString(k)] = []string{ToString(val)}
}
}
return m, nil
case map[interface{}][]string:
2015-07-31 04:13:54 +08:00
for k, val := range v {
m[ToString(k)] = ToStringSlice(val)
}
return m, nil
case map[interface{}]string:
2015-07-31 04:13:54 +08:00
for k, val := range v {
m[ToString(k)] = ToStringSlice(val)
}
return m, nil
case map[interface{}][]interface{}:
2015-07-31 04:13:54 +08:00
for k, val := range v {
m[ToString(k)] = ToStringSlice(val)
2015-07-31 04:13:54 +08:00
}
return m, nil
case map[interface{}]interface{}:
2015-07-31 04:13:54 +08:00
for k, val := range v {
key, err := ToStringE(k)
if err != nil {
return m, fmt.Errorf("Unable to Cast %#v to map[string][]string", i)
}
value, err := ToStringSliceE(val)
if err != nil {
return m, fmt.Errorf("Unable to Cast %#v to map[string][]string", i)
}
m[key] = value
2015-07-31 04:13:54 +08:00
}
default:
return m, fmt.Errorf("Unable to Cast %#v to map[string][]string", i)
2015-07-31 04:13:54 +08:00
}
return m, nil
2015-07-31 04:13:54 +08:00
}
2017-03-03 23:41:53 +08:00
// ToStringMapBoolE casts an interface to a map[string]bool type.
func ToStringMapBoolE(i interface{}) (map[string]bool, error) {
var m = map[string]bool{}
switch v := i.(type) {
case map[interface{}]interface{}:
for k, val := range v {
m[ToString(k)] = ToBool(val)
}
return m, nil
case map[string]interface{}:
for k, val := range v {
m[ToString(k)] = ToBool(val)
}
return m, nil
case map[string]bool:
return v, nil
default:
return m, fmt.Errorf("Unable to Cast %#v to map[string]bool", i)
}
}
2017-03-03 23:41:53 +08:00
// ToStringMapE casts an interface to a map[string]interface{} type.
func ToStringMapE(i interface{}) (map[string]interface{}, error) {
var m = map[string]interface{}{}
switch v := i.(type) {
case map[interface{}]interface{}:
for k, val := range v {
m[ToString(k)] = val
}
return m, nil
case map[string]interface{}:
return v, nil
default:
return m, fmt.Errorf("Unable to Cast %#v to map[string]interface{}", i)
}
}
2017-03-03 23:41:53 +08:00
// ToSliceE casts an interface to a []interface{} type.
2014-04-27 06:56:25 +08:00
func ToSliceE(i interface{}) ([]interface{}, error) {
var s []interface{}
switch v := i.(type) {
case []interface{}:
2017-03-04 00:27:06 +08:00
return append(s, v...), nil
2014-04-27 06:56:25 +08:00
case []map[string]interface{}:
for _, u := range v {
s = append(s, u)
}
return s, nil
default:
return s, fmt.Errorf("Unable to Cast %#v of type %v to []interface{}", i, reflect.TypeOf(i))
}
}
2017-03-03 23:41:53 +08:00
// ToBoolSliceE casts an interface to a []bool type.
2016-09-20 04:26:41 +08:00
func ToBoolSliceE(i interface{}) ([]bool, error) {
if i == nil {
return []bool{}, fmt.Errorf("Unable to Cast %#v to []bool", i)
}
switch v := i.(type) {
case []bool:
return v, nil
}
kind := reflect.TypeOf(i).Kind()
switch kind {
case reflect.Slice, reflect.Array:
s := reflect.ValueOf(i)
a := make([]bool, s.Len())
for j := 0; j < s.Len(); j++ {
val, err := ToBoolE(s.Index(j).Interface())
if err != nil {
return []bool{}, fmt.Errorf("Unable to Cast %#v to []bool", i)
}
a[j] = val
}
return a, nil
default:
return []bool{}, fmt.Errorf("Unable to Cast %#v to []bool", i)
}
}
2017-03-03 23:41:53 +08:00
// ToStringSliceE casts an interface to a []string type.
func ToStringSliceE(i interface{}) ([]string, error) {
var a []string
switch v := i.(type) {
case []interface{}:
for _, u := range v {
a = append(a, ToString(u))
}
return a, nil
case []string:
return v, nil
2015-02-19 10:44:31 +08:00
case string:
2015-02-20 01:58:55 +08:00
return strings.Fields(v), nil
case interface{}:
str, err := ToStringE(v)
if err != nil {
return a, fmt.Errorf("Unable to Cast %#v to []string", i)
}
return []string{str}, nil
default:
return a, fmt.Errorf("Unable to Cast %#v to []string", i)
}
}
2017-03-03 23:41:53 +08:00
// ToIntSliceE casts an interface to a []int type.
2015-02-24 18:58:16 +08:00
func ToIntSliceE(i interface{}) ([]int, error) {
if i == nil {
return []int{}, fmt.Errorf("Unable to Cast %#v to []int", i)
}
2015-02-24 18:58:16 +08:00
switch v := i.(type) {
case []int:
return v, nil
}
kind := reflect.TypeOf(i).Kind()
switch kind {
case reflect.Slice, reflect.Array:
s := reflect.ValueOf(i)
a := make([]int, s.Len())
for j := 0; j < s.Len(); j++ {
val, err := ToIntE(s.Index(j).Interface())
if err != nil {
return []int{}, fmt.Errorf("Unable to Cast %#v to []int", i)
}
a[j] = val
}
return a, nil
2015-02-24 18:58:16 +08:00
default:
return []int{}, fmt.Errorf("Unable to Cast %#v to []int", i)
2015-02-24 18:58:16 +08:00
}
}
2017-03-03 23:41:53 +08:00
// StringToDate attempts to parse a string into a time.Time type using a
// predefined list of formats. If no suitable format is found, an error is
// returned.
2014-04-05 13:21:52 +08:00
func StringToDate(s string) (time.Time, error) {
return parseDateWith(s, []string{
time.RFC3339,
"2006-01-02T15:04:05", // iso8601 without timezone
time.RFC1123Z,
time.RFC1123,
time.RFC822Z,
time.RFC822,
time.RFC850,
2014-04-05 13:21:52 +08:00
time.ANSIC,
time.UnixDate,
time.RubyDate,
"2006-01-02 15:04:05.999999999 -0700 MST", // Time.String()
2014-04-05 13:21:52 +08:00
"2006-01-02",
"02 Jan 2006",
"2006-01-02 15:04:05 -07:00",
"2006-01-02 15:04:05 -0700",
"2006-01-02 15:04:05Z07:00", // RFC3339 without T
"2006-01-02 15:04:05",
time.Kitchen,
time.Stamp,
time.StampMilli,
time.StampMicro,
time.StampNano,
2014-04-05 13:21:52 +08:00
})
}
func parseDateWith(s string, dates []string) (d time.Time, e error) {
for _, dateType := range dates {
if d, e = time.Parse(dateType, s); e == nil {
return
}
}
return d, fmt.Errorf("Unable to parse date: %s", s)
2014-04-05 13:21:52 +08:00
}