ratelimit

type Config struct {
	// RequestsPerMinute defines the maximum requests allowed per minute
	RequestsPerMinute int

	// BurstSize defines the maximum burst of requests allowed
	BurstSize int

	// TimeWindow defines the time window for rate limiting
	TimeWindow time.Duration

	// PerUser enables per-user rate limiting
	PerUser bool

	// PerEndpoint enables per-endpoint rate limiting
	PerEndpoint bool

	// EndpointLimits defines custom limits for specific endpoints
	EndpointLimits map[string]int

	// Storage defines the storage backend type
	StorageType string

	// RedisURL is the Redis connection URL (if using Redis storage)
	RedisURL string
}

type EndpointStat struct {
	Endpoint string
	Count    uint64
}

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

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

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

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

type Result struct {
	// Allowed indicates if the request is allowed
	Allowed bool

	// Limit is the maximum number of requests allowed
	Limit int64

	// Remaining is the number of requests remaining
	Remaining int64

	// ResetAt is when the rate limit resets
	ResetAt time.Time

	// RetryAfter is the duration to wait before retrying (if blocked)
	RetryAfter time.Duration
}

type Stats struct {
	TotalRequests    uint64
	AllowedRequests  uint64
	BlockedRequests  uint64
	EndpointRequests map[string]uint64
	UserRequests     map[string]uint64
	LastReset        time.Time
}

type Storage interface {
	// Get retrieves the current count and reset time for a key
	Get(ctx context.Context, key string) (count int64, resetAt time.Time, err error)

	// Increment increments the counter for a key and returns the new count
	Increment(ctx context.Context, key string, window time.Duration) (count int64, resetAt time.Time, err error)

	// Reset resets the counter for a key
	Reset(ctx context.Context, key string) error

	// Close closes the storage backend
	Close() error
}

type UserStat struct {
	User  string
	Count uint64
}