219 lines
6.7 KiB
Java
219 lines
6.7 KiB
Java
package com.imyeyu.api.util;
|
|
|
|
import com.imyeyu.java.TimiJava;
|
|
import com.imyeyu.java.bean.timi.TimiCode;
|
|
import com.imyeyu.java.bean.timi.TimiException;
|
|
import com.imyeyu.spring.TimiSpring;
|
|
import com.imyeyu.spring.util.Redis;
|
|
import com.imyeyu.utils.Calc;
|
|
import com.imyeyu.utils.Encoder;
|
|
import com.imyeyu.utils.Time;
|
|
import lombok.RequiredArgsConstructor;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import javax.imageio.ImageIO;
|
|
import java.awt.AlphaComposite;
|
|
import java.awt.Color;
|
|
import java.awt.Font;
|
|
import java.awt.GradientPaint;
|
|
import java.awt.Graphics2D;
|
|
import java.awt.geom.AffineTransform;
|
|
import java.awt.geom.QuadCurve2D;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
import java.util.UUID;
|
|
|
|
/**
|
|
* 验证码绘制
|
|
*
|
|
* @author 夜雨
|
|
* @version 2021-05-20 16:48
|
|
*/
|
|
@Slf4j
|
|
@Component
|
|
@RequiredArgsConstructor
|
|
public class CaptchaManager {
|
|
|
|
/** 会话频率限制键，插值会话 ID */
|
|
private static final String LOCK_KEY = "CAPTCHA:LOCK:%s";
|
|
|
|
/** 会话值缓存键，插值唯一 ID */
|
|
private static final String CACHE_KEY = "CAPTCHA:%s";
|
|
|
|
/** 绘制字符，移除 O,o,I,l */
|
|
private static final char[] CHARS = {
|
|
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
|
|
'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'm', 'n', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
|
|
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'
|
|
};
|
|
|
|
@Value("${spring.profiles.active}")
|
|
private String env;
|
|
|
|
private final Redis<String, String> redisCaptcha;
|
|
|
|
public Result generate(int width, int height) {
|
|
return generate(width, height, 4);
|
|
}
|
|
|
|
/**
|
|
* 生成并缓存验证码
|
|
*
|
|
* @param width 宽度
|
|
* @param height 高度
|
|
* @param length 长度
|
|
* @return 图片流
|
|
*/
|
|
public Result generate(int width, int height, int length) {
|
|
long lockTime = 1000;
|
|
// 加锁
|
|
String lockKey = LOCK_KEY.formatted(TimiSpring.getSession().getId());
|
|
Long lockAt = TimiSpring.getSessionAttr(lockKey, Long.class);
|
|
|
|
TimiSpring.setSessionAttr(lockKey, Time.now());
|
|
if (lockAt != null) {
|
|
long diff = Time.now() - lockAt;
|
|
if (diff < lockTime) {
|
|
// 限制频率
|
|
synchronized (this) {
|
|
try {
|
|
wait(lockTime - diff);
|
|
} catch (InterruptedException e) {
|
|
log.error("lock error", e);
|
|
throw new RuntimeException(e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
int fontWidth = (int) (width * .9 / 4);
|
|
int fontSize = (int) (height * .7);
|
|
int yOffset = (int) (height * .4);
|
|
StringBuilder value = new StringBuilder();
|
|
|
|
// 图片流
|
|
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g = (Graphics2D) image.getGraphics();
|
|
// 渐变背景
|
|
GradientPaint gradient = new GradientPaint(
|
|
0, 0,
|
|
new Color(Calc.random(200, 255), Calc.random(200, 255), Calc.random(200, 255)),
|
|
width,
|
|
height,
|
|
new Color(Calc.random(200, 255), Calc.random(200, 255), Calc.random(200, 255))
|
|
);
|
|
g.setPaint(gradient);
|
|
g.fillRect(0, 0, width, height);
|
|
|
|
// 半透明干扰层
|
|
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.4F));
|
|
for (int i = 0; i < 15; i++) {
|
|
g.setColor(new Color(Calc.random(0, 150), Calc.random(0, 150), Calc.random(0, 150)));
|
|
g.drawString(String.valueOf(CHARS[Calc.random(0, CHARS.length - 1)]), Calc.random(0, width), Calc.random(0, height));
|
|
}
|
|
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
|
|
|
|
// 绘制验证码字符
|
|
g.setFont(new Font("Fixedsys", Font.BOLD, Calc.random(fontSize - 2, fontSize + 2)));
|
|
for (int i = 0; i < length; i++) {
|
|
char c = CHARS[Calc.random(0, CHARS.length - 1)];
|
|
value.append(c);
|
|
// 颜色
|
|
g.setColor(new Color(Calc.random(50, 150), Calc.random(50, 150), Calc.random(50, 150)));
|
|
// 位置
|
|
int x = fontWidth * i + Calc.random(0, 10);
|
|
int y = height - yOffset + Calc.random(-4, 4);
|
|
// 随机角度
|
|
AffineTransform original = g.getTransform();
|
|
AffineTransform rotate = AffineTransform.getRotateInstance(Math.toRadians(Calc.random(0, 15)), x + fontSize / 2D, y - fontSize / 2D);
|
|
g.setTransform(rotate);
|
|
g.drawString(String.valueOf(c), x, y);
|
|
g.setTransform(original);
|
|
}
|
|
// 干扰线
|
|
for (int i = 0; i < height / 2; i++) {
|
|
int x1 = Calc.random(0, width);
|
|
int y1 = Calc.random(0, height);
|
|
int x2 = x1 + Calc.random(-20, 20);
|
|
int y2 = y1 + Calc.random(-20, 20);
|
|
g.setColor(new Color(Calc.random(100, 150), Calc.random(100, 150), Calc.random(100, 150)));
|
|
if (Calc.randomBoolean()) {
|
|
g.drawLine(x1, y1, x2, y2);
|
|
} else {
|
|
QuadCurve2D curve = new QuadCurve2D.Float();
|
|
double ctrlX = (x1 + x2) / 2D + Calc.random(-15, 15);
|
|
double ctrlY = (y1 + y2) / 2D + Calc.random(-15, 15);
|
|
curve.setCurve(x1, y1, ctrlX, ctrlY, x2, y2);
|
|
g.draw(curve);
|
|
}
|
|
}
|
|
|
|
try {
|
|
String uniqueId = UUID.randomUUID().toString();
|
|
// 写入缓存
|
|
redisCaptcha.set(CACHE_KEY.formatted(uniqueId), value.toString(), Time.parseToMS("3m"));
|
|
|
|
// 返回
|
|
ByteArrayOutputStream os = new ByteArrayOutputStream();
|
|
ImageIO.write(image, "png", os);
|
|
return new Result(uniqueId, "data:image/png;base64," + Encoder.base64(os.toByteArray()));
|
|
} catch (IOException e) {
|
|
throw new TimiException(TimiCode.ERROR);
|
|
}
|
|
}
|
|
|
|
/** @return 错误回调图像 */
|
|
public BufferedImage error(TimiCode code) {
|
|
final int width = 74, height = 24;
|
|
BufferedImage img = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
|
|
Graphics2D g = (Graphics2D) img.getGraphics();
|
|
g.setColor(Color.RED);
|
|
// 文本
|
|
g.setFont(new Font("Fixedsys", Font.BOLD, 13));
|
|
g.drawString("ERR." + code.getValue(), 2, 18);
|
|
g.setColor(Color.DARK_GRAY);
|
|
return img;
|
|
}
|
|
|
|
/**
|
|
* 验证
|
|
*
|
|
* <ul>
|
|
* <li>通过验证不抛出任何异常也不返回任何内容，否则抛出相应异常</li>
|
|
* <li>会验证非空和期限</li>
|
|
* </ul>
|
|
*
|
|
* @param uniqueId 唯一识别 ID
|
|
* @param value 提交的验证码
|
|
* @throws TimiException 验证异常
|
|
*/
|
|
public void verify(String uniqueId, String value) {
|
|
if (TimiJava.isEmpty(value)) {
|
|
throw new TimiException(TimiCode.ARG_MISS, "not found value");
|
|
}
|
|
if (env.startsWith("dev")) {
|
|
return;
|
|
}
|
|
String cacheKey = CACHE_KEY.formatted(uniqueId);
|
|
String cacheValue = redisCaptcha.get(cacheKey);
|
|
if (TimiJava.isEmpty(cacheValue)) {
|
|
throw new TimiException(TimiCode.ARG_EXPIRED, "captcha.expire");
|
|
}
|
|
if (!value.trim().equalsIgnoreCase(cacheValue)) {
|
|
throw new TimiException(TimiCode.RESULT_BAD, "captcha.error");
|
|
}
|
|
// 清除缓存
|
|
redisCaptcha.destroy(cacheKey);
|
|
}
|
|
|
|
/**
|
|
* @author 夜雨
|
|
* @since 2025-10-30 14:48
|
|
*/
|
|
public record Result(String id, String data) {
|
|
}
|
|
}
|