wePush/src/main/java/com/kimgo/wepush/service/SMSService.java

package com.kimgo.wepush.service;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.kimgo.wepush.model.TextMessage;
import com.kimgo.wepush.response.QyWeChatSendMessageApiResponse;
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 java.io.IOException;

@Service
public class SMSService {
    private final Logger logger = LoggerFactory.getLogger(SMSService.class);

    @Autowired
    private ApiSettingService apiSettingService;
    @Autowired
    private TokenService tokenService;
    @Autowired
    private QyWeChatURLService qyWeChatURLService;

    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();

        Request request = handleMessage(accessToken,smsInfo);
        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().getEnableDuplicateCheck());
        textCardMessage.setDuplicate_check_interval(apiSettingService.getApiSetting().getDuplicateCheckInterval());

        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;
    }
    public TextMessage setTextMessage(SMSInfo smsInfo){
        TextMessage textMessage = new TextMessage();
        textMessage.setTouser(apiSettingService.getApiSetting().getTouser());
        textMessage.setMsgtype("text");
        textMessage.setAgentid(apiSettingService.getApiSetting().getAgentid());
        textMessage.setEnable_duplicate_check(apiSettingService.getApiSetting().getEnableDuplicateCheck());
        textMessage.setDuplicate_check_interval(apiSettingService.getApiSetting().getDuplicateCheckInterval());
        TextMessage.Text text = new TextMessage.Text();
        text.setContent("你有一条新短信\n短信号码: "+smsInfo.getSmsNumber() + "\n接受时间: " + smsInfo.getSmsAcceptanceTime() + "\n内容: \n"+ smsInfo.getSmsContent());
        textMessage.setText(text);
        logger.info("TextMessage: {}", textMessage.toString());
        return textMessage;
    }
    private<T> Request bulidRequest(String accessToken,T object){
        String url = qyWeChatURLService.getSendTextCardMessageUrl() + "?access_token=" + accessToken;
        // 使用Jackson进行序列化
        ObjectMapper objectMapper = new ObjectMapper();
        String jsonBody = null;
        try {
            jsonBody = objectMapper.writeValueAsString(object);
            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();
        return request;
    }
    private Request handleMessage(String accessToken, SMSInfo smsInfo) {
        TextCardMessage textCardMessage = setTextCardMessage(smsInfo);
        int textCardMessageSize = checkMessageSize(textCardMessage.getTextcard().getDescription());

        if (textCardMessageSize <= 512) {
            return bulidRequest(accessToken, textCardMessage);
        } else {
            TextMessage textMessage = setTextMessage(smsInfo);
            int textMessageSize = checkMessageSize(textMessage.getText().getContent());

            if (textMessageSize <= 2048) {
                return bulidRequest(accessToken, textMessage);
            } else {
                TextCardMessage wrMsg = setWarningTextCardMessage();
                return bulidRequest(accessToken, wrMsg);
            }
        }
    }

    public TextCardMessage setWarningTextCardMessage() {
        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().getEnableDuplicateCheck());
        textCardMessage.setDuplicate_check_interval(apiSettingService.getApiSetting().getDuplicateCheckInterval());

        TextCardMessage.TextCard textCard = new TextCardMessage.TextCard();
        textCard.setTitle("警告消息");
        textCard.setDescription("<div class='highlight'>警告: 短信消息过长请查看手机。</div>");
        textCard.setUrl("https://kimgo.cn");
        // 将TextCard对象设置到TextCardMessage中
        textCardMessage.setTextcard(textCard);

        logger.info("TextCardMessage: {}", textCardMessage.toString());

        return textCardMessage;
    }
    public int checkMessageSize(String message){
        byte[] messageBytes = message.getBytes(java.nio.charset.StandardCharsets.UTF_8);
        int size = messageBytes.length;
        logger.info("Message size in bytes: " + size);
        return size;
    }
}