gslb

type Backend struct {
	Fqdn            string               // Fully qualified domain name
	Description     string               // Description of the backend
	Address         string               // IP address or hostname
	Priority        int                  // Priority for load balancing
	Weight          int                  // Weight for weighted load balancing
	Enable          bool                 // Enable or disable the backend
	Tags            []string             // List of tags for filtering or grouping
	HealthChecks    []GenericHealthCheck `yaml:"healthchecks"` // Health check configurations
	Timeout         string               // Timeout for requests
	Alive           bool                 // Indicates if the backend is alive
	Continent       string               // Continent code for GeoIP (e.g. EU)
	Country         string               // Country code for GeoIP
	Subdivision     string               // Subdivision/state code for GeoIP (e.g. CA, NY)
	City            string               // City name for GeoIP
	ASN             string               // ASN for GeoIP
	Location        string               // location
	Longitude       float64              // Longitude for distance-based GeoIP routing
	Latitude        float64              // Latitude for distance-based GeoIP routing
	LongitudeRad    float64              // Precomputed longitude in radians for distance calculations
	LatitudeRad     float64              // Precomputed latitude in radians for distance calculations
	HasCoordinates  bool                 // Indicates whether both coordinates were explicitly configured
	LastHealthcheck time.Time            // Last time a healthcheck was launched
	// contains filtered or unexported fields
}

type BackendInterface interface {
	GetFqdn() string
	SetFqdn(fqdn string)
	GetDescription() string
	GetAddress() string
	GetPriority() int
	GetWeight() int
	IsEnabled() bool
	GetTags() []string
	GetHealthChecks() []GenericHealthCheck
	GetTimeout() string
	GetContinent() string
	GetCountry() string
	GetSubdivision() string
	GetCity() string
	GetASN() string
	GetLocation() string
	GetLongitude() float64
	GetLatitude() float64
	GetLongitudeRad() float64
	GetLatitudeRad() float64
	HasGeoCoordinates() bool
	IsHealthy() bool

	Lock()
	Unlock()
	// contains filtered or unexported methods
}

type ClientInfo struct {
	IP        net.IP
	PrefixLen uint8
}

type GRPCHealthCheck struct {
	Host    string
	Port    int
	Service string
	Timeout time.Duration
}

type GSLB struct {
	Next                plugin.Handler
	Zones               map[string]string             // List of authoritative domains
	Records             map[string]map[string]*Record // zone -> fqdn -> record
	HealthcheckProfiles map[string]*HealthCheck       `yaml:"healthcheck_profiles"`

	Zone                      string   // Zone attendue pour la vérification des records
	LastResolution            sync.Map // key: domain (string), value: time.Time
	RoundRobinIndex           sync.Map
	MaxStaggerStart           string
	BatchSizeStart            int
	ResolutionIdleTimeout     string
	ResolutionIdleMultiplier  int // Multiplier for slow healthcheck interval
	HealthcheckIdleMultiplier int // Multiplier for slow healthcheck interval
	Mutex                     sync.RWMutex
	UseEDNSCSubnet            bool
	LocationMap               map[string]string
	GeoIPCountryDB            *geoip2.Reader // Loaded MaxMind DB (country)
	GeoIPCityDB               *geoip2.Reader // Loaded MaxMind DB (city)
	GeoIPASNDB                *geoip2.Reader // Loaded MaxMind DB (ASN)
	APIEnable                 bool           // Enable/disable API HTTP server
	APICertPath               string         // TLS certificate path for API
	APIKeyPath                string         // TLS key path for API
	APIListenAddr             string         // API listen address (default 0.0.0.0)
	APIListenPort             string         // API listen port (default 8080)
	APIBasicUser              string         // HTTP Basic Auth username (optional)
	APIBasicPass              string         // HTTP Basic Auth password (optional)
	// DisableTXT disables TXT record resolution if set to true
	DisableTXT bool
}

type GenericHealthCheck interface {
	// PerformCheck executes the health check for a backend.
	PerformCheck(backend *Backend, fdqn string, maxRetries int) bool

	// GetType returns the type of the health check (e.g., "http/80").
	GetType() string

	// Equals compares the current health check with another instance for equality.
	Equals(other GenericHealthCheck) bool
}

type HTTPHealthCheck struct {
	Port          int               `yaml:"port" default:"443"`
	EnableTLS     bool              `yaml:"enable_tls" default:"true"`
	Cert          string            `yaml:"cert_path" default:""`
	Key           string            `yaml:"key_path" default:""`
	CA            string            `yaml:"ca_path" default:""`
	URI           string            `yaml:"uri" default:"/"`
	Method        string            `yaml:"method" default:"GET"`
	Host          string            `yaml:"host" default:"localhost"`
	Headers       map[string]string `yaml:"headers"`
	Timeout       string            `yaml:"timeout" default:"5s"`
	ExpectedCode  int               `yaml:"expected_code" default:"200"`
	ExpectedBody  string            `yaml:"expected_body" default:""`
	SkipTLSVerify bool              `yaml:"skip_tls_verify" default:"false"`
}

type HealthCheck struct {
	Type   string                 `yaml:"type"`
	Params map[string]interface{} `yaml:"params"`
}

type ICMPHealthCheck struct {
	Count   int    `yaml:"count" default:"3"`    // Number of ICMP packets to send
	Timeout string `yaml:"timeout" default:"5s"` // Maximum duration for pings
}

type LuaHealthCheck struct {
	Script  string        `yaml:"script"`
	Timeout time.Duration `yaml:"timeout"`
}

type MockHealthCheck struct{}

type MySQLHealthCheck struct {
	Host     string `yaml:"host"`
	Port     int    `yaml:"port" default:"3306"`
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Database string `yaml:"database"`
	Timeout  string `yaml:"timeout" default:"3s"`
	Query    string `yaml:"query" default:"SELECT 1"`
}

type Pinger interface {
	Run() error
	Statistics() *probing.Statistics
	SetPrivileged(privileged bool)
}

type RealPinger struct {
	// contains filtered or unexported fields
}

type Record struct {
	Fqdn           string
	Mode           string
	Backends       []BackendInterface
	Owner          string
	Description    string
	RecordTTL      int
	ScrapeInterval string
	ScrapeRetries  int
	ScrapeTimeout  string
	// contains filtered or unexported fields
}

type TCPHealthCheck struct {
	Port    int    `yaml:"port" default:"80"`    // TCP port to connect to
	Timeout string `yaml:"timeout" default:"5s"` // Timeout for the TCP connection
}