add send captcha

src/main/java/com/huiju/estateagents/common/SMSUtils.java

@@ -0,0 +1,134 @@
+package com.huiju.estateagents.common;
+
+import com.huiju.estateagents.config.SMSProperties;
+import lombok.Data;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.*;
+import org.springframework.web.client.RestTemplate;
+
+import java.time.LocalDateTime;
+import java.util.*;
+
+@Slf4j
+@Configuration
+@EnableConfigurationProperties(SMSProperties.class)
+public class SMSUtils {
+    // 短信服务配置
+    SMSProperties smsProperties;
+
+    Map<String, Phone> allPhones;
+
+    @Autowired
+    public SMSUtils(SMSProperties properties) {
+        smsProperties = properties;
+        allPhones = new HashMap<>();
+    }
+
+    /**
+     * 发送验证码到 tel
+     * @param tel
+     * @return
+     */
+    public boolean sendCaptcha(String tel) {
+        Phone phone = allPhones.get(tel);
+        if (null == phone) {
+            phone = new Phone();
+            phone.setTel(tel);
+        }
+
+        phone.setCaptcha(genCaptcha(smsProperties.getCaptcha().getSize()));
+        phone.setExpire(LocalDateTime.now().plusSeconds(smsProperties.getCaptcha().getExpire()));
+        allPhones.put(tel, phone);
+
+        log.info("发送手机 {} 验证码 {}", tel, phone.getCaptcha());
+
+        String[] params = { phone.getCaptcha(), smsProperties.getCaptcha().getSign() };
+        return sendMessage(smsProperties.getCaptcha().getCode(), tel, params);
+    }
+
+    /**
+     * 校验验证码
+     * @param tel
+     * @param smsCaptcha
+     * @return
+     */
+    public boolean checkCaptcha(String tel, String smsCaptcha) {
+        Phone phone = allPhones.get(tel);
+        if (null == phone) {
+            log.error("手机号 {} 的验证码未找到", tel);
+            return false;
+        }
+
+        if (null == smsCaptcha || !smsCaptcha.equals(phone.getCaptcha())) {
+            log.error("手机号 {} 校验验证码出错, 正确: {}, 错误: {}", tel, phone.getCaptcha(), smsCaptcha);
+            return false;
+        }
+
+        LocalDateTime now = LocalDateTime.now();
+        if (now.isAfter(phone.getExpire())) {
+            log.error("手机号 {} 的验证码已过期", tel);
+            return false;
+        }
+
+        // 如果验证成功, 则从缓存中删除
+        allPhones.remove(tel);
+        return true;
+    }
+
+    public boolean sendVisitor(String tel, String name) {
+        log.info("发送手机 {} 客户名称 {}", tel, name);
+
+        String[] params = { name };
+        return sendMessage(smsProperties.getVisitor().getCode(), tel, params);
+    }
+
+    private String genCaptcha(int length) {
+        List<String> nums = new ArrayList<>();
+        for (int i = 0; i < length; i ++) {
+            int rand = (new Random()).nextInt(10);
+            nums.add(String.valueOf(rand));
+        }
+        return String.join("", nums);
+    }
+
+    private boolean sendMessage(String code, String tel, String[] params) {
+        // 发送内容
+        SMSProperties.Template message = new SMSProperties.Template();
+        message.setCode(code);
+        message.setTel(tel);
+        message.setParams(params);
+
+        // 构造 request
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.parseMediaType(smsProperties.getContentType()));
+//        String body = JSONObject.toJSONString(message);
+        HttpMethod method = HttpMethod.resolve(smsProperties.getMethod());
+        HttpEntity requestEntity = new HttpEntity(message, headers);
+
+        RestTemplate template = new RestTemplate();
+        HttpEntity<String> response = template.exchange(
+                smsProperties.getUrl(),
+                method,
+                requestEntity,
+                String.class,
+                new Object()
+        );
+
+        if (((ResponseEntity<String>) response).getStatusCodeValue() != HttpStatus.OK.value()) {
+            log.error("请求发送短信验证码失败: {}", response.toString());
+            return false;
+        }
+
+        return true;
+    }
+
+    @Data
+    public static class Phone {
+        String tel;
+        String captcha;
+        LocalDateTime expire;
+    }
+}

src/main/java/com/huiju/estateagents/config/SMSProperties.java

@@ -0,0 +1,35 @@
+package com.huiju.estateagents.config;
+
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+
+@Data
+@ConfigurationProperties(prefix = "sms")
+public class SMSProperties {
+    String url;
+    String method;
+    String contentType;
+    Captcha captcha;
+    Visitor visitor;
+
+    @Data
+    public static class Template {
+        String code;
+        String tel;
+        String[] params;
+    }
+
+    @Data
+    public static class Captcha {
+        String code;
+        Integer expire;
+        Integer size;
+        String sign;
+    }
+
+    @Data
+    public static class Visitor {
+        String code;
+    }
+}

src/main/java/com/huiju/estateagents/controller/CommonController.java

@@ -4,8 +4,10 @@ package com.huiju.estateagents.controller;
 import com.huiju.estateagents.base.ResponseBean;
 import com.huiju.estateagents.common.AliOSSUtils;
 import com.huiju.estateagents.common.CommConstant;
+import com.huiju.estateagents.common.SMSUtils;
 import com.huiju.estateagents.common.StringUtils;
 import org.apache.ibatis.annotations.Mapper;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -18,6 +20,8 @@ import java.util.Map;
 @RestController
 @RequestMapping("/api")
 public class CommonController {
+    @Autowired
+    SMSUtils smsUtils;
 
     /**
      * 图片上传
@@ -60,6 +64,16 @@ public class CommonController {
         return result;
     }
 
+    @PostMapping("{plat}/captcha")
+    public ResponseBean sendCaptcha(@RequestParam String tel) {
+        boolean success = smsUtils.sendCaptcha(tel);
+        if (success) {
+            return ResponseBean.success("发送验证码成功");
+        } else {
+            return ResponseBean.error("验证码发送失败, 请重试", ResponseBean.ERROR_UNAVAILABLE);
+        }
+    }
+
     @GetMapping("/wx/awesome/dict/{typeFor}")
     public ResponseBean dicts(@PathVariable String typeFor) {