common.go

package common

import (
	"reflect"
	"strconv"
)

// JWTType is the type of user JWTs that must be implemented.
type JWTType int

// Values of JWTType
const (
	PKI JWTType = iota
	OIDC
)

// convert an unknown int type to int64
func toInt64(i interface{}) int64 {
	switch i := i.(type) {
	case int:
		return int64(i)
	case int8:
		return int64(i)
	case int16:
		return int64(i)
	case int32:
		return int64(i)
	case int64:
		return i
	default:
		panic("toInt64(): expected a signed integer type, got " + reflect.TypeOf(i).String() + " instead")
	}
}

// convert an unknown int type to int64
func toUint64(i interface{}) uint64 {
	switch i := i.(type) {
	case uint:
		return uint64(i)
	case uint8:
		return uint64(i)
	case uint16:
		return uint64(i)
	case uint32:
		return uint64(i)
	case uint64:
		return i
	default:
		panic("toUint64(): expected an unsigned integer type, got " + reflect.TypeOf(i).String() + " instead")
	}
}

// FlattenClaim flattens all the generic claims in a flat array for strings.
func FlattenClaim(key string, claim interface{}) []string {
	attributes := []string{}

	switch claim := claim.(type) {
	case bool:
		attributes = append(attributes, key+"="+strconv.FormatBool(claim))
	case int, int8, int16, int32, int64:
		attributes = append(attributes, key+"="+strconv.FormatInt(toInt64(claim), 10))
	case uint, uint8, uint16, uint32, uint64:
		attributes = append(attributes, key+"="+strconv.FormatUint(toUint64(claim), 10))
	case float32:
		attributes = append(attributes, key+"="+strconv.FormatFloat(float64(claim), 'G', -1, 32))
	case float64:
		attributes = append(attributes, key+"="+strconv.FormatFloat(claim, 'G', -1, 64))
	case string:
		attributes = append(attributes, key+"="+claim)
	case []string:
		for _, data := range claim {
			attributes = append(attributes, key+"="+data)
		}
	case map[string]interface{}:
		for ikey, ivalue := range claim {
			for _, v := range FlattenClaim(ikey, ivalue) {
				attributes = append(attributes, key+":"+v)
			}
		}
	case []interface{}:
		for _, value := range claim {
			attributes = append(attributes, FlattenClaim(key, value)...)
		}
	default:
		// do nothing, just return attributes
	}
	return attributes
}