lbqms.git - Gitblit

			@@ -1,11 +1,10 @@
			package com.ruoyi.web.controller.common;

			import cn.dev33.satoken.annotation.SaIgnore;
			import cn.hutool.captcha.AbstractCaptcha;
			import cn.hutool.captcha.generator.CodeGenerator;
			import cn.hutool.core.convert.Convert;
			import cn.hutool.core.util.IdUtil;
			import cn.hutool.core.util.RandomUtil;
			import com.ruoyi.common.annotation.Anonymous;
			import com.ruoyi.common.constant.CacheConstants;
			import com.ruoyi.common.constant.Constants;
			import com.ruoyi.common.core.domain.R;
			@@ -19,16 +18,16 @@
			import com.ruoyi.sms.core.SmsTemplate;
			import com.ruoyi.sms.entity.SmsResult;
			import com.ruoyi.system.service.ISysConfigService;
			import io.swagger.v3.oas.annotations.Operation;
			import io.swagger.v3.oas.annotations.Parameter;
			import io.swagger.v3.oas.annotations.tags.Tag;
			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.validation.constraints.NotBlank;
			import jakarta.validation.constraints.NotBlank;
			import java.time.Duration;
			import java.util.HashMap;
			import java.util.Map;
			@@ -38,10 +37,9 @@
			*
			* @author Lion Li
			*/
			@Anonymous
			@SaIgnore
			@Slf4j
			@Validated
			@Tag(name ="验证码操作处理", description = "验证码管理")
			@RequiredArgsConstructor
			@RestController
			public class CaptchaController {
			@@ -52,14 +50,14 @@

			/**
			* 短信验证码
			*
			* @param phonenumber 用户手机号
			*/
			@Operation(summary = "短信验证码")
			@GetMapping("/captchaSms")
			public R<Void> smsCaptcha(@Parameter(name = "用户手机号")
			@NotBlank(message = "{user.phonenumber.not.blank}")
			public R<Void> smsCaptcha(@NotBlank(message = "{user.phonenumber.not.blank}")
			String phonenumber) {
			if (smsProperties.getEnabled()) {
			R.fail("当前系统没有开启短信功能！");
			if (!smsProperties.getEnabled()) {
			return R.fail("当前系统没有开启短信功能！");
			}
			String key = CacheConstants.CAPTCHA_CODE_KEY + phonenumber;
			String code = RandomUtil.randomNumbers(4);
			@@ -80,13 +78,12 @@
			/**
			* 生成验证码
			*/
			@Operation(summary = "生成验证码")
			@GetMapping("/captchaImage")
			public R<Map<String, Object>> getCode() {
			Map<String, Object> ajax = new HashMap<>();
			boolean captchaOnOff = configService.selectCaptchaOnOff();
			ajax.put("captchaOnOff", captchaOnOff);
			if (!captchaOnOff) {
			boolean captchaEnabled = configService.selectCaptchaEnabled();
			ajax.put("captchaEnabled", captchaEnabled);
			if (!captchaEnabled) {
			return R.ok(ajax);
			}
			// 保存验证码信息
			@@ -100,28 +97,16 @@
			AbstractCaptcha captcha = SpringUtils.getBean(captchaProperties.getCategory().getClazz());
			captcha.setGenerator(codeGenerator);
			captcha.createCode();
			String code = isMath ? getCodeResult(captcha.getCode()) : captcha.getCode();
			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);
			}

			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 Convert.toStr(a * b);
			case '+':
			return Convert.toStr(a + b);
			case '-':
			return Convert.toStr(a - b);
			default:
			return StringUtils.EMPTY;
			}
			}

			}