package cn.iocoder.yudao.module.system.framework.captcha.core; import com.anji.captcha.model.common.RepCodeEnum; import com.anji.captcha.model.common.ResponseModel; import com.anji.captcha.model.vo.CaptchaVO; import com.anji.captcha.service.impl.AbstractCaptchaService; import com.anji.captcha.service.impl.CaptchaServiceFactory; import com.anji.captcha.util.AESUtil; import com.anji.captcha.util.ImageUtils; import com.anji.captcha.util.RandomUtils; import cn.hutool.core.util.RandomUtil; import org.apache.commons.lang3.Strings; import java.awt.*; import java.awt.geom.AffineTransform; import java.awt.image.BufferedImage; import java.util.Properties; /** * 图片文字验证码 * * @author Tsui * @since 2025/7/23 20:44 */ public class PictureWordCaptchaServiceImpl extends AbstractCaptchaService { /** * 验证码的基础字符 */ private static final String CHARACTERS = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"; /** * 验证码长度 */ private static final Integer LENGTH = 4; private static final int WIDTH = 120; private static final int HEIGHT = 40; private static final int LINES = 10; @Override public void init(Properties config) { super.init(config); } @Override public void destroy(Properties config) { logger.info("start-clear-history-data-{}", captchaType()); } @Override public String captchaType() { return "pictureWord"; } @Override public ResponseModel get(CaptchaVO captchaVO) { String text = generateRandomText(LENGTH); CaptchaVO imageData = getImageData(text); // pointJson 不传到前端，只做后端校验，测试时放开 // imageData.setPointJson(text); return ResponseModel.successData(imageData); } @Override public ResponseModel check(CaptchaVO captchaVO) { ResponseModel r = super.check(captchaVO); if (!validatedReq(r)) { return r; } // 取出验证码 String codeKey = String.format(REDIS_CAPTCHA_KEY, captchaVO.getToken()); if (!CaptchaServiceFactory.getCache(cacheType).exists(codeKey)) { return ResponseModel.errorMsg(RepCodeEnum.API_CAPTCHA_INVALID); } // 正确的验证码 String codeValue = CaptchaServiceFactory.getCache(cacheType).get(codeKey); String code = getCodeByCodeValue(codeValue); String secretKey = getSecretKeyByCodeValue(codeValue); // 验证码只用一次，即刻失效 CaptchaServiceFactory.getCache(cacheType).delete(codeKey); // 用户输入的验证码(CaptchaVO 中 没有预留字段，暂时用 pointJson 无需加解密) String userCode = captchaVO.getPointJson(); if (!Strings.CI.equals(code, userCode)) { afterValidateFail(captchaVO); return ResponseModel.errorMsg(RepCodeEnum.API_CAPTCHA_COORDINATE_ERROR); } // 校验成功，将信息存入缓存 String value; try { value = AESUtil.aesEncrypt(captchaVO.getToken().concat("---").concat(userCode), secretKey); } catch (Exception e) { logger.error("AES 加密失败", e); afterValidateFail(captchaVO); return ResponseModel.errorMsg(e.getMessage()); } String secondKey = String.format(REDIS_SECOND_CAPTCHA_KEY, value); CaptchaServiceFactory.getCache(cacheType).set(secondKey, captchaVO.getToken(), EXPIRESIN_THREE); captchaVO.setResult(true); captchaVO.resetClientFlag(); return ResponseModel.successData(captchaVO); } @Override public ResponseModel verification(CaptchaVO captchaVO) { ResponseModel r = super.verification(captchaVO); if (!validatedReq(r)) { return r; } try { String codeKey = String.format(REDIS_SECOND_CAPTCHA_KEY, captchaVO.getCaptchaVerification()); if (!CaptchaServiceFactory.getCache(cacheType).exists(codeKey)) { return ResponseModel.errorMsg(RepCodeEnum.API_CAPTCHA_INVALID); } // 二次校验取值后，即刻失效 CaptchaServiceFactory.getCache(cacheType).delete(codeKey); } catch (Exception e) { logger.error("验证码解析失败", e); return ResponseModel.errorMsg(e.getMessage()); } return ResponseModel.success(); } private CaptchaVO getImageData(String text) { CaptchaVO dataVO = new CaptchaVO(); BufferedImage image = new BufferedImage(WIDTH, HEIGHT, BufferedImage.TYPE_INT_RGB); Graphics2D g = image.createGraphics(); // 设置背景色 g.setColor(getRandomColor(200, 250)); g.fillRect(0, 0, WIDTH, HEIGHT); // 绘制干扰线 for (int i = 0; i < LINES; i++) { g.setColor(getRandomColor(100, 200)); int x1 = RandomUtil.randomInt(WIDTH); int y1 = RandomUtil.randomInt(HEIGHT); int x2 = RandomUtil.randomInt(WIDTH); int y2 = RandomUtil.randomInt(HEIGHT); g.drawLine(x1, y1, x2, y2); } // 设置字体 g.setFont(new Font("Arial", Font.BOLD, 24)); // 绘制验证码文本 for (int i = 0; i < text.length(); i++) { g.setColor(getRandomColor(20, 130)); // 文字旋转 AffineTransform affineTransform = new AffineTransform(); int x = 20 + i * 20; int y = 24 + RandomUtil.randomInt(8); // 旋转范围 -45 ~ 45 affineTransform.setToRotation(Math.toRadians(RandomUtil.randomInt(-45, 45)), x, y); g.setTransform(affineTransform); g.drawString(text.charAt(i) + "", x, y); } // 添加噪点 for (int i = 0; i < 100; i++) { int x = RandomUtil.randomInt(WIDTH); int y = RandomUtil.randomInt(HEIGHT); image.setRGB(x, y, getRandomColor(0, 255).getRGB()); } g.dispose(); String secretKey = null; if (captchaAesStatus) { secretKey = AESUtil.getKey(); } dataVO.setSecretKey(secretKey); dataVO.setOriginalImageBase64(ImageUtils.getImageToBase64Str(image).replaceAll("\r|\n", "")); dataVO.setToken(RandomUtils.getUUID()); // dataVO.setSecretKey(secretKey); // 将坐标信息存入 redis 中 String codeKey = String.format(REDIS_CAPTCHA_KEY, dataVO.getToken()); CaptchaServiceFactory.getCache(cacheType).set(codeKey, getCodeValue(text, secretKey), EXPIRESIN_SECONDS); return dataVO; } private String getCodeValue(String text, String secretKey) { return text + "," + secretKey; } private String getCodeByCodeValue(String codeValue) { return codeValue.split(",")[0]; } private String getSecretKeyByCodeValue(String codeValue) { return codeValue.split(",")[1]; } private Color getRandomColor(int min, int max) { int minVal = Math.min(min, max); int maxVal = Math.max(min, max); int r = RandomUtil.randomInt(minVal, maxVal); int g = RandomUtil.randomInt(minVal, maxVal); int b = RandomUtil.randomInt(minVal, maxVal); return new Color(r, g, b); } /** * 生成指定长度的随机字符串 * * @param length 长度 * @return {@link String} */ public static String generateRandomText(int length) { return RandomUtil.randomString(CHARACTERS, length); } }