266 lines
7.0 KiB
Go
266 lines
7.0 KiB
Go
package dto
|
||
|
||
import (
|
||
"encoding/xml"
|
||
"github.com/duke-git/lancet/v2/slice"
|
||
"go-wechat/types"
|
||
"regexp"
|
||
"strings"
|
||
)
|
||
|
||
// Message
|
||
// @description: 消息
|
||
type Message struct {
|
||
MsgId int64 `json:"msgId"`
|
||
CreateTime int `json:"createTime"`
|
||
Content string `json:"content"`
|
||
DisplayFullContent string `json:"displayFullContent"`
|
||
FromUser string `json:"fromUser"`
|
||
GroupUser string `json:"-"`
|
||
MsgSequence int `json:"msgSequence"`
|
||
Pid int `json:"pid"`
|
||
Signature string `json:"signature"`
|
||
ToUser string `json:"toUser"`
|
||
Type types.MessageType `json:"type"`
|
||
Raw string `json:"raw"`
|
||
}
|
||
|
||
// systemMsgDataXml
|
||
// @description: 微信系统消息的xml结构
|
||
type systemMsgDataXml struct {
|
||
SysMsg sysMsg `xml:"sysmsg"`
|
||
Type string `xml:"type,attr"`
|
||
}
|
||
|
||
// appMsgDataXml
|
||
// @description: 微信app消息的xml结构
|
||
type appMsgDataXml struct {
|
||
XMLName xml.Name `xml:"msg"`
|
||
Text string `xml:",chardata"`
|
||
AppMsg struct {
|
||
Text string `xml:",chardata"`
|
||
Appid string `xml:"appid,attr"`
|
||
SdkVer string `xml:"sdkver,attr"`
|
||
Title string `xml:"title"`
|
||
Des string `xml:"des"`
|
||
Action string `xml:"action"`
|
||
Type string `xml:"type"`
|
||
ShowType string `xml:"showtype"`
|
||
Content string `xml:"content"`
|
||
URL string `xml:"url"`
|
||
ThumbUrl string `xml:"thumburl"`
|
||
LowUrl string `xml:"lowurl"`
|
||
AppAttach struct {
|
||
Text string `xml:",chardata"`
|
||
TotalLen string `xml:"totallen"`
|
||
AttachId string `xml:"attachid"`
|
||
FileExt string `xml:"fileext"`
|
||
} `xml:"appattach"`
|
||
ExtInfo string `xml:"extinfo"`
|
||
} `xml:"appmsg"`
|
||
AppInfo struct {
|
||
Text string `xml:",chardata"`
|
||
Version string `xml:"version"`
|
||
AppName string `xml:"appname"`
|
||
} `xml:"appinfo"`
|
||
}
|
||
|
||
// atMsgDataXml
|
||
// @description: 微信@消息的xml结构
|
||
type atMsgDataXml struct {
|
||
XMLName xml.Name `xml:"msgsource"`
|
||
Text string `xml:",chardata"`
|
||
AtUserList string `xml:"atuserlist"`
|
||
Silence string `xml:"silence"`
|
||
MemberCount string `xml:"membercount"`
|
||
Signature string `xml:"signature"`
|
||
TmpNode struct {
|
||
Text string `xml:",chardata"`
|
||
PublisherID string `xml:"publisher-id"`
|
||
} `xml:"tmp_node"`
|
||
}
|
||
|
||
// sysMsg
|
||
// @description: 消息主体
|
||
type sysMsg struct{}
|
||
|
||
func (m Message) IsGroup() bool {
|
||
return strings.HasSuffix(m.FromUser, "@chatroom")
|
||
}
|
||
|
||
// IsPat
|
||
// @description: 是否是拍一拍消息
|
||
// @receiver m
|
||
// @return bool
|
||
func (m Message) IsPat() bool {
|
||
// 解析xml
|
||
var d systemMsgDataXml
|
||
if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
|
||
return false
|
||
}
|
||
|
||
return m.Type == types.MsgTypeRecalled && d.Type == "pat"
|
||
}
|
||
|
||
// IsRevokeMsg
|
||
// @description: 是否是撤回消息
|
||
// @receiver m
|
||
// @return bool
|
||
func (m Message) IsRevokeMsg() bool {
|
||
// 解析xml
|
||
var d systemMsgDataXml
|
||
if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
|
||
return false
|
||
}
|
||
|
||
return m.Type == types.MsgTypeRecalled && d.Type == "revokemsg"
|
||
}
|
||
|
||
// IsNewUserJoin
|
||
// @description: 是否是新人入群
|
||
// @receiver m
|
||
// @return bool
|
||
func (m Message) IsNewUserJoin() bool {
|
||
if m.Type != types.MsgTypeSys {
|
||
return false
|
||
}
|
||
|
||
isInvitation := strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊")
|
||
isScanQrCode := strings.Contains(m.Content, "通过扫描") && strings.Contains(m.Content, "加入群聊")
|
||
sysFlag := isInvitation || isScanQrCode
|
||
if sysFlag {
|
||
return true
|
||
}
|
||
// 解析另一种情况
|
||
var d systemMsgDataXml
|
||
if err := xml.Unmarshal([]byte(m.Content), &d); err != nil {
|
||
return false
|
||
}
|
||
return d.Type == "delchatroommember"
|
||
}
|
||
|
||
// IsAt
|
||
// @description: 是否是At机器人的消息
|
||
// @receiver m
|
||
// @return bool
|
||
func (m Message) IsAt() bool {
|
||
return strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你")
|
||
}
|
||
|
||
// IsAtAll
|
||
// @description: 是否是At所有人的消息
|
||
// @receiver m
|
||
// @return bool
|
||
func (m Message) IsAtAll() bool {
|
||
// 解析raw里面的xml
|
||
var d atMsgDataXml
|
||
if err := xml.Unmarshal([]byte(m.Signature), &d); err != nil {
|
||
return false
|
||
}
|
||
// 转换@用户列表为数组
|
||
atUserList := strings.Split(d.AtUserList, ",")
|
||
// 判断是否包含@所有人
|
||
if slice.Contain(atUserList, "notify@all") {
|
||
return true
|
||
}
|
||
// 数据格式变动，再检查一下字符串是否包含 @所有人 字样
|
||
return m.IsAt() && strings.Contains(m.Content, "@所有人")
|
||
}
|
||
|
||
// IsPrivateText
|
||
// @description: 是否是私聊消息
|
||
// @receiver m
|
||
// @return bool
|
||
func (m Message) IsPrivateText() bool {
|
||
// 发信人不以@chatroom结尾且消息类型为文本
|
||
return !strings.HasSuffix(m.FromUser, "chatroom") && m.Type == types.MsgTypeText
|
||
}
|
||
|
||
// CleanContentStartWith
|
||
// @description: 判断是否包含指定消息前缀
|
||
// @receiver m
|
||
// @param prefix
|
||
// @return bool
|
||
func (m Message) CleanContentStartWith(prefix string) bool {
|
||
content := m.Content
|
||
|
||
// 如果是@消息，过滤掉@的内容
|
||
if m.IsAt() {
|
||
re := regexp.MustCompile(`@([^ | ]+)`)
|
||
matches := re.FindStringSubmatch(content)
|
||
if len(matches) > 0 {
|
||
// 过滤掉第一个匹配到的
|
||
content = strings.Replace(content, matches[0], "", 1)
|
||
}
|
||
}
|
||
|
||
// 去掉最前面的空格
|
||
content = strings.TrimLeft(content, " ")
|
||
content = strings.TrimLeft(content, " ")
|
||
|
||
return strings.HasPrefix(content, prefix)
|
||
}
|
||
|
||
// IsInvitationJoinGroup
|
||
// @description: 是否是邀请入群消息
|
||
// @receiver m
|
||
// @return bool 是否是邀请入群消息
|
||
// @return string 邀请入群消息内容
|
||
func (m Message) IsInvitationJoinGroup() (flag bool, str string) {
|
||
if m.Type == types.MsgTypeApp {
|
||
// 解析xml
|
||
var md appMsgDataXml
|
||
if err := xml.Unmarshal([]byte(m.Content), &md); err != nil {
|
||
return
|
||
}
|
||
flag = md.AppMsg.Type == "5" && md.AppMsg.Title == "邀请你加入群聊"
|
||
str = strings.ReplaceAll(md.AppMsg.Des, "，进入可查看详情。", "")
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
// IsNewFriendAdd
|
||
// @description: 是否是新好友添加消息
|
||
// @receiver m
|
||
// @return flag
|
||
func (m Message) IsNewFriendAdd() (flag bool) {
|
||
if m.Type != types.MsgTypeSys {
|
||
return
|
||
}
|
||
return strings.HasPrefix(m.Content, "你已添加了") && strings.HasSuffix(m.Content, "，现在可以开始聊天了。")
|
||
}
|
||
|
||
// IsOldFriendBack
|
||
// @description: 是否是老好友回归消息
|
||
// @receiver m
|
||
// @return flag
|
||
func (m Message) IsOldFriendBack() (flag bool) {
|
||
if m.Type != types.MsgTypeSys {
|
||
return
|
||
}
|
||
return m.Content == "以上是打招呼的内容"
|
||
}
|
||
|
||
// IsJoinToGroup
|
||
// @description: 是否是加入群聊消息
|
||
// @receiver m
|
||
// @return flag
|
||
func (m Message) IsJoinToGroup() (flag bool) {
|
||
if m.Type != types.MsgTypeSys {
|
||
return
|
||
}
|
||
return strings.Contains(m.Content, "\"邀请你加入了群聊，群聊参与人还有：")
|
||
}
|
||
|
||
// IsRemoveFromChatroom
|
||
// @description: 是否是被移出群聊消息
|
||
// @receiver m
|
||
// @return flag
|
||
func (m Message) IsRemoveFromChatroom() (flag bool) {
|
||
if m.Type != types.MsgTypeSys {
|
||
return
|
||
}
|
||
return strings.HasPrefix(m.Content, "你被\"") && strings.HasSuffix(m.Content, "\"移出群聊")
|
||
}
|