package auth

import (
	"context"
	"fmt"
	"time"

	"github.com/redis/go-redis/v9"
)

// RedisLimiter — счётчик попыток с фиксированным окном в Redis.
type RedisLimiter struct{ rdb redis.Cmdable }

// NewRedisLimiter создаёт лимитер.
func NewRedisLimiter(rdb redis.Cmdable) *RedisLimiter { return &RedisLimiter{rdb: rdb} }

// Первый INCR ставит срок окна; ответ — {счётчик, оставшийся TTL в мс}.
var incrWindow = redis.NewScript(`
local n = redis.call('INCR', KEYS[1])
if n == 1 then redis.call('PEXPIRE', KEYS[1], ARGV[1]) end
return {n, redis.call('PTTL', KEYS[1])}
`)

// Allow реализует Limiter.
func (l *RedisLimiter) Allow(ctx context.Context, key string, limit int, window time.Duration) (bool, int, error) {
	res, err := incrWindow.Run(ctx, l.rdb, []string{"rl:" + key}, window.Milliseconds()).Int64Slice()
	if err != nil {
		return false, 0, fmt.Errorf("auth: rate limit: %w", err)
	}
	if len(res) != 2 {
		return false, 0, fmt.Errorf("auth: rate limit: unexpected reply %v", res)
	}
	if res[0] <= int64(limit) {
		return true, 0, nil
	}
	retry := int((res[1] + 999) / 1000)
	if retry < 1 {
		retry = 1
	}
	return false, retry, nil
}