first commit

2023-12-05 01:39:52 +08:00
commit 767b08a025
46 changed files with 1739 additions and 0 deletions
--- a/src/main/java/com/kimgo/wepush/service/SMSService.java
+++ b/src/main/java/com/kimgo/wepush/service/SMSService.java
@@ -0,0 +1,107 @@
+package com.kimgo.wepush.service;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.kimgo.wepush.common.QyWeChatSendMessageApiResponse;
+import com.kimgo.wepush.config.URLConfig;
+import com.kimgo.wepush.model.SMSInfo;
+import com.kimgo.wepush.model.TextCardMessage;
+import com.kimgo.wepush.response.ServerResponseEntity;
+import okhttp3.*;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import com.kimgo.wepush.config.UserConfig;
+import java.io.IOException;
+
+@Service
+public class SMSService {
+    private final Logger logger = LoggerFactory.getLogger(SMSService.class);
+    @Autowired
+    private UserConfig userConfig;
+    @Autowired
+    private ApiSettingService apiSettingService;
+    @Autowired
+    private TokenService tokenService;
+    @Autowired
+    private URLConfig urlConfig;
+
+    public ServerResponseEntity getSMSInfo(String accessToken, SMSInfo smsInfo){
+        String correctAccessToken = tokenService.getApiAccessToken();
+        logger.info("accessToken: {} correctAccessToken: {}",accessToken,correctAccessToken);
+        if (!correctAccessToken.equals(accessToken)){
+            return ServerResponseEntity.fail("Invalid accessToken");
+        }
+        String qyAccessToken = tokenService.getAccessToken();
+        QyWeChatSendMessageApiResponse qyWeChatSendMessageApiResponse = requestWithOkhttp(qyAccessToken,smsInfo);
+        if (qyWeChatSendMessageApiResponse != null){
+            return ServerResponseEntity.success(qyWeChatSendMessageApiResponse.getMsgid());
+        }
+        return ServerResponseEntity.fail("request qyWeChat error,please check server error log.");
+    }
+
+    public QyWeChatSendMessageApiResponse requestWithOkhttp(String accessToken,SMSInfo smsInfo){
+        OkHttpClient client = new OkHttpClient();
+        String url = urlConfig.getSendTextCardMessageUrl() + accessToken;
+
+        TextCardMessage textCardMessage = setTextCardMessage(smsInfo);
+        // 使用Jackson进行序列化
+        ObjectMapper objectMapper = new ObjectMapper();
+        String jsonBody = null;
+        try {
+            jsonBody = objectMapper.writeValueAsString(textCardMessage);
+            logger.info("jsonBody: {}",jsonBody);
+        } catch (JsonProcessingException e) {
+            logger.error("JSON processing error", e);
+            return null;
+        }
+        // 构建请求体
+        RequestBody body = RequestBody.create(MediaType.parse("application/json; charset=utf-8"), jsonBody);
+        Request request = new Request.Builder().url(url).post(body).build();
+        try (Response response = client.newCall(request).execute()) {
+            String responseBody = response.body().string();
+            logger.info("Response: {}", responseBody);
+
+            ObjectMapper objectMapper1 = new ObjectMapper();
+            QyWeChatSendMessageApiResponse apiResponse = objectMapper1.readValue(responseBody, QyWeChatSendMessageApiResponse.class);
+
+            if (apiResponse.getErrcode() == 0) {
+                return apiResponse;
+            } else if (apiResponse.getErrcode() == 42001 || apiResponse.getErrcode() == 40014) {
+                logger.info("Access token expired. Refreshing token...");
+                tokenService.setAccessToken();  // 一个方法来刷新accessToken
+                String newAccessToken = tokenService.getAccessToken();
+                return requestWithOkhttp(newAccessToken, smsInfo);
+            } else {
+                // 处理其他错误情况
+                logger.error("Error: {}", apiResponse.getErrmsg());
+                return apiResponse;
+            }
+        } catch (IOException e) {
+            logger.error("OkHttp request error", e);
+            return null;
+        }
+    }
+
+    public TextCardMessage setTextCardMessage(SMSInfo smsInfo) {
+        TextCardMessage textCardMessage = new TextCardMessage();
+
+        textCardMessage.setTouser(apiSettingService.getApiSetting().getTouser());
+        textCardMessage.setMsgtype(apiSettingService.getApiSetting().getMsgtype());
+        textCardMessage.setAgentid(apiSettingService.getApiSetting().getAgentid());
+        textCardMessage.setEnable_duplicate_check(apiSettingService.getApiSetting().getEnable_duplicate_check());
+        textCardMessage.setDuplicate_check_interval(apiSettingService.getApiSetting().getDuplicate_check_interval());
+
+        TextCardMessage.TextCard textCard = new TextCardMessage.TextCard();
+        textCard.setTitle("新短信通知");
+        textCard.setDescription("<div class='gray'>您有一条新短信</div><div class='highlight'>短信号码: " + smsInfo.getSmsNumber() + "</div><div class='highlight'>接受短信时间: " + smsInfo.getSmsAcceptanceTime() + "</div>" + smsInfo.getSmsContent());
+        textCard.setUrl("https://kimgo.cn");
+        // 将TextCard对象设置到TextCardMessage中
+        textCardMessage.setTextcard(textCard);
+
+        logger.info("TextCardMessage: {}", textCardMessage.toString());
+
+        return textCardMessage;
+    }
+}