lbqms.git - Gitblit

			@@ -1,120 +1,112 @@
			package com.ruoyi.web.controller.common;

			import java.util.concurrent.TimeUnit;

			import cn.hutool.captcha.CircleCaptcha;
			import cn.hutool.captcha.ICaptcha;
			import cn.hutool.captcha.LineCaptcha;
			import cn.hutool.captcha.ShearCaptcha;
			import cn.dev33.satoken.annotation.SaIgnore;
			import cn.hutool.captcha.AbstractCaptcha;
			import cn.hutool.captcha.generator.CodeGenerator;
			import cn.hutool.captcha.generator.MathGenerator;
			import cn.hutool.captcha.generator.RandomGenerator;
			import cn.hutool.core.util.IdUtil;
			import cn.hutool.core.util.StrUtil;
			import com.ruoyi.common.core.text.Convert;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.beans.factory.annotation.Value;
			import cn.hutool.core.util.RandomUtil;
			import com.ruoyi.common.constant.CacheConstants;
			import com.ruoyi.common.constant.Constants;
			import com.ruoyi.common.core.domain.R;
			import com.ruoyi.common.enums.CaptchaType;
			import com.ruoyi.common.utils.StringUtils;
			import com.ruoyi.common.utils.redis.RedisUtils;
			import com.ruoyi.common.utils.reflect.ReflectUtils;
			import com.ruoyi.common.utils.spring.SpringUtils;
			import com.ruoyi.framework.config.properties.CaptchaProperties;
			import com.ruoyi.sms.config.properties.SmsProperties;
			import com.ruoyi.sms.core.SmsTemplate;
			import com.ruoyi.sms.entity.SmsResult;
			import com.ruoyi.system.service.ISysConfigService;
			import lombok.RequiredArgsConstructor;
			import lombok.extern.slf4j.Slf4j;
			import org.springframework.expression.Expression;
			import org.springframework.expression.ExpressionParser;
			import org.springframework.expression.spel.standard.SpelExpressionParser;
			import org.springframework.validation.annotation.Validated;
			import org.springframework.web.bind.annotation.GetMapping;
			import org.springframework.web.bind.annotation.RestController;
			import com.ruoyi.common.constant.Constants;
			import com.ruoyi.common.core.domain.AjaxResult;
			import com.ruoyi.common.core.redis.RedisCache;

			import javax.annotation.Resource;
			import jakarta.validation.constraints.NotBlank;
			import java.time.Duration;
			import java.util.HashMap;
			import java.util.Map;

			/**
			* 验证码操作处理
			*
			* @author Lion Li
			*/
			@SaIgnore
			@Slf4j
			@Validated
			@RequiredArgsConstructor
			@RestController
			public class CaptchaController {

			// 圆圈干扰验证码
			@Resource(name = "CircleCaptcha")
			private CircleCaptcha circleCaptcha;
			// 线段干扰的验证码
			@Resource(name = "LineCaptcha")
			private LineCaptcha lineCaptcha;
			// 扭曲干扰验证码
			@Resource(name = "ShearCaptcha")
			private ShearCaptcha shearCaptcha;
			private final CaptchaProperties captchaProperties;
			private final SmsProperties smsProperties;
			private final ISysConfigService configService;

			@Autowired
			private RedisCache redisCache;

			// 验证码类型
			@Value("${captcha.captchaType}")
			private String captchaType;
			// 验证码类别
			@Value("${captcha.captchaCategory}")
			private String captchaCategory;
			// 数字验证码位数
			@Value("${captcha.captchaNumberLength}")
			private int numberLength;
			// 字符验证码长度
			@Value("${captcha.captchaCharLength}")
			private int charLength;
			/**
			* 短信验证码
			*
			* @param phonenumber 用户手机号
			*/
			@GetMapping("/captchaSms")
			public R<Void> smsCaptcha(@NotBlank(message = "{user.phonenumber.not.blank}")
			String phonenumber) {
			if (!smsProperties.getEnabled()) {
			return R.fail("当前系统没有开启短信功能！");
			}
			String key = CacheConstants.CAPTCHA_CODE_KEY + phonenumber;
			String code = RandomUtil.randomNumbers(4);
			RedisUtils.setCacheObject(key, code, Duration.ofMinutes(Constants.CAPTCHA_EXPIRATION));
			// 验证码模板id 自行处理 (查数据库或写死均可)
			String templateId = "";
			Map<String, String> map = new HashMap<>(1);
			map.put("code", code);
			SmsTemplate smsTemplate = SpringUtils.getBean(SmsTemplate.class);
			SmsResult result = smsTemplate.send(phonenumber, templateId, map);
			if (!result.isSuccess()) {
			log.error("验证码短信发送异常 => {}", result);
			return R.fail(result.getMessage());
			}
			return R.ok();
			}

			/**
			* 生成验证码
			*/
			@GetMapping("/captchaImage")
			public AjaxResult getCode() {
			public R<Map<String, Object>> getCode() {
			Map<String, Object> ajax = new HashMap<>();
			boolean captchaEnabled = configService.selectCaptchaEnabled();
			ajax.put("captchaEnabled", captchaEnabled);
			if (!captchaEnabled) {
			return R.ok(ajax);
			}
			// 保存验证码信息
			String uuid = IdUtil.simpleUUID();
			String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid;
			String capStr = null, code = null;
			String verifyKey = CacheConstants.CAPTCHA_CODE_KEY + uuid;
			// 生成验证码
			CodeGenerator codeGenerator;
			if ("math".equals(captchaType)) {
			codeGenerator = new MathGenerator(numberLength);
			} else if ("char".equals(captchaType)) {
			codeGenerator = new RandomGenerator(charLength);
			} else {
			throw new IllegalArgumentException("验证码类型异常");
			CaptchaType captchaType = captchaProperties.getType();
			boolean isMath = CaptchaType.MATH == captchaType;
			Integer length = isMath ? captchaProperties.getNumberLength() : captchaProperties.getCharLength();
			CodeGenerator codeGenerator = ReflectUtils.newInstance(captchaType.getClazz(), length);
			AbstractCaptcha captcha = SpringUtils.getBean(captchaProperties.getCategory().getClazz());
			captcha.setGenerator(codeGenerator);
			captcha.createCode();
			String code = captcha.getCode();
			if (isMath) {
			ExpressionParser parser = new SpelExpressionParser();
			Expression exp = parser.parseExpression(StringUtils.remove(code, "="));
			code = exp.getValue(String.class);
			}
			if ("line".equals(captchaCategory)) {
			lineCaptcha.setGenerator(codeGenerator);
			capStr = lineCaptcha.getCode();
			} else if ("circle".equals(captchaCategory)) {
			circleCaptcha.setGenerator(codeGenerator);
			capStr = circleCaptcha.getCode();
			} else if ("shear".equals(captchaCategory)) {
			shearCaptcha.setGenerator(codeGenerator);
			capStr = shearCaptcha.getCode();
			} else {
			throw new IllegalArgumentException("验证码类别异常");
			}
			if ("math".equals(captchaType)) {
			code = getCodeResult(capStr);
			} else if ("char".equals(captchaType)) {
			code = capStr;
			}

			redisCache.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);

			circleCaptcha.createImage(capStr);
			AjaxResult ajax = AjaxResult.success();
			RedisUtils.setCacheObject(verifyKey, code, Duration.ofMinutes(Constants.CAPTCHA_EXPIRATION));
			ajax.put("uuid", uuid);
			ajax.put("img", circleCaptcha.getImageBase64());
			return ajax;
			}

			private String getCodeResult(String capStr) {
			int a = Convert.toInt(StrUtil.sub(capStr, 0, numberLength).trim());
			char operator = capStr.charAt(numberLength);
			int b = Convert.toInt(StrUtil.sub(capStr, numberLength + 1, numberLength + 1 + numberLength).trim());
			switch (operator) {
			case '*':
			return a * b + "";
			case '+':
			return a + b + "";
			case '-':
			return a - b + "";
			default:
			return "";
			}
			ajax.put("img", captcha.getImageBase64());
			return R.ok(ajax);
			}

			}