| | |
| | | import cn.hutool.captcha.generator.RandomGenerator; |
| | | import cn.hutool.core.convert.Convert; |
| | | import cn.hutool.core.util.IdUtil; |
| | | import cn.hutool.core.util.StrUtil; |
| | | import com.ruoyi.common.utils.StringUtils; |
| | | 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 com.ruoyi.system.service.ISysConfigService; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.web.bind.annotation.GetMapping; |
| | | import org.springframework.web.bind.annotation.RestController; |
| | |
| | | /** |
| | | * 验证码操作处理 |
| | | * |
| | | * @author Lion Li |
| | | * @author ruoyi |
| | | */ |
| | | @RestController |
| | | public class CaptchaController { |
| | |
| | | @Autowired |
| | | private CaptchaProperties captchaProperties; |
| | | |
| | | @Autowired |
| | | private ISysConfigService configService; |
| | | |
| | | /** |
| | | * 生成验证码 |
| | | */ |
| | | @GetMapping("/captchaImage") |
| | | public AjaxResult getCode() { |
| | | Map<String, Object> ajax = new HashMap<>(); |
| | | boolean captchaOnOff = configService.selectCaptchaOnOff(); |
| | | ajax.put("captchaOnOff", captchaOnOff); |
| | | if (!captchaOnOff) { |
| | | return AjaxResult.success(ajax); |
| | | } |
| | | // 保存验证码信息 |
| | | String uuid = IdUtil.simpleUUID(); |
| | | String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid; |
| | |
| | | code = captcha.getCode(); |
| | | } |
| | | redisCache.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES); |
| | | Map<String,Object> ajax = new HashMap<>(); |
| | | ajax.put("uuid", uuid); |
| | | ajax.put("img", captcha.getImageBase64()); |
| | | return AjaxResult.success(ajax); |
| | |
| | | |
| | | private String getCodeResult(String capStr) { |
| | | int numberLength = captchaProperties.getNumberLength(); |
| | | int a = Convert.toInt(StrUtil.sub(capStr, 0, numberLength).trim()); |
| | | int a = Convert.toInt(StringUtils.sub(capStr, 0, numberLength).trim()); |
| | | char operator = capStr.charAt(numberLength); |
| | | int b = Convert.toInt(StrUtil.sub(capStr, numberLength + 1, numberLength + 1 + numberLength).trim()); |
| | | int b = Convert.toInt(StringUtils.sub(capStr, numberLength + 1, numberLength + 1 + numberLength).trim()); |
| | | switch (operator) { |
| | | case '*': |
| | | return a * b + ""; |