@mtcute/core 0.8.0 → 0.9.1

package/esm/storage/index.d.ts

@@ -2,4 +2,5 @@ export * from './driver.js';
 export * from './memory/index.js';
 export * from './provider.js';
 export * from './repository/index.js';
+export * from './sqlite/index.js';
 export * from './storage.js';

package/esm/storage/index.js

@@ -2,5 +2,6 @@ export * from './driver.js';
 export * from './memory/index.js';
 export * from './provider.js';
 export * from './repository/index.js';
+export * from './sqlite/index.js';
 export * from './storage.js';
 //# sourceMappingURL=index.js.map

package/esm/storage/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/storage/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAA;AAC3B,cAAc,mBAAmB,CAAA;AACjC,cAAc,eAAe,CAAA;AAC7B,cAAc,uBAAuB,CAAA;AACrC,cAAc,cAAc,CAAA","sourcesContent":["export * from './driver.js'\nexport * from './memory/index.js'\nexport * from './provider.js'\nexport * from './repository/index.js'\nexport * from './storage.js'\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/storage/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAA;AAC3B,cAAc,mBAAmB,CAAA;AACjC,cAAc,eAAe,CAAA;AAC7B,cAAc,uBAAuB,CAAA;AACrC,cAAc,mBAAmB,CAAA;AACjC,cAAc,cAAc,CAAA","sourcesContent":["export * from './driver.js'\nexport * from './memory/index.js'\nexport * from './provider.js'\nexport * from './repository/index.js'\nexport * from './sqlite/index.js'\nexport * from './storage.js'\n"]}

package/esm/storage/sqlite/driver.d.ts

@@ -0,0 +1,24 @@
+import { BaseStorageDriver } from '../driver.js';
+import { ISqliteDatabase, ISqliteStatement } from './types.js';
+type MigrationFunction = (db: ISqliteDatabase) => void;
+export declare abstract class BaseSqliteStorageDriver extends BaseStorageDriver {
+    db: ISqliteDatabase;
+    private _pending;
+    private _runMany;
+    private _cleanup?;
+    private _migrations;
+    private _maxVersion;
+    private _legacyMigrations;
+    registerLegacyMigration(repo: string, migration: MigrationFunction): void;
+    registerMigration(repo: string, version: number, migration: MigrationFunction): void;
+    private _onLoad;
+    onLoad(cb: (db: ISqliteDatabase) => void): void;
+    _writeLater(stmt: ISqliteStatement, params: unknown[]): void;
+    private _runLegacyMigrations;
+    _initialize(): void;
+    abstract _createDatabase(): ISqliteDatabase;
+    _load(): void;
+    _save(): void;
+    _destroy(): void;
+}
+export {};

package/esm/storage/sqlite/driver.js

@@ -0,0 +1,125 @@
+import { getPlatform } from '../../platform.js';
+import { BaseStorageDriver } from '../driver.js';
+const MIGRATIONS_TABLE_NAME = 'mtcute_migrations';
+const MIGRATIONS_TABLE_SQL = `
+create table if not exists ${MIGRATIONS_TABLE_NAME} (
+    repo text not null primary key,
+    version integer not null
+);
+`.trim();
+export class BaseSqliteStorageDriver extends BaseStorageDriver {
+    constructor() {
+        super(...arguments);
+        this._pending = [];
+        this._migrations = new Map();
+        this._maxVersion = new Map();
+        // todo: remove in 1.0.0
+        this._legacyMigrations = new Map();
+        this._onLoad = new Set();
+        this._runLegacyMigrations = false;
+    }
+    registerLegacyMigration(repo, migration) {
+        if (this.loaded) {
+            throw new Error('Cannot register migrations after loading');
+        }
+        this._legacyMigrations.set(repo, migration);
+    }
+    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);
+        }
+    }
+    onLoad(cb) {
+        if (this.loaded) {
+            cb(this.db);
+        }
+        else {
+            this._onLoad.add(cb);
+        }
+    }
+    _writeLater(stmt, params) {
+        this._pending.push([stmt, params]);
+    }
+    _initialize() {
+        const hasLegacyTables = this.db
+            .prepare("select name from sqlite_master where type = 'table' and name = 'kv'")
+            .get();
+        if (hasLegacyTables) {
+            this._log.info('legacy tables detected, will run migrations');
+            this._runLegacyMigrations = true;
+        }
+        this.db.exec(MIGRATIONS_TABLE_SQL);
+        const writeVersion = this.db.prepare(`insert or replace into ${MIGRATIONS_TABLE_NAME} (repo, version) values (?, ?)`);
+        const getVersion = this.db.prepare(`select version from ${MIGRATIONS_TABLE_NAME} where repo = ?`);
+        const didUpgrade = new Set();
+        for (const repo of this._migrations.keys()) {
+            const res = getVersion.get(repo);
+            const startVersion = res?.version ?? 0;
+            let fromVersion = startVersion;
+            const migrations = this._migrations.get(repo);
+            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}`);
+                }
+                migration(this.db);
+                fromVersion = nextVersion;
+                didUpgrade.add(repo);
+            }
+            if (fromVersion !== startVersion) {
+                writeVersion.run(repo, targetVer);
+            }
+        }
+    }
+    _load() {
+        this.db = this._createDatabase();
+        this._runMany = this.db.transaction((stmts) => {
+            stmts.forEach((stmt) => {
+                stmt[0].run(stmt[1]);
+            });
+        });
+        this.db.transaction(() => this._initialize())();
+        this._cleanup = getPlatform().beforeExit(() => {
+            this._save();
+            this._destroy();
+        });
+        for (const cb of this._onLoad)
+            cb(this.db);
+        if (this._runLegacyMigrations) {
+            this.db.transaction(() => {
+                for (const migration of this._legacyMigrations.values()) {
+                    migration(this.db);
+                }
+                // in case _writeLater was used
+                this._runMany(this._pending);
+            })();
+        }
+    }
+    _save() {
+        if (!this._pending.length)
+            return;
+        this._runMany(this._pending);
+        this._pending = [];
+    }
+    _destroy() {
+        this.db.close();
+        this._cleanup?.();
+        this._cleanup = undefined;
+    }
+}
+//# sourceMappingURL=driver.js.map

package/esm/storage/sqlite/driver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"driver.js","sourceRoot":"","sources":["../../../../src/storage/sqlite/driver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAA;AAGhD,MAAM,qBAAqB,GAAG,mBAAmB,CAAA;AACjD,MAAM,oBAAoB,GAAG;6BACA,qBAAqB;;;;CAIjD,CAAC,IAAI,EAAE,CAAA;AAIR,MAAM,OAAgB,uBAAwB,SAAQ,iBAAiB;IAAvE;;QAGY,aAAQ,GAAoC,EAAE,CAAA;QAI9C,gBAAW,GAAgD,IAAI,GAAG,EAAE,CAAA;QACpE,gBAAW,GAAwB,IAAI,GAAG,EAAE,CAAA;QAEpD,wBAAwB;QAChB,sBAAiB,GAAmC,IAAI,GAAG,EAAE,CAAA;QAmC7D,YAAO,GAAG,IAAI,GAAG,EAAiC,CAAA;QAclD,yBAAoB,GAAG,KAAK,CAAA;IA6FxC,CAAC;IA5IG,uBAAuB,CAAC,IAAY,EAAE,SAA4B;QAC9D,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;SAC9D;QAED,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAA;IAC/C,CAAC;IAED,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;IAID,MAAM,CAAC,EAAiC;QACpC,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;SACd;aAAM;YACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;SACvB;IACL,CAAC;IAED,WAAW,CAAC,IAAsB,EAAE,MAAiB;QACjD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAA;IACtC,CAAC;IAID,WAAW;QACP,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE;aAC1B,OAAO,CAAC,qEAAqE,CAAC;aAC9E,GAAG,EAAE,CAAA;QAEV,IAAI,eAAe,EAAE;YACjB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAA;YAC7D,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAA;SACnC;QAED,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAA;QAElC,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAChC,0BAA0B,qBAAqB,gCAAgC,CAClF,CAAA;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uBAAuB,qBAAqB,iBAAiB,CAAC,CAAA;QAEjG,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAA;QAEpC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE;YACxC,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,IAAI,CAAoC,CAAA;YAEnE,MAAM,YAAY,GAAG,GAAG,EAAE,OAAO,IAAI,CAAC,CAAA;YACtC,IAAI,WAAW,GAAG,YAAY,CAAA;YAE9B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAE,CAAA;YAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAE,CAAA;YAE7C,OAAO,WAAW,GAAG,SAAS,EAAE;gBAC5B,MAAM,WAAW,GAAG,WAAW,GAAG,CAAC,CAAA;gBACnC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;gBAE7C,IAAI,CAAC,SAAS,EAAE;oBACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,eAAe,WAAW,EAAE,CAAC,CAAA;iBACxE;gBAED,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;gBAElB,WAAW,GAAG,WAAW,CAAA;gBACzB,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;aACvB;YAED,IAAI,WAAW,KAAK,YAAY,EAAE;gBAC9B,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAA;aACpC;SACJ;IACL,CAAC;IAID,KAAK;QACD,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,eAAe,EAAE,CAAA;QAEhC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,KAAsC,EAAE,EAAE;YAC3E,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACnB,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;YACxB,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,CAAA;QAE/C,IAAI,CAAC,QAAQ,GAAG,WAAW,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE;YAC1C,IAAI,CAAC,KAAK,EAAE,CAAA;YACZ,IAAI,CAAC,QAAQ,EAAE,CAAA;QACnB,CAAC,CAAC,CAAA;QACF,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO;YAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAE1C,IAAI,IAAI,CAAC,oBAAoB,EAAE;YAC3B,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;gBACrB,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,EAAE;oBACrD,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;iBACrB;gBAED,+BAA+B;gBAC/B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAChC,CAAC,CAAC,EAAE,CAAA;SACP;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM;YAAE,OAAM;QAEjC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC5B,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAA;IACtB,CAAC;IAED,QAAQ;QACJ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAA;QACf,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAA;QACjB,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAA;IAC7B,CAAC;CACJ","sourcesContent":["import { getPlatform } from '../../platform.js'\nimport { BaseStorageDriver } from '../driver.js'\nimport { ISqliteDatabase, ISqliteStatement } from './types.js'\n\nconst MIGRATIONS_TABLE_NAME = 'mtcute_migrations'\nconst MIGRATIONS_TABLE_SQL = `\ncreate table if not exists ${MIGRATIONS_TABLE_NAME} (\n    repo text not null primary key,\n    version integer not null\n);\n`.trim()\n\ntype MigrationFunction = (db: ISqliteDatabase) => void\n\nexport abstract class BaseSqliteStorageDriver extends BaseStorageDriver {\n    db!: ISqliteDatabase\n\n    private _pending: [ISqliteStatement, unknown[]][] = []\n    private _runMany!: (stmts: [ISqliteStatement, unknown[]][]) => void\n    private _cleanup?: () => void\n\n    private _migrations: Map<string, Map<number, MigrationFunction>> = new Map()\n    private _maxVersion: Map<string, number> = new Map()\n\n    // todo: remove in 1.0.0\n    private _legacyMigrations: Map<string, MigrationFunction> = new Map()\n\n    registerLegacyMigration(repo: string, migration: MigrationFunction): void {\n        if (this.loaded) {\n            throw new Error('Cannot register migrations after loading')\n        }\n\n        this._legacyMigrations.set(repo, migration)\n    }\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    private _onLoad = new Set<(db: ISqliteDatabase) => void>()\n\n    onLoad(cb: (db: ISqliteDatabase) => void): void {\n        if (this.loaded) {\n            cb(this.db)\n        } else {\n            this._onLoad.add(cb)\n        }\n    }\n\n    _writeLater(stmt: ISqliteStatement, params: unknown[]): void {\n        this._pending.push([stmt, params])\n    }\n\n    private _runLegacyMigrations = false\n\n    _initialize(): void {\n        const hasLegacyTables = this.db\n            .prepare(\"select name from sqlite_master where type = 'table' and name = 'kv'\")\n            .get()\n\n        if (hasLegacyTables) {\n            this._log.info('legacy tables detected, will run migrations')\n            this._runLegacyMigrations = true\n        }\n\n        this.db.exec(MIGRATIONS_TABLE_SQL)\n\n        const writeVersion = this.db.prepare(\n            `insert or replace into ${MIGRATIONS_TABLE_NAME} (repo, version) values (?, ?)`,\n        )\n        const getVersion = this.db.prepare(`select version from ${MIGRATIONS_TABLE_NAME} where repo = ?`)\n\n        const didUpgrade = new Set<string>()\n\n        for (const repo of this._migrations.keys()) {\n            const res = getVersion.get(repo) as { version: number } | undefined\n\n            const startVersion = res?.version ?? 0\n            let fromVersion = startVersion\n\n            const migrations = this._migrations.get(repo)!\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                migration(this.db)\n\n                fromVersion = nextVersion\n                didUpgrade.add(repo)\n            }\n\n            if (fromVersion !== startVersion) {\n                writeVersion.run(repo, targetVer)\n            }\n        }\n    }\n\n    abstract _createDatabase(): ISqliteDatabase\n\n    _load(): void {\n        this.db = this._createDatabase()\n\n        this._runMany = this.db.transaction((stmts: [ISqliteStatement, unknown[]][]) => {\n            stmts.forEach((stmt) => {\n                stmt[0].run(stmt[1])\n            })\n        })\n\n        this.db.transaction(() => this._initialize())()\n\n        this._cleanup = getPlatform().beforeExit(() => {\n            this._save()\n            this._destroy()\n        })\n        for (const cb of this._onLoad) cb(this.db)\n\n        if (this._runLegacyMigrations) {\n            this.db.transaction(() => {\n                for (const migration of this._legacyMigrations.values()) {\n                    migration(this.db)\n                }\n\n                // in case _writeLater was used\n                this._runMany(this._pending)\n            })()\n        }\n    }\n\n    _save(): void {\n        if (!this._pending.length) return\n\n        this._runMany(this._pending)\n        this._pending = []\n    }\n\n    _destroy(): void {\n        this.db.close()\n        this._cleanup?.()\n        this._cleanup = undefined\n    }\n}\n"]}

package/esm/storage/sqlite/index.d.ts

@@ -0,0 +1,17 @@
+import { ITelegramStorageProvider } from '../../highlevel/storage/provider.js';
+import { IMtStorageProvider } from '../provider.js';
+import { BaseSqliteStorageDriver } from './driver.js';
+import { SqliteAuthKeysRepository } from './repository/auth-keys.js';
+import { SqliteKeyValueRepository } from './repository/kv.js';
+import { SqlitePeersRepository } from './repository/peers.js';
+import { SqliteRefMessagesRepository } from './repository/ref-messages.js';
+export { BaseSqliteStorageDriver };
+export * from './types.js';
+export declare class BaseSqliteStorage implements IMtStorageProvider, ITelegramStorageProvider {
+    readonly driver: BaseSqliteStorageDriver;
+    constructor(driver: BaseSqliteStorageDriver);
+    readonly authKeys: SqliteAuthKeysRepository;
+    readonly kv: SqliteKeyValueRepository;
+    readonly refMessages: SqliteRefMessagesRepository;
+    readonly peers: SqlitePeersRepository;
+}

package/esm/storage/sqlite/index.js

@@ -0,0 +1,17 @@
+import { BaseSqliteStorageDriver } from './driver.js';
+import { SqliteAuthKeysRepository } from './repository/auth-keys.js';
+import { SqliteKeyValueRepository } from './repository/kv.js';
+import { SqlitePeersRepository } from './repository/peers.js';
+import { SqliteRefMessagesRepository } from './repository/ref-messages.js';
+export { BaseSqliteStorageDriver };
+export * from './types.js';
+export class BaseSqliteStorage {
+    constructor(driver) {
+        this.driver = driver;
+        this.authKeys = new SqliteAuthKeysRepository(this.driver);
+        this.kv = new SqliteKeyValueRepository(this.driver);
+        this.refMessages = new SqliteRefMessagesRepository(this.driver);
+        this.peers = new SqlitePeersRepository(this.driver);
+    }
+}
+//# sourceMappingURL=index.js.map

package/esm/storage/sqlite/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/storage/sqlite/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAA;AACrD,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAA;AACpE,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAA;AAC7D,OAAO,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAC7D,OAAO,EAAE,2BAA2B,EAAE,MAAM,8BAA8B,CAAA;AAE1E,OAAO,EAAE,uBAAuB,EAAE,CAAA;AAClC,cAAc,YAAY,CAAA;AAE1B,MAAM,OAAO,iBAAiB;IAC1B,YAAqB,MAA+B;QAA/B,WAAM,GAAN,MAAM,CAAyB;QAE3C,aAAQ,GAAG,IAAI,wBAAwB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACpD,OAAE,GAAG,IAAI,wBAAwB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAC9C,gBAAW,GAAG,IAAI,2BAA2B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAC1D,UAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IALA,CAAC;CAM3D","sourcesContent":["import { ITelegramStorageProvider } from '../../highlevel/storage/provider.js'\nimport { IMtStorageProvider } from '../provider.js'\nimport { BaseSqliteStorageDriver } from './driver.js'\nimport { SqliteAuthKeysRepository } from './repository/auth-keys.js'\nimport { SqliteKeyValueRepository } from './repository/kv.js'\nimport { SqlitePeersRepository } from './repository/peers.js'\nimport { SqliteRefMessagesRepository } from './repository/ref-messages.js'\n\nexport { BaseSqliteStorageDriver }\nexport * from './types.js'\n\nexport class BaseSqliteStorage implements IMtStorageProvider, ITelegramStorageProvider {\n    constructor(readonly driver: BaseSqliteStorageDriver) {}\n\n    readonly authKeys = new SqliteAuthKeysRepository(this.driver)\n    readonly kv = new SqliteKeyValueRepository(this.driver)\n    readonly refMessages = new SqliteRefMessagesRepository(this.driver)\n    readonly peers = new SqlitePeersRepository(this.driver)\n}\n"]}

package/esm/storage/sqlite/repository/auth-keys.d.ts

@@ -0,0 +1,20 @@
+import { IAuthKeysRepository } from '../../repository/auth-keys.js';
+import { BaseSqliteStorageDriver } from '../driver.js';
+export declare class SqliteAuthKeysRepository implements IAuthKeysRepository {
+    readonly _driver: BaseSqliteStorageDriver;
+    constructor(_driver: BaseSqliteStorageDriver);
+    private _set;
+    private _del;
+    set(dc: number, key: Uint8Array | null): void;
+    private _get;
+    get(dc: number): Uint8Array | null;
+    private _setTemp;
+    private _delTemp;
+    setTemp(dc: number, idx: number, key: Uint8Array | null, expires: number): void;
+    private _getTemp;
+    getTemp(dc: number, idx: number, now: number): Uint8Array | null;
+    private _delTempAll;
+    deleteByDc(dc: number): void;
+    private _delAll;
+    deleteAll(): void;
+}

package/esm/storage/sqlite/repository/auth-keys.js

@@ -0,0 +1,64 @@
+export class SqliteAuthKeysRepository {
+    constructor(_driver) {
+        this._driver = _driver;
+        _driver.registerMigration('auth_keys', 1, (db) => {
+            db.exec(`
+                create table if not exists auth_keys (
+                    dc integer primary key,
+                    key blob not null
+                );
+                create table if not exists temp_auth_keys (
+                    dc integer not null,
+                    idx integer not null,
+                    key blob not null,
+                    expires integer not null,
+                    primary key (dc, idx)
+                );
+            `);
+        });
+        _driver.onLoad((db) => {
+            this._get = db.prepare('select key from auth_keys where dc = ?');
+            this._getTemp = db.prepare('select key from temp_auth_keys where dc = ? and idx = ? and expires > ?');
+            this._set = db.prepare('insert or replace into auth_keys (dc, key) values (?, ?)');
+            this._setTemp = this._driver.db.prepare('insert or replace into temp_auth_keys (dc, idx, key, expires) values (?, ?, ?, ?)');
+            this._del = db.prepare('delete from auth_keys where dc = ?');
+            this._delTemp = db.prepare('delete from temp_auth_keys where dc = ? and idx = ?');
+            this._delTempAll = db.prepare('delete from temp_auth_keys where dc = ?');
+            this._delAll = db.prepare('delete from auth_keys');
+        });
+    }
+    set(dc, key) {
+        if (!key) {
+            this._del.run(dc);
+            return;
+        }
+        this._set.run(dc, key);
+    }
+    get(dc) {
+        const row = this._get.get(dc);
+        if (!row)
+            return null;
+        return row.key;
+    }
+    setTemp(dc, idx, key, expires) {
+        if (!key) {
+            this._delTemp.run(dc, idx);
+            return;
+        }
+        this._setTemp.run(dc, idx, key, expires);
+    }
+    getTemp(dc, idx, now) {
+        const row = this._getTemp.get(dc, idx, now);
+        if (!row)
+            return null;
+        return row.key;
+    }
+    deleteByDc(dc) {
+        this._del.run(dc);
+        this._delTempAll.run(dc);
+    }
+    deleteAll() {
+        this._delAll.run();
+    }
+}
+//# sourceMappingURL=auth-keys.js.map

package/esm/storage/sqlite/repository/auth-keys.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth-keys.js","sourceRoot":"","sources":["../../../../../src/storage/sqlite/repository/auth-keys.ts"],"names":[],"mappings":"AAcA,MAAM,OAAO,wBAAwB;IACjC,YAAqB,OAAgC;QAAhC,YAAO,GAAP,OAAO,CAAyB;QACjD,OAAO,CAAC,iBAAiB,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE;YAC7C,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;aAYP,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE;YAClB,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAA;YAChE,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,yEAAyE,CAAC,CAAA;YAErG,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,0DAA0D,CAAC,CAAA;YAClF,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CACnC,mFAAmF,CACtF,CAAA;YAED,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAA;YAC5D,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAA;YACjF,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAA;YACxE,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;IACN,CAAC;IAID,GAAG,CAAC,EAAU,EAAE,GAAsB;QAClC,IAAI,CAAC,GAAG,EAAE;YACN,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YAEjB,OAAM;SACT;QAED,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;IAC1B,CAAC;IAGD,GAAG,CAAC,EAAU;QACV,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAC7B,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAQ,GAAkB,CAAC,GAAG,CAAA;IAClC,CAAC;IAID,OAAO,CAAC,EAAU,EAAE,GAAW,EAAE,GAAsB,EAAE,OAAe;QACpE,IAAI,CAAC,GAAG,EAAE;YACN,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;YAE1B,OAAM;SACT;QAED,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC5C,CAAC;IAGD,OAAO,CAAC,EAAU,EAAE,GAAW,EAAE,GAAW;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;QAC3C,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAQ,GAAsB,CAAC,GAAG,CAAA;IACtC,CAAC;IAGD,UAAU,CAAC,EAAU;QACjB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACjB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAC5B,CAAC;IAGD,SAAS;QACL,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;IACtB,CAAC;CACJ","sourcesContent":["import { IAuthKeysRepository } from '../../repository/auth-keys.js'\nimport { BaseSqliteStorageDriver } from '../driver.js'\nimport { ISqliteStatement } from '../types.js'\n\ninterface AuthKeyDto {\n    dc: number\n    key: Uint8Array\n}\n\ninterface TempAuthKeyDto extends AuthKeyDto {\n    expires?: number\n    idx?: number\n}\n\nexport class SqliteAuthKeysRepository implements IAuthKeysRepository {\n    constructor(readonly _driver: BaseSqliteStorageDriver) {\n        _driver.registerMigration('auth_keys', 1, (db) => {\n            db.exec(`\n                create table if not exists auth_keys (\n                    dc integer primary key,\n                    key blob not null\n                );\n                create table if not exists temp_auth_keys (\n                    dc integer not null,\n                    idx integer not null,\n                    key blob not null,\n                    expires integer not null,\n                    primary key (dc, idx)\n                );\n            `)\n        })\n        _driver.onLoad((db) => {\n            this._get = db.prepare('select key from auth_keys where dc = ?')\n            this._getTemp = db.prepare('select key from temp_auth_keys where dc = ? and idx = ? and expires > ?')\n\n            this._set = db.prepare('insert or replace into auth_keys (dc, key) values (?, ?)')\n            this._setTemp = this._driver.db.prepare(\n                'insert or replace into temp_auth_keys (dc, idx, key, expires) values (?, ?, ?, ?)',\n            )\n\n            this._del = db.prepare('delete from auth_keys where dc = ?')\n            this._delTemp = db.prepare('delete from temp_auth_keys where dc = ? and idx = ?')\n            this._delTempAll = db.prepare('delete from temp_auth_keys where dc = ?')\n            this._delAll = db.prepare('delete from auth_keys')\n        })\n    }\n\n    private _set!: ISqliteStatement\n    private _del!: ISqliteStatement\n    set(dc: number, key: Uint8Array | null): void {\n        if (!key) {\n            this._del.run(dc)\n\n            return\n        }\n\n        this._set.run(dc, key)\n    }\n\n    private _get!: ISqliteStatement\n    get(dc: number): Uint8Array | null {\n        const row = this._get.get(dc)\n        if (!row) return null\n\n        return (row as AuthKeyDto).key\n    }\n\n    private _setTemp!: ISqliteStatement\n    private _delTemp!: ISqliteStatement\n    setTemp(dc: number, idx: number, key: Uint8Array | null, expires: number): void {\n        if (!key) {\n            this._delTemp.run(dc, idx)\n\n            return\n        }\n\n        this._setTemp.run(dc, idx, key, expires)\n    }\n\n    private _getTemp!: ISqliteStatement\n    getTemp(dc: number, idx: number, now: number): Uint8Array | null {\n        const row = this._getTemp.get(dc, idx, now)\n        if (!row) return null\n\n        return (row as TempAuthKeyDto).key\n    }\n\n    private _delTempAll!: ISqliteStatement\n    deleteByDc(dc: number): void {\n        this._del.run(dc)\n        this._delTempAll.run(dc)\n    }\n\n    private _delAll!: ISqliteStatement\n    deleteAll(): void {\n        this._delAll.run()\n    }\n}\n"]}

package/esm/storage/sqlite/repository/kv.d.ts

@@ -0,0 +1,14 @@
+import { IKeyValueRepository } from '../../repository/key-value.js';
+import { BaseSqliteStorageDriver } from '../driver.js';
+export declare class SqliteKeyValueRepository implements IKeyValueRepository {
+    readonly _driver: BaseSqliteStorageDriver;
+    constructor(_driver: BaseSqliteStorageDriver);
+    private _set;
+    set(key: string, value: Uint8Array): void;
+    private _get;
+    get(key: string): Uint8Array | null;
+    private _del;
+    delete(key: string): void;
+    private _delAll;
+    deleteAll(): void;
+}

package/esm/storage/sqlite/repository/kv.js

@@ -0,0 +1,92 @@
+import { __tlReaderMap } from '@mtcute/tl/binary/reader.js';
+import { __tlWriterMap } from '@mtcute/tl/binary/writer.js';
+import { CurrentUserService } from '../../../highlevel/storage/service/current-user.js';
+import { UpdatesStateService } from '../../../highlevel/storage/service/updates.js';
+import { DefaultDcsService } from '../../service/default-dcs.js';
+export class SqliteKeyValueRepository {
+    constructor(_driver) {
+        this._driver = _driver;
+        _driver.registerMigration('kv', 1, (db) => {
+            db.exec(`
+                create table key_value (
+                    key text primary key,
+                    value blob not null
+                );
+            `);
+        });
+        _driver.onLoad((db) => {
+            this._get = db.prepare('select value from key_value where key = ?');
+            this._set = db.prepare('insert or replace into key_value (key, value) values (?, ?)');
+            this._del = db.prepare('delete from key_value where key = ?');
+            this._delAll = db.prepare('delete from key_value');
+        });
+        // awkward dependencies, unsafe code, awful crutches
+        // all in the name of backwards compatibility
+        /* eslint-disable @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any */
+        /* eslint-disable @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-floating-promises */
+        /* eslint-disable @typescript-eslint/no-unsafe-argument */
+        _driver.registerLegacyMigration('kv', (db) => {
+            // fetch all values from the old table
+            const all = db.prepare('select key, value from kv').all();
+            const obj = {};
+            for (const { key, value } of all) {
+                obj[key] = JSON.parse(value);
+            }
+            db.exec('drop table kv');
+            // lol
+            const options = {
+                driver: this._driver,
+                readerMap: __tlReaderMap,
+                writerMap: __tlWriterMap,
+                // eslint-disable-next-line dot-notation
+                log: this._driver['_log'],
+            };
+            if (obj.self) {
+                new CurrentUserService(this, options).store({
+                    userId: obj.self.userId,
+                    isBot: obj.self.isBot,
+                    isPremium: false,
+                    usernames: [],
+                });
+            }
+            if (obj.pts) {
+                const svc = new UpdatesStateService(this, options);
+                svc.setPts(obj.pts);
+                if (obj.qts)
+                    svc.setQts(obj.qts);
+                if (obj.date)
+                    svc.setDate(obj.date);
+                if (obj.seq)
+                    svc.setSeq(obj.seq);
+                // also fetch channel states. they were moved to kv from a separate table
+                const channels = db.prepare('select * from pts').all();
+                for (const channel of channels) {
+                    svc.setChannelPts(channel.channel_id, channel.pts);
+                }
+            }
+            db.exec('drop table pts');
+            if (obj.def_dc) {
+                new DefaultDcsService(this, options).store(obj.def_dc);
+            }
+        });
+        /* eslint-enable @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any */
+        /* eslint-enable @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-floating-promises */
+        /* eslint-enable @typescript-eslint/no-unsafe-argument */
+    }
+    set(key, value) {
+        this._driver._writeLater(this._set, [key, value]);
+    }
+    get(key) {
+        const res = this._get.get(key);
+        if (!res)
+            return null;
+        return res.value;
+    }
+    delete(key) {
+        this._del.run(key);
+    }
+    deleteAll() {
+        this._delAll.run();
+    }
+}
+//# sourceMappingURL=kv.js.map

package/esm/storage/sqlite/repository/kv.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kv.js","sourceRoot":"","sources":["../../../../../src/storage/sqlite/repository/kv.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAA;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAA;AAE3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,oDAAoD,CAAA;AACvF,OAAO,EAAE,mBAAmB,EAAE,MAAM,+CAA+C,CAAA;AAGnF,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAA;AAShE,MAAM,OAAO,wBAAwB;IACjC,YAAqB,OAAgC;QAAhC,YAAO,GAAP,OAAO,CAAyB;QACjD,OAAO,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE;YACtC,EAAE,CAAC,IAAI,CAAC;;;;;aAKP,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE;YAClB,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,2CAA2C,CAAC,CAAA;YACnE,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,6DAA6D,CAAC,CAAA;YACrF,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAA;YAC7D,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;QAEF,oDAAoD;QACpD,6CAA6C;QAC7C,mGAAmG;QACnG,qGAAqG;QACrG,0DAA0D;QAC1D,OAAO,CAAC,uBAAuB,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;YACzC,sCAAsC;YACtC,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC,GAAG,EAAsC,CAAA;YAC7F,MAAM,GAAG,GAAwB,EAAE,CAAA;YAEnC,KAAK,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE;gBAC9B,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;aAC/B;YAED,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;YAExB,MAAM;YACN,MAAM,OAAO,GAAmB;gBAC5B,MAAM,EAAE,IAAI,CAAC,OAAO;gBACpB,SAAS,EAAE,aAAa;gBACxB,SAAS,EAAE,aAAa;gBACxB,wCAAwC;gBACxC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;aAC5B,CAAA;YAED,IAAI,GAAG,CAAC,IAAI,EAAE;gBACV,IAAI,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC;oBACxC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;oBACvB,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK;oBACrB,SAAS,EAAE,KAAK;oBAChB,SAAS,EAAE,EAAE;iBAChB,CAAC,CAAA;aACL;YAED,IAAI,GAAG,CAAC,GAAG,EAAE;gBACT,MAAM,GAAG,GAAG,IAAI,mBAAmB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;gBAClD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;gBACnB,IAAI,GAAG,CAAC,GAAG;oBAAE,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;gBAChC,IAAI,GAAG,CAAC,IAAI;oBAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBACnC,IAAI,GAAG,CAAC,GAAG;oBAAE,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;gBAEhC,yEAAyE;gBACzE,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAAG,EAAW,CAAA;gBAE/D,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;oBAC5B,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,CAAA;iBACrD;aACJ;YACD,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;YAEzB,IAAI,GAAG,CAAC,MAAM,EAAE;gBACZ,IAAI,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;aACzD;QACL,CAAC,CAAC,CAAA;QACF,kGAAkG;QAClG,oGAAoG;QACpG,yDAAyD;IAC7D,CAAC;IAGD,GAAG,CAAC,GAAW,EAAE,KAAiB;QAC9B,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAA;IACrD,CAAC;IAGD,GAAG,CAAC,GAAW;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QAC9B,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAQ,GAAmB,CAAC,KAAK,CAAA;IACrC,CAAC;IAGD,MAAM,CAAC,GAAW;QACd,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;IACtB,CAAC;IAGD,SAAS;QACL,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;IACtB,CAAC;CACJ","sourcesContent":["import { __tlReaderMap } from '@mtcute/tl/binary/reader.js'\nimport { __tlWriterMap } from '@mtcute/tl/binary/writer.js'\n\nimport { CurrentUserService } from '../../../highlevel/storage/service/current-user.js'\nimport { UpdatesStateService } from '../../../highlevel/storage/service/updates.js'\nimport { IKeyValueRepository } from '../../repository/key-value.js'\nimport { ServiceOptions } from '../../service/base.js'\nimport { DefaultDcsService } from '../../service/default-dcs.js'\nimport { BaseSqliteStorageDriver } from '../driver.js'\nimport { ISqliteStatement } from '../types.js'\n\ninterface KeyValueDto {\n    key: string\n    value: Uint8Array\n}\n\nexport class SqliteKeyValueRepository implements IKeyValueRepository {\n    constructor(readonly _driver: BaseSqliteStorageDriver) {\n        _driver.registerMigration('kv', 1, (db) => {\n            db.exec(`\n                create table key_value (\n                    key text primary key,\n                    value blob not null\n                );\n            `)\n        })\n        _driver.onLoad((db) => {\n            this._get = db.prepare('select value from key_value where key = ?')\n            this._set = db.prepare('insert or replace into key_value (key, value) values (?, ?)')\n            this._del = db.prepare('delete from key_value where key = ?')\n            this._delAll = db.prepare('delete from key_value')\n        })\n\n        // awkward dependencies, unsafe code, awful crutches\n        // all in the name of backwards compatibility\n        /* eslint-disable @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any */\n        /* eslint-disable @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-floating-promises */\n        /* eslint-disable @typescript-eslint/no-unsafe-argument */\n        _driver.registerLegacyMigration('kv', (db) => {\n            // fetch all values from the old table\n            const all = db.prepare('select key, value from kv').all() as { key: string; value: string }[]\n            const obj: Record<string, any> = {}\n\n            for (const { key, value } of all) {\n                obj[key] = JSON.parse(value)\n            }\n\n            db.exec('drop table kv')\n\n            // lol\n            const options: ServiceOptions = {\n                driver: this._driver,\n                readerMap: __tlReaderMap,\n                writerMap: __tlWriterMap,\n                // eslint-disable-next-line dot-notation\n                log: this._driver['_log'],\n            }\n\n            if (obj.self) {\n                new CurrentUserService(this, options).store({\n                    userId: obj.self.userId,\n                    isBot: obj.self.isBot,\n                    isPremium: false,\n                    usernames: [],\n                })\n            }\n\n            if (obj.pts) {\n                const svc = new UpdatesStateService(this, options)\n                svc.setPts(obj.pts)\n                if (obj.qts) svc.setQts(obj.qts)\n                if (obj.date) svc.setDate(obj.date)\n                if (obj.seq) svc.setSeq(obj.seq)\n\n                // also fetch channel states. they were moved to kv from a separate table\n                const channels = db.prepare('select * from pts').all() as any[]\n\n                for (const channel of channels) {\n                    svc.setChannelPts(channel.channel_id, channel.pts)\n                }\n            }\n            db.exec('drop table pts')\n\n            if (obj.def_dc) {\n                new DefaultDcsService(this, options).store(obj.def_dc)\n            }\n        })\n        /* eslint-enable @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any */\n        /* eslint-enable @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-floating-promises */\n        /* eslint-enable @typescript-eslint/no-unsafe-argument */\n    }\n\n    private _set!: ISqliteStatement\n    set(key: string, value: Uint8Array): void {\n        this._driver._writeLater(this._set, [key, value])\n    }\n\n    private _get!: ISqliteStatement\n    get(key: string): Uint8Array | null {\n        const res = this._get.get(key)\n        if (!res) return null\n\n        return (res as KeyValueDto).value\n    }\n\n    private _del!: ISqliteStatement\n    delete(key: string): void {\n        this._del.run(key)\n    }\n\n    private _delAll!: ISqliteStatement\n    deleteAll(): void {\n        this._delAll.run()\n    }\n}\n"]}

package/esm/storage/sqlite/repository/peers.d.ts

@@ -0,0 +1,16 @@
+import { IPeersRepository } from '../../../highlevel/storage/repository/peers.js';
+import { BaseSqliteStorageDriver } from '../driver.js';
+export declare class SqlitePeersRepository implements IPeersRepository {
+    readonly _driver: BaseSqliteStorageDriver;
+    constructor(_driver: BaseSqliteStorageDriver);
+    private _store;
+    store(peer: IPeersRepository.PeerInfo): void;
+    private _getById;
+    getById(id: number): IPeersRepository.PeerInfo | null;
+    private _getByUsername;
+    getByUsername(username: string): IPeersRepository.PeerInfo | null;
+    private _getByPhone;
+    getByPhone(phone: string): IPeersRepository.PeerInfo | null;
+    private _delAll;
+    deleteAll(): void;
+}

package/esm/storage/sqlite/repository/peers.js

@@ -0,0 +1,73 @@
+function mapPeerDto(dto) {
+    return {
+        id: dto.id,
+        accessHash: dto.hash,
+        usernames: JSON.parse(dto.usernames),
+        updated: dto.updated,
+        phone: dto.phone || undefined,
+        complete: dto.complete,
+    };
+}
+export class SqlitePeersRepository {
+    constructor(_driver) {
+        this._driver = _driver;
+        _driver.registerMigration('peers', 1, (db) => {
+            db.exec(`
+                create table peers (
+                    id integer primary key,
+                    hash text not null,
+                    usernames json not null,
+                    updated integer not null,
+                    phone text,
+                    complete blob
+                );
+                create index idx_peers_usernames on peers (usernames);
+                create index idx_peers_phone on peers (phone);
+            `);
+        });
+        _driver.onLoad((db) => {
+            this._store = db.prepare('insert or replace into peers (id, hash, usernames, updated, phone, complete) values (?, ?, ?, ?, ?, ?)');
+            this._getById = db.prepare('select * from peers where id = ?');
+            this._getByUsername = db.prepare('select * from peers where exists (select 1 from json_each(usernames) where value = ?)');
+            this._getByPhone = db.prepare('select * from peers where phone = ?');
+            this._delAll = db.prepare('delete from peers');
+        });
+        _driver.registerLegacyMigration('peers', (db) => {
+            // not too important information, just drop the table
+            db.exec('drop table entities');
+        });
+    }
+    store(peer) {
+        this._driver._writeLater(this._store, [
+            peer.id,
+            peer.accessHash,
+            // add commas to make it easier to search with LIKE
+            JSON.stringify(peer.usernames),
+            peer.updated,
+            peer.phone,
+            peer.complete,
+        ]);
+    }
+    getById(id) {
+        const row = this._getById.get(id);
+        if (!row)
+            return null;
+        return mapPeerDto(row);
+    }
+    getByUsername(username) {
+        const row = this._getByUsername.get(username);
+        if (!row)
+            return null;
+        return mapPeerDto(row);
+    }
+    getByPhone(phone) {
+        const row = this._getByPhone.get(phone);
+        if (!row)
+            return null;
+        return mapPeerDto(row);
+    }
+    deleteAll() {
+        this._delAll.run();
+    }
+}
+//# sourceMappingURL=peers.js.map

package/esm/storage/sqlite/repository/peers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"peers.js","sourceRoot":"","sources":["../../../../../src/storage/sqlite/repository/peers.ts"],"names":[],"mappings":"AAcA,SAAS,UAAU,CAAC,GAAY;IAC5B,OAAO;QACH,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,UAAU,EAAE,GAAG,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAa;QAChD,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,SAAS;QAC7B,QAAQ,EAAE,GAAG,CAAC,QAAQ;KACzB,CAAA;AACL,CAAC;AAED,MAAM,OAAO,qBAAqB;IAC9B,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,CAAC,EAAE,EAAE,EAAE;YAClB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,OAAO,CACpB,wGAAwG,CAC3G,CAAA;YAED,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAA;YAC9D,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC,OAAO,CAC5B,uFAAuF,CAC1F,CAAA;YACD,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAA;YAEpE,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,uBAAuB,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE;YAC5C,qDAAqD;YACrD,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAA;QAClC,CAAC,CAAC,CAAA;IACN,CAAC;IAGD,KAAK,CAAC,IAA+B;QACjC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE;YAClC,IAAI,CAAC,EAAE;YACP,IAAI,CAAC,UAAU;YACf,mDAAmD;YACnD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;YAC9B,IAAI,CAAC,OAAO;YACZ,IAAI,CAAC,KAAK;YACV,IAAI,CAAC,QAAQ;SAChB,CAAC,CAAA;IACN,CAAC;IAGD,OAAO,CAAC,EAAU;QACd,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACjC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAO,UAAU,CAAC,GAAc,CAAC,CAAA;IACrC,CAAC;IAGD,aAAa,CAAC,QAAgB;QAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAC7C,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAO,UAAU,CAAC,GAAc,CAAC,CAAA;IACrC,CAAC;IAGD,UAAU,CAAC,KAAa;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACvC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAO,UAAU,CAAC,GAAc,CAAC,CAAA;IACrC,CAAC;IAGD,SAAS;QACL,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;IACtB,CAAC;CACJ","sourcesContent":["import { IPeersRepository } from '../../../highlevel/storage/repository/peers.js'\nimport { BaseSqliteStorageDriver } from '../driver.js'\nimport { ISqliteStatement } from '../types.js'\n\ninterface PeerDto {\n    id: number\n    hash: string\n    usernames: string\n    updated: number\n    phone: string | null\n    // eslint-disable-next-line no-restricted-globals\n    complete: Buffer\n}\n\nfunction mapPeerDto(dto: PeerDto): IPeersRepository.PeerInfo {\n    return {\n        id: dto.id,\n        accessHash: dto.hash,\n        usernames: JSON.parse(dto.usernames) as string[],\n        updated: dto.updated,\n        phone: dto.phone || undefined,\n        complete: dto.complete,\n    }\n}\n\nexport class SqlitePeersRepository implements IPeersRepository {\n    constructor(readonly _driver: BaseSqliteStorageDriver) {\n        _driver.registerMigration('peers', 1, (db) => {\n            db.exec(`\n                create table peers (\n                    id integer primary key,\n                    hash text not null,\n                    usernames json not null,\n                    updated integer not null,\n                    phone text,\n                    complete blob\n                );\n                create index idx_peers_usernames on peers (usernames);\n                create index idx_peers_phone on peers (phone);\n            `)\n        })\n        _driver.onLoad((db) => {\n            this._store = db.prepare(\n                'insert or replace into peers (id, hash, usernames, updated, phone, complete) values (?, ?, ?, ?, ?, ?)',\n            )\n\n            this._getById = db.prepare('select * from peers where id = ?')\n            this._getByUsername = db.prepare(\n                'select * from peers where exists (select 1 from json_each(usernames) where value = ?)',\n            )\n            this._getByPhone = db.prepare('select * from peers where phone = ?')\n\n            this._delAll = db.prepare('delete from peers')\n        })\n        _driver.registerLegacyMigration('peers', (db) => {\n            // not too important information, just drop the table\n            db.exec('drop table entities')\n        })\n    }\n\n    private _store!: ISqliteStatement\n    store(peer: IPeersRepository.PeerInfo): void {\n        this._driver._writeLater(this._store, [\n            peer.id,\n            peer.accessHash,\n            // add commas to make it easier to search with LIKE\n            JSON.stringify(peer.usernames),\n            peer.updated,\n            peer.phone,\n            peer.complete,\n        ])\n    }\n\n    private _getById!: ISqliteStatement\n    getById(id: number): IPeersRepository.PeerInfo | null {\n        const row = this._getById.get(id)\n        if (!row) return null\n\n        return mapPeerDto(row as PeerDto)\n    }\n\n    private _getByUsername!: ISqliteStatement\n    getByUsername(username: string): IPeersRepository.PeerInfo | null {\n        const row = this._getByUsername.get(username)\n        if (!row) return null\n\n        return mapPeerDto(row as PeerDto)\n    }\n\n    private _getByPhone!: ISqliteStatement\n    getByPhone(phone: string): IPeersRepository.PeerInfo | null {\n        const row = this._getByPhone.get(phone)\n        if (!row) return null\n\n        return mapPeerDto(row as PeerDto)\n    }\n\n    private _delAll!: ISqliteStatement\n    deleteAll(): void {\n        this._delAll.run()\n    }\n}\n"]}

package/esm/storage/sqlite/repository/ref-messages.d.ts

@@ -0,0 +1,16 @@
+import { IReferenceMessagesRepository } from '@mtcute/core';
+import { BaseSqliteStorageDriver } from '../driver.js';
+export declare class SqliteRefMessagesRepository implements IReferenceMessagesRepository {
+    readonly _driver: BaseSqliteStorageDriver;
+    constructor(_driver: BaseSqliteStorageDriver);
+    private _store;
+    store(peerId: number, chatId: number, msgId: number): void;
+    private _getByPeer;
+    getByPeer(peerId: number): [number, number] | null;
+    private _del;
+    delete(chatId: number, msgIds: number[]): void;
+    private _delByPeer;
+    deleteByPeer(peerId: number): void;
+    private _delAll;
+    deleteAll(): void;
+}

package/esm/storage/sqlite/repository/ref-messages.js

@@ -0,0 +1,45 @@
+export class SqliteRefMessagesRepository {
+    constructor(_driver) {
+        this._driver = _driver;
+        _driver.registerMigration('ref_messages', 1, (db) => {
+            db.exec(`
+                create table if not exists message_refs (
+                    peer_id integer not null,
+                    chat_id integer not null,
+                    msg_id integer not null
+                );
+                create index if not exists idx_message_refs_peer on message_refs (peer_id);
+                create index if not exists idx_message_refs on message_refs (chat_id, msg_id);
+            `);
+        });
+        _driver.onLoad(() => {
+            this._store = this._driver.db.prepare('insert or replace into message_refs (peer_id, chat_id, msg_id) values (?, ?, ?)');
+            this._getByPeer = this._driver.db.prepare('select chat_id, msg_id from message_refs where peer_id = ?');
+            this._del = this._driver.db.prepare('delete from message_refs where chat_id = ? and msg_id = ?');
+            this._delByPeer = this._driver.db.prepare('delete from message_refs where peer_id = ?');
+            this._delAll = this._driver.db.prepare('delete from message_refs');
+        });
+    }
+    store(peerId, chatId, msgId) {
+        this._store.run(peerId, chatId, msgId);
+    }
+    getByPeer(peerId) {
+        const res = this._getByPeer.get(peerId);
+        if (!res)
+            return null;
+        const res_ = res;
+        return [res_.chat_id, res_.msg_id];
+    }
+    delete(chatId, msgIds) {
+        for (const msgId of msgIds) {
+            this._driver._writeLater(this._del, [chatId, msgId]);
+        }
+    }
+    deleteByPeer(peerId) {
+        this._delByPeer.run(peerId);
+    }
+    deleteAll() {
+        this._delAll.run();
+    }
+}
+//# sourceMappingURL=ref-messages.js.map

package/esm/storage/sqlite/repository/ref-messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ref-messages.js","sourceRoot":"","sources":["../../../../../src/storage/sqlite/repository/ref-messages.ts"],"names":[],"mappings":"AAWA,MAAM,OAAO,2BAA2B;IACpC,YAAqB,OAAgC;QAAhC,YAAO,GAAP,OAAO,CAAyB;QACjD,OAAO,CAAC,iBAAiB,CAAC,cAAc,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE;YAChD,EAAE,CAAC,IAAI,CAAC;;;;;;;;aAQP,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QACF,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE;YAChB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CACjC,iFAAiF,CACpF,CAAA;YAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,4DAA4D,CAAC,CAAA;YAEvG,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,2DAA2D,CAAC,CAAA;YAChG,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAA;YACvF,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAA;QACtE,CAAC,CAAC,CAAA;IACN,CAAC;IAGD,KAAK,CAAC,MAAc,EAAE,MAAc,EAAE,KAAa;QAC/C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,CAAA;IAC1C,CAAC;IAGD,SAAS,CAAC,MAAc;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QACvC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,MAAM,IAAI,GAAG,GAA0B,CAAA;QAEvC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAA;IACtC,CAAC;IAGD,MAAM,CAAC,MAAc,EAAE,MAAgB;QACnC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE;YACxB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAA;SACvD;IACL,CAAC;IAGD,YAAY,CAAC,MAAc;QACvB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;IAC/B,CAAC;IAGD,SAAS;QACL,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;IACtB,CAAC;CACJ","sourcesContent":["import { IReferenceMessagesRepository } from '@mtcute/core'\n\nimport { BaseSqliteStorageDriver } from '../driver.js'\nimport { ISqliteStatement } from '../types.js'\n\ninterface ReferenceMessageDto {\n    peer_id: number\n    chat_id: number\n    msg_id: number\n}\n\nexport class SqliteRefMessagesRepository implements IReferenceMessagesRepository {\n    constructor(readonly _driver: BaseSqliteStorageDriver) {\n        _driver.registerMigration('ref_messages', 1, (db) => {\n            db.exec(`\n                create table if not exists message_refs (\n                    peer_id integer not null,\n                    chat_id integer not null,\n                    msg_id integer not null\n                );\n                create index if not exists idx_message_refs_peer on message_refs (peer_id);\n                create index if not exists idx_message_refs on message_refs (chat_id, msg_id);\n            `)\n        })\n        _driver.onLoad(() => {\n            this._store = this._driver.db.prepare(\n                'insert or replace into message_refs (peer_id, chat_id, msg_id) values (?, ?, ?)',\n            )\n\n            this._getByPeer = this._driver.db.prepare('select chat_id, msg_id from message_refs where peer_id = ?')\n\n            this._del = this._driver.db.prepare('delete from message_refs where chat_id = ? and msg_id = ?')\n            this._delByPeer = this._driver.db.prepare('delete from message_refs where peer_id = ?')\n            this._delAll = this._driver.db.prepare('delete from message_refs')\n        })\n    }\n\n    private _store!: ISqliteStatement\n    store(peerId: number, chatId: number, msgId: number): void {\n        this._store.run(peerId, chatId, msgId)\n    }\n\n    private _getByPeer!: ISqliteStatement\n    getByPeer(peerId: number): [number, number] | null {\n        const res = this._getByPeer.get(peerId)\n        if (!res) return null\n\n        const res_ = res as ReferenceMessageDto\n\n        return [res_.chat_id, res_.msg_id]\n    }\n\n    private _del!: ISqliteStatement\n    delete(chatId: number, msgIds: number[]): void {\n        for (const msgId of msgIds) {\n            this._driver._writeLater(this._del, [chatId, msgId])\n        }\n    }\n\n    private _delByPeer!: ISqliteStatement\n    deleteByPeer(peerId: number): void {\n        this._delByPeer.run(peerId)\n    }\n\n    private _delAll!: ISqliteStatement\n    deleteAll(): void {\n        this._delAll.run()\n    }\n}\n"]}