| | |
|---|
| | | package com.ruoyi.web.controller.common; |
|---|
| | | |
|---|
| | | import cn.dev33.satoken.annotation.SaIgnore; |
|---|
| | | import cn.hutool.captcha.AbstractCaptcha; |
|---|
| | | import cn.hutool.captcha.CircleCaptcha; |
|---|
| | | import cn.hutool.captcha.LineCaptcha; |
|---|
| | | import cn.hutool.captcha.ShearCaptcha; |
|---|
| | | import cn.hutool.captcha.generator.CodeGenerator; |
|---|
| | | import cn.hutool.captcha.generator.RandomGenerator; |
|---|
| | | import cn.hutool.core.convert.Convert; |
|---|
| | | import cn.hutool.core.util.IdUtil; |
|---|
| | | import com.ruoyi.common.utils.StringUtils; |
|---|
| | | import cn.hutool.core.util.RandomUtil; |
|---|
| | | import com.ruoyi.common.constant.CacheConstants; |
|---|
| | | 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.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 org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | 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 javax.annotation.Resource; |
|---|
| | | import jakarta.validation.constraints.NotBlank; |
|---|
| | | import java.time.Duration; |
|---|
| | | import java.util.HashMap; |
|---|
| | | import java.util.Map; |
|---|
| | | import java.util.concurrent.TimeUnit; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 验证码操作处理 |
|---|
| | | * |
|---|
| | | * @author ruoyi |
|---|
| | | * @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; |
|---|
| | | /** |
|---|
| | | * 短信验证码 |
|---|
| | | * |
|---|
| | | * @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(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | @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; |
|---|
| | | String code = null; |
|---|
| | | // 生成验证码 |
|---|
| | | CodeGenerator codeGenerator; |
|---|
| | | AbstractCaptcha captcha; |
|---|
| | | switch (captchaProperties.getType()) { |
|---|
| | | case "math": |
|---|
| | | codeGenerator = new UnsignedMathGenerator(captchaProperties.getNumberLength()); |
|---|
| | | break; |
|---|
| | | case "char": |
|---|
| | | codeGenerator = new RandomGenerator(captchaProperties.getCharLength()); |
|---|
| | | break; |
|---|
| | | default: |
|---|
| | | throw new IllegalArgumentException("验证码类型异常"); |
|---|
| | | } |
|---|
| | | switch (captchaProperties.getCategory()) { |
|---|
| | | case "line": |
|---|
| | | captcha = lineCaptcha; |
|---|
| | | break; |
|---|
| | | case "circle": |
|---|
| | | captcha = circleCaptcha; |
|---|
| | | break; |
|---|
| | | case "shear": |
|---|
| | | captcha = shearCaptcha; |
|---|
| | | break; |
|---|
| | | default: |
|---|
| | | throw new IllegalArgumentException("验证码类别异常"); |
|---|
| | | } |
|---|
| | | captcha.setGenerator(codeGenerator); |
|---|
| | | captcha.createCode(); |
|---|
| | | if ("math".equals(captchaProperties.getType())) { |
|---|
| | | code = getCodeResult(captcha.getCode()); |
|---|
| | | } else if ("char".equals(captchaProperties.getType())) { |
|---|
| | | code = captcha.getCode(); |
|---|
| | | } |
|---|
| | | redisCache.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES); |
|---|
| | | 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(StringUtils.substring(capStr, 0, numberLength).trim()); |
|---|
| | | char operator = capStr.charAt(numberLength); |
|---|
| | | int b = Convert.toInt(StringUtils.substring(capStr, numberLength + 1, numberLength + 1 + numberLength).trim()); |
|---|
| | | switch (operator) { |
|---|
| | | case '*': |
|---|
| | | return a * b + ""; |
|---|
| | | case '+': |
|---|
| | | return a + b + ""; |
|---|
| | | case '-': |
|---|
| | | return a - b + ""; |
|---|
| | | default: |
|---|
| | | return ""; |
|---|
| | | } |
|---|
| | | } |
|---|
| | | /** |
|---|
| | | * 生成验证码 |
|---|
| | | */ |
|---|
| | | @GetMapping("/captchaImage") |
|---|
| | | 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 = CacheConstants.CAPTCHA_CODE_KEY + uuid; |
|---|
| | | // 生成验证码 |
|---|
| | | 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); |
|---|
| | | } |
|---|
| | | RedisUtils.setCacheObject(verifyKey, code, Duration.ofMinutes(Constants.CAPTCHA_EXPIRATION)); |
|---|
| | | ajax.put("uuid", uuid); |
|---|
| | | ajax.put("img", captcha.getImageBase64()); |
|---|
| | | return R.ok(ajax); |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
