Merge branch 'develop' of https://gitee.com/likeadmin/likeadmin_java into develop

This commit is contained in:
TinyAnts 2023-03-21 16:54:40 +08:00
commit f204e0d54b
3 changed files with 272 additions and 0 deletions

View File

@ -0,0 +1,44 @@
package com.mdd.admin.controller.channel;
import com.mdd.admin.service.IChannelOaCallBackService;
import com.mdd.common.aop.NotLogin;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
@RestController
@RequestMapping("api/channel/oa/callback")
@Api(tags = "公众号服务器验证及消息回复")
public class ChannelOaCallBackController {
@Resource
private IChannelOaCallBackService iChannelOaCallBackService;
@NotLogin
@GetMapping(produces = "text/plain;charset=utf-8")
@ApiOperation(value="公众号服务器地址验证")
public String authGet(@RequestParam(name = "signature", required = false) String signature,
@RequestParam(name = "timestamp", required = false) String timestamp,
@RequestParam(name = "nonce", required = false) String nonce,
@RequestParam(name = "echostr", required = false) String echostr) {
return iChannelOaCallBackService.checkSignature(signature, timestamp, nonce, echostr);
}
@NotLogin
@PostMapping(produces = "application/xml; charset=UTF-8")
@ApiOperation(value="公众号消息回复")
public String post(@RequestBody String requestBody,
@RequestParam("signature") String signature,
@RequestParam("timestamp") String timestamp,
@RequestParam("nonce") String nonce,
@RequestParam(name = "encrypt_type", required = false) String encType,
@RequestParam(name = "msg_signature", required = false) String msgSignature) {
return iChannelOaCallBackService.post(requestBody, signature, timestamp, nonce, encType, msgSignature);
}
}

View File

@ -0,0 +1,29 @@
package com.mdd.admin.service;
/**
* 公众号服务器回调
*/
public interface IChannelOaCallBackService {
/**
* 服务器验证
*/
String checkSignature(String signature, String timestamp, String nonce, String echostr);
/**
* 消息回复
*
* @param requestBody
* @param signature
* @param timestamp
* @param nonce
* @param encType
* @param msgSignature
* @return
*/
String post(String requestBody, String signature, String timestamp, String nonce, String encType, String msgSignature);
}

View File

@ -0,0 +1,199 @@
package com.mdd.admin.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.mdd.admin.service.IChannelOaCallBackService;
import com.mdd.common.entity.OfficialReply;
import com.mdd.common.exception.OperateException;
import com.mdd.common.mapper.OfficialReplyMapper;
import com.mdd.common.plugin.wechat.WxMnpDriver;
import me.chanjar.weixin.common.api.WxConsts;
import me.chanjar.weixin.mp.api.WxMpService;
import me.chanjar.weixin.mp.bean.message.WxMpXmlMessage;
import me.chanjar.weixin.mp.bean.message.WxMpXmlOutMessage;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.List;
import static me.chanjar.weixin.common.api.WxConsts.EventType.SUBSCRIBE;
@Service
public class ChannelOaCallBackServiceImpl implements IChannelOaCallBackService {
@Resource
private OfficialReplyMapper officialReplyMapper;
/**
* 服务器验证
*
* @param signature 微信加密签名
* @param timestamp 时间戳
* @param nonce 随机数
* @param echostr 随机字符串
* @return String
*/
@Override
public String checkSignature(String signature, String timestamp, String nonce, String echostr) {
WxMpService wxMpService = WxMnpDriver.oa();
if (wxMpService.checkSignature(timestamp, nonce, signature)) {
return echostr;
}
return "非法请求";
}
/**
* 消息回复
*
* @param requestBody 请求数据
* @param signature 微信加密签名
* @param timestamp 时间戳
* @param nonce 随机数
* @param encType 加密类型
* @param msgSignature 加密签名
* @return String
*/
@Override
public String post(String requestBody, String signature, String timestamp, String nonce, String encType, String msgSignature) {
WxMpService wxMpService = WxMnpDriver.oa();
if (!wxMpService.checkSignature(timestamp, nonce, signature)) {
throw new IllegalArgumentException("非法请求,可能属于伪造的请求!");
}
String outMsg = null;
if (encType == null) {
// 明文传输的消息
WxMpXmlMessage inMessage = WxMpXmlMessage.fromXml(requestBody);
WxMpXmlOutMessage outMessage = this.msgHandler(inMessage);
if (outMessage == null) {
return "";
}
outMsg = outMessage.toXml();
} else if ("aes".equalsIgnoreCase(encType)) {
// aes加密的消息
WxMpXmlMessage inMessage = WxMpXmlMessage.fromEncryptedXml(
requestBody,
wxMpService.getWxMpConfigStorage(),
timestamp,
nonce,
msgSignature);
WxMpXmlOutMessage outMessage = this.msgHandler(inMessage);
if (outMessage == null) {
return "";
}
outMsg = outMessage.toEncryptedXml(wxMpService.getWxMpConfigStorage());
}
return outMsg;
}
/**
* 消息处理
*
* @param wxMessage 微信回调信息
* @return WxMpXmlOutMessage
*/
private WxMpXmlOutMessage msgHandler(WxMpXmlMessage wxMessage) {
try {
// 文本消息
if (wxMessage.getMsgType().equals(WxConsts.XmlMsgType.TEXT)) {
String msg = keyMsg(wxMessage);
return textBuild(msg, wxMessage);
}
// 事件消息
if (wxMessage.getMsgType().equals(WxConsts.XmlMsgType.EVENT)) {
if (wxMessage.getEvent().equals(SUBSCRIBE)) {
// 关注回复
String msg = subMsg();
return textBuild(msg, wxMessage);
}
}
} catch (Exception e) {
throw new OperateException("公众号消息回调错误" + e.getMessage());
}
return null;
}
/**
* 返回文本消息
*
* @param content 文本内容
* @param wxMessage 微信回调信息
* @return WxMpXmlOutMessage
*/
private WxMpXmlOutMessage textBuild(String content, WxMpXmlMessage wxMessage) {
return WxMpXmlOutMessage.TEXT().content(content)
.fromUser(wxMessage.getToUser()).toUser(wxMessage.getFromUser())
.build();
}
/**
* 关键词回复
*
* @param wxMessage 微信回调信息
* @return String
*/
private String keyMsg(WxMpXmlMessage wxMessage) {
List<OfficialReply> oaReplyList = officialReplyMapper.selectList(
new QueryWrapper<OfficialReply>()
.eq("reply_type", 2)
.eq("is_delete", 0)
.eq("status", 1)
.orderByAsc("id"));
String msg = null;
for (OfficialReply oaReply : oaReplyList) {
// 全匹配
if (oaReply.getMatchingType() == 1 && oaReply.getKeyword().equals(wxMessage.getContent())) {
msg = oaReply.getContent();
}
// 模糊匹配
if (oaReply.getMatchingType() == 2 && wxMessage.getContent().contains(oaReply.getKeyword())) {
msg = oaReply.getContent();
}
}
return msg == null ? defaultMsg() : msg;
}
/**
* 默认回复
*
* @return String
*/
private String defaultMsg() {
OfficialReply officialReply = officialReplyMapper.selectOne(new QueryWrapper<OfficialReply>()
.eq("reply_type", 3)
.eq("is_delete", 0)
.eq("status", 1)
.last("limit 1"));
if (officialReply == null) {
return null;
}
return officialReply.getContent();
}
/**
* 关注回复内容
*
* @return String
*/
private String subMsg() {
OfficialReply officialReply = officialReplyMapper.selectOne(new QueryWrapper<OfficialReply>()
.eq("reply_type", 1)
.eq("is_delete", 0)
.eq("status", 1)
.last("limit 1"));
if (officialReply == null) {
return defaultMsg();
}
return officialReply.getContent();
}
}