ratelimit

type Backend interface {
	// Connection tracking
	GetConnectionCount(ip string) (int, error)
	IncrementConnectionCount(ip string, ttl time.Duration) error
	DecrementConnectionCount(ip string) error

	// Request rate tracking
	GetRequestCount(ip string, window time.Duration) (int, error)
	IncrementRequestCount(ip string, window time.Duration) error

	// Error rate tracking
	GetErrorCount(ip string, window time.Duration) (int, error)
	IncrementErrorCount(ip string, window time.Duration) error

	// Ban management
	IsBanned(ip string) (bool, time.Time, string, error)
	BanIP(ip string, duration time.Duration, reason string) error
	UnbanIP(ip string) error

	// Tor exit node management
	IsTorExit(ip string) (bool, error)
	SetTorExits(ips []string, ttl time.Duration) error
	GetTorExitCount() (int, error)

	// Global Tor rate limiting
	GetGlobalTorRequestCount(window time.Duration) (int, error)
	IncrementGlobalTorRequestCount(window time.Duration) error

	// Statistics
	GetStats() (BackendStats, error)

	// Cleanup
	Cleanup(ctx context.Context) error

	// Close
	Close() error
}

type BackendConfig struct {
	Type   string       `toml:"type"`
	Memory MemoryConfig `toml:"memory"`
	Redis  RedisConfig  `toml:"redis"`
}

type BackendStats struct {
	TrackedIPs    int       `json:"tracked_ips"`
	BannedIPs     int       `json:"banned_ips"`
	TorBanned     int       `json:"tor_banned"`
	BackendType   string    `json:"backend_type"`
	TorExitCount  int       `json:"tor_exits_count"`
	TorLastUpdate time.Time `json:"tor_last_updated"`
}

type BanRecord struct {
	ExpiresAt time.Time
	Reason    string
	BanType   string
}

type CacheInfo struct {
	Exists  bool      `json:"exists"`
	Path    string    `json:"path"`
	Size    int64     `json:"size"`
	ModTime time.Time `json:"mod_time"`
	Count   int       `json:"count"`
	IsStale bool      `json:"is_stale"`
}

type Config struct {
	Enabled                  bool          `toml:"enabled"`
	MaxConcurrentConnections int           `toml:"maxConcurrentConnections"`
	ConnectionRate           int           `toml:"connectionRate"`
	HTTPRequestRate          int           `toml:"httpRequestRate"`
	HTTPErrorRate            int           `toml:"httpErrorRate"`
	CrawlerBlockDuration     time.Duration `toml:"crawlerBlockDuration"`
	TrustProxyHeaders        bool          `toml:"trustProxyHeaders"`

	// Backend configuration
	Backend BackendConfig `toml:"backend"`

	// Tor configuration
	Tor TorConfig `toml:"tor"`

	// Whitelist configuration
	Whitelist WhitelistConfig `toml:"whitelist"`

	// Keyserver sync configuration
	KeyserverSync KeyserverSyncConfig `toml:"keyserverSync"`

	// Header configuration
	Headers HeaderConfig `toml:"headers"`
}

type HeaderConfig struct {
	Enabled   bool   `toml:"enabled"`
	TorHeader string `toml:"torHeader"`
	BanHeader string `toml:"banHeader"`
}

type KeyserverSyncConfig struct {
	Enabled bool `toml:"enabled"`
}

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

type MemoryConfig struct {
}

type RateLimitMetrics struct {
	ViolationsTotal *prometheus.CounterVec
	BannedIPs       *prometheus.GaugeVec
	TrackedIPs      prometheus.Gauge
	TorExitCount    prometheus.Gauge
	BackendDuration *prometheus.HistogramVec
}

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

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

type RedisConfig struct {
	Addr         string        `toml:"addr"`
	Password     string        `toml:"password"`
	DB           int           `toml:"db"`
	PoolSize     int           `toml:"poolSize"`
	DialTimeout  time.Duration `toml:"dialTimeout"`
	ReadTimeout  time.Duration `toml:"readTimeout"`
	WriteTimeout time.Duration `toml:"writeTimeout"`
	KeyPrefix    string        `toml:"keyPrefix"`
	TTL          time.Duration `toml:"ttl"`
	MaxRetries   int           `toml:"maxRetries"`
}

type TorConfig struct {
	Enabled                   bool          `toml:"enabled"`
	MaxRequestsPerConnection  int           `toml:"maxRequestsPerConnection"`
	MaxConcurrentConnections  int           `toml:"maxConcurrentConnections"`
	ConnectionRate            int           `toml:"connectionRate"`
	ConnectionRateWindow      time.Duration `toml:"connectionRateWindow"`
	BanDuration               time.Duration `toml:"banDuration"`
	RepeatOffenderBanDuration time.Duration `toml:"repeatOffenderBanDuration"`
	ExitNodeListURL           string        `toml:"exitNodeListURL"`
	UpdateInterval            time.Duration `toml:"updateInterval"`
	CacheFilePath             string        `toml:"cacheFilePath"`
	UserAgent                 string        `toml:"userAgent"`

	// Global Tor rate limiting
	GlobalRateLimit   bool          `toml:"globalRateLimit"`
	GlobalRequestRate int           `toml:"globalRequestRate"`
	GlobalRateWindow  time.Duration `toml:"globalRateWindow"`
	GlobalBanDuration time.Duration `toml:"globalBanDuration"`
}

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

type WhitelistConfig struct {
	IPs []string `toml:"ips"`
}