118 lines
3.4 KiB
Go
118 lines
3.4 KiB
Go
package model
|
||
|
||
import (
|
||
"bytes"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/sjzar/chatlog/internal/model/wxproto"
|
||
"github.com/sjzar/chatlog/pkg/util/zstd"
|
||
"google.golang.org/protobuf/proto"
|
||
)
|
||
|
||
// CREATE TABLE Msg_md5(talker)(
|
||
// local_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
// server_id INTEGER,
|
||
// local_type INTEGER,
|
||
// sort_seq INTEGER,
|
||
// real_sender_id INTEGER,
|
||
// create_time INTEGER,
|
||
// status INTEGER,
|
||
// upload_status INTEGER,
|
||
// download_status INTEGER,
|
||
// server_seq INTEGER,
|
||
// origin_source INTEGER,
|
||
// source TEXT,
|
||
// message_content TEXT,
|
||
// compress_content TEXT,
|
||
// packed_info_data BLOB,
|
||
// WCDB_CT_message_content INTEGER DEFAULT NULL,
|
||
// WCDB_CT_source INTEGER DEFAULT NULL
|
||
// )
|
||
type MessageV4 struct {
|
||
SortSeq int64 `json:"sort_seq"` // 消息序号，10位时间戳 + 3位序号
|
||
LocalType int64 `json:"local_type"` // 消息类型
|
||
RealSenderID int `json:"real_sender_id"` // 发送人 ID，对应 Name2Id 表序号
|
||
CreateTime int64 `json:"create_time"` // 消息创建时间，10位时间戳
|
||
MessageContent []byte `json:"message_content"` // 消息内容，文字聊天内容 或 zstd 压缩内容
|
||
PackedInfoData []byte `json:"packed_info_data"` // 额外数据，类似 proto，格式与 v3 有差异
|
||
Status int `json:"status"` // 消息状态，2 是已发送，4 是已接收，可以用于判断 IsSender（猜测）
|
||
|
||
// 非关键信息，后续有需要再加入
|
||
// LocalID int `json:"local_id"`
|
||
// ServerID int64 `json:"server_id"`
|
||
// UploadStatus int `json:"upload_status"`
|
||
// DownloadStatus int `json:"download_status"`
|
||
// ServerSeq int `json:"server_seq"`
|
||
// OriginSource int `json:"origin_source"`
|
||
// Source string `json:"source"`
|
||
// CompressContent string `json:"compress_content"`
|
||
}
|
||
|
||
func (m *MessageV4) Wrap(id2Name map[int]string, isChatRoom bool) *Message {
|
||
|
||
_m := &Message{
|
||
Sequence: m.SortSeq,
|
||
CreateTime: time.Unix(m.CreateTime, 0),
|
||
TalkerID: m.RealSenderID, // 依赖 Name2Id 表进行转换为 StrTalker
|
||
Type: m.LocalType,
|
||
Version: WeChatV4,
|
||
}
|
||
|
||
if name, ok := id2Name[m.RealSenderID]; ok {
|
||
_m.Talker = name
|
||
}
|
||
|
||
if m.Status == 2 {
|
||
_m.IsSender = 1
|
||
}
|
||
|
||
if bytes.HasPrefix(m.MessageContent, []byte{0x28, 0xb5, 0x2f, 0xfd}) {
|
||
if b, err := zstd.Decompress(m.MessageContent); err == nil {
|
||
_m.Content = string(b)
|
||
}
|
||
} else {
|
||
_m.Content = string(m.MessageContent)
|
||
}
|
||
|
||
if isChatRoom {
|
||
_m.IsChatRoom = true
|
||
split := strings.SplitN(_m.Content, ":\n", 2)
|
||
if len(split) == 2 {
|
||
_m.ChatRoomSender = split[0]
|
||
_m.Content = split[1]
|
||
}
|
||
}
|
||
|
||
if _m.Type != 1 {
|
||
mediaMessage, err := NewMediaMessage(_m.Type, _m.Content)
|
||
if err == nil {
|
||
_m.MediaMessage = mediaMessage
|
||
_m.Type = mediaMessage.Type
|
||
_m.SubType = mediaMessage.SubType
|
||
}
|
||
}
|
||
|
||
if len(m.PackedInfoData) != 0 {
|
||
if packedInfo := ParsePackedInfo(m.PackedInfoData); packedInfo != nil {
|
||
// FIXME 尝试解决 v4 版本 xml 数据无法匹配到 hardlink 记录的问题
|
||
if _m.Type == 3 && packedInfo.Image != nil {
|
||
_m.MediaMessage.MediaMD5 = packedInfo.Image.Md5
|
||
}
|
||
if _m.Type == 43 && packedInfo.Video != nil {
|
||
_m.MediaMessage.MediaMD5 = packedInfo.Video.Md5
|
||
}
|
||
}
|
||
}
|
||
|
||
return _m
|
||
}
|
||
|
||
func ParsePackedInfo(b []byte) *wxproto.PackedInfo {
|
||
var pbMsg wxproto.PackedInfo
|
||
if err := proto.Unmarshal(b, &pbMsg); err != nil {
|
||
return nil
|
||
}
|
||
return &pbMsg
|
||
}
|