mirror of
https://github.com/k3s-io/k3s.git
synced 2024-06-07 19:41:36 +00:00
281 lines
6.9 KiB
Go
281 lines
6.9 KiB
Go
|
package objx
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/base64"
|
||
|
"encoding/json"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"net/url"
|
||
|
"strconv"
|
||
|
)
|
||
|
|
||
|
// SignatureSeparator is the character that is used to
|
||
|
// separate the Base64 string from the security signature.
|
||
|
const SignatureSeparator = "_"
|
||
|
|
||
|
// URLValuesSliceKeySuffix is the character that is used to
|
||
|
// specify a suffic for slices parsed by URLValues.
|
||
|
// If the suffix is set to "[i]", then the index of the slice
|
||
|
// is used in place of i
|
||
|
// Ex: Suffix "[]" would have the form a[]=b&a[]=c
|
||
|
// OR Suffix "[i]" would have the form a[0]=b&a[1]=c
|
||
|
// OR Suffix "" would have the form a=b&a=c
|
||
|
var urlValuesSliceKeySuffix = "[]"
|
||
|
|
||
|
const (
|
||
|
URLValuesSliceKeySuffixEmpty = ""
|
||
|
URLValuesSliceKeySuffixArray = "[]"
|
||
|
URLValuesSliceKeySuffixIndex = "[i]"
|
||
|
)
|
||
|
|
||
|
// SetURLValuesSliceKeySuffix sets the character that is used to
|
||
|
// specify a suffic for slices parsed by URLValues.
|
||
|
// If the suffix is set to "[i]", then the index of the slice
|
||
|
// is used in place of i
|
||
|
// Ex: Suffix "[]" would have the form a[]=b&a[]=c
|
||
|
// OR Suffix "[i]" would have the form a[0]=b&a[1]=c
|
||
|
// OR Suffix "" would have the form a=b&a=c
|
||
|
func SetURLValuesSliceKeySuffix(s string) error {
|
||
|
if s == URLValuesSliceKeySuffixEmpty || s == URLValuesSliceKeySuffixArray || s == URLValuesSliceKeySuffixIndex {
|
||
|
urlValuesSliceKeySuffix = s
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
return errors.New("objx: Invalid URLValuesSliceKeySuffix provided.")
|
||
|
}
|
||
|
|
||
|
// JSON converts the contained object to a JSON string
|
||
|
// representation
|
||
|
func (m Map) JSON() (string, error) {
|
||
|
for k, v := range m {
|
||
|
m[k] = cleanUp(v)
|
||
|
}
|
||
|
|
||
|
result, err := json.Marshal(m)
|
||
|
if err != nil {
|
||
|
err = errors.New("objx: JSON encode failed with: " + err.Error())
|
||
|
}
|
||
|
return string(result), err
|
||
|
}
|
||
|
|
||
|
func cleanUpInterfaceArray(in []interface{}) []interface{} {
|
||
|
result := make([]interface{}, len(in))
|
||
|
for i, v := range in {
|
||
|
result[i] = cleanUp(v)
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
func cleanUpInterfaceMap(in map[interface{}]interface{}) Map {
|
||
|
result := Map{}
|
||
|
for k, v := range in {
|
||
|
result[fmt.Sprintf("%v", k)] = cleanUp(v)
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
func cleanUpStringMap(in map[string]interface{}) Map {
|
||
|
result := Map{}
|
||
|
for k, v := range in {
|
||
|
result[k] = cleanUp(v)
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
func cleanUpMSIArray(in []map[string]interface{}) []Map {
|
||
|
result := make([]Map, len(in))
|
||
|
for i, v := range in {
|
||
|
result[i] = cleanUpStringMap(v)
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
func cleanUpMapArray(in []Map) []Map {
|
||
|
result := make([]Map, len(in))
|
||
|
for i, v := range in {
|
||
|
result[i] = cleanUpStringMap(v)
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
func cleanUp(v interface{}) interface{} {
|
||
|
switch v := v.(type) {
|
||
|
case []interface{}:
|
||
|
return cleanUpInterfaceArray(v)
|
||
|
case []map[string]interface{}:
|
||
|
return cleanUpMSIArray(v)
|
||
|
case map[interface{}]interface{}:
|
||
|
return cleanUpInterfaceMap(v)
|
||
|
case Map:
|
||
|
return cleanUpStringMap(v)
|
||
|
case []Map:
|
||
|
return cleanUpMapArray(v)
|
||
|
default:
|
||
|
return v
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// MustJSON converts the contained object to a JSON string
|
||
|
// representation and panics if there is an error
|
||
|
func (m Map) MustJSON() string {
|
||
|
result, err := m.JSON()
|
||
|
if err != nil {
|
||
|
panic(err.Error())
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
// Base64 converts the contained object to a Base64 string
|
||
|
// representation of the JSON string representation
|
||
|
func (m Map) Base64() (string, error) {
|
||
|
var buf bytes.Buffer
|
||
|
|
||
|
jsonData, err := m.JSON()
|
||
|
if err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
|
||
|
encoder := base64.NewEncoder(base64.StdEncoding, &buf)
|
||
|
_, _ = encoder.Write([]byte(jsonData))
|
||
|
_ = encoder.Close()
|
||
|
|
||
|
return buf.String(), nil
|
||
|
}
|
||
|
|
||
|
// MustBase64 converts the contained object to a Base64 string
|
||
|
// representation of the JSON string representation and panics
|
||
|
// if there is an error
|
||
|
func (m Map) MustBase64() string {
|
||
|
result, err := m.Base64()
|
||
|
if err != nil {
|
||
|
panic(err.Error())
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
// SignedBase64 converts the contained object to a Base64 string
|
||
|
// representation of the JSON string representation and signs it
|
||
|
// using the provided key.
|
||
|
func (m Map) SignedBase64(key string) (string, error) {
|
||
|
base64, err := m.Base64()
|
||
|
if err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
|
||
|
sig := HashWithKey(base64, key)
|
||
|
return base64 + SignatureSeparator + sig, nil
|
||
|
}
|
||
|
|
||
|
// MustSignedBase64 converts the contained object to a Base64 string
|
||
|
// representation of the JSON string representation and signs it
|
||
|
// using the provided key and panics if there is an error
|
||
|
func (m Map) MustSignedBase64(key string) string {
|
||
|
result, err := m.SignedBase64(key)
|
||
|
if err != nil {
|
||
|
panic(err.Error())
|
||
|
}
|
||
|
return result
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
URL Query
|
||
|
------------------------------------------------
|
||
|
*/
|
||
|
|
||
|
// URLValues creates a url.Values object from an Obj. This
|
||
|
// function requires that the wrapped object be a map[string]interface{}
|
||
|
func (m Map) URLValues() url.Values {
|
||
|
vals := make(url.Values)
|
||
|
|
||
|
m.parseURLValues(m, vals, "")
|
||
|
|
||
|
return vals
|
||
|
}
|
||
|
|
||
|
func (m Map) parseURLValues(queryMap Map, vals url.Values, key string) {
|
||
|
useSliceIndex := false
|
||
|
if urlValuesSliceKeySuffix == "[i]" {
|
||
|
useSliceIndex = true
|
||
|
}
|
||
|
|
||
|
for k, v := range queryMap {
|
||
|
val := &Value{data: v}
|
||
|
switch {
|
||
|
case val.IsObjxMap():
|
||
|
if key == "" {
|
||
|
m.parseURLValues(val.ObjxMap(), vals, k)
|
||
|
} else {
|
||
|
m.parseURLValues(val.ObjxMap(), vals, key+"["+k+"]")
|
||
|
}
|
||
|
case val.IsObjxMapSlice():
|
||
|
sliceKey := k
|
||
|
if key != "" {
|
||
|
sliceKey = key + "[" + k + "]"
|
||
|
}
|
||
|
|
||
|
if useSliceIndex {
|
||
|
for i, sv := range val.MustObjxMapSlice() {
|
||
|
sk := sliceKey + "[" + strconv.FormatInt(int64(i), 10) + "]"
|
||
|
m.parseURLValues(sv, vals, sk)
|
||
|
}
|
||
|
} else {
|
||
|
sliceKey = sliceKey + urlValuesSliceKeySuffix
|
||
|
for _, sv := range val.MustObjxMapSlice() {
|
||
|
m.parseURLValues(sv, vals, sliceKey)
|
||
|
}
|
||
|
}
|
||
|
case val.IsMSISlice():
|
||
|
sliceKey := k
|
||
|
if key != "" {
|
||
|
sliceKey = key + "[" + k + "]"
|
||
|
}
|
||
|
|
||
|
if useSliceIndex {
|
||
|
for i, sv := range val.MustMSISlice() {
|
||
|
sk := sliceKey + "[" + strconv.FormatInt(int64(i), 10) + "]"
|
||
|
m.parseURLValues(New(sv), vals, sk)
|
||
|
}
|
||
|
} else {
|
||
|
sliceKey = sliceKey + urlValuesSliceKeySuffix
|
||
|
for _, sv := range val.MustMSISlice() {
|
||
|
m.parseURLValues(New(sv), vals, sliceKey)
|
||
|
}
|
||
|
}
|
||
|
case val.IsStrSlice(), val.IsBoolSlice(),
|
||
|
val.IsFloat32Slice(), val.IsFloat64Slice(),
|
||
|
val.IsIntSlice(), val.IsInt8Slice(), val.IsInt16Slice(), val.IsInt32Slice(), val.IsInt64Slice(),
|
||
|
val.IsUintSlice(), val.IsUint8Slice(), val.IsUint16Slice(), val.IsUint32Slice(), val.IsUint64Slice():
|
||
|
|
||
|
sliceKey := k
|
||
|
if key != "" {
|
||
|
sliceKey = key + "[" + k + "]"
|
||
|
}
|
||
|
|
||
|
if useSliceIndex {
|
||
|
for i, sv := range val.StringSlice() {
|
||
|
sk := sliceKey + "[" + strconv.FormatInt(int64(i), 10) + "]"
|
||
|
vals.Set(sk, sv)
|
||
|
}
|
||
|
} else {
|
||
|
sliceKey = sliceKey + urlValuesSliceKeySuffix
|
||
|
vals[sliceKey] = val.StringSlice()
|
||
|
}
|
||
|
|
||
|
default:
|
||
|
if key == "" {
|
||
|
vals.Set(k, val.String())
|
||
|
} else {
|
||
|
vals.Set(key+"["+k+"]", val.String())
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// URLQuery gets an encoded URL query representing the given
|
||
|
// Obj. This function requires that the wrapped object be a
|
||
|
// map[string]interface{}
|
||
|
func (m Map) URLQuery() (string, error) {
|
||
|
return m.URLValues().Encode(), nil
|
||
|
}
|