1
0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-04 19:17:38 +08:00
MiraiGo/message/message.go
icarus-ai e700751008
解析猜拳&随机骰子
安卓QQ发来的骰子Name值是随机骰子
2021-12-18 17:35:32 +08:00

627 lines
17 KiB
Go

package message
import (
"encoding/json"
"reflect"
"strconv"
"strings"
"github.com/Mrs4s/MiraiGo/binary"
"github.com/Mrs4s/MiraiGo/client/pb/msg"
"github.com/Mrs4s/MiraiGo/internal/proto"
"github.com/Mrs4s/MiraiGo/utils"
)
type (
PrivateMessage struct {
Id int32
InternalId int32
Self int64
Target int64
Time int32
Sender *Sender
Elements []IMessageElement
}
TempMessage struct {
Id int32
GroupCode int64
GroupName string
Self int64
Sender *Sender
Elements []IMessageElement
}
GroupMessage struct {
Id int32
InternalId int32
GroupCode int64
GroupName string
Sender *Sender
Time int32
Elements []IMessageElement
OriginalObject *msg.Message
// OriginalElements []*msg.Elem
}
SendingMessage struct {
Elements []IMessageElement
}
Sender struct {
Uin int64
Nickname string
CardName string
AnonymousInfo *AnonymousInfo
IsFriend bool
}
AnonymousInfo struct {
AnonymousId string
AnonymousNick string
}
IMessageElement interface {
Type() ElementType
}
IRichMessageElement interface {
Pack() []*msg.Elem
}
ElementType int
)
// MusicType values.
const (
QQMusic = iota // QQ音乐
CloudMusic // 网易云音乐
MiguMusic // 咪咕音乐
KugouMusic // 酷狗音乐
KuwoMusic // 酷我音乐
)
//go:generate stringer -type ElementType -linecomment
const (
Text ElementType = iota // 文本
Image // 图片
Face // 表情
At // 艾特
Reply // 回复
Service // 服务
Forward // 转发
File // 文件
Voice // 语音
Video // 视频
LightApp // 轻应用
RedBag // 红包
)
func (s *Sender) IsAnonymous() bool {
return s.Uin == 80000000
}
func NewSendingMessage() *SendingMessage {
return &SendingMessage{}
}
func (msg *PrivateMessage) ToString() (res string) {
for _, elem := range msg.Elements {
switch e := elem.(type) {
case *TextElement:
res += e.Content
case *FaceElement:
res += "[" + e.Name + "]"
case *AtElement:
res += e.Display
}
}
return
}
func (msg *TempMessage) ToString() (res string) {
for _, elem := range msg.Elements {
switch e := elem.(type) {
case *TextElement:
res += e.Content
case *FaceElement:
res += "[" + e.Name + "]"
case *AtElement:
res += e.Display
}
}
return
}
func (msg *GroupMessage) ToString() (res string) {
for _, elem := range msg.Elements {
switch e := elem.(type) {
case *TextElement:
res += e.Content
case *FaceElement:
res += "[" + e.Name + "]"
case *GroupImageElement:
res += "[Image: " + e.ImageId + "]"
case *AtElement:
res += e.Display
case *RedBagElement:
res += "[RedBag:" + e.Title + "]"
case *ReplyElement:
res += "[Reply:" + strconv.FormatInt(int64(e.ReplySeq), 10) + "]"
}
}
return
}
func (msg *SendingMessage) Append(e IMessageElement) *SendingMessage {
v := reflect.ValueOf(e)
if v.Kind() == reflect.Ptr && !v.IsNil() {
msg.Elements = append(msg.Elements, e)
}
return msg
}
func (msg *SendingMessage) Any(filter func(e IMessageElement) bool) bool {
for _, e := range msg.Elements {
if filter(e) {
return true
}
}
return false
}
func (msg *SendingMessage) FirstOrNil(filter func(e IMessageElement) bool) IMessageElement {
for _, e := range msg.Elements {
if filter(e) {
return e
}
}
return nil
}
func (msg *SendingMessage) Count(filter func(e IMessageElement) bool) (c int) {
for _, e := range msg.Elements {
if filter(e) {
c++
}
}
return
}
func (msg *SendingMessage) ToFragmented() [][]IMessageElement {
var fragmented [][]IMessageElement
for _, elem := range msg.Elements {
switch o := elem.(type) {
case *TextElement:
for _, text := range utils.ChunkString(o.Content, 80) {
fragmented = append(fragmented, []IMessageElement{NewText(text)})
}
default:
fragmented = append(fragmented, []IMessageElement{o})
}
}
return fragmented
}
func EstimateLength(elems []IMessageElement) int {
sum := 0
for _, elem := range elems {
switch e := elem.(type) {
case *TextElement:
sum += len(e.Content)
case *AtElement:
sum += len(e.Display)
case *ReplyElement:
sum += 444 + EstimateLength(e.Elements)
case *GroupImageElement, *FriendImageElement:
sum += 100
default:
sum += len(ToReadableString([]IMessageElement{elem}))
}
}
return sum
}
func (s *Sender) DisplayName() string {
if s.CardName == "" {
return s.Nickname
}
return s.CardName
}
func ToProtoElems(elems []IMessageElement, generalFlags bool) (r []*msg.Elem) {
if len(elems) == 0 {
return nil
}
for _, elem := range elems {
if reply, ok := elem.(*ReplyElement); ok {
r = append(r, &msg.Elem{
SrcMsg: &msg.SourceMsg{
OrigSeqs: []int32{reply.ReplySeq},
SenderUin: &reply.Sender,
Time: &reply.Time,
Flag: proto.Int32(1),
Elems: ToSrcProtoElems(reply.Elements),
RichMsg: []byte{},
PbReserve: []byte{},
SrcMsg: []byte{},
TroopName: []byte{},
},
})
if len(elems) > 1 {
if elems[0].Type() == Image || elems[1].Type() == Image {
r = append(r, &msg.Elem{Text: &msg.Text{Str: proto.String(" ")}})
}
}
}
}
for _, elem := range elems {
if e, ok := elem.(*ShortVideoElement); ok {
r = e.Pack()
break
}
if e, ok := elem.(IRichMessageElement); ok {
r = append(r, e.Pack()...)
}
}
if generalFlags {
L:
for _, elem := range elems {
switch e := elem.(type) {
case *ServiceElement:
if e.SubType == "Long" {
r = append(r, &msg.Elem{
GeneralFlags: &msg.GeneralFlags{
LongTextFlag: proto.Int32(1),
LongTextResid: &e.ResId,
PbReserve: []byte{0x78, 0x00, 0xF8, 0x01, 0x00, 0xC8, 0x02, 0x00},
},
})
break L
}
// d, _ := hex.DecodeString("08097800C80100F00100F80100900200C80200980300A00320B00300C00300D00300E803008A04020803900480808010B80400C00400")
r = append(r, &msg.Elem{
GeneralFlags: &msg.GeneralFlags{
PbReserve: []byte{
0x08, 0x09, 0x78, 0x00, 0xC8, 0x01, 0x00, 0xF0, 0x01, 0x00, 0xF8, 0x01, 0x00, 0x90, 0x02, 0x00,
0xC8, 0x02, 0x00, 0x98, 0x03, 0x00, 0xA0, 0x03, 0x20, 0xB0, 0x03, 0x00, 0xC0, 0x03, 0x00, 0xD0,
0x03, 0x00, 0xE8, 0x03, 0x00, 0x8A, 0x04, 0x02, 0x08, 0x03, 0x90, 0x04, 0x80, 0x80, 0x80, 0x10,
0xB8, 0x04, 0x00, 0xC0, 0x04, 0x00,
},
},
})
break L
}
}
}
return
}
var photoTextElem IMessageElement = NewText("[图片]")
func ToSrcProtoElems(elems []IMessageElement) []*msg.Elem {
elems2 := make([]IMessageElement, len(elems))
copy(elems2, elems)
for i, elem := range elems2 {
if elem.Type() == Image {
elems2[i] = photoTextElem
}
}
return ToProtoElems(elems2, false)
}
func ParseMessageElems(elems []*msg.Elem) []IMessageElement {
var res []IMessageElement
for _, elem := range elems {
if elem.SrcMsg != nil && len(elem.SrcMsg.OrigSeqs) != 0 {
r := &ReplyElement{
ReplySeq: elem.SrcMsg.OrigSeqs[0],
Time: elem.SrcMsg.GetTime(),
Sender: elem.SrcMsg.GetSenderUin(),
GroupID: elem.SrcMsg.GetToUin(),
Elements: ParseMessageElems(elem.SrcMsg.Elems),
}
res = append(res, r)
}
if elem.TransElemInfo != nil {
if elem.TransElemInfo.GetElemType() == 24 { // QFile
i3 := len(elem.TransElemInfo.ElemValue)
r := binary.NewReader(elem.TransElemInfo.ElemValue)
if i3 > 3 {
if r.ReadByte() == 1 {
pb := r.ReadBytes(int(r.ReadUInt16()))
objMsg := msg.ObjMsg{}
if err := proto.Unmarshal(pb, &objMsg); err == nil && len(objMsg.MsgContentInfo) > 0 {
info := objMsg.MsgContentInfo[0]
res = append(res, &GroupFileElement{
Name: info.MsgFile.FileName,
Size: info.MsgFile.FileSize,
Path: string(info.MsgFile.FilePath),
Busid: info.MsgFile.BusId,
})
}
}
}
}
}
if elem.LightApp != nil && len(elem.LightApp.Data) > 1 {
var content []byte
if elem.LightApp.Data[0] == 0 {
content = elem.LightApp.Data[1:]
}
if elem.LightApp.Data[0] == 1 {
content = binary.ZlibUncompress(elem.LightApp.Data[1:])
}
if len(content) > 0 && len(content) < 1024*1024*1024 { // 解析出错 or 非法内容
// TODO: 解析具体的APP
return append(res, &LightAppElement{Content: string(content)})
}
}
if elem.VideoFile != nil {
return []IMessageElement{
&ShortVideoElement{
Name: string(elem.VideoFile.FileName),
Uuid: elem.VideoFile.FileUuid,
Size: elem.VideoFile.GetFileSize(),
ThumbSize: elem.VideoFile.GetThumbFileSize(),
Md5: elem.VideoFile.FileMd5,
ThumbMd5: elem.VideoFile.GetThumbFileMd5(),
},
}
}
if elem.Text != nil {
switch {
case len(elem.Text.Attr6Buf) > 0:
att6 := binary.NewReader(elem.Text.Attr6Buf)
att6.ReadBytes(7)
target := int64(uint32(att6.ReadInt32()))
at := NewAt(target, elem.Text.GetStr())
at.SubType = AtTypeGroupMember
res = append(res, at)
case len(elem.Text.PbReserve) > 0:
resv := new(msg.TextResvAttr)
_ = proto.Unmarshal(elem.Text.PbReserve, resv)
if resv.GetAtType() == 2 {
at := NewAt(int64(resv.GetAtMemberTinyid()), elem.Text.GetStr())
at.SubType = AtTypeGuildMember
res = append(res, at)
break
}
if resv.GetAtType() == 4 {
at := NewAt(int64(resv.AtChannelInfo.GetChannelId()), elem.Text.GetStr())
at.SubType = AtTypeGuildChannel
res = append(res, at)
break
}
fallthrough
default:
res = append(res, NewText(func() string {
// 这么处理应该没问题
if strings.Contains(elem.Text.GetStr(), "\r") && !strings.Contains(elem.Text.GetStr(), "\r\n") {
return strings.ReplaceAll(elem.Text.GetStr(), "\r", "\r\n")
}
return elem.Text.GetStr()
}()))
}
}
if elem.RichMsg != nil {
var content string
if elem.RichMsg.Template1[0] == 0 {
content = string(elem.RichMsg.Template1[1:])
}
if elem.RichMsg.Template1[0] == 1 {
content = string(binary.ZlibUncompress(elem.RichMsg.Template1[1:]))
}
if content != "" {
if elem.RichMsg.GetServiceId() == 35 {
elem := forwardMsgFromXML(content)
if elem != nil {
res = append(res, elem)
continue
}
}
if elem.RichMsg.GetServiceId() == 33 {
continue // 前面一个 elem 已经解析到链接
}
if isOk := strings.Contains(content, "<?xml"); isOk {
res = append(res, NewRichXml(content, int64(elem.RichMsg.GetServiceId())))
continue
} else if json.Valid(utils.S2B(content)) {
res = append(res, NewRichJson(content))
continue
}
res = append(res, NewText(content))
}
}
if elem.CustomFace != nil {
if len(elem.CustomFace.Md5) == 0 {
continue
}
var url string
if elem.CustomFace.GetOrigUrl() == "" {
url = "https://gchat.qpic.cn/gchatpic_new/0/0-0-" + strings.ReplaceAll(binary.CalculateImageResourceId(elem.CustomFace.Md5)[1:37], "-", "") + "/0?term=2"
} else {
url = "https://gchat.qpic.cn" + elem.CustomFace.GetOrigUrl()
}
if strings.Contains(elem.CustomFace.GetOrigUrl(), "qmeet") {
res = append(res, &GuildImageElement{
FileId: int64(elem.CustomFace.GetFileId()),
FilePath: elem.CustomFace.GetFilePath(),
Size: elem.CustomFace.GetSize(),
Width: elem.CustomFace.GetWidth(),
Height: elem.CustomFace.GetHeight(),
Url: url,
Md5: elem.CustomFace.Md5,
})
continue
}
res = append(res, &GroupImageElement{
FileId: int64(elem.CustomFace.GetFileId()),
ImageId: elem.CustomFace.GetFilePath(),
Size: elem.CustomFace.GetSize(),
Width: elem.CustomFace.GetWidth(),
Height: elem.CustomFace.GetHeight(),
Url: url,
ImageBizType: func() ImageBizType {
if len(elem.CustomFace.PbReserve) == 0 {
return UnknownBizType
}
attr := new(msg.ResvAttr)
if proto.Unmarshal(elem.CustomFace.PbReserve, attr) != nil {
return UnknownBizType
}
return ImageBizType(attr.GetImageBizType())
}(),
Md5: elem.CustomFace.Md5,
})
}
if elem.MarketFace != nil {
face := &MarketFaceElement{
Name: utils.B2S(elem.MarketFace.GetFaceName()),
FaceId: elem.MarketFace.FaceId,
TabId: int32(elem.MarketFace.GetTabId()),
ItemType: int32(elem.MarketFace.GetItemType()),
SubType: int32(elem.MarketFace.GetSubType()),
MediaType: int32(elem.MarketFace.GetMediaType()),
EncryptKey: elem.MarketFace.GetKey(),
MagicValue: utils.B2S(elem.MarketFace.Mobileparam),
}
if face.Name == "[骰子]" || face.Name == "[随机骰子]" {
return []IMessageElement{
&DiceElement{
MarketFaceElement: face,
Value: func() int32 {
v := strings.SplitN(face.MagicValue, "=", 2)[1]
t, _ := strconv.ParseInt(v, 10, 32)
return int32(t) + 1
}(),
},
}
}
if face.Name == "[猜拳]" {
v := strings.SplitN(face.MagicValue, "=", 2)[1]
t, _ := strconv.ParseInt(v, 10, 32)
return []IMessageElement{
&FingerGuessingElement{
MarketFaceElement: face,
Value: int32(t),
Name : fingerGuessingName[int32(t)],
},
}
}
return []IMessageElement{face}
}
if elem.NotOnlineImage != nil {
var img string
if elem.NotOnlineImage.GetOrigUrl() != "" {
img = "https://c2cpicdw.qpic.cn" + elem.NotOnlineImage.GetOrigUrl()
} else {
img = "https://c2cpicdw.qpic.cn/offpic_new/0"
downloadPath := elem.NotOnlineImage.GetResId()
if elem.NotOnlineImage.GetDownloadPath() != "" {
downloadPath = elem.NotOnlineImage.GetDownloadPath()
}
if !strings.HasPrefix(downloadPath, "/") {
img += "/"
}
img += downloadPath + "/0?term=3"
}
res = append(res, &FriendImageElement{
ImageId: elem.NotOnlineImage.GetFilePath(),
Size: elem.NotOnlineImage.GetFileLen(),
Url: img,
Md5: elem.NotOnlineImage.PicMd5,
})
}
if elem.QQWalletMsg != nil && elem.QQWalletMsg.AioBody != nil {
// /com/tencent/mobileqq/data/MessageForQQWalletMsg.java#L366
msgType := elem.QQWalletMsg.AioBody.GetMsgType()
if msgType <= 1000 && elem.QQWalletMsg.AioBody.RedType != nil {
return []IMessageElement{
&RedBagElement{
MsgType: RedBagMessageType(msgType),
Title: elem.QQWalletMsg.AioBody.Receiver.GetTitle(),
},
}
}
}
if elem.Face != nil {
res = append(res, NewFace(elem.Face.GetIndex()))
}
if elem.CommonElem != nil {
switch elem.CommonElem.GetServiceType() {
case 3:
flash := &msg.MsgElemInfoServtype3{}
_ = proto.Unmarshal(elem.CommonElem.PbElem, flash)
if flash.FlashTroopPic != nil {
res = append(res, &GroupImageElement{
FileId: int64(flash.FlashTroopPic.GetFileId()),
ImageId: flash.FlashTroopPic.GetFilePath(),
Size: flash.FlashTroopPic.GetSize(),
Width: flash.FlashTroopPic.GetWidth(),
Height: flash.FlashTroopPic.GetHeight(),
Md5: flash.FlashTroopPic.Md5,
Flash: true,
})
return res
}
if flash.FlashC2CPic != nil {
res = append(res, &FriendImageElement{
ImageId: flash.FlashC2CPic.GetFilePath(),
Size: flash.FlashC2CPic.GetFileLen(),
Md5: flash.FlashC2CPic.PicMd5,
Flash: true,
})
return res
}
case 33:
newSysFaceMsg := &msg.MsgElemInfoServtype33{}
_ = proto.Unmarshal(elem.CommonElem.PbElem, newSysFaceMsg)
res = append(res, NewFace(int32(newSysFaceMsg.GetIndex())))
case 37:
animatedStickerMsg := &msg.MsgElemInfoServtype37{}
_ = proto.Unmarshal(elem.CommonElem.PbElem, animatedStickerMsg)
sticker := &AnimatedSticker{
ID: int32(animatedStickerMsg.GetQsid()),
Name: strings.TrimPrefix(string(animatedStickerMsg.GetText()), "/"),
}
return []IMessageElement{sticker} // sticker 永远为单独消息
}
}
}
return res
}
func ToReadableString(m []IMessageElement) string {
sb := new(strings.Builder)
for _, elem := range m {
switch e := elem.(type) {
case *TextElement:
sb.WriteString(e.Content)
case *GroupImageElement, *FriendImageElement:
sb.WriteString("[图片]")
case *FaceElement:
sb.WriteByte('/')
sb.WriteString(e.Name)
case *ForwardElement:
sb.WriteString("[聊天记录]")
// NOTE: flash pic is singular
// To be clarified
// case *GroupFlashImgElement:
// return "[闪照]"
// case *FriendFlashImgElement:
// return "[闪照]"
case *AtElement:
sb.WriteString(e.Display)
}
}
return sb.String()
}
func FaceNameById(id int) string {
if name, ok := faceMap[id]; ok {
return name
}
return "未知表情"
}