@mtcute/core 0.1.0 → 0.1.2

package/cjs/network/config-manager.test.js

@@ -0,0 +1,170 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const test_1 = require("@mtcute/test");
+const config_manager_js_1 = require("./config-manager.js");
+(0, vitest_1.describe)('ConfigManager', () => {
+    const config = (0, test_1.createStub)('config', {
+        expires: 300,
+    });
+    const getConfig = vitest_1.vi.fn();
+    (0, vitest_1.beforeEach)(() => {
+        vitest_1.vi.useFakeTimers();
+        vitest_1.vi.setSystemTime(0);
+        getConfig.mockClear().mockImplementation(() => Promise.resolve(config));
+    });
+    (0, vitest_1.afterEach)(() => void vitest_1.vi.useRealTimers());
+    (0, vitest_1.it)('should fetch initial config', async () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        const fetchedConfig = await cm.get();
+        (0, vitest_1.expect)(getConfig).toHaveBeenCalledTimes(1);
+        (0, vitest_1.expect)(fetchedConfig).toEqual(config);
+        (0, vitest_1.expect)(cm.getNow()).toEqual(config);
+    });
+    (0, vitest_1.it)('should automatically update config', async () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        await cm.update();
+        getConfig.mockImplementation(() => Promise.resolve({
+            ...config,
+            expires: 600,
+        }));
+        await vitest_1.vi.advanceTimersByTimeAsync(301000);
+        (0, vitest_1.expect)(getConfig).toHaveBeenCalledTimes(2);
+    });
+    (0, vitest_1.it)('should correctly determine stale config', () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        (0, vitest_1.expect)(cm.isStale).toBe(true);
+        cm.setConfig(config);
+        (0, vitest_1.expect)(cm.isStale).toBe(false);
+        vitest_1.vi.setSystemTime(300000);
+        (0, vitest_1.expect)(cm.isStale).toBe(true);
+    });
+    (0, vitest_1.it)('should not update config if not stale', async () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        await cm.update();
+        getConfig.mockClear();
+        await cm.update();
+        (0, vitest_1.expect)(getConfig).not.toHaveBeenCalled();
+    });
+    (0, vitest_1.it)('should not update config twice', async () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        await cm.update();
+        vitest_1.vi.setSystemTime(300000);
+        getConfig.mockClear();
+        await Promise.all([cm.update(), cm.update()]);
+        (0, vitest_1.expect)(getConfig).toHaveBeenCalledOnce();
+    });
+    (0, vitest_1.it)('should call listeners on config update', async () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        const listener = vitest_1.vi.fn();
+        cm.onConfigUpdate(listener);
+        await cm.update();
+        vitest_1.vi.setSystemTime(300000);
+        cm.offConfigUpdate(listener);
+        await cm.update();
+        (0, vitest_1.expect)(listener).toHaveBeenCalledOnce();
+        (0, vitest_1.expect)(listener).toHaveBeenCalledWith(config);
+    });
+    (0, vitest_1.it)('should correctly destroy', async () => {
+        const cm = new config_manager_js_1.ConfigManager(getConfig);
+        await cm.update();
+        cm.destroy();
+        getConfig.mockClear();
+        await vitest_1.vi.advanceTimersByTimeAsync(301000);
+        (0, vitest_1.expect)(getConfig).not.toHaveBeenCalled();
+    });
+    (0, vitest_1.describe)('findOption', () => {
+        const useDcOptions = (options) => {
+            getConfig.mockImplementation(() => Promise.resolve({
+                ...config,
+                dcOptions: options,
+            }));
+        };
+        const findOption = async (params) => {
+            const cm = new config_manager_js_1.ConfigManager(getConfig);
+            await cm.update();
+            return cm.findOption(params);
+        };
+        (0, vitest_1.it)('should find option by dc id', async () => {
+            useDcOptions([
+                (0, test_1.createStub)('dcOption', { id: 1, ipAddress: '1.1.1.1' }),
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.2' }),
+            ]);
+            (0, vitest_1.expect)(await findOption({ dcId: 1 })).toMatchObject({
+                id: 1,
+                ipAddress: '1.1.1.1',
+            });
+            (0, vitest_1.expect)(await findOption({ dcId: 2 })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.2',
+            });
+        });
+        (0, vitest_1.it)('should ignore tcpoOnly options', async () => {
+            useDcOptions([
+                (0, test_1.createStub)('dcOption', { id: 1, ipAddress: '1.1.1.1', tcpoOnly: true }),
+                (0, test_1.createStub)('dcOption', { id: 1, ipAddress: '1.1.1.2' }),
+            ]);
+            (0, vitest_1.expect)(await findOption({ dcId: 1 })).toMatchObject({
+                id: 1,
+                ipAddress: '1.1.1.2',
+            });
+        });
+        (0, vitest_1.it)('should respect allowMedia flag', async () => {
+            useDcOptions([
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.2', mediaOnly: true }),
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.3' }),
+            ]);
+            (0, vitest_1.expect)(await findOption({ dcId: 2 })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.3',
+            });
+            (0, vitest_1.expect)(await findOption({ dcId: 2, allowMedia: true })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.2',
+            });
+        });
+        (0, vitest_1.it)('should respect preferMedia flag', async () => {
+            useDcOptions([
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.3' }),
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.2', mediaOnly: true }),
+            ]);
+            (0, vitest_1.expect)(await findOption({ dcId: 2 })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.3',
+            });
+            (0, vitest_1.expect)(await findOption({ dcId: 2, allowMedia: true, preferMedia: true })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.2',
+            });
+        });
+        (0, vitest_1.it)('should respect allowIpv6 flag', async () => {
+            useDcOptions([
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '::1', ipv6: true }),
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.3' }),
+            ]);
+            (0, vitest_1.expect)(await findOption({ dcId: 2 })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.3',
+            });
+            (0, vitest_1.expect)(await findOption({ dcId: 2, allowIpv6: true })).toMatchObject({
+                id: 2,
+                ipAddress: '::1',
+            });
+        });
+        (0, vitest_1.it)('should respect preferIpv6 flag', async () => {
+            useDcOptions([
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '2.2.2.3' }),
+                (0, test_1.createStub)('dcOption', { id: 2, ipAddress: '::1', ipv6: true }),
+            ]);
+            (0, vitest_1.expect)(await findOption({ dcId: 2 })).toMatchObject({
+                id: 2,
+                ipAddress: '2.2.2.3',
+            });
+            (0, vitest_1.expect)(await findOption({ dcId: 2, allowIpv6: true, preferIpv6: true })).toMatchObject({
+                id: 2,
+                ipAddress: '::1',
+            });
+        });
+    });
+});
+//# sourceMappingURL=config-manager.test.js.map

package/cjs/network/config-manager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config-manager.test.js","sourceRoot":"","sources":["../../../src/network/config-manager.test.ts"],"names":[],"mappings":";;AAAA,mCAAwE;AAExE,uCAAyC;AAGzC,2DAAmD;AAEnD,IAAA,iBAAQ,EAAC,eAAe,EAAE,GAAG,EAAE;IAC3B,MAAM,MAAM,GAAG,IAAA,iBAAU,EAAC,QAAQ,EAAE;QAChC,OAAO,EAAE,GAAG;KACf,CAAC,CAAA;IACF,MAAM,SAAS,GAAG,WAAE,CAAC,EAAE,EAAE,CAAA;IAEzB,IAAA,mBAAU,EAAC,GAAG,EAAE;QACZ,WAAE,CAAC,aAAa,EAAE,CAAA;QAClB,WAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;QACnB,SAAS,CAAC,SAAS,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IACF,IAAA,kBAAS,EAAC,GAAG,EAAE,CAAC,KAAK,WAAE,CAAC,aAAa,EAAE,CAAC,CAAA;IAExC,IAAA,WAAE,EAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QAEvC,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,GAAG,EAAE,CAAA;QAEpC,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC1C,IAAA,eAAM,EAAC,aAAa,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QACrC,IAAA,eAAM,EAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACvC,CAAC,CAAC,CAAA;IAEF,IAAA,WAAE,EAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QACvC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,SAAS,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAC9B,OAAO,CAAC,OAAO,CAAC;YACZ,GAAG,MAAM;YACT,OAAO,EAAE,GAAG;SACf,CAAC,CACL,CAAA;QAED,MAAM,WAAE,CAAC,wBAAwB,CAAC,MAAO,CAAC,CAAA;QAE1C,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,IAAA,WAAE,EAAC,yCAAyC,EAAE,GAAG,EAAE;QAC/C,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QACvC,IAAA,eAAM,EAAC,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE7B,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QACpB,IAAA,eAAM,EAAC,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAE9B,WAAE,CAAC,aAAa,CAAC,MAAO,CAAC,CAAA;QACzB,IAAA,eAAM,EAAC,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,IAAA,WAAE,EAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QACvC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,SAAS,CAAC,SAAS,EAAE,CAAA;QACrB,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,IAAA,WAAE,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QACvC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,WAAE,CAAC,aAAa,CAAC,MAAO,CAAC,CAAA;QACzB,SAAS,CAAC,SAAS,EAAE,CAAA;QACrB,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAE7C,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,oBAAoB,EAAE,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,IAAA,WAAE,EAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QACvC,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAA;QACxB,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAA;QAC3B,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,WAAE,CAAC,aAAa,CAAC,MAAO,CAAC,CAAA;QACzB,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAA;QAC5B,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,CAAA;QACvC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,IAAA,WAAE,EAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;QACvC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;QAEjB,EAAE,CAAC,OAAO,EAAE,CAAA;QAEZ,SAAS,CAAC,SAAS,EAAE,CAAA;QACrB,MAAM,WAAE,CAAC,wBAAwB,CAAC,MAAO,CAAC,CAAA;QAE1C,IAAA,eAAM,EAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,IAAA,iBAAQ,EAAC,YAAY,EAAE,GAAG,EAAE;QACxB,MAAM,YAAY,GAAG,CAAC,OAAyB,EAAE,EAAE;YAC/C,SAAS,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAC9B,OAAO,CAAC,OAAO,CAAC;gBACZ,GAAG,MAAM;gBACT,SAAS,EAAE,OAAO;aACrB,CAAC,CACL,CAAA;QACL,CAAC,CAAA;QAED,MAAM,UAAU,GAAG,KAAK,EAAE,MAAkD,EAAE,EAAE;YAC5E,MAAM,EAAE,GAAG,IAAI,iCAAa,CAAC,SAAS,CAAC,CAAA;YACvC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAA;YAEjB,OAAO,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;QAChC,CAAC,CAAA;QAED,IAAA,WAAE,EAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YACzC,YAAY,CAAC;gBACT,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;gBACvD,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;aAC1D,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;YACF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC5C,YAAY,CAAC;gBACT,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;gBACvE,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;aAC1D,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC5C,YAAY,CAAC;gBACT,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;gBACxE,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;aAC1D,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAClE,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC7C,YAAY,CAAC;gBACT,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;gBACvD,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;aAC3E,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBACrF,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC3C,YAAY,CAAC;gBACT,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAC/D,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;aAC1D,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBACjE,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,KAAK;aACnB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,IAAA,WAAE,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC5C,YAAY,CAAC;gBACT,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;gBACvD,IAAA,iBAAU,EAAC,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aAClE,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBAChD,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,SAAS;aACvB,CAAC,CAAA;YAEF,IAAA,eAAM,EAAC,MAAM,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;gBACnF,EAAE,EAAE,CAAC;gBACL,SAAS,EAAE,KAAK;aACnB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA","sourcesContent":["import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'\n\nimport { createStub } from '@mtcute/test'\nimport { tl } from '@mtcute/tl'\n\nimport { ConfigManager } from './config-manager.js'\n\ndescribe('ConfigManager', () => {\n    const config = createStub('config', {\n        expires: 300,\n    })\n    const getConfig = vi.fn()\n\n    beforeEach(() => {\n        vi.useFakeTimers()\n        vi.setSystemTime(0)\n        getConfig.mockClear().mockImplementation(() => Promise.resolve(config))\n    })\n    afterEach(() => void vi.useRealTimers())\n\n    it('should fetch initial config', async () => {\n        const cm = new ConfigManager(getConfig)\n\n        const fetchedConfig = await cm.get()\n\n        expect(getConfig).toHaveBeenCalledTimes(1)\n        expect(fetchedConfig).toEqual(config)\n        expect(cm.getNow()).toEqual(config)\n    })\n\n    it('should automatically update config', async () => {\n        const cm = new ConfigManager(getConfig)\n        await cm.update()\n\n        getConfig.mockImplementation(() =>\n            Promise.resolve({\n                ...config,\n                expires: 600,\n            }),\n        )\n\n        await vi.advanceTimersByTimeAsync(301_000)\n\n        expect(getConfig).toHaveBeenCalledTimes(2)\n    })\n\n    it('should correctly determine stale config', () => {\n        const cm = new ConfigManager(getConfig)\n        expect(cm.isStale).toBe(true)\n\n        cm.setConfig(config)\n        expect(cm.isStale).toBe(false)\n\n        vi.setSystemTime(300_000)\n        expect(cm.isStale).toBe(true)\n    })\n\n    it('should not update config if not stale', async () => {\n        const cm = new ConfigManager(getConfig)\n        await cm.update()\n\n        getConfig.mockClear()\n        await cm.update()\n\n        expect(getConfig).not.toHaveBeenCalled()\n    })\n\n    it('should not update config twice', async () => {\n        const cm = new ConfigManager(getConfig)\n        await cm.update()\n\n        vi.setSystemTime(300_000)\n        getConfig.mockClear()\n        await Promise.all([cm.update(), cm.update()])\n\n        expect(getConfig).toHaveBeenCalledOnce()\n    })\n\n    it('should call listeners on config update', async () => {\n        const cm = new ConfigManager(getConfig)\n        const listener = vi.fn()\n        cm.onConfigUpdate(listener)\n        await cm.update()\n\n        vi.setSystemTime(300_000)\n        cm.offConfigUpdate(listener)\n        await cm.update()\n\n        expect(listener).toHaveBeenCalledOnce()\n        expect(listener).toHaveBeenCalledWith(config)\n    })\n\n    it('should correctly destroy', async () => {\n        const cm = new ConfigManager(getConfig)\n        await cm.update()\n\n        cm.destroy()\n\n        getConfig.mockClear()\n        await vi.advanceTimersByTimeAsync(301_000)\n\n        expect(getConfig).not.toHaveBeenCalled()\n    })\n\n    describe('findOption', () => {\n        const useDcOptions = (options: tl.RawDcOption[]) => {\n            getConfig.mockImplementation(() =>\n                Promise.resolve({\n                    ...config,\n                    dcOptions: options,\n                }),\n            )\n        }\n\n        const findOption = async (params: Parameters<ConfigManager['findOption']>[0]) => {\n            const cm = new ConfigManager(getConfig)\n            await cm.update()\n\n            return cm.findOption(params)\n        }\n\n        it('should find option by dc id', async () => {\n            useDcOptions([\n                createStub('dcOption', { id: 1, ipAddress: '1.1.1.1' }),\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.2' }),\n            ])\n\n            expect(await findOption({ dcId: 1 })).toMatchObject({\n                id: 1,\n                ipAddress: '1.1.1.1',\n            })\n            expect(await findOption({ dcId: 2 })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.2',\n            })\n        })\n\n        it('should ignore tcpoOnly options', async () => {\n            useDcOptions([\n                createStub('dcOption', { id: 1, ipAddress: '1.1.1.1', tcpoOnly: true }),\n                createStub('dcOption', { id: 1, ipAddress: '1.1.1.2' }),\n            ])\n\n            expect(await findOption({ dcId: 1 })).toMatchObject({\n                id: 1,\n                ipAddress: '1.1.1.2',\n            })\n        })\n\n        it('should respect allowMedia flag', async () => {\n            useDcOptions([\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.2', mediaOnly: true }),\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.3' }),\n            ])\n\n            expect(await findOption({ dcId: 2 })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.3',\n            })\n\n            expect(await findOption({ dcId: 2, allowMedia: true })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.2',\n            })\n        })\n\n        it('should respect preferMedia flag', async () => {\n            useDcOptions([\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.3' }),\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.2', mediaOnly: true }),\n            ])\n\n            expect(await findOption({ dcId: 2 })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.3',\n            })\n\n            expect(await findOption({ dcId: 2, allowMedia: true, preferMedia: true })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.2',\n            })\n        })\n\n        it('should respect allowIpv6 flag', async () => {\n            useDcOptions([\n                createStub('dcOption', { id: 2, ipAddress: '::1', ipv6: true }),\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.3' }),\n            ])\n\n            expect(await findOption({ dcId: 2 })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.3',\n            })\n\n            expect(await findOption({ dcId: 2, allowIpv6: true })).toMatchObject({\n                id: 2,\n                ipAddress: '::1',\n            })\n        })\n\n        it('should respect preferIpv6 flag', async () => {\n            useDcOptions([\n                createStub('dcOption', { id: 2, ipAddress: '2.2.2.3' }),\n                createStub('dcOption', { id: 2, ipAddress: '::1', ipv6: true }),\n            ])\n\n            expect(await findOption({ dcId: 2 })).toMatchObject({\n                id: 2,\n                ipAddress: '2.2.2.3',\n            })\n\n            expect(await findOption({ dcId: 2, allowIpv6: true, preferIpv6: true })).toMatchObject({\n                id: 2,\n                ipAddress: '::1',\n            })\n        })\n    })\n})\n"]}

package/cjs/network/mtproto-session.d.ts

@@ -107,9 +107,9 @@ export declare class MtprotoSession {
     getMessageId(): Long;
     getSeqNo(isContentRelated?: boolean): number;
     /** Encrypt a single MTProto message using session's keys */
-    encryptMessage(message: Uint8Array): Promise<Uint8Array>;
+    encryptMessage(message: Uint8Array): Uint8Array;
     /** Decrypt a single MTProto message using session's keys */
-    decryptMessage(data: Uint8Array, callback: Parameters<AuthKey['decryptMessage']>[2]): Promise<void>;
+    decryptMessage(data: Uint8Array, callback: Parameters<AuthKey['decryptMessage']>[2]): void;
     writeMessage(writer: TlBinaryWriter, content: tl.TlObject | mtp.TlObject | Uint8Array, isContentRelated?: boolean): Long;
     onTransportFlood(callback: () => void): number | undefined;
     resetLastPing(withTime?: boolean): void;

package/cjs/network/mtproto-session.js

@@ -136,12 +136,12 @@ class MtprotoSession {
         return seqNo;
     }
     /** Encrypt a single MTProto message using session's keys */
-    async encryptMessage(message) {
+    encryptMessage(message) {
         const key = this._authKeyTemp.ready ? this._authKeyTemp : this._authKey;
         return key.encryptMessage(message, this.serverSalt, this._sessionId);
     }
     /** Decrypt a single MTProto message using session's keys */
-    async decryptMessage(data, callback) {
+    decryptMessage(data, callback) {
         if (!this._authKey.ready)
             throw new index_js_1.MtcuteError('Keys are not set up!');
         const authKeyId = data.subarray(0, 8);

package/cjs/network/mtproto-session.js.map

@@ -1 +1 @@
-{"version":3,"file":"mtproto-session.js","sourceRoot":"","sources":["../../../src/network/mtproto-session.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAuB;AAGvB,mDAAqG;AAErG,gDAA+C;AAC/C,gDAU0B;AAC1B,+CAAuC;AAgEvC;;;GAGG;AACH,MAAa,cAAc;IA0CvB,YACa,OAAwB,EACxB,GAAW,EACX,UAAuB,EACvB,UAAuB;QAHvB,YAAO,GAAP,OAAO,CAAiB;QACxB,QAAG,GAAH,GAAG,CAAQ;QACX,eAAU,GAAV,UAAU,CAAa;QACvB,eAAU,GAAV,UAAU,CAAa;QA7CpC,eAAU,GAAG,IAAA,qBAAU,GAAE,CAAA;QAEzB,aAAQ,GAAG,IAAI,qBAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAC/D,iBAAY,GAAG,IAAI,qBAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,0BAAqB,GAAG,IAAI,qBAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAE5E,gBAAW,GAAG,CAAC,CAAA;QACf,mBAAc,GAAG,cAAI,CAAC,IAAI,CAAA;QAC1B,WAAM,GAAG,CAAC,CAAA;QAEV,eAAU,GAAG,cAAI,CAAC,IAAI,CAAA;QAEtB,aAAa;QACb,iBAAiB;QACjB,yBAAoB,GAAG,IAAI,iBAAM,CAAO,IAAI,EAAE,IAAI,CAAC,CAAA;QACnD,yBAAoB,GAAG,IAAI,iBAAM,CAAO,IAAI,EAAE,IAAI,CAAC,CAAA;QAEnD,SAAS;QACT,cAAS,GAAG,IAAI,gBAAK,EAAc,CAAA;QACnC,eAAU,GAAW,EAAE,CAAA;QACvB,mBAAc,GAAW,EAAE,CAAA;QAC3B,oBAAe,GAAW,EAAE,CAAA;QAC5B,oBAAe,GAAW,EAAE,CAAA;QAC5B,qBAAgB,GAAG,IAAI,sBAAW,CAAa,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAS,GAAG,CAAC,CAAC,QAAS,CAAC,CAAA;QAEvF,gBAAgB;QAChB,oBAAe,GAAG,IAAI,kBAAO,EAAkB,CAAA;QAC/C,4BAAuB,GAAG,IAAI,kBAAO,EAAQ,CAAA;QAE7C,gBAAW,GAAG,GAAG,CAAA;QACjB,iBAAY,GAAG,CAAC,CAAA;QAChB,kBAAa,GAAG,cAAI,CAAC,IAAI,CAAA;QACzB,0BAAqB,GAAG,cAAI,CAAC,IAAI,CAAA;QAEjC,yBAAoB,GAAG,KAAK,CAAA;QAC5B,yBAAoB,GAAG,KAAK,CAAA;QAE5B,mBAAc,GAAG,IAAI,CAAA;QAUjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,YAAY,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAA;IAClE,CAAC;IAED,IAAI,kBAAkB;QAClB,OAAO,OAAO,CACV,IAAI,CAAC,SAAS,CAAC,MAAM;YACjB,IAAI,CAAC,UAAU,CAAC,MAAM;YACtB,IAAI,CAAC,cAAc,CAAC,MAAM;YAC1B,IAAI,CAAC,eAAe,CAAC,MAAM,CAClC,CAAA;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,GAAG,KAAK;QACrB,IAAI,WAAW,EAAE;YACb,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;YACrB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;YACzB,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAA;SACrC;QAED,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QACpC,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;IAC5B,CAAC;IAED;;;;;OAKG;IACH,UAAU,CAAC,WAAW,GAAG,KAAK;QAC1B,IAAI,CAAC,cAAc,GAAG,cAAI,CAAC,IAAI,CAAA;QAC/B,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA;QAEf,IAAI,CAAC,UAAU,GAAG,IAAA,qBAAU,GAAE,CAAA;QAC9B,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,6BAA6B,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAC9D,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,YAAY,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAA;QAE9D,sBAAsB;QAEtB,IAAI,CAAC,WAAW,EAAE;YACd,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE;gBAC9C,IAAI,IAAI,CAAC,CAAC,KAAK,KAAK,EAAE;oBAClB,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,sBAAW,CAAC,kBAAkB,CAAC,CAAC,CAAA;iBAC/D;aACJ;YACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAA;SAC/B;QAED,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAA;QACjC,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAA;QAEjC,IAAI,CAAC,WAAW,EAAE;YACd,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE;gBAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAG,CAAA;gBAEtC,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE;oBACpB,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,sBAAW,CAAC,kBAAkB,CAAC,CAAC,CAAA;iBAC1D;aACJ;SACJ;QAED,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAA;QAC1B,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAA;QAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAA;QAC/B,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAA;IACjC,CAAC;IAED,UAAU,CAAC,GAAe,EAAE,KAAe;QACvC,8BAA8B;QAC9B,IAAI,CAAC,CAAC,KAAK,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,SAAS;YAAE,OAAO,KAAK,CAAA;QAExD,GAAG,CAAC,IAAI,GAAG,KAAK,CAAA;QAChB,GAAG,CAAC,WAAW,GAAG,SAAS,CAAA;QAC3B,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,CAAA;QACvF,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;QAE5B,OAAO,IAAI,CAAA;IACf,CAAC;IAED,YAAY;QACR,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,WAAW,CAAA;QAC/D,MAAM,QAAQ,GAAG,SAAS,GAAG,IAAI,CAAA;QACjC,MAAM,MAAM,GAAG,IAAA,uBAAY,EAAC,MAAM,CAAC,CAAA;QAEnC,IAAI,SAAS,GAAG,IAAI,cAAI,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAA;QAEvE,IAAI,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE;YACnC,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;SACzC;QAED,IAAI,CAAC,cAAc,GAAG,SAAS,CAAA;QAE/B,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,QAAQ,CAAC,gBAAgB,GAAG,IAAI;QAC5B,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,CAAA;QAEvB,IAAI,gBAAgB,EAAE;YAClB,KAAK,IAAI,CAAC,CAAA;YACV,IAAI,CAAC,MAAM,IAAI,CAAC,CAAA;SACnB;QAED,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,4DAA4D;IAC5D,KAAK,CAAC,cAAc,CAAC,OAAmB;QACpC,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAA;QAEvE,OAAO,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;IACxE,CAAC;IAED,4DAA4D;IAC5D,KAAK,CAAC,cAAc,CAAC,IAAgB,EAAE,QAAkD;QACrF,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK;YAAE,MAAM,IAAI,sBAAW,CAAC,sBAAsB,CAAC,CAAA;QAEvE,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;QAErC,IAAI,GAAY,CAAA;QAEhB,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;YAChC,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAA;SACtB;aAAM,IAAI,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;YAC3C,GAAG,GAAG,IAAI,CAAC,YAAY,CAAA;SAC1B;aAAM,IAAI,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;YACpD,GAAG,GAAG,IAAI,CAAC,qBAAqB,CAAA;SACnC;aAAM;YACH,IAAI,CAAC,GAAG,CAAC,IAAI,CACT,sEAAsE,EACtE,SAAS,EACT,IAAI,CAAC,QAAQ,CAAC,EAAE,EAChB,IAAI,CAAC,YAAY,CAAC,EAAE,EACpB,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAChC,CAAA;YAED,OAAM;SACT;QAED,OAAO,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;IAC9D,CAAC;IAED,YAAY,CACR,MAAsB,EACtB,OAAgD,EAChD,gBAAgB,GAAG,IAAI;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAA;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAA;QAE7C,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YACxC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChB,mCAAsB,CAAC,gBAAgB,CAAC,MAAM,CAAC,SAAU,EAAE,OAAO,CAAC,CAAA;QAEvE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QACtB,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACjB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACnB,IAAI,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;;YAC/C,MAAM,CAAC,MAAM,CAAC,OAAsB,CAAC,CAAA;QAE1C,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,gBAAgB,CAAC,QAAoB;QACjC,IAAI,IAAI,CAAC,iBAAiB;YAAE,OAAM,CAAC,kBAAkB;QAErD,oDAAoD;QACpD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;QAExB,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAA;QAEnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,EAAE,KAAK,CAAC,CAAA;QAC9D,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QACpC,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAEtC,IAAI,CAAC,iBAAiB,GAAG,UAAU,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAA;YAClC,QAAQ,EAAE,CAAA;QACd,CAAC,EAAE,OAAO,CAAC,CAAA;QACX,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC,GAAG,EAAE;YACvC,IAAI,CAAC,mBAAmB,GAAG,SAAS,CAAA;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAA;QAC9B,CAAC,EAAE,KAAK,CAAC,CAAA;QAET,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,sDAAsD,EAAE,OAAO,CAAC,CAAA;QAE/E,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAA;IAC/B,CAAC;IAED,aAAa,CAAC,QAAQ,GAAG,KAAK;QAC1B,IAAI,QAAQ;YAAE,IAAI,CAAC,YAAY,GAAG,CAAC,CAAA;QAEnC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;SAClD;QAED,IAAI,CAAC,aAAa,GAAG,cAAI,CAAC,IAAI,CAAA;IAClC,CAAC;CACJ;AA3PD,wCA2PC","sourcesContent":["import Long from 'long'\n\nimport { mtp, tl } from '@mtcute/tl'\nimport { TlBinaryWriter, TlReaderMap, TlSerializationCounter, TlWriterMap } from '@mtcute/tl-runtime'\n\nimport { MtcuteError } from '../types/index.js'\nimport {\n    ControllablePromise,\n    Deque,\n    getRandomInt,\n    ICryptoProvider,\n    Logger,\n    LongMap,\n    LruSet,\n    randomLong,\n    SortedArray,\n} from '../utils/index.js'\nimport { AuthKey } from './auth-key.js'\n\nexport interface PendingRpc {\n    method: string\n    data: Uint8Array\n    promise: ControllablePromise\n    stack?: string\n    gzipOverhead?: number\n\n    sent?: boolean\n    done?: boolean\n    msgId?: Long\n    seqNo?: number\n    containerId?: Long\n    acked?: boolean\n    initConn?: boolean\n    getState?: number\n    cancelled?: boolean\n    timeout?: NodeJS.Timeout\n}\n\nexport type PendingMessage =\n    | {\n          _: 'rpc'\n          rpc: PendingRpc\n      }\n    | {\n          _: 'container'\n          msgIds: Long[]\n      }\n    | {\n          _: 'state'\n          msgIds: Long[]\n          containerId: Long\n      }\n    | {\n          _: 'resend'\n          msgIds: Long[]\n          containerId: Long\n      }\n    | {\n          _: 'ping'\n          pingId: Long\n          containerId: Long\n      }\n    | {\n          _: 'destroy_session'\n          sessionId: Long\n          containerId: Long\n      }\n    | {\n          _: 'cancel'\n          msgId: Long\n          containerId: Long\n      }\n    | {\n          _: 'future_salts'\n          containerId: Long\n      }\n    | {\n          _: 'bind'\n          promise: ControllablePromise\n      }\n\n/**\n * Class encapsulating a single MTProto session and storing\n * all the relevant state\n */\nexport class MtprotoSession {\n    _sessionId = randomLong()\n\n    _authKey = new AuthKey(this._crypto, this.log, this._readerMap)\n    _authKeyTemp = new AuthKey(this._crypto, this.log, this._readerMap)\n    _authKeyTempSecondary = new AuthKey(this._crypto, this.log, this._readerMap)\n\n    _timeOffset = 0\n    _lastMessageId = Long.ZERO\n    _seqNo = 0\n\n    serverSalt = Long.ZERO\n\n    /// state ///\n    // recent msg ids\n    recentOutgoingMsgIds = new LruSet<Long>(1000, true)\n    recentIncomingMsgIds = new LruSet<Long>(1000, true)\n\n    // queues\n    queuedRpc = new Deque<PendingRpc>()\n    queuedAcks: Long[] = []\n    queuedStateReq: Long[] = []\n    queuedResendReq: Long[] = []\n    queuedCancelReq: Long[] = []\n    getStateSchedule = new SortedArray<PendingRpc>([], (a, b) => a.getState! - b.getState!)\n\n    // requests info\n    pendingMessages = new LongMap<PendingMessage>()\n    destroySessionIdToMsgId = new LongMap<Long>()\n\n    lastPingRtt = NaN\n    lastPingTime = 0\n    lastPingMsgId = Long.ZERO\n    lastSessionCreatedUid = Long.ZERO\n\n    initConnectionCalled = false\n    authorizationPending = false\n\n    next429Timeout = 1000\n    current429Timeout?: NodeJS.Timeout\n    next429ResetTimeout?: NodeJS.Timeout\n\n    constructor(\n        readonly _crypto: ICryptoProvider,\n        readonly log: Logger,\n        readonly _readerMap: TlReaderMap,\n        readonly _writerMap: TlWriterMap,\n    ) {\n        this.log.prefix = `[SESSION ${this._sessionId.toString(16)}] `\n    }\n\n    get hasPendingMessages(): boolean {\n        return Boolean(\n            this.queuedRpc.length ||\n                this.queuedAcks.length ||\n                this.queuedStateReq.length ||\n                this.queuedResendReq.length,\n        )\n    }\n\n    /**\n     * Reset session by resetting auth key(s) and session state\n     */\n    reset(withAuthKey = false): void {\n        if (withAuthKey) {\n            this._authKey.reset()\n            this._authKeyTemp.reset()\n            this._authKeyTempSecondary.reset()\n        }\n\n        clearTimeout(this.current429Timeout)\n        this.resetState()\n        this.resetLastPing(true)\n    }\n\n    /**\n     * Reset session state and generate a new session ID.\n     *\n     * By default, also cancels any pending RPC requests.\n     * If `keepPending` is set to `true`, pending requests will be kept\n     */\n    resetState(keepPending = false): void {\n        this._lastMessageId = Long.ZERO\n        this._seqNo = 0\n\n        this._sessionId = randomLong()\n        this.log.debug('session reset, new sid = %h', this._sessionId)\n        this.log.prefix = `[SESSION ${this._sessionId.toString(16)}] `\n\n        // reset session state\n\n        if (!keepPending) {\n            for (const info of this.pendingMessages.values()) {\n                if (info._ === 'rpc') {\n                    info.rpc.promise.reject(new MtcuteError('Session is reset'))\n                }\n            }\n            this.pendingMessages.clear()\n        }\n\n        this.recentOutgoingMsgIds.clear()\n        this.recentIncomingMsgIds.clear()\n\n        if (!keepPending) {\n            while (this.queuedRpc.length) {\n                const rpc = this.queuedRpc.popFront()!\n\n                if (rpc.sent === false) {\n                    rpc.promise.reject(new MtcuteError('Session is reset'))\n                }\n            }\n        }\n\n        this.queuedAcks.length = 0\n        this.queuedStateReq.length = 0\n        this.queuedResendReq.length = 0\n        this.getStateSchedule.clear()\n    }\n\n    enqueueRpc(rpc: PendingRpc, force?: boolean): boolean {\n        // already queued or cancelled\n        if ((!force && !rpc.sent) || rpc.cancelled) return false\n\n        rpc.sent = false\n        rpc.containerId = undefined\n        this.log.debug('enqueued %s for sending (msg_id = %s)', rpc.method, rpc.msgId || 'n/a')\n        this.queuedRpc.pushBack(rpc)\n\n        return true\n    }\n\n    getMessageId(): Long {\n        const timeTicks = Date.now()\n        const timeSec = Math.floor(timeTicks / 1000) + this._timeOffset\n        const timeMSec = timeTicks % 1000\n        const random = getRandomInt(0xffff)\n\n        let messageId = new Long((timeMSec << 21) | (random << 3) | 4, timeSec)\n\n        if (this._lastMessageId.gt(messageId)) {\n            messageId = this._lastMessageId.add(4)\n        }\n\n        this._lastMessageId = messageId\n\n        return messageId\n    }\n\n    getSeqNo(isContentRelated = true): number {\n        let seqNo = this._seqNo\n\n        if (isContentRelated) {\n            seqNo += 1\n            this._seqNo += 2\n        }\n\n        return seqNo\n    }\n\n    /** Encrypt a single MTProto message using session's keys */\n    async encryptMessage(message: Uint8Array): Promise<Uint8Array> {\n        const key = this._authKeyTemp.ready ? this._authKeyTemp : this._authKey\n\n        return key.encryptMessage(message, this.serverSalt, this._sessionId)\n    }\n\n    /** Decrypt a single MTProto message using session's keys */\n    async decryptMessage(data: Uint8Array, callback: Parameters<AuthKey['decryptMessage']>[2]): Promise<void> {\n        if (!this._authKey.ready) throw new MtcuteError('Keys are not set up!')\n\n        const authKeyId = data.subarray(0, 8)\n\n        let key: AuthKey\n\n        if (this._authKey.match(authKeyId)) {\n            key = this._authKey\n        } else if (this._authKeyTemp.match(authKeyId)) {\n            key = this._authKeyTemp\n        } else if (this._authKeyTempSecondary.match(authKeyId)) {\n            key = this._authKeyTempSecondary\n        } else {\n            this.log.warn(\n                'received message with unknown authKey = %h (expected %h or %h or %h)',\n                authKeyId,\n                this._authKey.id,\n                this._authKeyTemp.id,\n                this._authKeyTempSecondary.id,\n            )\n\n            return\n        }\n\n        return key.decryptMessage(data, this._sessionId, callback)\n    }\n\n    writeMessage(\n        writer: TlBinaryWriter,\n        content: tl.TlObject | mtp.TlObject | Uint8Array,\n        isContentRelated = true,\n    ): Long {\n        const messageId = this.getMessageId()\n        const seqNo = this.getSeqNo(isContentRelated)\n\n        const length = ArrayBuffer.isView(content) ?\n            content.length :\n            TlSerializationCounter.countNeededBytes(writer.objectMap!, content)\n\n        writer.long(messageId)\n        writer.int(seqNo)\n        writer.uint(length)\n        if (ArrayBuffer.isView(content)) writer.raw(content)\n        else writer.object(content as tl.TlObject)\n\n        return messageId\n    }\n\n    onTransportFlood(callback: () => void) {\n        if (this.current429Timeout) return // already waiting\n\n        // all active queries must be resent after a timeout\n        this.resetLastPing(true)\n\n        const timeout = this.next429Timeout\n\n        this.next429Timeout = Math.min(this.next429Timeout * 2, 32000)\n        clearTimeout(this.current429Timeout)\n        clearTimeout(this.next429ResetTimeout)\n\n        this.current429Timeout = setTimeout(() => {\n            this.current429Timeout = undefined\n            callback()\n        }, timeout)\n        this.next429ResetTimeout = setTimeout(() => {\n            this.next429ResetTimeout = undefined\n            this.next429Timeout = 1000\n        }, 60000)\n\n        this.log.debug('transport flood, waiting for %d ms before proceeding', timeout)\n\n        return Date.now() + timeout\n    }\n\n    resetLastPing(withTime = false): void {\n        if (withTime) this.lastPingTime = 0\n\n        if (!this.lastPingMsgId.isZero()) {\n            this.pendingMessages.delete(this.lastPingMsgId)\n        }\n\n        this.lastPingMsgId = Long.ZERO\n    }\n}\n"]}
+{"version":3,"file":"mtproto-session.js","sourceRoot":"","sources":["../../../src/network/mtproto-session.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAuB;AAGvB,mDAAqG;AAErG,gDAA+C;AAC/C,gDAU0B;AAC1B,+CAAuC;AAgEvC;;;GAGG;AACH,MAAa,cAAc;IA0CvB,YACa,OAAwB,EACxB,GAAW,EACX,UAAuB,EACvB,UAAuB;QAHvB,YAAO,GAAP,OAAO,CAAiB;QACxB,QAAG,GAAH,GAAG,CAAQ;QACX,eAAU,GAAV,UAAU,CAAa;QACvB,eAAU,GAAV,UAAU,CAAa;QA7CpC,eAAU,GAAG,IAAA,qBAAU,GAAE,CAAA;QAEzB,aAAQ,GAAG,IAAI,qBAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAC/D,iBAAY,GAAG,IAAI,qBAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,0BAAqB,GAAG,IAAI,qBAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAE5E,gBAAW,GAAG,CAAC,CAAA;QACf,mBAAc,GAAG,cAAI,CAAC,IAAI,CAAA;QAC1B,WAAM,GAAG,CAAC,CAAA;QAEV,eAAU,GAAG,cAAI,CAAC,IAAI,CAAA;QAEtB,aAAa;QACb,iBAAiB;QACjB,yBAAoB,GAAG,IAAI,iBAAM,CAAO,IAAI,EAAE,IAAI,CAAC,CAAA;QACnD,yBAAoB,GAAG,IAAI,iBAAM,CAAO,IAAI,EAAE,IAAI,CAAC,CAAA;QAEnD,SAAS;QACT,cAAS,GAAG,IAAI,gBAAK,EAAc,CAAA;QACnC,eAAU,GAAW,EAAE,CAAA;QACvB,mBAAc,GAAW,EAAE,CAAA;QAC3B,oBAAe,GAAW,EAAE,CAAA;QAC5B,oBAAe,GAAW,EAAE,CAAA;QAC5B,qBAAgB,GAAG,IAAI,sBAAW,CAAa,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAS,GAAG,CAAC,CAAC,QAAS,CAAC,CAAA;QAEvF,gBAAgB;QAChB,oBAAe,GAAG,IAAI,kBAAO,EAAkB,CAAA;QAC/C,4BAAuB,GAAG,IAAI,kBAAO,EAAQ,CAAA;QAE7C,gBAAW,GAAG,GAAG,CAAA;QACjB,iBAAY,GAAG,CAAC,CAAA;QAChB,kBAAa,GAAG,cAAI,CAAC,IAAI,CAAA;QACzB,0BAAqB,GAAG,cAAI,CAAC,IAAI,CAAA;QAEjC,yBAAoB,GAAG,KAAK,CAAA;QAC5B,yBAAoB,GAAG,KAAK,CAAA;QAE5B,mBAAc,GAAG,IAAI,CAAA;QAUjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,YAAY,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAA;IAClE,CAAC;IAED,IAAI,kBAAkB;QAClB,OAAO,OAAO,CACV,IAAI,CAAC,SAAS,CAAC,MAAM;YACjB,IAAI,CAAC,UAAU,CAAC,MAAM;YACtB,IAAI,CAAC,cAAc,CAAC,MAAM;YAC1B,IAAI,CAAC,eAAe,CAAC,MAAM,CAClC,CAAA;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,GAAG,KAAK;QACrB,IAAI,WAAW,EAAE;YACb,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;YACrB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;YACzB,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAA;SACrC;QAED,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QACpC,IAAI,CAAC,UAAU,EAAE,CAAA;QACjB,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;IAC5B,CAAC;IAED;;;;;OAKG;IACH,UAAU,CAAC,WAAW,GAAG,KAAK;QAC1B,IAAI,CAAC,cAAc,GAAG,cAAI,CAAC,IAAI,CAAA;QAC/B,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA;QAEf,IAAI,CAAC,UAAU,GAAG,IAAA,qBAAU,GAAE,CAAA;QAC9B,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,6BAA6B,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAC9D,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,YAAY,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAA;QAE9D,sBAAsB;QAEtB,IAAI,CAAC,WAAW,EAAE;YACd,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE;gBAC9C,IAAI,IAAI,CAAC,CAAC,KAAK,KAAK,EAAE;oBAClB,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,sBAAW,CAAC,kBAAkB,CAAC,CAAC,CAAA;iBAC/D;aACJ;YACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAA;SAC/B;QAED,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAA;QACjC,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAA;QAEjC,IAAI,CAAC,WAAW,EAAE;YACd,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE;gBAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAG,CAAA;gBAEtC,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE;oBACpB,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,sBAAW,CAAC,kBAAkB,CAAC,CAAC,CAAA;iBAC1D;aACJ;SACJ;QAED,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAA;QAC1B,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAA;QAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAA;QAC/B,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAA;IACjC,CAAC;IAED,UAAU,CAAC,GAAe,EAAE,KAAe;QACvC,8BAA8B;QAC9B,IAAI,CAAC,CAAC,KAAK,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,SAAS;YAAE,OAAO,KAAK,CAAA;QAExD,GAAG,CAAC,IAAI,GAAG,KAAK,CAAA;QAChB,GAAG,CAAC,WAAW,GAAG,SAAS,CAAA;QAC3B,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,CAAA;QACvF,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;QAE5B,OAAO,IAAI,CAAA;IACf,CAAC;IAED,YAAY;QACR,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,WAAW,CAAA;QAC/D,MAAM,QAAQ,GAAG,SAAS,GAAG,IAAI,CAAA;QACjC,MAAM,MAAM,GAAG,IAAA,uBAAY,EAAC,MAAM,CAAC,CAAA;QAEnC,IAAI,SAAS,GAAG,IAAI,cAAI,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAA;QAEvE,IAAI,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE;YACnC,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;SACzC;QAED,IAAI,CAAC,cAAc,GAAG,SAAS,CAAA;QAE/B,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,QAAQ,CAAC,gBAAgB,GAAG,IAAI;QAC5B,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,CAAA;QAEvB,IAAI,gBAAgB,EAAE;YAClB,KAAK,IAAI,CAAC,CAAA;YACV,IAAI,CAAC,MAAM,IAAI,CAAC,CAAA;SACnB;QAED,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,4DAA4D;IAC5D,cAAc,CAAC,OAAmB;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAA;QAEvE,OAAO,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;IACxE,CAAC;IAED,4DAA4D;IAC5D,cAAc,CAAC,IAAgB,EAAE,QAAkD;QAC/E,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK;YAAE,MAAM,IAAI,sBAAW,CAAC,sBAAsB,CAAC,CAAA;QAEvE,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;QAErC,IAAI,GAAY,CAAA;QAEhB,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;YAChC,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAA;SACtB;aAAM,IAAI,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;YAC3C,GAAG,GAAG,IAAI,CAAC,YAAY,CAAA;SAC1B;aAAM,IAAI,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE;YACpD,GAAG,GAAG,IAAI,CAAC,qBAAqB,CAAA;SACnC;aAAM;YACH,IAAI,CAAC,GAAG,CAAC,IAAI,CACT,sEAAsE,EACtE,SAAS,EACT,IAAI,CAAC,QAAQ,CAAC,EAAE,EAChB,IAAI,CAAC,YAAY,CAAC,EAAE,EACpB,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAChC,CAAA;YAED,OAAM;SACT;QAED,OAAO,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;IAC9D,CAAC;IAED,YAAY,CACR,MAAsB,EACtB,OAAgD,EAChD,gBAAgB,GAAG,IAAI;QAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAA;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAA;QAE7C,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YACxC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChB,mCAAsB,CAAC,gBAAgB,CAAC,MAAM,CAAC,SAAU,EAAE,OAAO,CAAC,CAAA;QAEvE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QACtB,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACjB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACnB,IAAI,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;;YAC/C,MAAM,CAAC,MAAM,CAAC,OAAsB,CAAC,CAAA;QAE1C,OAAO,SAAS,CAAA;IACpB,CAAC;IAED,gBAAgB,CAAC,QAAoB;QACjC,IAAI,IAAI,CAAC,iBAAiB;YAAE,OAAM,CAAC,kBAAkB;QAErD,oDAAoD;QACpD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAA;QAExB,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAA;QAEnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,EAAE,KAAK,CAAC,CAAA;QAC9D,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;QACpC,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAEtC,IAAI,CAAC,iBAAiB,GAAG,UAAU,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAA;YAClC,QAAQ,EAAE,CAAA;QACd,CAAC,EAAE,OAAO,CAAC,CAAA;QACX,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC,GAAG,EAAE;YACvC,IAAI,CAAC,mBAAmB,GAAG,SAAS,CAAA;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAA;QAC9B,CAAC,EAAE,KAAK,CAAC,CAAA;QAET,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,sDAAsD,EAAE,OAAO,CAAC,CAAA;QAE/E,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAA;IAC/B,CAAC;IAED,aAAa,CAAC,QAAQ,GAAG,KAAK;QAC1B,IAAI,QAAQ;YAAE,IAAI,CAAC,YAAY,GAAG,CAAC,CAAA;QAEnC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;SAClD;QAED,IAAI,CAAC,aAAa,GAAG,cAAI,CAAC,IAAI,CAAA;IAClC,CAAC;CACJ;AA3PD,wCA2PC","sourcesContent":["import Long from 'long'\n\nimport { mtp, tl } from '@mtcute/tl'\nimport { TlBinaryWriter, TlReaderMap, TlSerializationCounter, TlWriterMap } from '@mtcute/tl-runtime'\n\nimport { MtcuteError } from '../types/index.js'\nimport {\n    ControllablePromise,\n    Deque,\n    getRandomInt,\n    ICryptoProvider,\n    Logger,\n    LongMap,\n    LruSet,\n    randomLong,\n    SortedArray,\n} from '../utils/index.js'\nimport { AuthKey } from './auth-key.js'\n\nexport interface PendingRpc {\n    method: string\n    data: Uint8Array\n    promise: ControllablePromise\n    stack?: string\n    gzipOverhead?: number\n\n    sent?: boolean\n    done?: boolean\n    msgId?: Long\n    seqNo?: number\n    containerId?: Long\n    acked?: boolean\n    initConn?: boolean\n    getState?: number\n    cancelled?: boolean\n    timeout?: NodeJS.Timeout\n}\n\nexport type PendingMessage =\n    | {\n          _: 'rpc'\n          rpc: PendingRpc\n      }\n    | {\n          _: 'container'\n          msgIds: Long[]\n      }\n    | {\n          _: 'state'\n          msgIds: Long[]\n          containerId: Long\n      }\n    | {\n          _: 'resend'\n          msgIds: Long[]\n          containerId: Long\n      }\n    | {\n          _: 'ping'\n          pingId: Long\n          containerId: Long\n      }\n    | {\n          _: 'destroy_session'\n          sessionId: Long\n          containerId: Long\n      }\n    | {\n          _: 'cancel'\n          msgId: Long\n          containerId: Long\n      }\n    | {\n          _: 'future_salts'\n          containerId: Long\n      }\n    | {\n          _: 'bind'\n          promise: ControllablePromise\n      }\n\n/**\n * Class encapsulating a single MTProto session and storing\n * all the relevant state\n */\nexport class MtprotoSession {\n    _sessionId = randomLong()\n\n    _authKey = new AuthKey(this._crypto, this.log, this._readerMap)\n    _authKeyTemp = new AuthKey(this._crypto, this.log, this._readerMap)\n    _authKeyTempSecondary = new AuthKey(this._crypto, this.log, this._readerMap)\n\n    _timeOffset = 0\n    _lastMessageId = Long.ZERO\n    _seqNo = 0\n\n    serverSalt = Long.ZERO\n\n    /// state ///\n    // recent msg ids\n    recentOutgoingMsgIds = new LruSet<Long>(1000, true)\n    recentIncomingMsgIds = new LruSet<Long>(1000, true)\n\n    // queues\n    queuedRpc = new Deque<PendingRpc>()\n    queuedAcks: Long[] = []\n    queuedStateReq: Long[] = []\n    queuedResendReq: Long[] = []\n    queuedCancelReq: Long[] = []\n    getStateSchedule = new SortedArray<PendingRpc>([], (a, b) => a.getState! - b.getState!)\n\n    // requests info\n    pendingMessages = new LongMap<PendingMessage>()\n    destroySessionIdToMsgId = new LongMap<Long>()\n\n    lastPingRtt = NaN\n    lastPingTime = 0\n    lastPingMsgId = Long.ZERO\n    lastSessionCreatedUid = Long.ZERO\n\n    initConnectionCalled = false\n    authorizationPending = false\n\n    next429Timeout = 1000\n    current429Timeout?: NodeJS.Timeout\n    next429ResetTimeout?: NodeJS.Timeout\n\n    constructor(\n        readonly _crypto: ICryptoProvider,\n        readonly log: Logger,\n        readonly _readerMap: TlReaderMap,\n        readonly _writerMap: TlWriterMap,\n    ) {\n        this.log.prefix = `[SESSION ${this._sessionId.toString(16)}] `\n    }\n\n    get hasPendingMessages(): boolean {\n        return Boolean(\n            this.queuedRpc.length ||\n                this.queuedAcks.length ||\n                this.queuedStateReq.length ||\n                this.queuedResendReq.length,\n        )\n    }\n\n    /**\n     * Reset session by resetting auth key(s) and session state\n     */\n    reset(withAuthKey = false): void {\n        if (withAuthKey) {\n            this._authKey.reset()\n            this._authKeyTemp.reset()\n            this._authKeyTempSecondary.reset()\n        }\n\n        clearTimeout(this.current429Timeout)\n        this.resetState()\n        this.resetLastPing(true)\n    }\n\n    /**\n     * Reset session state and generate a new session ID.\n     *\n     * By default, also cancels any pending RPC requests.\n     * If `keepPending` is set to `true`, pending requests will be kept\n     */\n    resetState(keepPending = false): void {\n        this._lastMessageId = Long.ZERO\n        this._seqNo = 0\n\n        this._sessionId = randomLong()\n        this.log.debug('session reset, new sid = %h', this._sessionId)\n        this.log.prefix = `[SESSION ${this._sessionId.toString(16)}] `\n\n        // reset session state\n\n        if (!keepPending) {\n            for (const info of this.pendingMessages.values()) {\n                if (info._ === 'rpc') {\n                    info.rpc.promise.reject(new MtcuteError('Session is reset'))\n                }\n            }\n            this.pendingMessages.clear()\n        }\n\n        this.recentOutgoingMsgIds.clear()\n        this.recentIncomingMsgIds.clear()\n\n        if (!keepPending) {\n            while (this.queuedRpc.length) {\n                const rpc = this.queuedRpc.popFront()!\n\n                if (rpc.sent === false) {\n                    rpc.promise.reject(new MtcuteError('Session is reset'))\n                }\n            }\n        }\n\n        this.queuedAcks.length = 0\n        this.queuedStateReq.length = 0\n        this.queuedResendReq.length = 0\n        this.getStateSchedule.clear()\n    }\n\n    enqueueRpc(rpc: PendingRpc, force?: boolean): boolean {\n        // already queued or cancelled\n        if ((!force && !rpc.sent) || rpc.cancelled) return false\n\n        rpc.sent = false\n        rpc.containerId = undefined\n        this.log.debug('enqueued %s for sending (msg_id = %s)', rpc.method, rpc.msgId || 'n/a')\n        this.queuedRpc.pushBack(rpc)\n\n        return true\n    }\n\n    getMessageId(): Long {\n        const timeTicks = Date.now()\n        const timeSec = Math.floor(timeTicks / 1000) + this._timeOffset\n        const timeMSec = timeTicks % 1000\n        const random = getRandomInt(0xffff)\n\n        let messageId = new Long((timeMSec << 21) | (random << 3) | 4, timeSec)\n\n        if (this._lastMessageId.gt(messageId)) {\n            messageId = this._lastMessageId.add(4)\n        }\n\n        this._lastMessageId = messageId\n\n        return messageId\n    }\n\n    getSeqNo(isContentRelated = true): number {\n        let seqNo = this._seqNo\n\n        if (isContentRelated) {\n            seqNo += 1\n            this._seqNo += 2\n        }\n\n        return seqNo\n    }\n\n    /** Encrypt a single MTProto message using session's keys */\n    encryptMessage(message: Uint8Array): Uint8Array {\n        const key = this._authKeyTemp.ready ? this._authKeyTemp : this._authKey\n\n        return key.encryptMessage(message, this.serverSalt, this._sessionId)\n    }\n\n    /** Decrypt a single MTProto message using session's keys */\n    decryptMessage(data: Uint8Array, callback: Parameters<AuthKey['decryptMessage']>[2]): void {\n        if (!this._authKey.ready) throw new MtcuteError('Keys are not set up!')\n\n        const authKeyId = data.subarray(0, 8)\n\n        let key: AuthKey\n\n        if (this._authKey.match(authKeyId)) {\n            key = this._authKey\n        } else if (this._authKeyTemp.match(authKeyId)) {\n            key = this._authKeyTemp\n        } else if (this._authKeyTempSecondary.match(authKeyId)) {\n            key = this._authKeyTempSecondary\n        } else {\n            this.log.warn(\n                'received message with unknown authKey = %h (expected %h or %h or %h)',\n                authKeyId,\n                this._authKey.id,\n                this._authKeyTemp.id,\n                this._authKeyTempSecondary.id,\n            )\n\n            return\n        }\n\n        return key.decryptMessage(data, this._sessionId, callback)\n    }\n\n    writeMessage(\n        writer: TlBinaryWriter,\n        content: tl.TlObject | mtp.TlObject | Uint8Array,\n        isContentRelated = true,\n    ): Long {\n        const messageId = this.getMessageId()\n        const seqNo = this.getSeqNo(isContentRelated)\n\n        const length = ArrayBuffer.isView(content) ?\n            content.length :\n            TlSerializationCounter.countNeededBytes(writer.objectMap!, content)\n\n        writer.long(messageId)\n        writer.int(seqNo)\n        writer.uint(length)\n        if (ArrayBuffer.isView(content)) writer.raw(content)\n        else writer.object(content as tl.TlObject)\n\n        return messageId\n    }\n\n    onTransportFlood(callback: () => void) {\n        if (this.current429Timeout) return // already waiting\n\n        // all active queries must be resent after a timeout\n        this.resetLastPing(true)\n\n        const timeout = this.next429Timeout\n\n        this.next429Timeout = Math.min(this.next429Timeout * 2, 32000)\n        clearTimeout(this.current429Timeout)\n        clearTimeout(this.next429ResetTimeout)\n\n        this.current429Timeout = setTimeout(() => {\n            this.current429Timeout = undefined\n            callback()\n        }, timeout)\n        this.next429ResetTimeout = setTimeout(() => {\n            this.next429ResetTimeout = undefined\n            this.next429Timeout = 1000\n        }, 60000)\n\n        this.log.debug('transport flood, waiting for %d ms before proceeding', timeout)\n\n        return Date.now() + timeout\n    }\n\n    resetLastPing(withTime = false): void {\n        if (withTime) this.lastPingTime = 0\n\n        if (!this.lastPingMsgId.isZero()) {\n            this.pendingMessages.delete(this.lastPingMsgId)\n        }\n\n        this.lastPingMsgId = Long.ZERO\n    }\n}\n"]}

package/cjs/network/multi-session-connection.d.ts

@@ -22,7 +22,7 @@ export declare class MultiSessionConnection extends EventEmitter {
     sendRpc<T extends tl.RpcMethod>(request: T, stack?: string, timeout?: number, abortSignal?: AbortSignal): Promise<tl.RpcCallReturn[T['_']]>;
     connect(): void;
     ensureConnected(): void;
-    setAuthKey(authKey: Uint8Array | null, temp?: boolean, idx?: number): Promise<void>;
+    setAuthKey(authKey: Uint8Array | null, temp?: boolean, idx?: number): void;
     resetAuthKeys(): void;
     setInactivityTimeout(timeout?: number): void;
     notifyKeyChange(): void;

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

@@ -185,10 +185,10 @@ class MultiSessionConnection extends events_1.default {
             return;
         this.connect();
     }
-    async setAuthKey(authKey, temp = false, idx = 0) {
+    setAuthKey(authKey, temp = false, idx = 0) {
         const session = this._sessions[idx];
         const key = temp ? session._authKeyTemp : session._authKey;
-        await key.setup(authKey);
+        key.setup(authKey);
     }
     resetAuthKeys() {
         for (const session of this._sessions) {

package/cjs/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,oDAAiC;AAKjC,6DAAqD;AACrD,mEAAoF;AAGpF,MAAa,sBAAuB,SAAQ,gBAAY;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,mCAAc,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,yCAAiB,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,KAAK,CAAC,UAAU,CAAC,OAA0B,EAAE,IAAI,GAAG,KAAK,EAAE,GAAG,GAAG,CAAC;QAC9D,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,MAAM,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IAC5B,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;AAvSD,wDAuSC","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    async setAuthKey(authKey: Uint8Array | null, temp = false, idx = 0): Promise<void> {\n        const session = this._sessions[idx]\n        const key = temp ? session._authKeyTemp : session._authKey\n        await 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,oDAAiC;AAKjC,6DAAqD;AACrD,mEAAoF;AAGpF,MAAa,sBAAuB,SAAQ,gBAAY;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,mCAAc,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,yCAAiB,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;AAvSD,wDAuSC","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"]}

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

@@ -33,6 +33,7 @@ export interface NetworkManagerParams {
     isPremium: boolean;
     _emitError: (err: Error, connection?: SessionConnection) => void;
     keepAliveAction: () => void;
+    onUsable: () => void;
 }
 export type ConnectionCountDelegate = (kind: ConnectionKind, dcId: number, isPremium: boolean) => number;
 /**

package/cjs/network/network-manager.js

@@ -99,12 +99,10 @@ class DcConnectionManager {
             }
             this.manager._log.debug('key change for dc %d from connection %d', this.dcId, idx);
             // send key to other connections
-            Promise.all([
-                this.manager._storage.setAuthKeyFor(this.dcId, key),
-                this.upload.setAuthKey(key),
-                this.download.setAuthKey(key),
-                this.downloadSmall.setAuthKey(key),
-            ])
+            this.upload.setAuthKey(key);
+            this.download.setAuthKey(key);
+            this.downloadSmall.setAuthKey(key);
+            Promise.resolve(this.manager._storage.setAuthKeyFor(this.dcId, key))
                 .then(() => {
                 this.upload.notifyKeyChange();
                 this.download.notifyKeyChange();
@@ -119,12 +117,10 @@ class DcConnectionManager {
             }
             this.manager._log.debug('temp key change for dc %d from connection %d', this.dcId, idx);
             // send key to other connections
-            Promise.all([
-                this.manager._storage.setTempAuthKeyFor(this.dcId, idx, key, expires * 1000),
-                this.upload.setAuthKey(key, true),
-                this.download.setAuthKey(key, true),
-                this.downloadSmall.setAuthKey(key, true),
-            ])
+            this.upload.setAuthKey(key, true);
+            this.download.setAuthKey(key, true);
+            this.downloadSmall.setAuthKey(key, true);
+            Promise.resolve(this.manager._storage.setTempAuthKeyFor(this.dcId, idx, key, expires * 1000))
                 .then(() => {
                 this.upload.notifyKeyChange();
                 this.download.notifyKeyChange();
@@ -170,25 +166,21 @@ class DcConnectionManager {
     }
     async loadKeys() {
         const permanent = await this.manager._storage.getAuthKeyFor(this.dcId);
-        await Promise.all([
-            this.main.setAuthKey(permanent),
-            this.upload.setAuthKey(permanent),
-            this.download.setAuthKey(permanent),
-            this.downloadSmall.setAuthKey(permanent),
-        ]);
+        this.main.setAuthKey(permanent);
+        this.upload.setAuthKey(permanent);
+        this.download.setAuthKey(permanent);
+        this.downloadSmall.setAuthKey(permanent);
         if (!permanent) {
             return false;
         }
         if (this.manager.params.usePfs) {
             await Promise.all(this.main._sessions.map(async (_, i) => {
                 const temp = await this.manager._storage.getAuthKeyFor(this.dcId, i);
-                await this.main.setAuthKey(temp, true, i);
+                this.main.setAuthKey(temp, true, i);
                 if (i === 0) {
-                    await Promise.all([
-                        this.upload.setAuthKey(temp, true),
-                        this.download.setAuthKey(temp, true),
-                        this.downloadSmall.setAuthKey(temp, true),
-                    ]);
+                    this.upload.setAuthKey(temp, true);
+                    this.download.setAuthKey(temp, true);
+                    this.downloadSmall.setAuthKey(temp, true);
                 }
             }));
         }
@@ -227,7 +219,7 @@ class NetworkManager {
             _: 'initConnection',
             deviceModel,
             systemVersion: '1.0',
-            appVersion: '0.1.0',
+            appVersion: '0.1.2',
             systemLangCode: 'en',
             langPack: '',
             langCode: 'en',
@@ -271,6 +263,7 @@ class NetworkManager {
         dc.setIsPrimary(true);
         dc.main.on('usable', () => {
             this._lastUpdateTime = Date.now();
+            this.params.onUsable();
             if (this._keepAliveInterval)
                 clearInterval(this._keepAliveInterval);
             this._keepAliveInterval = setInterval(() => {
@@ -454,8 +447,9 @@ class NetworkManager {
             }
             catch (e) {
                 lastError = e;
-                if (!tl_1.tl.RpcError.is(e))
-                    continue;
+                if (!tl_1.tl.RpcError.is(e)) {
+                    throw e;
+                }
                 if (!(e.code in CLIENT_ERRORS)) {
                     this._log.warn('Telegram is having internal issues: %d %s, retrying', e.code, e.message);
                     if (e.text === 'WORKER_BUSY_TOO_LONG_RETRY') {