mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-05 03:23:49 +08:00
612 lines
18 KiB
Go
612 lines
18 KiB
Go
package coolq
|
||
|
||
import (
|
||
"encoding/hex"
|
||
"os"
|
||
"path"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/Mrs4s/go-cqhttp/global"
|
||
|
||
"github.com/Mrs4s/MiraiGo/binary"
|
||
"github.com/Mrs4s/MiraiGo/client"
|
||
"github.com/Mrs4s/MiraiGo/message"
|
||
log "github.com/sirupsen/logrus"
|
||
)
|
||
|
||
var format = "string"
|
||
|
||
// SetMessageFormat 设置消息上报格式，默认为string
|
||
func SetMessageFormat(f string) {
|
||
format = f
|
||
}
|
||
|
||
// ToFormattedMessage 将给定[]message.IMessageElement转换为通过coolq.SetMessageFormat所定义的消息上报格式
|
||
func ToFormattedMessage(e []message.IMessageElement, groupID int64, isRaw ...bool) (r interface{}) {
|
||
if format == "string" {
|
||
r = ToStringMessage(e, groupID, isRaw...)
|
||
} else if format == "array" {
|
||
r = ToArrayMessage(e, groupID)
|
||
}
|
||
return
|
||
}
|
||
|
||
func (bot *CQBot) privateMessageEvent(c *client.QQClient, m *message.PrivateMessage) {
|
||
bot.checkMedia(m.Elements)
|
||
cqm := ToStringMessage(m.Elements, 0, true)
|
||
if !m.Sender.IsFriend {
|
||
bot.oneWayMsgCache.Store(m.Sender.Uin, "")
|
||
}
|
||
id := m.Id
|
||
if bot.db != nil {
|
||
id = bot.InsertPrivateMessage(m)
|
||
}
|
||
log.Infof("收到好友 %v(%v) 的消息: %v (%v)", m.Sender.DisplayName(), m.Sender.Uin, cqm, id)
|
||
fm := MSG{
|
||
"post_type": func() string {
|
||
if m.Sender.Uin == bot.Client.Uin {
|
||
return "message_sent"
|
||
}
|
||
return "message"
|
||
}(),
|
||
"message_type": "private",
|
||
"sub_type": "friend",
|
||
"message_id": id,
|
||
"user_id": m.Sender.Uin,
|
||
"target_id": m.Target,
|
||
"message": ToFormattedMessage(m.Elements, 0, false),
|
||
"raw_message": cqm,
|
||
"font": 0,
|
||
"self_id": c.Uin,
|
||
"time": time.Now().Unix(),
|
||
"sender": MSG{
|
||
"user_id": m.Sender.Uin,
|
||
"nickname": m.Sender.Nickname,
|
||
"sex": "unknown",
|
||
"age": 0,
|
||
},
|
||
}
|
||
bot.dispatchEventMessage(fm)
|
||
if m.Sender.Uin != c.Uin {
|
||
c.MarkPrivateMessageReaded(m.Sender.Uin, int64(m.Time))
|
||
}
|
||
}
|
||
|
||
func (bot *CQBot) groupMessageEvent(c *client.QQClient, m *message.GroupMessage) {
|
||
bot.checkMedia(m.Elements)
|
||
for _, elem := range m.Elements {
|
||
if file, ok := elem.(*message.GroupFileElement); ok {
|
||
log.Infof("群 %v(%v) 内 %v(%v) 上传了文件: %v", m.GroupName, m.GroupCode, m.Sender.DisplayName(), m.Sender.Uin, file.Name)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "group_upload",
|
||
"group_id": m.GroupCode,
|
||
"user_id": m.Sender.Uin,
|
||
"file": MSG{
|
||
"id": file.Path,
|
||
"name": file.Name,
|
||
"size": file.Size,
|
||
"busid": file.Busid,
|
||
"url": c.GetGroupFileUrl(m.GroupCode, file.Path, file.Busid),
|
||
},
|
||
"self_id": c.Uin,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
return
|
||
}
|
||
}
|
||
cqm := ToStringMessage(m.Elements, m.GroupCode, true)
|
||
id := m.Id
|
||
if bot.db != nil {
|
||
id = bot.InsertGroupMessage(m)
|
||
}
|
||
log.Infof("收到群 %v(%v) 内 %v(%v) 的消息: %v (%v)", m.GroupName, m.GroupCode, m.Sender.DisplayName(), m.Sender.Uin, cqm, id)
|
||
gm := bot.formatGroupMessage(m)
|
||
if gm == nil {
|
||
return
|
||
}
|
||
gm["message_id"] = id
|
||
bot.dispatchEventMessage(gm)
|
||
if m.Sender.Uin != c.Uin {
|
||
c.MarkGroupMessageReaded(m.GroupCode, int64(m.Id))
|
||
}
|
||
}
|
||
|
||
func (bot *CQBot) tempMessageEvent(c *client.QQClient, e *client.TempMessageEvent) {
|
||
m := e.Message
|
||
bot.checkMedia(m.Elements)
|
||
cqm := ToStringMessage(m.Elements, 0, true)
|
||
bot.tempSessionCache.Store(m.Sender.Uin, e.Session)
|
||
id := m.Id
|
||
if bot.db != nil {
|
||
id = bot.InsertTempMessage(m.Sender.Uin, m)
|
||
}
|
||
log.Infof("收到来自群 %v(%v) 内 %v(%v) 的临时会话消息: %v", m.GroupName, m.GroupCode, m.Sender.DisplayName(), m.Sender.Uin, cqm)
|
||
tm := MSG{
|
||
"post_type": "message",
|
||
"message_type": "private",
|
||
"sub_type": "group",
|
||
"temp_source": e.Session.Source,
|
||
"message_id": id,
|
||
"user_id": m.Sender.Uin,
|
||
"message": ToFormattedMessage(m.Elements, 0, false),
|
||
"raw_message": cqm,
|
||
"font": 0,
|
||
"self_id": c.Uin,
|
||
"time": time.Now().Unix(),
|
||
"sender": MSG{
|
||
"user_id": m.Sender.Uin,
|
||
"group_id": m.GroupCode,
|
||
"nickname": m.Sender.Nickname,
|
||
"sex": "unknown",
|
||
"age": 0,
|
||
},
|
||
}
|
||
bot.dispatchEventMessage(tm)
|
||
}
|
||
|
||
func (bot *CQBot) groupMutedEvent(c *client.QQClient, e *client.GroupMuteEvent) {
|
||
g := c.FindGroup(e.GroupCode)
|
||
if e.TargetUin == 0 {
|
||
if e.Time != 0 {
|
||
log.Infof("群 %v 被 %v 开启全员禁言.",
|
||
formatGroupName(g), formatMemberName(g.FindMember(e.OperatorUin)))
|
||
} else {
|
||
log.Infof("群 %v 被 %v 解除全员禁言.",
|
||
formatGroupName(g), formatMemberName(g.FindMember(e.OperatorUin)))
|
||
}
|
||
} else {
|
||
if e.Time > 0 {
|
||
log.Infof("群 %v 内 %v 被 %v 禁言了 %v 秒.",
|
||
formatGroupName(g), formatMemberName(g.FindMember(e.TargetUin)), formatMemberName(g.FindMember(e.OperatorUin)), e.Time)
|
||
} else {
|
||
log.Infof("群 %v 内 %v 被 %v 解除禁言.",
|
||
formatGroupName(g), formatMemberName(g.FindMember(e.TargetUin)), formatMemberName(g.FindMember(e.OperatorUin)))
|
||
}
|
||
}
|
||
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"duration": e.Time,
|
||
"group_id": e.GroupCode,
|
||
"notice_type": "group_ban",
|
||
"operator_id": e.OperatorUin,
|
||
"self_id": c.Uin,
|
||
"user_id": e.TargetUin,
|
||
"time": time.Now().Unix(),
|
||
"sub_type": func() string {
|
||
if e.Time == 0 {
|
||
return "lift_ban"
|
||
}
|
||
return "ban"
|
||
}(),
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) groupRecallEvent(c *client.QQClient, e *client.GroupMessageRecalledEvent) {
|
||
g := c.FindGroup(e.GroupCode)
|
||
gid := toGlobalID(e.GroupCode, e.MessageId)
|
||
log.Infof("群 %v 内 %v 撤回了 %v 的消息: %v.",
|
||
formatGroupName(g), formatMemberName(g.FindMember(e.OperatorUin)), formatMemberName(g.FindMember(e.AuthorUin)), gid)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"group_id": e.GroupCode,
|
||
"notice_type": "group_recall",
|
||
"self_id": c.Uin,
|
||
"user_id": e.AuthorUin,
|
||
"operator_id": e.OperatorUin,
|
||
"time": e.Time,
|
||
"message_id": gid,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) groupNotifyEvent(c *client.QQClient, e client.INotifyEvent) {
|
||
group := c.FindGroup(e.From())
|
||
switch notify := e.(type) {
|
||
case *client.GroupPokeNotifyEvent:
|
||
sender := group.FindMember(notify.Sender)
|
||
receiver := group.FindMember(notify.Receiver)
|
||
log.Infof("群 %v 内 %v 戳了戳 %v", formatGroupName(group), formatMemberName(sender), formatMemberName(receiver))
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"group_id": group.Code,
|
||
"notice_type": "notify",
|
||
"sub_type": "poke",
|
||
"self_id": c.Uin,
|
||
"user_id": notify.Sender,
|
||
"sender_id": notify.Sender,
|
||
"target_id": notify.Receiver,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
case *client.GroupRedBagLuckyKingNotifyEvent:
|
||
sender := group.FindMember(notify.Sender)
|
||
luckyKing := group.FindMember(notify.LuckyKing)
|
||
log.Infof("群 %v 内 %v 的红包被抢完, %v 是运气王", formatGroupName(group), formatMemberName(sender), formatMemberName(luckyKing))
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"group_id": group.Code,
|
||
"notice_type": "notify",
|
||
"sub_type": "lucky_king",
|
||
"self_id": c.Uin,
|
||
"user_id": notify.Sender,
|
||
"sender_id": notify.Sender,
|
||
"target_id": notify.LuckyKing,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
case *client.MemberHonorChangedNotifyEvent:
|
||
log.Info(notify.Content())
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"group_id": group.Code,
|
||
"notice_type": "notify",
|
||
"sub_type": "honor",
|
||
"self_id": c.Uin,
|
||
"user_id": notify.Uin,
|
||
"time": time.Now().Unix(),
|
||
"honor_type": func() string {
|
||
switch notify.Honor {
|
||
case client.Talkative:
|
||
return "talkative"
|
||
case client.Performer:
|
||
return "performer"
|
||
case client.Emotion:
|
||
return "emotion"
|
||
case client.Legend:
|
||
return "legend"
|
||
case client.StrongNewbie:
|
||
return "strong_newbie"
|
||
default:
|
||
return "ERROR"
|
||
}
|
||
}(),
|
||
})
|
||
}
|
||
}
|
||
|
||
func (bot *CQBot) friendNotifyEvent(c *client.QQClient, e client.INotifyEvent) {
|
||
friend := c.FindFriend(e.From())
|
||
if notify, ok := e.(*client.FriendPokeNotifyEvent); ok {
|
||
if notify.Receiver == notify.Sender {
|
||
log.Infof("好友 %v 戳了戳自己.", friend.Nickname)
|
||
} else {
|
||
log.Infof("好友 %v 戳了戳你.", friend.Nickname)
|
||
}
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "notify",
|
||
"sub_type": "poke",
|
||
"self_id": c.Uin,
|
||
"user_id": notify.Sender,
|
||
"sender_id": notify.Sender,
|
||
"target_id": notify.Receiver,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
}
|
||
}
|
||
|
||
func (bot *CQBot) memberTitleUpdatedEvent(c *client.QQClient, e *client.MemberSpecialTitleUpdatedEvent) {
|
||
group := c.FindGroup(e.GroupCode)
|
||
mem := group.FindMember(e.Uin)
|
||
log.Infof("群 %v(%v) 内成员 %v(%v) 获得了新的头衔: %v", group.Name, group.Code, mem.DisplayName(), mem.Uin, e.NewTitle)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "notify",
|
||
"sub_type": "title",
|
||
"self_id": c.Uin,
|
||
"user_id": e.Uin,
|
||
"time": time.Now().Unix(),
|
||
"title": e.NewTitle,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) friendRecallEvent(c *client.QQClient, e *client.FriendMessageRecalledEvent) {
|
||
f := c.FindFriend(e.FriendUin)
|
||
gid := toGlobalID(e.FriendUin, e.MessageId)
|
||
if f != nil {
|
||
log.Infof("好友 %v(%v) 撤回了消息: %v", f.Nickname, f.Uin, gid)
|
||
} else {
|
||
log.Infof("好友 %v 撤回了消息: %v", e.FriendUin, gid)
|
||
}
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "friend_recall",
|
||
"self_id": c.Uin,
|
||
"user_id": e.FriendUin,
|
||
"time": e.Time,
|
||
"message_id": gid,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) offlineFileEvent(c *client.QQClient, e *client.OfflineFileEvent) {
|
||
f := c.FindFriend(e.Sender)
|
||
if f == nil {
|
||
return
|
||
}
|
||
log.Infof("好友 %v(%v) 发送了离线文件 %v", f.Nickname, f.Uin, e.FileName)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "offline_file",
|
||
"user_id": e.Sender,
|
||
"file": MSG{
|
||
"name": e.FileName,
|
||
"size": e.FileSize,
|
||
"url": e.DownloadUrl,
|
||
},
|
||
"self_id": c.Uin,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) joinGroupEvent(c *client.QQClient, group *client.GroupInfo) {
|
||
log.Infof("Bot进入了群 %v.", formatGroupName(group))
|
||
bot.dispatchEventMessage(bot.groupIncrease(group.Code, 0, c.Uin))
|
||
}
|
||
|
||
func (bot *CQBot) leaveGroupEvent(c *client.QQClient, e *client.GroupLeaveEvent) {
|
||
if e.Operator != nil {
|
||
log.Infof("Bot被 %v T出了群 %v.", formatMemberName(e.Operator), formatGroupName(e.Group))
|
||
} else {
|
||
log.Infof("Bot退出了群 %v.", formatGroupName(e.Group))
|
||
}
|
||
bot.dispatchEventMessage(bot.groupDecrease(e.Group.Code, c.Uin, e.Operator))
|
||
}
|
||
|
||
func (bot *CQBot) memberPermissionChangedEvent(c *client.QQClient, e *client.MemberPermissionChangedEvent) {
|
||
st := func() string {
|
||
if e.NewPermission == client.Administrator {
|
||
return "set"
|
||
}
|
||
return "unset"
|
||
}()
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "group_admin",
|
||
"sub_type": st,
|
||
"group_id": e.Group.Code,
|
||
"user_id": e.Member.Uin,
|
||
"time": time.Now().Unix(),
|
||
"self_id": c.Uin,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) memberCardUpdatedEvent(c *client.QQClient, e *client.MemberCardUpdatedEvent) {
|
||
log.Infof("群 %v 的 %v 更新了名片 %v -> %v", formatGroupName(e.Group), formatMemberName(e.Member), e.OldCard, e.Member.CardName)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "group_card",
|
||
"group_id": e.Group.Code,
|
||
"user_id": e.Member.Uin,
|
||
"card_new": e.Member.CardName,
|
||
"card_old": e.OldCard,
|
||
"time": time.Now().Unix(),
|
||
"self_id": c.Uin,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) memberJoinEvent(_ *client.QQClient, e *client.MemberJoinGroupEvent) {
|
||
log.Infof("新成员 %v 进入了群 %v.", formatMemberName(e.Member), formatGroupName(e.Group))
|
||
bot.dispatchEventMessage(bot.groupIncrease(e.Group.Code, 0, e.Member.Uin))
|
||
}
|
||
|
||
func (bot *CQBot) memberLeaveEvent(_ *client.QQClient, e *client.MemberLeaveGroupEvent) {
|
||
if e.Operator != nil {
|
||
log.Infof("成员 %v 被 %v T出了群 %v.", formatMemberName(e.Member), formatMemberName(e.Operator), formatGroupName(e.Group))
|
||
} else {
|
||
log.Infof("成员 %v 离开了群 %v.", formatMemberName(e.Member), formatGroupName(e.Group))
|
||
}
|
||
bot.dispatchEventMessage(bot.groupDecrease(e.Group.Code, e.Member.Uin, e.Operator))
|
||
}
|
||
|
||
func (bot *CQBot) friendRequestEvent(c *client.QQClient, e *client.NewFriendRequest) {
|
||
log.Infof("收到来自 %v(%v) 的好友请求: %v", e.RequesterNick, e.RequesterUin, e.Message)
|
||
flag := strconv.FormatInt(e.RequestId, 10)
|
||
bot.friendReqCache.Store(flag, e)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "request",
|
||
"request_type": "friend",
|
||
"user_id": e.RequesterUin,
|
||
"comment": e.Message,
|
||
"flag": flag,
|
||
"time": time.Now().Unix(),
|
||
"self_id": c.Uin,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) friendAddedEvent(c *client.QQClient, e *client.NewFriendEvent) {
|
||
log.Infof("添加了新好友: %v(%v)", e.Friend.Nickname, e.Friend.Uin)
|
||
bot.tempSessionCache.Delete(e.Friend.Uin)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "friend_add",
|
||
"self_id": c.Uin,
|
||
"user_id": e.Friend.Uin,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) groupInvitedEvent(c *client.QQClient, e *client.GroupInvitedRequest) {
|
||
log.Infof("收到来自群 %v(%v) 内用户 %v(%v) 的加群邀请.", e.GroupName, e.GroupCode, e.InvitorNick, e.InvitorUin)
|
||
flag := strconv.FormatInt(e.RequestId, 10)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "request",
|
||
"request_type": "group",
|
||
"sub_type": "invite",
|
||
"group_id": e.GroupCode,
|
||
"user_id": e.InvitorUin,
|
||
"comment": "",
|
||
"flag": flag,
|
||
"time": time.Now().Unix(),
|
||
"self_id": c.Uin,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) groupJoinReqEvent(c *client.QQClient, e *client.UserJoinGroupRequest) {
|
||
log.Infof("群 %v(%v) 收到来自用户 %v(%v) 的加群请求.", e.GroupName, e.GroupCode, e.RequesterNick, e.RequesterUin)
|
||
flag := strconv.FormatInt(e.RequestId, 10)
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "request",
|
||
"request_type": "group",
|
||
"sub_type": "add",
|
||
"group_id": e.GroupCode,
|
||
"user_id": e.RequesterUin,
|
||
"comment": e.Message,
|
||
"flag": flag,
|
||
"time": time.Now().Unix(),
|
||
"self_id": c.Uin,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) otherClientStatusChangedEvent(c *client.QQClient, e *client.OtherClientStatusChangedEvent) {
|
||
if e.Online {
|
||
log.Infof("Bot 账号在客户端 %v (%v) 登录.", e.Client.DeviceName, e.Client.DeviceKind)
|
||
} else {
|
||
log.Infof("Bot 账号在客户端 %v (%v) 登出.", e.Client.DeviceName, e.Client.DeviceKind)
|
||
}
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "client_status",
|
||
"online": e.Online,
|
||
"client": MSG{
|
||
"app_id": e.Client.AppId,
|
||
"device_name": e.Client.DeviceName,
|
||
"device_kind": e.Client.DeviceKind,
|
||
},
|
||
"self_id": c.Uin,
|
||
"time": time.Now().Unix(),
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) groupEssenceMsg(c *client.QQClient, e *client.GroupDigestEvent) {
|
||
g := c.FindGroup(e.GroupCode)
|
||
gid := toGlobalID(e.GroupCode, e.MessageID)
|
||
if e.OperationType == 1 {
|
||
log.Infof(
|
||
"群 %v 内 %v 将 %v 的消息(%v)设为了精华消息.",
|
||
formatGroupName(g),
|
||
formatMemberName(g.FindMember(e.OperatorUin)),
|
||
formatMemberName(g.FindMember(e.SenderUin)),
|
||
gid,
|
||
)
|
||
} else {
|
||
log.Infof(
|
||
"群 %v 内 %v 将 %v 的消息(%v)移出了精华消息.",
|
||
formatGroupName(g),
|
||
formatMemberName(g.FindMember(e.OperatorUin)),
|
||
formatMemberName(g.FindMember(e.SenderUin)),
|
||
gid,
|
||
)
|
||
}
|
||
if e.OperatorUin == bot.Client.Uin {
|
||
return
|
||
}
|
||
bot.dispatchEventMessage(MSG{
|
||
"post_type": "notice",
|
||
"group_id": e.GroupCode,
|
||
"notice_type": "essence",
|
||
"sub_type": func() string {
|
||
if e.OperationType == 1 {
|
||
return "add"
|
||
}
|
||
return "delete"
|
||
}(),
|
||
"self_id": c.Uin,
|
||
"sender_id": e.SenderUin,
|
||
"operator_id": e.OperatorUin,
|
||
"time": time.Now().Unix(),
|
||
"message_id": gid,
|
||
})
|
||
}
|
||
|
||
func (bot *CQBot) groupIncrease(groupCode, operatorUin, userUin int64) MSG {
|
||
return MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "group_increase",
|
||
"group_id": groupCode,
|
||
"operator_id": operatorUin,
|
||
"self_id": bot.Client.Uin,
|
||
"sub_type": "approve",
|
||
"time": time.Now().Unix(),
|
||
"user_id": userUin,
|
||
}
|
||
}
|
||
|
||
func (bot *CQBot) groupDecrease(groupCode, userUin int64, operator *client.GroupMemberInfo) MSG {
|
||
return MSG{
|
||
"post_type": "notice",
|
||
"notice_type": "group_decrease",
|
||
"group_id": groupCode,
|
||
"operator_id": func() int64 {
|
||
if operator != nil {
|
||
return operator.Uin
|
||
}
|
||
return userUin
|
||
}(),
|
||
"self_id": bot.Client.Uin,
|
||
"sub_type": func() string {
|
||
if operator != nil {
|
||
if userUin == bot.Client.Uin {
|
||
return "kick_me"
|
||
}
|
||
return "kick"
|
||
}
|
||
return "leave"
|
||
}(),
|
||
"time": time.Now().Unix(),
|
||
"user_id": userUin,
|
||
}
|
||
}
|
||
|
||
func (bot *CQBot) checkMedia(e []message.IMessageElement) {
|
||
for _, elem := range e {
|
||
switch i := elem.(type) {
|
||
case *message.GroupImageElement:
|
||
filename := hex.EncodeToString(i.Md5) + ".image"
|
||
if !global.PathExists(path.Join(global.ImagePath, filename)) {
|
||
_ = os.WriteFile(path.Join(global.ImagePath, filename), binary.NewWriterF(func(w *binary.Writer) {
|
||
w.Write(i.Md5)
|
||
w.WriteUInt32(uint32(i.Size))
|
||
w.WriteString(i.ImageId)
|
||
w.WriteString(i.Url)
|
||
}), 0o644)
|
||
}
|
||
case *message.FriendImageElement:
|
||
filename := hex.EncodeToString(i.Md5) + ".image"
|
||
if !global.PathExists(path.Join(global.ImagePath, filename)) {
|
||
_ = os.WriteFile(path.Join(global.ImagePath, filename), binary.NewWriterF(func(w *binary.Writer) {
|
||
w.Write(i.Md5)
|
||
w.WriteUInt32(uint32(i.Size))
|
||
w.WriteString(i.ImageId)
|
||
w.WriteString(i.Url)
|
||
}), 0o644)
|
||
}
|
||
case *message.VoiceElement:
|
||
i.Name = strings.ReplaceAll(i.Name, "{", "")
|
||
i.Name = strings.ReplaceAll(i.Name, "}", "")
|
||
if !global.PathExists(path.Join(global.VoicePath, i.Name)) {
|
||
b, err := global.GetBytes(i.Url)
|
||
if err != nil {
|
||
log.Warnf("语音文件 %v 下载失败: %v", i.Name, err)
|
||
continue
|
||
}
|
||
_ = os.WriteFile(path.Join(global.VoicePath, i.Name), b, 0o644)
|
||
}
|
||
case *message.ShortVideoElement:
|
||
filename := hex.EncodeToString(i.Md5) + ".video"
|
||
if !global.PathExists(path.Join(global.VideoPath, filename)) {
|
||
_ = os.WriteFile(path.Join(global.VideoPath, filename), binary.NewWriterF(func(w *binary.Writer) {
|
||
w.Write(i.Md5)
|
||
w.Write(i.ThumbMd5)
|
||
w.WriteUInt32(uint32(i.Size))
|
||
w.WriteUInt32(uint32(i.ThumbSize))
|
||
w.WriteString(i.Name)
|
||
w.Write(i.Uuid)
|
||
}), 0o644)
|
||
}
|
||
i.Name = filename
|
||
i.Url = bot.Client.GetShortVideoUrl(i.Uuid, i.Md5)
|
||
}
|
||
}
|
||
}
|