121 lines
3.4 KiB
Go
121 lines
3.4 KiB
Go
|
package wxtask
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"regexp"
|
|||
|
|
"strings"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
"xiawan/wx/clientsdk/baseinfo"
|
|||
|
|
"xiawan/wx/db"
|
|||
|
|
|
|||
|
|
"xiawan/wx/protobuf/wechat"
|
|||
|
|
"xiawan/wx/srv/wxface"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// WXRevokeTask 消息防撤回管理器
|
|||
|
|
type WXRevokeTask struct {
|
|||
|
|
wxConn wxface.IWXConnect
|
|||
|
|
// 防撤回开关
|
|||
|
|
bAvoidRevoke bool
|
|||
|
|
// 读写数据的读写锁
|
|||
|
|
revokeLock sync.RWMutex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewWXRevokeTask 新建消息防撤回任务管理器
|
|||
|
|
func NewWXRevokeTask(wxConn wxface.IWXConnect) *WXRevokeTask {
|
|||
|
|
return &WXRevokeTask{
|
|||
|
|
wxConn: wxConn,
|
|||
|
|
bAvoidRevoke: false,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Start 开启防消息撤回任务
|
|||
|
|
func (wxrt *WXRevokeTask) Start() {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Stop 停止防消息撤回任务
|
|||
|
|
func (wxrt *WXRevokeTask) Stop() {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddNewMsg 新增缓存消息
|
|||
|
|
func (wxrt *WXRevokeTask) AddNewMsg(addMsg *wechat.AddMsg, uuid string) {
|
|||
|
|
wxrt.revokeLock.Lock()
|
|||
|
|
defer wxrt.revokeLock.Unlock()
|
|||
|
|
_ = db.SaveMsg(*addMsg, uuid)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// OnRevokeMsg 某人移除了消息
|
|||
|
|
func (wxrt *WXRevokeTask) OnRevokeMsg(revokeMsg baseinfo.RevokeMsg, groupWXID string, toGroupWXID string, uuid string) {
|
|||
|
|
wxrt.revokeLock.Lock()
|
|||
|
|
defer wxrt.revokeLock.Unlock()
|
|||
|
|
// fmt.Printf("撤回消息：groupWXID %s\n", groupWXID)
|
|||
|
|
currentReqInvoker := wxrt.wxConn.GetWXReqInvoker()
|
|||
|
|
now := time.Now()
|
|||
|
|
timeStr := now.Format("2006-01-02 15:04:05")
|
|||
|
|
groupName := ""
|
|||
|
|
if groupWXID != "" {
|
|||
|
|
groupInfo, _, _ := currentReqInvoker.SendGetContactRequestForCache(groupWXID, true)
|
|||
|
|
if groupInfo != nil {
|
|||
|
|
groupName = groupInfo.GetNickName().GetStr()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if groupName != "" {
|
|||
|
|
groupName = "【" + groupName + "】\n"
|
|||
|
|
}
|
|||
|
|
groupName = timeStr + "\n" + groupName
|
|||
|
|
|
|||
|
|
currentWXFileHelperMgr := wxrt.wxConn.GetWXFileHelperMgr()
|
|||
|
|
|
|||
|
|
tmpAddMsg := db.GetAndDeleteMsg(revokeMsg.NewMsgID, uuid)
|
|||
|
|
if tmpAddMsg == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
strMsg := tmpAddMsg.GetContent().GetStr()
|
|||
|
|
// fmt.Println("撤回消息 MsgType：" + fmt.Sprintf("%d", tmpAddMsg.GetMsgType()))
|
|||
|
|
if tmpAddMsg.GetMsgType() == 1 {
|
|||
|
|
if groupWXID != "" && toGroupWXID == "" && strings.Contains(strMsg, ":") {
|
|||
|
|
// 群聊，去掉第一个冒号之前的内容
|
|||
|
|
strMsg = strings.Split(strMsg, ":")[1]
|
|||
|
|
}
|
|||
|
|
showMsg := groupName + revokeMsg.ReplaceMsg + ":" + strMsg
|
|||
|
|
currentWXFileHelperMgr.AddNewTipMsg(showMsg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if tmpAddMsg.GetMsgType() == baseinfo.MMAddMsgTypeRefer && strings.Contains(strMsg, "<type>57</type>") {
|
|||
|
|
// 引用消息，只可能是文本
|
|||
|
|
// <title>了解一下</title>
|
|||
|
|
titleRegex := regexp.MustCompile(`<title>([\s\S]*?)<\/title>`)
|
|||
|
|
match := titleRegex.FindStringSubmatch(strMsg)
|
|||
|
|
if len(match) > 1 {
|
|||
|
|
strMsg = match[1]
|
|||
|
|
showMsg := groupName + revokeMsg.ReplaceMsg + ":" + strMsg
|
|||
|
|
currentWXFileHelperMgr.AddNewTipMsg(showMsg)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 图片消息
|
|||
|
|
if tmpAddMsg.GetMsgType() == baseinfo.MMAddMsgTypeImage {
|
|||
|
|
strMsgByte := []byte(strMsg)
|
|||
|
|
showMsg := groupName + revokeMsg.ReplaceMsg + ":[图片]"
|
|||
|
|
currentWXFileHelperMgr.AddNewTipMsg(showMsg)
|
|||
|
|
currentWXFileHelperMgr.ForwardImageMsg(strMsgByte)
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
// 表情消息
|
|||
|
|
if tmpAddMsg.GetMsgType() == baseinfo.MMAddMsgTypePic {
|
|||
|
|
strMsgByte := []byte(strMsg)
|
|||
|
|
showMsg := groupName + revokeMsg.ReplaceMsg + ":[表情]"
|
|||
|
|
currentWXFileHelperMgr.AddNewTipMsg(showMsg)
|
|||
|
|
currentWXFileHelperMgr.ForwardEmoticonMsg(strMsgByte)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SetAvoidRevoke 设置是否防消息撤回
|
|||
|
|
func (wxrt *WXRevokeTask) SetAvoidRevoke(bFlag bool) {
|
|||
|
|
wxrt.bAvoidRevoke = bFlag
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsAvoidRevoke 是否防消息撤回
|
|||
|
|
func (wxrt *WXRevokeTask) IsAvoidRevoke() bool {
|
|||
|
|
return wxrt.bAvoidRevoke
|
|||
|
|
}
|