package com.nq.utils.captcha;

import com.nq.utils.redis.RedisShardedPoolUtils;
import org.apache.commons.lang3.StringUtils;

public final class CaptchaUtil {

    public static final String REDIS_PREFIX = "Captcha:";
    /** 图形验证码有效期（秒） */
    public static final int CAPTCHA_TTL_SECONDS = 300;

    private CaptchaUtil() {
    }

    public static void save(String token, String code) {
        if (StringUtils.isBlank(token) || StringUtils.isBlank(code)) {
            return;
        }
        RedisShardedPoolUtils.setEx(REDIS_PREFIX + token.trim(), code, CAPTCHA_TTL_SECONDS);
    }

    public static String get(String token) {
        if (StringUtils.isBlank(token)) {
            return null;
        }
        return RedisShardedPoolUtils.get(REDIS_PREFIX + token.trim());
    }

    /** 校验失败不删除，避免输错一次就必须刷新；校验成功才删除防重复使用 */
    public static boolean verifyAndRemove(String token, String inputCode) {
        if (StringUtils.isBlank(token) || StringUtils.isBlank(inputCode)) {
            return false;
        }
        String key = REDIS_PREFIX + token.trim();
        String original = RedisShardedPoolUtils.get(key);
        if (StringUtils.isBlank(original)) {
            return false;
        }
        if (!inputCode.trim().equalsIgnoreCase(original.trim())) {
            return false;
        }
        RedisShardedPoolUtils.del(key);
        return true;
    }
}