@mtcute/core 0.6.0 → 0.7.0

package/cjs/highlevel/methods/users/update-profile.d.ts

@@ -0,0 +1,23 @@
+import { ITelegramClient } from '../../client.types.js';
+import { User } from '../../types/index.js';
+/**
+ * Update your profile details.
+ *
+ * Only pass fields that you want to change.
+ *
+ * @param params
+ */
+export declare function updateProfile(client: ITelegramClient, params: {
+    /**
+     * New first name
+     */
+    firstName?: string;
+    /**
+     * New last name. Pass `''` (empty string) to remove it
+     */
+    lastName?: string;
+    /**
+     * New bio (max 70 chars). Pass `''` (empty string) to remove it
+     */
+    bio?: string;
+}): Promise<User>;

package/cjs/highlevel/methods/users/update-profile.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateProfile = void 0;
+const index_js_1 = require("../../types/index.js");
+/**
+ * Update your profile details.
+ *
+ * Only pass fields that you want to change.
+ *
+ * @param params
+ */
+async function updateProfile(client, params) {
+    const res = await client.call({
+        _: 'account.updateProfile',
+        firstName: params.firstName,
+        lastName: params.lastName,
+        about: params.bio,
+    });
+    return new index_js_1.User(res);
+}
+exports.updateProfile = updateProfile;
+//# sourceMappingURL=update-profile.js.map

package/cjs/highlevel/methods/users/update-profile.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-profile.js","sourceRoot":"","sources":["../../../../../src/highlevel/methods/users/update-profile.ts"],"names":[],"mappings":";;;AACA,mDAA2C;AAE3C;;;;;;GAMG;AACI,KAAK,UAAU,aAAa,CAC/B,MAAuB,EACvB,MAeC;IAED,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;QAC1B,CAAC,EAAE,uBAAuB;QAC1B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,GAAG;KACpB,CAAC,CAAA;IAEF,OAAO,IAAI,eAAI,CAAC,GAAG,CAAC,CAAA;AACxB,CAAC;AA3BD,sCA2BC","sourcesContent":["import { ITelegramClient } from '../../client.types.js'\nimport { User } from '../../types/index.js'\n\n/**\n * Update your profile details.\n *\n * Only pass fields that you want to change.\n *\n * @param params\n */\nexport async function updateProfile(\n    client: ITelegramClient,\n    params: {\n        /**\n         * New first name\n         */\n        firstName?: string\n\n        /**\n         * New last name. Pass `''` (empty string) to remove it\n         */\n        lastName?: string\n\n        /**\n         * New bio (max 70 chars). Pass `''` (empty string) to remove it\n         */\n        bio?: string\n    },\n): Promise<User> {\n    const res = await client.call({\n        _: 'account.updateProfile',\n        firstName: params.firstName,\n        lastName: params.lastName,\n        about: params.bio,\n    })\n\n    return new User(res)\n}\n"]}

package/cjs/highlevel/storage/index.d.ts

@@ -0,0 +1,4 @@
+export * from './provider.js';
+export * from './repository/peers.js';
+export * from './repository/ref-messages.js';
+export * from './storage.js';

package/cjs/highlevel/storage/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./provider.js"), exports);
+__exportStar(require("./repository/peers.js"), exports);
+__exportStar(require("./repository/ref-messages.js"), exports);
+__exportStar(require("./storage.js"), exports);
+//# sourceMappingURL=index.js.map

package/cjs/highlevel/storage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/highlevel/storage/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA6B;AAC7B,wDAAqC;AACrC,+DAA4C;AAC5C,+CAA4B","sourcesContent":["export * from './provider.js'\nexport * from './repository/peers.js'\nexport * from './repository/ref-messages.js'\nexport * from './storage.js'\n"]}

package/cjs/highlevel/storage/provider.d.ts

@@ -0,0 +1,7 @@
+import { IMtStorageProvider } from '../../storage/provider.js';
+import { IPeersRepository } from './repository/peers.js';
+import { IReferenceMessagesRepository } from './repository/ref-messages.js';
+export interface ITelegramStorageProvider extends IMtStorageProvider {
+    readonly peers: IPeersRepository;
+    readonly refMessages: IReferenceMessagesRepository;
+}

package/cjs/highlevel/storage/provider.js

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

package/cjs/highlevel/storage/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../../src/highlevel/storage/provider.ts"],"names":[],"mappings":"","sourcesContent":["import { IMtStorageProvider } from '../../storage/provider.js'\nimport { IPeersRepository } from './repository/peers.js'\nimport { IReferenceMessagesRepository } from './repository/ref-messages.js'\n\nexport interface ITelegramStorageProvider extends IMtStorageProvider {\n    readonly peers: IPeersRepository\n    readonly refMessages: IReferenceMessagesRepository\n}\n"]}

package/cjs/highlevel/storage/repository/peers.d.ts

@@ -0,0 +1,32 @@
+import { MaybePromise } from '../../../types/utils.js';
+export declare namespace IPeersRepository {
+    /** Information about a cached peer */
+    interface PeerInfo {
+        /** Peer marked ID */
+        id: number;
+        /** Peer access hash, as a fast string representation */
+        accessHash: string;
+        /** Peer usernames, if any */
+        usernames: string[];
+        /** Timestamp (in seconds) when the peer was last updated */
+        updated: number;
+        /** Peer phone number, if available */
+        phone?: string;
+        /**
+         * Complete information about the peer,
+         * serialization of {@link tl.TypeUser} or {@link tl.TypeChat}
+         */
+        complete: Uint8Array;
+    }
+}
+export interface IPeersRepository {
+    /** Store the given peer*/
+    store(peer: IPeersRepository.PeerInfo): MaybePromise<void>;
+    /** Find a peer by their `id` */
+    getById(id: number): MaybePromise<IPeersRepository.PeerInfo | null>;
+    /** Find a peer by their username (where `usernames` includes `username`) */
+    getByUsername(username: string): MaybePromise<IPeersRepository.PeerInfo | null>;
+    /** Find a peer by their `phone` */
+    getByPhone(phone: string): MaybePromise<IPeersRepository.PeerInfo | null>;
+    deleteAll(): MaybePromise<void>;
+}

package/cjs/highlevel/storage/repository/peers.js

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

package/cjs/highlevel/storage/repository/peers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"peers.js","sourceRoot":"","sources":["../../../../../src/highlevel/storage/repository/peers.ts"],"names":[],"mappings":"","sourcesContent":["import { MaybePromise } from '../../../types/utils.js'\n\n// eslint-disable-next-line @typescript-eslint/no-namespace\nexport namespace IPeersRepository {\n    /** Information about a cached peer */\n    export interface PeerInfo {\n        /** Peer marked ID */\n        id: number\n        /** Peer access hash, as a fast string representation */\n        accessHash: string\n        /** Peer usernames, if any */\n        usernames: string[]\n        /** Timestamp (in seconds) when the peer was last updated */\n        updated: number\n        /** Peer phone number, if available */\n        phone?: string\n\n        /**\n         * Complete information about the peer,\n         * serialization of {@link tl.TypeUser} or {@link tl.TypeChat}\n         */\n        complete: Uint8Array\n    }\n}\n\nexport interface IPeersRepository {\n    /** Store the given peer*/\n    store(peer: IPeersRepository.PeerInfo): MaybePromise<void>\n    /** Find a peer by their `id` */\n    getById(id: number): MaybePromise<IPeersRepository.PeerInfo | null>\n    /** Find a peer by their username (where `usernames` includes `username`) */\n    getByUsername(username: string): MaybePromise<IPeersRepository.PeerInfo | null>\n    /** Find a peer by their `phone` */\n    getByPhone(phone: string): MaybePromise<IPeersRepository.PeerInfo | null>\n\n    deleteAll(): MaybePromise<void>\n}\n"]}

package/cjs/highlevel/storage/repository/ref-messages.d.ts

@@ -0,0 +1,20 @@
+import { MaybePromise } from '../../../types/utils.js';
+export interface IReferenceMessagesRepository {
+    /** Store a reference message */
+    store(peerId: number, chatId: number, msgId: number): MaybePromise<void>;
+    /**
+     * Get the reference message for the given `peerId`.
+     *
+     * If more than one reference message is stored for the given `peerId`,
+     * the one with the highest `msgId` should be returned, but this is not
+     * really important.
+     */
+    getByPeer(peerId: number): MaybePromise<[number, number] | null>;
+    /**
+     * Delete reference messages given the `chatId`
+     * where `msgId` is one of `msgIds`
+     */
+    delete(chatId: number, msgIds: number[]): MaybePromise<void>;
+    deleteByPeer(peerId: number): MaybePromise<void>;
+    deleteAll(): MaybePromise<void>;
+}

package/cjs/highlevel/storage/repository/ref-messages.js

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

package/cjs/highlevel/storage/repository/ref-messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ref-messages.js","sourceRoot":"","sources":["../../../../../src/highlevel/storage/repository/ref-messages.ts"],"names":[],"mappings":"","sourcesContent":["import { MaybePromise } from '../../../types/utils.js'\n\nexport interface IReferenceMessagesRepository {\n    /** Store a reference message */\n    store(peerId: number, chatId: number, msgId: number): MaybePromise<void>\n    /**\n     * Get the reference message for the given `peerId`.\n     *\n     * If more than one reference message is stored for the given `peerId`,\n     * the one with the highest `msgId` should be returned, but this is not\n     * really important.\n     */\n    getByPeer(peerId: number): MaybePromise<[number, number] | null>\n\n    /**\n     * Delete reference messages given the `chatId`\n     * where `msgId` is one of `msgIds`\n     */\n    delete(chatId: number, msgIds: number[]): MaybePromise<void>\n    deleteByPeer(peerId: number): MaybePromise<void>\n    deleteAll(): MaybePromise<void>\n}\n"]}

package/cjs/highlevel/storage/service/current-user.d.ts

@@ -0,0 +1,23 @@
+import { tl } from '@mtcute/tl';
+import { IKeyValueRepository } from '../../../storage/repository/key-value.js';
+import { BaseService, ServiceOptions } from '../../../storage/service/base.js';
+export interface CurrentUserInfo {
+    userId: number;
+    isBot: boolean;
+    isPremium: boolean;
+    usernames: string[];
+}
+export declare class CurrentUserService extends BaseService {
+    private _kv;
+    constructor(_kv: IKeyValueRepository, opts: ServiceOptions);
+    private _cached?;
+    store(info: CurrentUserInfo | null): Promise<void>;
+    storeFrom(user: tl.TypeUser): Promise<CurrentUserInfo>;
+    fetch(): Promise<CurrentUserInfo | null>;
+    getCached(safe?: boolean): CurrentUserInfo | null;
+    update(params: {
+        username?: string;
+        usernames?: string[];
+        isPremium?: boolean;
+    }): Promise<void>;
+}

package/cjs/highlevel/storage/service/current-user.js

@@ -0,0 +1,129 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CurrentUserService = void 0;
+const tl_runtime_1 = require("@mtcute/tl-runtime");
+const base_js_1 = require("../../../storage/service/base.js");
+const index_js_1 = require("../../../types/index.js");
+const type_assertions_js_1 = require("../../../utils/type-assertions.js");
+const peer_utils_js_1 = require("../../utils/peer-utils.js");
+const KV_CURRENT_USER = 'current_user';
+function serialize(info) {
+    if (!info)
+        return new Uint8Array(0);
+    const hasUsernames = info.usernames.length > 0;
+    let usernamesOverhead = hasUsernames ? 4 : 0;
+    for (const username of info.usernames) {
+        // since usernames are always ASCII, string length is the same as byte length
+        usernamesOverhead += tl_runtime_1.TlSerializationCounter.countBytesOverhead(username.length) + username.length;
+    }
+    const writer = tl_runtime_1.TlBinaryWriter.manual(16 + usernamesOverhead);
+    writer.int(1); // version
+    let flags = 0;
+    if (info.isBot)
+        flags |= 1;
+    if (hasUsernames)
+        flags |= 2;
+    if (info.isPremium)
+        flags |= 4;
+    writer.int(flags);
+    writer.int53(info.userId);
+    if (hasUsernames) {
+        writer.int(info.usernames.length);
+        for (const username of info.usernames) {
+            writer.string(username);
+        }
+    }
+    return writer.result();
+}
+function parse(data) {
+    if (data.length === 0)
+        return null;
+    const reader = tl_runtime_1.TlBinaryReader.manual(data);
+    if (reader.int() !== 1)
+        return null;
+    const flags = reader.int();
+    const userId = reader.int53();
+    let usernames = [];
+    if (flags & 2) {
+        const len = reader.int();
+        usernames = new Array(len);
+        for (let i = 0; i < len; i++) {
+            usernames[i] = reader.string();
+        }
+    }
+    return {
+        userId,
+        isBot: (flags & 1) !== 0,
+        isPremium: (flags & 4) !== 0,
+        usernames,
+    };
+}
+// todo: add testMode here
+class CurrentUserService extends base_js_1.BaseService {
+    constructor(_kv, opts) {
+        super(opts);
+        this._kv = _kv;
+    }
+    async store(info) {
+        if (info && this._cached) {
+            // update the existing object so the references to it are still valid
+            if (this._cached.userId === info.userId) {
+                return;
+            }
+            this._cached.userId = info.userId;
+            this._cached.isBot = info.isBot;
+        }
+        else {
+            this._cached = info;
+        }
+        await this._kv.set(KV_CURRENT_USER, serialize(info));
+        await this._driver.save?.();
+    }
+    async storeFrom(user) {
+        (0, type_assertions_js_1.assertTypeIs)('storeFrom', user, 'user');
+        const obj = {
+            userId: user.id,
+            isBot: user.bot,
+            isPremium: user.premium,
+            usernames: (0, peer_utils_js_1.extractUsernames)(user),
+        };
+        await this.store(obj);
+        return obj;
+    }
+    async fetch() {
+        if (this._cached)
+            return this._cached;
+        const data = await this._kv.get(KV_CURRENT_USER);
+        if (!data)
+            return null;
+        const info = parse(data);
+        this._cached = info;
+        return info;
+    }
+    getCached(safe = false) {
+        if (this._cached === undefined) {
+            if (safe)
+                return null;
+            throw new index_js_1.MtArgumentError('User info is not cached yet');
+        }
+        return this._cached;
+    }
+    async update(params) {
+        const info = await this.fetch();
+        if (!info)
+            return;
+        const { username, usernames, isPremium } = params;
+        if (isPremium !== undefined)
+            info.isPremium = isPremium;
+        if (username !== undefined) {
+            // "main" username is always the first one
+            info.usernames[0] = username;
+        }
+        else if (usernames !== undefined) {
+            info.usernames = usernames;
+        }
+        return this.store(info);
+    }
+}
+exports.CurrentUserService = CurrentUserService;
+//# sourceMappingURL=current-user.js.map

package/cjs/highlevel/storage/service/current-user.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"current-user.js","sourceRoot":"","sources":["../../../../../src/highlevel/storage/service/current-user.ts"],"names":[],"mappings":";;;AACA,mDAA2F;AAG3F,8DAA8E;AAC9E,sDAAyD;AACzD,0EAAgE;AAChE,6DAA4D;AAS5D,MAAM,eAAe,GAAG,cAAc,CAAA;AAEtC,SAAS,SAAS,CAAC,IAA4B;IAC3C,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;IAEnC,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAA;IAE9C,IAAI,iBAAiB,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAE5C,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE;QACnC,6EAA6E;QAC7E,iBAAiB,IAAI,mCAAsB,CAAC,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAA;KACpG;IAED,MAAM,MAAM,GAAG,2BAAc,CAAC,MAAM,CAAC,EAAE,GAAG,iBAAiB,CAAC,CAAA;IAC5D,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA,CAAC,UAAU;IAExB,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,IAAI,CAAC,KAAK;QAAE,KAAK,IAAI,CAAC,CAAA;IAC1B,IAAI,YAAY;QAAE,KAAK,IAAI,CAAC,CAAA;IAC5B,IAAI,IAAI,CAAC,SAAS;QAAE,KAAK,IAAI,CAAC,CAAA;IAE9B,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;IACjB,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAEzB,IAAI,YAAY,EAAE;QACd,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAEjC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE;YACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;SAC1B;KACJ;IAED,OAAO,MAAM,CAAC,MAAM,EAAE,CAAA;AAC1B,CAAC;AAED,SAAS,KAAK,CAAC,IAAgB;IAC3B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAA;IAElC,MAAM,MAAM,GAAG,2BAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAC1C,IAAI,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC;QAAE,OAAO,IAAI,CAAA;IAEnC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,EAAE,CAAA;IAC1B,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,CAAA;IAE7B,IAAI,SAAS,GAAa,EAAE,CAAA;IAE5B,IAAI,KAAK,GAAG,CAAC,EAAE;QACX,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,CAAA;QACxB,SAAS,GAAG,IAAI,KAAK,CAAS,GAAG,CAAC,CAAA;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE;YAC1B,SAAS,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAA;SACjC;KACJ;IAED,OAAO;QACH,MAAM;QACN,KAAK,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QACxB,SAAS,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC5B,SAAS;KACZ,CAAA;AACL,CAAC;AAED,0BAA0B;AAE1B,MAAa,kBAAmB,SAAQ,qBAAW;IAC/C,YACY,GAAwB,EAChC,IAAoB;QAEpB,KAAK,CAAC,IAAI,CAAC,CAAA;QAHH,QAAG,GAAH,GAAG,CAAqB;IAIpC,CAAC;IAID,KAAK,CAAC,KAAK,CAAC,IAA4B;QACpC,IAAI,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE;YACtB,qEAAqE;YACrE,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM,EAAE;gBACrC,OAAM;aACT;YAED,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;YACjC,IAAI,CAAC,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;SAClC;aAAM;YACH,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;SACtB;QAED,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAA;QACpD,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAA;IAC/B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAiB;QAC7B,IAAA,iCAAY,EAAC,WAAW,EAAE,IAAI,EAAE,MAAM,CAAC,CAAA;QAEvC,MAAM,GAAG,GAAoB;YACzB,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,KAAK,EAAE,IAAI,CAAC,GAAI;YAChB,SAAS,EAAE,IAAI,CAAC,OAAQ;YACxB,SAAS,EAAE,IAAA,gCAAgB,EAAC,IAAI,CAAC;SACpC,CAAA;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;QAErB,OAAO,GAAG,CAAA;IACd,CAAC;IAED,KAAK,CAAC,KAAK;QACP,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC,OAAO,CAAA;QAErC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,CAAA;QAChD,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAA;QAEtB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAA;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QAEnB,OAAO,IAAI,CAAA;IACf,CAAC;IAED,SAAS,CAAC,IAAI,GAAG,KAAK;QAClB,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE;YAC5B,IAAI,IAAI;gBAAE,OAAO,IAAI,CAAA;YAErB,MAAM,IAAI,0BAAe,CAAC,6BAA6B,CAAC,CAAA;SAC3D;QAED,OAAO,IAAI,CAAC,OAAO,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAwE;QACjF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,CAAA;QAC/B,IAAI,CAAC,IAAI;YAAE,OAAM;QAEjB,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,MAAM,CAAA;QAEjD,IAAI,SAAS,KAAK,SAAS;YAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;QAEvD,IAAI,QAAQ,KAAK,SAAS,EAAE;YACxB,0CAA0C;YAC1C,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAA;SAC/B;aAAM,IAAI,SAAS,KAAK,SAAS,EAAE;YAChC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;SAC7B;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IAC3B,CAAC;CACJ;AAhFD,gDAgFC","sourcesContent":["import { tl } from '@mtcute/tl'\nimport { TlBinaryReader, TlBinaryWriter, TlSerializationCounter } from '@mtcute/tl-runtime'\n\nimport { IKeyValueRepository } from '../../../storage/repository/key-value.js'\nimport { BaseService, ServiceOptions } from '../../../storage/service/base.js'\nimport { MtArgumentError } from '../../../types/index.js'\nimport { assertTypeIs } from '../../../utils/type-assertions.js'\nimport { extractUsernames } from '../../utils/peer-utils.js'\n\nexport interface CurrentUserInfo {\n    userId: number\n    isBot: boolean\n    isPremium: boolean\n    usernames: string[]\n}\n\nconst KV_CURRENT_USER = 'current_user'\n\nfunction serialize(info: CurrentUserInfo | null): Uint8Array {\n    if (!info) return new Uint8Array(0)\n\n    const hasUsernames = info.usernames.length > 0\n\n    let usernamesOverhead = hasUsernames ? 4 : 0\n\n    for (const username of info.usernames) {\n        // since usernames are always ASCII, string length is the same as byte length\n        usernamesOverhead += TlSerializationCounter.countBytesOverhead(username.length) + username.length\n    }\n\n    const writer = TlBinaryWriter.manual(16 + usernamesOverhead)\n    writer.int(1) // version\n\n    let flags = 0\n    if (info.isBot) flags |= 1\n    if (hasUsernames) flags |= 2\n    if (info.isPremium) flags |= 4\n\n    writer.int(flags)\n    writer.int53(info.userId)\n\n    if (hasUsernames) {\n        writer.int(info.usernames.length)\n\n        for (const username of info.usernames) {\n            writer.string(username)\n        }\n    }\n\n    return writer.result()\n}\n\nfunction parse(data: Uint8Array): CurrentUserInfo | null {\n    if (data.length === 0) return null\n\n    const reader = TlBinaryReader.manual(data)\n    if (reader.int() !== 1) return null\n\n    const flags = reader.int()\n    const userId = reader.int53()\n\n    let usernames: string[] = []\n\n    if (flags & 2) {\n        const len = reader.int()\n        usernames = new Array<string>(len)\n\n        for (let i = 0; i < len; i++) {\n            usernames[i] = reader.string()\n        }\n    }\n\n    return {\n        userId,\n        isBot: (flags & 1) !== 0,\n        isPremium: (flags & 4) !== 0,\n        usernames,\n    }\n}\n\n// todo: add testMode here\n\nexport class CurrentUserService extends BaseService {\n    constructor(\n        private _kv: IKeyValueRepository,\n        opts: ServiceOptions,\n    ) {\n        super(opts)\n    }\n\n    private _cached?: CurrentUserInfo | null\n\n    async store(info: CurrentUserInfo | null): Promise<void> {\n        if (info && this._cached) {\n            // update the existing object so the references to it are still valid\n            if (this._cached.userId === info.userId) {\n                return\n            }\n\n            this._cached.userId = info.userId\n            this._cached.isBot = info.isBot\n        } else {\n            this._cached = info\n        }\n\n        await this._kv.set(KV_CURRENT_USER, serialize(info))\n        await this._driver.save?.()\n    }\n\n    async storeFrom(user: tl.TypeUser): Promise<CurrentUserInfo> {\n        assertTypeIs('storeFrom', user, 'user')\n\n        const obj: CurrentUserInfo = {\n            userId: user.id,\n            isBot: user.bot!,\n            isPremium: user.premium!,\n            usernames: extractUsernames(user),\n        }\n        await this.store(obj)\n\n        return obj\n    }\n\n    async fetch(): Promise<CurrentUserInfo | null> {\n        if (this._cached) return this._cached\n\n        const data = await this._kv.get(KV_CURRENT_USER)\n        if (!data) return null\n\n        const info = parse(data)\n        this._cached = info\n\n        return info\n    }\n\n    getCached(safe = false): CurrentUserInfo | null {\n        if (this._cached === undefined) {\n            if (safe) return null\n\n            throw new MtArgumentError('User info is not cached yet')\n        }\n\n        return this._cached\n    }\n\n    async update(params: { username?: string; usernames?: string[]; isPremium?: boolean }): Promise<void> {\n        const info = await this.fetch()\n        if (!info) return\n\n        const { username, usernames, isPremium } = params\n\n        if (isPremium !== undefined) info.isPremium = isPremium\n\n        if (username !== undefined) {\n            // \"main\" username is always the first one\n            info.usernames[0] = username\n        } else if (usernames !== undefined) {\n            info.usernames = usernames\n        }\n\n        return this.store(info)\n    }\n}\n"]}

package/cjs/highlevel/storage/service/peers.d.ts

@@ -0,0 +1,23 @@
+import { tl } from '@mtcute/tl';
+import { BaseService, ServiceOptions } from '../../../storage/service/base.js';
+import { IPeersRepository } from '../repository/peers.js';
+import { RefMessagesService } from './ref-messages.js';
+export interface PeersServiceOptions {
+    cacheSize?: number;
+    updatesWriteInterval?: number;
+}
+export declare class PeersService extends BaseService {
+    private options;
+    private _peers;
+    private _refs;
+    private _cache;
+    private _pendingWrites;
+    constructor(options: PeersServiceOptions, _peers: IPeersRepository, _refs: RefMessagesService, common: ServiceOptions);
+    updatePeersFrom(obj: tl.TlObject | tl.TlObject[]): Promise<boolean>;
+    store(peer: tl.TypeUser | tl.TypeChat): Promise<void>;
+    private _returnCaching;
+    getById(id: number, allowRefs?: boolean): Promise<tl.TypeInputPeer | null>;
+    getByPhone(phone: string): Promise<tl.TypeInputPeer | null>;
+    getByUsername(username: string): Promise<tl.TypeInputPeer | null>;
+    getCompleteById(id: number): Promise<tl.TypeUser | tl.TypeChat | null>;
+}

package/cjs/highlevel/storage/service/peers.js

@@ -0,0 +1,217 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PeersService = void 0;
+const long_1 = __importDefault(require("long"));
+const base_js_1 = require("../../../storage/service/base.js");
+const long_utils_js_1 = require("../../../utils/long-utils.js");
+const lru_map_js_1 = require("../../../utils/lru-map.js");
+const peer_utils_js_1 = require("../../../utils/peer-utils.js");
+const peer_utils_js_2 = require("../../utils/peer-utils.js");
+const USERNAME_TTL = 24 * 60 * 60 * 1000; // 1 day
+function getInputPeer(dto) {
+    const [type, id] = (0, peer_utils_js_1.parseMarkedPeerId)(dto.id);
+    switch (type) {
+        case 'user':
+            return {
+                _: 'inputPeerUser',
+                userId: id,
+                accessHash: (0, long_utils_js_1.longFromFastString)(dto.accessHash),
+            };
+        case 'chat':
+            return {
+                _: 'inputPeerChat',
+                chatId: id,
+            };
+        case 'channel':
+            return {
+                _: 'inputPeerChannel',
+                channelId: id,
+                accessHash: (0, long_utils_js_1.longFromFastString)(dto.accessHash),
+            };
+    }
+}
+class PeersService extends base_js_1.BaseService {
+    constructor(options, _peers, _refs, common) {
+        super(common);
+        this.options = options;
+        this._peers = _peers;
+        this._refs = _refs;
+        this._pendingWrites = new Map();
+        this._cache = new lru_map_js_1.LruMap(options.cacheSize ?? 100);
+    }
+    async updatePeersFrom(obj) {
+        let count = 0;
+        for (const peer of (0, peer_utils_js_1.getAllPeersFrom)(obj)) {
+            // no point in caching min peers as we can't use them
+            if (peer.min)
+                continue;
+            count += 1;
+            await this.store(peer);
+        }
+        if (count > 0) {
+            await this._driver.save?.();
+            this._log.debug('cached %d peers', count);
+            return true;
+        }
+        return false;
+    }
+    async store(peer) {
+        let dto;
+        let accessHash;
+        switch (peer._) {
+            case 'user': {
+                if (!peer.accessHash) {
+                    this._log.warn('received user without access hash: %j', peer);
+                    return;
+                }
+                dto = {
+                    id: peer.id,
+                    accessHash: (0, long_utils_js_1.longToFastString)(peer.accessHash),
+                    phone: peer.phone,
+                    usernames: (0, peer_utils_js_2.extractUsernames)(peer),
+                    updated: Date.now(),
+                    complete: this._serializeTl(peer),
+                };
+                accessHash = peer.accessHash;
+                break;
+            }
+            case 'chat':
+            case 'chatForbidden': {
+                dto = {
+                    id: -peer.id,
+                    accessHash: '',
+                    updated: Date.now(),
+                    complete: this._serializeTl(peer),
+                    usernames: [],
+                };
+                accessHash = long_1.default.ZERO;
+                break;
+            }
+            case 'channel':
+            case 'channelForbidden': {
+                if (!peer.accessHash) {
+                    this._log.warn('received channel without access hash: %j', peer);
+                    return;
+                }
+                dto = {
+                    id: (0, peer_utils_js_1.toggleChannelIdMark)(peer.id),
+                    accessHash: (0, long_utils_js_1.longToFastString)(peer.accessHash),
+                    usernames: (0, peer_utils_js_2.extractUsernames)(peer),
+                    updated: Date.now(),
+                    complete: this._serializeTl(peer),
+                };
+                accessHash = peer.accessHash;
+                break;
+            }
+            default:
+                return;
+        }
+        const cached = this._cache.get(peer.id);
+        if (cached && this.options.updatesWriteInterval !== 0) {
+            const oldAccessHash = cached.peer.accessHash;
+            if (oldAccessHash?.eq(accessHash)) {
+                // when entity is cached and hash is the same, an update query is needed,
+                // since some field in the full entity might have changed, or the username/phone
+                //
+                // to avoid too many DB calls, and since these updates are pretty common,
+                // they are grouped and applied in batches no more than once every 30sec (or user-defined).
+                //
+                // until then, they are either served from in-memory cache,
+                // or an older version is fetched from DB
+                this._pendingWrites.set(peer.id, dto);
+                cached.complete = peer;
+                return;
+            }
+        }
+        // entity is not cached in memory, or the access hash has changed
+        // we need to update it in the DB asap, and also update the in-memory cache
+        await this._peers.store(dto);
+        this._cache.set(peer.id, {
+            peer: getInputPeer(dto),
+            complete: peer,
+        });
+        // todo: if (!this._cachedSelf?.isBot) {
+        // we have the full peer, we no longer need the references
+        // we can skip this in the other branch, since in that case it would've already been deleted
+        await this._refs.deleteByPeer(peer.id);
+    }
+    _returnCaching(id, dto) {
+        const peer = getInputPeer(dto);
+        const complete = this._deserializeTl(dto.complete);
+        this._cache.set(id, {
+            peer,
+            complete: complete,
+        });
+        return peer;
+    }
+    async getById(id, allowRefs = true) {
+        const cached = this._cache.get(id);
+        if (cached)
+            return cached.peer;
+        const dto = await this._peers.getById(id);
+        if (dto) {
+            return this._returnCaching(id, dto);
+        }
+        if (allowRefs) {
+            const ref = await this._refs.getForPeer(id);
+            if (!ref)
+                return null;
+            const [chatId, msgId] = ref;
+            const chat = await this.getById(chatId, false);
+            if (!chat)
+                return null;
+            if (id > 0) {
+                // user
+                return {
+                    _: 'inputPeerUserFromMessage',
+                    peer: chat,
+                    msgId,
+                    userId: id,
+                };
+            }
+            // channel
+            return {
+                _: 'inputPeerChannelFromMessage',
+                peer: chat,
+                msgId,
+                channelId: (0, peer_utils_js_1.toggleChannelIdMark)(id),
+            };
+        }
+        return null;
+    }
+    async getByPhone(phone) {
+        const dto = await this._peers.getByPhone(phone);
+        if (!dto)
+            return null;
+        return this._returnCaching(dto.id, dto);
+    }
+    async getByUsername(username) {
+        const dto = await this._peers.getByUsername(username.toLowerCase());
+        if (!dto)
+            return null;
+        if (Date.now() - dto.updated > USERNAME_TTL) {
+            // username is too old, we can't trust it. ask the client to re-fetch it
+            return null;
+        }
+        return this._returnCaching(dto.id, dto);
+    }
+    async getCompleteById(id) {
+        const cached = this._cache.get(id);
+        if (cached)
+            return cached.complete;
+        const dto = await this._peers.getById(id);
+        if (!dto)
+            return null;
+        const cacheItem = {
+            peer: getInputPeer(dto),
+            complete: this._deserializeTl(dto.complete),
+        };
+        this._cache.set(id, cacheItem);
+        return cacheItem.complete;
+    }
+}
+exports.PeersService = PeersService;
+//# sourceMappingURL=peers.js.map

package/cjs/highlevel/storage/service/peers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"peers.js","sourceRoot":"","sources":["../../../../../src/highlevel/storage/service/peers.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAuB;AAIvB,8DAA8E;AAC9E,gEAAmF;AACnF,0DAAkD;AAClD,gEAAsG;AACtG,6DAA4D;AAc5D,MAAM,YAAY,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAA,CAAC,QAAQ;AAEjD,SAAS,YAAY,CAAC,GAA8B;IAChD,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAA,iCAAiB,EAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAE5C,QAAQ,IAAI,EAAE;QACV,KAAK,MAAM;YACP,OAAO;gBACH,CAAC,EAAE,eAAe;gBAClB,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,IAAA,kCAAkB,EAAC,GAAG,CAAC,UAAU,CAAC;aACjD,CAAA;QACL,KAAK,MAAM;YACP,OAAO;gBACH,CAAC,EAAE,eAAe;gBAClB,MAAM,EAAE,EAAE;aACb,CAAA;QACL,KAAK,SAAS;YACV,OAAO;gBACH,CAAC,EAAE,kBAAkB;gBACrB,SAAS,EAAE,EAAE;gBACb,UAAU,EAAE,IAAA,kCAAkB,EAAC,GAAG,CAAC,UAAU,CAAC;aACjD,CAAA;KACR;AACL,CAAC;AAED,MAAa,YAAa,SAAQ,qBAAW;IAIzC,YACY,OAA4B,EAC5B,MAAwB,EACxB,KAAyB,EACjC,MAAsB;QAEtB,KAAK,CAAC,MAAM,CAAC,CAAA;QALL,YAAO,GAAP,OAAO,CAAqB;QAC5B,WAAM,GAAN,MAAM,CAAkB;QACxB,UAAK,GAAL,KAAK,CAAoB;QAL7B,mBAAc,GAAG,IAAI,GAAG,EAAqC,CAAA;QAUjE,IAAI,CAAC,MAAM,GAAG,IAAI,mBAAM,CAAC,OAAO,CAAC,SAAS,IAAI,GAAG,CAAC,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,GAAgC;QAClD,IAAI,KAAK,GAAG,CAAC,CAAA;QAEb,KAAK,MAAM,IAAI,IAAI,IAAA,+BAAe,EAAC,GAAG,CAAC,EAAE;YACrC,qDAAqD;YACrD,IAAK,IAAgD,CAAC,GAAG;gBAAE,SAAQ;YAEnE,KAAK,IAAI,CAAC,CAAA;YAEV,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;SACzB;QAED,IAAI,KAAK,GAAG,CAAC,EAAE;YACX,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAA;YAC3B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAA;YAEzC,OAAO,IAAI,CAAA;SACd;QAED,OAAO,KAAK,CAAA;IAChB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAA+B;QACvC,IAAI,GAA8B,CAAA;QAClC,IAAI,UAAmB,CAAA;QAEvB,QAAQ,IAAI,CAAC,CAAC,EAAE;YACZ,KAAK,MAAM,CAAC,CAAC;gBACT,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;oBAClB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,IAAI,CAAC,CAAA;oBAE7D,OAAM;iBACT;gBAED,GAAG,GAAG;oBACF,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,UAAU,EAAE,IAAA,gCAAgB,EAAC,IAAI,CAAC,UAAU,CAAC;oBAC7C,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,SAAS,EAAE,IAAA,gCAAgB,EAAC,IAAI,CAAC;oBACjC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;oBACnB,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;iBACpC,CAAA;gBACD,UAAU,GAAG,IAAI,CAAC,UAAU,CAAA;gBAC5B,MAAK;aACR;YACD,KAAK,MAAM,CAAC;YACZ,KAAK,eAAe,CAAC,CAAC;gBAClB,GAAG,GAAG;oBACF,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE;oBACZ,UAAU,EAAE,EAAE;oBACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;oBACnB,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;oBACjC,SAAS,EAAE,EAAE;iBAChB,CAAA;gBACD,UAAU,GAAG,cAAI,CAAC,IAAI,CAAA;gBACtB,MAAK;aACR;YACD,KAAK,SAAS,CAAC;YACf,KAAK,kBAAkB,CAAC,CAAC;gBACrB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;oBAClB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,0CAA0C,EAAE,IAAI,CAAC,CAAA;oBAEhE,OAAM;iBACT;gBAED,GAAG,GAAG;oBACF,EAAE,EAAE,IAAA,mCAAmB,EAAC,IAAI,CAAC,EAAE,CAAC;oBAChC,UAAU,EAAE,IAAA,gCAAgB,EAAC,IAAI,CAAC,UAAU,CAAC;oBAC7C,SAAS,EAAE,IAAA,gCAAgB,EAAC,IAAqB,CAAC;oBAClD,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;oBACnB,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;iBACpC,CAAA;gBACD,UAAU,GAAG,IAAI,CAAC,UAAU,CAAA;gBAC5B,MAAK;aACR;YACD;gBACI,OAAM;SACb;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAEvC,IAAI,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,oBAAoB,KAAK,CAAC,EAAE;YACnD,MAAM,aAAa,GAAI,MAAM,CAAC,IAA4D,CAAC,UAAU,CAAA;YAErG,IAAI,aAAa,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE;gBAC/B,yEAAyE;gBACzE,gFAAgF;gBAChF,EAAE;gBACF,yEAAyE;gBACzE,2FAA2F;gBAC3F,EAAE;gBACF,2DAA2D;gBAC3D,yCAAyC;gBAEzC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;gBACrC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAA;gBAEtB,OAAM;aACT;SACJ;QAED,iEAAiE;QACjE,2EAA2E;QAC3E,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;QAC5B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACrB,IAAI,EAAE,YAAY,CAAC,GAAG,CAAC;YACvB,QAAQ,EAAE,IAAI;SACjB,CAAC,CAAA;QAEF,wCAAwC;QACxC,0DAA0D;QAC1D,4FAA4F;QAC5F,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAC1C,CAAC;IAEO,cAAc,CAAC,EAAU,EAAE,GAA8B;QAC7D,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAElD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE;YAChB,IAAI;YACJ,QAAQ,EAAE,QAA4C;SACzD,CAAC,CAAA;QAEF,OAAO,IAAI,CAAA;IACf,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,EAAU,EAAE,SAAS,GAAG,IAAI;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAClC,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC,IAAI,CAAA;QAE9B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;QAEzC,IAAI,GAAG,EAAE;YACL,OAAO,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;SACtC;QAED,IAAI,SAAS,EAAE;YACX,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAA;YAC3C,IAAI,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAA;YAErB,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,GAAG,CAAA;YAC3B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;YAC9C,IAAI,CAAC,IAAI;gBAAE,OAAO,IAAI,CAAA;YAEtB,IAAI,EAAE,GAAG,CAAC,EAAE;gBACR,OAAO;gBACP,OAAO;oBACH,CAAC,EAAE,0BAA0B;oBAC7B,IAAI,EAAE,IAAI;oBACV,KAAK;oBACL,MAAM,EAAE,EAAE;iBACb,CAAA;aACJ;YAED,UAAU;YACV,OAAO;gBACH,CAAC,EAAE,6BAA6B;gBAChC,IAAI,EAAE,IAAI;gBACV,KAAK;gBACL,SAAS,EAAE,IAAA,mCAAmB,EAAC,EAAE,CAAC;aACrC,CAAA;SACJ;QAED,OAAO,IAAI,CAAA;IACf,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAa;QAC1B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAA;QAC/C,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,QAAgB;QAChC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAA;QACnE,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,OAAO,GAAG,YAAY,EAAE;YACzC,wEAAwE;YACxE,OAAO,IAAI,CAAA;SACd;QAED,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAClC,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC,QAAQ,CAAA;QAElC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAA;QAErB,MAAM,SAAS,GAAc;YACzB,IAAI,EAAE,YAAY,CAAC,GAAG,CAAC;YACvB,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAqC;SAClF,CAAA;QACD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,CAAA;QAE9B,OAAO,SAAS,CAAC,QAAQ,CAAA;IAC7B,CAAC;CACJ;AAxND,oCAwNC","sourcesContent":["import Long from 'long'\n\nimport { tl } from '@mtcute/tl'\n\nimport { BaseService, ServiceOptions } from '../../../storage/service/base.js'\nimport { longFromFastString, longToFastString } from '../../../utils/long-utils.js'\nimport { LruMap } from '../../../utils/lru-map.js'\nimport { getAllPeersFrom, parseMarkedPeerId, toggleChannelIdMark } from '../../../utils/peer-utils.js'\nimport { extractUsernames } from '../../utils/peer-utils.js'\nimport { IPeersRepository } from '../repository/peers.js'\nimport { RefMessagesService } from './ref-messages.js'\n\ninterface CacheItem {\n    peer: tl.TypeInputPeer\n    complete: tl.TypeUser | tl.TypeChat | null\n}\n\nexport interface PeersServiceOptions {\n    cacheSize?: number\n    updatesWriteInterval?: number\n}\n\nconst USERNAME_TTL = 24 * 60 * 60 * 1000 // 1 day\n\nfunction getInputPeer(dto: IPeersRepository.PeerInfo): tl.TypeInputPeer {\n    const [type, id] = parseMarkedPeerId(dto.id)\n\n    switch (type) {\n        case 'user':\n            return {\n                _: 'inputPeerUser',\n                userId: id,\n                accessHash: longFromFastString(dto.accessHash),\n            }\n        case 'chat':\n            return {\n                _: 'inputPeerChat',\n                chatId: id,\n            }\n        case 'channel':\n            return {\n                _: 'inputPeerChannel',\n                channelId: id,\n                accessHash: longFromFastString(dto.accessHash),\n            }\n    }\n}\n\nexport class PeersService extends BaseService {\n    private _cache: LruMap<number, CacheItem>\n    private _pendingWrites = new Map<number, IPeersRepository.PeerInfo>()\n\n    constructor(\n        private options: PeersServiceOptions,\n        private _peers: IPeersRepository,\n        private _refs: RefMessagesService,\n        common: ServiceOptions,\n    ) {\n        super(common)\n\n        this._cache = new LruMap(options.cacheSize ?? 100)\n    }\n\n    async updatePeersFrom(obj: tl.TlObject | tl.TlObject[]) {\n        let count = 0\n\n        for (const peer of getAllPeersFrom(obj)) {\n            // no point in caching min peers as we can't use them\n            if ((peer as Extract<typeof peer, { min?: unknown }>).min) continue\n\n            count += 1\n\n            await this.store(peer)\n        }\n\n        if (count > 0) {\n            await this._driver.save?.()\n            this._log.debug('cached %d peers', count)\n\n            return true\n        }\n\n        return false\n    }\n\n    async store(peer: tl.TypeUser | tl.TypeChat): Promise<void> {\n        let dto: IPeersRepository.PeerInfo\n        let accessHash: tl.Long\n\n        switch (peer._) {\n            case 'user': {\n                if (!peer.accessHash) {\n                    this._log.warn('received user without access hash: %j', peer)\n\n                    return\n                }\n\n                dto = {\n                    id: peer.id,\n                    accessHash: longToFastString(peer.accessHash),\n                    phone: peer.phone,\n                    usernames: extractUsernames(peer),\n                    updated: Date.now(),\n                    complete: this._serializeTl(peer),\n                }\n                accessHash = peer.accessHash\n                break\n            }\n            case 'chat':\n            case 'chatForbidden': {\n                dto = {\n                    id: -peer.id,\n                    accessHash: '',\n                    updated: Date.now(),\n                    complete: this._serializeTl(peer),\n                    usernames: [],\n                }\n                accessHash = Long.ZERO\n                break\n            }\n            case 'channel':\n            case 'channelForbidden': {\n                if (!peer.accessHash) {\n                    this._log.warn('received channel without access hash: %j', peer)\n\n                    return\n                }\n\n                dto = {\n                    id: toggleChannelIdMark(peer.id),\n                    accessHash: longToFastString(peer.accessHash),\n                    usernames: extractUsernames(peer as tl.RawChannel),\n                    updated: Date.now(),\n                    complete: this._serializeTl(peer),\n                }\n                accessHash = peer.accessHash\n                break\n            }\n            default:\n                return\n        }\n\n        const cached = this._cache.get(peer.id)\n\n        if (cached && this.options.updatesWriteInterval !== 0) {\n            const oldAccessHash = (cached.peer as Extract<tl.TypeInputPeer, { accessHash?: unknown }>).accessHash\n\n            if (oldAccessHash?.eq(accessHash)) {\n                // when entity is cached and hash is the same, an update query is needed,\n                // since some field in the full entity might have changed, or the username/phone\n                //\n                // to avoid too many DB calls, and since these updates are pretty common,\n                // they are grouped and applied in batches no more than once every 30sec (or user-defined).\n                //\n                // until then, they are either served from in-memory cache,\n                // or an older version is fetched from DB\n\n                this._pendingWrites.set(peer.id, dto)\n                cached.complete = peer\n\n                return\n            }\n        }\n\n        // entity is not cached in memory, or the access hash has changed\n        // we need to update it in the DB asap, and also update the in-memory cache\n        await this._peers.store(dto)\n        this._cache.set(peer.id, {\n            peer: getInputPeer(dto),\n            complete: peer,\n        })\n\n        // todo: if (!this._cachedSelf?.isBot) {\n        // we have the full peer, we no longer need the references\n        // we can skip this in the other branch, since in that case it would've already been deleted\n        await this._refs.deleteByPeer(peer.id)\n    }\n\n    private _returnCaching(id: number, dto: IPeersRepository.PeerInfo) {\n        const peer = getInputPeer(dto)\n        const complete = this._deserializeTl(dto.complete)\n\n        this._cache.set(id, {\n            peer,\n            complete: complete as tl.TypeUser | tl.TypeChat | null,\n        })\n\n        return peer\n    }\n\n    async getById(id: number, allowRefs = true): Promise<tl.TypeInputPeer | null> {\n        const cached = this._cache.get(id)\n        if (cached) return cached.peer\n\n        const dto = await this._peers.getById(id)\n\n        if (dto) {\n            return this._returnCaching(id, dto)\n        }\n\n        if (allowRefs) {\n            const ref = await this._refs.getForPeer(id)\n            if (!ref) return null\n\n            const [chatId, msgId] = ref\n            const chat = await this.getById(chatId, false)\n            if (!chat) return null\n\n            if (id > 0) {\n                // user\n                return {\n                    _: 'inputPeerUserFromMessage',\n                    peer: chat,\n                    msgId,\n                    userId: id,\n                }\n            }\n\n            // channel\n            return {\n                _: 'inputPeerChannelFromMessage',\n                peer: chat,\n                msgId,\n                channelId: toggleChannelIdMark(id),\n            }\n        }\n\n        return null\n    }\n\n    async getByPhone(phone: string): Promise<tl.TypeInputPeer | null> {\n        const dto = await this._peers.getByPhone(phone)\n        if (!dto) return null\n\n        return this._returnCaching(dto.id, dto)\n    }\n\n    async getByUsername(username: string): Promise<tl.TypeInputPeer | null> {\n        const dto = await this._peers.getByUsername(username.toLowerCase())\n        if (!dto) return null\n\n        if (Date.now() - dto.updated > USERNAME_TTL) {\n            // username is too old, we can't trust it. ask the client to re-fetch it\n            return null\n        }\n\n        return this._returnCaching(dto.id, dto)\n    }\n\n    async getCompleteById(id: number): Promise<tl.TypeUser | tl.TypeChat | null> {\n        const cached = this._cache.get(id)\n        if (cached) return cached.complete\n\n        const dto = await this._peers.getById(id)\n        if (!dto) return null\n\n        const cacheItem: CacheItem = {\n            peer: getInputPeer(dto),\n            complete: this._deserializeTl(dto.complete) as tl.TypeUser | tl.TypeChat | null,\n        }\n        this._cache.set(id, cacheItem)\n\n        return cacheItem.complete\n    }\n}\n"]}

package/cjs/highlevel/storage/service/ref-messages.d.ts

@@ -0,0 +1,15 @@
+import { BaseService, ServiceOptions } from '../../../storage/service/base.js';
+import { IReferenceMessagesRepository } from '../repository/ref-messages.js';
+export interface RefMessagesServiceOptions {
+    cacheSize?: number;
+}
+export declare class RefMessagesService extends BaseService {
+    readonly options: RefMessagesServiceOptions;
+    readonly _refs: IReferenceMessagesRepository;
+    private _cache;
+    constructor(options: RefMessagesServiceOptions, _refs: IReferenceMessagesRepository, common: ServiceOptions);
+    store(peerId: number, chatId: number, msgId: number): Promise<void>;
+    getForPeer(peerId: number): Promise<[number, number] | null>;
+    delete(chatId: number, msgIds: number[]): Promise<void>;
+    deleteByPeer(peerId: number): Promise<void>;
+}