ratelimit

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

type DatabaseStorageConfig struct {
	// CleanupInterval specifies how often to remove expired entries from the database
	// Defaults to 1 minute if not specified
	CleanupInterval time.Duration `json:"cleanup_interval" toml:"cleanup_interval"`
}

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

type MemoryStorageConfig struct {
	// CleanupInterval specifies how often to remove expired entries
	// Defaults to 1 minute if not specified
	CleanupInterval time.Duration `json:"cleanup_interval" toml:"cleanup_interval"`
}

type RateLimit struct {
	bun.BaseModel `bun:"table:rate_limits"`

	Key       string    `json:"key" bun:"column:key,pk"`
	Count     int       `json:"count" bun:"column:count"`
	ExpiresAt time.Time `json:"expires_at" bun:"column:expires_at"`
}

type RateLimitCheckRequest struct {
	ClientIP   string
	Path       string
	HTTPMethod string
}

type RateLimitCheckResponse struct {
	// Allowed indicates whether the request should be allowed
	Allowed bool
	// Limit is the maximum number of requests allowed
	Limit int
	// Window is the time window for the rate limit in seconds
	Window int
	// RetryAfter is the number of seconds to wait before retrying (only set if Allowed is false)
	RetryAfter int
}

type RateLimitEntry struct {
	Count     int
	FirstReq  time.Time
	LastReset time.Time
}

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

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

type RateLimitPluginConfig struct {
	Enabled bool `json:"enabled" toml:"enabled"`

	// Time window for the rate limit
	Window time.Duration `json:"window" toml:"window"`

	// Max number of requests allowed within the window
	Max int `json:"max" toml:"max"`

	// Optional override for the storage namespace prefix
	// Other plugins may also use their own prefixes when accessing secondary storage
	Prefix string `json:"prefix,omitempty" toml:"prefix"`

	// Custom rules for specific paths or methods
	CustomRules map[string]RateLimitRule `json:"custom_rules" toml:"custom_rules"`

	// Provider specifies which rate limit backend to use
	// Options: "memory", "redis", "database"
	// If not specified or if the selected provider is unavailable, defaults to "memory"
	//
	// Note: rate limit plugin can use either its own provider (if specified) or the
	// secondary-storage plugin if available. If secondary-storage plugin is configured
	// and enabled, that takes precedence over the provider selection here.
	Provider RateLimitProviderType `json:"provider" toml:"provider"`

	// Memory contains configuration options for in-memory storage
	Memory *MemoryStorageConfig `json:"memory,omitempty" toml:"memory"`

	// Database contains configuration options for database storage
	Database *DatabaseStorageConfig `json:"database,omitempty" toml:"database"`
}

type RateLimitProvider interface {
	// GetName returns the name of the provider
	GetName() string
	// CheckAndIncrement checks if a request is allowed and increments the counter if so
	// key is the fully-qualified key (with prefix already included)
	// window is the time window for expiration
	// maxRequests is the maximum number of requests allowed in the window
	// Returns: (allowed bool, currentCount int, resetTime time.Time, error)
	CheckAndIncrement(ctx context.Context, key string, window time.Duration, maxRequests int) (bool, int, time.Time, error)
	// Close closes any resources held by the provider
	Close() error
}

type RateLimitProviderType string

type RateLimitRepository interface {
	// GetByKey retrieves a rate limit record by its key
	GetByKey(ctx context.Context, key string) (*RateLimit, error)

	// UpdateOrCreate updates count for an existing record, or creates a new one
	UpdateOrCreate(ctx context.Context, key string, window time.Duration) (*RateLimit, error)

	// CleanupExpired removes expired rate limit records
	CleanupExpired(ctx context.Context, now time.Time) error
}

type RateLimitRule struct {
	// Disable rate limiting for this endpoint entirely
	Disabled bool `json:"disabled" toml:"disabled"`

	// Time window for the rate limit
	Window time.Duration `json:"window" toml:"window"`

	// Max number of requests allowed within the window
	Max int `json:"max" toml:"max"`

	// Optional override for the storage namespace
	Prefix string `json:"prefix,omitempty" toml:"prefix"`
}

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

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