mirror of
https://github.com/Mrs4s/MiraiGo.git
synced 2025-05-04 19:17:38 +08:00
291 lines
6.9 KiB
Go
291 lines
6.9 KiB
Go
package message
|
|
|
|
import (
|
|
"github.com/Mrs4s/MiraiGo/binary"
|
|
"github.com/Mrs4s/MiraiGo/client/pb/msg"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type PrivateMessage struct {
|
|
Id int32
|
|
InternalId int32
|
|
Target int64
|
|
Time int32
|
|
Sender *Sender
|
|
Elements []IMessageElement
|
|
}
|
|
|
|
type TempMessage struct {
|
|
Id int32
|
|
GroupCode int64
|
|
GroupName string
|
|
Sender *Sender
|
|
Elements []IMessageElement
|
|
}
|
|
|
|
type GroupMessage struct {
|
|
Id int32
|
|
InternalId int32
|
|
GroupCode int64
|
|
GroupName string
|
|
Sender *Sender
|
|
Time int32
|
|
Elements []IMessageElement
|
|
//OriginalElements []*msg.Elem
|
|
}
|
|
|
|
type SendingMessage struct {
|
|
Elements []IMessageElement
|
|
}
|
|
|
|
type Sender struct {
|
|
Uin int64
|
|
Nickname string
|
|
CardName string
|
|
IsFriend bool
|
|
}
|
|
|
|
type IMessageElement interface {
|
|
Type() ElementType
|
|
}
|
|
|
|
type ElementType int
|
|
|
|
const (
|
|
Text ElementType = iota
|
|
Image
|
|
Face
|
|
At
|
|
Reply
|
|
)
|
|
|
|
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 *ImageElement:
|
|
res += "[Image:" + e.Filename + "]"
|
|
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 *ImageElement:
|
|
res += "[Image:" + e.Filename + "]"
|
|
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 *ImageElement:
|
|
res += "[Image:" + e.Filename + "]"
|
|
case *FaceElement:
|
|
res += "[" + e.Name + "]"
|
|
case *GroupImageElement:
|
|
res += "[Image: " + e.ImageId + "]"
|
|
case *AtElement:
|
|
res += e.Display
|
|
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 (s *Sender) DisplayName() string {
|
|
if s.CardName == "" {
|
|
return s.Nickname
|
|
}
|
|
return s.CardName
|
|
}
|
|
|
|
func ToProtoElems(elems []IMessageElement) (r []*msg.Elem) {
|
|
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: 1,
|
|
Elems: ToProtoElems(reply.Elements),
|
|
RichMsg: []byte{},
|
|
PbReserve: []byte{},
|
|
SrcMsg: []byte{},
|
|
TroopName: []byte{},
|
|
},
|
|
})
|
|
}
|
|
}
|
|
for _, elem := range elems {
|
|
switch e := elem.(type) {
|
|
case *TextElement:
|
|
r = append(r, &msg.Elem{
|
|
Text: &msg.Text{
|
|
Str: e.Content,
|
|
},
|
|
})
|
|
case *FaceElement:
|
|
r = append(r, &msg.Elem{
|
|
Face: &msg.Face{
|
|
Index: e.Index,
|
|
Old: binary.ToBytes(int16(0x1445 - 4 + e.Index)),
|
|
Buf: []byte{0x00, 0x01, 0x00, 0x04, 0x52, 0xCC, 0xF5, 0xD0},
|
|
},
|
|
})
|
|
case *AtElement:
|
|
r = append(r, &msg.Elem{
|
|
Text: &msg.Text{
|
|
Str: e.Display,
|
|
Attr6Buf: binary.NewWriterF(func(w *binary.Writer) {
|
|
w.WriteUInt16(1)
|
|
w.WriteUInt16(0)
|
|
w.WriteUInt16(uint16(len(e.Display)))
|
|
w.WriteByte(func() byte {
|
|
if e.Target == 0 {
|
|
return 1
|
|
}
|
|
return 0
|
|
}())
|
|
w.WriteUInt32(uint32(e.Target))
|
|
w.WriteUInt16(0)
|
|
}),
|
|
},
|
|
})
|
|
r = append(r, &msg.Elem{Text: &msg.Text{Str: " "}})
|
|
case *GroupImageElement:
|
|
r = append(r, &msg.Elem{
|
|
CustomFace: &msg.CustomFace{
|
|
FilePath: e.ImageId,
|
|
Md5: e.Md5[:],
|
|
Flag: make([]byte, 4),
|
|
OldData: []byte{0x15, 0x36, 0x20, 0x39, 0x32, 0x6B, 0x41, 0x31, 0x00, 0x38, 0x37, 0x32, 0x66, 0x30, 0x36, 0x36, 0x30, 0x33, 0x61, 0x65, 0x31, 0x30, 0x33, 0x62, 0x37, 0x20, 0x20, 0x20, 0x20, 0x20,
|
|
0x20, 0x35, 0x30, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x7B, 0x30, 0x31, 0x45, 0x39, 0x34, 0x35, 0x31, 0x42, 0x2D, 0x37, 0x30, 0x45, 0x44,
|
|
0x2D, 0x45, 0x41, 0x45, 0x33, 0x2D, 0x42, 0x33, 0x37, 0x43, 0x2D, 0x31, 0x30, 0x31, 0x46, 0x31, 0x45, 0x45, 0x42, 0x46, 0x35, 0x42, 0x35, 0x7D, 0x2E, 0x70, 0x6E, 0x67, 0x41},
|
|
},
|
|
})
|
|
case *FriendImageElement:
|
|
r = append(r, &msg.Elem{
|
|
NotOnlineImage: &msg.NotOnlineImage{
|
|
FilePath: e.ImageId,
|
|
ResId: e.ImageId,
|
|
OldPicMd5: false,
|
|
PicMd5: e.Md5,
|
|
DownloadPath: e.ImageId,
|
|
Original: 1,
|
|
PbReserve: []byte{0x78, 0x02},
|
|
},
|
|
})
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func ParseMessageElems(elems []*msg.Elem) []IMessageElement {
|
|
var res []IMessageElement
|
|
for _, elem := range elems {
|
|
if elem.SrcMsg != nil {
|
|
if len(elem.SrcMsg.OrigSeqs) != 0 {
|
|
r := &ReplyElement{
|
|
ReplySeq: elem.SrcMsg.OrigSeqs[0],
|
|
Time: elem.SrcMsg.Time,
|
|
Sender: elem.SrcMsg.SenderUin,
|
|
Elements: ParseMessageElems(elem.SrcMsg.Elems),
|
|
}
|
|
res = append(res, r)
|
|
}
|
|
continue
|
|
}
|
|
if elem.Text != nil {
|
|
if len(elem.Text.Attr6Buf) == 0 {
|
|
res = append(res, NewText(elem.Text.Str))
|
|
} else {
|
|
att6 := binary.NewReader(elem.Text.Attr6Buf)
|
|
att6.ReadBytes(7)
|
|
target := int64(uint32(att6.ReadInt32()))
|
|
res = append(res, NewAt(target, elem.Text.Str))
|
|
}
|
|
}
|
|
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 != "" {
|
|
res = append(res, NewText(content))
|
|
}
|
|
}
|
|
if elem.CustomFace != nil {
|
|
res = append(res, &ImageElement{
|
|
Filename: elem.CustomFace.FilePath,
|
|
Size: elem.CustomFace.Size,
|
|
Url: func() string {
|
|
if elem.CustomFace.OrigUrl == "" {
|
|
return "http://gchat.qpic.cn/gchatpic_new/0/0-0-" + strings.ReplaceAll(binary.CalculateImageResourceId(elem.CustomFace.Md5)[1:37], "-", "") + "/0?term=2"
|
|
}
|
|
return "http://gchat.qpic.cn" + elem.CustomFace.OrigUrl
|
|
}(),
|
|
Md5: elem.CustomFace.Md5,
|
|
})
|
|
}
|
|
if elem.NotOnlineImage != nil {
|
|
var img string
|
|
if elem.NotOnlineImage.OrigUrl != "" {
|
|
img = "http://c2cpicdw.qpic.cn" + elem.NotOnlineImage.OrigUrl
|
|
} else {
|
|
img = "http://c2cpicdw.qpic.cn/offpic_new/0/" + elem.NotOnlineImage.ResId + "/0?term=2"
|
|
}
|
|
res = append(res, &ImageElement{
|
|
Filename: elem.NotOnlineImage.FilePath,
|
|
Size: elem.NotOnlineImage.FileLen,
|
|
Url: img,
|
|
Md5: elem.NotOnlineImage.PicMd5,
|
|
})
|
|
}
|
|
if elem.Face != nil {
|
|
res = append(res, NewFace(elem.Face.Index))
|
|
}
|
|
}
|
|
return res
|
|
}
|