woodpecker/pipeline/frontend/yaml/compiler/params.go

package compiler

import (
	"fmt"
	"reflect"
	"strconv"
	"strings"

	"codeberg.org/6543/go-yaml2json"
	"gopkg.in/yaml.v3"
)

// paramsToEnv uses reflection to convert a map[string]interface to a list
// of environment variables.
func paramsToEnv(from map[string]interface{}, to map[string]string, secrets map[string]Secret) (err error) {
	if to == nil {
		return fmt.Errorf("no map to write to")
	}
	for k, v := range from {
		if v == nil || len(k) == 0 {
			continue
		}
		to[sanitizeParamKey(k)], err = sanitizeParamValue(v, secrets)
		if err != nil {
			return err
		}
	}
	return nil
}

func sanitizeParamKey(k string) string {
	return "PLUGIN_" +
		strings.ToUpper(
			strings.ReplaceAll(k, ".", "_"),
		)
}

func isComplex(t reflect.Kind) bool {
	switch t {
	case reflect.Bool,
		reflect.String,
		reflect.Int, reflect.Int16, reflect.Int32, reflect.Int64,
		reflect.Float32, reflect.Float64:
		return false
	default:
		return true
	}
}

func sanitizeParamValue(v interface{}, secrets map[string]Secret) (string, error) {
	t := reflect.TypeOf(v)
	vv := reflect.ValueOf(v)

	switch t.Kind() {
	case reflect.Bool:
		return strconv.FormatBool(vv.Bool()), nil

	case reflect.String:
		return vv.String(), nil

	case reflect.Int, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Int8:
		return fmt.Sprintf("%v", vv.Int()), nil

	case reflect.Float32, reflect.Float64:
		return fmt.Sprintf("%v", vv.Float()), nil

	case reflect.Map:
		if fromSecret, ok := v.(map[string]interface{}); ok {
			if secretNameI, ok := fromSecret["from_secret"]; ok {
				if secretName, ok := secretNameI.(string); ok {
					if secret, ok := secrets[strings.ToLower(secretName)]; ok {
						return secret.Value, nil
					}
					return "", fmt.Errorf("no secret found for %q", secretName)
				}
			}
		}
		ymlOut, _ := yaml.Marshal(vv.Interface())
		out, _ := yaml2json.Convert(ymlOut)
		return string(out), nil

	case reflect.Slice, reflect.Array:
		if vv.Len() == 0 {
			return "", nil
		}
		if !isComplex(t.Elem().Kind()) || t.Elem().Kind() == reflect.Interface {
			in := make([]string, vv.Len())
			for i := 0; i < vv.Len(); i++ {
				var err error
				if in[i], err = sanitizeParamValue(vv.Index(i).Interface(), secrets); err != nil {
					return "", err
				}
			}
			return strings.Join(in, ","), nil
		}

		// it's complex use yml.ToJSON
		fallthrough

	default:
		out, err := yaml.Marshal(vv.Interface())
		if err != nil {
			return "", err
		}
		out, err = yaml2json.Convert(out)
		if err != nil {
			return "", err
		}
		return string(out), nil
	}
}