kratos/x/map_json.go

80 lines
1.6 KiB
Go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package x
import (
"bytes"
"encoding/json"
"regexp"
"strconv"
"strings"
)
var (
regexMatchInt = regexp.MustCompile(`^[0-9]+$`)
regexMatchFloat = regexp.MustCompile(`^[0-9]+\.[0-9]+$`)
regexMatchBool = regexp.MustCompile(`^(?i)false|true|on$`)
)
func TypeMap(m map[string]string) (map[string]interface{}, error) {
jm := make(map[string]interface{})
for k, v := range m {
if regexMatchInt.MatchString(v) {
vv, err := strconv.ParseInt(v, 10, 64)
if err != nil {
return nil, err
}
jm[k] = vv
} else if regexMatchFloat.MatchString(v) {
vv, err := strconv.ParseFloat(v, 64)
if err != nil {
return nil, err
}
jm[k] = vv
} else if regexMatchBool.MatchString(v) {
// Checkboxes have default values of `on` when checked, so set this to true.
if strings.ToLower(v) == "on" {
v = "true"
}
vv, err := strconv.ParseBool(strings.ToLower(v))
if err != nil {
return nil, err
}
jm[k] = vv
} else {
jm[k] = v
}
}
return jm, nil
}
func UntypedMapToJSON(m map[string]string) (json.RawMessage, error) {
jm, err := TypeMap(m)
if err != nil {
return nil, err
}
var b bytes.Buffer
if err := json.NewEncoder(&b).Encode(jm); err != nil {
return nil, err
}
return b.Bytes(), nil
}
func StructToMap(s interface{}) (map[string]interface{}, error) {
b, err := json.Marshal(s)
if err != nil {
return nil, err
}
mapData := make(map[string]interface{})
err = json.Unmarshal(b, &mapData)
if err != nil {
return nil, err
}
return mapData, nil
}