package cache

import (
	"context"
	"errors"
	"fmt"
	"strconv"
	"time"

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

const (
	UserSmsCodeTTL        = 10 * time.Minute
	UserSendSmsLimitTTL   = 24 * time.Hour
	UserSendSmsLimitCount = 5
)

const (
	UserSmsCodePrefix = "user:sms_code:%s"
	UserSendSmsLimit  = "user:send_sms_limit:%s"
)

func GetUserSmsCode(ctx context.Context, phone string) (string, error) {
	code, err := RedisClient.Get(ctx, fmt.Sprintf(UserSmsCodePrefix, phone)).Result()
	if err != nil {
		if err == redis.Nil {
			return "", nil
		}
		return "", err
	}
	return code, nil
}

func SetUserSmsCode(ctx context.Context, phone, code string) error {
	return RedisClient.Set(ctx, fmt.Sprintf(UserSmsCodePrefix, phone), code, UserSmsCodeTTL).Err()
}

func GetUserSendSmsLimit(ctx context.Context, phone string) (int, error) {
	limit, err := RedisClient.Get(ctx, fmt.Sprintf(UserSendSmsLimit, phone)).Result()
	if err != nil {
		if err == redis.Nil {
			return 0, nil
		}
		return 0, err
	}
	return strconv.Atoi(limit)
}

func SetUserSendSmsLimit(ctx context.Context, phone string) error {
	count, err := RedisClient.Incr(ctx, fmt.Sprintf(UserSendSmsLimit, phone)).Result()
	if err != nil {
		return err
	}
	if count > UserSendSmsLimitCount {
		return errors.New("send sms limit")
	}
	return RedisClient.Expire(ctx, fmt.Sprintf(UserSendSmsLimit, phone), UserSendSmsLimitTTL).Err()
}

func DeleteUserSmsCode(ctx context.Context, phone string) error {
	return RedisClient.Del(ctx, fmt.Sprintf(UserSmsCodePrefix, phone)).Err()
}