proxy-pool/internal/config/config.go

package config

import (
	"fmt"
	"time"
)

type Duration time.Duration

func (d *Duration) UnmarshalText(text []byte) error {
	value, err := time.ParseDuration(string(text))
	if err != nil {
		return fmt.Errorf("parse duration %q: %w", text, err)
	}
	*d = Duration(value)
	return nil
}

func (d Duration) Value() time.Duration { return time.Duration(d) }

type Config struct {
	Version      int                 `yaml:"version"`
	Defaults     Defaults            `yaml:"defaults"`
	Security     Security            `yaml:"security"`
	Gateway      Listener            `yaml:"gateway"`
	Distribution Distribution        `yaml:"distribution"`
	Admin        Listener            `yaml:"admin"`
	Metrics      Metrics             `yaml:"metrics"`
	Storage      Storage             `yaml:"storage"`
	Routing      []Routing           `yaml:"routing"`
	Upstreams    map[string]Upstream `yaml:"upstreams"`
}

type Defaults struct {
	Fetch Fetch `yaml:"fetch"`
	Check Check `yaml:"check"`
}

type Security struct {
	RequireProtectionOnPublicListen bool `yaml:"requireProtectionOnPublicListen"`
}

type Listener struct {
	Enabled           bool              `yaml:"enabled"`
	Listen            string            `yaml:"listen"`
	Access            Access            `yaml:"access"`
	Auth              Auth              `yaml:"auth"`
	Limits            Limits            `yaml:"limits"`
	Retry             Retry             `yaml:"retry"`
	DestinationPolicy DestinationPolicy `yaml:"destinationPolicy"`
}

type Distribution struct {
	Listener             `yaml:",inline"`
	ClientIdentification ClientIdentification `yaml:"clientIdentification"`
	Extraction           Extraction           `yaml:"extraction"`
}

type Access struct {
	AllowCIDRs     []string `yaml:"allowCIDRs"`
	TrustedProxies []string `yaml:"trustedProxies"`
}

type Auth struct {
	Mode         string       `yaml:"mode"`
	Username     string       `yaml:"username"`
	Password     string       `yaml:"password"`
	PasswordFile string       `yaml:"passwordFile"`
	Token        string       `yaml:"token"`
	TokenFile    string       `yaml:"tokenFile"`
	Header       string       `yaml:"header"`
	CIDRs        []string     `yaml:"cidrs"`
	Methods      []AuthMethod `yaml:"methods"`
}

type AuthMethod struct {
	Mode         string   `yaml:"mode"`
	Username     string   `yaml:"username"`
	Password     string   `yaml:"password"`
	PasswordFile string   `yaml:"passwordFile"`
	Header       string   `yaml:"header"`
	Value        string   `yaml:"value"`
	ValueFile    string   `yaml:"valueFile"`
	CIDRs        []string `yaml:"cidrs"`
}

type Limits struct {
	MaxConcurrentConnections   int `yaml:"maxConcurrentConnections"`
	RequestsPerMinute          int `yaml:"requestsPerMinute"`
	RequestsPerMinutePerClient int `yaml:"requestsPerMinutePerClient"`
}

type Retry struct {
	MaxAttempts  int      `yaml:"maxAttempts"`
	RetryMethods []string `yaml:"retryMethods"`
}

type DestinationPolicy struct {
	DenyPrivateNetworks bool     `yaml:"denyPrivateNetworks"`
	DenyLoopback        bool     `yaml:"denyLoopback"`
	DenyLinkLocal       bool     `yaml:"denyLinkLocal"`
	DenyCIDRs           []string `yaml:"denyCIDRs"`
}

type ClientIdentification struct {
	Mode string `yaml:"mode"`
}

type Extraction struct {
	Fulfillment        string   `yaml:"fulfillment"`
	MaxCountPerRequest int      `yaml:"maxCountPerRequest"`
	MinRemainingTTL    Duration `yaml:"minRemainingTTL"`
	MaxHealthCheckAge  Duration `yaml:"maxHealthCheckAge"`
	ReserveForGateway  int      `yaml:"reserveForGateway"`
}

type Metrics struct {
	Enabled bool   `yaml:"enabled"`
	Listen  string `yaml:"listen"`
}

type Storage struct {
	PostgresURL string `yaml:"postgresURL"`
	RedisURL    string `yaml:"redisURL"`
}

type Routing struct {
	Name          string        `yaml:"name"`
	Enabled       bool          `yaml:"enabled"`
	Purpose       string        `yaml:"purpose"`
	Match         RoutingMatch  `yaml:"match"`
	Upstreams     []string      `yaml:"upstreams"`
	Strategy      Strategy      `yaml:"strategy"`
	OnUnavailable OnUnavailable `yaml:"onUnavailable"`
}

type RoutingMatch struct {
	HostRegex string            `yaml:"hostRegex"`
	Methods   []string          `yaml:"methods"`
	PathRegex string            `yaml:"pathRegex"`
	Headers   map[string]string `yaml:"headers"`
}

type Strategy struct {
	Type                  string         `yaml:"type"`
	SwitchAfterEmptyFetch int            `yaml:"switchAfterEmptyFetch"`
	EndBehavior           string         `yaml:"endBehavior"`
	Weights               map[string]int `yaml:"weights"`
}

type OnUnavailable struct {
	Action      string   `yaml:"action"`
	WaitTimeout Duration `yaml:"waitTimeout"`
}

type Upstream struct {
	Enabled   bool        `yaml:"enabled"`
	Exposure  []string    `yaml:"exposure"`
	Provider  Provider    `yaml:"provider"`
	API       ProviderAPI `yaml:"api"`
	ProxyAuth ProxyAuth   `yaml:"proxyAuth"`
	Pool      Pool        `yaml:"pool"`
	Capacity  Capacity    `yaml:"capacity"`
	Lifecycle Lifecycle   `yaml:"lifecycle"`
	Fetch     Fetch       `yaml:"fetch"`
	Check     Check       `yaml:"check"`
}

type Provider struct {
	BillingMode string   `yaml:"billingMode"`
	Protocols   []string `yaml:"protocols"`
}

type ProviderAPI struct {
	URL      string            `yaml:"url"`
	Method   string            `yaml:"method"`
	Auth     ProviderAuth      `yaml:"auth"`
	Headers  map[string]string `yaml:"headers"`
	Query    map[string]string `yaml:"query"`
	Body     APIBody           `yaml:"body"`
	Template string            `yaml:"template"`
}

type ProviderAuth struct {
	Type         string `yaml:"type"`
	Username     string `yaml:"username"`
	Password     string `yaml:"password"`
	PasswordFile string `yaml:"passwordFile"`
	Token        string `yaml:"token"`
	TokenFile    string `yaml:"tokenFile"`
	Location     string `yaml:"location"`
	Name         string `yaml:"name"`
	Value        string `yaml:"value"`
	ValueFile    string `yaml:"valueFile"`
}

type APIBody struct {
	Type  string            `yaml:"type"`
	Value map[string]string `yaml:"value"`
}

type ProxyAuth struct {
	Type         string `yaml:"type"`
	Username     string `yaml:"username"`
	Password     string `yaml:"password"`
	PasswordFile string `yaml:"passwordFile"`
}

type Pool struct {
	MaxSize     int      `yaml:"maxSize"`
	ShrinkDelay Duration `yaml:"shrinkDelay"`
}

type Capacity struct {
	MaxConcurrencyPerProxy int `yaml:"maxConcurrencyPerProxy"`
}

type Lifecycle struct {
	TTL                    Duration `yaml:"ttl"`
	AllocationSafetyMargin Duration `yaml:"allocationSafetyMargin"`
}

type Fetch struct {
	RequestInterval  Duration `yaml:"requestInterval"`
	Timeout          Duration `yaml:"timeout"`
	MaxAttempts      int      `yaml:"maxAttempts"`
	MaxInFlight      int      `yaml:"maxInFlight"`
	MaxTotal         int      `yaml:"maxTotal"`
	MaxResponseBytes int64    `yaml:"maxResponseBytes"`
	TemplateTimeout  Duration `yaml:"templateTimeout"`
	Retry            Backoff  `yaml:"retry"`
}

type Backoff struct {
	Initial Duration `yaml:"initial"`
	Max     Duration `yaml:"max"`
	Jitter  int      `yaml:"jitter"`
}

type Check struct {
	Interval               Duration `yaml:"interval"`
	Jitter                 int      `yaml:"jitter"`
	MaxInFlight            int      `yaml:"maxInFlight"`
	Timeout                Duration `yaml:"timeout"`
	MaxAttempts            int      `yaml:"maxAttempts"`
	MaxConsecutiveFailures int      `yaml:"maxConsecutiveFailures"`
	URLs                   []string `yaml:"urls"`
}