x

internal/wechatdb/repository/message.go

@@ -0,0 +1,68 @@
+package repository
+
+import (
+	"context"
+	"time"
+
+	"github.com/sjzar/chatlog/internal/model"
+
+	log "github.com/sirupsen/logrus"
+)
+
+// GetMessages 实现 Repository 接口的 GetMessages 方法
+func (r *Repository) GetMessages(ctx context.Context, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+
+	if contact, _ := r.GetContact(ctx, talker); contact != nil {
+		talker = contact.UserName
+	} else if chatRoom, _ := r.GetChatRoom(ctx, talker); chatRoom != nil {
+		talker = chatRoom.Name
+	}
+
+	messages, err := r.ds.GetMessages(ctx, startTime, endTime, talker, limit, offset)
+	if err != nil {
+		return nil, err
+	}
+
+	// 补充消息信息
+	if err := r.EnrichMessages(ctx, messages); err != nil {
+		log.Debugf("EnrichMessages failed: %v", err)
+	}
+
+	return messages, nil
+}
+
+// EnrichMessages 补充消息的额外信息
+func (r *Repository) EnrichMessages(ctx context.Context, messages []*model.Message) error {
+	for _, msg := range messages {
+		r.enrichMessage(msg)
+	}
+	return nil
+}
+
+// enrichMessage 补充单条消息的额外信息
+func (r *Repository) enrichMessage(msg *model.Message) {
+	talker := msg.Talker
+
+	// 处理群聊消息
+	if msg.IsChatRoom {
+		talker = msg.ChatRoomSender
+
+		// 补充群聊名称
+		if chatRoom, ok := r.chatRoomCache[msg.Talker]; ok {
+			msg.ChatRoomName = chatRoom.DisplayName()
+
+			// 补充发送者在群里的显示名称
+			if displayName, ok := chatRoom.User2DisplayName[talker]; ok {
+				msg.DisplayName = displayName
+			}
+		}
+	}
+
+	// 如果不是自己发送的消息且还没有显示名称，尝试补充发送者信息
+	if msg.DisplayName == "" && msg.IsSender != 1 {
+		contact := r.getFullContact(talker)
+		if contact != nil {
+			msg.DisplayName = contact.DisplayName()
+		}
+	}
+}