import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 短信衛(wèi)士 - 專(zhuān)治各種手速過(guò)快
 */
public class SmsGuard {
    
    // 使用Guava Cache存儲(chǔ)訪問(wèn)頻率
    private static final Cache<String, AtomicInteger> IP_CACHE = 
        CacheBuilder.newBuilder()
            .expireAfterWrite(1, TimeUnit.HOURS)
            .build();
    
    private static final Cache<String, AtomicInteger> PHONE_CACHE = 
        CacheBuilder.newBuilder()
            .expireAfterWrite(1, TimeUnit.HOURS)
            .build();
    
    /**
     * 檢查這個(gè)IP是不是在開(kāi)掛
     * @param ip 客戶(hù)端IP
     * @param maxAttempts 最大嘗試次數(shù)（比如1小時(shí)10次）
     * @return true=正常用戶(hù)，false=疑似黑客
     */
    public static boolean isIpAllowed(String ip, int maxAttempts) {
        try {
            AtomicInteger counter = IP_CACHE.get(ip, () -> new AtomicInteger(0));
            int attempts = counter.incrementAndGet();
            
            if (attempts > maxAttempts) {
                System.out.println("檢測(cè)到IP " + ip + " 疑似開(kāi)掛，已攔截！");
                return false;
            }
            return true;
        } catch (Exception e) {
            return false; // 出錯(cuò)時(shí)保守一點(diǎn)，拒絕訪問(wèn)
        }
    }
    
    /**
     * 檢查這個(gè)手機(jī)號(hào)是不是在刷驗(yàn)證碼
     * @param phone 手機(jī)號(hào)
     * @param maxSmsPerHour 每小時(shí)最多發(fā)幾條
     * @return true=可以發(fā)，false=發(fā)太多了
     */
    public static boolean isPhoneAllowed(String phone, int maxSmsPerHour) {
        try {
            AtomicInteger counter = PHONE_CACHE.get(phone, () -> new AtomicInteger(0));
            int sentCount = counter.incrementAndGet();
            
            if (sentCount > maxSmsPerHour) {
                System.out.println("手機(jī)號(hào) " + phone + " 今天已經(jīng)收到" + sentCount + "條驗(yàn)證碼，讓它歇會(huì)兒吧");
                return false;
            }
            return true;
        } catch (Exception e) {
            return false;
        }
    }
}

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.util.Random;

/**
 * 驗(yàn)證碼生成器 - 專(zhuān)治眼瞎的機(jī)器人
 */
public class CaptchaGenerator {
    
    /**
     * 生成能讓機(jī)器人懷疑人生的驗(yàn)證碼
     * @return [0]=圖片Base64, [1]=驗(yàn)證碼答案
     */
    public static String[] generateCaptcha() {
        int width = 120;
        int height = 40;
        
        // 創(chuàng)建一張讓機(jī)器人哭泣的圖片
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = image.createGraphics();
        
        // 設(shè)置背景色（隨機(jī)淺色）
        g.setColor(getRandomLightColor());
        g.fillRect(0, 0, width, height);
        
        // 畫(huà)干擾線（讓機(jī)器人眼花繚亂）
        g.setColor(Color.BLACK);
        Random random = new Random();
        for (int i = 0; i < 10; i++) {
            int x1 = random.nextInt(width);
            int y1 = random.nextInt(height);
            int x2 = random.nextInt(width);
            int y2 = random.nextInt(height);
            g.drawLine(x1, y1, x2, y2);
        }
        
        // 生成隨機(jī)驗(yàn)證碼（避開(kāi)容易混淆的字符）
        String chars = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789";
        StringBuilder captchaText = new StringBuilder();
        for (int i = 0; i < 4; i++) {
            char c = chars.charAt(random.nextInt(chars.length()));
            captchaText.append(c);
            
            // 扭曲、旋轉(zhuǎn)、變色 - 三連擊！
            g.setFont(new Font("Arial", Font.BOLD | Font.ITALIC, 30 + random.nextInt(5)));
            g.setColor(getRandomDarkColor());
            
            // 輕微旋轉(zhuǎn)字符
            double theta = random.nextDouble() * 0.5 - 0.25;
            g.rotate(theta, 20 + i * 25, 25);
            g.drawString(String.valueOf(c), 20 + i * 25, 25);
            g.rotate(-theta, 20 + i * 25, 25);
        }
        
        g.dispose();
        
        try {
            // 轉(zhuǎn)換為Base64
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageIO.write(image, "png", baos);
            String base64Image = java.util.Base64.getEncoder().encodeToString(baos.toByteArray());
            
            return new String[]{"data:image/png;base64," + base64Image, captchaText.toString()};
        } catch (Exception e) {
            throw new RuntimeException("驗(yàn)證碼生成失敗", e);
        }
    }
    
    private static Color getRandomLightColor() {
        Random random = new Random();
        return new Color(200 + random.nextInt(55), 
                        200 + random.nextInt(55), 
                        200 + random.nextInt(55));
    }
    
    private static Color getRandomDarkColor() {
        Random random = new Random();
        return new Color(random.nextInt(150), 
                        random.nextInt(150), 
                        random.nextInt(150));
    }
}

import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 滑動(dòng)驗(yàn)證碼 - 專(zhuān)治不會(huì)用鼠標(biāo)的機(jī)器人
 */
public class SlideCaptchaService {
    
    // 存儲(chǔ)驗(yàn)證會(huì)話
    private static final ConcurrentHashMap<String, SlideCaptchaData> SESSIONS = 
        new ConcurrentHashMap<>();
    
    /**
     * 生成滑動(dòng)驗(yàn)證碼挑戰(zhàn)
     */
    public static SlideChallenge generateChallenge() {
        String sessionId = UUID.randomUUID().toString();
        
        // 隨機(jī)生成目標(biāo)位置（這里簡(jiǎn)化了，實(shí)際應(yīng)該有圖片處理）
        int targetX = 100 + new Random().nextInt(200);
        int targetY = 50 + new Random().nextInt(100);
        
        SlideCaptchaData data = new SlideCaptchaData(targetX, targetY);
        SESSIONS.put(sessionId, data);
        
        // 設(shè)置5分鐘過(guò)期
        new Timer().schedule(new TimerTask() {
            @Override
            public void run() {
                SESSIONS.remove(sessionId);
            }
        }, 5 * 60 * 1000);
        
        return new SlideChallenge(sessionId, targetX, targetY);
    }
    
    /**
     * 驗(yàn)證滑動(dòng)結(jié)果
     */
    public static boolean verify(String sessionId, int userX, int userY) {
        SlideCaptchaData data = SESSIONS.get(sessionId);
        if (data == null) {
            return false; // 會(huì)話過(guò)期
        }
        
        // 允許±5像素的誤差（人類(lèi)手抖，機(jī)器人太精確反而可疑）
        boolean isValid = Math.abs(userX - data.targetX) <= 5 && 
                         Math.abs(userY - data.targetY) <= 5;
        
        if (isValid) {
            SESSIONS.remove(sessionId); // 一次性使用
        }
        
        return isValid;
    }
    
    static class SlideCaptchaData {
        int targetX;
        int targetY;
        
        SlideCaptchaData(int targetX, int targetY) {
            this.targetX = targetX;
            this.targetY = targetY;
        }
    }
    
    static class SlideChallenge {
        String sessionId;
        int targetX;
        int targetY;
        
        SlideChallenge(String sessionId, int targetX, int targetY) {
            this.sessionId = sessionId;
            this.targetX = targetX;
            this.targetY = targetY;
        }
    }
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import java.util.concurrent.TimeUnit;

/**
 * 短信發(fā)送服務(wù) - 武裝到牙齒的版本
 */
@Service
public class SmsService {
    
    @Autowired
    private RedisTemplate<String, String> redisTemplate;
    
    @Autowired
    private RiskControlService riskControlService;
    
    /**
     * 發(fā)送驗(yàn)證碼（安全加強(qiáng)版）
     */
    public ApiResponse sendVerificationCode(String phone, String ip, String captchaCode, String sessionId) {
        
        // 1. 檢查IP風(fēng)險(xiǎn)
        if (!riskControlService.checkIpRisk(ip)) {
            return ApiResponse.error("您的網(wǎng)絡(luò)環(huán)境存在風(fēng)險(xiǎn)，請(qǐng)稍后再試");
        }
        
        // 2. 驗(yàn)證圖形驗(yàn)證碼（如果有）
        if (captchaCode != null && !validateCaptcha(sessionId, captchaCode)) {
            return ApiResponse.error("驗(yàn)證碼錯(cuò)誤，請(qǐng)重新輸入");
        }
        
        // 3. 頻率控制：同一手機(jī)號(hào)1分鐘內(nèi)只能發(fā)1次
        String minuteKey = "sms:minute:" + phone;
        if (Boolean.TRUE.equals(redisTemplate.hasKey(minuteKey))) {
            return ApiResponse.error("操作過(guò)于頻繁，請(qǐng)1分鐘后再試");
        }
        
        // 4. 頻率控制：同一手機(jī)號(hào)1小時(shí)內(nèi)最多5次
        String hourKey = "sms:hour:" + phone;
        Long hourCount = redisTemplate.opsForValue().increment(hourKey);
        if (hourCount != null && hourCount == 1) {
            redisTemplate.expire(hourKey, 1, TimeUnit.HOURS);
        }
        if (hourCount != null && hourCount > 5) {
            return ApiResponse.error("今日驗(yàn)證碼發(fā)送次數(shù)已達(dá)上限");
        }
        
        // 5. 生成6位隨機(jī)驗(yàn)證碼
        String code = String.format("%06d", new Random().nextInt(999999));
        
        // 6. 存儲(chǔ)驗(yàn)證碼（5分鐘過(guò)期）
        String codeKey = "sms:code:" + phone;
        redisTemplate.opsForValue().set(codeKey, code, 5, TimeUnit.MINUTES);
        
        // 7. 設(shè)置1分鐘冷卻期
        redisTemplate.opsForValue().set(minuteKey, "1", 1, TimeUnit.MINUTES);
        
        // 8. 記錄發(fā)送日志（用于分析）
        logSmsSent(phone, ip, code);
        
        // 9. 調(diào)用第三方短信服務(wù)（實(shí)際發(fā)送）
        boolean sent = realSendSms(phone, code);
        
        if (sent) {
            // 10. 返回脫敏的手機(jī)號(hào)
            String maskedPhone = phone.substring(0, 3) + "****" + phone.substring(7);
            return ApiResponse.success("驗(yàn)證碼已發(fā)送至" + maskedPhone);
        } else {
            return ApiResponse.error("短信發(fā)送失敗，請(qǐng)稍后重試");
        }
    }
    
    /**
     * 驗(yàn)證短信驗(yàn)證碼
     */
    public boolean verifyCode(String phone, String userCode) {
        String codeKey = "sms:code:" + phone;
        String correctCode = redisTemplate.opsForValue().get(codeKey);
        
        if (correctCode == null) {
            return false; // 驗(yàn)證碼已過(guò)期
        }
        
        // 驗(yàn)證成功后刪除驗(yàn)證碼（防止重復(fù)使用）
        boolean isValid = correctCode.equals(userCode);
        if (isValid) {
            redisTemplate.delete(codeKey);
        }
        
        return isValid;
    }
    
    private void logSmsSent(String phone, String ip, String code) {
        // 這里應(yīng)該記錄到數(shù)據(jù)庫(kù)或日志系統(tǒng)
        System.out.println(String.format(
            "短信發(fā)送日志: phone=%s, ip=%s, code=%s, time=%s",
            phone, ip, code, new java.util.Date()
        ));
    }
    
    private boolean realSendSms(String phone, String code) {
        // 調(diào)用真實(shí)的短信服務(wù)商接口
        // 這里簡(jiǎn)化處理，實(shí)際應(yīng)該用HTTP客戶(hù)端調(diào)用
        try {
            System.out.println(String.format(
                "發(fā)送短信到 %s: 您的驗(yàn)證碼是%s，5分鐘內(nèi)有效，打死也不要告訴別人哦！",
                phone, code
            ));
            return true;
        } catch (Exception e) {
            return false;
        }
    }
}

/**
 * 風(fēng)控服務(wù) - 專(zhuān)治各種不服
 */
@Service
public class RiskControlService {
    
    @Autowired
    private RedisTemplate<String, String> redisTemplate;
    
    /**
     * 綜合風(fēng)險(xiǎn)評(píng)估
     */
    public RiskLevel assessRisk(String phone, String ip, String userAgent) {
        int riskScore = 0;
        
        // 1. IP地址檢查
        if (isSuspiciousIp(ip)) {
            riskScore += 30;
        }
        
        // 2. User-Agent檢查
        if (isSuspiciousUserAgent(userAgent)) {
            riskScore += 20;
        }
        
        // 3. 請(qǐng)求頻率檢查
        if (isHighFrequency(ip)) {
            riskScore += 40;
        }
        
        // 4. 手機(jī)號(hào)歸屬地 vs IP歸屬地
        if (!isLocationConsistent(phone, ip)) {
            riskScore += 20;
        }
        
        // 5. 歷史行為檢查
        if (hasBadHistory(ip)) {
            riskScore += 50;
        }
        
        // 根據(jù)分?jǐn)?shù)返回風(fēng)險(xiǎn)等級(jí)
        if (riskScore >= 80) {
            return RiskLevel.HIGH;
        } else if (riskScore >= 50) {
            return RiskLevel.MEDIUM;
        } else {
            return RiskLevel.LOW;
        }
    }
    
    /**
     * 檢查IP風(fēng)險(xiǎn)
     */
    public boolean checkIpRisk(String ip) {
        String key = "risk:ip:" + ip;
        Long count = redisTemplate.opsForValue().increment(key);
        
        if (count == 1) {
            redisTemplate.expire(key, 1, TimeUnit.HOURS);
        }
        
        // 1小時(shí)內(nèi)超過(guò)50次請(qǐng)求視為風(fēng)險(xiǎn)
        return count == null || count <= 50;
    }
    
    enum RiskLevel {
        LOW,    // 低風(fēng)險(xiǎn)：正常通過(guò)
        MEDIUM, // 中風(fēng)險(xiǎn)：需要額外驗(yàn)證
        HIGH    // 高風(fēng)險(xiǎn)：直接拒絕
    }
    
    // 其他檢測(cè)方法...
}

/**
 * 監(jiān)控服務(wù) - 短信接口的"心電圖"
 */
@Service
public class SmsMonitorService {
    
    // 關(guān)鍵指標(biāo)監(jiān)控
    public void monitorMetrics() {
        // 1. 成功率監(jiān)控
        // 2. 響應(yīng)時(shí)間監(jiān)控
        // 3. 異常請(qǐng)求監(jiān)控
        // 4. 費(fèi)用消耗監(jiān)控
        
        // 發(fā)現(xiàn)異常立即告警
        // - 短信量突增
        // - 成功率突降
        // - 特定IP大量請(qǐng)求
    }
    
    /**
     * 自動(dòng)熔斷機(jī)制
     */
    public void circuitBreaker(String phonePrefix) {
        // 如果某個(gè)號(hào)段異常，自動(dòng)臨時(shí)屏蔽
        // 比如：170號(hào)段被大量攻擊，自動(dòng)限制該號(hào)段
    }
}