chore: refresh gramsrv public release

internal/store/memory/message_polls.go

@@ -0,0 +1,89 @@
+package memory
+
+import (
+	"context"
+	"time"
+
+	"telesrv/internal/domain"
+)
+
+// 私聊消息 poll 投票/关闭：消息可见性在本 store 校验，poll 级校验与状态全部
+// 委托共享 PollStore（与 postgres 实现共用 domain 纯函数语义）。
+
+func (s *MessageStore) VoteMessagePoll(_ context.Context, req domain.VotePrivateMessagePollRequest) (domain.PrivateMessagePollResult, error) {
+	target, err := s.pollMessageTarget(req.UserID, req.Peer, req.MessageID)
+	if err != nil {
+		return domain.PrivateMessagePollResult{}, err
+	}
+	if req.Date == 0 {
+		req.Date = int(time.Now().Unix())
+	}
+	if err := s.polls.Vote(target.Media.Poll.ID, req.UserID, req.Options, req.Date); err != nil {
+		return domain.PrivateMessagePollResult{}, err
+	}
+	return s.privatePollResult(target.UID, target.Media.Poll.ID, req.Date), nil
+}
+
+func (s *MessageStore) CloseMessagePoll(_ context.Context, req domain.ClosePrivateMessagePollRequest) (domain.PrivateMessagePollResult, error) {
+	target, err := s.pollMessageTarget(req.UserID, req.Peer, req.MessageID)
+	if err != nil {
+		return domain.PrivateMessagePollResult{}, err
+	}
+	if req.Date == 0 {
+		req.Date = int(time.Now().Unix())
+	}
+	if err := s.polls.Close(target.Media.Poll.ID, req.UserID); err != nil {
+		return domain.PrivateMessagePollResult{}, err
+	}
+	return s.privatePollResult(target.UID, target.Media.Poll.ID, req.Date), nil
+}
+
+// pollMessageTarget 定位 viewer box 中带 poll 的目标消息。
+func (s *MessageStore) pollMessageTarget(userID int64, peer domain.Peer, messageID int) (domain.Message, error) {
+	if s == nil || s.polls == nil {
+		return domain.Message{}, domain.ErrMessageIDInvalid
+	}
+	if userID == 0 || peer.Type != domain.PeerTypeUser || peer.ID == 0 || messageID <= 0 || messageID > domain.MaxMessageBoxID {
+		return domain.Message{}, domain.ErrMessageIDInvalid
+	}
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+	for _, msg := range s.m[userID] {
+		if msg.ID != messageID || msg.Peer != peer {
+			continue
+		}
+		if msg.Media == nil || msg.Media.Kind != domain.MessageMediaKindPoll || msg.Media.Poll == nil || msg.Media.Poll.ID == 0 {
+			return domain.Message{}, domain.ErrMessageIDInvalid
+		}
+		return msg, nil
+	}
+	return domain.Message{}, domain.ErrMessageIDInvalid
+}
+
+// privatePollResult 收集同一 UID 的全部 owner 副本，并按各自 owner 视角 enrich poll。
+func (s *MessageStore) privatePollResult(uid, pollID int64, now int) domain.PrivateMessagePollResult {
+	out := domain.PrivateMessagePollResult{PollID: pollID}
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+	for _, messages := range s.m {
+		for _, msg := range messages {
+			if msg.UID != uid {
+				continue
+			}
+			item := cloneMessage(msg)
+			item.Media = enrichPollMediaForViewer(s.polls, item.Media, item.OwnerUserID, now)
+			out.Messages = append(out.Messages, item)
+		}
+	}
+	return out
+}
+
+// enrichPrivateMessagePolls 是私聊读路径的 poll enrichment 入口（与 reactions 填充并列）。
+func (s *MessageStore) enrichPrivateMessagePolls(messages []domain.Message, now int) {
+	if s == nil || s.polls == nil {
+		return
+	}
+	for i := range messages {
+		messages[i].Media = enrichPollMediaForViewer(s.polls, messages[i].Media, messages[i].OwnerUserID, now)
+	}
+}