@mtcute/dispatcher 0.16.7 → 0.16.13

package/cjs/state/providers/memory.js

@@ -1,81 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MemoryStateStorage = void 0;
-const core_1 = require("@mtcute/core");
-class MemoryStateRepository {
-    _driver;
-    state;
-    rl;
-    constructor(_driver) {
-        this._driver = _driver;
-        this.state = this._driver.getState('dispatcher_fsm', () => new Map());
-        this.rl = this._driver.getState('rl', () => new Map());
-    }
-    setState(key, state, ttl) {
-        this.state.set(key, {
-            value: state,
-            expiresAt: ttl ? Date.now() + ttl * 1000 : undefined,
-        });
-    }
-    getState(key, now) {
-        const state = this.state.get(key);
-        if (!state)
-            return null;
-        if (state.expiresAt && state.expiresAt < now) {
-            this.state.delete(key);
-            return null;
-        }
-        return state.value;
-    }
-    deleteState(key) {
-        this.state.delete(key);
-    }
-    vacuum(now) {
-        for (const [key, state] of this.state.entries()) {
-            if (state.expiresAt && state.expiresAt < now) {
-                this.state.delete(key);
-            }
-        }
-        for (const [key, state] of this.rl.entries()) {
-            if (state.reset < now) {
-                this.rl.delete(key);
-            }
-        }
-    }
-    getRateLimit(key, now, limit, window) {
-        // leaky bucket
-        const item = this.rl.get(key);
-        if (!item) {
-            const state = {
-                reset: now + window * 1000,
-                remaining: limit,
-            };
-            this.rl.set(key, state);
-            return [state.remaining, state.reset];
-        }
-        if (item.reset < now) {
-            // expired
-            const state = {
-                reset: now + window * 1000,
-                remaining: limit,
-            };
-            this.rl.set(key, state);
-            return [state.remaining, state.reset];
-        }
-        item.remaining = item.remaining > 0 ? item.remaining - 1 : 0;
-        return [item.remaining, item.reset];
-    }
-    resetRateLimit(key) {
-        this.rl.delete(key);
-    }
-}
-class MemoryStateStorage {
-    driver;
-    state;
-    constructor(driver = new core_1.MemoryStorageDriver()) {
-        this.driver = driver;
-        this.state = new MemoryStateRepository(this.driver);
-    }
-}
-exports.MemoryStateStorage = MemoryStateStorage;
-//# sourceMappingURL=memory.js.map

package/cjs/state/providers/memory.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/state/providers/memory.ts"],"names":[],"mappings":";;;AACA,uCAAkD;AAelD,MAAM,qBAAqB;IAGF;IAFZ,KAAK,CAAuB;IAC5B,EAAE,CAA2B;IACtC,YAAqB,OAA4B;QAA5B,YAAO,GAAP,OAAO,CAAqB;QAC7C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAwB,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,CAAA;QAC5F,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAA4B,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,CAAA;IACrF,CAAC;IAED,QAAQ,CAAC,GAAW,EAAE,KAAa,EAAE,GAAwB;QACzD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE;YAChB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS;SACvD,CAAC,CAAA;IACN,CAAC;IAED,QAAQ,CAAC,GAAW,EAAE,GAAW;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QACjC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAA;QAEvB,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;YAC3C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;YAEtB,OAAO,IAAI,CAAA;QACf,CAAC;QAED,OAAO,KAAK,CAAC,KAAK,CAAA;IACtB,CAAC;IAED,WAAW,CAAC,GAAW;QACnB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IAC1B,CAAC;IAED,MAAM,CAAC,GAAW;QACd,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YAC9C,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;gBAC3C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;YAC1B,CAAC;QACL,CAAC;QAED,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;gBACpB,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;YACvB,CAAC;QACL,CAAC;IACL,CAAC;IAED,YAAY,CAAC,GAAW,EAAE,GAAW,EAAE,KAAa,EAAE,MAAc;QAChE,eAAe;QACf,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QAE7B,IAAI,CAAC,IAAI,EAAE,CAAC;YACR,MAAM,KAAK,GAAiB;gBACxB,KAAK,EAAE,GAAG,GAAG,MAAM,GAAG,IAAI;gBAC1B,SAAS,EAAE,KAAK;aACnB,CAAA;YAED,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;YAEvB,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;QACzC,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;YACnB,UAAU;YAEV,MAAM,KAAK,GAAiB;gBACxB,KAAK,EAAE,GAAG,GAAG,MAAM,GAAG,IAAI;gBAC1B,SAAS,EAAE,KAAK;aACnB,CAAA;YAED,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;YAEvB,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;QACzC,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QAE5D,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;IACvC,CAAC;IAED,cAAc,CAAC,GAAW;QACtB,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IACvB,CAAC;CACJ;AAED,MAAa,kBAAkB;IAGN;IAFZ,KAAK,CAAuB;IAErC,YAAqB,SAA8B,IAAI,0BAAmB,EAAE;QAAvD,WAAM,GAAN,MAAM,CAAiD;QACxE,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACvD,CAAC;CACJ;AAND,gDAMC","sourcesContent":["import type { MaybePromise } from '@mtcute/core'\nimport { MemoryStorageDriver } from '@mtcute/core'\n\nimport type { IStateStorageProvider } from '../provider.js'\nimport type { IStateRepository } from '../repository.js'\n\ninterface StateDto {\n    value: string\n    expiresAt?: number\n}\n\ninterface RateLimitDto {\n    reset: number\n    remaining: number\n}\n\nclass MemoryStateRepository implements IStateRepository {\n    readonly state: Map<string, StateDto>\n    readonly rl: Map<string, RateLimitDto>\n    constructor(readonly _driver: MemoryStorageDriver) {\n        this.state = this._driver.getState<Map<string, StateDto>>('dispatcher_fsm', () => new Map())\n        this.rl = this._driver.getState<Map<string, RateLimitDto>>('rl', () => new Map())\n    }\n\n    setState(key: string, state: string, ttl?: number | undefined): void {\n        this.state.set(key, {\n            value: state,\n            expiresAt: ttl ? Date.now() + ttl * 1000 : undefined,\n        })\n    }\n\n    getState(key: string, now: number): string | null {\n        const state = this.state.get(key)\n        if (!state) return null\n\n        if (state.expiresAt && state.expiresAt < now) {\n            this.state.delete(key)\n\n            return null\n        }\n\n        return state.value\n    }\n\n    deleteState(key: string): void {\n        this.state.delete(key)\n    }\n\n    vacuum(now: number): void {\n        for (const [key, state] of this.state.entries()) {\n            if (state.expiresAt && state.expiresAt < now) {\n                this.state.delete(key)\n            }\n        }\n\n        for (const [key, state] of this.rl.entries()) {\n            if (state.reset < now) {\n                this.rl.delete(key)\n            }\n        }\n    }\n\n    getRateLimit(key: string, now: number, limit: number, window: number): [number, number] {\n        // leaky bucket\n        const item = this.rl.get(key)\n\n        if (!item) {\n            const state: RateLimitDto = {\n                reset: now + window * 1000,\n                remaining: limit,\n            }\n\n            this.rl.set(key, state)\n\n            return [state.remaining, state.reset]\n        }\n\n        if (item.reset < now) {\n            // expired\n\n            const state: RateLimitDto = {\n                reset: now + window * 1000,\n                remaining: limit,\n            }\n\n            this.rl.set(key, state)\n\n            return [state.remaining, state.reset]\n        }\n\n        item.remaining = item.remaining > 0 ? item.remaining - 1 : 0\n\n        return [item.remaining, item.reset]\n    }\n\n    resetRateLimit(key: string): MaybePromise<void> {\n        this.rl.delete(key)\n    }\n}\n\nexport class MemoryStateStorage implements IStateStorageProvider {\n    readonly state: MemoryStateRepository\n\n    constructor(readonly driver: MemoryStorageDriver = new MemoryStorageDriver()) {\n        this.state = new MemoryStateRepository(this.driver)\n    }\n}\n"]}

package/cjs/state/providers/sqlite.d.ts

@@ -1,28 +0,0 @@
-import type { BaseSqliteStorage, BaseSqliteStorageDriver, MaybePromise } from '@mtcute/core';
-import type { IStateStorageProvider } from '../provider.js';
-import type { IStateRepository } from '../repository.js';
-declare class SqliteStateRepository implements IStateRepository {
-    readonly _driver: BaseSqliteStorageDriver;
-    constructor(_driver: BaseSqliteStorageDriver);
-    private _setState;
-    setState(key: string, state: string, ttl?: number | undefined): MaybePromise<void>;
-    private _getState;
-    getState(key: string, now: number): MaybePromise<string | null>;
-    private _deleteState;
-    deleteState(key: string): MaybePromise<void>;
-    private _deleteOldState;
-    private _deleteOldRl;
-    vacuum(now: number): MaybePromise<void>;
-    private _setRl;
-    private _getRl;
-    private _deleteRl;
-    getRateLimit(key: string, now: number, limit: number, window: number): [number, number];
-    resetRateLimit(key: string): MaybePromise<void>;
-}
-export declare class SqliteStateStorage implements IStateStorageProvider {
-    readonly driver: BaseSqliteStorageDriver;
-    readonly state: SqliteStateRepository;
-    constructor(driver: BaseSqliteStorageDriver);
-    static from(provider: BaseSqliteStorage): SqliteStateStorage;
-}
-export {};

package/cjs/state/providers/sqlite.js

@@ -1,100 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SqliteStateStorage = void 0;
-class SqliteStateRepository {
-    _driver;
-    constructor(_driver) {
-        this._driver = _driver;
-        _driver.registerMigration('state', 1, (db) => {
-            db.exec(`
-                create table fsm_state (
-                    key text primary key,
-                    value text not null,
-                    expires_at integer
-                );
-                create table rl_state (
-                    key text primary key,
-                    reset integer not null,
-                    remaining integer not null
-                );
-            `);
-        });
-        _driver.onLoad(() => {
-            this._setState = _driver.db.prepare('insert or replace into fsm_state (key, value, expires_at) values (?, ?, ?)');
-            this._getState = _driver.db.prepare('select value, expires_at from fsm_state where key = ?');
-            this._deleteState = _driver.db.prepare('delete from fsm_state where key = ?');
-            this._deleteOldState = _driver.db.prepare('delete from fsm_state where expires_at < ?');
-            this._setRl = _driver.db.prepare('insert or replace into rl_state (key, reset, remaining) values (?, ?, ?)');
-            this._getRl = _driver.db.prepare('select reset, remaining from rl_state where key = ?');
-            this._deleteRl = _driver.db.prepare('delete from rl_state where key = ?');
-            this._deleteOldRl = _driver.db.prepare('delete from rl_state where reset < ?');
-        });
-        _driver.registerLegacyMigration('state', (db) => {
-            // not too important information, just drop the table
-            db.exec('drop table state');
-        });
-    }
-    _setState;
-    setState(key, state, ttl) {
-        this._setState.run(key, state, ttl ? Date.now() + ttl * 1000 : undefined);
-    }
-    _getState;
-    getState(key, now) {
-        const res_ = this._getState.get(key);
-        if (!res_)
-            return null;
-        const res = res_;
-        if (res.expires_at && res.expires_at < now) {
-            this._deleteState.run(key);
-            return null;
-        }
-        return res.value;
-    }
-    _deleteState;
-    deleteState(key) {
-        this._deleteState.run(key);
-    }
-    _deleteOldState;
-    _deleteOldRl;
-    vacuum(now) {
-        this._deleteOldState.run(now);
-        this._deleteOldRl.run(now);
-    }
-    _setRl;
-    _getRl;
-    _deleteRl;
-    getRateLimit(key, now, limit, window) {
-        const val = this._getRl.get(key);
-        // hot path. rate limit fsm entries always have an expiration date
-        if (!val || val.reset < now) {
-            // expired or does not exist
-            const item = {
-                reset: now + window * 1000,
-                remaining: limit,
-            };
-            this._setRl.run(key, item.reset, item.remaining);
-            return [item.remaining, item.reset];
-        }
-        if (val.remaining > 0) {
-            val.remaining -= 1;
-            this._setRl.run(key, val.reset, val.remaining);
-        }
-        return [val.remaining, val.reset];
-    }
-    resetRateLimit(key) {
-        this._deleteRl.run(key);
-    }
-}
-class SqliteStateStorage {
-    driver;
-    state;
-    constructor(driver) {
-        this.driver = driver;
-        this.state = new SqliteStateRepository(driver);
-    }
-    static from(provider) {
-        return new SqliteStateStorage(provider.driver);
-    }
-}
-exports.SqliteStateStorage = SqliteStateStorage;
-//# sourceMappingURL=sqlite.js.map

package/cjs/state/providers/sqlite.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"sqlite.js","sourceRoot":"","sources":["../../../../src/state/providers/sqlite.ts"],"names":[],"mappings":";;;AAeA,MAAM,qBAAqB;IACF;IAArB,YAAqB,OAAgC;QAAhC,YAAO,GAAP,OAAO,CAAyB;QACjD,OAAO,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE;YACzC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;aAWP,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE;YAChB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAC/B,4EAA4E,CAC/E,CAAA;YACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,uDAAuD,CAAC,CAAA;YAC5F,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAA;YAC7E,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAA;YAEvF,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,0EAA0E,CAAC,CAAA;YAC5G,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAA;YACvF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAA;YACzE,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAA;QAClF,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,uBAAuB,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE;YAC5C,qDAAqD;YACrD,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;QAC/B,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,SAAS,CAAmB;IACpC,QAAQ,CAAC,GAAW,EAAE,KAAa,EAAE,GAAwB;QACzD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;IAC7E,CAAC;IAEO,SAAS,CAAmB;IACpC,QAAQ,CAAC,GAAW,EAAE,GAAW;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QACpC,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAA;QACtB,MAAM,GAAG,GAAG,IAAgB,CAAA;QAE5B,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;YAE1B,OAAO,IAAI,CAAA;QACf,CAAC;QAED,OAAO,GAAG,CAAC,KAAK,CAAA;IACpB,CAAC;IAEO,YAAY,CAAmB;IACvC,WAAW,CAAC,GAAW;QACnB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;IAC9B,CAAC;IAEO,eAAe,CAAmB;IAClC,YAAY,CAAmB;IACvC,MAAM,CAAC,GAAW;QACd,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QAC7B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;IAC9B,CAAC;IAEO,MAAM,CAAmB;IACzB,MAAM,CAAmB;IACzB,SAAS,CAAmB;IAEpC,YAAY,CAAC,GAAW,EAAE,GAAW,EAAE,KAAa,EAAE,MAAc;QAChE,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAA6B,CAAA;QAE5D,kEAAkE;QAElE,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;YAC1B,4BAA4B;YAC5B,MAAM,IAAI,GAAiB;gBACvB,KAAK,EAAE,GAAG,GAAG,MAAM,GAAG,IAAI;gBAC1B,SAAS,EAAE,KAAK;aACnB,CAAA;YAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;YAEhD,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAA;QACvC,CAAC;QAED,IAAI,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;YACpB,GAAG,CAAC,SAAS,IAAI,CAAC,CAAA;YAElB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,SAAS,CAAC,CAAA;QAClD,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;IACrC,CAAC;IAED,cAAc,CAAC,GAAW;QACtB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;IAC3B,CAAC;CACJ;AAED,MAAa,kBAAkB;IAEN;IADZ,KAAK,CAAuB;IACrC,YAAqB,MAA+B;QAA/B,WAAM,GAAN,MAAM,CAAyB;QAChD,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAqB,CAAC,MAAM,CAAC,CAAA;IAClD,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,QAA2B;QACnC,OAAO,IAAI,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IAClD,CAAC;CACJ;AATD,gDASC","sourcesContent":["import type { BaseSqliteStorage, BaseSqliteStorageDriver, ISqliteStatement, MaybePromise } from '@mtcute/core'\n\nimport type { IStateStorageProvider } from '../provider.js'\nimport type { IStateRepository } from '../repository.js'\n\ninterface StateDto {\n    value: string\n    expires_at: number | null\n}\n\ninterface RateLimitDto {\n    reset: number\n    remaining: number\n}\n\nclass SqliteStateRepository implements IStateRepository {\n    constructor(readonly _driver: BaseSqliteStorageDriver) {\n        _driver.registerMigration('state', 1, (db) => {\n            db.exec(`\n                create table fsm_state (\n                    key text primary key,\n                    value text not null,\n                    expires_at integer\n                );\n                create table rl_state (\n                    key text primary key,\n                    reset integer not null,\n                    remaining integer not null\n                );\n            `)\n        })\n        _driver.onLoad(() => {\n            this._setState = _driver.db.prepare(\n                'insert or replace into fsm_state (key, value, expires_at) values (?, ?, ?)',\n            )\n            this._getState = _driver.db.prepare('select value, expires_at from fsm_state where key = ?')\n            this._deleteState = _driver.db.prepare('delete from fsm_state where key = ?')\n            this._deleteOldState = _driver.db.prepare('delete from fsm_state where expires_at < ?')\n\n            this._setRl = _driver.db.prepare('insert or replace into rl_state (key, reset, remaining) values (?, ?, ?)')\n            this._getRl = _driver.db.prepare('select reset, remaining from rl_state where key = ?')\n            this._deleteRl = _driver.db.prepare('delete from rl_state where key = ?')\n            this._deleteOldRl = _driver.db.prepare('delete from rl_state where reset < ?')\n        })\n        _driver.registerLegacyMigration('state', (db) => {\n            // not too important information, just drop the table\n            db.exec('drop table state')\n        })\n    }\n\n    private _setState!: ISqliteStatement\n    setState(key: string, state: string, ttl?: number | undefined): MaybePromise<void> {\n        this._setState.run(key, state, ttl ? Date.now() + ttl * 1000 : undefined)\n    }\n\n    private _getState!: ISqliteStatement\n    getState(key: string, now: number): MaybePromise<string | null> {\n        const res_ = this._getState.get(key)\n        if (!res_) return null\n        const res = res_ as StateDto\n\n        if (res.expires_at && res.expires_at < now) {\n            this._deleteState.run(key)\n\n            return null\n        }\n\n        return res.value\n    }\n\n    private _deleteState!: ISqliteStatement\n    deleteState(key: string): MaybePromise<void> {\n        this._deleteState.run(key)\n    }\n\n    private _deleteOldState!: ISqliteStatement\n    private _deleteOldRl!: ISqliteStatement\n    vacuum(now: number): MaybePromise<void> {\n        this._deleteOldState.run(now)\n        this._deleteOldRl.run(now)\n    }\n\n    private _setRl!: ISqliteStatement\n    private _getRl!: ISqliteStatement\n    private _deleteRl!: ISqliteStatement\n\n    getRateLimit(key: string, now: number, limit: number, window: number): [number, number] {\n        const val = this._getRl.get(key) as RateLimitDto | undefined\n\n        // hot path. rate limit fsm entries always have an expiration date\n\n        if (!val || val.reset < now) {\n            // expired or does not exist\n            const item: RateLimitDto = {\n                reset: now + window * 1000,\n                remaining: limit,\n            }\n\n            this._setRl.run(key, item.reset, item.remaining)\n\n            return [item.remaining, item.reset]\n        }\n\n        if (val.remaining > 0) {\n            val.remaining -= 1\n\n            this._setRl.run(key, val.reset, val.remaining)\n        }\n\n        return [val.remaining, val.reset]\n    }\n\n    resetRateLimit(key: string): MaybePromise<void> {\n        this._deleteRl.run(key)\n    }\n}\n\nexport class SqliteStateStorage implements IStateStorageProvider {\n    readonly state: SqliteStateRepository\n    constructor(readonly driver: BaseSqliteStorageDriver) {\n        this.state = new SqliteStateRepository(driver)\n    }\n\n    static from(provider: BaseSqliteStorage): SqliteStateStorage {\n        return new SqliteStateStorage(provider.driver)\n    }\n}\n"]}

package/cjs/state/repository.js

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

package/cjs/state/repository.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"repository.js","sourceRoot":"","sources":["../../../src/state/repository.ts"],"names":[],"mappings":"","sourcesContent":["import type { MaybePromise } from '@mtcute/core'\n\n/**\n * Interface for FSM storage for the dispatcher.\n *\n * All of the officially supported storages already implement\n * this interface, so you can just re-use it.\n *\n * Current scene is a special case of a `string` state,\n * Most of the time you can just store it the same way\n * as normal state, prefixing with something like `$current_state_`\n * (scene name can't start with `$`).\n * Alternatively, you can store them as simple strings\n */\nexport interface IStateRepository {\n    /**\n     * Retrieve state from the storage\n     * If state is not found or has expired, return `null`\n     *\n     * @param key  Key of the state, as defined by {@link StateKeyDelegate}\n     */\n    getState: (key: string, now: number) => MaybePromise<string | null>\n\n    /**\n     * Save state to the storage\n     *\n     * @param key  Key of the state, as defined by {@link StateKeyDelegate}\n     * @param state  String representing the state\n     * @param ttl  TTL for the state, in seconds\n     */\n    setState: (key: string, state: string, ttl?: number) => MaybePromise<void>\n\n    /**\n     * Delete state from the storage\n     *\n     * @param key  Key of the state, as defined by {@link StateKeyDelegate}\n     */\n    deleteState: (key: string) => MaybePromise<void>\n\n    /**\n     * Clean up expired states and rate limits.\n     *\n     * @param now  Current unix time in ms\n     */\n    vacuum: (now: number) => MaybePromise<void>\n\n    /**\n     * Get information about a rate limit.\n     *\n     * It is recommended that you use sliding window or leaky bucket\n     * to implement rate limiting ([learn more](https://konghq.com/blog/how-to-design-a-scalable-rate-limiting-algorithm/)),\n     *\n     * @param key  Key of the rate limit\n     * @param now  Current unix time in ms\n     * @param limit  Maximum number of requests in `window`\n     * @param window  Window size in seconds\n     * @returns  Tuple containing the number of remaining and\n     *   unix time in ms when the user can try again\n     */\n    getRateLimit: (key: string, now: number, limit: number, window: number) => MaybePromise<[number, number]>\n\n    /**\n     * Reset a rate limit.\n     *\n     * @param key  Key of the rate limit\n     */\n    resetRateLimit: (key: string) => MaybePromise<void>\n}\n"]}

package/cjs/state/service.d.ts

@@ -1,20 +0,0 @@
-import type { MaybePromise } from '@mtcute/core';
-import type { IStateStorageProvider } from './provider.js';
-export declare class StateService {
-    readonly provider: IStateStorageProvider;
-    constructor(provider: IStateStorageProvider);
-    private _cache;
-    private _vacuumTimer?;
-    private _loaded;
-    private _load;
-    load(): Promise<void>;
-    destroy(): Promise<void>;
-    getState<T>(key: string): Promise<T | null>;
-    setState<T>(key: string, state: T, ttl?: number): Promise<void>;
-    deleteState(key: string): Promise<void>;
-    getCurrentScene(key: string): Promise<string | null>;
-    setCurrentScene(key: string, scene: string, ttl?: number): Promise<void>;
-    deleteCurrentScene(key: string): Promise<void>;
-    getRateLimit(key: string, limit: number, window: number): MaybePromise<[number, number]>;
-    resetRateLimit(key: string): MaybePromise<void>;
-}

package/cjs/state/service.js

@@ -1,70 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.StateService = void 0;
-const utils_js_1 = require("@mtcute/core/utils.js");
-const makeCurrentSceneKey = (key) => `$current_scene_${key}`;
-class StateService {
-    provider;
-    constructor(provider) {
-        this.provider = provider;
-    }
-    _cache = new utils_js_1.LruMap(100);
-    _vacuumTimer;
-    _loaded = false;
-    _load = (0, utils_js_1.asyncResettable)(async () => {
-        await this.provider.driver.load?.();
-        this._loaded = true;
-    });
-    async load() {
-        await this._load.run();
-        this._vacuumTimer = setInterval(() => {
-            Promise.resolve(this.provider.state.vacuum(Date.now())).catch(() => { });
-        }, 300_000);
-    }
-    async destroy() {
-        await this.provider.driver.save?.();
-        await this.provider.driver.destroy?.();
-        clearInterval(this._vacuumTimer);
-        this._loaded = false;
-    }
-    async getState(key) {
-        if (!this._loaded)
-            await this.load();
-        const cached = this._cache.get(key);
-        if (cached)
-            return cached;
-        const state = await this.provider.state.getState(key, Date.now());
-        if (!state)
-            return null;
-        return JSON.parse(state);
-    }
-    async setState(key, state, ttl) {
-        if (!this._loaded)
-            await this.load();
-        this._cache.set(key, state);
-        await this.provider.state.setState(key, JSON.stringify(state), ttl);
-    }
-    async deleteState(key) {
-        if (!this._loaded)
-            await this.load();
-        this._cache.delete(key);
-        await this.provider.state.deleteState(key);
-    }
-    getCurrentScene(key) {
-        return this.getState(makeCurrentSceneKey(key));
-    }
-    setCurrentScene(key, scene, ttl) {
-        return this.setState(makeCurrentSceneKey(key), scene, ttl);
-    }
-    deleteCurrentScene(key) {
-        return this.deleteState(makeCurrentSceneKey(key));
-    }
-    getRateLimit(key, limit, window) {
-        return this.provider.state.getRateLimit(key, Date.now(), limit, window);
-    }
-    resetRateLimit(key) {
-        return this.provider.state.resetRateLimit(key);
-    }
-}
-exports.StateService = StateService;
-//# sourceMappingURL=service.js.map

package/cjs/state/service.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"service.js","sourceRoot":"","sources":["../../../src/state/service.ts"],"names":[],"mappings":";;;AAAA,oDAA+D;AAK/D,MAAM,mBAAmB,GAAG,CAAC,GAAW,EAAE,EAAE,CAAC,kBAAkB,GAAG,EAAE,CAAA;AAEpE,MAAa,YAAY;IACA;IAArB,YAAqB,QAA+B;QAA/B,aAAQ,GAAR,QAAQ,CAAuB;IAAG,CAAC;IAEhD,MAAM,GAA4B,IAAI,iBAAM,CAAC,GAAG,CAAC,CAAA;IACjD,YAAY,CAAiB;IAE7B,OAAO,GAAG,KAAK,CAAA;IACf,KAAK,GAAG,IAAA,0BAAe,EAAC,KAAK,IAAI,EAAE;QACvC,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAA;QACnC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;IACvB,CAAC,CAAC,CAAA;IAEF,KAAK,CAAC,IAAI;QACN,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAA;QACtB,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;YACjC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAC3E,CAAC,EAAE,OAAO,CAAC,CAAA;IACf,CAAC;IAED,KAAK,CAAC,OAAO;QACT,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAA;QACnC,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAA;QACtC,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QAChC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAA;IACxB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAI,GAAW;QACzB,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,CAAC,IAAI,EAAE,CAAA;QAEpC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QACnC,IAAI,MAAM;YAAE,OAAO,MAAW,CAAA;QAE9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAA;QACjE,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAA;QAEvB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAM,CAAA;IACjC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAI,GAAW,EAAE,KAAQ,EAAE,GAAY;QACjD,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,CAAC,IAAI,EAAE,CAAA;QAEpC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;QAC3B,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAA;IACvE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAW;QACzB,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,CAAC,IAAI,EAAE,CAAA;QAEpC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;QACvB,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAA;IAC9C,CAAC;IAED,eAAe,CAAC,GAAW;QACvB,OAAO,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAA;IAClD,CAAC;IAED,eAAe,CAAC,GAAW,EAAE,KAAa,EAAE,GAAY;QACpD,OAAO,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,CAAA;IAC9D,CAAC;IAED,kBAAkB,CAAC,GAAW;QAC1B,OAAO,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAA;IACrD,CAAC;IAED,YAAY,CAAC,GAAW,EAAE,KAAa,EAAE,MAAc;QACnD,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;IAC3E,CAAC;IAED,cAAc,CAAC,GAAW;QACtB,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,CAAA;IAClD,CAAC;CACJ;AAvED,oCAuEC","sourcesContent":["import { LruMap, asyncResettable } from '@mtcute/core/utils.js'\nimport type { MaybePromise } from '@mtcute/core'\n\nimport type { IStateStorageProvider } from './provider.js'\n\nconst makeCurrentSceneKey = (key: string) => `$current_scene_${key}`\n\nexport class StateService {\n    constructor(readonly provider: IStateStorageProvider) {}\n\n    private _cache: LruMap<string, unknown> = new LruMap(100)\n    private _vacuumTimer?: NodeJS.Timeout\n\n    private _loaded = false\n    private _load = asyncResettable(async () => {\n        await this.provider.driver.load?.()\n        this._loaded = true\n    })\n\n    async load(): Promise<void> {\n        await this._load.run()\n        this._vacuumTimer = setInterval(() => {\n            Promise.resolve(this.provider.state.vacuum(Date.now())).catch(() => {})\n        }, 300_000)\n    }\n\n    async destroy(): Promise<void> {\n        await this.provider.driver.save?.()\n        await this.provider.driver.destroy?.()\n        clearInterval(this._vacuumTimer)\n        this._loaded = false\n    }\n\n    async getState<T>(key: string): Promise<T | null> {\n        if (!this._loaded) await this.load()\n\n        const cached = this._cache.get(key)\n        if (cached) return cached as T\n\n        const state = await this.provider.state.getState(key, Date.now())\n        if (!state) return null\n\n        return JSON.parse(state) as T\n    }\n\n    async setState<T>(key: string, state: T, ttl?: number): Promise<void> {\n        if (!this._loaded) await this.load()\n\n        this._cache.set(key, state)\n        await this.provider.state.setState(key, JSON.stringify(state), ttl)\n    }\n\n    async deleteState(key: string): Promise<void> {\n        if (!this._loaded) await this.load()\n\n        this._cache.delete(key)\n        await this.provider.state.deleteState(key)\n    }\n\n    getCurrentScene(key: string): Promise<string | null> {\n        return this.getState(makeCurrentSceneKey(key))\n    }\n\n    setCurrentScene(key: string, scene: string, ttl?: number): Promise<void> {\n        return this.setState(makeCurrentSceneKey(key), scene, ttl)\n    }\n\n    deleteCurrentScene(key: string): Promise<void> {\n        return this.deleteState(makeCurrentSceneKey(key))\n    }\n\n    getRateLimit(key: string, limit: number, window: number): MaybePromise<[number, number]> {\n        return this.provider.state.getRateLimit(key, Date.now(), limit, window)\n    }\n\n    resetRateLimit(key: string): MaybePromise<void> {\n        return this.provider.state.resetRateLimit(key)\n    }\n}\n"]}

package/cjs/state/update-state.js

@@ -1,219 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.UpdateState = exports.RateLimitError = void 0;
-const core_1 = require("@mtcute/core");
-const utils_js_1 = require("@mtcute/core/utils.js");
-/**
- * Error thrown by `.rateLimit()`
- */
-class RateLimitError extends core_1.MtcuteError {
-    reset;
-    constructor(reset) {
-        super('You are being rate limited.');
-        this.reset = reset;
-    }
-}
-exports.RateLimitError = RateLimitError;
-/**
- * State of the current update.
- *
- * @template State  Type that represents the state
- * @template SceneName  Possible scene names
- */
-class UpdateState {
-    _key;
-    _localKey;
-    _storage;
-    _scene;
-    _scoped;
-    _cached;
-    _localStorage;
-    _localKeyBase;
-    constructor(storage, key, scene, scoped, customStorage, customKey) {
-        this._storage = storage;
-        this._key = key;
-        this._scene = scene;
-        this._scoped = scoped;
-        this._localStorage = customStorage ?? storage;
-        this._localKeyBase = customKey ?? key;
-        this._updateLocalKey();
-    }
-    /** Name of the current scene */
-    get scene() {
-        return this._scene;
-    }
-    _updateLocalKey() {
-        if (!this._scoped) {
-            this._localKey = this._localKeyBase;
-        }
-        else {
-            this._localKey = this._scene ? `${this._scene}_${this._localKeyBase}` : this._localKeyBase;
-        }
-    }
-    async get(fallback, force) {
-        if (typeof fallback === 'boolean') {
-            force = fallback;
-            fallback = undefined;
-        }
-        if (!force && this._cached !== undefined) {
-            if (!this._cached && fallback) {
-                return typeof fallback === 'function' ? fallback() : fallback;
-            }
-            return this._cached;
-        }
-        let res = (await this._localStorage.getState(this._localKey));
-        if (!res && fallback) {
-            res = typeof fallback === 'function' ? fallback() : fallback;
-        }
-        this._cached = res;
-        return res;
-    }
-    /**
-     * Set new state to the storage
-     *
-     * @param state  New state
-     * @param ttl  TTL for the new state (in seconds)
-     */
-    async set(state, ttl) {
-        this._cached = state;
-        await this._localStorage.setState(this._localKey, state, ttl);
-    }
-    /**
-     * Merge the given object to the current state.
-     *
-     * > **Note**: If the storage currently has no state,
-     * > then `fallback` must be provided.
-     *
-     * Basically a shorthand to calling `.get()`,
-     * modifying and then calling `.set()`
-     *
-     * @param state  State to be merged
-     * @param fallback  Default state
-     * @param ttl  TTL for the new state (in seconds)
-     * @param forceLoad  Whether to force load the old state from storage
-     */
-    async merge(state, params = {}) {
-        const { fallback, ttl, forceLoad } = params;
-        const old = await this.get(forceLoad);
-        if (!old) {
-            if (!fallback) {
-                throw new core_1.MtArgumentError('Cannot use merge on empty state without fallback.');
-            }
-            const fallback_ = typeof fallback === 'function' ? fallback() : fallback;
-            await this.set({ ...fallback_, ...state }, ttl);
-        }
-        else {
-            await this.set({ ...old, ...state }, ttl);
-        }
-        // _cached is set by .set()
-        return this._cached;
-    }
-    /**
-     * Delete the state from the storage
-     */
-    async delete() {
-        this._cached = null;
-        await this._localStorage.deleteState(this._localKey);
-    }
-    /**
-     * Enter some scene
-     */
-    async enter(scene, params) {
-        const { with: with_, ttl, reset = true } = params ?? {};
-        if (reset && this._scoped)
-            await this.delete();
-        if (!scene['_scene']) {
-            throw new core_1.MtArgumentError('Cannot enter a non-scene Dispatcher');
-        }
-        if (!scene['_parent']) {
-            throw new core_1.MtArgumentError('This scene has not been registered');
-        }
-        this._scene = scene['_scene'];
-        this._scoped = scene['_sceneScoped'];
-        this._updateLocalKey();
-        await this._storage.setCurrentScene(this._key, this._scene, ttl);
-        if (with_) {
-            if (scene['_customStateKeyDelegate']) {
-                throw new core_1.MtArgumentError('Cannot use `with` parameter when the scene uses a custom state key delegate');
-            }
-            await scene.getState(this._key).set(with_, ttl);
-        }
-    }
-    /**
-     * Exit from current scene to the root
-     *
-     * @param reset
-     *     Whether to reset scene state (only applicable in case this is a scoped scene)
-     */
-    async exit(reset = true) {
-        if (reset && this._scoped)
-            await this.delete();
-        this._scene = null;
-        this._updateLocalKey();
-        await this._storage.deleteCurrentScene(this._key);
-    }
-    /**
-     * Rate limit some handler.
-     *
-     * When the rate limit exceeds, {@link RateLimitError} is thrown.
-     *
-     * This is a simple rate-limiting solution that uses
-     * the same key as the state. If you need something more
-     * sophisticated and/or customizable, you'll have to implement
-     * your own rate-limiter.
-     *
-     * > **Note**: `key` is used to prefix the local key
-     * > derived using the given key delegate.
-     *
-     * @param key  Key of the rate limit
-     * @param limit  Maximum number of requests in `window`
-     * @param window  Window size in seconds
-     * @returns  Tuple containing the number of remaining and
-     *   unix time in ms when the user can try again
-     */
-    async rateLimit(key, limit, window) {
-        const [remaining, reset] = await this._localStorage.getRateLimit(`${key}:${this._localKey}`, limit, window);
-        if (!remaining) {
-            throw new RateLimitError(reset);
-        }
-        return [remaining - 1, reset];
-    }
-    /**
-     * Throttle some handler.
-     *
-     * When the rate limit exceeds, this function waits for it to reset.
-     *
-     * This is a simple wrapper over {@link rateLimit}, and follows the same logic.
-     *
-     * > **Note**: `key` is used to prefix the local key
-     * > derived using the given key delegate.
-     *
-     * @param key  Key of the rate limit
-     * @param limit  Maximum number of requests in `window`
-     * @param window  Window size in seconds
-     * @returns  Tuple containing the number of remaining and
-     *   unix time in ms when the user can try again
-     */
-    async throttle(key, limit, window) {
-        try {
-            return await this.rateLimit(key, limit, window);
-        }
-        catch (e) {
-            if (e instanceof RateLimitError) {
-                await (0, utils_js_1.sleep)(e.reset - Date.now());
-                return this.throttle(key, limit, window);
-            }
-            throw e;
-        }
-    }
-    /**
-     * Reset the rate limit
-     *
-     * @param key  Key of the rate limit
-     */
-    async resetRateLimit(key) {
-        await this._localStorage.resetRateLimit(`${key}:${this._localKey}`);
-    }
-}
-exports.UpdateState = UpdateState;
-//# sourceMappingURL=update-state.js.map

package/cjs/state/update-state.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"update-state.js","sourceRoot":"","sources":["../../../src/state/update-state.ts"],"names":[],"mappings":";;;AAAA,uCAA2D;AAC3D,oDAA6C;AAM7C;;GAEG;AACH,MAAa,cAAe,SAAQ,kBAAW;IACtB;IAArB,YAAqB,KAAa;QAC9B,KAAK,CAAC,6BAA6B,CAAC,CAAA;QADnB,UAAK,GAAL,KAAK,CAAQ;IAElC,CAAC;CACJ;AAJD,wCAIC;AAED;;;;;GAKG;AACH,MAAa,WAAW;IACZ,IAAI,CAAQ;IACZ,SAAS,CAAS;IAElB,QAAQ,CAAc;IAEtB,MAAM,CAAe;IACrB,OAAO,CAAU;IACjB,OAAO,CAAe;IAEtB,aAAa,CAAc;IAC3B,aAAa,CAAQ;IAE7B,YACI,OAAqB,EACrB,GAAW,EACX,KAAoB,EACpB,MAAgB,EAChB,aAA4B,EAC5B,SAAkB;QAElB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAA;QACvB,IAAI,CAAC,IAAI,GAAG,GAAG,CAAA;QACf,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACnB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QAErB,IAAI,CAAC,aAAa,GAAG,aAAa,IAAI,OAAO,CAAA;QAC7C,IAAI,CAAC,aAAa,GAAG,SAAS,IAAI,GAAG,CAAA;QAErC,IAAI,CAAC,eAAe,EAAE,CAAA;IAC1B,CAAC;IAED,gCAAgC;IAChC,IAAI,KAAK;QACL,OAAO,IAAI,CAAC,MAAM,CAAA;IACtB,CAAC;IAEO,eAAe;QACnB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAA;QACvC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAA;QAC9F,CAAC;IACL,CAAC;IAyBD,KAAK,CAAC,GAAG,CAAC,QAA0C,EAAE,KAAe;QACjE,IAAI,OAAO,QAAQ,KAAK,SAAS,EAAE,CAAC;YAChC,KAAK,GAAG,QAAQ,CAAA;YAChB,QAAQ,GAAG,SAAS,CAAA;QACxB,CAAC;QAED,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC5B,OAAO,OAAO,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAA;YACjE,CAAC;YAED,OAAO,IAAI,CAAC,OAAO,CAAA;QACvB,CAAC;QAED,IAAI,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAiB,CAAA;QAE7E,IAAI,CAAC,GAAG,IAAI,QAAQ,EAAE,CAAC;YACnB,GAAG,GAAG,OAAO,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAA;QAChE,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,GAAG,CAAA;QAElB,OAAO,GAAG,CAAA;IACd,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,GAAY;QAChC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAA;QACpB,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,GAAG,CAAC,CAAA;IACjE,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,KAAK,CAAC,KAAK,CACP,KAAqB,EACrB,SAII,EAAE;QAEN,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,MAAM,CAAA;QAE3C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QAErC,IAAI,CAAC,GAAG,EAAE,CAAC;YACP,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACZ,MAAM,IAAI,sBAAe,CAAC,mDAAmD,CAAC,CAAA;YAClF,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAA;YAExE,MAAM,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,EAAE,GAAG,KAAK,EAAE,EAAE,GAAG,CAAC,CAAA;QACnD,CAAC;aAAM,CAAC;YACJ,MAAM,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,EAAE,GAAG,KAAK,EAAE,EAAE,GAAG,CAAC,CAAA;QAC7C,CAAC;QAED,2BAA2B;QAE3B,OAAO,IAAI,CAAC,OAAQ,CAAA;IACxB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM;QACR,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;IACxD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CACP,KAAY,EACZ,MAiBC;QAED,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,GAAG,IAAI,EAAE,GAAG,MAAM,IAAI,EAAE,CAAA;QAEvD,IAAI,KAAK,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,CAAC,MAAM,EAAE,CAAA;QAE9C,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,sBAAe,CAAC,qCAAqC,CAAC,CAAA;QACpE,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;YACpB,MAAM,IAAI,sBAAe,CAAC,oCAAoC,CAAC,CAAA;QACnE,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAA;QAC7B,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,CAAA;QACpC,IAAI,CAAC,eAAe,EAAE,CAAA;QAEtB,MAAM,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;QAEhE,IAAI,KAAK,EAAE,CAAC;YACR,IAAI,KAAK,CAAC,yBAAyB,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,sBAAe,CAAC,6EAA6E,CAAC,CAAA;YAC5G,CAAC;YAED,MAAM,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;QACnD,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,IAAI;QACnB,IAAI,KAAK,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,CAAC,MAAM,EAAE,CAAA;QAC9C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAA;QAClB,IAAI,CAAC,eAAe,EAAE,CAAA;QACtB,MAAM,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrD,CAAC;IAED;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,CAAC,SAAS,CAAC,GAAW,EAAE,KAAa,EAAE,MAAc;QACtD,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,GAAG,IAAI,IAAI,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;QAE3G,IAAI,CAAC,SAAS,EAAE,CAAC;YACb,MAAM,IAAI,cAAc,CAAC,KAAK,CAAC,CAAA;QACnC,CAAC;QAED,OAAO,CAAC,SAAS,GAAG,CAAC,EAAE,KAAK,CAAC,CAAA;IACjC,CAAC;IAED;;;;;;;;;;;;;;;OAeG;IACH,KAAK,CAAC,QAAQ,CAAC,GAAW,EAAE,KAAa,EAAE,MAAc;QACrD,IAAI,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;QACnD,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,cAAc,EAAE,CAAC;gBAC9B,MAAM,IAAA,gBAAK,EAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAA;gBAEjC,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;YAC5C,CAAC;YAED,MAAM,CAAC,CAAA;QACX,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,cAAc,CAAC,GAAW;QAC5B,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,GAAG,GAAG,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC,CAAA;IACvE,CAAC;CACJ;AA7RD,kCA6RC","sourcesContent":["import { MtArgumentError, MtcuteError } from '@mtcute/core'\nimport { sleep } from '@mtcute/core/utils.js'\n\nimport type { Dispatcher } from '../dispatcher.js'\n\nimport type { StateService } from './service.js'\n\n/**\n * Error thrown by `.rateLimit()`\n */\nexport class RateLimitError extends MtcuteError {\n    constructor(readonly reset: number) {\n        super('You are being rate limited.')\n    }\n}\n\n/**\n * State of the current update.\n *\n * @template State  Type that represents the state\n * @template SceneName  Possible scene names\n */\nexport class UpdateState<State extends object> {\n    private _key: string\n    private _localKey!: string\n\n    private _storage: StateService\n\n    private _scene: string | null\n    private _scoped?: boolean\n    private _cached?: State | null\n\n    private _localStorage: StateService\n    private _localKeyBase: string\n\n    constructor(\n        storage: StateService,\n        key: string,\n        scene: string | null,\n        scoped?: boolean,\n        customStorage?: StateService,\n        customKey?: string,\n    ) {\n        this._storage = storage\n        this._key = key\n        this._scene = scene\n        this._scoped = scoped\n\n        this._localStorage = customStorage ?? storage\n        this._localKeyBase = customKey ?? key\n\n        this._updateLocalKey()\n    }\n\n    /** Name of the current scene */\n    get scene(): string | null {\n        return this._scene\n    }\n\n    private _updateLocalKey(): void {\n        if (!this._scoped) {\n            this._localKey = this._localKeyBase\n        } else {\n            this._localKey = this._scene ? `${this._scene}_${this._localKeyBase}` : this._localKeyBase\n        }\n    }\n\n    /**\n     * Retrieve the state from the storage, falling back to default\n     * if not found\n     *\n     * @param fallback  Default state value\n     * @param force  Whether to ignore cached state (def. `false`)\n     */\n    async get(fallback: State | (() => State), force?: boolean): Promise<State>\n\n    /**\n     * Retrieve the state from the storage, falling back to default\n     * if not found\n     *\n     * @param fallback  Default state value\n     * @param force  Whether to ignore cached state (def. `false`)\n     */\n    async get(fallback?: State | (() => State), force?: boolean): Promise<State | null>\n    /**\n     * Retrieve the state from the storage\n     *\n     * @param force  Whether to ignore cached state (def. `false`)\n     */\n    async get(force?: boolean): Promise<State | null>\n    async get(fallback?: State | (() => State) | boolean, force?: boolean): Promise<State | null> {\n        if (typeof fallback === 'boolean') {\n            force = fallback\n            fallback = undefined\n        }\n\n        if (!force && this._cached !== undefined) {\n            if (!this._cached && fallback) {\n                return typeof fallback === 'function' ? fallback() : fallback\n            }\n\n            return this._cached\n        }\n\n        let res = (await this._localStorage.getState(this._localKey)) as State | null\n\n        if (!res && fallback) {\n            res = typeof fallback === 'function' ? fallback() : fallback\n        }\n        this._cached = res\n\n        return res\n    }\n\n    /**\n     * Set new state to the storage\n     *\n     * @param state  New state\n     * @param ttl  TTL for the new state (in seconds)\n     */\n    async set(state: State, ttl?: number): Promise<void> {\n        this._cached = state\n        await this._localStorage.setState(this._localKey, state, ttl)\n    }\n\n    /**\n     * Merge the given object to the current state.\n     *\n     * > **Note**: If the storage currently has no state,\n     * > then `fallback` must be provided.\n     *\n     * Basically a shorthand to calling `.get()`,\n     * modifying and then calling `.set()`\n     *\n     * @param state  State to be merged\n     * @param fallback  Default state\n     * @param ttl  TTL for the new state (in seconds)\n     * @param forceLoad  Whether to force load the old state from storage\n     */\n    async merge(\n        state: Partial<State>,\n        params: {\n            fallback?: State | (() => State)\n            ttl?: number\n            forceLoad?: boolean\n        } = {},\n    ): Promise<State> {\n        const { fallback, ttl, forceLoad } = params\n\n        const old = await this.get(forceLoad)\n\n        if (!old) {\n            if (!fallback) {\n                throw new MtArgumentError('Cannot use merge on empty state without fallback.')\n            }\n\n            const fallback_ = typeof fallback === 'function' ? fallback() : fallback\n\n            await this.set({ ...fallback_, ...state }, ttl)\n        } else {\n            await this.set({ ...old, ...state }, ttl)\n        }\n\n        // _cached is set by .set()\n\n        return this._cached!\n    }\n\n    /**\n     * Delete the state from the storage\n     */\n    async delete(): Promise<void> {\n        this._cached = null\n        await this._localStorage.deleteState(this._localKey)\n    }\n\n    /**\n     * Enter some scene\n     */\n    async enter<SceneState extends object, Scene extends Dispatcher<SceneState>>(\n        scene: Scene,\n        params?: {\n            /**\n             * Initial state for the scene\n             *\n             * Note that this will only work if the scene uses the same key delegate as this state.\n             */\n            with?: SceneState\n\n            /** TTL for the scene (in seconds) */\n            ttl?: number\n\n            /**\n             * If currently in a scoped scene, whether to reset the state\n             *\n             * @default  true\n             */\n            reset?: boolean\n        },\n    ): Promise<void> {\n        const { with: with_, ttl, reset = true } = params ?? {}\n\n        if (reset && this._scoped) await this.delete()\n\n        if (!scene['_scene']) {\n            throw new MtArgumentError('Cannot enter a non-scene Dispatcher')\n        }\n\n        if (!scene['_parent']) {\n            throw new MtArgumentError('This scene has not been registered')\n        }\n\n        this._scene = scene['_scene']\n        this._scoped = scene['_sceneScoped']\n        this._updateLocalKey()\n\n        await this._storage.setCurrentScene(this._key, this._scene, ttl)\n\n        if (with_) {\n            if (scene['_customStateKeyDelegate']) {\n                throw new MtArgumentError('Cannot use `with` parameter when the scene uses a custom state key delegate')\n            }\n\n            await scene.getState(this._key).set(with_, ttl)\n        }\n    }\n\n    /**\n     * Exit from current scene to the root\n     *\n     * @param reset\n     *     Whether to reset scene state (only applicable in case this is a scoped scene)\n     */\n    async exit(reset = true): Promise<void> {\n        if (reset && this._scoped) await this.delete()\n        this._scene = null\n        this._updateLocalKey()\n        await this._storage.deleteCurrentScene(this._key)\n    }\n\n    /**\n     * Rate limit some handler.\n     *\n     * When the rate limit exceeds, {@link RateLimitError} is thrown.\n     *\n     * This is a simple rate-limiting solution that uses\n     * the same key as the state. If you need something more\n     * sophisticated and/or customizable, you'll have to implement\n     * your own rate-limiter.\n     *\n     * > **Note**: `key` is used to prefix the local key\n     * > derived using the given key delegate.\n     *\n     * @param key  Key of the rate limit\n     * @param limit  Maximum number of requests in `window`\n     * @param window  Window size in seconds\n     * @returns  Tuple containing the number of remaining and\n     *   unix time in ms when the user can try again\n     */\n    async rateLimit(key: string, limit: number, window: number): Promise<[number, number]> {\n        const [remaining, reset] = await this._localStorage.getRateLimit(`${key}:${this._localKey}`, limit, window)\n\n        if (!remaining) {\n            throw new RateLimitError(reset)\n        }\n\n        return [remaining - 1, reset]\n    }\n\n    /**\n     * Throttle some handler.\n     *\n     * When the rate limit exceeds, this function waits for it to reset.\n     *\n     * This is a simple wrapper over {@link rateLimit}, and follows the same logic.\n     *\n     * > **Note**: `key` is used to prefix the local key\n     * > derived using the given key delegate.\n     *\n     * @param key  Key of the rate limit\n     * @param limit  Maximum number of requests in `window`\n     * @param window  Window size in seconds\n     * @returns  Tuple containing the number of remaining and\n     *   unix time in ms when the user can try again\n     */\n    async throttle(key: string, limit: number, window: number): Promise<[number, number]> {\n        try {\n            return await this.rateLimit(key, limit, window)\n        } catch (e: unknown) {\n            if (e instanceof RateLimitError) {\n                await sleep(e.reset - Date.now())\n\n                return this.throttle(key, limit, window)\n            }\n\n            throw e\n        }\n    }\n\n    /**\n     * Reset the rate limit\n     *\n     * @param key  Key of the rate limit\n     */\n    async resetRateLimit(key: string): Promise<void> {\n        await this._localStorage.resetRateLimit(`${key}:${this._localKey}`)\n    }\n}\n"]}