add 增加验证码开关

朱杰

2021-05-25 5dc873209e5b6ec2c4289e928831fad6888cfb98

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java

@@ -5,7 +5,6 @@
import cn.hutool.captcha.LineCaptcha;
import cn.hutool.captcha.ShearCaptcha;
import cn.hutool.captcha.generator.CodeGenerator;
import cn.hutool.captcha.generator.MathGenerator;
import cn.hutool.captcha.generator.RandomGenerator;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.IdUtil;
@@ -13,12 +12,15 @@
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.framework.captcha.UnsignedMathGenerator;
import com.ruoyi.framework.config.properties.CaptchaProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
@@ -42,24 +44,20 @@
   @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;
   @Autowired
   private CaptchaProperties captchaProperties;

   /**
    * 生成验证码
    */
   @GetMapping("/captchaImage")
   public AjaxResult getCode() {
      Map<String, Object> ajax = new HashMap<>();
      Boolean enabled = captchaProperties.getEnabled();
      ajax.put("enabled", enabled);
      if (!enabled) {
         return AjaxResult.success(ajax);
      }
      // 保存验证码信息
      String uuid = IdUtil.simpleUUID();
      String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid;
@@ -67,17 +65,17 @@
      // 生成验证码
      CodeGenerator codeGenerator;
      AbstractCaptcha captcha;
      switch (captchaType) {
      switch (captchaProperties.getType()) {
         case "math":
            codeGenerator = new MathGenerator(numberLength);
            codeGenerator = new UnsignedMathGenerator(captchaProperties.getNumberLength());
            break;
         case "char":
            codeGenerator = new RandomGenerator(charLength);
            codeGenerator = new RandomGenerator(captchaProperties.getCharLength());
            break;
         default:
            throw new IllegalArgumentException("验证码类型异常");
      }
      switch (captchaCategory) {
      switch (captchaProperties.getCategory()) {
         case "line":
            captcha = lineCaptcha;
            break;
@@ -92,19 +90,19 @@
      }
      captcha.setGenerator(codeGenerator);
      captcha.createCode();
      if ("math".equals(captchaType)) {
      if ("math".equals(captchaProperties.getType())) {
         code = getCodeResult(captcha.getCode());
      } else if ("char".equals(captchaType)) {
      } else if ("char".equals(captchaProperties.getType())) {
         code = captcha.getCode();
      }
      redisCache.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);
      AjaxResult ajax = AjaxResult.success();
      ajax.put("uuid", uuid);
      ajax.put("img", captcha.getImageBase64());
      return ajax;
      return AjaxResult.success(ajax);
   }

   private String getCodeResult(String capStr) {
      int numberLength = captchaProperties.getNumberLength();
      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());