@mtcute/core 0.13.2 → 0.13.3

package/esm/highlevel/updates/manager.js

@@ -100,6 +100,8 @@ export class UpdatesManager {
     log;
     _handler = () => { };
     _onCatchingUp = () => { };
+    // eslint-disable-next-line @typescript-eslint/ban-types
+    _channelPtsLimit;
     auth; // todo: do we need a local copy?
     keepAliveInterval;
     constructor(client, params = {}) {
@@ -116,6 +118,18 @@ export class UpdatesManager {
         this.log = client.log.create('updates');
         this.catchUpOnStart = params.catchUp ?? false;
         this.noDispatchEnabled = !params.disableNoDispatch;
+        if (params.channelPtsLimit) {
+            if (typeof params.channelPtsLimit === 'function') {
+                this._channelPtsLimit = params.channelPtsLimit;
+            }
+            else {
+                const limit = params.channelPtsLimit;
+                this._channelPtsLimit = () => limit;
+            }
+        }
+        else {
+            this._channelPtsLimit = () => (this.auth?.isBot ? 100000 : 100);
+        }
     }
     setHandler(handler) {
         this._handler = handler;
@@ -335,6 +349,12 @@ export class UpdatesManager {
             log.debug('loaded initial state: pts=%d, qts=%d, date=%d, seq=%d', this.pts, this.qts, this.date, this.seq);
         }
         catch (e) {
+            if (tl.RpcError.is(e, 'AUTH_KEY_UNREGISTERED')) {
+                // we are logged out, stop updates loop
+                lock.release();
+                this.stopLoop();
+                return;
+            }
             if (this.client.isConnected) {
                 log.error('failed to fetch updates state: %s', e);
             }
@@ -666,7 +686,7 @@ export class UpdatesManager {
         const channel = toInputChannel(channelPeer);
         // to make TS happy
         let pts = _pts;
-        let limit = this.auth?.isBot ? 100000 : 100;
+        let limit = this._channelPtsLimit(channelId);
         if (pts <= 0) {
             pts = 1;
             limit = 1;
@@ -681,8 +701,11 @@ export class UpdatesManager {
                 limit,
                 filter: { _: 'channelMessagesFilterEmpty' },
             });
-            if (diff.timeout)
-                lastTimeout = diff.timeout;
+            if (diff.timeout) {
+                lastTimeout = this.params.overrideOpenChatTimeout ?
+                    this.params.overrideOpenChatTimeout(diff) :
+                    diff.timeout;
+            }
             if (diff._ === 'updates.channelDifferenceEmpty') {
                 log.debug('getChannelDifference (cid = %d) returned channelDifferenceEmpty', channelId);
                 break;
@@ -692,13 +715,18 @@ export class UpdatesManager {
                 if (diff.dialog._ === 'dialog') {
                     pts = diff.dialog.pts;
                 }
-                log.warn('getChannelDifference (cid = %d) returned channelDifferenceTooLong. new pts: %d, recent msgs: %d', channelId, pts, diff.messages.length);
-                diff.messages.forEach((message) => {
-                    log.debug('processing message %d (%s) from TooLong diff for channel %d', message.id, message._, channelId);
-                    if (message._ === 'messageEmpty')
-                        return;
-                    pendingUnorderedUpdates.pushBack(toPendingUpdate(messageToUpdate(message), peers, true));
-                });
+                if (this.params.onChannelTooLong) {
+                    this.params.onChannelTooLong(channelId, diff);
+                }
+                else {
+                    log.warn('getChannelDifference (cid = %d) returned channelDifferenceTooLong. new pts: %d, recent msgs: %d', channelId, pts, diff.messages.length);
+                    diff.messages.forEach((message) => {
+                        log.debug('processing message %d (%s) from TooLong diff for channel %d', message.id, message._, channelId);
+                        if (message._ === 'messageEmpty')
+                            return;
+                        pendingUnorderedUpdates.pushBack(toPendingUpdate(messageToUpdate(message), peers, true));
+                    });
+                }
                 break;
             }
             log.debug('getChannelDifference (cid = %d) returned %d messages, %d updates. new pts: %d, final: %b', channelId, diff.newMessages.length, diff.otherUpdates.length, diff.pts, diff.final);