chore: refresh gramsrv public release

internal/rpc/bots_hooks.go

@@ -0,0 +1,102 @@
+package rpc
+
+import (
+	"context"
+	"time"
+
+	"github.com/gotd/td/tg"
+	"go.uber.org/zap"
+
+	"telesrv/internal/domain"
+)
+
+// 本文件实现 app/bots 的 RouterHooks 回调：token revoke 后的 session 失效闭环，
+// 以及命令变更后的 updateBotCommands 在线推送。Router 创建后经
+// botsService.SetRouterHooks(router) 装配（见 cmd/telesrv/main.go）。
+
+// maxBotCommandsPushPeers 限制单次命令变更的推送扇出（bot 的最近 dialog peer 数）。
+// 超出的离线/长尾用户靠 bot_info_version bump 在下次 getFullUser 时拿到新命令。
+const maxBotCommandsPushPeers = 100
+
+// RevokeBotSessions 撤销 bot 的全部已登录 session：删除全部 authorization 行并
+// 强制断开在线连接（与 account.resetAuthorization 被踢闭环同款顺序）。
+func (r *Router) RevokeBotSessions(ctx context.Context, botUserID int64) error {
+	if r.deps.Auth == nil || botUserID == 0 {
+		return nil
+	}
+	deleted, err := r.deps.Auth.ResetAuthorizations(ctx, botUserID, [8]byte{})
+	if err != nil {
+		return err
+	}
+	for _, a := range deleted {
+		r.revokeAuthKeySessions(a.AuthKeyID)
+		if err := r.clearAuthKeyState(ctx, a.AuthKeyID); err != nil {
+			r.log.Warn("revoke bot sessions: clear auth key state",
+				zap.Int64("bot_user_id", botUserID), zap.Error(err))
+		}
+	}
+	if len(deleted) > 0 {
+		r.log.Info("revoked bot sessions", zap.Int64("bot_user_id", botUserID), zap.Int("count", len(deleted)))
+	}
+	return nil
+}
+
+// PushBotCommandsChanged 给「与该 bot 有私聊 dialog 且在线」的用户推
+// updateBotCommands（peer = 该 bot 的 user peer，对齐 TDesktop/DrKLO 消费语义）。
+// updateBotCommands 无 pts/qts，不进 getDifference——离线用户由随写库一起完成的
+// bot_info_version bump 兜底（下次 getFullUser 重拉命令）。
+//
+// fire-and-forget：在独立 goroutine 内执行（脱离已返回的 setBotCommands RPC ctx），
+// 否则 GetDialogs + 最多 maxBotCommandsPushPeers 次 best-effort 推送会把 RPC 响应
+// 拖到拥塞超时之和、并跨用户占住 BotFather 条带锁。推送是纯通知，丢失靠 version
+// bump 兜底，不需保证投达。扇出有界：只取 bot dialog 列表前 maxBotCommandsPushPeers
+// 个 user peer，再按在线快照过滤；超界部分走版本兜底（有意取舍）。
+func (r *Router) PushBotCommandsChanged(ctx context.Context, botUserID int64, commands []domain.BotCommand) {
+	if r.deps.Dialogs == nil || r.deps.Sessions == nil || botUserID == 0 {
+		return
+	}
+	// 拷贝命令切片：调用方（service）可能复用底层数组。
+	cmds := append([]domain.BotCommand(nil), commands...)
+	go r.pushBotCommandsChanged(context.WithoutCancel(ctx), botUserID, cmds)
+}
+
+func (r *Router) pushBotCommandsChanged(ctx context.Context, botUserID int64, commands []domain.BotCommand) {
+	defer func() {
+		if rec := recover(); rec != nil {
+			r.log.Error("push bot commands panicked", zap.Int64("bot_user_id", botUserID), zap.Any("panic", rec))
+		}
+	}()
+	ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
+	defer cancel()
+	list, err := r.deps.Dialogs.GetDialogs(ctx, botUserID, domain.DialogFilter{Limit: maxBotCommandsPushPeers})
+	if err != nil {
+		r.log.Warn("push bot commands: list bot dialogs", zap.Int64("bot_user_id", botUserID), zap.Error(err))
+		return
+	}
+	candidates := make([]int64, 0, len(list.Dialogs))
+	for _, dialog := range list.Dialogs {
+		if dialog.Peer.Type == domain.PeerTypeUser && dialog.Peer.ID != 0 && dialog.Peer.ID != botUserID {
+			candidates = append(candidates, dialog.Peer.ID)
+		}
+	}
+	if len(candidates) == 0 {
+		return
+	}
+	if provider, ok := r.deps.Sessions.(OnlineUserProvider); ok {
+		candidates = provider.OnlineUserIDsForCandidates(candidates, maxBotCommandsPushPeers)
+	}
+	if len(candidates) == 0 {
+		return
+	}
+	update := &tg.Updates{
+		Updates: []tg.UpdateClass{&tg.UpdateBotCommands{
+			Peer:     &tg.PeerUser{UserID: botUserID},
+			BotID:    botUserID,
+			Commands: tgBotCommands(commands),
+		}},
+		Date: int(r.clock.Now().Unix()),
+	}
+	for _, userID := range candidates {
+		r.pushUserMessage(ctx, userID, "push bot commands", update)
+	}
+}