MiraiGo/message/pack.go

package message

import (
	"encoding/hex"
	"fmt"

	"github.com/Mrs4s/MiraiGo/binary"
	"github.com/Mrs4s/MiraiGo/client/pb/msg"
	"github.com/Mrs4s/MiraiGo/internal/proto"
)

/*
var imgOld = []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,
}
*/

func (e *TextElement) Pack() (r []*msg.Elem) {
	r = append(r, &msg.Elem{
		Text: &msg.Text{
			Str: &e.Content,
		},
	})
	return
}

func (e *FaceElement) Pack() (r []*msg.Elem) {
	r = []*msg.Elem{}
	if e.Index >= 260 {
		elem := &msg.MsgElemInfoServtype33{
			Index:  proto.Uint32(uint32(e.Index)),
			Text:   []byte("/" + e.Name),
			Compat: []byte("/" + e.Name),
		}
		b, _ := proto.Marshal(elem)
		r = append(r, &msg.Elem{
			CommonElem: &msg.CommonElem{
				ServiceType:  proto.Int32(33),
				PbElem:       b,
				BusinessType: proto.Int32(1),
			},
		})
	} else {
		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},
			},
		})
	}
	return
}

func (e *AtElement) Pack() (r []*msg.Elem) {
	r = []*msg.Elem{}
	if e.Guild {
		pb, _ := proto.Marshal(&msg.TextResvAttr{AtType: proto.Uint32(2), AtMemberTinyid: proto.Uint64(uint64(e.Target))})
		r = append(r, &msg.Elem{
			Text: &msg.Text{
				Str:       &e.Display,
				PbReserve: pb,
			},
		})
	} else {
		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([]rune(e.Display))))
					if e.Target == 0 {
						w.WriteByte(1)
					} else {
						w.WriteByte(0)
					}
					w.WriteUInt32(uint32(e.Target))
					w.WriteUInt16(0)
				}),
			},
		})
	}
	r = append(r, &msg.Elem{Text: &msg.Text{Str: proto.String(" ")}})
	return
}

func (e *ServiceElement) Pack() (r []*msg.Elem) {
	r = []*msg.Elem{}
	// id =35 已移至 ForwardElement
	if e.Id == 1 {
		r = append(r, &msg.Elem{
			Text: &msg.Text{Str: &e.ResId},
		})
	}
	r = append(r, &msg.Elem{
		RichMsg: &msg.RichMsg{
			Template1: append([]byte{1}, binary.ZlibCompress([]byte(e.Content))...),
			ServiceId: &e.Id,
		},
	})
	return
}

func (e *LightAppElement) Pack() (r []*msg.Elem) {
	r = []*msg.Elem{}
	r = append(r, &msg.Elem{
		LightApp: &msg.LightAppElem{
			Data: append([]byte{1}, binary.ZlibCompress([]byte(e.Content))...),
			// MsgResid: []byte{1},
		},
	})
	return
}

func (e *ShortVideoElement) Pack() (r []*msg.Elem) {
	r = append(r, &msg.Elem{
		Text: &msg.Text{
			Str: proto.String("你的QQ暂不支持查看视频短片，请期待后续版本。"),
		},
	})

	video := &msg.VideoFile{
		FileUuid:               e.Uuid,
		FileMd5:                e.Md5,
		FileName:               []byte(hex.EncodeToString(e.Md5) + ".mp4"),
		FileFormat:             proto.Int32(3),
		FileTime:               proto.Int32(10),
		FileSize:               proto.Int32(e.Size),
		ThumbWidth:             proto.Int32(1280),
		ThumbHeight:            proto.Int32(720),
		ThumbFileMd5:           e.ThumbMd5,
		ThumbFileSize:          proto.Int32(e.ThumbSize),
		BusiType:               proto.Int32(0),
		FromChatType:           proto.Int32(-1),
		ToChatType:             proto.Int32(-1),
		BoolSupportProgressive: proto.Bool(true),
		FileWidth:              proto.Int32(1280),
		FileHeight:             proto.Int32(720),
	}

	if e.Guild {
		video.BusiType = proto.Int32(4601)
		video.SubBusiType = proto.Int32(4601)
		video.FileWidth = proto.Int32(0)
		video.FileHeight = proto.Int32(0)
		video.VideoAttr = proto.Int32(0)
	}

	r = append(r, &msg.Elem{VideoFile: video})
	return
}

func (e *AnimatedSticker) Pack() []*msg.Elem {
	if e.Name == "" {
		e.Name = faceMap[int(e.ID)]
	}
	name := "/" + e.Name
	business := int32(1)
	if e.ID == 114 {
		// 对篮球适配，否则篮球不会投出
		business = 2
	}

	pbElem, _ := proto.Marshal(&msg.MsgElemInfoServtype37{
		Packid:      []byte("1"),
		Stickerid:   []byte(stickerMap[int(e.ID)]),
		Qsid:        proto.Uint32(uint32(e.ID)),
		Sourcetype:  proto.Uint32(1),
		Stickertype: proto.Uint32(uint32(business)),
		Resultid:    []byte{},
		Text:        []byte(name),
		Surpriseid:  []byte{},
		Randomtype:  proto.Uint32(1),
	})
	common := &msg.Elem{
		CommonElem: &msg.CommonElem{
			ServiceType:  proto.Int32(37),
			PbElem:       pbElem,
			BusinessType: &business,
		},
	}

	pbReverse, _ := proto.Marshal(&msg.Text{Str: proto.String(fmt.Sprintf("[%s]请使用最新版手机QQ体验新功能", e.Name))})
	text := &msg.Elem{
		Text: &msg.Text{
			Str:       &name,
			PbReserve: pbReverse,
		},
	}

	return []*msg.Elem{common, text}
}