feat: wake backgrounded accounts via MTProto internal push connection

internal/rpc/account.go

@@ -2,10 +2,13 @@ package rpc
 
 import (
 	"context"
+	"encoding/hex"
 	"errors"
+	"strconv"
 	"strings"
 
 	"github.com/iamxvbaba/td/tg"
+	"go.uber.org/zap"
 
 	"github.com/iamxvbaba/td/tlprofile"
 	"telesrv/internal/branding"
@@ -14,6 +17,12 @@ import (
 	"telesrv/internal/domain"
 )
 
+// mtprotoPushTokenType 是 account.registerDevice/unregisterDevice 的 token_type=7：
+// 客户端在没有 FCM/APNs 的场景下（自建 owpg:// 服务器、大陆/去 Google 化设备）自己
+// 建立的「MTProto 内部推送」通道——一条独立、常驻、只发 ping 的连接，用自己的
+// session_id 作为 token 上报。见 PushSessionRegistrar。
+const mtprotoPushTokenType = 7
+
 // registerAccount 注册 account.* RPC handler。
 func (r *Router) registerAccount(d *tlprofile.Dispatcher) {
 	registerRPC[*tg.AccountDeleteAccountRequest](d, tlprofile.SemanticMethodAccountDeleteAccount, func(ctx context.Context, req *tg.AccountDeleteAccountRequest) (any, error) {
@@ -26,9 +35,11 @@ func (r *Router) registerAccount(d *tlprofile.Dispatcher) {
 		return r.onAccountConfirmPhone(ctx, req)
 	})
 	registerRPC[*tg.AccountRegisterDeviceRequest](d, tlprofile.SemanticMethodAccountRegisterDevice, func(ctx context.Context, req *tg.AccountRegisterDeviceRequest) (any, error) {
+		r.registerPushSession(ctx, req.TokenType, req.Token)
 		return true, nil
 	})
 	registerRPC[*tg.AccountUnregisterDeviceRequest](d, tlprofile.SemanticMethodAccountUnregisterDevice, func(ctx context.Context, req *tg.AccountUnregisterDeviceRequest) (any, error) {
+		r.unregisterPushSession(ctx, req.TokenType, req.Token)
 		return true, nil
 	})
 	registerRPC[*tg.AccountUpdateDeviceLockedRequest](d, tlprofile.SemanticMethodAccountUpdateDeviceLocked, func(ctx context.Context, layerRequest *tg.AccountUpdateDeviceLockedRequest) (any, error) {
@@ -460,6 +471,65 @@ func (r *Router) registerAccount(d *tlprofile.Dispatcher) {
 
 }
 
+// registerPushSession 把 token_type=7 的 registerDevice 登记为发起本次 RPC 的连接
+// （raw auth_key_id + session_id）的推送通道身份，见 mtprotoPushTokenType 与
+// PushSessionRegistrar。其它 token_type（FCM/APNs/…）本服务器不发外部推送，忽略。
+func (r *Router) registerPushSession(ctx context.Context, tokenType int, token string) {
+	if tokenType != mtprotoPushTokenType {
+		return
+	}
+	registrar, ok := r.deps.Sessions.(PushSessionRegistrar)
+	if !ok {
+		return
+	}
+	rawAuthKeyID, ok := RawAuthKeyIDFrom(ctx)
+	if !ok {
+		return
+	}
+	sessionID, ok := SessionIDFrom(ctx)
+	if !ok {
+		return
+	}
+	// token 上报的是客户端自己的 pushSessionId（ConnectionsManager.cpp 的
+	// to_string_uint64((uint64_t) pushSessionId)），理论上应与本次 RPC 所在的
+	// session_id 一致（registerDevice 走主连接发起，携带 push 连接自己的 session_id
+	// 作为 token）。以 token 解析出的 session_id 为准登记——若客户端将来在别的连接上
+	// 上报，仍能正确登记到「那条」连接，而不是发起 RPC 的这条。
+	if parsed, err := strconv.ParseInt(token, 10, 64); err == nil && parsed != 0 {
+		sessionID = parsed
+	}
+	registrar.MarkPushSession(rawAuthKeyID, sessionID)
+	if r.log != nil {
+		r.log.Info("registered MTProto push session",
+			zap.String("auth_key_id", hex.EncodeToString(rawAuthKeyID[:])),
+			zap.Int64("push_session_id", sessionID),
+		)
+	}
+}
+
+// unregisterPushSession 撤销 registerPushSession 的登记。
+func (r *Router) unregisterPushSession(ctx context.Context, tokenType int, token string) {
+	if tokenType != mtprotoPushTokenType {
+		return
+	}
+	registrar, ok := r.deps.Sessions.(PushSessionRegistrar)
+	if !ok {
+		return
+	}
+	rawAuthKeyID, ok := RawAuthKeyIDFrom(ctx)
+	if !ok {
+		return
+	}
+	sessionID, ok := SessionIDFrom(ctx)
+	if !ok {
+		return
+	}
+	if parsed, err := strconv.ParseInt(token, 10, 64); err == nil && parsed != 0 {
+		sessionID = parsed
+	}
+	registrar.UnmarkPushSession(rawAuthKeyID, sessionID)
+}
+
 func (r *Router) onAccountGetPassword(ctx context.Context) (*tg.AccountPassword, error) {
 	if r.deps.Account == nil {
 		return tgPassword(domain.PasswordSettings{SecureRandom: []byte("telesrv-tdesktop-dev-secure-rand")}), nil

internal/rpc/deps.go

@@ -165,6 +165,18 @@ type BestEffortSessionBinder interface {
 	PushToUserExceptAuthKeySessionBestEffort(ctx context.Context, userID int64, excludeAuthKeyID [8]byte, excludeSessionID int64, t proto.MessageType, msg tg.UpdatesClass, timeout time.Duration) (int, error)
 }
 
+// PushSessionRegistrar 登记/撤销 account.registerDevice(token_type=7，MTProto 内部
+// 推送通道) 的 session。登记后该 session 在推送 fan-out 中被视为永久就绪，绕过
+// receivesUpdates 门槛——它只发 ping、永远不会调 updates.getState，但在账号后台/
+// 未选中（客户端把主连接 setAppPaused 挂起）时是唯一还连着服务器的连接，是官方
+// Telegram 在无 FCM/APNs 场景（大陆、去 Google 化设备）下仍能收到来电、消息等实时
+// 推送的机制。SessionManager 实现；未装配时 registerDevice 静默跳过登记（只影响
+// 后台唤醒，不影响功能正确性）。See memory: call-inactive-account-network-pause。
+type PushSessionRegistrar interface {
+	MarkPushSession(rawAuthKeyID [8]byte, sessionID int64)
+	UnmarkPushSession(rawAuthKeyID [8]byte, sessionID int64)
+}
+
 // TransientSessionBinder 推送短命、不写 durable log 的 update（typing / presence）。
 // 与普通推送的关键区别：目标 session 未就绪时直接跳过、不进 pending——transient 数据
 // getDifference 无法补，就绪后由 getState 快照/下次状态变化重建，囤积过期 transient 无意义。

internal/rpc/sticker_link_smoke_test.go

@@ -89,7 +89,7 @@ func TestCustomStickerPackLinkInstallAndSendSmoke(t *testing.T) {
 		t.Fatalf("sticker link status = %d body=%q, want 200", rr.Code, rr.Body.String())
 	}
 	body := rr.Body.String()
-	for _, want := range []string{"https://telesrv.net/addstickers/alice_fresh_pack", "telesrv://addstickers?set=alice_fresh_pack"} {
+	for _, want := range []string{"https://telesrv.net/addstickers/alice_fresh_pack", "telesrv://telesrv.net/addstickers/alice_fresh_pack"} {
 		if !strings.Contains(body, want) {
 			t.Fatalf("sticker link body missing %q:\n%s", want, body)
 		}