mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-31 13:50:01 +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
 | |
| }
 |