package com.cl.util;

import org.springframework.context.annotation.Bean;
import org.springframework.scheduling.annotation.Scheduled;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

// 使用ConcurrentHashMap作为临时存储

public class LoginAttemptService {
    private static final int MAX_ATTEMPTS = 5;
    private static final long LOCK_TIME_MS = 5 * 60 * 1000; // 5分钟
    
    private final Map<String, LoginAttempt> attemptsCache = new ConcurrentHashMap<>();

    /**
     * 是否被锁了
     */
    public boolean isLocked(String phone) {
        LoginAttempt attempt = attemptsCache.get(phone);
        return attempt != null 
               && attempt.getLockCount() >= MAX_ATTEMPTS
               && System.currentTimeMillis() < attempt.getLastAttemptTime() + LOCK_TIME_MS;
    }

    /**
     * 添加密码错误记录
     */
    public void recordFailedAttempt(String phone) {
        attemptsCache.compute(phone, (k, v) -> {
            long now = System.currentTimeMillis();
            if (v == null) {
                return new LoginAttempt(1, now);
            }
            
            // 如果上次尝试超过锁定时间，则重置
            if (now - v.getLastAttemptTime() > LOCK_TIME_MS) {
                return new LoginAttempt(1, now);
            }
            
            return new LoginAttempt(v.getLockCount() + 1, now);
        });
    }

    /**
     * 清除
     */
    public void clearAttempts(String phone) {
        attemptsCache.remove(phone);
    }
    
    @Scheduled(fixedRate = 60 * 60 * 1000) // 每小时清理一次过期记录
    public void cleanupExpiredAttempts() {
        long now = System.currentTimeMillis();
        attemptsCache.entrySet().removeIf(entry -> 
            now - entry.getValue().getLastAttemptTime() > LOCK_TIME_MS * 2
        );
    }
}