@mtcute/core 0.1.0

package/esm/network/transports/obfuscated.js

@@ -0,0 +1,79 @@
+import { concatBuffers, dataViewFromBuffer } from '../../utils/buffer-utils.js';
+import { randomBytes } from '../../utils/index.js';
+import { WrappedCodec } from './wrapped.js';
+export class ObfuscatedPacketCodec extends WrappedCodec {
+    constructor(inner, proxy) {
+        super(inner);
+        this._proxy = proxy;
+    }
+    async tag() {
+        let random;
+        let dv;
+        for (;;) {
+            random = randomBytes(64);
+            if (random[0] === 0xef)
+                continue;
+            dv = dataViewFromBuffer(random);
+            const firstInt = dv.getInt32(0, true);
+            if (firstInt === 0x44414548 || // HEAD
+                firstInt === 0x54534f50 || // POST
+                firstInt === 0x20544547 || // GET(space)
+                firstInt === 0x4954504f || // OPTI
+                firstInt === 0xdddddddd || // padded intermediate
+                firstInt === 0xeeeeeeee || // intermediate
+                firstInt === 0x02010316 // no idea, taken from tdlib
+            ) {
+                continue;
+            }
+            if (dv.getInt32(4, true) === 0)
+                continue;
+            break;
+        }
+        let innerTag = await this._inner.tag();
+        if (innerTag.length !== 4) {
+            const b = innerTag[0];
+            innerTag = new Uint8Array([b, b, b, b]);
+        }
+        random.set(innerTag, 56);
+        if (this._proxy) {
+            let dcId = this._proxy.dcId;
+            if (this._proxy.test)
+                dcId += 10000;
+            if (this._proxy.media)
+                dcId = -dcId;
+            dv.setInt16(60, dcId, true);
+        }
+        // randomBytes may return a Buffer in Node.js, whose .slice() doesn't copy
+        const randomRev = Uint8Array.prototype.slice.call(random, 8, 56).reverse();
+        let encryptKey = random.subarray(8, 40);
+        const encryptIv = random.subarray(40, 56);
+        let decryptKey = randomRev.subarray(0, 32);
+        const decryptIv = randomRev.subarray(32, 48);
+        if (this._proxy) {
+            encryptKey = await this._crypto.sha256(concatBuffers([encryptKey, this._proxy.secret]));
+            decryptKey = await this._crypto.sha256(concatBuffers([decryptKey, this._proxy.secret]));
+        }
+        this._encryptor = await this._crypto.createAesCtr(encryptKey, encryptIv, true);
+        this._decryptor = await this._crypto.createAesCtr(decryptKey, decryptIv, false);
+        const encrypted = await this._encryptor.encrypt(random);
+        random.set(encrypted.subarray(56, 64), 56);
+        return random;
+    }
+    async encode(packet) {
+        return this._encryptor.encrypt(await this._inner.encode(packet));
+    }
+    feed(data) {
+        const dec = this._decryptor.decrypt(data);
+        if (ArrayBuffer.isView(dec))
+            this._inner.feed(dec);
+        else {
+            dec.then((dec) => this._inner.feed(dec)).catch((err) => this.emit('error', err));
+        }
+    }
+    reset() {
+        this._inner.reset();
+        delete this._encryptor;
+        delete this._decryptor;
+    }
+}
+//# sourceMappingURL=obfuscated.js.map

package/esm/network/transports/obfuscated.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"obfuscated.js","sourceRoot":"","sources":["../../../../src/network/transports/obfuscated.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAA;AAC/E,OAAO,EAAqB,WAAW,EAAE,MAAM,sBAAsB,CAAA;AAErE,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAA;AAS3C,MAAM,OAAO,qBAAsB,SAAQ,YAAY;IAMnD,YAAY,KAAmB,EAAE,KAAmB;QAChD,KAAK,CAAC,KAAK,CAAC,CAAA;QACZ,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,GAAG;QACL,IAAI,MAAkB,CAAA;QACtB,IAAI,EAAY,CAAA;QAEhB,SAAS;YACL,MAAM,GAAG,WAAW,CAAC,EAAE,CAAC,CAAA;YACxB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI;gBAAE,SAAQ;YAEhC,EAAE,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAA;YAC/B,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAErC,IACI,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,aAAa;gBACxC,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,sBAAsB;gBACjD,QAAQ,KAAK,UAAU,IAAI,eAAe;gBAC1C,QAAQ,KAAK,UAAU,CAAC,4BAA4B;cACtD;gBACE,SAAQ;aACX;YACD,IAAI,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC;gBAAE,SAAQ;YAExC,MAAK;SACR;QAED,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAA;QAEtC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;YACvB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAA;YACrB,QAAQ,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;SAC1C;QACD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAA;QAExB,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAA;YAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAAE,IAAI,IAAI,KAAK,CAAA;YACnC,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK;gBAAE,IAAI,GAAG,CAAC,IAAI,CAAA;YAEnC,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAA;SAC9B;QAED,0EAA0E;QAC1E,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAA;QAE1E,IAAI,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACvC,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAEzC,IAAI,UAAU,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAC1C,MAAM,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAE5C,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;YACvF,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;SAC1F;QAED,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,CAAA;QAC9E,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,SAAS,EAAE,KAAK,CAAC,CAAA;QAE/E,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QACvD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;QAE1C,OAAO,MAAM,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAkB;QAC3B,OAAO,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;IACrE,CAAC;IAED,IAAI,CAAC,IAAgB;QACjB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAE1C,IAAI,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;aAC7C;YACD,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;SACnF;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;QACnB,OAAO,IAAI,CAAC,UAAU,CAAA;QACtB,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;CACJ","sourcesContent":["import { concatBuffers, dataViewFromBuffer } from '../../utils/buffer-utils.js'\nimport { IEncryptionScheme, randomBytes } from '../../utils/index.js'\nimport { IPacketCodec } from './abstract.js'\nimport { WrappedCodec } from './wrapped.js'\n\nexport interface MtProxyInfo {\n    dcId: number\n    secret: Uint8Array\n    test: boolean\n    media: boolean\n}\n\nexport class ObfuscatedPacketCodec extends WrappedCodec implements IPacketCodec {\n    private _encryptor?: IEncryptionScheme\n    private _decryptor?: IEncryptionScheme\n\n    private _proxy?: MtProxyInfo\n\n    constructor(inner: IPacketCodec, proxy?: MtProxyInfo) {\n        super(inner)\n        this._proxy = proxy\n    }\n\n    async tag(): Promise<Uint8Array> {\n        let random: Uint8Array\n        let dv: DataView\n\n        for (;;) {\n            random = randomBytes(64)\n            if (random[0] === 0xef) continue\n\n            dv = dataViewFromBuffer(random)\n            const firstInt = dv.getInt32(0, true)\n\n            if (\n                firstInt === 0x44414548 || // HEAD\n                firstInt === 0x54534f50 || // POST\n                firstInt === 0x20544547 || // GET(space)\n                firstInt === 0x4954504f || // OPTI\n                firstInt === 0xdddddddd || // padded intermediate\n                firstInt === 0xeeeeeeee || // intermediate\n                firstInt === 0x02010316 // no idea, taken from tdlib\n            ) {\n                continue\n            }\n            if (dv.getInt32(4, true) === 0) continue\n\n            break\n        }\n\n        let innerTag = await this._inner.tag()\n\n        if (innerTag.length !== 4) {\n            const b = innerTag[0]\n            innerTag = new Uint8Array([b, b, b, b])\n        }\n        random.set(innerTag, 56)\n\n        if (this._proxy) {\n            let dcId = this._proxy.dcId\n            if (this._proxy.test) dcId += 10000\n            if (this._proxy.media) dcId = -dcId\n\n            dv.setInt16(60, dcId, true)\n        }\n\n        // randomBytes may return a Buffer in Node.js, whose .slice() doesn't copy\n        const randomRev = Uint8Array.prototype.slice.call(random, 8, 56).reverse()\n\n        let encryptKey = random.subarray(8, 40)\n        const encryptIv = random.subarray(40, 56)\n\n        let decryptKey = randomRev.subarray(0, 32)\n        const decryptIv = randomRev.subarray(32, 48)\n\n        if (this._proxy) {\n            encryptKey = await this._crypto.sha256(concatBuffers([encryptKey, this._proxy.secret]))\n            decryptKey = await this._crypto.sha256(concatBuffers([decryptKey, this._proxy.secret]))\n        }\n\n        this._encryptor = await this._crypto.createAesCtr(encryptKey, encryptIv, true)\n        this._decryptor = await this._crypto.createAesCtr(decryptKey, decryptIv, false)\n\n        const encrypted = await this._encryptor.encrypt(random)\n        random.set(encrypted.subarray(56, 64), 56)\n\n        return random\n    }\n\n    async encode(packet: Uint8Array): Promise<Uint8Array> {\n        return this._encryptor!.encrypt(await this._inner.encode(packet))\n    }\n\n    feed(data: Uint8Array): void {\n        const dec = this._decryptor!.decrypt(data)\n\n        if (ArrayBuffer.isView(dec)) this._inner.feed(dec)\n        else {\n            dec.then((dec) => this._inner.feed(dec)).catch((err) => this.emit('error', err))\n        }\n    }\n\n    reset(): void {\n        this._inner.reset()\n        delete this._encryptor\n        delete this._decryptor\n    }\n}\n"]}

package/esm/network/transports/streamed.d.ts

@@ -0,0 +1,24 @@
+/// <reference types="node" resolution-mode="require"/>
+import EventEmitter from 'events';
+/**
+ * Base for streamed codecs.
+ *
+ * Streamed means that MTProto packet can be divided into
+ * multiple transport packets.
+ */
+export declare abstract class StreamedCodec extends EventEmitter {
+    protected _stream: Uint8Array;
+    /**
+     * Should return whether a full packet is available
+     * in `_stream` buffer
+     */
+    protected abstract _packetAvailable(): boolean;
+    /**
+     * Handle a single (!) packet from `_stream` and emit `packet` or `error`.
+     *
+     * Should return true if there are more packets available, false otherwise
+     */
+    protected abstract _handlePacket(): boolean;
+    feed(data: Uint8Array): void;
+    reset(): void;
+}

package/esm/network/transports/streamed.js

@@ -0,0 +1,25 @@
+import EventEmitter from 'events';
+import { concatBuffers } from '../../utils/index.js';
+/**
+ * Base for streamed codecs.
+ *
+ * Streamed means that MTProto packet can be divided into
+ * multiple transport packets.
+ */
+export class StreamedCodec extends EventEmitter {
+    constructor() {
+        super(...arguments);
+        this._stream = new Uint8Array(0);
+    }
+    feed(data) {
+        this._stream = concatBuffers([this._stream, data]);
+        while (this._packetAvailable()) {
+            if (!this._handlePacket())
+                break;
+        }
+    }
+    reset() {
+        this._stream = new Uint8Array(0);
+    }
+}
+//# sourceMappingURL=streamed.js.map

package/esm/network/transports/streamed.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamed.js","sourceRoot":"","sources":["../../../../src/network/transports/streamed.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AAEjC,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAA;AAEpD;;;;;GAKG;AACH,MAAM,OAAgB,aAAc,SAAQ,YAAY;IAAxD;;QACc,YAAO,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;IA0BzC,CAAC;IAXG,IAAI,CAAC,IAAgB;QACjB,IAAI,CAAC,OAAO,GAAG,aAAa,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAA;QAElD,OAAO,IAAI,CAAC,gBAAgB,EAAE,EAAE;YAC5B,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;gBAAE,MAAK;SACnC;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;IACpC,CAAC;CACJ","sourcesContent":["import EventEmitter from 'events'\n\nimport { concatBuffers } from '../../utils/index.js'\n\n/**\n * Base for streamed codecs.\n *\n * Streamed means that MTProto packet can be divided into\n * multiple transport packets.\n */\nexport abstract class StreamedCodec extends EventEmitter {\n    protected _stream = new Uint8Array(0)\n\n    /**\n     * Should return whether a full packet is available\n     * in `_stream` buffer\n     */\n    protected abstract _packetAvailable(): boolean\n\n    /**\n     * Handle a single (!) packet from `_stream` and emit `packet` or `error`.\n     *\n     * Should return true if there are more packets available, false otherwise\n     */\n    protected abstract _handlePacket(): boolean\n\n    feed(data: Uint8Array): void {\n        this._stream = concatBuffers([this._stream, data])\n\n        while (this._packetAvailable()) {\n            if (!this._handlePacket()) break\n        }\n    }\n\n    reset(): void {\n        this._stream = new Uint8Array(0)\n    }\n}\n"]}

package/esm/network/transports/tcp.d.ts

@@ -0,0 +1,33 @@
+/// <reference types="node" resolution-mode="require"/>
+/// <reference types="node" resolution-mode="require"/>
+import EventEmitter from 'events';
+import { Socket } from 'net';
+import { tl } from '@mtcute/tl';
+import { ICryptoProvider, Logger } from '../../utils/index.js';
+import { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js';
+import { IntermediatePacketCodec } from './intermediate.js';
+/**
+ * Base for TCP transports.
+ * Subclasses must provide packet codec in `_packetCodec` property
+ */
+export declare abstract class BaseTcpTransport extends EventEmitter implements ITelegramTransport {
+    protected _currentDc: tl.RawDcOption | null;
+    protected _state: TransportState;
+    protected _socket: Socket | null;
+    abstract _packetCodec: IPacketCodec;
+    protected _crypto: ICryptoProvider;
+    protected log: Logger;
+    packetCodecInitialized: boolean;
+    private _updateLogPrefix;
+    setup(crypto: ICryptoProvider, log: Logger): void;
+    state(): TransportState;
+    currentDc(): tl.RawDcOption | null;
+    connect(dc: tl.RawDcOption, testMode: boolean): void;
+    close(): void;
+    handleError(error: Error): void;
+    handleConnect(): void;
+    send(bytes: Uint8Array): Promise<void>;
+}
+export declare class TcpTransport extends BaseTcpTransport {
+    _packetCodec: IntermediatePacketCodec;
+}

package/esm/network/transports/tcp.js

@@ -0,0 +1,122 @@
+import EventEmitter from 'events';
+import { connect } from 'net';
+import { MtcuteError } from '../../types/errors.js';
+import { TransportState } from './abstract.js';
+import { IntermediatePacketCodec } from './intermediate.js';
+/**
+ * Base for TCP transports.
+ * Subclasses must provide packet codec in `_packetCodec` property
+ */
+export class BaseTcpTransport extends EventEmitter {
+    constructor() {
+        super(...arguments);
+        this._currentDc = null;
+        this._state = TransportState.Idle;
+        this._socket = null;
+        this.packetCodecInitialized = false;
+    }
+    _updateLogPrefix() {
+        if (this._currentDc) {
+            this.log.prefix = `[TCP:${this._currentDc.ipAddress}:${this._currentDc.port}] `;
+        }
+        else {
+            this.log.prefix = '[TCP:disconnected] ';
+        }
+    }
+    setup(crypto, log) {
+        this._crypto = crypto;
+        this.log = log.create('tcp');
+        this._updateLogPrefix();
+    }
+    state() {
+        return this._state;
+    }
+    currentDc() {
+        return this._currentDc;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    connect(dc, testMode) {
+        if (this._state !== TransportState.Idle) {
+            throw new MtcuteError('Transport is not IDLE');
+        }
+        if (!this.packetCodecInitialized) {
+            this._packetCodec.setup?.(this._crypto, this.log);
+            this._packetCodec.on('error', (err) => this.emit('error', err));
+            this._packetCodec.on('packet', (buf) => this.emit('message', buf));
+            this.packetCodecInitialized = true;
+        }
+        this._state = TransportState.Connecting;
+        this._currentDc = dc;
+        this._updateLogPrefix();
+        this.log.debug('connecting to %j', dc);
+        this._socket = connect(dc.port, dc.ipAddress, this.handleConnect.bind(this));
+        this._socket.on('data', (data) => {
+            this._packetCodec.feed(data);
+        });
+        this._socket.on('error', this.handleError.bind(this));
+        this._socket.on('close', this.close.bind(this));
+    }
+    close() {
+        if (this._state === TransportState.Idle)
+            return;
+        this.log.info('connection closed');
+        this.emit('close');
+        this._state = TransportState.Idle;
+        this._socket.removeAllListeners();
+        this._socket.destroy();
+        this._socket = null;
+        this._currentDc = null;
+        this._packetCodec.reset();
+    }
+    handleError(error) {
+        this.log.error('error: %s', error.stack);
+        this.emit('error', error);
+    }
+    handleConnect() {
+        this.log.info('connected');
+        Promise.resolve(this._packetCodec.tag())
+            .then((initialMessage) => {
+            if (initialMessage.length) {
+                this._socket.write(initialMessage, (err) => {
+                    if (err) {
+                        this.log.error('failed to write initial message: %s', err.stack);
+                        this.emit('error');
+                        this.close();
+                    }
+                    else {
+                        this._state = TransportState.Ready;
+                        this.emit('ready');
+                    }
+                });
+            }
+            else {
+                this._state = TransportState.Ready;
+                this.emit('ready');
+            }
+        })
+            .catch((err) => this.emit('error', err));
+    }
+    async send(bytes) {
+        if (this._state !== TransportState.Ready) {
+            throw new MtcuteError('Transport is not READY');
+        }
+        const framed = await this._packetCodec.encode(bytes);
+        return new Promise((resolve, reject) => {
+            this._socket.write(framed, (error) => {
+                if (error) {
+                    reject(error);
+                }
+                else {
+                    resolve();
+                }
+            });
+        });
+    }
+}
+export class TcpTransport extends BaseTcpTransport {
+    constructor() {
+        super(...arguments);
+        this._packetCodec = new IntermediatePacketCodec();
+    }
+}
+//# sourceMappingURL=tcp.js.map

package/esm/network/transports/tcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tcp.js","sourceRoot":"","sources":["../../../../src/network/transports/tcp.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AACjC,OAAO,EAAE,OAAO,EAAU,MAAM,KAAK,CAAA;AAIrC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAA;AAEnD,OAAO,EAAoC,cAAc,EAAE,MAAM,eAAe,CAAA;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAE3D;;;GAGG;AACH,MAAM,OAAgB,gBAAiB,SAAQ,YAAY;IAA3D;;QACc,eAAU,GAA0B,IAAI,CAAA;QACxC,WAAM,GAAmB,cAAc,CAAC,IAAI,CAAA;QAC5C,YAAO,GAAkB,IAAI,CAAA;QAMvC,2BAAsB,GAAG,KAAK,CAAA;IA+GlC,CAAC;IA7GW,gBAAgB;QACpB,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,QAAQ,IAAI,CAAC,UAAU,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAA;SAClF;aAAM;YACH,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,qBAAqB,CAAA;SAC1C;IACL,CAAC;IAED,KAAK,CAAC,MAAuB,EAAE,GAAW;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC5B,IAAI,CAAC,gBAAgB,EAAE,CAAA;IAC3B,CAAC;IAED,KAAK;QACD,OAAO,IAAI,CAAC,MAAM,CAAA;IACtB,CAAC;IAED,SAAS;QACL,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;IAED,6DAA6D;IAC7D,OAAO,CAAC,EAAkB,EAAE,QAAiB;QACzC,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,WAAW,CAAC,uBAAuB,CAAC,CAAA;SACjD;QAED,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAA;YACjD,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;YAC/D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAA;YAClE,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAA;SACrC;QAED,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,UAAU,CAAA;QACvC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QACpB,IAAI,CAAC,gBAAgB,EAAE,CAAA;QAEvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAA;QAEtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAE5E,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACrD,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IACnD,CAAC;IAED,KAAK;QACD,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI;YAAE,OAAM;QAC/C,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAElC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAClB,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,IAAI,CAAA;QACjC,IAAI,CAAC,OAAQ,CAAC,kBAAkB,EAAE,CAAA;QAClC,IAAI,CAAC,OAAQ,CAAC,OAAO,EAAE,CAAA;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;QACtB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;IAC7B,CAAC;IAED,WAAW,CAAC,KAAY;QACpB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;QACxC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IAC7B,CAAC;IAED,aAAa;QACT,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAE1B,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;aACnC,IAAI,CAAC,CAAC,cAAc,EAAE,EAAE;YACrB,IAAI,cAAc,CAAC,MAAM,EAAE;gBACvB,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,GAAG,EAAE,EAAE;oBACxC,IAAI,GAAG,EAAE;wBACL,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;wBAChE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;wBAClB,IAAI,CAAC,KAAK,EAAE,CAAA;qBACf;yBAAM;wBACH,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;wBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;qBACrB;gBACL,CAAC,CAAC,CAAA;aACL;iBAAM;gBACH,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;gBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;aACrB;QACL,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;IAChD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAiB;QACxB,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,WAAW,CAAC,wBAAwB,CAAC,CAAA;SAClD;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEpD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBAClC,IAAI,KAAK,EAAE;oBACP,MAAM,CAAC,KAAK,CAAC,CAAA;iBAChB;qBAAM;oBACH,OAAO,EAAE,CAAA;iBACZ;YACL,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC;CACJ;AAED,MAAM,OAAO,YAAa,SAAQ,gBAAgB;IAAlD;;QACI,iBAAY,GAAG,IAAI,uBAAuB,EAAE,CAAA;IAChD,CAAC;CAAA","sourcesContent":["import EventEmitter from 'events'\nimport { connect, Socket } from 'net'\n\nimport { tl } from '@mtcute/tl'\n\nimport { MtcuteError } from '../../types/errors.js'\nimport { ICryptoProvider, Logger } from '../../utils/index.js'\nimport { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js'\nimport { IntermediatePacketCodec } from './intermediate.js'\n\n/**\n * Base for TCP transports.\n * Subclasses must provide packet codec in `_packetCodec` property\n */\nexport abstract class BaseTcpTransport extends EventEmitter implements ITelegramTransport {\n    protected _currentDc: tl.RawDcOption | null = null\n    protected _state: TransportState = TransportState.Idle\n    protected _socket: Socket | null = null\n\n    abstract _packetCodec: IPacketCodec\n    protected _crypto!: ICryptoProvider\n    protected log!: Logger\n\n    packetCodecInitialized = false\n\n    private _updateLogPrefix() {\n        if (this._currentDc) {\n            this.log.prefix = `[TCP:${this._currentDc.ipAddress}:${this._currentDc.port}] `\n        } else {\n            this.log.prefix = '[TCP:disconnected] '\n        }\n    }\n\n    setup(crypto: ICryptoProvider, log: Logger): void {\n        this._crypto = crypto\n        this.log = log.create('tcp')\n        this._updateLogPrefix()\n    }\n\n    state(): TransportState {\n        return this._state\n    }\n\n    currentDc(): tl.RawDcOption | null {\n        return this._currentDc\n    }\n\n    // eslint-disable-next-line @typescript-eslint/no-unused-vars\n    connect(dc: tl.RawDcOption, testMode: boolean): void {\n        if (this._state !== TransportState.Idle) {\n            throw new MtcuteError('Transport is not IDLE')\n        }\n\n        if (!this.packetCodecInitialized) {\n            this._packetCodec.setup?.(this._crypto, this.log)\n            this._packetCodec.on('error', (err) => this.emit('error', err))\n            this._packetCodec.on('packet', (buf) => this.emit('message', buf))\n            this.packetCodecInitialized = true\n        }\n\n        this._state = TransportState.Connecting\n        this._currentDc = dc\n        this._updateLogPrefix()\n\n        this.log.debug('connecting to %j', dc)\n\n        this._socket = connect(dc.port, dc.ipAddress, this.handleConnect.bind(this))\n\n        this._socket.on('data', (data) => {\n            this._packetCodec.feed(data)\n        })\n        this._socket.on('error', this.handleError.bind(this))\n        this._socket.on('close', this.close.bind(this))\n    }\n\n    close(): void {\n        if (this._state === TransportState.Idle) return\n        this.log.info('connection closed')\n\n        this.emit('close')\n        this._state = TransportState.Idle\n        this._socket!.removeAllListeners()\n        this._socket!.destroy()\n        this._socket = null\n        this._currentDc = null\n        this._packetCodec.reset()\n    }\n\n    handleError(error: Error): void {\n        this.log.error('error: %s', error.stack)\n        this.emit('error', error)\n    }\n\n    handleConnect(): void {\n        this.log.info('connected')\n\n        Promise.resolve(this._packetCodec.tag())\n            .then((initialMessage) => {\n                if (initialMessage.length) {\n                    this._socket!.write(initialMessage, (err) => {\n                        if (err) {\n                            this.log.error('failed to write initial message: %s', err.stack)\n                            this.emit('error')\n                            this.close()\n                        } else {\n                            this._state = TransportState.Ready\n                            this.emit('ready')\n                        }\n                    })\n                } else {\n                    this._state = TransportState.Ready\n                    this.emit('ready')\n                }\n            })\n            .catch((err) => this.emit('error', err))\n    }\n\n    async send(bytes: Uint8Array): Promise<void> {\n        if (this._state !== TransportState.Ready) {\n            throw new MtcuteError('Transport is not READY')\n        }\n\n        const framed = await this._packetCodec.encode(bytes)\n\n        return new Promise((resolve, reject) => {\n            this._socket!.write(framed, (error) => {\n                if (error) {\n                    reject(error)\n                } else {\n                    resolve()\n                }\n            })\n        })\n    }\n}\n\nexport class TcpTransport extends BaseTcpTransport {\n    _packetCodec = new IntermediatePacketCodec()\n}\n"]}

package/esm/network/transports/websocket.d.ts

@@ -0,0 +1,47 @@
+/// <reference types="node" resolution-mode="require"/>
+import EventEmitter from 'events';
+import { tl } from '@mtcute/tl';
+import { ICryptoProvider, Logger } from '../../utils/index.js';
+import { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js';
+import { ObfuscatedPacketCodec } from './obfuscated.js';
+export type WebSocketConstructor = {
+    new (address: string, protocol?: string): WebSocket;
+};
+/**
+ * Base for WebSocket transports.
+ * Subclasses must provide packet codec in `_packetCodec` property
+ */
+export declare abstract class BaseWebSocketTransport extends EventEmitter implements ITelegramTransport {
+    private _currentDc;
+    private _state;
+    private _socket;
+    private _crypto;
+    protected log: Logger;
+    abstract _packetCodec: IPacketCodec;
+    packetCodecInitialized: boolean;
+    private _baseDomain;
+    private _subdomains;
+    private _WebSocket;
+    constructor({ ws, baseDomain, subdomains, }?: {
+        /** Custom implementation of WebSocket (e.g. https://npm.im/ws) */
+        ws?: WebSocketConstructor;
+        /** Base WebSocket domain */
+        baseDomain?: string;
+        /** Map of sub-domains (key is DC ID, value is string) */
+        subdomains?: Record<string, string>;
+    });
+    private _updateLogPrefix;
+    setup(crypto: ICryptoProvider, log: Logger): void;
+    state(): TransportState;
+    currentDc(): tl.RawDcOption | null;
+    connect(dc: tl.RawDcOption, testMode: boolean): void;
+    close(): void;
+    handleError(event: Event | {
+        error: Error;
+    }): void;
+    handleConnect(): void;
+    send(bytes: Uint8Array): Promise<void>;
+}
+export declare class WebSocketTransport extends BaseWebSocketTransport {
+    _packetCodec: ObfuscatedPacketCodec;
+}

package/esm/network/transports/websocket.js

@@ -0,0 +1,116 @@
+import EventEmitter from 'events';
+import { MtcuteError, MtUnsupportedError } from '../../types/errors.js';
+import { TransportState } from './abstract.js';
+import { IntermediatePacketCodec } from './intermediate.js';
+import { ObfuscatedPacketCodec } from './obfuscated.js';
+const subdomainsMap = {
+    '1': 'pluto',
+    '2': 'venus',
+    '3': 'aurora',
+    '4': 'vesta',
+    '5': 'flora',
+};
+/**
+ * Base for WebSocket transports.
+ * Subclasses must provide packet codec in `_packetCodec` property
+ */
+export class BaseWebSocketTransport extends EventEmitter {
+    constructor({ ws = WebSocket, baseDomain = 'web.telegram.org', subdomains = subdomainsMap, } = {}) {
+        super();
+        this._currentDc = null;
+        this._state = TransportState.Idle;
+        this._socket = null;
+        this.packetCodecInitialized = false;
+        if (!ws) {
+            throw new MtUnsupportedError('To use WebSocket transport with NodeJS, install `ws` package and pass it to constructor');
+        }
+        // gotta love cjs/esm compat
+        if ('default' in ws) {
+            ws = ws.default;
+        }
+        this._baseDomain = baseDomain;
+        this._subdomains = subdomains;
+        this._WebSocket = ws;
+        this.close = this.close.bind(this);
+    }
+    _updateLogPrefix() {
+        if (this._currentDc) {
+            this.log.prefix = `[WS:${this._subdomains[this._currentDc.id]}.${this._baseDomain}] `;
+        }
+        else {
+            this.log.prefix = '[WS:disconnected] ';
+        }
+    }
+    setup(crypto, log) {
+        this._crypto = crypto;
+        this.log = log.create('ws');
+    }
+    state() {
+        return this._state;
+    }
+    currentDc() {
+        return this._currentDc;
+    }
+    connect(dc, testMode) {
+        if (this._state !== TransportState.Idle) {
+            throw new MtcuteError('Transport is not IDLE');
+        }
+        if (!this.packetCodecInitialized) {
+            this._packetCodec.setup?.(this._crypto, this.log);
+            this._packetCodec.on('error', (err) => this.emit('error', err));
+            this._packetCodec.on('packet', (buf) => this.emit('message', buf));
+            this.packetCodecInitialized = true;
+        }
+        this._state = TransportState.Connecting;
+        this._currentDc = dc;
+        this._socket = new this._WebSocket(`wss://${this._subdomains[dc.id]}.${this._baseDomain}/apiws${testMode ? '_test' : ''}`, 'binary');
+        this._updateLogPrefix();
+        this.log.debug('connecting to %s (%j)', this._socket.url, dc);
+        this._socket.binaryType = 'arraybuffer';
+        this._socket.addEventListener('message', (evt) => this._packetCodec.feed(new Uint8Array(evt.data)));
+        this._socket.addEventListener('open', this.handleConnect.bind(this));
+        this._socket.addEventListener('error', this.handleError.bind(this));
+        this._socket.addEventListener('close', this.close);
+    }
+    close() {
+        if (this._state === TransportState.Idle)
+            return;
+        this.log.info('close');
+        this.emit('connection closed');
+        this._state = TransportState.Idle;
+        this._socket.removeEventListener('close', this.close);
+        this._socket.close();
+        this._socket = null;
+        this._currentDc = null;
+        this._packetCodec.reset();
+    }
+    handleError(event) {
+        const error = 'error' in event ? event.error : new Error('unknown WebSocket error');
+        this.log.error('error: %s', error.stack);
+        this.emit('error', error);
+    }
+    handleConnect() {
+        this.log.info('connected');
+        Promise.resolve(this._packetCodec.tag())
+            .then((initialMessage) => {
+            this._socket.send(initialMessage);
+            this._state = TransportState.Ready;
+            this.emit('ready');
+        })
+            .catch((err) => this.emit('error', err));
+    }
+    async send(bytes) {
+        if (this._state !== TransportState.Ready) {
+            throw new MtcuteError('Transport is not READY');
+        }
+        const framed = await this._packetCodec.encode(bytes);
+        this._socket.send(framed);
+    }
+}
+export class WebSocketTransport extends BaseWebSocketTransport {
+    constructor() {
+        super(...arguments);
+        this._packetCodec = new ObfuscatedPacketCodec(new IntermediatePacketCodec());
+    }
+}
+//# sourceMappingURL=websocket.js.map

package/esm/network/transports/websocket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket.js","sourceRoot":"","sources":["../../../../src/network/transports/websocket.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AAIjC,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAA;AAEvE,OAAO,EAAoC,cAAc,EAAE,MAAM,eAAe,CAAA;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAA;AAMvD,MAAM,aAAa,GAA2B;IAC1C,GAAG,EAAE,OAAO;IACZ,GAAG,EAAE,OAAO;IACZ,GAAG,EAAE,QAAQ;IACb,GAAG,EAAE,OAAO;IACZ,GAAG,EAAE,OAAO;CACf,CAAA;AAED;;;GAGG;AACH,MAAM,OAAgB,sBAAuB,SAAQ,YAAY;IAc7D,YAAY,EACR,EAAE,GAAG,SAAS,EACd,UAAU,GAAG,kBAAkB,EAC/B,UAAU,GAAG,aAAa,MAQ1B,EAAE;QACF,KAAK,EAAE,CAAA;QAzBH,eAAU,GAA0B,IAAI,CAAA;QACxC,WAAM,GAAmB,cAAc,CAAC,IAAI,CAAA;QAC5C,YAAO,GAAqB,IAAI,CAAA;QAKxC,2BAAsB,GAAG,KAAK,CAAA;QAoB1B,IAAI,CAAC,EAAE,EAAE;YACL,MAAM,IAAI,kBAAkB,CACxB,yFAAyF,CAC5F,CAAA;SACJ;QAED,4BAA4B;QAC5B,IAAI,SAAS,IAAI,EAAE,EAAE;YACjB,EAAE,GAAG,EAAE,CAAC,OAA+B,CAAA;SAC1C;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAA;QAC7B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAA;QAC7B,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QAEpB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACtC,CAAC;IAEO,gBAAgB;QACpB,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,WAAW,IAAI,CAAA;SACxF;aAAM;YACH,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,oBAAoB,CAAA;SACzC;IACL,CAAC;IAED,KAAK,CAAC,MAAuB,EAAE,GAAW;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAC/B,CAAC;IAED,KAAK;QACD,OAAO,IAAI,CAAC,MAAM,CAAA;IACtB,CAAC;IAED,SAAS;QACL,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;IAED,OAAO,CAAC,EAAkB,EAAE,QAAiB;QACzC,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,WAAW,CAAC,uBAAuB,CAAC,CAAA;SACjD;QAED,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAA;YACjD,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;YAC/D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAA;YAClE,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAA;SACrC;QAED,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,UAAU,CAAA;QACvC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QACpB,IAAI,CAAC,OAAO,GAAG,IAAI,IAAI,CAAC,UAAU,CAC9B,SAAS,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,WAAW,SAAS,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,EACtF,QAAQ,CACX,CAAA;QAED,IAAI,CAAC,gBAAgB,EAAE,CAAA;QACvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,uBAAuB,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QAE7D,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,aAAa,CAAA;QAEvC,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAC7C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,IAAmB,CAAC,CAAC,CAClE,CAAA;QACD,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACpE,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACnE,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;IACtD,CAAC;IAED,KAAK;QACD,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI;YAAE,OAAM;QAC/C,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAEtB,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAC9B,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,IAAI,CAAA;QACjC,IAAI,CAAC,OAAQ,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;QACtD,IAAI,CAAC,OAAQ,CAAC,KAAK,EAAE,CAAA;QACrB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;QACtB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;IAC7B,CAAC;IAED,WAAW,CAAC,KAA+B;QACvC,MAAM,KAAK,GAAG,OAAO,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;QAEnF,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;QACxC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IAC7B,CAAC;IAED,aAAa;QACT,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAE1B,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;aACnC,IAAI,CAAC,CAAC,cAAc,EAAE,EAAE;YACrB,IAAI,CAAC,OAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;YAClC,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;YAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACtB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;IAChD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAiB;QACxB,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,WAAW,CAAC,wBAAwB,CAAC,CAAA;SAClD;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEpD,IAAI,CAAC,OAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAC9B,CAAC;CACJ;AAED,MAAM,OAAO,kBAAmB,SAAQ,sBAAsB;IAA9D;;QACI,iBAAY,GAAG,IAAI,qBAAqB,CAAC,IAAI,uBAAuB,EAAE,CAAC,CAAA;IAC3E,CAAC;CAAA","sourcesContent":["import EventEmitter from 'events'\n\nimport { tl } from '@mtcute/tl'\n\nimport { MtcuteError, MtUnsupportedError } from '../../types/errors.js'\nimport { ICryptoProvider, Logger } from '../../utils/index.js'\nimport { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js'\nimport { IntermediatePacketCodec } from './intermediate.js'\nimport { ObfuscatedPacketCodec } from './obfuscated.js'\n\nexport type WebSocketConstructor = {\n    new (address: string, protocol?: string): WebSocket\n}\n\nconst subdomainsMap: Record<string, string> = {\n    '1': 'pluto',\n    '2': 'venus',\n    '3': 'aurora',\n    '4': 'vesta',\n    '5': 'flora',\n}\n\n/**\n * Base for WebSocket transports.\n * Subclasses must provide packet codec in `_packetCodec` property\n */\nexport abstract class BaseWebSocketTransport extends EventEmitter implements ITelegramTransport {\n    private _currentDc: tl.RawDcOption | null = null\n    private _state: TransportState = TransportState.Idle\n    private _socket: WebSocket | null = null\n    private _crypto!: ICryptoProvider\n    protected log!: Logger\n\n    abstract _packetCodec: IPacketCodec\n    packetCodecInitialized = false\n\n    private _baseDomain: string\n    private _subdomains: Record<string, string>\n    private _WebSocket: WebSocketConstructor\n\n    constructor({\n        ws = WebSocket,\n        baseDomain = 'web.telegram.org',\n        subdomains = subdomainsMap,\n    }: {\n        /** Custom implementation of WebSocket (e.g. https://npm.im/ws) */\n        ws?: WebSocketConstructor\n        /** Base WebSocket domain */\n        baseDomain?: string\n        /** Map of sub-domains (key is DC ID, value is string) */\n        subdomains?: Record<string, string>\n    } = {}) {\n        super()\n\n        if (!ws) {\n            throw new MtUnsupportedError(\n                'To use WebSocket transport with NodeJS, install `ws` package and pass it to constructor',\n            )\n        }\n\n        // gotta love cjs/esm compat\n        if ('default' in ws) {\n            ws = ws.default as WebSocketConstructor\n        }\n\n        this._baseDomain = baseDomain\n        this._subdomains = subdomains\n        this._WebSocket = ws\n\n        this.close = this.close.bind(this)\n    }\n\n    private _updateLogPrefix() {\n        if (this._currentDc) {\n            this.log.prefix = `[WS:${this._subdomains[this._currentDc.id]}.${this._baseDomain}] `\n        } else {\n            this.log.prefix = '[WS:disconnected] '\n        }\n    }\n\n    setup(crypto: ICryptoProvider, log: Logger): void {\n        this._crypto = crypto\n        this.log = log.create('ws')\n    }\n\n    state(): TransportState {\n        return this._state\n    }\n\n    currentDc(): tl.RawDcOption | null {\n        return this._currentDc\n    }\n\n    connect(dc: tl.RawDcOption, testMode: boolean): void {\n        if (this._state !== TransportState.Idle) {\n            throw new MtcuteError('Transport is not IDLE')\n        }\n\n        if (!this.packetCodecInitialized) {\n            this._packetCodec.setup?.(this._crypto, this.log)\n            this._packetCodec.on('error', (err) => this.emit('error', err))\n            this._packetCodec.on('packet', (buf) => this.emit('message', buf))\n            this.packetCodecInitialized = true\n        }\n\n        this._state = TransportState.Connecting\n        this._currentDc = dc\n        this._socket = new this._WebSocket(\n            `wss://${this._subdomains[dc.id]}.${this._baseDomain}/apiws${testMode ? '_test' : ''}`,\n            'binary',\n        )\n\n        this._updateLogPrefix()\n        this.log.debug('connecting to %s (%j)', this._socket.url, dc)\n\n        this._socket.binaryType = 'arraybuffer'\n\n        this._socket.addEventListener('message', (evt) =>\n            this._packetCodec.feed(new Uint8Array(evt.data as ArrayBuffer)),\n        )\n        this._socket.addEventListener('open', this.handleConnect.bind(this))\n        this._socket.addEventListener('error', this.handleError.bind(this))\n        this._socket.addEventListener('close', this.close)\n    }\n\n    close(): void {\n        if (this._state === TransportState.Idle) return\n        this.log.info('close')\n\n        this.emit('connection closed')\n        this._state = TransportState.Idle\n        this._socket!.removeEventListener('close', this.close)\n        this._socket!.close()\n        this._socket = null\n        this._currentDc = null\n        this._packetCodec.reset()\n    }\n\n    handleError(event: Event | { error: Error }): void {\n        const error = 'error' in event ? event.error : new Error('unknown WebSocket error')\n\n        this.log.error('error: %s', error.stack)\n        this.emit('error', error)\n    }\n\n    handleConnect(): void {\n        this.log.info('connected')\n\n        Promise.resolve(this._packetCodec.tag())\n            .then((initialMessage) => {\n                this._socket!.send(initialMessage)\n                this._state = TransportState.Ready\n                this.emit('ready')\n            })\n            .catch((err) => this.emit('error', err))\n    }\n\n    async send(bytes: Uint8Array): Promise<void> {\n        if (this._state !== TransportState.Ready) {\n            throw new MtcuteError('Transport is not READY')\n        }\n\n        const framed = await this._packetCodec.encode(bytes)\n\n        this._socket!.send(framed)\n    }\n}\n\nexport class WebSocketTransport extends BaseWebSocketTransport {\n    _packetCodec = new ObfuscatedPacketCodec(new IntermediatePacketCodec())\n}\n"]}

package/esm/network/transports/wrapped.d.ts

@@ -0,0 +1,11 @@
+/// <reference types="node" resolution-mode="require"/>
+import EventEmitter from 'events';
+import { ICryptoProvider, Logger } from '../../utils/index.js';
+import { IPacketCodec } from './abstract.js';
+export declare abstract class WrappedCodec extends EventEmitter {
+    protected _crypto: ICryptoProvider;
+    protected _inner: IPacketCodec;
+    constructor(inner: IPacketCodec);
+    removeAllListeners(): this;
+    setup(crypto: ICryptoProvider, log: Logger): void;
+}

package/esm/network/transports/wrapped.js

@@ -0,0 +1,19 @@
+import EventEmitter from 'events';
+export class WrappedCodec extends EventEmitter {
+    constructor(inner) {
+        super();
+        this._inner = inner;
+        this._inner.on('error', (err) => this.emit('error', err));
+        this._inner.on('packet', (buf) => this.emit('packet', buf));
+    }
+    removeAllListeners() {
+        super.removeAllListeners();
+        this._inner.removeAllListeners();
+        return this;
+    }
+    setup(crypto, log) {
+        this._crypto = crypto;
+        this._inner.setup?.(crypto, log);
+    }
+}
+//# sourceMappingURL=wrapped.js.map

package/esm/network/transports/wrapped.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapped.js","sourceRoot":"","sources":["../../../../src/network/transports/wrapped.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AAKjC,MAAM,OAAgB,YAAa,SAAQ,YAAY;IAInD,YAAY,KAAmB;QAC3B,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACnB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;QACzD,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAA;IAC/D,CAAC;IAED,kBAAkB;QACd,KAAK,CAAC,kBAAkB,EAAE,CAAA;QAC1B,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAA;QAEhC,OAAO,IAAI,CAAA;IACf,CAAC;IAED,KAAK,CAAC,MAAuB,EAAE,GAAW;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;IACpC,CAAC;CACJ","sourcesContent":["import EventEmitter from 'events'\n\nimport { ICryptoProvider, Logger } from '../../utils/index.js'\nimport { IPacketCodec } from './abstract.js'\n\nexport abstract class WrappedCodec extends EventEmitter {\n    protected _crypto!: ICryptoProvider\n    protected _inner: IPacketCodec\n\n    constructor(inner: IPacketCodec) {\n        super()\n        this._inner = inner\n        this._inner.on('error', (err) => this.emit('error', err))\n        this._inner.on('packet', (buf) => this.emit('packet', buf))\n    }\n\n    removeAllListeners(): this {\n        super.removeAllListeners()\n        this._inner.removeAllListeners()\n\n        return this\n    }\n\n    setup(crypto: ICryptoProvider, log: Logger): void {\n        this._crypto = crypto\n        this._inner.setup?.(crypto, log)\n    }\n}\n"]}