3fa0c6f1e4
Follow up gramsrv#27 with the remaining peer-bearing messages.Messages boundaries and route-level wire regression coverage from the telesrv canonical implementation. Origin-PR: gramsrv#27 Origin-Head:f0bd6c9699Origin-Merge:7713847b2aCanonical: f1ba406015a2d5e37802b963757a5cac8ee3eb0e Co-authored-by: epilepticseizureee <business.egor.sg@gmail.com>
338 lines
13 KiB
Go
338 lines
13 KiB
Go
package rpc
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
|
||
"github.com/iamxvbaba/td/tg"
|
||
|
||
"telesrv/internal/domain"
|
||
)
|
||
|
||
// resolveMonoforumForAdmin 解析 TDesktop messages.getSavedDialogs/getSavedHistory 的 parent_peer，
|
||
// 并校验当前用户可管理母广播频道的 Direct Messages。TDesktop 的 SavedSublist 实际会把
|
||
// parentChat()->input() 作为 parent_peer；根据客户端 materialize 路径，它既可能是 monoforum
|
||
// 虚拟频道，也可能是与之关联的母广播频道。因此这里把两种 wire peer 归一到同一个 monoforum，
|
||
// 授权仍只认母频道的 creator / ManageDirectMessages，绝不能把普通 admin 放进管理者视图。
|
||
//
|
||
// 返回 (monoforum频道, isMonoforum, err)：parent 是有效但未关联 Direct Messages 的普通频道时
|
||
// 返回 (零, false, nil)，由调用方保留良性空响应；关联频道的非管理者返回 CHAT_ADMIN_REQUIRED。
|
||
func (r *Router) resolveMonoforumForAdmin(ctx context.Context, userID int64, parent domain.Peer) (domain.Channel, bool, error) {
|
||
if r.deps.Channels == nil {
|
||
return domain.Channel{}, false, notImplementedErr()
|
||
}
|
||
if parent.Type != domain.PeerTypeChannel || parent.ID == 0 {
|
||
return domain.Channel{}, false, parentPeerInvalidErr()
|
||
}
|
||
mono, isAdmin, err := r.deps.Channels.ResolveMonoforumSend(ctx, userID, parent.ID)
|
||
if errors.Is(err, domain.ErrChannelInvalid) {
|
||
// TDesktop 当前的 Direct Messages subsection 会传母广播频道。只接受显式的
|
||
// linked_monoforum 关系，不能把任意普通频道猜成 monoforum。
|
||
views, viewErr := r.deps.Channels.GetChannels(ctx, userID, []int64{parent.ID})
|
||
if viewErr != nil {
|
||
return domain.Channel{}, false, internalErr()
|
||
}
|
||
if len(views) != 1 {
|
||
return domain.Channel{}, false, nil
|
||
}
|
||
parentChannel := views[0].Channel
|
||
if parentChannel.ID != parent.ID || parentChannel.Deleted || parentChannel.Monoforum || parentChannel.LinkedMonoforumID == 0 {
|
||
return domain.Channel{}, false, nil
|
||
}
|
||
mono, isAdmin, err = r.deps.Channels.ResolveMonoforumSend(ctx, userID, parentChannel.LinkedMonoforumID)
|
||
if err != nil {
|
||
// A visible parent that advertises linked_monoforum_id but cannot resolve
|
||
// that target violates the durable channel-link invariant. Do not disguise
|
||
// it as an ordinary channel probe.
|
||
return domain.Channel{}, false, internalErr()
|
||
}
|
||
}
|
||
if err != nil {
|
||
if errors.Is(err, domain.ErrChannelInvalid) {
|
||
return domain.Channel{}, false, nil
|
||
}
|
||
return domain.Channel{}, false, internalErr()
|
||
}
|
||
if !isAdmin {
|
||
// 是 monoforum,但调用者不是母频道管理员:读私信列表/历史仅限管理员。
|
||
return domain.Channel{}, false, tgerr400("CHAT_ADMIN_REQUIRED")
|
||
}
|
||
return mono, true, nil
|
||
}
|
||
|
||
// monoforumSavedDialogs 返回 monoforum 的订阅者子会话列表(管理员视角的私信列表)。mono 已解析+鉴权。
|
||
func (r *Router) monoforumSavedDialogs(ctx context.Context, userID int64, mono domain.Channel, limit, offsetID int) (tg.MessagesSavedDialogsClass, error) {
|
||
list, err := r.deps.Channels.ListMonoforumDialogs(ctx, domain.MonoforumDialogsFilter{
|
||
MonoforumID: mono.ID,
|
||
Limit: limit,
|
||
OffsetID: offsetID,
|
||
})
|
||
if err != nil {
|
||
return nil, internalErr()
|
||
}
|
||
dialogs := make([]tg.SavedDialogClass, 0, len(list.Dialogs))
|
||
for _, d := range list.Dialogs {
|
||
peer := tgPeer(d.SavedPeer)
|
||
if peer == nil {
|
||
continue
|
||
}
|
||
md := &tg.MonoForumDialog{
|
||
Peer: peer,
|
||
TopMessage: d.TopMessageID,
|
||
ReadInboxMaxID: d.ReadInboxMaxID,
|
||
ReadOutboxMaxID: d.ReadOutboxMaxID,
|
||
UnreadCount: d.UnreadCount,
|
||
}
|
||
dialogs = append(dialogs, md)
|
||
}
|
||
messages := make([]tg.MessageClass, 0, len(list.Messages))
|
||
for _, m := range list.Messages {
|
||
if item := tgChannelMessage(userID, m); item != nil {
|
||
messages = append(messages, item)
|
||
}
|
||
}
|
||
return &tg.MessagesSavedDialogs{
|
||
Dialogs: dialogs,
|
||
Messages: messages,
|
||
Chats: r.monoforumChats(ctx, userID, mono),
|
||
Users: r.monoforumSubscriberUsers(ctx, userID, list.Dialogs, list.Messages),
|
||
}, nil
|
||
}
|
||
|
||
// monoforumSavedHistory 返回某订阅者在 monoforum 内的私信历史。mono 已解析+鉴权。
|
||
func (r *Router) monoforumSavedHistory(ctx context.Context, userID int64, mono domain.Channel, savedPeer domain.Peer, limit, offsetID int) (tg.MessagesMessagesClass, error) {
|
||
if savedPeer.ID == 0 {
|
||
return nil, peerIDInvalidErr()
|
||
}
|
||
hist, err := r.deps.Channels.ListMonoforumHistory(ctx, domain.MonoforumHistoryFilter{
|
||
MonoforumID: mono.ID,
|
||
SavedPeer: savedPeer,
|
||
Limit: limit,
|
||
OffsetID: offsetID,
|
||
})
|
||
if err != nil {
|
||
return nil, internalErr()
|
||
}
|
||
messages := make([]tg.MessageClass, 0, len(hist.Messages))
|
||
for _, m := range hist.Messages {
|
||
if item := tgChannelMessage(userID, m); item != nil {
|
||
messages = append(messages, item)
|
||
}
|
||
}
|
||
result := &tg.MessagesMessagesSlice{
|
||
Count: hist.Count,
|
||
Messages: messages,
|
||
Chats: r.monoforumChats(ctx, userID, mono),
|
||
Users: r.monoforumSubscriberUsers(ctx, userID, nil, hist.Messages),
|
||
}
|
||
r.applyPeerReadModelsToMessages(ctx, userID, result)
|
||
return result, nil
|
||
}
|
||
|
||
// monoforumChats 投影客户端 materialize monoforum 私信所需的频道:monoforum 自身直接投影，
|
||
// 再按 viewer 补母广播频道。订阅者没有 monoforum member row，管理员身份也只来自母频道。
|
||
func (r *Router) monoforumChats(ctx context.Context, userID int64, mono domain.Channel) []tg.ChatClass {
|
||
chats := []tg.ChatClass{tgChannelChatForView(userID, domain.ChannelView{Channel: mono})}
|
||
if mono.LinkedMonoforumID != 0 && r.deps.Channels != nil {
|
||
if views, err := r.deps.Channels.GetChannels(ctx, userID, []int64{mono.LinkedMonoforumID}); err == nil {
|
||
for _, view := range views {
|
||
if view.Channel.ID != 0 {
|
||
chats = append(chats, tgChannelChatForView(userID, view))
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return chats
|
||
}
|
||
|
||
// monoforumSubscriberUsers 投影订阅者用户(子会话 saved_peer + 消息发件人)。
|
||
func (r *Router) monoforumSubscriberUsers(ctx context.Context, userID int64, dialogs []domain.MonoforumDialog, messages []domain.ChannelMessage) []tg.UserClass {
|
||
ids := make([]int64, 0, len(dialogs)+len(messages))
|
||
seen := map[int64]struct{}{}
|
||
add := func(id int64) {
|
||
if id == 0 {
|
||
return
|
||
}
|
||
if _, ok := seen[id]; ok {
|
||
return
|
||
}
|
||
seen[id] = struct{}{}
|
||
ids = append(ids, id)
|
||
}
|
||
for _, d := range dialogs {
|
||
if d.SavedPeer.Type == domain.PeerTypeUser {
|
||
add(d.SavedPeer.ID)
|
||
}
|
||
}
|
||
for _, m := range messages {
|
||
add(m.SenderUserID)
|
||
}
|
||
if len(ids) == 0 || r.deps.Users == nil {
|
||
return []tg.UserClass{}
|
||
}
|
||
found, err := r.deps.Users.ByIDs(ctx, userID, ids)
|
||
if err != nil {
|
||
return []tg.UserClass{}
|
||
}
|
||
return r.tgUsers(found)
|
||
}
|
||
|
||
// monoforumReplyPresent 判断 sendMessage 的 reply_to 是否显式携带 monoforum_peer_id。
|
||
// 管理员回复必须带目标订阅者；普通订阅者按官方 TDesktop 行为不携带 reply_to，目标由调用者推导。
|
||
func monoforumReplyPresent(input tg.InputReplyToClass) bool {
|
||
switch v := input.(type) {
|
||
case *tg.InputReplyToMonoForum:
|
||
return v != nil
|
||
case *tg.InputReplyToMessage:
|
||
if v == nil {
|
||
return false
|
||
}
|
||
_, ok := v.GetMonoforumPeerID()
|
||
return ok
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
// monoforumReplyTargetPeer 从 reply_to 的 monoforum_peer_id 解析订阅者子会话 peer。
|
||
func (r *Router) monoforumReplyTargetPeer(userID int64, input tg.InputReplyToClass) (domain.Peer, bool) {
|
||
var inputPeer tg.InputPeerClass
|
||
switch v := input.(type) {
|
||
case *tg.InputReplyToMonoForum:
|
||
if v != nil {
|
||
inputPeer = v.MonoforumPeerID
|
||
}
|
||
case *tg.InputReplyToMessage:
|
||
if v != nil {
|
||
if p, ok := v.GetMonoforumPeerID(); ok {
|
||
inputPeer = p
|
||
}
|
||
}
|
||
}
|
||
if inputPeer == nil {
|
||
return domain.Peer{}, false
|
||
}
|
||
return r.domainPeerFromInputPeer(userID, inputPeer)
|
||
}
|
||
|
||
func (r *Router) monoforumSavedPeerForSender(userID int64, isAdmin bool, replyTo tg.InputReplyToClass) (domain.Peer, error) {
|
||
savedPeer := domain.Peer{Type: domain.PeerTypeUser, ID: userID}
|
||
if monoforumReplyPresent(replyTo) {
|
||
var valid bool
|
||
savedPeer, valid = r.monoforumReplyTargetPeer(userID, replyTo)
|
||
if !valid || savedPeer.Type != domain.PeerTypeUser || savedPeer.ID == 0 {
|
||
return domain.Peer{}, replyToMonoforumPeerInvalidErr()
|
||
}
|
||
} else if isAdmin {
|
||
return domain.Peer{}, replyToMonoforumPeerInvalidErr()
|
||
}
|
||
if !isAdmin && savedPeer.ID != userID {
|
||
return domain.Peer{}, replyToMonoforumPeerInvalidErr()
|
||
}
|
||
return savedPeer, nil
|
||
}
|
||
|
||
// monoforumMessageReplyFromInput separates the sub-dialog selector from the actual message reply.
|
||
// InputReplyToMonoForum only selects a subscriber; InputReplyToMessage may carry both the selector
|
||
// and a real reply_to_msg_id, so clear flags.5 before reusing the common structural validator.
|
||
func (r *Router) monoforumMessageReplyFromInput(ctx context.Context, userID int64, peer domain.Peer, input tg.InputReplyToClass) (*domain.MessageReply, error) {
|
||
switch value := input.(type) {
|
||
case nil, *tg.InputReplyToMonoForum:
|
||
return nil, nil
|
||
case *tg.InputReplyToMessage:
|
||
if value == nil {
|
||
return nil, nil
|
||
}
|
||
clean := *value
|
||
clean.Flags.Unset(5)
|
||
clean.MonoforumPeerID = nil
|
||
return r.messageReplyFromInput(ctx, userID, peer, &clean)
|
||
default:
|
||
return r.messageReplyFromInput(ctx, userID, peer, input)
|
||
}
|
||
}
|
||
|
||
// sendMonoforumMessage 处理向频道私信(monoforum)发送:订阅者发到自己的子会话,管理员回复到目标订阅者。
|
||
// saved_peer 对订阅者由调用者推导、对管理员来自 reply_to;管理员可写任意订阅者子会话,订阅者只能写自己的。
|
||
func (r *Router) sendMonoforumMessage(ctx context.Context, userID int64, peer domain.Peer, mono domain.Channel, isAdmin bool, req domain.SendMonoforumMessageRequest) (tg.UpdatesClass, error) {
|
||
if r.deps.Channels == nil {
|
||
return nil, notImplementedErr()
|
||
}
|
||
if peer.Type != domain.PeerTypeChannel || peer.ID == 0 {
|
||
return nil, peerIDInvalidErr()
|
||
}
|
||
if mono.ID != peer.ID || !mono.Monoforum || req.SavedPeer.Type != domain.PeerTypeUser || req.SavedPeer.ID == 0 {
|
||
return nil, replyToMonoforumPeerInvalidErr()
|
||
}
|
||
if !isAdmin && req.SavedPeer.ID != userID {
|
||
// 普通订阅者只能写自己的子会话,不能写他人的。
|
||
return nil, replyToMonoforumPeerInvalidErr()
|
||
}
|
||
req.MonoforumID = mono.ID
|
||
req.SenderUserID = userID
|
||
if req.Date == 0 {
|
||
req.Date = int(r.clock.Now().Unix())
|
||
}
|
||
res, err := r.deps.Channels.SendMonoforumMessage(ctx, req)
|
||
if err != nil {
|
||
return nil, messageSendErr(err)
|
||
}
|
||
if req.ClearDraft {
|
||
r.clearDraftAfterSend(ctx, userID, peer, req.ReplyTo)
|
||
}
|
||
if !res.Duplicate {
|
||
r.enqueueMonoforumMessageFanout(ctx, userID, mono, req.SavedPeer, res)
|
||
}
|
||
return r.monoforumSendUpdates(ctx, userID, mono, req.SavedPeer, res), nil
|
||
}
|
||
|
||
// monoforumSendUpdates 给发送者构造回声 Updates:updateMessageID(关联 random_id)+ updateNewChannelMessage
|
||
// (monoforum 走 channel pts)。另一方经 monoforum 频道的 getChannelDifference 收取该 durable 事件。
|
||
func (r *Router) monoforumSendUpdates(ctx context.Context, userID int64, mono domain.Channel, savedPeer domain.Peer, res domain.SendChannelMessageResult) tg.UpdatesClass {
|
||
updates := make([]tg.UpdateClass, 0, 3)
|
||
if res.Message.RandomID != 0 {
|
||
updates = append(updates, &tg.UpdateMessageID{ID: res.Message.ID, RandomID: res.Message.RandomID})
|
||
}
|
||
newMsg := &tg.UpdateNewChannelMessage{Pts: res.Event.Pts, PtsCount: res.Event.PtsCount}
|
||
if channelMsg := tgChannelMessage(userID, res.Message); channelMsg != nil {
|
||
newMsg.Message = channelMsg
|
||
} else {
|
||
newMsg.Message = &tg.MessageEmpty{ID: res.Message.ID}
|
||
}
|
||
updates = append(updates, newMsg)
|
||
if res.SenderStarsBalance != nil && res.Message.SenderUserID == userID {
|
||
updates = append(updates, &tg.UpdateStarsBalance{
|
||
Balance: &tg.StarsAmount{Amount: res.SenderStarsBalance.Balance},
|
||
})
|
||
}
|
||
date := int(r.clock.Now().Unix())
|
||
if res.Duplicate && res.ReplayDeleteEvent != nil {
|
||
if deleted := tgChannelUpdate(userID, *res.ReplayDeleteEvent); deleted != nil {
|
||
updates = append(updates, deleted)
|
||
}
|
||
if res.ReplayDeleteEvent.Date > date {
|
||
date = res.ReplayDeleteEvent.Date
|
||
}
|
||
}
|
||
return &tg.Updates{
|
||
Updates: updates,
|
||
Chats: r.monoforumChats(ctx, userID, mono),
|
||
Users: r.monoforumSubscriberUsers(ctx, userID, []domain.MonoforumDialog{{SavedPeer: savedPeer}}, []domain.ChannelMessage{res.Message}),
|
||
Date: date,
|
||
}
|
||
}
|
||
|
||
func (r *Router) monoforumDeliveryUpdates(ctx context.Context, userID int64, mono domain.Channel, savedPeer domain.Peer, res domain.SendChannelMessageResult) *tg.Updates {
|
||
updates, _ := r.monoforumSendUpdates(ctx, userID, mono, savedPeer, res).(*tg.Updates)
|
||
if updates == nil {
|
||
return nil
|
||
}
|
||
filtered := make([]tg.UpdateClass, 0, len(updates.Updates))
|
||
for _, update := range updates.Updates {
|
||
if _, randomMapping := update.(*tg.UpdateMessageID); !randomMapping {
|
||
filtered = append(filtered, update)
|
||
}
|
||
}
|
||
updates.Updates = filtered
|
||
return updates
|
||
}
|