package config

import (
	"fmt"
	"strings"

	"gopkg.in/yaml.v3"
)

type IPSource string

const (
	Legacy        IPSource = ""
	Direct        IPSource = "direct"
	XForwardedFor IPSource = "x-forwarded-for"
	XRealIP       IPSource = "x-real-ip"
)

func (ips *IPSource) UnmarshalYAML(val *yaml.Node) error {
	var str string
	err := val.Decode(&str)
	if err != nil {
		return err
	}

	switch v := IPSource(strings.ToLower(str)); v {
	case Legacy, Direct, XForwardedFor, XRealIP:
		*ips = v
	default:
		return fmt.Errorf("invalid IP source '%s'", str)
	}

	return nil
}

type Config struct {
	Bind           string   `yaml:"bind"`
	IPSource       IPSource `yaml:"ip_source"`
	TrustedProxies []string `yaml:"trusted_proxies"`
}