godoxy-yusing/src/docker/label_parser.go

package docker

import (
	"strings"

	E "github.com/yusing/go-proxy/error"
	"gopkg.in/yaml.v3"
)

func yamlListParser(value string) (any, E.NestedError) {
	value = strings.TrimSpace(value)
	if value == "" {
		return []string{}, nil
	}
	var data []string
	err := E.From(yaml.Unmarshal([]byte(value), &data))
	return data, err
}

func yamlStringMappingParser(value string) (any, E.NestedError) {
	value = strings.TrimSpace(value)
	lines := strings.Split(value, "\n")
	h := make(map[string]string)
	for _, line := range lines {
		parts := strings.SplitN(line, ":", 2)
		if len(parts) != 2 {
			return nil, E.Invalid("set header statement", line)
		}
		key := strings.TrimSpace(parts[0])
		val := strings.TrimSpace(parts[1])
		if existing, ok := h[key]; ok {
			h[key] = existing + ", " + val
		} else {
			h[key] = val
		}
	}
	return h, nil
}

func boolParser(value string) (any, E.NestedError) {
	switch strings.ToLower(value) {
	case "true", "yes", "1":
		return true, nil
	case "false", "no", "0":
		return false, nil
	default:
		return nil, E.Invalid("boolean value", value)
	}
}

const NSProxy = "proxy"

var _ = func() int {
	RegisterNamespace(NSProxy, ValueParserMap{
		"path_patterns": yamlListParser,
		"set_headers":   yamlStringMappingParser,
		"hide_headers":  yamlListParser,
		"no_tls_verify": boolParser,
	})
	return 0
}()