@mtcute/core 0.1.0

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

@@ -0,0 +1,236 @@
+import EventEmitter from 'events';
+import { MtprotoSession } from './mtproto-session.js';
+import { SessionConnection } from './session-connection.js';
+export class MultiSessionConnection extends EventEmitter {
+    constructor(params, _count, log, logPrefix = '') {
+        super();
+        this.params = params;
+        this._count = _count;
+        this._enforcePfs = false;
+        this._connections = [];
+        this._destroyed = false;
+        this._nextConnection = 0;
+        this._log = log.create('multi');
+        if (logPrefix)
+            this._log.prefix = `[${logPrefix}] `;
+        this._enforcePfs = _count > 1 && params.isMainConnection;
+        this._sessions = [];
+        this._updateConnections();
+    }
+    setCount(count, connect = this.params.isMainConnection) {
+        this._count = count;
+        this._updateConnections(connect);
+    }
+    _updateSessions() {
+        // there are two cases
+        // 1. this msc is main, in which case every connection should have its own session
+        // 2. this msc is not main, in which case all connections should share the same session
+        // if (!this.params.isMainConnection) {
+        //     // case 2
+        //     this._log.debug(
+        //         'updating sessions count: %d -> 1',
+        //         this._sessions.length,
+        //     )
+        //
+        //     if (this._sessions.length === 0) {
+        //         this._sessions.push(
+        //             new MtprotoSession(
+        //                 this.params.crypto,
+        //                 this._log.create('session'),
+        //                 this.params.readerMap,
+        //                 this.params.writerMap,
+        //             ),
+        //         )
+        //     }
+        //
+        //     // shouldn't happen, but just in case
+        //     while (this._sessions.length > 1) {
+        //         this._sessions.pop()!.reset()
+        //     }
+        //
+        //     return
+        // }
+        this._log.debug('updating sessions count: %d -> %d', this._sessions.length, this._count);
+        // case 1
+        if (this._sessions.length === this._count)
+            return;
+        if (this._sessions.length > this._count) {
+            // destroy extra sessions
+            for (let i = this._sessions.length - 1; i >= this._count; i--) {
+                this._sessions[i].reset();
+            }
+            this._sessions.splice(this._count);
+            return;
+        }
+        while (this._sessions.length < this._count) {
+            const idx = this._sessions.length;
+            const session = new MtprotoSession(this.params.crypto, this._log.create('session'), this.params.readerMap, this.params.writerMap);
+            // brvh
+            if (idx !== 0)
+                session._authKey = this._sessions[0]._authKey;
+            this._sessions.push(session);
+        }
+    }
+    _updateConnections(connect = false) {
+        this._updateSessions();
+        if (this._connections.length === this._count)
+            return;
+        this._log.debug('updating connections count: %d -> %d', this._connections.length, this._count);
+        const newEnforcePfs = this._count > 1 && this.params.isMainConnection;
+        const enforcePfsChanged = newEnforcePfs !== this._enforcePfs;
+        if (enforcePfsChanged) {
+            this._log.debug('enforcePfs changed: %s -> %s', this._enforcePfs, newEnforcePfs);
+            this._enforcePfs = newEnforcePfs;
+        }
+        if (this._connections.length > this._count) {
+            // destroy extra connections
+            for (let i = this._connections.length - 1; i >= this._count; i--) {
+                this._connections[i].removeAllListeners();
+                this._connections[i].destroy();
+            }
+            this._connections.splice(this._count);
+            return;
+        }
+        if (enforcePfsChanged) {
+            this._connections.forEach((conn) => {
+                conn.setUsePfs(this.params.usePfs || this._enforcePfs);
+            });
+        }
+        // create new connections
+        for (let i = this._connections.length; i < this._count; i++) {
+            const session = this._sessions[i]; // this.params.isMainConnection ? // :
+            // this._sessions[0]
+            const conn = new SessionConnection({
+                ...this.params,
+                usePfs: this.params.usePfs || this._enforcePfs,
+                isMainConnection: this.params.isMainConnection && i === 0,
+                withUpdates: this.params.isMainConnection && this.params.isMainDcConnection && !this.params.disableUpdates,
+            }, session);
+            if (this.params.isMainConnection) {
+                conn.on('update', (update) => this.emit('update', update));
+            }
+            conn.on('error', (err) => this.emit('error', err, conn));
+            conn.on('key-change', (key) => {
+                this.emit('key-change', i, key);
+                // notify other connections
+                for (const conn_ of this._connections) {
+                    if (conn_ === conn)
+                        continue;
+                    conn_.onConnected();
+                }
+            });
+            conn.on('tmp-key-change', (key, expires) => this.emit('tmp-key-change', i, key, expires));
+            conn.on('auth-begin', () => {
+                this._log.debug('received auth-begin from connection %d', i);
+                this.emit('auth-begin', i);
+                // we need to reset temp auth keys if there are any left
+                this._connections.forEach((conn_) => {
+                    conn_._session._authKeyTemp.reset();
+                    if (conn_ !== conn)
+                        conn_.reconnect();
+                });
+            });
+            conn.on('usable', () => this.emit('usable', i));
+            conn.on('request-auth', () => this.emit('request-auth', i));
+            conn.on('flood-done', () => {
+                this._log.debug('received flood-done from connection %d', i);
+                this._connections.forEach((it) => it.flushWhenIdle());
+            });
+            this._connections.push(conn);
+            if (connect)
+                conn.connect();
+        }
+    }
+    destroy() {
+        this._connections.forEach((conn) => conn.destroy());
+        this._sessions.forEach((sess) => sess.reset());
+        this.removeAllListeners();
+        this._destroyed = true;
+    }
+    sendRpc(request, stack, timeout, abortSignal) {
+        // if (this.params.isMainConnection) {
+        // find the least loaded connection
+        let min = Infinity;
+        let minIdx = 0;
+        for (let i = 0; i < this._connections.length; i++) {
+            const conn = this._connections[i];
+            const total = conn._session.queuedRpc.length + conn._session.pendingMessages.size();
+            if (total < min) {
+                min = total;
+                minIdx = i;
+            }
+        }
+        return this._connections[minIdx].sendRpc(request, stack, timeout, abortSignal);
+        // }
+        // round-robin connections
+        // since they all share the same session, it doesn't matter which one we use
+        // the connection chosen here will only affect the first attempt at sending
+        // return this._connections[
+        //     this._nextConnection++ % this._connections.length
+        // ].sendRpc(request, stack, timeout)
+    }
+    connect() {
+        for (const conn of this._connections) {
+            conn.connect();
+        }
+    }
+    ensureConnected() {
+        if (this._connections[0].isConnected)
+            return;
+        this.connect();
+    }
+    async setAuthKey(authKey, temp = false, idx = 0) {
+        const session = this._sessions[idx];
+        const key = temp ? session._authKeyTemp : session._authKey;
+        await key.setup(authKey);
+    }
+    resetAuthKeys() {
+        for (const session of this._sessions) {
+            session.reset(true);
+        }
+        this.notifyKeyChange();
+    }
+    setInactivityTimeout(timeout) {
+        this._log.debug('setting inactivity timeout to %s', timeout);
+        // for future connections (if any)
+        this.params.inactivityTimeout = timeout;
+        // for current connections
+        for (const conn of this._connections) {
+            conn.setInactivityTimeout(timeout);
+        }
+    }
+    notifyKeyChange() {
+        // only expected to be called on non-main connections
+        const session = this._sessions[0];
+        if (this.params.usePfs && !session._authKeyTemp.ready) {
+            this._log.debug('temp auth key needed but not ready, ignoring key change');
+            return;
+        }
+        if (this._sessions[0].queuedRpc.length) {
+            // there are pending requests, we need to reconnect.
+            this._log.debug('notifying key change on the connection due to queued rpc');
+            this._connections[0].onConnected();
+        }
+        // connection is idle, we don't need to notify it
+    }
+    requestAuth() {
+        this._connections[0]._authorize();
+    }
+    resetSessions() {
+        if (this.params.isMainConnection) {
+            for (const conn of this._connections) {
+                conn._resetSession();
+            }
+        }
+        else {
+            this._connections[0]._resetSession();
+        }
+    }
+    changeTransport(factory) {
+        this._connections.forEach((conn) => conn.changeTransport(factory));
+    }
+    getPoolSize() {
+        return this._connections.length;
+    }
+}
+//# sourceMappingURL=multi-session-connection.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"multi-session-connection.js","sourceRoot":"","sources":["../../../src/network/multi-session-connection.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AAKjC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAA;AACrD,OAAO,EAAE,iBAAiB,EAA2B,MAAM,yBAAyB,CAAA;AAGpF,MAAM,OAAO,sBAAuB,SAAQ,YAAY;IAKpD,YACa,MAA+B,EAChC,MAAc,EACtB,GAAW,EACX,SAAS,GAAG,EAAE;QAEd,KAAK,EAAE,CAAA;QALE,WAAM,GAAN,MAAM,CAAyB;QAChC,WAAM,GAAN,MAAM,CAAQ;QAJlB,gBAAW,GAAG,KAAK,CAAA;QAiBjB,iBAAY,GAAwB,EAAE,CAAA;QA2JhD,eAAU,GAAG,KAAK,CAAA;QASV,oBAAe,GAAG,CAAC,CAAA;QA5KvB,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAC/B,IAAI,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,IAAI,CAAA;QACnD,IAAI,CAAC,WAAW,GAAG,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,gBAAgB,CAAA;QAExD,IAAI,CAAC,SAAS,GAAG,EAAE,CAAA;QACnB,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC7B,CAAC;IAID,QAAQ,CAAC,KAAa,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;QAC1D,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QAEnB,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC;IAEO,eAAe;QACnB,sBAAsB;QACtB,kFAAkF;QAClF,uFAAuF;QACvF,uCAAuC;QACvC,gBAAgB;QAChB,uBAAuB;QACvB,8CAA8C;QAC9C,iCAAiC;QACjC,QAAQ;QACR,EAAE;QACF,yCAAyC;QACzC,+BAA+B;QAC/B,kCAAkC;QAClC,sCAAsC;QACtC,+CAA+C;QAC/C,yCAAyC;QACzC,yCAAyC;QACzC,iBAAiB;QACjB,YAAY;QACZ,QAAQ;QACR,EAAE;QACF,4CAA4C;QAC5C,0CAA0C;QAC1C,wCAAwC;QACxC,QAAQ;QACR,EAAE;QACF,aAAa;QACb,IAAI;QAEJ,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,mCAAmC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;QAExF,SAAS;QACT,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAM;QAEjD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACrC,yBAAyB;YACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC3D,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAA;aAC5B;YAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAElC,OAAM;SACT;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACxC,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAA;YACjC,MAAM,OAAO,GAAG,IAAI,cAAc,CAC9B,IAAI,CAAC,MAAM,CAAC,MAAM,EAClB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAC3B,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,IAAI,CAAC,MAAM,CAAC,SAAS,CACxB,CAAA;YAED,OAAO;YACP,IAAI,GAAG,KAAK,CAAC;gBAAE,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAA;YAE5D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;SAC/B;IACL,CAAC;IAEO,kBAAkB,CAAC,OAAO,GAAG,KAAK;QACtC,IAAI,CAAC,eAAe,EAAE,CAAA;QACtB,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;YAAE,OAAM;QAEpD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,sCAAsC,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;QAE9F,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAA;QACrE,MAAM,iBAAiB,GAAG,aAAa,KAAK,IAAI,CAAC,WAAW,CAAA;QAE5D,IAAI,iBAAiB,EAAE;YACnB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,8BAA8B,EAAE,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,CAAA;YAChF,IAAI,CAAC,WAAW,GAAG,aAAa,CAAA;SACnC;QAED,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE;YACxC,4BAA4B;YAC5B,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC9D,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAA;gBACzC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;aACjC;YAED,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAErC,OAAM;SACT;QAED,IAAI,iBAAiB,EAAE;YACnB,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,CAAA;YAC1D,CAAC,CAAC,CAAA;SACL;QAED,yBAAyB;QACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACzD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA,CAAC,sCAAsC;YACxE,oBAAoB;YACpB,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAC9B;gBACI,GAAG,IAAI,CAAC,MAAM;gBACd,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW;gBAC9C,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,KAAK,CAAC;gBACzD,WAAW,EACP,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc;aACpG,EACD,OAAO,CACV,CAAA;YAED,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE;gBAC9B,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;aAC7D;YACD,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAA;YACxD,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC1B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,GAAG,CAAC,CAAA;gBAE/B,2BAA2B;gBAC3B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE;oBACnC,IAAI,KAAK,KAAK,IAAI;wBAAE,SAAQ;oBAC5B,KAAK,CAAC,WAAW,EAAE,CAAA;iBACtB;YACL,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC,CAAA;YACzF,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAA;gBAC5D,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAA;gBAE1B,wDAAwD;gBAExD,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;oBACnC,IAAI,KAAK,KAAK,IAAI;wBAAE,KAAK,CAAC,SAAS,EAAE,CAAA;gBACzC,CAAC,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAA;YAC/C,IAAI,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAA;YAC3D,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAA;gBAE5D,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,CAAA;YACzD,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC5B,IAAI,OAAO;gBAAE,IAAI,CAAC,OAAO,EAAE,CAAA;SAC9B;IACL,CAAC;IAGD,OAAO;QACH,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAA;QACnD,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;QAC9C,IAAI,CAAC,kBAAkB,EAAE,CAAA;QAEzB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;IAC1B,CAAC;IAID,OAAO,CACH,OAAU,EACV,KAAc,EACd,OAAgB,EAChB,WAAyB;QAEzB,sCAAsC;QACtC,mCAAmC;QACnC,IAAI,GAAG,GAAG,QAAQ,CAAA;QAClB,IAAI,MAAM,GAAG,CAAC,CAAA;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACjC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,EAAE,CAAA;YAEnF,IAAI,KAAK,GAAG,GAAG,EAAE;gBACb,GAAG,GAAG,KAAK,CAAA;gBACX,MAAM,GAAG,CAAC,CAAA;aACb;SACJ;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC,CAAA;QAC9E,IAAI;QAEJ,0BAA0B;QAC1B,4EAA4E;QAC5E,2EAA2E;QAC3E,4BAA4B;QAC5B,wDAAwD;QACxD,qCAAqC;IACzC,CAAC;IAED,OAAO;QACH,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE;YAClC,IAAI,CAAC,OAAO,EAAE,CAAA;SACjB;IACL,CAAC;IAED,eAAe;QACX,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW;YAAE,OAAM;QAE5C,IAAI,CAAC,OAAO,EAAE,CAAA;IAClB,CAAC;IAED,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","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"]}

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

@@ -0,0 +1,190 @@
+import { tl } from '@mtcute/tl';
+import { TlReaderMap, TlWriterMap } from '@mtcute/tl-runtime';
+import { ITelegramStorage } from '../storage/index.js';
+import { ICryptoProvider, Logger } from '../utils/index.js';
+import { ConfigManager } from './config-manager.js';
+import { MultiSessionConnection } from './multi-session-connection.js';
+import { PersistentConnectionParams } from './persistent-connection.js';
+import { ReconnectionStrategy } from './reconnection.js';
+import { SessionConnection } from './session-connection.js';
+import { TransportFactory } from './transports/index.js';
+export type ConnectionKind = 'main' | 'upload' | 'download' | 'downloadSmall';
+/**
+ * Params passed into {@link NetworkManager} by {@link TelegramClient}.
+ * This type is intended for internal usage only.
+ */
+export interface NetworkManagerParams {
+    storage: ITelegramStorage;
+    crypto: ICryptoProvider;
+    log: Logger;
+    enableErrorReporting: boolean;
+    apiId: number;
+    initConnectionOptions?: Partial<Omit<tl.RawInitConnectionRequest, 'apiId' | 'query'>>;
+    transport?: TransportFactory;
+    reconnectionStrategy?: ReconnectionStrategy<PersistentConnectionParams>;
+    floodSleepThreshold: number;
+    maxRetryCount: number;
+    disableUpdates?: boolean;
+    testMode: boolean;
+    layer: number;
+    useIpv6: boolean;
+    readerMap: TlReaderMap;
+    writerMap: TlWriterMap;
+    isPremium: boolean;
+    _emitError: (err: Error, connection?: SessionConnection) => void;
+    keepAliveAction: () => void;
+}
+export type ConnectionCountDelegate = (kind: ConnectionKind, dcId: number, isPremium: boolean) => number;
+/**
+ * Additional params passed into {@link NetworkManager} by the user
+ * that customize the behavior of the manager
+ */
+export interface NetworkManagerExtraParams {
+    /**
+     * Whether to use PFS (Perfect Forward Secrecy) for all connections.
+     * This is disabled by default
+     */
+    usePfs?: boolean;
+    /**
+     * Connection count for each connection kind.
+     * The function should be pure to avoid unexpected behavior.
+     *
+     * Defaults to TDLib logic:
+     *   - main: handled internally, **cannot be changed here**
+     *   - upload: if premium or dc id is other than 2 or 4, then 8, otherwise 4
+     *   - download: if premium then 8, otherwise 2
+     *   - downloadSmall: 2
+     */
+    connectionCount?: ConnectionCountDelegate;
+    /**
+     * Idle timeout for non-main connections, in ms
+     *
+     * @default  60000 (60 seconds).
+     */
+    inactivityTimeout?: number;
+}
+/** Options that can be customized when making an RPC call */
+export interface RpcCallOptions {
+    /**
+     * If the call results in a `FLOOD_WAIT_X` error,
+     * the maximum amount of time to wait before retrying.
+     *
+     * If set to `0`, the call will not be retried.
+     *
+     * @default {@link BaseTelegramClientOptions.floodSleepThreshold}
+     */
+    floodSleepThreshold?: number;
+    /**
+     * If the call results in an internal server error or a flood wait,
+     * the maximum amount of times to retry the call.
+     *
+     * @default {@link BaseTelegramClientOptions.maxRetryCount}
+     */
+    maxRetryCount?: number;
+    /**
+     * Timeout for the call, in milliseconds.
+     *
+     * @default Infinity
+     */
+    timeout?: number;
+    /**
+     * Kind of connection to use for this call.
+     *
+     * @default 'main'
+     */
+    kind?: ConnectionKind;
+    /**
+     * ID of the DC to use for this call
+     */
+    dcId?: number;
+    /**
+     * DC connection manager to use for this call.
+     * Overrides `dcId` if set.
+     */
+    manager?: DcConnectionManager;
+    /**
+     * Abort signal for the call.
+     */
+    abortSignal?: AbortSignal;
+}
+/**
+ * Wrapper over all connection pools for a single DC.
+ */
+export declare class DcConnectionManager {
+    /** Network manager instance */
+    readonly manager: NetworkManager;
+    /** DC ID */
+    readonly dcId: number;
+    /** DC options to use */
+    readonly _dcs: ITelegramStorage.DcOptions;
+    /** Whether this DC is the primary one */
+    isPrimary: boolean;
+    private __baseConnectionParams;
+    private _log;
+    /** Main connection pool */
+    main: MultiSessionConnection;
+    /** Upload connection pool */
+    upload: MultiSessionConnection;
+    /** Download connection pool */
+    download: MultiSessionConnection;
+    /** Download connection pool (for small files) */
+    downloadSmall: MultiSessionConnection;
+    private get _mainConnectionCount();
+    constructor(
+    /** Network manager instance */
+    manager: NetworkManager, 
+    /** DC ID */
+    dcId: number, 
+    /** DC options to use */
+    _dcs: ITelegramStorage.DcOptions, 
+    /** Whether this DC is the primary one */
+    isPrimary?: boolean);
+    private _setupMulti;
+    setIsPrimary(isPrimary: boolean): void;
+    setIsPremium(isPremium: boolean): void;
+    loadKeys(): Promise<boolean>;
+}
+/**
+ * Class that manages all connections to Telegram servers.
+ */
+export declare class NetworkManager {
+    readonly params: NetworkManagerParams & NetworkManagerExtraParams;
+    readonly config: ConfigManager;
+    readonly _log: Logger;
+    readonly _storage: ITelegramStorage;
+    readonly _initConnectionParams: tl.RawInitConnectionRequest;
+    readonly _transportFactory: TransportFactory;
+    readonly _reconnectionStrategy: ReconnectionStrategy<PersistentConnectionParams>;
+    readonly _connectionCount: ConnectionCountDelegate;
+    protected readonly _dcConnections: Map<number, DcConnectionManager>;
+    protected _primaryDc?: DcConnectionManager;
+    private _keepAliveInterval?;
+    private _lastUpdateTime;
+    private _updateHandler;
+    constructor(params: NetworkManagerParams & NetworkManagerExtraParams, config: ConfigManager);
+    private _findDcOptions;
+    private _switchPrimaryDc;
+    private _dcCreationPromise;
+    _getOtherDc(dcId: number): Promise<DcConnectionManager>;
+    /**
+     * Perform initial connection to the default DC
+     *
+     * @param defaultDcs  Default DCs to connect to
+     */
+    connect(defaultDcs: ITelegramStorage.DcOptions): Promise<void>;
+    private _pendingExports;
+    private _exportAuthTo;
+    setIsPremium(isPremium: boolean): void;
+    notifyLoggedIn(auth: tl.auth.TypeAuthorization): void;
+    resetSessions(): void;
+    private _onConfigChanged;
+    changePrimaryDc(newDc: number): Promise<void>;
+    private _floodWaitedRequests;
+    call<T extends tl.RpcMethod>(message: T, params?: RpcCallOptions, stack?: string): Promise<tl.RpcCallReturn[T['_']]>;
+    setUpdateHandler(handler: (upd: tl.TypeUpdates, fromClient: boolean) => void): void;
+    handleUpdate(update: tl.TypeUpdates, fromClient?: boolean): void;
+    changeTransport(factory: TransportFactory): void;
+    getPoolSize(kind: ConnectionKind, dcId?: number): number;
+    getPrimaryDcId(): number;
+    destroy(): void;
+}