query.go

package fireblocksdk

import (
	"fmt"
	"net/url"
	"reflect"
	"strings"
)

type QueryItems []QueryItem

//go:generate stringer -type=EnvEntry
type QueryItem struct {
	Key, Value string
}

func (e QueryItem) String() string {
	return fmt.Sprintf("%s = %s", e.Key, e.Value)
}

// BuildQuery uses `env` and `envDefault` as tag to bind config to viper bindings
// Example: `env:"USERNAME" envDefault:"admin"`
func BuildQuery(in any) QueryItems {
	if in == nil || reflect.ValueOf(in).IsNil() {
		return nil
	}

	var vars QueryItems
	var t = reflect.TypeOf(in)
	var kind = t.Kind()
	if kind == reflect.Ptr {
		t = t.Elem()
		kind = t.Kind()
	}

	if kind == reflect.Struct {
		iterateStructFields(t, in, &vars)
	}

	return vars
}

func (items QueryItems) URLValues() url.Values {
	values := make(url.Values, len(items))

	for _, val := range items {
		if val.Value != "" {
			values.Add(normalize(val.Key, val.Value))
		}
	}

	return values
}

func normalize(key, value string) (string, string) {
	substrings := strings.Split(key, ",")

	return substrings[0], value
}

func iterateStructFields(t reflect.Type, v any, vars *QueryItems) {
	val := reflect.ValueOf(v)
	if val.Kind() == reflect.Ptr {
		val = val.Elem()
	}

	for i := 0; i < t.NumField(); i++ {
		var field = t.Field(i)
		tag := field.Tag.Get("json")
		value := val.Field(i)
		kind := value.Kind()

		var vv any
		if kind == reflect.Ptr {
			value = value.Elem()
			vv = value.Interface()
		} else {
			vv = value.Interface()
		}

		entry := QueryItem{
			Key:   tag,
			Value: fmt.Sprintf("%v", vv),
		}

		*vars = append(*vars, entry)
	}
}