@mtcute/core 0.5.0 → 0.7.0

package/cjs/network/transports/abstract.d.ts

@@ -1,8 +1,8 @@
 /// <reference types="node" resolution-mode="require"/>
 import EventEmitter from 'events';
 import { tl } from '@mtcute/tl';
-import { MaybeAsync } from '../../types/index.js';
-import { ICryptoProvider, Logger } from '../../utils/index.js';
+import { MaybePromise } from '../../types/index.js';
+import { BasicDcOption, ICryptoProvider, Logger } from '../../utils/index.js';
 /** Current state of the transport */
 export declare enum TransportState {
     /**
@@ -37,12 +37,12 @@ export interface ITelegramTransport extends EventEmitter {
     /** returns current state */
     state(): TransportState;
     /** returns current DC. should return null if state == IDLE */
-    currentDc(): tl.RawDcOption | null;
+    currentDc(): BasicDcOption | null;
     /**
      * Start trying to connect to a specified DC.
      * Will throw an error if state != IDLE
      */
-    connect(dc: tl.RawDcOption, testMode: boolean): void;
+    connect(dc: BasicDcOption, testMode: boolean): void;
     /** call to close existing connection to some DC */
     close(): void;
     /** send a message */
@@ -65,9 +65,9 @@ export type TransportFactory = () => ITelegramTransport;
  */
 export interface IPacketCodec {
     /** Initial tag of the codec. Will be sent immediately once connected. */
-    tag(): MaybeAsync<Uint8Array>;
+    tag(): MaybePromise<Uint8Array>;
     /** Encodes and frames a single packet */
-    encode(packet: Uint8Array): MaybeAsync<Uint8Array>;
+    encode(packet: Uint8Array): MaybePromise<Uint8Array>;
     /** Feed packet to the codec. Once packet is processed, codec is supposed to emit `packet` or `error` */
     feed(data: Uint8Array): void;
     /** Reset codec state (for example, reset buffer) */

package/cjs/network/transports/abstract.js

@@ -21,7 +21,7 @@ var TransportState;
      * Transport has an active connection
      */
     TransportState["Ready"] = "ready";
-})(TransportState = exports.TransportState || (exports.TransportState = {}));
+})(TransportState || (exports.TransportState = TransportState = {}));
 /**
  * Error that represents an error in an underlying transport
  * (and usually means either server-side problems or bug in library)

package/cjs/network/transports/abstract.js.map

@@ -1 +1 @@
-{"version":3,"file":"abstract.js","sourceRoot":"","sources":["../../../../src/network/transports/abstract.ts"],"names":[],"mappings":";;;AAOA,qCAAqC;AACrC,IAAY,cAkBX;AAlBD,WAAY,cAAc;IACtB;;;;;;;OAOG;IACH,+BAAa,CAAA;IACb;;OAEG;IACH,2CAAyB,CAAA;IACzB;;OAEG;IACH,iCAAe,CAAA;AACnB,CAAC,EAlBW,cAAc,GAAd,sBAAc,KAAd,sBAAc,QAkBzB;AA6ED;;;;;GAKG;AACH,MAAa,cAAe,SAAQ,KAAK;IAGrC,YAAY,IAAY;QACpB,KAAK,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAA;QACjC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;IACpB,CAAC;CACJ;AAPD,wCAOC","sourcesContent":["import EventEmitter from 'events'\n\nimport { tl } from '@mtcute/tl'\n\nimport { MaybeAsync } from '../../types/index.js'\nimport { ICryptoProvider, Logger } from '../../utils/index.js'\n\n/** Current state of the transport */\nexport enum TransportState {\n    /**\n     * Transport has no active connections nor trying to connect to anything\n     * Can be a result of network failure, close() call, or connect() call was\n     * never called on this instance of Transport.\n     *\n     * This state is usually immediately handled by more higher-level components,\n     * thus rarely seen\n     */\n    Idle = 'idle',\n    /**\n     * Transport has no active connections, but is trying to connect to something\n     */\n    Connecting = 'connecting',\n    /**\n     * Transport has an active connection\n     */\n    Ready = 'ready',\n}\n\n/**\n * Interface implementing a transport to interact with Telegram servers.\n *\n * Events:\n *  - `ready` event is emitted once connection has been established: `() => void`\n *  - `close` event is emitted once connection has been closed: `() => void`\n *  - `error` event is event is emitted when there was some error\n *    (either mtproto related or network related): `(error: Error) => void`\n *  - `message` event is emitted when a mtproto message is received: `(message: Buffer) => void`\n */\nexport interface ITelegramTransport extends EventEmitter {\n    /** returns current state */\n    state(): TransportState\n    /** returns current DC. should return null if state == IDLE */\n    currentDc(): tl.RawDcOption | null\n\n    /**\n     * Start trying to connect to a specified DC.\n     * Will throw an error if state != IDLE\n     */\n    connect(dc: tl.RawDcOption, testMode: boolean): void\n    /** call to close existing connection to some DC */\n    close(): void\n    /** send a message */\n    send(data: Uint8Array): Promise<void>\n\n    /**\n     * Provides crypto and logging for the transport.\n     * Not done in constructor to simplify factory.\n     *\n     * This method is called before any other.\n     */\n    setup?(crypto: ICryptoProvider, log: Logger): void\n\n    getMtproxyInfo?(): tl.RawInputClientProxy\n}\n\n/** Transport factory function */\nexport type TransportFactory = () => ITelegramTransport\n\n/**\n * Interface declaring handling of received packets.\n * When receiving a packet, its content is sent to feed(),\n * and codec is supposed to emit `packet` or `error` event when packet is parsed.\n */\nexport interface IPacketCodec {\n    /** Initial tag of the codec. Will be sent immediately once connected. */\n    tag(): MaybeAsync<Uint8Array>\n\n    /** Encodes and frames a single packet */\n    encode(packet: Uint8Array): MaybeAsync<Uint8Array>\n\n    /** Feed packet to the codec. Once packet is processed, codec is supposed to emit `packet` or `error` */\n    feed(data: Uint8Array): void\n    /** Reset codec state (for example, reset buffer) */\n    reset(): void\n\n    /** Remove all listeners from a codec, used to safely dispose it */\n    removeAllListeners(): void\n\n    /**\n     * Emitted when a packet containing a\n     * (Transport error)[https://core.telegram.org/mtproto/mtproto-transports#transport-errors] is encountered.\n     */\n    on(event: 'error', handler: (error: Error) => void): void\n    /** Emitted when a full packet has been processed. */\n    on(event: 'packet', handler: (packet: Uint8Array) => void): void\n\n    /**\n     * For codecs that use crypto functions and/or logging.\n     * This method is called before any other.\n     */\n    setup?(crypto: ICryptoProvider, log: Logger): void\n}\n\n/**\n * Error that represents an error in an underlying transport\n * (and usually means either server-side problems or bug in library)\n *\n * More: https://core.telegram.org/mtproto/mtproto-transports#transport-errors\n */\nexport class TransportError extends Error {\n    code: number\n\n    constructor(code: number) {\n        super(`Transport error: ${code}`)\n        this.code = code\n    }\n}\n"]}
+{"version":3,"file":"abstract.js","sourceRoot":"","sources":["../../../../src/network/transports/abstract.ts"],"names":[],"mappings":";;;AAOA,qCAAqC;AACrC,IAAY,cAkBX;AAlBD,WAAY,cAAc;IACtB;;;;;;;OAOG;IACH,+BAAa,CAAA;IACb;;OAEG;IACH,2CAAyB,CAAA;IACzB;;OAEG;IACH,iCAAe,CAAA;AACnB,CAAC,EAlBW,cAAc,8BAAd,cAAc,QAkBzB;AA6ED;;;;;GAKG;AACH,MAAa,cAAe,SAAQ,KAAK;IAGrC,YAAY,IAAY;QACpB,KAAK,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAA;QACjC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;IACpB,CAAC;CACJ;AAPD,wCAOC","sourcesContent":["import EventEmitter from 'events'\n\nimport { tl } from '@mtcute/tl'\n\nimport { MaybePromise } from '../../types/index.js'\nimport { BasicDcOption, ICryptoProvider, Logger } from '../../utils/index.js'\n\n/** Current state of the transport */\nexport enum TransportState {\n    /**\n     * Transport has no active connections nor trying to connect to anything\n     * Can be a result of network failure, close() call, or connect() call was\n     * never called on this instance of Transport.\n     *\n     * This state is usually immediately handled by more higher-level components,\n     * thus rarely seen\n     */\n    Idle = 'idle',\n    /**\n     * Transport has no active connections, but is trying to connect to something\n     */\n    Connecting = 'connecting',\n    /**\n     * Transport has an active connection\n     */\n    Ready = 'ready',\n}\n\n/**\n * Interface implementing a transport to interact with Telegram servers.\n *\n * Events:\n *  - `ready` event is emitted once connection has been established: `() => void`\n *  - `close` event is emitted once connection has been closed: `() => void`\n *  - `error` event is event is emitted when there was some error\n *    (either mtproto related or network related): `(error: Error) => void`\n *  - `message` event is emitted when a mtproto message is received: `(message: Buffer) => void`\n */\nexport interface ITelegramTransport extends EventEmitter {\n    /** returns current state */\n    state(): TransportState\n    /** returns current DC. should return null if state == IDLE */\n    currentDc(): BasicDcOption | null\n\n    /**\n     * Start trying to connect to a specified DC.\n     * Will throw an error if state != IDLE\n     */\n    connect(dc: BasicDcOption, testMode: boolean): void\n    /** call to close existing connection to some DC */\n    close(): void\n    /** send a message */\n    send(data: Uint8Array): Promise<void>\n\n    /**\n     * Provides crypto and logging for the transport.\n     * Not done in constructor to simplify factory.\n     *\n     * This method is called before any other.\n     */\n    setup?(crypto: ICryptoProvider, log: Logger): void\n\n    getMtproxyInfo?(): tl.RawInputClientProxy\n}\n\n/** Transport factory function */\nexport type TransportFactory = () => ITelegramTransport\n\n/**\n * Interface declaring handling of received packets.\n * When receiving a packet, its content is sent to feed(),\n * and codec is supposed to emit `packet` or `error` event when packet is parsed.\n */\nexport interface IPacketCodec {\n    /** Initial tag of the codec. Will be sent immediately once connected. */\n    tag(): MaybePromise<Uint8Array>\n\n    /** Encodes and frames a single packet */\n    encode(packet: Uint8Array): MaybePromise<Uint8Array>\n\n    /** Feed packet to the codec. Once packet is processed, codec is supposed to emit `packet` or `error` */\n    feed(data: Uint8Array): void\n    /** Reset codec state (for example, reset buffer) */\n    reset(): void\n\n    /** Remove all listeners from a codec, used to safely dispose it */\n    removeAllListeners(): void\n\n    /**\n     * Emitted when a packet containing a\n     * (Transport error)[https://core.telegram.org/mtproto/mtproto-transports#transport-errors] is encountered.\n     */\n    on(event: 'error', handler: (error: Error) => void): void\n    /** Emitted when a full packet has been processed. */\n    on(event: 'packet', handler: (packet: Uint8Array) => void): void\n\n    /**\n     * For codecs that use crypto functions and/or logging.\n     * This method is called before any other.\n     */\n    setup?(crypto: ICryptoProvider, log: Logger): void\n}\n\n/**\n * Error that represents an error in an underlying transport\n * (and usually means either server-side problems or bug in library)\n *\n * More: https://core.telegram.org/mtproto/mtproto-transports#transport-errors\n */\nexport class TransportError extends Error {\n    code: number\n\n    constructor(code: number) {\n        super(`Transport error: ${code}`)\n        this.code = code\n    }\n}\n"]}

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

@@ -2,8 +2,7 @@
 /// <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 { BasicDcOption, ICryptoProvider, Logger } from '../../utils/index.js';
 import { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js';
 import { IntermediatePacketCodec } from './intermediate.js';
 /**
@@ -11,7 +10,7 @@ import { IntermediatePacketCodec } from './intermediate.js';
  * Subclasses must provide packet codec in `_packetCodec` property
  */
 export declare abstract class BaseTcpTransport extends EventEmitter implements ITelegramTransport {
-    protected _currentDc: tl.RawDcOption | null;
+    protected _currentDc: BasicDcOption | null;
     protected _state: TransportState;
     protected _socket: Socket | null;
     abstract _packetCodec: IPacketCodec;
@@ -21,8 +20,8 @@ export declare abstract class BaseTcpTransport extends EventEmitter implements I
     private _updateLogPrefix;
     setup(crypto: ICryptoProvider, log: Logger): void;
     state(): TransportState;
-    currentDc(): tl.RawDcOption | null;
-    connect(dc: tl.RawDcOption, testMode: boolean): void;
+    currentDc(): BasicDcOption | null;
+    connect(dc: BasicDcOption, testMode: boolean): void;
     close(): void;
     handleError(error: Error): void;
     handleConnect(): void;

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

@@ -1 +1 @@
-{"version":3,"file":"tcp.js","sourceRoot":"","sources":["../../../../src/network/transports/tcp.ts"],"names":[],"mappings":";;;;;;AAAA,oDAAiC;AACjC,6BAAqC;AAIrC,qDAAmD;AAEnD,+CAAgF;AAChF,uDAA2D;AAE3D;;;GAGG;AACH,MAAsB,gBAAiB,SAAQ,gBAAY;IAA3D;;QACc,eAAU,GAA0B,IAAI,CAAA;QACxC,WAAM,GAAmB,4BAAc,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,4BAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,uBAAW,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,4BAAc,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,IAAA,aAAO,EAAC,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,4BAAc,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,4BAAc,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,4BAAc,CAAC,KAAK,CAAA;wBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;qBACrB;gBACL,CAAC,CAAC,CAAA;aACL;iBAAM;gBACH,IAAI,CAAC,MAAM,GAAG,4BAAc,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,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEpD,IAAI,IAAI,CAAC,MAAM,KAAK,4BAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,uBAAW,CAAC,wBAAwB,CAAC,CAAA;SAClD;QAED,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;AAxHD,4CAwHC;AAED,MAAa,YAAa,SAAQ,gBAAgB;IAAlD;;QACI,iBAAY,GAAG,IAAI,yCAAuB,EAAE,CAAA;IAChD,CAAC;CAAA;AAFD,oCAEC","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        const framed = await this._packetCodec.encode(bytes)\n\n        if (this._state !== TransportState.Ready) {\n            throw new MtcuteError('Transport is not READY')\n        }\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"]}
+{"version":3,"file":"tcp.js","sourceRoot":"","sources":["../../../../src/network/transports/tcp.ts"],"names":[],"mappings":";;;;;;AAAA,oDAAiC;AACjC,6BAAqC;AAErC,qDAAmD;AAEnD,+CAAgF;AAChF,uDAA2D;AAE3D;;;GAGG;AACH,MAAsB,gBAAiB,SAAQ,gBAAY;IAA3D;;QACc,eAAU,GAAyB,IAAI,CAAA;QACvC,WAAM,GAAmB,4BAAc,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,EAAiB,EAAE,QAAiB;QACxC,IAAI,IAAI,CAAC,MAAM,KAAK,4BAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,uBAAW,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,4BAAc,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,IAAA,aAAO,EAAC,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,4BAAc,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,4BAAc,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,4BAAc,CAAC,KAAK,CAAA;wBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;qBACrB;gBACL,CAAC,CAAC,CAAA;aACL;iBAAM;gBACH,IAAI,CAAC,MAAM,GAAG,4BAAc,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,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEpD,IAAI,IAAI,CAAC,MAAM,KAAK,4BAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,uBAAW,CAAC,wBAAwB,CAAC,CAAA;SAClD;QAED,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;AAxHD,4CAwHC;AAED,MAAa,YAAa,SAAQ,gBAAgB;IAAlD;;QACI,iBAAY,GAAG,IAAI,yCAAuB,EAAE,CAAA;IAChD,CAAC;CAAA;AAFD,oCAEC","sourcesContent":["import EventEmitter from 'events'\nimport { connect, Socket } from 'net'\n\nimport { MtcuteError } from '../../types/errors.js'\nimport { BasicDcOption, 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: BasicDcOption | 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(): BasicDcOption | null {\n        return this._currentDc\n    }\n\n    // eslint-disable-next-line @typescript-eslint/no-unused-vars\n    connect(dc: BasicDcOption, 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        const framed = await this._packetCodec.encode(bytes)\n\n        if (this._state !== TransportState.Ready) {\n            throw new MtcuteError('Transport is not READY')\n        }\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/cjs/network/transports/websocket.d.ts

@@ -1,7 +1,6 @@
 /// <reference types="node" resolution-mode="require"/>
 import EventEmitter from 'events';
-import { tl } from '@mtcute/tl';
-import { ICryptoProvider, Logger } from '../../utils/index.js';
+import { BasicDcOption, ICryptoProvider, Logger } from '../../utils/index.js';
 import { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js';
 import { ObfuscatedPacketCodec } from './obfuscated.js';
 export type WebSocketConstructor = {
@@ -33,8 +32,8 @@ export declare abstract class BaseWebSocketTransport extends EventEmitter implem
     private _updateLogPrefix;
     setup(crypto: ICryptoProvider, log: Logger): void;
     state(): TransportState;
-    currentDc(): tl.RawDcOption | null;
-    connect(dc: tl.RawDcOption, testMode: boolean): void;
+    currentDc(): BasicDcOption | null;
+    connect(dc: BasicDcOption, testMode: boolean): void;
     close(): void;
     handleError(event: Event | {
         error: Error;

package/cjs/network/transports/websocket.js

@@ -81,8 +81,8 @@ class BaseWebSocketTransport extends events_1.default {
     close() {
         if (this._state === abstract_js_1.TransportState.Idle)
             return;
-        this.log.info('close');
-        this.emit('connection closed');
+        this.log.info('connection closed');
+        this.emit('close');
         this._state = abstract_js_1.TransportState.Idle;
         this._socket.removeEventListener('close', this.close);
         this._socket.close();

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

@@ -1 +1 @@
-{"version":3,"file":"websocket.js","sourceRoot":"","sources":["../../../../src/network/transports/websocket.ts"],"names":[],"mappings":";;;;;;AAAA,oDAAiC;AAIjC,qDAAuE;AAEvE,+CAAgF;AAChF,uDAA2D;AAC3D,mDAAuD;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,MAAsB,sBAAuB,SAAQ,gBAAY;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,4BAAc,CAAC,IAAI,CAAA;QAC5C,YAAO,GAAqB,IAAI,CAAA;QAKxC,2BAAsB,GAAG,KAAK,CAAA;QAoB1B,IAAI,CAAC,EAAE,EAAE;YACL,MAAM,IAAI,8BAAkB,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,4BAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,uBAAW,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,4BAAc,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,4BAAc,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,4BAAc,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,4BAAc,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,4BAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,uBAAW,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;AA5ID,wDA4IC;AAED,MAAa,kBAAmB,SAAQ,sBAAsB;IAA9D;;QACI,iBAAY,GAAG,IAAI,qCAAqB,CAAC,IAAI,yCAAuB,EAAE,CAAC,CAAA;IAC3E,CAAC;CAAA;AAFD,gDAEC","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"]}
+{"version":3,"file":"websocket.js","sourceRoot":"","sources":["../../../../src/network/transports/websocket.ts"],"names":[],"mappings":";;;;;;AAAA,oDAAiC;AAEjC,qDAAuE;AAEvE,+CAAgF;AAChF,uDAA2D;AAC3D,mDAAuD;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,MAAsB,sBAAuB,SAAQ,gBAAY;IAc7D,YAAY,EACR,EAAE,GAAG,SAAS,EACd,UAAU,GAAG,kBAAkB,EAC/B,UAAU,GAAG,aAAa,MAQ1B,EAAE;QACF,KAAK,EAAE,CAAA;QAzBH,eAAU,GAAyB,IAAI,CAAA;QACvC,WAAM,GAAmB,4BAAc,CAAC,IAAI,CAAA;QAC5C,YAAO,GAAqB,IAAI,CAAA;QAKxC,2BAAsB,GAAG,KAAK,CAAA;QAoB1B,IAAI,CAAC,EAAE,EAAE;YACL,MAAM,IAAI,8BAAkB,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,EAAiB,EAAE,QAAiB;QACxC,IAAI,IAAI,CAAC,MAAM,KAAK,4BAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,uBAAW,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,4BAAc,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,4BAAc,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,4BAAc,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,4BAAc,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,4BAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,uBAAW,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;AA5ID,wDA4IC;AAED,MAAa,kBAAmB,SAAQ,sBAAsB;IAA9D;;QACI,iBAAY,GAAG,IAAI,qCAAqB,CAAC,IAAI,yCAAuB,EAAE,CAAC,CAAA;IAC3E,CAAC;CAAA;AAFD,gDAEC","sourcesContent":["import EventEmitter from 'events'\n\nimport { MtcuteError, MtUnsupportedError } from '../../types/errors.js'\nimport { BasicDcOption, 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: BasicDcOption | 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(): BasicDcOption | null {\n        return this._currentDc\n    }\n\n    connect(dc: BasicDcOption, 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('connection closed')\n\n        this.emit('close')\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/cjs/storage/driver.d.ts

@@ -0,0 +1,56 @@
+import { MaybePromise } from '../types/utils.js';
+import { Logger } from '../utils/logger.js';
+/**
+ * Basic storage driver interface,
+ * describing the lifecycle of a storage driver
+ */
+export interface IStorageDriver {
+    /**
+     * Load session from some external storage.
+     * Should be used either to load data from file/network/etc
+     * to memory, or to open required connections to fetch data on demand
+     *
+     * May be called more than once, handle this with care
+     * (or use {@link BaseStorageDriver} that handles this for you)
+     */
+    load?(): MaybePromise<void>;
+    /**
+     * Save session to some external storage.
+     * Should be used to commit pending changes in the session.
+     * For example, saving session content to file/network/etc,
+     * or committing a database transaction
+     *
+     * It is safe to batch all changes and only commit them here,
+     * unless stated otherwise in the method description
+     */
+    save?(): MaybePromise<void>;
+    /**
+     * Cleanup session and release all used resources.
+     *
+     * May be called more than once, handle this with care
+     * (or use {@link BaseStorageDriver} that handles this for you)
+     */
+    destroy?(): MaybePromise<void>;
+    /**
+     * Setup the driver, passing the logger instance,
+     * in case your driver needs it
+     */
+    setup?(log: Logger): void;
+}
+/**
+ * Base storage driver class, implementing {@link IStorageDriver}
+ * and handling the lifecycle for you
+ */
+export declare abstract class BaseStorageDriver implements IStorageDriver {
+    abstract _load(): MaybePromise<void>;
+    abstract _destroy(): MaybePromise<void>;
+    abstract _save?(): MaybePromise<void>;
+    private _loadedTimes;
+    private _destroyed;
+    protected _log: Logger;
+    setup(log: Logger): void;
+    protected get loaded(): boolean;
+    load(): Promise<void>;
+    destroy(): Promise<void>;
+    save(): MaybePromise<void>;
+}

package/cjs/storage/driver.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseStorageDriver = void 0;
+/**
+ * Base storage driver class, implementing {@link IStorageDriver}
+ * and handling the lifecycle for you
+ */
+class BaseStorageDriver {
+    constructor() {
+        this._loadedTimes = 0;
+        this._destroyed = false;
+    }
+    setup(log) {
+        this._log = log;
+    }
+    get loaded() {
+        return this._loadedTimes > 0;
+    }
+    async load() {
+        if (this._loadedTimes === 0) {
+            await this._load();
+            this._destroyed = false;
+        }
+        this._loadedTimes++;
+    }
+    async destroy() {
+        if (this._destroyed) {
+            return;
+        }
+        this._loadedTimes--;
+        if (this._loadedTimes === 0) {
+            await this._destroy();
+            this._destroyed = true;
+        }
+    }
+    save() {
+        return this._save?.();
+    }
+}
+exports.BaseStorageDriver = BaseStorageDriver;
+//# sourceMappingURL=driver.js.map

package/cjs/storage/driver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"driver.js","sourceRoot":"","sources":["../../../src/storage/driver.ts"],"names":[],"mappings":";;;AA0CA;;;GAGG;AACH,MAAsB,iBAAiB;IAAvC;QAKY,iBAAY,GAAG,CAAC,CAAA;QAChB,eAAU,GAAG,KAAK,CAAA;IAqC9B,CAAC;IAjCG,KAAK,CAAC,GAAW;QACb,IAAI,CAAC,IAAI,GAAG,GAAG,CAAA;IACnB,CAAC;IAED,IAAc,MAAM;QAChB,OAAO,IAAI,CAAC,YAAY,GAAG,CAAC,CAAA;IAChC,CAAC;IAED,KAAK,CAAC,IAAI;QACN,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE;YACzB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAA;YAClB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAA;SAC1B;QAED,IAAI,CAAC,YAAY,EAAE,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,OAAO;QACT,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,OAAM;SACT;QAED,IAAI,CAAC,YAAY,EAAE,CAAA;QAEnB,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE;YACzB,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAA;YACrB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;SACzB;IACL,CAAC;IAED,IAAI;QACA,OAAO,IAAI,CAAC,KAAK,EAAE,EAAE,CAAA;IACzB,CAAC;CACJ;AA3CD,8CA2CC","sourcesContent":["import { MaybePromise } from '../types/utils.js'\nimport { Logger } from '../utils/logger.js'\n\n/**\n * Basic storage driver interface,\n * describing the lifecycle of a storage driver\n */\nexport interface IStorageDriver {\n    /**\n     * Load session from some external storage.\n     * Should be used either to load data from file/network/etc\n     * to memory, or to open required connections to fetch data on demand\n     *\n     * May be called more than once, handle this with care\n     * (or use {@link BaseStorageDriver} that handles this for you)\n     */\n    load?(): MaybePromise<void>\n    /**\n     * Save session to some external storage.\n     * Should be used to commit pending changes in the session.\n     * For example, saving session content to file/network/etc,\n     * or committing a database transaction\n     *\n     * It is safe to batch all changes and only commit them here,\n     * unless stated otherwise in the method description\n     */\n    save?(): MaybePromise<void>\n    /**\n     * Cleanup session and release all used resources.\n     *\n     * May be called more than once, handle this with care\n     * (or use {@link BaseStorageDriver} that handles this for you)\n     */\n    destroy?(): MaybePromise<void>\n\n    /**\n     * Setup the driver, passing the logger instance,\n     * in case your driver needs it\n     */\n    setup?(log: Logger): void\n}\n\n/**\n * Base storage driver class, implementing {@link IStorageDriver}\n * and handling the lifecycle for you\n */\nexport abstract class BaseStorageDriver implements IStorageDriver {\n    abstract _load(): MaybePromise<void>\n    abstract _destroy(): MaybePromise<void>\n    abstract _save?(): MaybePromise<void>\n\n    private _loadedTimes = 0\n    private _destroyed = false\n\n    protected _log!: Logger\n\n    setup(log: Logger): void {\n        this._log = log\n    }\n\n    protected get loaded(): boolean {\n        return this._loadedTimes > 0\n    }\n\n    async load(): Promise<void> {\n        if (this._loadedTimes === 0) {\n            await this._load()\n            this._destroyed = false\n        }\n\n        this._loadedTimes++\n    }\n\n    async destroy(): Promise<void> {\n        if (this._destroyed) {\n            return\n        }\n\n        this._loadedTimes--\n\n        if (this._loadedTimes === 0) {\n            await this._destroy()\n            this._destroyed = true\n        }\n    }\n\n    save(): MaybePromise<void> {\n        return this._save?.()\n    }\n}\n"]}

package/cjs/storage/index.d.ts

@@ -1,2 +1,6 @@
-export * from './abstract.js';
-export * from './memory.js';
+export * from './driver.js';
+export * from './provider.js';
+export * from './providers/idb/index.js';
+export * from './providers/memory/index.js';
+export * from './repository/index.js';
+export * from './storage.js';

package/cjs/storage/index.js

@@ -14,6 +14,10 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./abstract.js"), exports);
-__exportStar(require("./memory.js"), exports);
+__exportStar(require("./driver.js"), exports);
+__exportStar(require("./provider.js"), exports);
+__exportStar(require("./providers/idb/index.js"), exports);
+__exportStar(require("./providers/memory/index.js"), exports);
+__exportStar(require("./repository/index.js"), exports);
+__exportStar(require("./storage.js"), exports);
 //# sourceMappingURL=index.js.map

package/cjs/storage/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/storage/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA6B;AAC7B,8CAA2B","sourcesContent":["export * from './abstract.js'\nexport * from './memory.js'\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/storage/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA2B;AAC3B,gDAA6B;AAC7B,2DAAwC;AACxC,8DAA2C;AAC3C,wDAAqC;AACrC,+CAA4B","sourcesContent":["export * from './driver.js'\nexport * from './provider.js'\nexport * from './providers/idb/index.js'\nexport * from './providers/memory/index.js'\nexport * from './repository/index.js'\nexport * from './storage.js'\n"]}

package/cjs/storage/provider.d.ts

@@ -0,0 +1,10 @@
+import { IStorageDriver } from './driver.js';
+import { IAuthKeysRepository } from './repository/auth-keys.js';
+import { IKeyValueRepository } from './repository/key-value.js';
+export type IStorageProvider<T> = T & {
+    readonly driver: IStorageDriver;
+};
+export type IMtStorageProvider = IStorageProvider<{
+    readonly kv: IKeyValueRepository;
+    readonly authKeys: IAuthKeysRepository;
+}>;

package/cjs/storage/provider.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=provider.js.map

package/cjs/storage/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../src/storage/provider.ts"],"names":[],"mappings":"","sourcesContent":["import { IStorageDriver } from './driver.js'\nimport { IAuthKeysRepository } from './repository/auth-keys.js'\nimport { IKeyValueRepository } from './repository/key-value.js'\n\nexport type IStorageProvider<T> = T & {\n    readonly driver: IStorageDriver\n}\n\nexport type IMtStorageProvider = IStorageProvider<{\n    readonly kv: IKeyValueRepository\n    readonly authKeys: IAuthKeysRepository\n}>\n"]}

package/cjs/storage/providers/idb/driver.d.ts

@@ -0,0 +1,18 @@
+import { BaseStorageDriver } from '../../driver.js';
+export type PostMigrationFunction = (db: IDBDatabase) => Promise<void>;
+type MigrationFunction = (db: IDBDatabase) => void | PostMigrationFunction;
+export declare class IdbStorageDriver extends BaseStorageDriver {
+    readonly _dbName: string;
+    db: IDBDatabase;
+    constructor(_dbName: string);
+    private _pendingWrites;
+    private _pendingWritesOses;
+    private _migrations;
+    private _maxVersion;
+    registerMigration(repo: string, version: number, migration: MigrationFunction): void;
+    writeLater(os: string, obj: unknown): void;
+    _load(): Promise<void>;
+    _save(): Promise<void>;
+    _destroy(): void;
+}
+export {};

package/cjs/storage/providers/idb/driver.js

@@ -0,0 +1,138 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IdbStorageDriver = void 0;
+const errors_js_1 = require("../../../types/errors.js");
+const driver_js_1 = require("../../driver.js");
+const utils_js_1 = require("./utils.js");
+const REPO_VERSION_PREFIX = '__version:';
+class IdbStorageDriver extends driver_js_1.BaseStorageDriver {
+    constructor(_dbName) {
+        super();
+        this._dbName = _dbName;
+        this._pendingWrites = [];
+        this._pendingWritesOses = new Set();
+        this._migrations = new Map();
+        this._maxVersion = new Map();
+        if (typeof indexedDB === 'undefined') {
+            throw new errors_js_1.MtUnsupportedError('IndexedDB is not available');
+        }
+    }
+    registerMigration(repo, version, migration) {
+        if (this.loaded) {
+            throw new Error('Cannot register migrations after loading');
+        }
+        let map = this._migrations.get(repo);
+        if (!map) {
+            map = new Map();
+            this._migrations.set(repo, map);
+        }
+        if (map.has(version)) {
+            throw new Error(`Migration for ${repo} version ${version} is already registered`);
+        }
+        map.set(version, migration);
+        const prevMax = this._maxVersion.get(repo) ?? 0;
+        if (version > prevMax) {
+            this._maxVersion.set(repo, version);
+        }
+    }
+    writeLater(os, obj) {
+        this._pendingWrites.push([os, obj]);
+        this._pendingWritesOses.add(os);
+    }
+    async _load() {
+        this.db = await new Promise((resolve, reject) => {
+            // indexed db fucking sucks - we can't create tables once we have loaded
+            // and making an ever-incrementing version number is pretty hard
+            // since migrations are added dynamically.
+            //
+            // force the database to always emit `upgradeneeded` by passing current time
+            const req = indexedDB.open(this._dbName, Date.now());
+            req.onerror = () => reject(req.error);
+            const postUpgrade = [];
+            req.onsuccess = async () => {
+                try {
+                    for (const cb of postUpgrade) {
+                        await cb(req.result);
+                    }
+                    resolve(req.result);
+                }
+                catch (e) {
+                    reject(e);
+                }
+            };
+            req.onupgradeneeded = () => {
+                // indexed db still fucking sucks. we can't fetch anything from here,
+                // since migrations must be sync, and any fetch is inherently async
+                // what we do have, however, is the list of object stores.
+                // we can abuse them to store the current migrations status as plain strings
+                const db = req.result;
+                const didUpgrade = new Set();
+                const doUpgrade = (repo, fromVersion) => {
+                    const migrations = this._migrations.get(repo);
+                    if (!migrations)
+                        return;
+                    const targetVer = this._maxVersion.get(repo);
+                    while (fromVersion < targetVer) {
+                        const nextVersion = fromVersion + 1;
+                        const migration = migrations.get(nextVersion);
+                        if (!migration) {
+                            throw new Error(`No migration for ${repo} to version ${nextVersion}`);
+                        }
+                        const result = migration(db);
+                        if (result) {
+                            // guess what? IDB still. fucking. sucks!
+                            // if we want to do something except creating/removing
+                            // databases, we should do this outside of migration
+                            postUpgrade.push(result);
+                        }
+                        fromVersion = nextVersion;
+                    }
+                    didUpgrade.add(repo);
+                    db.createObjectStore(`${REPO_VERSION_PREFIX}${repo}:${targetVer}`);
+                };
+                for (const key of db.objectStoreNames) {
+                    if (!key.startsWith(REPO_VERSION_PREFIX))
+                        continue;
+                    const [, repo, version] = key.split(':');
+                    const currentVer = Number(version);
+                    db.deleteObjectStore(key);
+                    doUpgrade(repo, currentVer);
+                    didUpgrade.add(repo);
+                }
+                for (const repo of this._migrations.keys()) {
+                    if (didUpgrade.has(repo))
+                        continue;
+                    doUpgrade(repo, 0);
+                }
+            };
+        });
+    }
+    async _save() {
+        if (this._pendingWritesOses.size === 0)
+            return;
+        const writes = this._pendingWrites;
+        const oses = this._pendingWritesOses;
+        this._pendingWrites = [];
+        this._pendingWritesOses = new Set();
+        const tx = this.db.transaction(oses, 'readwrite');
+        const osMap = new Map();
+        for (const table of oses) {
+            osMap.set(table, tx.objectStore(table));
+        }
+        for (const [table, obj] of writes) {
+            const os = osMap.get(table);
+            if (obj === null) {
+                os.delete(table);
+            }
+            else {
+                os.put(obj);
+            }
+        }
+        await (0, utils_js_1.txToPromise)(tx);
+    }
+    _destroy() {
+        this.db.close();
+    }
+}
+exports.IdbStorageDriver = IdbStorageDriver;
+//# sourceMappingURL=driver.js.map

package/cjs/storage/providers/idb/driver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"driver.js","sourceRoot":"","sources":["../../../../../src/storage/providers/idb/driver.ts"],"names":[],"mappings":";;;AAAA,wDAA6D;AAC7D,+CAAmD;AACnD,yCAAwC;AAKxC,MAAM,mBAAmB,GAAG,YAAY,CAAA;AAExC,MAAa,gBAAiB,SAAQ,6BAAiB;IAGnD,YAAqB,OAAe;QAChC,KAAK,EAAE,CAAA;QADU,YAAO,GAAP,OAAO,CAAQ;QAQ5B,mBAAc,GAAwB,EAAE,CAAA;QACxC,uBAAkB,GAAG,IAAI,GAAG,EAAU,CAAA;QACtC,gBAAW,GAAgD,IAAI,GAAG,EAAE,CAAA;QACpE,gBAAW,GAAwB,IAAI,GAAG,EAAE,CAAA;QARhD,IAAI,OAAO,SAAS,KAAK,WAAW,EAAE;YAClC,MAAM,IAAI,8BAAkB,CAAC,4BAA4B,CAAC,CAAA;SAC7D;IACL,CAAC;IAOD,iBAAiB,CAAC,IAAY,EAAE,OAAe,EAAE,SAA4B;QACzE,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;SAC9D;QAED,IAAI,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QAEpC,IAAI,CAAC,GAAG,EAAE;YACN,GAAG,GAAG,IAAI,GAAG,EAAE,CAAA;YACf,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;SAClC;QAED,IAAI,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;YAClB,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,YAAY,OAAO,wBAAwB,CAAC,CAAA;SACpF;QAED,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,CAAC,CAAA;QAE3B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE/C,IAAI,OAAO,GAAG,OAAO,EAAE;YACnB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;SACtC;IACL,CAAC;IAED,UAAU,CAAC,EAAU,EAAE,GAAY;QAC/B,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAA;QACnC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IACnC,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,CAAC,EAAE,GAAG,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC5C,wEAAwE;YACxE,gEAAgE;YAChE,0CAA0C;YAC1C,EAAE;YACF,4EAA4E;YAC5E,MAAM,GAAG,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAA;YAEpD,GAAG,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;YAErC,MAAM,WAAW,GAA4B,EAAE,CAAA;YAE/C,GAAG,CAAC,SAAS,GAAG,KAAK,IAAI,EAAE;gBACvB,IAAI;oBACA,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE;wBAC1B,MAAM,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;qBACvB;oBACD,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;iBACtB;gBAAC,OAAO,CAAC,EAAE;oBACR,MAAM,CAAC,CAAC,CAAC,CAAA;iBACZ;YACL,CAAC,CAAA;YACD,GAAG,CAAC,eAAe,GAAG,GAAG,EAAE;gBACvB,qEAAqE;gBACrE,mEAAmE;gBACnE,0DAA0D;gBAC1D,4EAA4E;gBAC5E,MAAM,EAAE,GAAG,GAAG,CAAC,MAAM,CAAA;gBAErB,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAA;gBAEpC,MAAM,SAAS,GAAG,CAAC,IAAY,EAAE,WAAmB,EAAE,EAAE;oBACpD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;oBAC7C,IAAI,CAAC,UAAU;wBAAE,OAAM;oBAEvB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAE,CAAA;oBAE7C,OAAO,WAAW,GAAG,SAAS,EAAE;wBAC5B,MAAM,WAAW,GAAG,WAAW,GAAG,CAAC,CAAA;wBACnC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;wBAE7C,IAAI,CAAC,SAAS,EAAE;4BACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,eAAe,WAAW,EAAE,CAAC,CAAA;yBACxE;wBAED,MAAM,MAAM,GAAG,SAAS,CAAC,EAAE,CAAC,CAAA;wBAE5B,IAAI,MAAM,EAAE;4BACR,yCAAyC;4BACzC,sDAAsD;4BACtD,oDAAoD;4BACpD,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;yBAC3B;wBAED,WAAW,GAAG,WAAW,CAAA;qBAC5B;oBAED,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;oBACpB,EAAE,CAAC,iBAAiB,CAAC,GAAG,mBAAmB,GAAG,IAAI,IAAI,SAAS,EAAE,CAAC,CAAA;gBACtE,CAAC,CAAA;gBAED,KAAK,MAAM,GAAG,IAAI,EAAE,CAAC,gBAAgB,EAAE;oBACnC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,mBAAmB,CAAC;wBAAE,SAAQ;oBAClD,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;oBAExC,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAA;oBAClC,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAA;oBACzB,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;oBAC3B,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;iBACvB;gBAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE;oBACxC,IAAI,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC;wBAAE,SAAQ;oBAElC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;iBACrB;YACL,CAAC,CAAA;QACL,CAAC,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,IAAI,CAAC,kBAAkB,CAAC,IAAI,KAAK,CAAC;YAAE,OAAM;QAE9C,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAA;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,CAAA;QACpC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAA;QACxB,IAAI,CAAC,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAA;QAEnC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,CAAA;QAEjD,MAAM,KAAK,GAAG,IAAI,GAAG,EAA0B,CAAA;QAE/C,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE;YACtB,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAA;SAC1C;QAED,KAAK,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,MAAM,EAAE;YAC/B,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAE,CAAA;YAE5B,IAAI,GAAG,KAAK,IAAI,EAAE;gBACd,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;aACnB;iBAAM;gBACH,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;aACd;SACJ;QAED,MAAM,IAAA,sBAAW,EAAC,EAAE,CAAC,CAAA;IACzB,CAAC;IAED,QAAQ;QACJ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAA;IACnB,CAAC;CACJ;AA/JD,4CA+JC","sourcesContent":["import { MtUnsupportedError } from '../../../types/errors.js'\nimport { BaseStorageDriver } from '../../driver.js'\nimport { txToPromise } from './utils.js'\n\nexport type PostMigrationFunction = (db: IDBDatabase) => Promise<void>\ntype MigrationFunction = (db: IDBDatabase) => void | PostMigrationFunction\n\nconst REPO_VERSION_PREFIX = '__version:'\n\nexport class IdbStorageDriver extends BaseStorageDriver {\n    db!: IDBDatabase\n\n    constructor(readonly _dbName: string) {\n        super()\n\n        if (typeof indexedDB === 'undefined') {\n            throw new MtUnsupportedError('IndexedDB is not available')\n        }\n    }\n\n    private _pendingWrites: [string, unknown][] = []\n    private _pendingWritesOses = new Set<string>()\n    private _migrations: Map<string, Map<number, MigrationFunction>> = new Map()\n    private _maxVersion: Map<string, number> = new Map()\n\n    registerMigration(repo: string, version: number, migration: MigrationFunction): void {\n        if (this.loaded) {\n            throw new Error('Cannot register migrations after loading')\n        }\n\n        let map = this._migrations.get(repo)\n\n        if (!map) {\n            map = new Map()\n            this._migrations.set(repo, map)\n        }\n\n        if (map.has(version)) {\n            throw new Error(`Migration for ${repo} version ${version} is already registered`)\n        }\n\n        map.set(version, migration)\n\n        const prevMax = this._maxVersion.get(repo) ?? 0\n\n        if (version > prevMax) {\n            this._maxVersion.set(repo, version)\n        }\n    }\n\n    writeLater(os: string, obj: unknown): void {\n        this._pendingWrites.push([os, obj])\n        this._pendingWritesOses.add(os)\n    }\n\n    async _load(): Promise<void> {\n        this.db = await new Promise((resolve, reject) => {\n            // indexed db fucking sucks - we can't create tables once we have loaded\n            // and making an ever-incrementing version number is pretty hard\n            // since migrations are added dynamically.\n            //\n            // force the database to always emit `upgradeneeded` by passing current time\n            const req = indexedDB.open(this._dbName, Date.now())\n\n            req.onerror = () => reject(req.error)\n\n            const postUpgrade: PostMigrationFunction[] = []\n\n            req.onsuccess = async () => {\n                try {\n                    for (const cb of postUpgrade) {\n                        await cb(req.result)\n                    }\n                    resolve(req.result)\n                } catch (e) {\n                    reject(e)\n                }\n            }\n            req.onupgradeneeded = () => {\n                // indexed db still fucking sucks. we can't fetch anything from here,\n                // since migrations must be sync, and any fetch is inherently async\n                // what we do have, however, is the list of object stores.\n                // we can abuse them to store the current migrations status as plain strings\n                const db = req.result\n\n                const didUpgrade = new Set<string>()\n\n                const doUpgrade = (repo: string, fromVersion: number) => {\n                    const migrations = this._migrations.get(repo)\n                    if (!migrations) return\n\n                    const targetVer = this._maxVersion.get(repo)!\n\n                    while (fromVersion < targetVer) {\n                        const nextVersion = fromVersion + 1\n                        const migration = migrations.get(nextVersion)\n\n                        if (!migration) {\n                            throw new Error(`No migration for ${repo} to version ${nextVersion}`)\n                        }\n\n                        const result = migration(db)\n\n                        if (result) {\n                            // guess what? IDB still. fucking. sucks!\n                            // if we want to do something except creating/removing\n                            // databases, we should do this outside of migration\n                            postUpgrade.push(result)\n                        }\n\n                        fromVersion = nextVersion\n                    }\n\n                    didUpgrade.add(repo)\n                    db.createObjectStore(`${REPO_VERSION_PREFIX}${repo}:${targetVer}`)\n                }\n\n                for (const key of db.objectStoreNames) {\n                    if (!key.startsWith(REPO_VERSION_PREFIX)) continue\n                    const [, repo, version] = key.split(':')\n\n                    const currentVer = Number(version)\n                    db.deleteObjectStore(key)\n                    doUpgrade(repo, currentVer)\n                    didUpgrade.add(repo)\n                }\n\n                for (const repo of this._migrations.keys()) {\n                    if (didUpgrade.has(repo)) continue\n\n                    doUpgrade(repo, 0)\n                }\n            }\n        })\n    }\n\n    async _save() {\n        if (this._pendingWritesOses.size === 0) return\n\n        const writes = this._pendingWrites\n        const oses = this._pendingWritesOses\n        this._pendingWrites = []\n        this._pendingWritesOses = new Set()\n\n        const tx = this.db.transaction(oses, 'readwrite')\n\n        const osMap = new Map<string, IDBObjectStore>()\n\n        for (const table of oses) {\n            osMap.set(table, tx.objectStore(table))\n        }\n\n        for (const [table, obj] of writes) {\n            const os = osMap.get(table)!\n\n            if (obj === null) {\n                os.delete(table)\n            } else {\n                os.put(obj)\n            }\n        }\n\n        await txToPromise(tx)\n    }\n\n    _destroy(): void {\n        this.db.close()\n    }\n}\n"]}

package/cjs/storage/providers/idb/idb.test.d.ts

@@ -0,0 +1 @@
+export {};

package/cjs/storage/providers/idb/idb.test.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const test_1 = require("@mtcute/test");
+const index_js_1 = require("./index.js");
+if (import.meta.env.TEST_ENV === 'browser') {
+    (0, vitest_1.describe)('idb storage', () => {
+        const idbName = 'mtcute_test_' + Math.random().toString(36).slice(2);
+        const storage = new index_js_1.IdbStorage(idbName);
+        (0, vitest_1.beforeAll)(() => storage.driver.load());
+        (0, test_1.testAuthKeysRepository)(storage.authKeys);
+        (0, test_1.testKeyValueRepository)(storage.kv, storage.driver);
+        (0, test_1.testPeersRepository)(storage.peers, storage.driver);
+        (0, test_1.testRefMessagesRepository)(storage.refMessages, storage.driver);
+        (0, vitest_1.afterAll)(async () => {
+            await storage.driver.destroy();
+            const req = indexedDB.deleteDatabase(idbName);
+            await new Promise((resolve, reject) => {
+                req.onerror = () => reject(req.error);
+                req.onsuccess = () => resolve();
+                req.onblocked = () => resolve();
+            });
+        });
+    });
+}
+else {
+    vitest_1.describe.skip('idb storage', () => { });
+}
+//# sourceMappingURL=idb.test.js.map

package/cjs/storage/providers/idb/idb.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"idb.test.js","sourceRoot":"","sources":["../../../../../src/storage/providers/idb/idb.test.ts"],"names":[],"mappings":";;AAAA,mCAAsD;AAEtD,uCAKqB;AAErB,yCAAuC;AAEvC,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,KAAK,SAAS,EAAE;IACxC,IAAA,iBAAQ,EAAC,aAAa,EAAE,GAAG,EAAE;QACzB,MAAM,OAAO,GAAG,cAAc,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;QAEpE,MAAM,OAAO,GAAG,IAAI,qBAAU,CAAC,OAAO,CAAC,CAAA;QAEvC,IAAA,kBAAS,EAAC,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA;QAEtC,IAAA,6BAAsB,EAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QACxC,IAAA,6BAAsB,EAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QAClD,IAAA,0BAAmB,EAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QAClD,IAAA,gCAAyB,EAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QAE9D,IAAA,iBAAQ,EAAC,KAAK,IAAI,EAAE;YAChB,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAA;YAE9B,MAAM,GAAG,GAAG,SAAS,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;YAC7C,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACxC,GAAG,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;gBACrC,GAAG,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,CAAA;gBAC/B,GAAG,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,CAAA;YACnC,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;CACL;KAAM;IACH,iBAAQ,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;CACzC","sourcesContent":["import { afterAll, beforeAll, describe } from 'vitest'\n\nimport {\n    testAuthKeysRepository,\n    testKeyValueRepository,\n    testPeersRepository,\n    testRefMessagesRepository,\n} from '@mtcute/test'\n\nimport { IdbStorage } from './index.js'\n\nif (import.meta.env.TEST_ENV === 'browser') {\n    describe('idb storage', () => {\n        const idbName = 'mtcute_test_' + Math.random().toString(36).slice(2)\n\n        const storage = new IdbStorage(idbName)\n\n        beforeAll(() => storage.driver.load())\n\n        testAuthKeysRepository(storage.authKeys)\n        testKeyValueRepository(storage.kv, storage.driver)\n        testPeersRepository(storage.peers, storage.driver)\n        testRefMessagesRepository(storage.refMessages, storage.driver)\n\n        afterAll(async () => {\n            await storage.driver.destroy()\n\n            const req = indexedDB.deleteDatabase(idbName)\n            await new Promise<void>((resolve, reject) => {\n                req.onerror = () => reject(req.error)\n                req.onsuccess = () => resolve()\n                req.onblocked = () => resolve()\n            })\n        })\n    })\n} else {\n    describe.skip('idb storage', () => {})\n}\n"]}