@mtcute/core 0.1.2 → 0.2.2

package/esm/base-client.d.ts

@@ -20,10 +20,9 @@ export interface BaseTelegramClientOptions {
      */
     apiHash: string;
     /**
-     * Telegram storage to use.
-     * If omitted, {@link MemoryStorage} is used
+     * Storage to use for this client.
      */
-    storage?: ITelegramStorage;
+    storage: ITelegramStorage;
     /**
      * Cryptography provider factory to allow delegating
      * crypto to native addon, worker, etc.
@@ -260,8 +259,6 @@ export declare class BaseTelegramClient extends EventEmitter {
     _emitError(err: unknown, connection?: SessionConnection): void;
     /**
      * Adds all peers from a given object to entity cache in storage.
-     *
-     * @returns  `true` if there were any `min` peers
      */
     _cachePeersFrom(obj: object): Promise<boolean>;
     /**

package/esm/base-client.js

@@ -6,7 +6,6 @@ import { __tlReaderMap as defaultReaderMap } from '@mtcute/tl/binary/reader.js';
 import { __tlWriterMap as defaultWriterMap } from '@mtcute/tl/binary/writer.js';
 import { ConfigManager } from './network/config-manager.js';
 import { NetworkManager } from './network/network-manager.js';
-import { MemoryStorage } from './storage/index.js';
 import { createControllablePromise, defaultCryptoProviderFactory, defaultProductionDc, defaultProductionIpv6Dc, defaultTestDc, defaultTestIpv6Dc, getAllPeersFrom, LogManager, readStringSession, toggleChannelIdMark, writeStringSession, } from './utils/index.js';
 /**
  * Basic Telegram client that only implements the bare minimum
@@ -30,7 +29,7 @@ export class BaseTelegramClient extends EventEmitter {
             this.log.level = opts.logLevel;
         }
         this.crypto = (opts.crypto ?? defaultCryptoProviderFactory)();
-        this.storage = opts.storage ?? new MemoryStorage();
+        this.storage = opts.storage;
         this._apiHash = opts.apiHash;
         this._useIpv6 = Boolean(opts.useIpv6);
         this._testMode = Boolean(opts.testMode);
@@ -164,7 +163,9 @@ export class BaseTelegramClient extends EventEmitter {
         }
         const stack = this._niceStacks ? new Error().stack : undefined;
         const res = await this.network.call(message, params, stack);
-        await this._cachePeersFrom(res);
+        if (await this._cachePeersFrom(res)) {
+            await this.saveStorage();
+        }
         // eslint-disable-next-line @typescript-eslint/no-unsafe-return
         return res;
     }
@@ -207,19 +208,13 @@ export class BaseTelegramClient extends EventEmitter {
     }
     /**
      * Adds all peers from a given object to entity cache in storage.
-     *
-     * @returns  `true` if there were any `min` peers
      */
     async _cachePeersFrom(obj) {
         const parsedPeers = [];
-        let hadMin = false;
         let count = 0;
         for (const peer of getAllPeersFrom(obj)) {
             if (peer.min) {
-                // absolutely incredible min peer handling, courtesy of levlam.
-                // see this thread: https://t.me/tdlibchat/15084
-                hadMin = true;
-                this.log.debug('received min peer: %j', peer);
+                // no point in caching min peers as we can't use them
                 continue;
             }
             count += 1;
@@ -266,8 +261,9 @@ export class BaseTelegramClient extends EventEmitter {
         if (count > 0) {
             await this.storage.updatePeers(parsedPeers);
             this.log.debug('cached %d peers', count);
+            return true;
         }
-        return hadMin;
+        return false;
     }
     /**
      * Export current session to a single *LONG* string, containing

package/esm/base-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"base-client.js","sourceRoot":"","sources":["../../src/base-client.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,YAAY,MAAM,QAAQ,CAAA;AACjC,OAAO,IAAI,MAAM,MAAM,CAAA;AAEvB,OAAO,EAAE,EAAE,EAAE,MAAM,YAAY,CAAA;AAC/B,OAAO,EAAE,aAAa,IAAI,gBAAgB,EAAE,MAAM,6BAA6B,CAAA;AAC/E,OAAO,EAAE,aAAa,IAAI,gBAAgB,EAAE,MAAM,6BAA6B,CAAA;AAG/E,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAA;AAE3D,OAAO,EAAE,cAAc,EAA6C,MAAM,8BAA8B,CAAA;AAExG,OAAO,EAAoB,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAEpE,OAAO,EAEH,yBAAyB,EAEzB,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,eAAe,EAEf,UAAU,EACV,iBAAiB,EACjB,mBAAmB,EACnB,kBAAkB,GACrB,MAAM,kBAAkB,CAAA;AAyKzB;;;GAGG;AACH,MAAM,OAAO,kBAAmB,SAAQ,YAAY;IAwDhD,YAAY,IAA+B;QACvC,KAAK,EAAE,CAAA;QAjBX,uDAAuD;QAC7C,oBAAe,GAAG,CAAC,CAAA;QAEpB,YAAO,GAAG,IAAI,aAAa,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QAE9E,0DAA0D;QAClD,eAAU,GAAwC,KAAK,CAAA;QAOtD,QAAG,GAAG,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAA;QA6EvC,uBAAkB,GAA4B,EAAE,CAAA;QAvE5C,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAA;QAEhF,IAAI,KAAK,CAAC,KAAK,CAAC,EAAE;YACd,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAA;SACjE;QAED,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE;YAC7B,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAA;SACjC;QAED,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,4BAA4B,CAAC,EAAE,CAAA;QAC7D,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,aAAa,EAAE,CAAA;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAA;QAC5B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACrC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAEvC,IAAI,EAAE,GAAG,IAAI,CAAC,UAAU,CAAA;QAExB,IAAI,CAAC,EAAE,EAAE;YACL,IAAI,IAAI,CAAC,SAAS,EAAE;gBAChB,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAA;aACzD;iBAAM;gBACH,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,mBAAmB,CAAA;aACrE;SACJ;QAED,IAAI,CAAC,WAAW,GAAG,EAAE,CAAA;QACrB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAA;QAE1C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,KAAK,CAAA;QAC5C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,IAAI,gBAAgB,CAAA;QACpD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,IAAI,gBAAgB,CAAA;QAEpD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAC7B;YACI,KAAK;YACL,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,cAAc,EAAE,IAAI,CAAC,cAAc,IAAI,KAAK;YAC5C,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,SAAS;YACxB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;YACtC,mBAAmB,EAAE,IAAI,CAAC,mBAAmB,IAAI,KAAK;YACtD,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;YAC9B,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC;YACjD,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,IAAI,KAAK;YACxD,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;SAC1B,EACD,IAAI,CAAC,OAAO,CACf,CAAA;QAED,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;IACpE,CAAC;IAES,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IAC3B,CAAC;IAES,KAAK,CAAC,YAAY;QACxB,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAA;IAC/B,CAAC;IAID,iBAAiB,CAAC,EAAuB;QACrC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACpC,CAAC;IAED,oBAAoB,CAAC,EAAuB;QACxC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAC7E,CAAC;IAED,KAAK,CAAC,WAAW;QACb,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,kBAAkB,EAAE;YACtC,MAAM,EAAE,EAAE,CAAA;SACb;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAA;IAC/B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,uBAAuB;YACvB,MAAM,IAAI,CAAC,UAAU,CAAA;YAErB,OAAM;SACT;QAED,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,UAAU,GAAG,yBAAyB,EAAE,CAAC,CAAA;QAE/D,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAA;QAChC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAA;QACzB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAA;QACpD,IAAI,SAAS,KAAK,IAAI;YAAE,IAAI,CAAC,WAAW,GAAG,SAAS,CAAA;QAEpD,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAEnF,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE;YAC9D,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,CAAA;YAEjE,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,SAAS,EAAE;gBAClC,MAAM,IAAI,KAAK,CACX,sDAAsD;oBAClD,cAAc,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,uBACzC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAC9B,EAAE,CACT,CAAA;aACJ;YAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,CAAA;YAClC,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;YAEjD,IAAI,IAAI,CAAC,IAAI,EAAE;gBACX,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;aACxC;YAED,uDAAuD;YACvD,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;YAEvE,MAAM,IAAI,CAAC,WAAW,EAAE,CAAA;SAC3B;QAED,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE3B,IAAI,CAAC,OAAO;aACP,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC;aACzB,IAAI,CAAC,GAAG,EAAE;YACP,OAAO,CAAC,OAAO,EAAE,CAAA;YACjB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;QAC1B,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAA;IACpD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACP,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAEzB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAA;QACtB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAA;QAEtB,MAAM,IAAI,CAAC,WAAW,EAAE,CAAA;QACxB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAA;QAE9B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACvB,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,KAAK,CAAC,IAAI,CACN,OAAmC,EACnC,MAAuB;QAEvB,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE;YAC1B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;SACvB;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAA;QAE9D,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,CAAA;QAC3D,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;QAE/B,+DAA+D;QAC/D,OAAO,GAAG,CAAA;IACd,CAAC;IAED;;;;;;;;;OASG;IACH,eAAe,CAAC,OAAyB;QACrC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;IACzC,CAAC;IAED;;;;;;;;OAQG;IACH,OAAO,CAAC,OAA+D;QACnE,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAA;IAC3B,CAAC;IAED,cAAc,CAAC,IAA8B;QACzC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;QACjC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAA;IAChC,CAAC;IAED,UAAU,CAAC,GAAY,EAAE,UAA8B;QACnD,IAAI,IAAI,CAAC,QAAQ,EAAE;YACf,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,UAAU,CAAC,CAAA;SACjC;aAAM;YACH,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;SACrB;IACL,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe,CAAC,GAAW;QAC7B,MAAM,WAAW,GAAgC,EAAE,CAAA;QAEnD,IAAI,MAAM,GAAG,KAAK,CAAA;QAClB,IAAI,KAAK,GAAG,CAAC,CAAA;QAEb,KAAK,MAAM,IAAI,IAAI,eAAe,CAAC,GAAkB,CAAC,EAAE;YACpD,IAAK,IAAY,CAAC,GAAG,EAAE;gBACnB,+DAA+D;gBAC/D,gDAAgD;gBAChD,MAAM,GAAG,IAAI,CAAA;gBACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAA;gBAC7C,SAAQ;aACX;YAED,KAAK,IAAI,CAAC,CAAA;YAEV,QAAQ,IAAI,CAAC,CAAC,EAAE;gBACZ,KAAK,MAAM;oBACP,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;wBAClB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,uCAAuC,EAAE,IAAI,CAAC,CAAA;wBAC5D,SAAQ;qBACX;oBAED,WAAW,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,IAAI,CAAC,EAAE;wBACX,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE;wBACtC,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI;qBACb,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,MAAM,CAAC;gBACZ,KAAK,eAAe;oBAChB,WAAW,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE;wBACZ,UAAU,EAAE,IAAI,CAAC,IAAI;wBACrB,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI;qBACb,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,SAAS,CAAC;gBACf,KAAK,kBAAkB;oBACnB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;wBAClB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,uCAAuC,EAAE,IAAI,CAAC,CAAA;wBAC5D,SAAQ;qBACX;oBACD,WAAW,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;wBAChC,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS;wBACzE,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,IAAI;qBACb,CAAC,CAAA;oBACF,MAAK;aACZ;SACJ;QAED,IAAI,KAAK,GAAG,CAAC,EAAE;YACX,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAA;YAC3C,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAA;SAC3C;QAED,OAAO,MAAM,CAAA;IACjB,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,aAAa;QACf,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAA;QACrD,IAAI,CAAC,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAA;QAErD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QACpE,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;QAE1D,OAAO,kBAAkB,CAAC,IAAI,CAAC,UAAU,EAAE;YACvC,OAAO,EAAE,CAAC;YACV,IAAI,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YAClC,QAAQ,EAAE,IAAI,CAAC,SAAS;YACxB,UAAU;YACV,OAAO;SACV,CAAC,CAAA;IACN,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,aAAa,CAAC,OAAe,EAAE,KAAK,GAAG,KAAK;QACxC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAA;QAC1B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;IAC7B,CAAC;CACJ","sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport EventEmitter from 'events'\nimport Long from 'long'\n\nimport { tl } from '@mtcute/tl'\nimport { __tlReaderMap as defaultReaderMap } from '@mtcute/tl/binary/reader.js'\nimport { __tlWriterMap as defaultWriterMap } from '@mtcute/tl/binary/writer.js'\nimport { TlReaderMap, TlWriterMap } from '@mtcute/tl-runtime'\n\nimport { ConfigManager } from './network/config-manager.js'\nimport { ReconnectionStrategy, SessionConnection, TransportFactory } from './network/index.js'\nimport { NetworkManager, NetworkManagerExtraParams, RpcCallOptions } from './network/network-manager.js'\nimport { PersistentConnectionParams } from './network/persistent-connection.js'\nimport { ITelegramStorage, MemoryStorage } from './storage/index.js'\nimport { MustEqual } from './types/index.js'\nimport {\n    ControllablePromise,\n    createControllablePromise,\n    CryptoProviderFactory,\n    defaultCryptoProviderFactory,\n    defaultProductionDc,\n    defaultProductionIpv6Dc,\n    defaultTestDc,\n    defaultTestIpv6Dc,\n    getAllPeersFrom,\n    ICryptoProvider,\n    LogManager,\n    readStringSession,\n    toggleChannelIdMark,\n    writeStringSession,\n} from './utils/index.js'\n\n/** Options for {@link BaseTelegramClient} */\nexport interface BaseTelegramClientOptions {\n    /**\n     * API ID from my.telegram.org\n     */\n    apiId: number | string\n    /**\n     * API hash from my.telegram.org\n     */\n    apiHash: string\n\n    /**\n     * Telegram storage to use.\n     * If omitted, {@link MemoryStorage} is used\n     */\n    storage?: ITelegramStorage\n\n    /**\n     * Cryptography provider factory to allow delegating\n     * crypto to native addon, worker, etc.\n     */\n    crypto?: CryptoProviderFactory\n\n    /**\n     * Whether to use IPv6 datacenters\n     * (IPv6 will be preferred when choosing a DC by id)\n     * (default: false)\n     */\n    useIpv6?: boolean\n\n    /**\n     * Primary DC to use for initial connection.\n     * This does not mean this will be the only DC used,\n     * nor that this DC will actually be primary, this only\n     * determines the first DC the library will try to connect to.\n     * Can be used to connect to other networks (like test DCs).\n     *\n     * When session already contains primary DC, this parameter is ignored.\n     *\n     * @default  Production DC 2.\n     */\n    defaultDcs?: ITelegramStorage.DcOptions\n\n    /**\n     * Whether to connect to test servers.\n     *\n     * If passed, {@link defaultDc} defaults to Test DC 2.\n     *\n     * **Must** be passed if using test servers, even if\n     * you passed custom {@link defaultDc}\n     */\n    testMode?: boolean\n\n    /**\n     * Additional options for initConnection call.\n     * `apiId` and `query` are not available and will be ignored.\n     * Omitted values will be filled with defaults\n     */\n    initConnectionOptions?: Partial<Omit<tl.RawInitConnectionRequest, 'apiId' | 'query'>>\n\n    /**\n     * Transport factory to use in the client.\n     *\n     * @default  platform-specific transport: WebSocket on the web, TCP in node\n     */\n    transport?: TransportFactory\n\n    /**\n     * Reconnection strategy.\n     *\n     * @default  simple reconnection strategy: first 0ms, then up to 5s (increasing by 1s)\n     */\n    reconnectionStrategy?: ReconnectionStrategy<PersistentConnectionParams>\n\n    /**\n     * Maximum duration of a flood_wait that will be waited automatically.\n     * Flood waits above this threshold will throw a FloodWaitError.\n     * Set to 0 to disable. Can be overridden with `throwFlood` parameter in call() params\n     *\n     * @default 10000\n     */\n    floodSleepThreshold?: number\n\n    /**\n     * Maximum number of retries when calling RPC methods.\n     * Call is retried when InternalError or FloodWaitError is encountered.\n     * Can be set to Infinity.\n     *\n     * @default 5\n     */\n    maxRetryCount?: number\n\n    /**\n     * If true, every single API call will be wrapped with `tl.invokeWithoutUpdates`,\n     * effectively disabling the server-sent events for the clients.\n     * May be useful in some cases.\n     *\n     * Note that this only wraps calls made with `.call()` within the primary\n     * connection. Additional connections and direct `.sendForResult()` calls\n     * must be wrapped manually.\n     *\n     * @default false\n     */\n    disableUpdates?: boolean\n\n    /**\n     * mtcute can send all unknown RPC errors to [danog](https://github.com/danog)'s\n     * [error reporting service](https://rpc.pwrtelegram.xyz/).\n     *\n     * This is fully anonymous (except maybe IP) and is only used to improve the library\n     * and developer experience for everyone working with MTProto. This is fully opt-in,\n     * and if you're too paranoid, you can disable it by manually passing `enableErrorReporting: false` to the client.\n     *\n     * @default false\n     */\n    enableErrorReporting?: boolean\n\n    /**\n     * If true, RPC errors will have a stack trace of the initial `.call()`\n     * or `.sendForResult()` call position, which drastically improves\n     * debugging experience.<br>\n     * If false, they will have a stack trace of mtcute internals.\n     *\n     * Internally this creates a stack capture before every RPC call\n     * and stores it until the result is received. This might\n     * use a lot more memory than normal, thus can be disabled here.\n     *\n     * @default true\n     */\n    niceStacks?: boolean\n\n    /**\n     * Extra parameters for {@link NetworkManager}\n     */\n    network?: NetworkManagerExtraParams\n\n    /**\n     * Set logging level for the client.\n     *\n     * See static members of {@link LogManager} for possible values.\n     */\n    logLevel?: number\n\n    /**\n     * **EXPERT USE ONLY!**\n     *\n     * Override TL layer used for the connection.\n     *\n     * **Does not** change the schema used.\n     */\n    overrideLayer?: number\n\n    /**\n     * **EXPERT USE ONLY**\n     *\n     * Override reader map used for the connection.\n     */\n    readerMap?: TlReaderMap\n\n    /**\n     * **EXPERT USE ONLY**\n     *\n     * Override writer map used for the connection.\n     */\n    writerMap?: TlWriterMap\n}\n\n/**\n * Basic Telegram client that only implements the bare minimum\n * to make RPC calls and receive low-level updates.\n */\nexport class BaseTelegramClient extends EventEmitter {\n    /**\n     * Crypto provider taken from {@link BaseTelegramClientOptions.crypto}\n     */\n    readonly crypto: ICryptoProvider\n\n    /**\n     * Telegram storage taken from {@link BaseTelegramClientOptions.storage}\n     */\n    readonly storage: ITelegramStorage\n\n    /**\n     * API hash taken from {@link BaseTelegramClientOptions.apiHash}\n     */\n    protected readonly _apiHash: string\n\n    /**\n     * \"Use IPv6\" taken from {@link BaseTelegramClientOptions.useIpv6}\n     */\n    protected readonly _useIpv6: boolean\n\n    /**\n     * \"Test mode\" taken from {@link BaseTelegramClientOptions.testMode}\n     */\n    protected readonly _testMode: boolean\n\n    /**\n     * Primary DCs taken from {@link BaseTelegramClientOptions.defaultDcs},\n     * loaded from session or changed by other means (like redirecting).\n     */\n    protected _defaultDcs: ITelegramStorage.DcOptions\n\n    private _niceStacks: boolean\n    /** TL layer used by the client */\n    readonly _layer: number\n    /** TL readers map used by the client */\n    readonly _readerMap: TlReaderMap\n    /** TL writers map used by the client */\n    readonly _writerMap: TlWriterMap\n\n    /** Unix timestamp when the last update was received */\n    protected _lastUpdateTime = 0\n\n    readonly _config = new ConfigManager(() => this.call({ _: 'help.getConfig' }))\n\n    // not really connected, but rather \"connect() was called\"\n    private _connected: ControllablePromise<void> | boolean = false\n\n    private _onError?: (err: unknown, connection?: SessionConnection) => void\n\n    private _importFrom?: string\n    private _importForce?: boolean\n\n    readonly log = new LogManager('client')\n    readonly network: NetworkManager\n\n    constructor(opts: BaseTelegramClientOptions) {\n        super()\n\n        const apiId = typeof opts.apiId === 'string' ? parseInt(opts.apiId) : opts.apiId\n\n        if (isNaN(apiId)) {\n            throw new Error('apiId must be a number or a numeric string!')\n        }\n\n        if (opts.logLevel !== undefined) {\n            this.log.level = opts.logLevel\n        }\n\n        this.crypto = (opts.crypto ?? defaultCryptoProviderFactory)()\n        this.storage = opts.storage ?? new MemoryStorage()\n        this._apiHash = opts.apiHash\n        this._useIpv6 = Boolean(opts.useIpv6)\n        this._testMode = Boolean(opts.testMode)\n\n        let dc = opts.defaultDcs\n\n        if (!dc) {\n            if (this._testMode) {\n                dc = this._useIpv6 ? defaultTestIpv6Dc : defaultTestDc\n            } else {\n                dc = this._useIpv6 ? defaultProductionIpv6Dc : defaultProductionDc\n            }\n        }\n\n        this._defaultDcs = dc\n        this._niceStacks = opts.niceStacks ?? true\n\n        this._layer = opts.overrideLayer ?? tl.LAYER\n        this._readerMap = opts.readerMap ?? defaultReaderMap\n        this._writerMap = opts.writerMap ?? defaultWriterMap\n\n        this.network = new NetworkManager(\n            {\n                apiId,\n                crypto: this.crypto,\n                disableUpdates: opts.disableUpdates ?? false,\n                initConnectionOptions: opts.initConnectionOptions,\n                layer: this._layer,\n                log: this.log,\n                readerMap: this._readerMap,\n                writerMap: this._writerMap,\n                reconnectionStrategy: opts.reconnectionStrategy,\n                storage: this.storage,\n                testMode: this._testMode,\n                transport: opts.transport,\n                _emitError: this._emitError.bind(this),\n                floodSleepThreshold: opts.floodSleepThreshold ?? 10000,\n                maxRetryCount: opts.maxRetryCount ?? 5,\n                isPremium: false,\n                useIpv6: Boolean(opts.useIpv6),\n                keepAliveAction: this._keepAliveAction.bind(this),\n                enableErrorReporting: opts.enableErrorReporting ?? false,\n                onUsable: () => this.emit('usable'),\n                ...(opts.network ?? {}),\n            },\n            this._config,\n        )\n\n        this.storage.setup?.(this.log, this._readerMap, this._writerMap)\n    }\n\n    protected _keepAliveAction(): void {\n        this.emit('keep_alive')\n    }\n\n    protected async _loadStorage(): Promise<void> {\n        await this.storage.load?.()\n    }\n\n    _beforeStorageSave: (() => Promise<void>)[] = []\n\n    beforeStorageSave(cb: () => Promise<void>): void {\n        this._beforeStorageSave.push(cb)\n    }\n\n    offBeforeStorageSave(cb: () => Promise<void>): void {\n        this._beforeStorageSave = this._beforeStorageSave.filter((x) => x !== cb)\n    }\n\n    async saveStorage(): Promise<void> {\n        for (const cb of this._beforeStorageSave) {\n            await cb()\n        }\n        await this.storage.save?.()\n    }\n\n    /**\n     * Initialize the connection to the primary DC.\n     *\n     * You shouldn't usually call this method directly as it is called\n     * implicitly the first time you call {@link call}.\n     */\n    async connect(): Promise<void> {\n        if (this._connected) {\n            // avoid double-connect\n            await this._connected\n\n            return\n        }\n\n        const promise = (this._connected = createControllablePromise())\n\n        await this.crypto.initialize?.()\n        await this._loadStorage()\n        const primaryDc = await this.storage.getDefaultDcs()\n        if (primaryDc !== null) this._defaultDcs = primaryDc\n\n        const defaultDcAuthKey = await this.storage.getAuthKeyFor(this._defaultDcs.main.id)\n\n        if ((this._importForce || !defaultDcAuthKey) && this._importFrom) {\n            const data = readStringSession(this._readerMap, this._importFrom)\n\n            if (data.testMode !== this._testMode) {\n                throw new Error(\n                    'This session string is not for the current backend. ' +\n                        `Session is ${data.testMode ? 'test' : 'prod'}, but the client is ${\n                            this._testMode ? 'test' : 'prod'\n                        }`,\n                )\n            }\n\n            this._defaultDcs = data.primaryDcs\n            await this.storage.setDefaultDcs(data.primaryDcs)\n\n            if (data.self) {\n                await this.storage.setSelf(data.self)\n            }\n\n            // await this.primaryConnection.setupKeys(data.authKey)\n            await this.storage.setAuthKeyFor(data.primaryDcs.main.id, data.authKey)\n\n            await this.saveStorage()\n        }\n\n        this.emit('before_connect')\n\n        this.network\n            .connect(this._defaultDcs)\n            .then(() => {\n                promise.resolve()\n                this._connected = true\n            })\n            .catch((err: Error) => this._emitError(err))\n    }\n\n    /**\n     * Close all connections and finalize the client.\n     */\n    async close(): Promise<void> {\n        this.emit('before_close')\n\n        this._config.destroy()\n        this.network.destroy()\n\n        await this.saveStorage()\n        await this.storage.destroy?.()\n\n        this.emit('closed')\n    }\n\n    /**\n     * Make an RPC call to the primary DC.\n     * This method handles DC migration, flood waits and retries automatically.\n     *\n     * If you want more low-level control, use\n     * `primaryConnection.sendForResult()` (which is what this method wraps)\n     *\n     * This method is still quite low-level and you shouldn't use this\n     * when using high-level API provided by `@mtcute/client`.\n     *\n     * @param message  RPC method to call\n     * @param params  Additional call parameters\n     */\n    async call<T extends tl.RpcMethod>(\n        message: MustEqual<T, tl.RpcMethod>,\n        params?: RpcCallOptions,\n    ): Promise<tl.RpcCallReturn[T['_']]> {\n        if (this._connected !== true) {\n            await this.connect()\n        }\n\n        const stack = this._niceStacks ? new Error().stack : undefined\n\n        const res = await this.network.call(message, params, stack)\n        await this._cachePeersFrom(res)\n\n        // eslint-disable-next-line @typescript-eslint/no-unsafe-return\n        return res\n    }\n\n    /**\n     * Change transport for the client.\n     *\n     * Can be used, for example, to change proxy at runtime\n     *\n     * This effectively calls `changeTransport()` on\n     * `primaryConnection` and all additional connections.\n     *\n     * @param factory  New transport factory\n     */\n    changeTransport(factory: TransportFactory): void {\n        this.network.changeTransport(factory)\n    }\n\n    /**\n     * Register an error handler for the client\n     *\n     * @param handler\n     *     Error handler. Called with one or two parameters.\n     *     The first one is always the error, and the second is\n     *     the connection in which the error has occurred, in case\n     *     this was connection-related error.\n     */\n    onError(handler: (err: unknown, connection?: SessionConnection) => void): void {\n        this._onError = handler\n    }\n\n    notifyLoggedIn(auth: tl.auth.RawAuthorization): void {\n        this.network.notifyLoggedIn(auth)\n        this.emit('logged_in', auth)\n    }\n\n    _emitError(err: unknown, connection?: SessionConnection): void {\n        if (this._onError) {\n            this._onError(err, connection)\n        } else {\n            console.error(err)\n        }\n    }\n\n    /**\n     * Adds all peers from a given object to entity cache in storage.\n     *\n     * @returns  `true` if there were any `min` peers\n     */\n    async _cachePeersFrom(obj: object): Promise<boolean> {\n        const parsedPeers: ITelegramStorage.PeerInfo[] = []\n\n        let hadMin = false\n        let count = 0\n\n        for (const peer of getAllPeersFrom(obj as tl.TlObject)) {\n            if ((peer as any).min) {\n                // absolutely incredible min peer handling, courtesy of levlam.\n                // see this thread: https://t.me/tdlibchat/15084\n                hadMin = true\n                this.log.debug('received min peer: %j', peer)\n                continue\n            }\n\n            count += 1\n\n            switch (peer._) {\n                case 'user':\n                    if (!peer.accessHash) {\n                        this.log.warn('received user without access hash: %j', peer)\n                        continue\n                    }\n\n                    parsedPeers.push({\n                        id: peer.id,\n                        accessHash: peer.accessHash,\n                        username: peer.username?.toLowerCase(),\n                        phone: peer.phone,\n                        type: 'user',\n                        full: peer,\n                    })\n                    break\n                case 'chat':\n                case 'chatForbidden':\n                    parsedPeers.push({\n                        id: -peer.id,\n                        accessHash: Long.ZERO,\n                        type: 'chat',\n                        full: peer,\n                    })\n                    break\n                case 'channel':\n                case 'channelForbidden':\n                    if (!peer.accessHash) {\n                        this.log.warn('received user without access hash: %j', peer)\n                        continue\n                    }\n                    parsedPeers.push({\n                        id: toggleChannelIdMark(peer.id),\n                        accessHash: peer.accessHash,\n                        username: peer._ === 'channel' ? peer.username?.toLowerCase() : undefined,\n                        type: 'channel',\n                        full: peer,\n                    })\n                    break\n            }\n        }\n\n        if (count > 0) {\n            await this.storage.updatePeers(parsedPeers)\n            this.log.debug('cached %d peers', count)\n        }\n\n        return hadMin\n    }\n\n    /**\n     * Export current session to a single *LONG* string, containing\n     * all the needed information.\n     *\n     * > **Warning!** Anyone with this string will be able\n     * > to authorize as you and do anything. Treat this\n     * > as your password, and never give it away!\n     * >\n     * > In case you have accidentally leaked this string,\n     * > make sure to revoke this session in account settings:\n     * > \"Privacy & Security\" > \"Active sessions\" >\n     * > find the one containing `mtcute` > Revoke,\n     * > or, in case this is a bot, revoke bot token\n     * > with [@BotFather](//t.me/botfather)\n     */\n    async exportSession(): Promise<string> {\n        const primaryDcs = await this.storage.getDefaultDcs()\n        if (!primaryDcs) throw new Error('No default DC set')\n\n        const authKey = await this.storage.getAuthKeyFor(primaryDcs.main.id)\n        if (!authKey) throw new Error('Auth key is not ready yet')\n\n        return writeStringSession(this._writerMap, {\n            version: 2,\n            self: await this.storage.getSelf(),\n            testMode: this._testMode,\n            primaryDcs,\n            authKey,\n        })\n    }\n\n    /**\n     * Request the session to be imported from the given session string.\n     *\n     * Note that the string will not be parsed and imported right away,\n     * instead, it will be imported when `connect()` is called\n     *\n     * Also note that the session will only be imported in case\n     * the storage is missing authorization (i.e. does not contain\n     * auth key for the primary DC), otherwise it will be ignored (unless `force).\n     *\n     * @param session  Session string to import\n     * @param force  Whether to overwrite existing session\n     */\n    importSession(session: string, force = false): void {\n        this._importFrom = session\n        this._importForce = force\n    }\n}\n"]}
+{"version":3,"file":"base-client.js","sourceRoot":"","sources":["../../src/base-client.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,YAAY,MAAM,QAAQ,CAAA;AACjC,OAAO,IAAI,MAAM,MAAM,CAAA;AAEvB,OAAO,EAAE,EAAE,EAAE,MAAM,YAAY,CAAA;AAC/B,OAAO,EAAE,aAAa,IAAI,gBAAgB,EAAE,MAAM,6BAA6B,CAAA;AAC/E,OAAO,EAAE,aAAa,IAAI,gBAAgB,EAAE,MAAM,6BAA6B,CAAA;AAG/E,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAA;AAE3D,OAAO,EAAE,cAAc,EAA6C,MAAM,8BAA8B,CAAA;AAIxG,OAAO,EAEH,yBAAyB,EAEzB,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,eAAe,EAEf,UAAU,EACV,iBAAiB,EACjB,mBAAmB,EACnB,kBAAkB,GACrB,MAAM,kBAAkB,CAAA;AAwKzB;;;GAGG;AACH,MAAM,OAAO,kBAAmB,SAAQ,YAAY;IAwDhD,YAAY,IAA+B;QACvC,KAAK,EAAE,CAAA;QAjBX,uDAAuD;QAC7C,oBAAe,GAAG,CAAC,CAAA;QAEpB,YAAO,GAAG,IAAI,aAAa,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QAE9E,0DAA0D;QAClD,eAAU,GAAwC,KAAK,CAAA;QAOtD,QAAG,GAAG,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAA;QA8EvC,uBAAkB,GAA4B,EAAE,CAAA;QAxE5C,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAA;QAEhF,IAAI,KAAK,CAAC,KAAK,CAAC,EAAE;YACd,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAA;SACjE;QAED,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE;YAC7B,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAA;SACjC;QAED,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,4BAA4B,CAAC,EAAE,CAAA;QAE7D,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;QAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAA;QAC5B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACrC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAEvC,IAAI,EAAE,GAAG,IAAI,CAAC,UAAU,CAAA;QAExB,IAAI,CAAC,EAAE,EAAE;YACL,IAAI,IAAI,CAAC,SAAS,EAAE;gBAChB,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa,CAAA;aACzD;iBAAM;gBACH,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,mBAAmB,CAAA;aACrE;SACJ;QAED,IAAI,CAAC,WAAW,GAAG,EAAE,CAAA;QACrB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAA;QAE1C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,KAAK,CAAA;QAC5C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,IAAI,gBAAgB,CAAA;QACpD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,IAAI,gBAAgB,CAAA;QAEpD,IAAI,CAAC,OAAO,GAAG,IAAI,cAAc,CAC7B;YACI,KAAK;YACL,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,cAAc,EAAE,IAAI,CAAC,cAAc,IAAI,KAAK;YAC5C,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,SAAS;YACxB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;YACtC,mBAAmB,EAAE,IAAI,CAAC,mBAAmB,IAAI,KAAK;YACtD,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;YAC9B,eAAe,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC;YACjD,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,IAAI,KAAK;YACxD,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;YACnC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;SAC1B,EACD,IAAI,CAAC,OAAO,CACf,CAAA;QAED,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;IACpE,CAAC;IAES,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IAC3B,CAAC;IAES,KAAK,CAAC,YAAY;QACxB,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAA;IAC/B,CAAC;IAID,iBAAiB,CAAC,EAAuB;QACrC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACpC,CAAC;IAED,oBAAoB,CAAC,EAAuB;QACxC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAA;IAC7E,CAAC;IAED,KAAK,CAAC,WAAW;QACb,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,kBAAkB,EAAE;YACtC,MAAM,EAAE,EAAE,CAAA;SACb;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAA;IAC/B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,uBAAuB;YACvB,MAAM,IAAI,CAAC,UAAU,CAAA;YAErB,OAAM;SACT;QAED,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,UAAU,GAAG,yBAAyB,EAAE,CAAC,CAAA;QAE/D,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAA;QAChC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAA;QACzB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAA;QACpD,IAAI,SAAS,KAAK,IAAI;YAAE,IAAI,CAAC,WAAW,GAAG,SAAS,CAAA;QAEpD,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAEnF,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE;YAC9D,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,CAAA;YAEjE,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,SAAS,EAAE;gBAClC,MAAM,IAAI,KAAK,CACX,sDAAsD;oBAClD,cAAc,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,uBACzC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAC9B,EAAE,CACT,CAAA;aACJ;YAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,CAAA;YAClC,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;YAEjD,IAAI,IAAI,CAAC,IAAI,EAAE;gBACX,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;aACxC;YAED,uDAAuD;YACvD,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;YAEvE,MAAM,IAAI,CAAC,WAAW,EAAE,CAAA;SAC3B;QAED,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE3B,IAAI,CAAC,OAAO;aACP,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC;aACzB,IAAI,CAAC,GAAG,EAAE;YACP,OAAO,CAAC,OAAO,EAAE,CAAA;YACjB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;QAC1B,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAA;IACpD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACP,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;QAEzB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAA;QACtB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAA;QAEtB,MAAM,IAAI,CAAC,WAAW,EAAE,CAAA;QACxB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAA;QAE9B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACvB,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,KAAK,CAAC,IAAI,CACN,OAAmC,EACnC,MAAuB;QAEvB,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE;YAC1B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;SACvB;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAA;QAE9D,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,CAAA;QAE3D,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE;YACjC,MAAM,IAAI,CAAC,WAAW,EAAE,CAAA;SAC3B;QAED,+DAA+D;QAC/D,OAAO,GAAG,CAAA;IACd,CAAC;IAED;;;;;;;;;OASG;IACH,eAAe,CAAC,OAAyB;QACrC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;IACzC,CAAC;IAED;;;;;;;;OAQG;IACH,OAAO,CAAC,OAA+D;QACnE,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAA;IAC3B,CAAC;IAED,cAAc,CAAC,IAA8B;QACzC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,CAAA;QACjC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAA;IAChC,CAAC;IAED,UAAU,CAAC,GAAY,EAAE,UAA8B;QACnD,IAAI,IAAI,CAAC,QAAQ,EAAE;YACf,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,UAAU,CAAC,CAAA;SACjC;aAAM;YACH,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;SACrB;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,GAAW;QAC7B,MAAM,WAAW,GAAgC,EAAE,CAAA;QAEnD,IAAI,KAAK,GAAG,CAAC,CAAA;QAEb,KAAK,MAAM,IAAI,IAAI,eAAe,CAAC,GAAkB,CAAC,EAAE;YACpD,IAAK,IAAY,CAAC,GAAG,EAAE;gBACnB,qDAAqD;gBACrD,SAAQ;aACX;YAED,KAAK,IAAI,CAAC,CAAA;YAEV,QAAQ,IAAI,CAAC,CAAC,EAAE;gBACZ,KAAK,MAAM;oBACP,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;wBAClB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,uCAAuC,EAAE,IAAI,CAAC,CAAA;wBAC5D,SAAQ;qBACX;oBAED,WAAW,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,IAAI,CAAC,EAAE;wBACX,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE;wBACtC,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI;qBACb,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,MAAM,CAAC;gBACZ,KAAK,eAAe;oBAChB,WAAW,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE;wBACZ,UAAU,EAAE,IAAI,CAAC,IAAI;wBACrB,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI;qBACb,CAAC,CAAA;oBACF,MAAK;gBACT,KAAK,SAAS,CAAC;gBACf,KAAK,kBAAkB;oBACnB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;wBAClB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,uCAAuC,EAAE,IAAI,CAAC,CAAA;wBAC5D,SAAQ;qBACX;oBACD,WAAW,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;wBAChC,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS;wBACzE,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,IAAI;qBACb,CAAC,CAAA;oBACF,MAAK;aACZ;SACJ;QAED,IAAI,KAAK,GAAG,CAAC,EAAE;YACX,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAA;YAC3C,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAA;YAExC,OAAO,IAAI,CAAA;SACd;QAED,OAAO,KAAK,CAAA;IAChB,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,aAAa;QACf,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAA;QACrD,IAAI,CAAC,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAA;QAErD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QACpE,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;QAE1D,OAAO,kBAAkB,CAAC,IAAI,CAAC,UAAU,EAAE;YACvC,OAAO,EAAE,CAAC;YACV,IAAI,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YAClC,QAAQ,EAAE,IAAI,CAAC,SAAS;YACxB,UAAU;YACV,OAAO;SACV,CAAC,CAAA;IACN,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,aAAa,CAAC,OAAe,EAAE,KAAK,GAAG,KAAK;QACxC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAA;QAC1B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;IAC7B,CAAC;CACJ","sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport EventEmitter from 'events'\nimport Long from 'long'\n\nimport { tl } from '@mtcute/tl'\nimport { __tlReaderMap as defaultReaderMap } from '@mtcute/tl/binary/reader.js'\nimport { __tlWriterMap as defaultWriterMap } from '@mtcute/tl/binary/writer.js'\nimport { TlReaderMap, TlWriterMap } from '@mtcute/tl-runtime'\n\nimport { ConfigManager } from './network/config-manager.js'\nimport { ReconnectionStrategy, SessionConnection, TransportFactory } from './network/index.js'\nimport { NetworkManager, NetworkManagerExtraParams, RpcCallOptions } from './network/network-manager.js'\nimport { PersistentConnectionParams } from './network/persistent-connection.js'\nimport { ITelegramStorage } from './storage/index.js'\nimport { MustEqual } from './types/index.js'\nimport {\n    ControllablePromise,\n    createControllablePromise,\n    CryptoProviderFactory,\n    defaultCryptoProviderFactory,\n    defaultProductionDc,\n    defaultProductionIpv6Dc,\n    defaultTestDc,\n    defaultTestIpv6Dc,\n    getAllPeersFrom,\n    ICryptoProvider,\n    LogManager,\n    readStringSession,\n    toggleChannelIdMark,\n    writeStringSession,\n} from './utils/index.js'\n\n/** Options for {@link BaseTelegramClient} */\nexport interface BaseTelegramClientOptions {\n    /**\n     * API ID from my.telegram.org\n     */\n    apiId: number | string\n    /**\n     * API hash from my.telegram.org\n     */\n    apiHash: string\n\n    /**\n     * Storage to use for this client.\n     */\n    storage: ITelegramStorage\n\n    /**\n     * Cryptography provider factory to allow delegating\n     * crypto to native addon, worker, etc.\n     */\n    crypto?: CryptoProviderFactory\n\n    /**\n     * Whether to use IPv6 datacenters\n     * (IPv6 will be preferred when choosing a DC by id)\n     * (default: false)\n     */\n    useIpv6?: boolean\n\n    /**\n     * Primary DC to use for initial connection.\n     * This does not mean this will be the only DC used,\n     * nor that this DC will actually be primary, this only\n     * determines the first DC the library will try to connect to.\n     * Can be used to connect to other networks (like test DCs).\n     *\n     * When session already contains primary DC, this parameter is ignored.\n     *\n     * @default  Production DC 2.\n     */\n    defaultDcs?: ITelegramStorage.DcOptions\n\n    /**\n     * Whether to connect to test servers.\n     *\n     * If passed, {@link defaultDc} defaults to Test DC 2.\n     *\n     * **Must** be passed if using test servers, even if\n     * you passed custom {@link defaultDc}\n     */\n    testMode?: boolean\n\n    /**\n     * Additional options for initConnection call.\n     * `apiId` and `query` are not available and will be ignored.\n     * Omitted values will be filled with defaults\n     */\n    initConnectionOptions?: Partial<Omit<tl.RawInitConnectionRequest, 'apiId' | 'query'>>\n\n    /**\n     * Transport factory to use in the client.\n     *\n     * @default  platform-specific transport: WebSocket on the web, TCP in node\n     */\n    transport?: TransportFactory\n\n    /**\n     * Reconnection strategy.\n     *\n     * @default  simple reconnection strategy: first 0ms, then up to 5s (increasing by 1s)\n     */\n    reconnectionStrategy?: ReconnectionStrategy<PersistentConnectionParams>\n\n    /**\n     * Maximum duration of a flood_wait that will be waited automatically.\n     * Flood waits above this threshold will throw a FloodWaitError.\n     * Set to 0 to disable. Can be overridden with `throwFlood` parameter in call() params\n     *\n     * @default 10000\n     */\n    floodSleepThreshold?: number\n\n    /**\n     * Maximum number of retries when calling RPC methods.\n     * Call is retried when InternalError or FloodWaitError is encountered.\n     * Can be set to Infinity.\n     *\n     * @default 5\n     */\n    maxRetryCount?: number\n\n    /**\n     * If true, every single API call will be wrapped with `tl.invokeWithoutUpdates`,\n     * effectively disabling the server-sent events for the clients.\n     * May be useful in some cases.\n     *\n     * Note that this only wraps calls made with `.call()` within the primary\n     * connection. Additional connections and direct `.sendForResult()` calls\n     * must be wrapped manually.\n     *\n     * @default false\n     */\n    disableUpdates?: boolean\n\n    /**\n     * mtcute can send all unknown RPC errors to [danog](https://github.com/danog)'s\n     * [error reporting service](https://rpc.pwrtelegram.xyz/).\n     *\n     * This is fully anonymous (except maybe IP) and is only used to improve the library\n     * and developer experience for everyone working with MTProto. This is fully opt-in,\n     * and if you're too paranoid, you can disable it by manually passing `enableErrorReporting: false` to the client.\n     *\n     * @default false\n     */\n    enableErrorReporting?: boolean\n\n    /**\n     * If true, RPC errors will have a stack trace of the initial `.call()`\n     * or `.sendForResult()` call position, which drastically improves\n     * debugging experience.<br>\n     * If false, they will have a stack trace of mtcute internals.\n     *\n     * Internally this creates a stack capture before every RPC call\n     * and stores it until the result is received. This might\n     * use a lot more memory than normal, thus can be disabled here.\n     *\n     * @default true\n     */\n    niceStacks?: boolean\n\n    /**\n     * Extra parameters for {@link NetworkManager}\n     */\n    network?: NetworkManagerExtraParams\n\n    /**\n     * Set logging level for the client.\n     *\n     * See static members of {@link LogManager} for possible values.\n     */\n    logLevel?: number\n\n    /**\n     * **EXPERT USE ONLY!**\n     *\n     * Override TL layer used for the connection.\n     *\n     * **Does not** change the schema used.\n     */\n    overrideLayer?: number\n\n    /**\n     * **EXPERT USE ONLY**\n     *\n     * Override reader map used for the connection.\n     */\n    readerMap?: TlReaderMap\n\n    /**\n     * **EXPERT USE ONLY**\n     *\n     * Override writer map used for the connection.\n     */\n    writerMap?: TlWriterMap\n}\n\n/**\n * Basic Telegram client that only implements the bare minimum\n * to make RPC calls and receive low-level updates.\n */\nexport class BaseTelegramClient extends EventEmitter {\n    /**\n     * Crypto provider taken from {@link BaseTelegramClientOptions.crypto}\n     */\n    readonly crypto: ICryptoProvider\n\n    /**\n     * Telegram storage taken from {@link BaseTelegramClientOptions.storage}\n     */\n    readonly storage: ITelegramStorage\n\n    /**\n     * API hash taken from {@link BaseTelegramClientOptions.apiHash}\n     */\n    protected readonly _apiHash: string\n\n    /**\n     * \"Use IPv6\" taken from {@link BaseTelegramClientOptions.useIpv6}\n     */\n    protected readonly _useIpv6: boolean\n\n    /**\n     * \"Test mode\" taken from {@link BaseTelegramClientOptions.testMode}\n     */\n    protected readonly _testMode: boolean\n\n    /**\n     * Primary DCs taken from {@link BaseTelegramClientOptions.defaultDcs},\n     * loaded from session or changed by other means (like redirecting).\n     */\n    protected _defaultDcs: ITelegramStorage.DcOptions\n\n    private _niceStacks: boolean\n    /** TL layer used by the client */\n    readonly _layer: number\n    /** TL readers map used by the client */\n    readonly _readerMap: TlReaderMap\n    /** TL writers map used by the client */\n    readonly _writerMap: TlWriterMap\n\n    /** Unix timestamp when the last update was received */\n    protected _lastUpdateTime = 0\n\n    readonly _config = new ConfigManager(() => this.call({ _: 'help.getConfig' }))\n\n    // not really connected, but rather \"connect() was called\"\n    private _connected: ControllablePromise<void> | boolean = false\n\n    private _onError?: (err: unknown, connection?: SessionConnection) => void\n\n    private _importFrom?: string\n    private _importForce?: boolean\n\n    readonly log = new LogManager('client')\n    readonly network: NetworkManager\n\n    constructor(opts: BaseTelegramClientOptions) {\n        super()\n\n        const apiId = typeof opts.apiId === 'string' ? parseInt(opts.apiId) : opts.apiId\n\n        if (isNaN(apiId)) {\n            throw new Error('apiId must be a number or a numeric string!')\n        }\n\n        if (opts.logLevel !== undefined) {\n            this.log.level = opts.logLevel\n        }\n\n        this.crypto = (opts.crypto ?? defaultCryptoProviderFactory)()\n\n        this.storage = opts.storage\n        this._apiHash = opts.apiHash\n        this._useIpv6 = Boolean(opts.useIpv6)\n        this._testMode = Boolean(opts.testMode)\n\n        let dc = opts.defaultDcs\n\n        if (!dc) {\n            if (this._testMode) {\n                dc = this._useIpv6 ? defaultTestIpv6Dc : defaultTestDc\n            } else {\n                dc = this._useIpv6 ? defaultProductionIpv6Dc : defaultProductionDc\n            }\n        }\n\n        this._defaultDcs = dc\n        this._niceStacks = opts.niceStacks ?? true\n\n        this._layer = opts.overrideLayer ?? tl.LAYER\n        this._readerMap = opts.readerMap ?? defaultReaderMap\n        this._writerMap = opts.writerMap ?? defaultWriterMap\n\n        this.network = new NetworkManager(\n            {\n                apiId,\n                crypto: this.crypto,\n                disableUpdates: opts.disableUpdates ?? false,\n                initConnectionOptions: opts.initConnectionOptions,\n                layer: this._layer,\n                log: this.log,\n                readerMap: this._readerMap,\n                writerMap: this._writerMap,\n                reconnectionStrategy: opts.reconnectionStrategy,\n                storage: this.storage,\n                testMode: this._testMode,\n                transport: opts.transport,\n                _emitError: this._emitError.bind(this),\n                floodSleepThreshold: opts.floodSleepThreshold ?? 10000,\n                maxRetryCount: opts.maxRetryCount ?? 5,\n                isPremium: false,\n                useIpv6: Boolean(opts.useIpv6),\n                keepAliveAction: this._keepAliveAction.bind(this),\n                enableErrorReporting: opts.enableErrorReporting ?? false,\n                onUsable: () => this.emit('usable'),\n                ...(opts.network ?? {}),\n            },\n            this._config,\n        )\n\n        this.storage.setup?.(this.log, this._readerMap, this._writerMap)\n    }\n\n    protected _keepAliveAction(): void {\n        this.emit('keep_alive')\n    }\n\n    protected async _loadStorage(): Promise<void> {\n        await this.storage.load?.()\n    }\n\n    _beforeStorageSave: (() => Promise<void>)[] = []\n\n    beforeStorageSave(cb: () => Promise<void>): void {\n        this._beforeStorageSave.push(cb)\n    }\n\n    offBeforeStorageSave(cb: () => Promise<void>): void {\n        this._beforeStorageSave = this._beforeStorageSave.filter((x) => x !== cb)\n    }\n\n    async saveStorage(): Promise<void> {\n        for (const cb of this._beforeStorageSave) {\n            await cb()\n        }\n        await this.storage.save?.()\n    }\n\n    /**\n     * Initialize the connection to the primary DC.\n     *\n     * You shouldn't usually call this method directly as it is called\n     * implicitly the first time you call {@link call}.\n     */\n    async connect(): Promise<void> {\n        if (this._connected) {\n            // avoid double-connect\n            await this._connected\n\n            return\n        }\n\n        const promise = (this._connected = createControllablePromise())\n\n        await this.crypto.initialize?.()\n        await this._loadStorage()\n        const primaryDc = await this.storage.getDefaultDcs()\n        if (primaryDc !== null) this._defaultDcs = primaryDc\n\n        const defaultDcAuthKey = await this.storage.getAuthKeyFor(this._defaultDcs.main.id)\n\n        if ((this._importForce || !defaultDcAuthKey) && this._importFrom) {\n            const data = readStringSession(this._readerMap, this._importFrom)\n\n            if (data.testMode !== this._testMode) {\n                throw new Error(\n                    'This session string is not for the current backend. ' +\n                        `Session is ${data.testMode ? 'test' : 'prod'}, but the client is ${\n                            this._testMode ? 'test' : 'prod'\n                        }`,\n                )\n            }\n\n            this._defaultDcs = data.primaryDcs\n            await this.storage.setDefaultDcs(data.primaryDcs)\n\n            if (data.self) {\n                await this.storage.setSelf(data.self)\n            }\n\n            // await this.primaryConnection.setupKeys(data.authKey)\n            await this.storage.setAuthKeyFor(data.primaryDcs.main.id, data.authKey)\n\n            await this.saveStorage()\n        }\n\n        this.emit('before_connect')\n\n        this.network\n            .connect(this._defaultDcs)\n            .then(() => {\n                promise.resolve()\n                this._connected = true\n            })\n            .catch((err: Error) => this._emitError(err))\n    }\n\n    /**\n     * Close all connections and finalize the client.\n     */\n    async close(): Promise<void> {\n        this.emit('before_close')\n\n        this._config.destroy()\n        this.network.destroy()\n\n        await this.saveStorage()\n        await this.storage.destroy?.()\n\n        this.emit('closed')\n    }\n\n    /**\n     * Make an RPC call to the primary DC.\n     * This method handles DC migration, flood waits and retries automatically.\n     *\n     * If you want more low-level control, use\n     * `primaryConnection.sendForResult()` (which is what this method wraps)\n     *\n     * This method is still quite low-level and you shouldn't use this\n     * when using high-level API provided by `@mtcute/client`.\n     *\n     * @param message  RPC method to call\n     * @param params  Additional call parameters\n     */\n    async call<T extends tl.RpcMethod>(\n        message: MustEqual<T, tl.RpcMethod>,\n        params?: RpcCallOptions,\n    ): Promise<tl.RpcCallReturn[T['_']]> {\n        if (this._connected !== true) {\n            await this.connect()\n        }\n\n        const stack = this._niceStacks ? new Error().stack : undefined\n\n        const res = await this.network.call(message, params, stack)\n\n        if (await this._cachePeersFrom(res)) {\n            await this.saveStorage()\n        }\n\n        // eslint-disable-next-line @typescript-eslint/no-unsafe-return\n        return res\n    }\n\n    /**\n     * Change transport for the client.\n     *\n     * Can be used, for example, to change proxy at runtime\n     *\n     * This effectively calls `changeTransport()` on\n     * `primaryConnection` and all additional connections.\n     *\n     * @param factory  New transport factory\n     */\n    changeTransport(factory: TransportFactory): void {\n        this.network.changeTransport(factory)\n    }\n\n    /**\n     * Register an error handler for the client\n     *\n     * @param handler\n     *     Error handler. Called with one or two parameters.\n     *     The first one is always the error, and the second is\n     *     the connection in which the error has occurred, in case\n     *     this was connection-related error.\n     */\n    onError(handler: (err: unknown, connection?: SessionConnection) => void): void {\n        this._onError = handler\n    }\n\n    notifyLoggedIn(auth: tl.auth.RawAuthorization): void {\n        this.network.notifyLoggedIn(auth)\n        this.emit('logged_in', auth)\n    }\n\n    _emitError(err: unknown, connection?: SessionConnection): void {\n        if (this._onError) {\n            this._onError(err, connection)\n        } else {\n            console.error(err)\n        }\n    }\n\n    /**\n     * Adds all peers from a given object to entity cache in storage.\n     */\n    async _cachePeersFrom(obj: object): Promise<boolean> {\n        const parsedPeers: ITelegramStorage.PeerInfo[] = []\n\n        let count = 0\n\n        for (const peer of getAllPeersFrom(obj as tl.TlObject)) {\n            if ((peer as any).min) {\n                // no point in caching min peers as we can't use them\n                continue\n            }\n\n            count += 1\n\n            switch (peer._) {\n                case 'user':\n                    if (!peer.accessHash) {\n                        this.log.warn('received user without access hash: %j', peer)\n                        continue\n                    }\n\n                    parsedPeers.push({\n                        id: peer.id,\n                        accessHash: peer.accessHash,\n                        username: peer.username?.toLowerCase(),\n                        phone: peer.phone,\n                        type: 'user',\n                        full: peer,\n                    })\n                    break\n                case 'chat':\n                case 'chatForbidden':\n                    parsedPeers.push({\n                        id: -peer.id,\n                        accessHash: Long.ZERO,\n                        type: 'chat',\n                        full: peer,\n                    })\n                    break\n                case 'channel':\n                case 'channelForbidden':\n                    if (!peer.accessHash) {\n                        this.log.warn('received user without access hash: %j', peer)\n                        continue\n                    }\n                    parsedPeers.push({\n                        id: toggleChannelIdMark(peer.id),\n                        accessHash: peer.accessHash,\n                        username: peer._ === 'channel' ? peer.username?.toLowerCase() : undefined,\n                        type: 'channel',\n                        full: peer,\n                    })\n                    break\n            }\n        }\n\n        if (count > 0) {\n            await this.storage.updatePeers(parsedPeers)\n            this.log.debug('cached %d peers', count)\n\n            return true\n        }\n\n        return false\n    }\n\n    /**\n     * Export current session to a single *LONG* string, containing\n     * all the needed information.\n     *\n     * > **Warning!** Anyone with this string will be able\n     * > to authorize as you and do anything. Treat this\n     * > as your password, and never give it away!\n     * >\n     * > In case you have accidentally leaked this string,\n     * > make sure to revoke this session in account settings:\n     * > \"Privacy & Security\" > \"Active sessions\" >\n     * > find the one containing `mtcute` > Revoke,\n     * > or, in case this is a bot, revoke bot token\n     * > with [@BotFather](//t.me/botfather)\n     */\n    async exportSession(): Promise<string> {\n        const primaryDcs = await this.storage.getDefaultDcs()\n        if (!primaryDcs) throw new Error('No default DC set')\n\n        const authKey = await this.storage.getAuthKeyFor(primaryDcs.main.id)\n        if (!authKey) throw new Error('Auth key is not ready yet')\n\n        return writeStringSession(this._writerMap, {\n            version: 2,\n            self: await this.storage.getSelf(),\n            testMode: this._testMode,\n            primaryDcs,\n            authKey,\n        })\n    }\n\n    /**\n     * Request the session to be imported from the given session string.\n     *\n     * Note that the string will not be parsed and imported right away,\n     * instead, it will be imported when `connect()` is called\n     *\n     * Also note that the session will only be imported in case\n     * the storage is missing authorization (i.e. does not contain\n     * auth key for the primary DC), otherwise it will be ignored (unless `force).\n     *\n     * @param session  Session string to import\n     * @param force  Whether to overwrite existing session\n     */\n    importSession(session: string, force = false): void {\n        this._importFrom = session\n        this._importForce = force\n    }\n}\n"]}

package/esm/index.d.ts

@@ -4,7 +4,7 @@ export * from './storage/index.js';
 export * from './types/index.js';
 export * from './utils/peer-utils.js';
 /** @hidden */
-export * from '@mtcute/tl';
+export { mtp, tl } from '@mtcute/tl';
 import Long from 'long';
 /** @hidden */
 export { Long };

package/esm/index.js

@@ -4,7 +4,7 @@ export * from './storage/index.js';
 export * from './types/index.js';
 export * from './utils/peer-utils.js';
 /** @hidden */
-export * from '@mtcute/tl';
+export { mtp, tl } from '@mtcute/tl';
 import Long from 'long';
 /** @hidden */
 export { Long };

package/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAA;AAChC,cAAc,oBAAoB,CAAA;AAClC,cAAc,oBAAoB,CAAA;AAClC,cAAc,kBAAkB,CAAA;AAChC,cAAc,uBAAuB,CAAA;AAErC,cAAc;AACd,cAAc,YAAY,CAAA;AAC1B,OAAO,IAAI,MAAM,MAAM,CAAA;AACvB,cAAc;AACd,OAAO,EAAE,IAAI,EAAE,CAAA","sourcesContent":["export * from './base-client.js'\nexport * from './network/index.js'\nexport * from './storage/index.js'\nexport * from './types/index.js'\nexport * from './utils/peer-utils.js'\n\n/** @hidden */\nexport * from '@mtcute/tl'\nimport Long from 'long'\n/** @hidden */\nexport { Long }\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAA;AAChC,cAAc,oBAAoB,CAAA;AAClC,cAAc,oBAAoB,CAAA;AAClC,cAAc,kBAAkB,CAAA;AAChC,cAAc,uBAAuB,CAAA;AAErC,cAAc;AACd,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,YAAY,CAAA;AACpC,OAAO,IAAI,MAAM,MAAM,CAAA;AACvB,cAAc;AACd,OAAO,EAAE,IAAI,EAAE,CAAA","sourcesContent":["export * from './base-client.js'\nexport * from './network/index.js'\nexport * from './storage/index.js'\nexport * from './types/index.js'\nexport * from './utils/peer-utils.js'\n\n/** @hidden */\nexport { mtp, tl } from '@mtcute/tl'\nimport Long from 'long'\n/** @hidden */\nexport { Long }\n"]}

package/esm/network/index.d.ts

@@ -1,4 +1,4 @@
-export { ConnectionKind, NetworkManagerExtraParams, RpcCallOptions } from './network-manager.js';
+export type { ConnectionKind, NetworkManagerExtraParams, RpcCallOptions } from './network-manager.js';
 export * from './reconnection.js';
 export * from './session-connection.js';
 export * from './transports/index.js';

package/esm/network/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/network/index.ts"],"names":[],"mappings":"AACA,cAAc,mBAAmB,CAAA;AACjC,cAAc,yBAAyB,CAAA;AACvC,cAAc,uBAAuB,CAAA","sourcesContent":["export { ConnectionKind, NetworkManagerExtraParams, RpcCallOptions } from './network-manager.js'\nexport * from './reconnection.js'\nexport * from './session-connection.js'\nexport * from './transports/index.js'\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/network/index.ts"],"names":[],"mappings":"AACA,cAAc,mBAAmB,CAAA;AACjC,cAAc,yBAAyB,CAAA;AACvC,cAAc,uBAAuB,CAAA","sourcesContent":["export type { ConnectionKind, NetworkManagerExtraParams, RpcCallOptions } from './network-manager.js'\nexport * from './reconnection.js'\nexport * from './session-connection.js'\nexport * from './transports/index.js'\n"]}

package/esm/network/multi-session-connection.js

@@ -1,4 +1,5 @@
 import EventEmitter from 'events';
+import { createControllablePromise } from '../utils/index.js';
 import { MtprotoSession } from './mtproto-session.js';
 import { SessionConnection } from './session-connection.js';
 export class MultiSessionConnection extends EventEmitter {
@@ -92,8 +93,19 @@ export class MultiSessionConnection extends EventEmitter {
             return;
         }
         if (enforcePfsChanged) {
-            this._connections.forEach((conn) => {
-                conn.setUsePfs(this.params.usePfs || this._enforcePfs);
+            // we need to fetch new auth keys first
+            const promise = createControllablePromise();
+            this.emit('request-keys', promise);
+            promise
+                .then(() => {
+                this._connections.forEach((conn) => {
+                    conn.setUsePfs(this.params.usePfs || this._enforcePfs);
+                    if (connect)
+                        conn.connect();
+                });
+            })
+                .catch((err) => {
+                this.emit('error', err);
             });
         }
         // create new connections
@@ -106,7 +118,7 @@ export class MultiSessionConnection extends EventEmitter {
                 isMainConnection: this.params.isMainConnection && i === 0,
                 withUpdates: this.params.isMainConnection && this.params.isMainDcConnection && !this.params.disableUpdates,
             }, session);
-            if (this.params.isMainConnection) {
+            if (this.params.isMainConnection && this.params.isMainDcConnection) {
                 conn.on('update', (update) => this.emit('update', update));
             }
             conn.on('error', (err) => this.emit('error', err, conn));
@@ -137,7 +149,8 @@ export class MultiSessionConnection extends EventEmitter {
                 this._connections.forEach((it) => it.flushWhenIdle());
             });
             this._connections.push(conn);
-            if (connect)
+            // if enforcePfsChanged, we need to connect after setting the new auth key
+            if (connect && !enforcePfsChanged)
                 conn.connect();
         }
     }

package/esm/network/multi-session-connection.js.map

@@ -1 +1 @@
-{"version":3,"file":"multi-session-connection.js","sourceRoot":"","sources":["../../../src/network/multi-session-connection.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AAKjC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAA;AACrD,OAAO,EAAE,iBAAiB,EAA2B,MAAM,yBAAyB,CAAA;AAGpF,MAAM,OAAO,sBAAuB,SAAQ,YAAY;IAKpD,YACa,MAA+B,EAChC,MAAc,EACtB,GAAW,EACX,SAAS,GAAG,EAAE;QAEd,KAAK,EAAE,CAAA;QALE,WAAM,GAAN,MAAM,CAAyB;QAChC,WAAM,GAAN,MAAM,CAAQ;QAJlB,gBAAW,GAAG,KAAK,CAAA;QAiBjB,iBAAY,GAAwB,EAAE,CAAA;QA2JhD,eAAU,GAAG,KAAK,CAAA;QASV,oBAAe,GAAG,CAAC,CAAA;QA5KvB,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAC/B,IAAI,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,IAAI,CAAA;QACnD,IAAI,CAAC,WAAW,GAAG,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,gBAAgB,CAAA;QAExD,IAAI,CAAC,SAAS,GAAG,EAAE,CAAA;QACnB,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC7B,CAAC;IAID,QAAQ,CAAC,KAAa,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;QAC1D,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QAEnB,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC;IAEO,eAAe;QACnB,sBAAsB;QACtB,kFAAkF;QAClF,uFAAuF;QACvF,uCAAuC;QACvC,gBAAgB;QAChB,uBAAuB;QACvB,8CAA8C;QAC9C,iCAAiC;QACjC,QAAQ;QACR,EAAE;QACF,yCAAyC;QACzC,+BAA+B;QAC/B,kCAAkC;QAClC,sCAAsC;QACtC,+CAA+C;QAC/C,yCAAyC;QACzC,yCAAyC;QACzC,iBAAiB;QACjB,YAAY;QACZ,QAAQ;QACR,EAAE;QACF,4CAA4C;QAC5C,0CAA0C;QAC1C,wCAAwC;QACxC,QAAQ;QACR,EAAE;QACF,aAAa;QACb,IAAI;QAEJ,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,mCAAmC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;QAExF,SAAS;QACT,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAM;QAEjD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACrC,yBAAyB;YACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC3D,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAA;aAC5B;YAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAElC,OAAM;SACT;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACxC,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAA;YACjC,MAAM,OAAO,GAAG,IAAI,cAAc,CAC9B,IAAI,CAAC,MAAM,CAAC,MAAM,EAClB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAC3B,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CACxB,CAAA;YAED,OAAO;YACP,IAAI,GAAG,KAAK,CAAC;gBAAE,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAA;YAE5D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;SAC/B;IACL,CAAC;IAEO,kBAAkB,CAAC,OAAO,GAAG,KAAK;QACtC,IAAI,CAAC,eAAe,EAAE,CAAA;QACtB,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAM;QAEpD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,sCAAsC,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;QAE9F,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAA;QACrE,MAAM,iBAAiB,GAAG,aAAa,KAAK,IAAI,CAAC,WAAW,CAAA;QAE5D,IAAI,iBAAiB,EAAE;YACnB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,EAAE,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,CAAA;YAChF,IAAI,CAAC,WAAW,GAAG,aAAa,CAAA;SACnC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACxC,4BAA4B;YAC5B,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC9D,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAA;gBACzC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;aACjC;YAED,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAErC,OAAM;SACT;QAED,IAAI,iBAAiB,EAAE;YACnB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,CAAA;YAC1D,CAAC,CAAC,CAAA;SACL;QAED,yBAAyB;QACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACzD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA,CAAC,sCAAsC;YACxE,oBAAoB;YACpB,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAC9B;gBACI,GAAG,IAAI,CAAC,MAAM;gBACd,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW;gBAC9C,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,KAAK,CAAC;gBACzD,WAAW,EACP,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc;aACpG,EACD,OAAO,CACV,CAAA;YAED,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE;gBAC9B,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;aAC7D;YACD,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAA;YACxD,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC1B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,CAAC,CAAA;gBAE/B,2BAA2B;gBAC3B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE;oBACnC,IAAI,KAAK,KAAK,IAAI;wBAAE,SAAQ;oBAC5B,KAAK,CAAC,WAAW,EAAE,CAAA;iBACtB;YACL,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC,CAAA;YACzF,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAA;gBAC5D,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAA;gBAE1B,wDAAwD;gBAExD,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;oBACnC,IAAI,KAAK,KAAK,IAAI;wBAAE,KAAK,CAAC,SAAS,EAAE,CAAA;gBACzC,CAAC,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAA;YAC/C,IAAI,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAA;YAC3D,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAA;gBAE5D,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,CAAA;YACzD,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC5B,IAAI,OAAO;gBAAE,IAAI,CAAC,OAAO,EAAE,CAAA;SAC9B;IACL,CAAC;IAGD,OAAO;QACH,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAA;QACnD,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;QAC9C,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAEzB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;IAC1B,CAAC;IAID,OAAO,CACH,OAAU,EACV,KAAc,EACd,OAAgB,EAChB,WAAyB;QAEzB,sCAAsC;QACtC,mCAAmC;QACnC,IAAI,GAAG,GAAG,QAAQ,CAAA;QAClB,IAAI,MAAM,GAAG,CAAC,CAAA;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACjC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,EAAE,CAAA;YAEnF,IAAI,KAAK,GAAG,GAAG,EAAE;gBACb,GAAG,GAAG,KAAK,CAAA;gBACX,MAAM,GAAG,CAAC,CAAA;aACb;SACJ;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC,CAAA;QAC9E,IAAI;QAEJ,0BAA0B;QAC1B,4EAA4E;QAC5E,2EAA2E;QAC3E,4BAA4B;QAC5B,wDAAwD;QACxD,qCAAqC;IACzC,CAAC;IAED,OAAO;QACH,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;YAClC,IAAI,CAAC,OAAO,EAAE,CAAA;SACjB;IACL,CAAC;IAED,eAAe;QACX,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW;YAAE,OAAM;QAE5C,IAAI,CAAC,OAAO,EAAE,CAAA;IAClB,CAAC;IAED,UAAU,CAAC,OAA0B,EAAE,IAAI,GAAG,KAAK,EAAE,GAAG,GAAG,CAAC;QACxD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAA;QACnC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAA;QAC1D,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IACtB,CAAC;IAED,aAAa;QACT,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,SAAS,EAAE;YAClC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;SACtB;QACD,IAAI,CAAC,eAAe,EAAE,CAAA;IAC1B,CAAC;IAED,oBAAoB,CAAC,OAAgB;QACjC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,kCAAkC,EAAE,OAAO,CAAC,CAAA;QAE5D,kCAAkC;QAClC,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,OAAO,CAAA;QAEvC,0BAA0B;QAC1B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;YAClC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;SACrC;IACL,CAAC;IAED,eAAe;QACX,qDAAqD;QACrD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;QAEjC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE;YACnD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,yDAAyD,CAAC,CAAA;YAE1E,OAAM;SACT;QAED,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE;YACpC,oDAAoD;YACpD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,0DAA0D,CAAC,CAAA;YAC3E,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAA;SACrC;QAED,iDAAiD;IACrD,CAAC;IAED,WAAW;QACP,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAA;IACrC,CAAC;IAED,aAAa;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE;YAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;gBAClC,IAAI,CAAC,aAAa,EAAE,CAAA;aACvB;SACJ;aAAM;YACH,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAA;SACvC;IACL,CAAC;IAED,eAAe,CAAC,OAAyB;QACrC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAA;IACtE,CAAC;IAED,WAAW;QACP,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAA;IACnC,CAAC;CACJ","sourcesContent":["import EventEmitter from 'events'\n\nimport { tl } from '@mtcute/tl'\n\nimport { Logger } from '../utils/index.js'\nimport { MtprotoSession } from './mtproto-session.js'\nimport { SessionConnection, SessionConnectionParams } from './session-connection.js'\nimport { TransportFactory } from './transports/index.js'\n\nexport class MultiSessionConnection extends EventEmitter {\n    private _log: Logger\n    readonly _sessions: MtprotoSession[]\n    private _enforcePfs = false\n\n    constructor(\n        readonly params: SessionConnectionParams,\n        private _count: number,\n        log: Logger,\n        logPrefix = '',\n    ) {\n        super()\n        this._log = log.create('multi')\n        if (logPrefix) this._log.prefix = `[${logPrefix}] `\n        this._enforcePfs = _count > 1 && params.isMainConnection\n\n        this._sessions = []\n        this._updateConnections()\n    }\n\n    protected _connections: SessionConnection[] = []\n\n    setCount(count: number, connect = this.params.isMainConnection): void {\n        this._count = count\n\n        this._updateConnections(connect)\n    }\n\n    private _updateSessions(): void {\n        // there are two cases\n        // 1. this msc is main, in which case every connection should have its own session\n        // 2. this msc is not main, in which case all connections should share the same session\n        // if (!this.params.isMainConnection) {\n        //     // case 2\n        //     this._log.debug(\n        //         'updating sessions count: %d -> 1',\n        //         this._sessions.length,\n        //     )\n        //\n        //     if (this._sessions.length === 0) {\n        //         this._sessions.push(\n        //             new MtprotoSession(\n        //                 this.params.crypto,\n        //                 this._log.create('session'),\n        //                 this.params.readerMap,\n        //                 this.params.writerMap,\n        //             ),\n        //         )\n        //     }\n        //\n        //     // shouldn't happen, but just in case\n        //     while (this._sessions.length > 1) {\n        //         this._sessions.pop()!.reset()\n        //     }\n        //\n        //     return\n        // }\n\n        this._log.debug('updating sessions count: %d -> %d', this._sessions.length, this._count)\n\n        // case 1\n        if (this._sessions.length === this._count) return\n\n        if (this._sessions.length > this._count) {\n            // destroy extra sessions\n            for (let i = this._sessions.length - 1; i >= this._count; i--) {\n                this._sessions[i].reset()\n            }\n\n            this._sessions.splice(this._count)\n\n            return\n        }\n\n        while (this._sessions.length < this._count) {\n            const idx = this._sessions.length\n            const session = new MtprotoSession(\n                this.params.crypto,\n                this._log.create('session'),\n                this.params.readerMap,\n                this.params.writerMap,\n            )\n\n            // brvh\n            if (idx !== 0) session._authKey = this._sessions[0]._authKey\n\n            this._sessions.push(session)\n        }\n    }\n\n    private _updateConnections(connect = false): void {\n        this._updateSessions()\n        if (this._connections.length === this._count) return\n\n        this._log.debug('updating connections count: %d -> %d', this._connections.length, this._count)\n\n        const newEnforcePfs = this._count > 1 && this.params.isMainConnection\n        const enforcePfsChanged = newEnforcePfs !== this._enforcePfs\n\n        if (enforcePfsChanged) {\n            this._log.debug('enforcePfs changed: %s -> %s', this._enforcePfs, newEnforcePfs)\n            this._enforcePfs = newEnforcePfs\n        }\n\n        if (this._connections.length > this._count) {\n            // destroy extra connections\n            for (let i = this._connections.length - 1; i >= this._count; i--) {\n                this._connections[i].removeAllListeners()\n                this._connections[i].destroy()\n            }\n\n            this._connections.splice(this._count)\n\n            return\n        }\n\n        if (enforcePfsChanged) {\n            this._connections.forEach((conn) => {\n                conn.setUsePfs(this.params.usePfs || this._enforcePfs)\n            })\n        }\n\n        // create new connections\n        for (let i = this._connections.length; i < this._count; i++) {\n            const session = this._sessions[i] // this.params.isMainConnection ? // :\n            // this._sessions[0]\n            const conn = new SessionConnection(\n                {\n                    ...this.params,\n                    usePfs: this.params.usePfs || this._enforcePfs,\n                    isMainConnection: this.params.isMainConnection && i === 0,\n                    withUpdates:\n                        this.params.isMainConnection && this.params.isMainDcConnection && !this.params.disableUpdates,\n                },\n                session,\n            )\n\n            if (this.params.isMainConnection) {\n                conn.on('update', (update) => this.emit('update', update))\n            }\n            conn.on('error', (err) => this.emit('error', err, conn))\n            conn.on('key-change', (key) => {\n                this.emit('key-change', i, key)\n\n                // notify other connections\n                for (const conn_ of this._connections) {\n                    if (conn_ === conn) continue\n                    conn_.onConnected()\n                }\n            })\n            conn.on('tmp-key-change', (key, expires) => this.emit('tmp-key-change', i, key, expires))\n            conn.on('auth-begin', () => {\n                this._log.debug('received auth-begin from connection %d', i)\n                this.emit('auth-begin', i)\n\n                // we need to reset temp auth keys if there are any left\n\n                this._connections.forEach((conn_) => {\n                    conn_._session._authKeyTemp.reset()\n                    if (conn_ !== conn) conn_.reconnect()\n                })\n            })\n            conn.on('usable', () => this.emit('usable', i))\n            conn.on('request-auth', () => this.emit('request-auth', i))\n            conn.on('flood-done', () => {\n                this._log.debug('received flood-done from connection %d', i)\n\n                this._connections.forEach((it) => it.flushWhenIdle())\n            })\n\n            this._connections.push(conn)\n            if (connect) conn.connect()\n        }\n    }\n\n    _destroyed = false\n    destroy(): void {\n        this._connections.forEach((conn) => conn.destroy())\n        this._sessions.forEach((sess) => sess.reset())\n        this.removeAllListeners()\n\n        this._destroyed = true\n    }\n\n    private _nextConnection = 0\n\n    sendRpc<T extends tl.RpcMethod>(\n        request: T,\n        stack?: string,\n        timeout?: number,\n        abortSignal?: AbortSignal,\n    ): Promise<tl.RpcCallReturn[T['_']]> {\n        // if (this.params.isMainConnection) {\n        // find the least loaded connection\n        let min = Infinity\n        let minIdx = 0\n\n        for (let i = 0; i < this._connections.length; i++) {\n            const conn = this._connections[i]\n            const total = conn._session.queuedRpc.length + conn._session.pendingMessages.size()\n\n            if (total < min) {\n                min = total\n                minIdx = i\n            }\n        }\n\n        return this._connections[minIdx].sendRpc(request, stack, timeout, abortSignal)\n        // }\n\n        // round-robin connections\n        // since they all share the same session, it doesn't matter which one we use\n        // the connection chosen here will only affect the first attempt at sending\n        // return this._connections[\n        //     this._nextConnection++ % this._connections.length\n        // ].sendRpc(request, stack, timeout)\n    }\n\n    connect(): void {\n        for (const conn of this._connections) {\n            conn.connect()\n        }\n    }\n\n    ensureConnected(): void {\n        if (this._connections[0].isConnected) return\n\n        this.connect()\n    }\n\n    setAuthKey(authKey: Uint8Array | null, temp = false, idx = 0): void {\n        const session = this._sessions[idx]\n        const key = temp ? session._authKeyTemp : session._authKey\n        key.setup(authKey)\n    }\n\n    resetAuthKeys(): void {\n        for (const session of this._sessions) {\n            session.reset(true)\n        }\n        this.notifyKeyChange()\n    }\n\n    setInactivityTimeout(timeout?: number): void {\n        this._log.debug('setting inactivity timeout to %s', timeout)\n\n        // for future connections (if any)\n        this.params.inactivityTimeout = timeout\n\n        // for current connections\n        for (const conn of this._connections) {\n            conn.setInactivityTimeout(timeout)\n        }\n    }\n\n    notifyKeyChange(): void {\n        // only expected to be called on non-main connections\n        const session = this._sessions[0]\n\n        if (this.params.usePfs && !session._authKeyTemp.ready) {\n            this._log.debug('temp auth key needed but not ready, ignoring key change')\n\n            return\n        }\n\n        if (this._sessions[0].queuedRpc.length) {\n            // there are pending requests, we need to reconnect.\n            this._log.debug('notifying key change on the connection due to queued rpc')\n            this._connections[0].onConnected()\n        }\n\n        // connection is idle, we don't need to notify it\n    }\n\n    requestAuth(): void {\n        this._connections[0]._authorize()\n    }\n\n    resetSessions(): void {\n        if (this.params.isMainConnection) {\n            for (const conn of this._connections) {\n                conn._resetSession()\n            }\n        } else {\n            this._connections[0]._resetSession()\n        }\n    }\n\n    changeTransport(factory: TransportFactory): void {\n        this._connections.forEach((conn) => conn.changeTransport(factory))\n    }\n\n    getPoolSize(): number {\n        return this._connections.length\n    }\n}\n"]}
+{"version":3,"file":"multi-session-connection.js","sourceRoot":"","sources":["../../../src/network/multi-session-connection.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AAIjC,OAAO,EAAE,yBAAyB,EAAU,MAAM,mBAAmB,CAAA;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAA;AACrD,OAAO,EAAE,iBAAiB,EAA2B,MAAM,yBAAyB,CAAA;AAGpF,MAAM,OAAO,sBAAuB,SAAQ,YAAY;IAKpD,YACa,MAA+B,EAChC,MAAc,EACtB,GAAW,EACX,SAAS,GAAG,EAAE;QAEd,KAAK,EAAE,CAAA;QALE,WAAM,GAAN,MAAM,CAAyB;QAChC,WAAM,GAAN,MAAM,CAAQ;QAJlB,gBAAW,GAAG,KAAK,CAAA;QAiBjB,iBAAY,GAAwB,EAAE,CAAA;QAwKhD,eAAU,GAAG,KAAK,CAAA;QASV,oBAAe,GAAG,CAAC,CAAA;QAzLvB,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAC/B,IAAI,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,IAAI,CAAA;QACnD,IAAI,CAAC,WAAW,GAAG,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,gBAAgB,CAAA;QAExD,IAAI,CAAC,SAAS,GAAG,EAAE,CAAA;QACnB,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC7B,CAAC;IAID,QAAQ,CAAC,KAAa,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;QAC1D,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QAEnB,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC;IAEO,eAAe;QACnB,sBAAsB;QACtB,kFAAkF;QAClF,uFAAuF;QACvF,uCAAuC;QACvC,gBAAgB;QAChB,uBAAuB;QACvB,8CAA8C;QAC9C,iCAAiC;QACjC,QAAQ;QACR,EAAE;QACF,yCAAyC;QACzC,+BAA+B;QAC/B,kCAAkC;QAClC,sCAAsC;QACtC,+CAA+C;QAC/C,yCAAyC;QACzC,yCAAyC;QACzC,iBAAiB;QACjB,YAAY;QACZ,QAAQ;QACR,EAAE;QACF,4CAA4C;QAC5C,0CAA0C;QAC1C,wCAAwC;QACxC,QAAQ;QACR,EAAE;QACF,aAAa;QACb,IAAI;QAEJ,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,mCAAmC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;QAExF,SAAS;QACT,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAM;QAEjD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACrC,yBAAyB;YACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC3D,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAA;aAC5B;YAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAElC,OAAM;SACT;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACxC,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAA;YACjC,MAAM,OAAO,GAAG,IAAI,cAAc,CAC9B,IAAI,CAAC,MAAM,CAAC,MAAM,EAClB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAC3B,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CACxB,CAAA;YAED,OAAO;YACP,IAAI,GAAG,KAAK,CAAC;gBAAE,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAA;YAE5D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;SAC/B;IACL,CAAC;IAEO,kBAAkB,CAAC,OAAO,GAAG,KAAK;QACtC,IAAI,CAAC,eAAe,EAAE,CAAA;QACtB,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAM;QAEpD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,sCAAsC,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;QAE9F,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAA;QACrE,MAAM,iBAAiB,GAAG,aAAa,KAAK,IAAI,CAAC,WAAW,CAAA;QAE5D,IAAI,iBAAiB,EAAE;YACnB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,EAAE,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,CAAA;YAChF,IAAI,CAAC,WAAW,GAAG,aAAa,CAAA;SACnC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACxC,4BAA4B;YAC5B,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC9D,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAA;gBACzC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;aACjC;YAED,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAErC,OAAM;SACT;QAED,IAAI,iBAAiB,EAAE;YACnB,uCAAuC;YACvC,MAAM,OAAO,GAAG,yBAAyB,EAAQ,CAAA;YACjD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;YAElC,OAAO;iBACF,IAAI,CAAC,GAAG,EAAE;gBACP,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,CAAA;oBAEtD,IAAI,OAAO;wBAAE,IAAI,CAAC,OAAO,EAAE,CAAA;gBAC/B,CAAC,CAAC,CAAA;YACN,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACX,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;YAC3B,CAAC,CAAC,CAAA;SACT;QAED,yBAAyB;QACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACzD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA,CAAC,sCAAsC;YACxE,oBAAoB;YACpB,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAC9B;gBACI,GAAG,IAAI,CAAC,MAAM;gBACd,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW;gBAC9C,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,KAAK,CAAC;gBACzD,WAAW,EACP,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc;aACpG,EACD,OAAO,CACV,CAAA;YAED,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE;gBAChE,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;aAC7D;YACD,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAA;YACxD,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC1B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,CAAC,CAAA;gBAE/B,2BAA2B;gBAC3B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE;oBACnC,IAAI,KAAK,KAAK,IAAI;wBAAE,SAAQ;oBAC5B,KAAK,CAAC,WAAW,EAAE,CAAA;iBACtB;YACL,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC,CAAA;YACzF,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAA;gBAC5D,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAA;gBAE1B,wDAAwD;gBAExD,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;oBACnC,IAAI,KAAK,KAAK,IAAI;wBAAE,KAAK,CAAC,SAAS,EAAE,CAAA;gBACzC,CAAC,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAA;YAC/C,IAAI,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAA;YAC3D,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAA;gBAE5D,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,CAAA;YACzD,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC5B,0EAA0E;YAC1E,IAAI,OAAO,IAAI,CAAC,iBAAiB;gBAAE,IAAI,CAAC,OAAO,EAAE,CAAA;SACpD;IACL,CAAC;IAGD,OAAO;QACH,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAA;QACnD,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;QAC9C,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAEzB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;IAC1B,CAAC;IAID,OAAO,CACH,OAAU,EACV,KAAc,EACd,OAAgB,EAChB,WAAyB;QAEzB,sCAAsC;QACtC,mCAAmC;QACnC,IAAI,GAAG,GAAG,QAAQ,CAAA;QAClB,IAAI,MAAM,GAAG,CAAC,CAAA;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACjC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,EAAE,CAAA;YAEnF,IAAI,KAAK,GAAG,GAAG,EAAE;gBACb,GAAG,GAAG,KAAK,CAAA;gBACX,MAAM,GAAG,CAAC,CAAA;aACb;SACJ;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC,CAAA;QAC9E,IAAI;QAEJ,0BAA0B;QAC1B,4EAA4E;QAC5E,2EAA2E;QAC3E,4BAA4B;QAC5B,wDAAwD;QACxD,qCAAqC;IACzC,CAAC;IAED,OAAO;QACH,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;YAClC,IAAI,CAAC,OAAO,EAAE,CAAA;SACjB;IACL,CAAC;IAED,eAAe;QACX,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW;YAAE,OAAM;QAE5C,IAAI,CAAC,OAAO,EAAE,CAAA;IAClB,CAAC;IAED,UAAU,CAAC,OAA0B,EAAE,IAAI,GAAG,KAAK,EAAE,GAAG,GAAG,CAAC;QACxD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAA;QACnC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAA;QAC1D,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IACtB,CAAC;IAED,aAAa;QACT,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,SAAS,EAAE;YAClC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;SACtB;QACD,IAAI,CAAC,eAAe,EAAE,CAAA;IAC1B,CAAC;IAED,oBAAoB,CAAC,OAAgB;QACjC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,kCAAkC,EAAE,OAAO,CAAC,CAAA;QAE5D,kCAAkC;QAClC,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,OAAO,CAAA;QAEvC,0BAA0B;QAC1B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;YAClC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;SACrC;IACL,CAAC;IAED,eAAe;QACX,qDAAqD;QACrD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;QAEjC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE;YACnD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,yDAAyD,CAAC,CAAA;YAE1E,OAAM;SACT;QAED,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE;YACpC,oDAAoD;YACpD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,0DAA0D,CAAC,CAAA;YAC3E,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAA;SACrC;QAED,iDAAiD;IACrD,CAAC;IAED,WAAW;QACP,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAA;IACrC,CAAC;IAED,aAAa;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE;YAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;gBAClC,IAAI,CAAC,aAAa,EAAE,CAAA;aACvB;SACJ;aAAM;YACH,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAA;SACvC;IACL,CAAC;IAED,eAAe,CAAC,OAAyB;QACrC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAA;IACtE,CAAC;IAED,WAAW;QACP,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAA;IACnC,CAAC;CACJ","sourcesContent":["import EventEmitter from 'events'\n\nimport { tl } from '@mtcute/tl'\n\nimport { createControllablePromise, Logger } from '../utils/index.js'\nimport { MtprotoSession } from './mtproto-session.js'\nimport { SessionConnection, SessionConnectionParams } from './session-connection.js'\nimport { TransportFactory } from './transports/index.js'\n\nexport class MultiSessionConnection extends EventEmitter {\n    private _log: Logger\n    readonly _sessions: MtprotoSession[]\n    private _enforcePfs = false\n\n    constructor(\n        readonly params: SessionConnectionParams,\n        private _count: number,\n        log: Logger,\n        logPrefix = '',\n    ) {\n        super()\n        this._log = log.create('multi')\n        if (logPrefix) this._log.prefix = `[${logPrefix}] `\n        this._enforcePfs = _count > 1 && params.isMainConnection\n\n        this._sessions = []\n        this._updateConnections()\n    }\n\n    protected _connections: SessionConnection[] = []\n\n    setCount(count: number, connect = this.params.isMainConnection): void {\n        this._count = count\n\n        this._updateConnections(connect)\n    }\n\n    private _updateSessions(): void {\n        // there are two cases\n        // 1. this msc is main, in which case every connection should have its own session\n        // 2. this msc is not main, in which case all connections should share the same session\n        // if (!this.params.isMainConnection) {\n        //     // case 2\n        //     this._log.debug(\n        //         'updating sessions count: %d -> 1',\n        //         this._sessions.length,\n        //     )\n        //\n        //     if (this._sessions.length === 0) {\n        //         this._sessions.push(\n        //             new MtprotoSession(\n        //                 this.params.crypto,\n        //                 this._log.create('session'),\n        //                 this.params.readerMap,\n        //                 this.params.writerMap,\n        //             ),\n        //         )\n        //     }\n        //\n        //     // shouldn't happen, but just in case\n        //     while (this._sessions.length > 1) {\n        //         this._sessions.pop()!.reset()\n        //     }\n        //\n        //     return\n        // }\n\n        this._log.debug('updating sessions count: %d -> %d', this._sessions.length, this._count)\n\n        // case 1\n        if (this._sessions.length === this._count) return\n\n        if (this._sessions.length > this._count) {\n            // destroy extra sessions\n            for (let i = this._sessions.length - 1; i >= this._count; i--) {\n                this._sessions[i].reset()\n            }\n\n            this._sessions.splice(this._count)\n\n            return\n        }\n\n        while (this._sessions.length < this._count) {\n            const idx = this._sessions.length\n            const session = new MtprotoSession(\n                this.params.crypto,\n                this._log.create('session'),\n                this.params.readerMap,\n                this.params.writerMap,\n            )\n\n            // brvh\n            if (idx !== 0) session._authKey = this._sessions[0]._authKey\n\n            this._sessions.push(session)\n        }\n    }\n\n    private _updateConnections(connect = false): void {\n        this._updateSessions()\n        if (this._connections.length === this._count) return\n\n        this._log.debug('updating connections count: %d -> %d', this._connections.length, this._count)\n\n        const newEnforcePfs = this._count > 1 && this.params.isMainConnection\n        const enforcePfsChanged = newEnforcePfs !== this._enforcePfs\n\n        if (enforcePfsChanged) {\n            this._log.debug('enforcePfs changed: %s -> %s', this._enforcePfs, newEnforcePfs)\n            this._enforcePfs = newEnforcePfs\n        }\n\n        if (this._connections.length > this._count) {\n            // destroy extra connections\n            for (let i = this._connections.length - 1; i >= this._count; i--) {\n                this._connections[i].removeAllListeners()\n                this._connections[i].destroy()\n            }\n\n            this._connections.splice(this._count)\n\n            return\n        }\n\n        if (enforcePfsChanged) {\n            // we need to fetch new auth keys first\n            const promise = createControllablePromise<void>()\n            this.emit('request-keys', promise)\n\n            promise\n                .then(() => {\n                    this._connections.forEach((conn) => {\n                        conn.setUsePfs(this.params.usePfs || this._enforcePfs)\n\n                        if (connect) conn.connect()\n                    })\n                })\n                .catch((err) => {\n                    this.emit('error', err)\n                })\n        }\n\n        // create new connections\n        for (let i = this._connections.length; i < this._count; i++) {\n            const session = this._sessions[i] // this.params.isMainConnection ? // :\n            // this._sessions[0]\n            const conn = new SessionConnection(\n                {\n                    ...this.params,\n                    usePfs: this.params.usePfs || this._enforcePfs,\n                    isMainConnection: this.params.isMainConnection && i === 0,\n                    withUpdates:\n                        this.params.isMainConnection && this.params.isMainDcConnection && !this.params.disableUpdates,\n                },\n                session,\n            )\n\n            if (this.params.isMainConnection && this.params.isMainDcConnection) {\n                conn.on('update', (update) => this.emit('update', update))\n            }\n            conn.on('error', (err) => this.emit('error', err, conn))\n            conn.on('key-change', (key) => {\n                this.emit('key-change', i, key)\n\n                // notify other connections\n                for (const conn_ of this._connections) {\n                    if (conn_ === conn) continue\n                    conn_.onConnected()\n                }\n            })\n            conn.on('tmp-key-change', (key, expires) => this.emit('tmp-key-change', i, key, expires))\n            conn.on('auth-begin', () => {\n                this._log.debug('received auth-begin from connection %d', i)\n                this.emit('auth-begin', i)\n\n                // we need to reset temp auth keys if there are any left\n\n                this._connections.forEach((conn_) => {\n                    conn_._session._authKeyTemp.reset()\n                    if (conn_ !== conn) conn_.reconnect()\n                })\n            })\n            conn.on('usable', () => this.emit('usable', i))\n            conn.on('request-auth', () => this.emit('request-auth', i))\n            conn.on('flood-done', () => {\n                this._log.debug('received flood-done from connection %d', i)\n\n                this._connections.forEach((it) => it.flushWhenIdle())\n            })\n\n            this._connections.push(conn)\n            // if enforcePfsChanged, we need to connect after setting the new auth key\n            if (connect && !enforcePfsChanged) conn.connect()\n        }\n    }\n\n    _destroyed = false\n    destroy(): void {\n        this._connections.forEach((conn) => conn.destroy())\n        this._sessions.forEach((sess) => sess.reset())\n        this.removeAllListeners()\n\n        this._destroyed = true\n    }\n\n    private _nextConnection = 0\n\n    sendRpc<T extends tl.RpcMethod>(\n        request: T,\n        stack?: string,\n        timeout?: number,\n        abortSignal?: AbortSignal,\n    ): Promise<tl.RpcCallReturn[T['_']]> {\n        // if (this.params.isMainConnection) {\n        // find the least loaded connection\n        let min = Infinity\n        let minIdx = 0\n\n        for (let i = 0; i < this._connections.length; i++) {\n            const conn = this._connections[i]\n            const total = conn._session.queuedRpc.length + conn._session.pendingMessages.size()\n\n            if (total < min) {\n                min = total\n                minIdx = i\n            }\n        }\n\n        return this._connections[minIdx].sendRpc(request, stack, timeout, abortSignal)\n        // }\n\n        // round-robin connections\n        // since they all share the same session, it doesn't matter which one we use\n        // the connection chosen here will only affect the first attempt at sending\n        // return this._connections[\n        //     this._nextConnection++ % this._connections.length\n        // ].sendRpc(request, stack, timeout)\n    }\n\n    connect(): void {\n        for (const conn of this._connections) {\n            conn.connect()\n        }\n    }\n\n    ensureConnected(): void {\n        if (this._connections[0].isConnected) return\n\n        this.connect()\n    }\n\n    setAuthKey(authKey: Uint8Array | null, temp = false, idx = 0): void {\n        const session = this._sessions[idx]\n        const key = temp ? session._authKeyTemp : session._authKey\n        key.setup(authKey)\n    }\n\n    resetAuthKeys(): void {\n        for (const session of this._sessions) {\n            session.reset(true)\n        }\n        this.notifyKeyChange()\n    }\n\n    setInactivityTimeout(timeout?: number): void {\n        this._log.debug('setting inactivity timeout to %s', timeout)\n\n        // for future connections (if any)\n        this.params.inactivityTimeout = timeout\n\n        // for current connections\n        for (const conn of this._connections) {\n            conn.setInactivityTimeout(timeout)\n        }\n    }\n\n    notifyKeyChange(): void {\n        // only expected to be called on non-main connections\n        const session = this._sessions[0]\n\n        if (this.params.usePfs && !session._authKeyTemp.ready) {\n            this._log.debug('temp auth key needed but not ready, ignoring key change')\n\n            return\n        }\n\n        if (this._sessions[0].queuedRpc.length) {\n            // there are pending requests, we need to reconnect.\n            this._log.debug('notifying key change on the connection due to queued rpc')\n            this._connections[0].onConnected()\n        }\n\n        // connection is idle, we don't need to notify it\n    }\n\n    requestAuth(): void {\n        this._connections[0]._authorize()\n    }\n\n    resetSessions(): void {\n        if (this.params.isMainConnection) {\n            for (const conn of this._connections) {\n                conn._resetSession()\n            }\n        } else {\n            this._connections[0]._resetSession()\n        }\n    }\n\n    changeTransport(factory: TransportFactory): void {\n        this._connections.forEach((conn) => conn.changeTransport(factory))\n    }\n\n    getPoolSize(): number {\n        return this._connections.length\n    }\n}\n"]}

package/esm/network/network-manager.d.ts

@@ -143,7 +143,7 @@ export declare class DcConnectionManager {
     private _setupMulti;
     setIsPrimary(isPrimary: boolean): void;
     setIsPremium(isPremium: boolean): void;
-    loadKeys(): Promise<boolean>;
+    loadKeys(forcePfs?: boolean): Promise<boolean>;
 }
 /**
  * Class that manages all connections to Telegram servers.

package/esm/network/network-manager.js

@@ -141,6 +141,12 @@ export class DcConnectionManager {
         connection.on('request-auth', () => {
             this.main.requestAuth();
         });
+        // fucking awesome architecture, but whatever
+        connection.on('request-keys', (promise) => {
+            this.loadKeys(true)
+                .then(() => promise.resolve())
+                .catch((e) => promise.reject(e));
+        });
         connection.on('error', (err, conn) => {
             this.manager.params._emitError(err, conn);
         });
@@ -149,6 +155,7 @@ export class DcConnectionManager {
         if (this.isPrimary === isPrimary)
             return;
         this.isPrimary = isPrimary;
+        this.main.params.isMainDcConnection = isPrimary;
         if (isPrimary) {
             this.main.setInactivityTimeout(undefined);
         }
@@ -161,7 +168,7 @@ export class DcConnectionManager {
         this.download.setCount(this.manager._connectionCount('download', this.dcId, isPremium));
         this.downloadSmall.setCount(this.manager._connectionCount('downloadSmall', this.dcId, isPremium));
     }
-    async loadKeys() {
+    async loadKeys(forcePfs = false) {
         const permanent = await this.manager._storage.getAuthKeyFor(this.dcId);
         this.main.setAuthKey(permanent);
         this.upload.setAuthKey(permanent);
@@ -170,7 +177,7 @@ export class DcConnectionManager {
         if (!permanent) {
             return false;
         }
-        if (this.manager.params.usePfs) {
+        if (this.manager.params.usePfs || forcePfs) {
             await Promise.all(this.main._sessions.map(async (_, i) => {
                 const temp = await this.manager._storage.getAuthKeyFor(this.dcId, i);
                 this.main.setAuthKey(temp, true, i);
@@ -215,7 +222,7 @@ export class NetworkManager {
             _: 'initConnection',
             deviceModel,
             systemVersion: '1.0',
-            appVersion: '0.1.2',
+            appVersion: '0.2.2',
             systemLangCode: 'en',
             langPack: '',
             langCode: 'en',
@@ -323,7 +330,7 @@ export class NetworkManager {
             // shouldn't happen
             throw new MtArgumentError('DC manager already exists');
         }
-        const dc = new DcConnectionManager(this, defaultDcs.main.id, defaultDcs);
+        const dc = new DcConnectionManager(this, defaultDcs.main.id, defaultDcs, true);
         this._dcConnections.set(defaultDcs.main.id, dc);
         await this._switchPrimaryDc(dc);
     }
@@ -390,7 +397,7 @@ export class NetworkManager {
             return;
         const options = await this._findDcOptions(newDc);
         if (!this._dcConnections.has(newDc)) {
-            this._dcConnections.set(newDc, new DcConnectionManager(this, newDc, options));
+            this._dcConnections.set(newDc, new DcConnectionManager(this, newDc, options, true));
         }
         await this._storage.setDefaultDcs(options);
         await this._switchPrimaryDc(this._dcConnections.get(newDc));