blocklist

type BlocklistConfig struct {
	// MaxFailedAttempts 最大失败尝试次数，超过则封禁
	MaxFailedAttempts int
	// BlockDuration 封禁时长
	BlockDuration time.Duration
	// ClearInterval 清理间隔，定期清理过期的记录
	ClearInterval time.Duration
	// OnBlocked 封禁时的处理函数
	OnBlocked func(w http.ResponseWriter, r *http.Request)
	// RecordExpiry 记录过期时间，过期后失败次数重置
	RecordExpiry time.Duration
	// WhitelistIPs 白名单IP，这些IP不会被封禁
	WhitelistIPs []string

	// Storage 存储实现，默认为内存存储
	Storage Storage
	// UseProgressiveBlocking 是否使用递增封禁时长
	UseProgressiveBlocking bool
	// ProgressiveBlockingFactor 递增封禁时长因子
	ProgressiveBlockingFactor float64
	// MaxBlockDuration 最大封禁时长
	MaxBlockDuration time.Duration
	// GeoRestriction 地理位置限制，可选
	GeoRestriction *GeoRestriction
	// AllowPrivateIPs 是否允许私有IP地址
	AllowPrivateIPs bool
	// contains filtered or unexported fields
}

type CountryInfo struct {
	// Code 国家代码
	Code string `json:"code"`
	// Name 国家名称
	Name string `json:"name"`
	// Continent 大陆
	Continent string `json:"continent"`
}

type GeoRestriction struct {
	// Mode 限制模式
	Mode GeoRestrictionMode
	// Countries 国家代码列表
	Countries []string

	// DB GeoIP2数据库
	DB *geoip2.Reader
	// contains filtered or unexported fields
}

type GeoRestrictionMode int

type IPRecord struct {
	// IP地址
	IP string `json:"ip"`
	// LastActivity 最后活动时间
	LastActivity time.Time `json:"last_activity"`
	// FailedAttempts 失败尝试次数
	FailedAttempts int `json:"failed_attempts"`
	// BlockedUntil 封禁解除时间
	BlockedUntil time.Time `json:"blocked_until"`
	// BlockCount 封禁次数，用于递增封禁时长
	BlockCount int `json:"block_count"`
	// CountryCode 国家/地区代码
	CountryCode string `json:"country_code,omitempty"`
}

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

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

type MistBlocklistConfig struct {
	// 原始黑名单配置
	Config *BlocklistConfig
	// 当IP被封禁时的处理函数（适用于Mist框架）
	OnBlocked func(*mist.Context)
}

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

type Storage interface {
	// SaveIPRecord 保存IP记录
	SaveIPRecord(record *IPRecord) error
	// GetIPRecord 获取IP记录
	GetIPRecord(ip string) (*IPRecord, error)
	// DeleteIPRecord 删除IP记录
	DeleteIPRecord(ip string) error
	// ListIPRecords 列出所有IP记录
	ListIPRecords() ([]*IPRecord, error)
	// ListBlockedIPs 列出所有被封禁的IP
	ListBlockedIPs() ([]*IPRecord, error)
	// Close 关闭存储连接
	Close() error
}