@metamask-previews/assets-controllers 45.1.1-preview-175f4f5d → 45.1.1-preview-e1a582c9

package/dist/MultichainBalancesController/BalancesTracker.cjs

@@ -0,0 +1,117 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _BalancesTracker_poller, _BalancesTracker_updateBalance, _BalancesTracker_balances;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BalancesTracker = void 0;
+const Poller_1 = require("./Poller.cjs");
+const BALANCES_TRACKING_INTERVAL = 30 * 1000; // Every 30s in milliseconds.
+class BalancesTracker {
+    constructor(updateBalanceCallback) {
+        _BalancesTracker_poller.set(this, void 0);
+        _BalancesTracker_updateBalance.set(this, void 0);
+        _BalancesTracker_balances.set(this, {});
+        __classPrivateFieldSet(this, _BalancesTracker_updateBalance, updateBalanceCallback, "f");
+        __classPrivateFieldSet(this, _BalancesTracker_poller, new Poller_1.Poller(this.updateBalances, BALANCES_TRACKING_INTERVAL), "f");
+    }
+    /**
+     * Starts the tracking process.
+     */
+    start() {
+        __classPrivateFieldGet(this, _BalancesTracker_poller, "f").start();
+    }
+    /**
+     * Stops the tracking process.
+     */
+    stop() {
+        __classPrivateFieldGet(this, _BalancesTracker_poller, "f").stop();
+    }
+    /**
+     * Checks if an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @returns True if the account is being tracker, false otherwise.
+     */
+    isTracked(accountId) {
+        return accountId in __classPrivateFieldGet(this, _BalancesTracker_balances, "f");
+    }
+    /**
+     * Asserts that an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    assertBeingTracked(accountId) {
+        if (!this.isTracked(accountId)) {
+            throw new Error(`Account is not being tracked: ${accountId}`);
+        }
+    }
+    /**
+     * Starts tracking a new account ID. This method has no effect on already tracked
+     * accounts.
+     *
+     * @param accountId - The account ID.
+     * @param blockTime - The block time (used when refreshing the account balances).
+     */
+    track(accountId, blockTime) {
+        // Do not overwrite current info if already being tracked!
+        if (!this.isTracked(accountId)) {
+            __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId] = {
+                lastUpdated: 0,
+                blockTime,
+            };
+        }
+    }
+    /**
+     * Stops tracking a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    untrack(accountId) {
+        this.assertBeingTracked(accountId);
+        delete __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId];
+    }
+    /**
+     * Update the balances for a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    async updateBalance(accountId) {
+        this.assertBeingTracked(accountId);
+        // We check if the balance is outdated (by comparing to the block time associated
+        // with this kind of account).
+        //
+        // This might not be super accurate, but we could probably compute this differently
+        // and try to sync with the "real block time"!
+        const info = __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId];
+        const isOutdated = Date.now() - info.lastUpdated >= info.blockTime;
+        const hasNoBalanceYet = info.lastUpdated === 0;
+        if (hasNoBalanceYet || isOutdated) {
+            await __classPrivateFieldGet(this, _BalancesTracker_updateBalance, "f").call(this, accountId);
+            __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId].lastUpdated = Date.now();
+        }
+    }
+    /**
+     * Update the balances of all tracked accounts (only if the balances
+     * is considered outdated).
+     */
+    async updateBalances() {
+        await Promise.allSettled(Object.keys(__classPrivateFieldGet(this, _BalancesTracker_balances, "f")).map(async (accountId) => {
+            await this.updateBalance(accountId);
+        }));
+    }
+}
+exports.BalancesTracker = BalancesTracker;
+_BalancesTracker_poller = new WeakMap(), _BalancesTracker_updateBalance = new WeakMap(), _BalancesTracker_balances = new WeakMap();
+//# sourceMappingURL=BalancesTracker.cjs.map

package/dist/MultichainBalancesController/BalancesTracker.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"BalancesTracker.cjs","sourceRoot":"","sources":["../../src/MultichainBalancesController/BalancesTracker.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,yCAAkC;AAOlC,MAAM,0BAA0B,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,6BAA6B;AAE3E,MAAa,eAAe;IAO1B,YAAY,qBAA2D;QANvE,0CAAgB;QAEhB,iDAAqD;QAErD,oCAAyC,EAAE,EAAC;QAG1C,uBAAA,IAAI,kCAAkB,qBAAqB,MAAA,CAAC;QAE5C,uBAAA,IAAI,2BAAW,IAAI,eAAM,CAAC,IAAI,CAAC,cAAc,EAAE,0BAA0B,CAAC,MAAA,CAAC;IAC7E,CAAC;IAED;;OAEG;IACH,KAAK;QACH,uBAAA,IAAI,+BAAQ,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,uBAAA,IAAI,+BAAQ,CAAC,IAAI,EAAE,CAAC;IACtB,CAAC;IAED;;;;;OAKG;IACH,SAAS,CAAC,SAAiB;QACzB,OAAO,SAAS,IAAI,uBAAA,IAAI,iCAAU,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,kBAAkB,CAAC,SAAiB;QAClC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC;SAC/D;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,SAAiB,EAAE,SAAiB;QACxC,0DAA0D;QAC1D,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;YAC9B,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,GAAG;gBAC1B,WAAW,EAAE,CAAC;gBACd,SAAS;aACV,CAAC;SACH;IACH,CAAC;IAED;;;;;OAKG;IACH,OAAO,CAAC,SAAiB;QACvB,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACnC,OAAO,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAEnC,iFAAiF;QACjF,8BAA8B;QAC9B,EAAE;QACF,mFAAmF;QACnF,8CAA8C;QAC9C,MAAM,IAAI,GAAG,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC;QACnE,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,KAAK,CAAC,CAAC;QAC/C,IAAI,eAAe,IAAI,UAAU,EAAE;YACjC,MAAM,uBAAA,IAAI,sCAAe,MAAnB,IAAI,EAAgB,SAAS,CAAC,CAAC;YACrC,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;SACpD;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,CAAC,UAAU,CACtB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,iCAAU,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;YAClD,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AA/GD,0CA+GC","sourcesContent":["import { Poller } from './Poller';\n\ntype BalanceInfo = {\n  lastUpdated: number;\n  blockTime: number;\n};\n\nconst BALANCES_TRACKING_INTERVAL = 30 * 1000; // Every 30s in milliseconds.\n\nexport class BalancesTracker {\n  #poller: Poller;\n\n  #updateBalance: (accountId: string) => Promise<void>;\n\n  #balances: Record<string, BalanceInfo> = {};\n\n  constructor(updateBalanceCallback: (accountId: string) => Promise<void>) {\n    this.#updateBalance = updateBalanceCallback;\n\n    this.#poller = new Poller(this.updateBalances, BALANCES_TRACKING_INTERVAL);\n  }\n\n  /**\n   * Starts the tracking process.\n   */\n  start(): void {\n    this.#poller.start();\n  }\n\n  /**\n   * Stops the tracking process.\n   */\n  stop(): void {\n    this.#poller.stop();\n  }\n\n  /**\n   * Checks if an account ID is being tracked.\n   *\n   * @param accountId - The account ID.\n   * @returns True if the account is being tracker, false otherwise.\n   */\n  isTracked(accountId: string) {\n    return accountId in this.#balances;\n  }\n\n  /**\n   * Asserts that an account ID is being tracked.\n   *\n   * @param accountId - The account ID.\n   * @throws If the account ID is not being tracked.\n   */\n  assertBeingTracked(accountId: string) {\n    if (!this.isTracked(accountId)) {\n      throw new Error(`Account is not being tracked: ${accountId}`);\n    }\n  }\n\n  /**\n   * Starts tracking a new account ID. This method has no effect on already tracked\n   * accounts.\n   *\n   * @param accountId - The account ID.\n   * @param blockTime - The block time (used when refreshing the account balances).\n   */\n  track(accountId: string, blockTime: number) {\n    // Do not overwrite current info if already being tracked!\n    if (!this.isTracked(accountId)) {\n      this.#balances[accountId] = {\n        lastUpdated: 0,\n        blockTime,\n      };\n    }\n  }\n\n  /**\n   * Stops tracking a tracked account ID.\n   *\n   * @param accountId - The account ID.\n   * @throws If the account ID is not being tracked.\n   */\n  untrack(accountId: string) {\n    this.assertBeingTracked(accountId);\n    delete this.#balances[accountId];\n  }\n\n  /**\n   * Update the balances for a tracked account ID.\n   *\n   * @param accountId - The account ID.\n   * @throws If the account ID is not being tracked.\n   */\n  async updateBalance(accountId: string) {\n    this.assertBeingTracked(accountId);\n\n    // We check if the balance is outdated (by comparing to the block time associated\n    // with this kind of account).\n    //\n    // This might not be super accurate, but we could probably compute this differently\n    // and try to sync with the \"real block time\"!\n    const info = this.#balances[accountId];\n    const isOutdated = Date.now() - info.lastUpdated >= info.blockTime;\n    const hasNoBalanceYet = info.lastUpdated === 0;\n    if (hasNoBalanceYet || isOutdated) {\n      await this.#updateBalance(accountId);\n      this.#balances[accountId].lastUpdated = Date.now();\n    }\n  }\n\n  /**\n   * Update the balances of all tracked accounts (only if the balances\n   * is considered outdated).\n   */\n  async updateBalances() {\n    await Promise.allSettled(\n      Object.keys(this.#balances).map(async (accountId) => {\n        await this.updateBalance(accountId);\n      }),\n    );\n  }\n}\n"]}

package/dist/MultichainBalancesController/BalancesTracker.d.cts

@@ -0,0 +1,54 @@
+export declare class BalancesTracker {
+    #private;
+    constructor(updateBalanceCallback: (accountId: string) => Promise<void>);
+    /**
+     * Starts the tracking process.
+     */
+    start(): void;
+    /**
+     * Stops the tracking process.
+     */
+    stop(): void;
+    /**
+     * Checks if an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @returns True if the account is being tracker, false otherwise.
+     */
+    isTracked(accountId: string): boolean;
+    /**
+     * Asserts that an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    assertBeingTracked(accountId: string): void;
+    /**
+     * Starts tracking a new account ID. This method has no effect on already tracked
+     * accounts.
+     *
+     * @param accountId - The account ID.
+     * @param blockTime - The block time (used when refreshing the account balances).
+     */
+    track(accountId: string, blockTime: number): void;
+    /**
+     * Stops tracking a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    untrack(accountId: string): void;
+    /**
+     * Update the balances for a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    updateBalance(accountId: string): Promise<void>;
+    /**
+     * Update the balances of all tracked accounts (only if the balances
+     * is considered outdated).
+     */
+    updateBalances(): Promise<void>;
+}
+//# sourceMappingURL=BalancesTracker.d.cts.map

package/dist/MultichainBalancesController/BalancesTracker.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BalancesTracker.d.cts","sourceRoot":"","sources":["../../src/MultichainBalancesController/BalancesTracker.ts"],"names":[],"mappings":"AASA,qBAAa,eAAe;;gBAOd,qBAAqB,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC;IAMvE;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;OAEG;IACH,IAAI,IAAI,IAAI;IAIZ;;;;;OAKG;IACH,SAAS,CAAC,SAAS,EAAE,MAAM;IAI3B;;;;;OAKG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM;IAMpC;;;;;;OAMG;IACH,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAU1C;;;;;OAKG;IACH,OAAO,CAAC,SAAS,EAAE,MAAM;IAKzB;;;;;OAKG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM;IAiBrC;;;OAGG;IACG,cAAc;CAOrB"}

package/dist/MultichainBalancesController/BalancesTracker.d.mts

@@ -0,0 +1,54 @@
+export declare class BalancesTracker {
+    #private;
+    constructor(updateBalanceCallback: (accountId: string) => Promise<void>);
+    /**
+     * Starts the tracking process.
+     */
+    start(): void;
+    /**
+     * Stops the tracking process.
+     */
+    stop(): void;
+    /**
+     * Checks if an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @returns True if the account is being tracker, false otherwise.
+     */
+    isTracked(accountId: string): boolean;
+    /**
+     * Asserts that an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    assertBeingTracked(accountId: string): void;
+    /**
+     * Starts tracking a new account ID. This method has no effect on already tracked
+     * accounts.
+     *
+     * @param accountId - The account ID.
+     * @param blockTime - The block time (used when refreshing the account balances).
+     */
+    track(accountId: string, blockTime: number): void;
+    /**
+     * Stops tracking a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    untrack(accountId: string): void;
+    /**
+     * Update the balances for a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    updateBalance(accountId: string): Promise<void>;
+    /**
+     * Update the balances of all tracked accounts (only if the balances
+     * is considered outdated).
+     */
+    updateBalances(): Promise<void>;
+}
+//# sourceMappingURL=BalancesTracker.d.mts.map

package/dist/MultichainBalancesController/BalancesTracker.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BalancesTracker.d.mts","sourceRoot":"","sources":["../../src/MultichainBalancesController/BalancesTracker.ts"],"names":[],"mappings":"AASA,qBAAa,eAAe;;gBAOd,qBAAqB,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC;IAMvE;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;OAEG;IACH,IAAI,IAAI,IAAI;IAIZ;;;;;OAKG;IACH,SAAS,CAAC,SAAS,EAAE,MAAM;IAI3B;;;;;OAKG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM;IAMpC;;;;;;OAMG;IACH,KAAK,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAU1C;;;;;OAKG;IACH,OAAO,CAAC,SAAS,EAAE,MAAM;IAKzB;;;;;OAKG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM;IAiBrC;;;OAGG;IACG,cAAc;CAOrB"}

package/dist/MultichainBalancesController/BalancesTracker.mjs

@@ -0,0 +1,113 @@
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _BalancesTracker_poller, _BalancesTracker_updateBalance, _BalancesTracker_balances;
+import { Poller } from "./Poller.mjs";
+const BALANCES_TRACKING_INTERVAL = 30 * 1000; // Every 30s in milliseconds.
+export class BalancesTracker {
+    constructor(updateBalanceCallback) {
+        _BalancesTracker_poller.set(this, void 0);
+        _BalancesTracker_updateBalance.set(this, void 0);
+        _BalancesTracker_balances.set(this, {});
+        __classPrivateFieldSet(this, _BalancesTracker_updateBalance, updateBalanceCallback, "f");
+        __classPrivateFieldSet(this, _BalancesTracker_poller, new Poller(this.updateBalances, BALANCES_TRACKING_INTERVAL), "f");
+    }
+    /**
+     * Starts the tracking process.
+     */
+    start() {
+        __classPrivateFieldGet(this, _BalancesTracker_poller, "f").start();
+    }
+    /**
+     * Stops the tracking process.
+     */
+    stop() {
+        __classPrivateFieldGet(this, _BalancesTracker_poller, "f").stop();
+    }
+    /**
+     * Checks if an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @returns True if the account is being tracker, false otherwise.
+     */
+    isTracked(accountId) {
+        return accountId in __classPrivateFieldGet(this, _BalancesTracker_balances, "f");
+    }
+    /**
+     * Asserts that an account ID is being tracked.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    assertBeingTracked(accountId) {
+        if (!this.isTracked(accountId)) {
+            throw new Error(`Account is not being tracked: ${accountId}`);
+        }
+    }
+    /**
+     * Starts tracking a new account ID. This method has no effect on already tracked
+     * accounts.
+     *
+     * @param accountId - The account ID.
+     * @param blockTime - The block time (used when refreshing the account balances).
+     */
+    track(accountId, blockTime) {
+        // Do not overwrite current info if already being tracked!
+        if (!this.isTracked(accountId)) {
+            __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId] = {
+                lastUpdated: 0,
+                blockTime,
+            };
+        }
+    }
+    /**
+     * Stops tracking a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    untrack(accountId) {
+        this.assertBeingTracked(accountId);
+        delete __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId];
+    }
+    /**
+     * Update the balances for a tracked account ID.
+     *
+     * @param accountId - The account ID.
+     * @throws If the account ID is not being tracked.
+     */
+    async updateBalance(accountId) {
+        this.assertBeingTracked(accountId);
+        // We check if the balance is outdated (by comparing to the block time associated
+        // with this kind of account).
+        //
+        // This might not be super accurate, but we could probably compute this differently
+        // and try to sync with the "real block time"!
+        const info = __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId];
+        const isOutdated = Date.now() - info.lastUpdated >= info.blockTime;
+        const hasNoBalanceYet = info.lastUpdated === 0;
+        if (hasNoBalanceYet || isOutdated) {
+            await __classPrivateFieldGet(this, _BalancesTracker_updateBalance, "f").call(this, accountId);
+            __classPrivateFieldGet(this, _BalancesTracker_balances, "f")[accountId].lastUpdated = Date.now();
+        }
+    }
+    /**
+     * Update the balances of all tracked accounts (only if the balances
+     * is considered outdated).
+     */
+    async updateBalances() {
+        await Promise.allSettled(Object.keys(__classPrivateFieldGet(this, _BalancesTracker_balances, "f")).map(async (accountId) => {
+            await this.updateBalance(accountId);
+        }));
+    }
+}
+_BalancesTracker_poller = new WeakMap(), _BalancesTracker_updateBalance = new WeakMap(), _BalancesTracker_balances = new WeakMap();
+//# sourceMappingURL=BalancesTracker.mjs.map

package/dist/MultichainBalancesController/BalancesTracker.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"BalancesTracker.mjs","sourceRoot":"","sources":["../../src/MultichainBalancesController/BalancesTracker.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,MAAM,EAAE,qBAAiB;AAOlC,MAAM,0BAA0B,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,6BAA6B;AAE3E,MAAM,OAAO,eAAe;IAO1B,YAAY,qBAA2D;QANvE,0CAAgB;QAEhB,iDAAqD;QAErD,oCAAyC,EAAE,EAAC;QAG1C,uBAAA,IAAI,kCAAkB,qBAAqB,MAAA,CAAC;QAE5C,uBAAA,IAAI,2BAAW,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,0BAA0B,CAAC,MAAA,CAAC;IAC7E,CAAC;IAED;;OAEG;IACH,KAAK;QACH,uBAAA,IAAI,+BAAQ,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,uBAAA,IAAI,+BAAQ,CAAC,IAAI,EAAE,CAAC;IACtB,CAAC;IAED;;;;;OAKG;IACH,SAAS,CAAC,SAAiB;QACzB,OAAO,SAAS,IAAI,uBAAA,IAAI,iCAAU,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,kBAAkB,CAAC,SAAiB;QAClC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC;SAC/D;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,SAAiB,EAAE,SAAiB;QACxC,0DAA0D;QAC1D,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;YAC9B,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,GAAG;gBAC1B,WAAW,EAAE,CAAC;gBACd,SAAS;aACV,CAAC;SACH;IACH,CAAC;IAED;;;;;OAKG;IACH,OAAO,CAAC,SAAiB;QACvB,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACnC,OAAO,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAEnC,iFAAiF;QACjF,8BAA8B;QAC9B,EAAE;QACF,mFAAmF;QACnF,8CAA8C;QAC9C,MAAM,IAAI,GAAG,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC;QACnE,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,KAAK,CAAC,CAAC;QAC/C,IAAI,eAAe,IAAI,UAAU,EAAE;YACjC,MAAM,uBAAA,IAAI,sCAAe,MAAnB,IAAI,EAAgB,SAAS,CAAC,CAAC;YACrC,uBAAA,IAAI,iCAAU,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;SACpD;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,CAAC,UAAU,CACtB,MAAM,CAAC,IAAI,CAAC,uBAAA,IAAI,iCAAU,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;YAClD,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["import { Poller } from './Poller';\n\ntype BalanceInfo = {\n  lastUpdated: number;\n  blockTime: number;\n};\n\nconst BALANCES_TRACKING_INTERVAL = 30 * 1000; // Every 30s in milliseconds.\n\nexport class BalancesTracker {\n  #poller: Poller;\n\n  #updateBalance: (accountId: string) => Promise<void>;\n\n  #balances: Record<string, BalanceInfo> = {};\n\n  constructor(updateBalanceCallback: (accountId: string) => Promise<void>) {\n    this.#updateBalance = updateBalanceCallback;\n\n    this.#poller = new Poller(this.updateBalances, BALANCES_TRACKING_INTERVAL);\n  }\n\n  /**\n   * Starts the tracking process.\n   */\n  start(): void {\n    this.#poller.start();\n  }\n\n  /**\n   * Stops the tracking process.\n   */\n  stop(): void {\n    this.#poller.stop();\n  }\n\n  /**\n   * Checks if an account ID is being tracked.\n   *\n   * @param accountId - The account ID.\n   * @returns True if the account is being tracker, false otherwise.\n   */\n  isTracked(accountId: string) {\n    return accountId in this.#balances;\n  }\n\n  /**\n   * Asserts that an account ID is being tracked.\n   *\n   * @param accountId - The account ID.\n   * @throws If the account ID is not being tracked.\n   */\n  assertBeingTracked(accountId: string) {\n    if (!this.isTracked(accountId)) {\n      throw new Error(`Account is not being tracked: ${accountId}`);\n    }\n  }\n\n  /**\n   * Starts tracking a new account ID. This method has no effect on already tracked\n   * accounts.\n   *\n   * @param accountId - The account ID.\n   * @param blockTime - The block time (used when refreshing the account balances).\n   */\n  track(accountId: string, blockTime: number) {\n    // Do not overwrite current info if already being tracked!\n    if (!this.isTracked(accountId)) {\n      this.#balances[accountId] = {\n        lastUpdated: 0,\n        blockTime,\n      };\n    }\n  }\n\n  /**\n   * Stops tracking a tracked account ID.\n   *\n   * @param accountId - The account ID.\n   * @throws If the account ID is not being tracked.\n   */\n  untrack(accountId: string) {\n    this.assertBeingTracked(accountId);\n    delete this.#balances[accountId];\n  }\n\n  /**\n   * Update the balances for a tracked account ID.\n   *\n   * @param accountId - The account ID.\n   * @throws If the account ID is not being tracked.\n   */\n  async updateBalance(accountId: string) {\n    this.assertBeingTracked(accountId);\n\n    // We check if the balance is outdated (by comparing to the block time associated\n    // with this kind of account).\n    //\n    // This might not be super accurate, but we could probably compute this differently\n    // and try to sync with the \"real block time\"!\n    const info = this.#balances[accountId];\n    const isOutdated = Date.now() - info.lastUpdated >= info.blockTime;\n    const hasNoBalanceYet = info.lastUpdated === 0;\n    if (hasNoBalanceYet || isOutdated) {\n      await this.#updateBalance(accountId);\n      this.#balances[accountId].lastUpdated = Date.now();\n    }\n  }\n\n  /**\n   * Update the balances of all tracked accounts (only if the balances\n   * is considered outdated).\n   */\n  async updateBalances() {\n    await Promise.allSettled(\n      Object.keys(this.#balances).map(async (accountId) => {\n        await this.updateBalance(accountId);\n      }),\n    );\n  }\n}\n"]}

package/dist/MultichainBalancesController/MultichainBalancesController.cjs

@@ -0,0 +1,203 @@
+"use strict";
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var _MultichainBalancesController_instances, _MultichainBalancesController_tracker, _MultichainBalancesController_networkAssetsMap, _MultichainBalancesController_trackAccount, _MultichainBalancesController_listMultichainAccounts, _MultichainBalancesController_listAccounts, _MultichainBalancesController_getAccount, _MultichainBalancesController_updateBalance, _MultichainBalancesController_isNonEvmAccount, _MultichainBalancesController_handleOnAccountAdded, _MultichainBalancesController_handleOnAccountRemoved, _MultichainBalancesController_getBalances, _MultichainBalancesController_getClient;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MultichainBalancesController = exports.defaultState = void 0;
+const base_controller_1 = require("@metamask/base-controller");
+const keyring_api_1 = require("@metamask/keyring-api");
+const snaps_utils_1 = require("@metamask/snaps-utils");
+const _1 = require("./index.cjs");
+const utils_1 = require("./utils.cjs");
+const controllerName = 'MultichainBalancesController';
+/**
+ * Default state of the {@link MultichainBalancesController}.
+ */
+exports.defaultState = { balances: {} };
+/**
+ * {@link multichainBalancesController}'s metadata.
+ *
+ * This allows us to choose if fields of the state should be persisted or not
+ * using the `persist` flag; and if they can be sent to Sentry or not, using
+ * the `anonymous` flag.
+ */
+const multichainBalancesControllerMetadata = {
+    balances: {
+        persist: true,
+        anonymous: false,
+    },
+};
+/**
+ * The MultichainBalancesController is responsible for fetching and caching account
+ * balances.
+ */
+class MultichainBalancesController extends base_controller_1.BaseController {
+    constructor({ messenger, state, }) {
+        super({
+            messenger,
+            name: controllerName,
+            metadata: multichainBalancesControllerMetadata,
+            state: {
+                ...exports.defaultState,
+                ...state,
+            },
+        });
+        _MultichainBalancesController_instances.add(this);
+        _MultichainBalancesController_tracker.set(this, void 0);
+        // As a temporary solution, we are using a map to store the assets
+        // that are hardcoded in the module. In the future, this mapping
+        // should be dynamic to allow a client to register and unregister assets
+        _MultichainBalancesController_networkAssetsMap.set(this, _1.NETWORK_ASSETS_MAP);
+        __classPrivateFieldSet(this, _MultichainBalancesController_tracker, new _1.BalancesTracker(async (accountId) => await __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_updateBalance).call(this, accountId)), "f");
+        // Register all non-EVM accounts into the tracker
+        for (const account of __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_listAccounts).call(this)) {
+            __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_trackAccount).call(this, account);
+        }
+        this.messagingSystem.subscribe('AccountsController:accountAdded', (account) => __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_handleOnAccountAdded).call(this, account));
+        this.messagingSystem.subscribe('AccountsController:accountRemoved', (account) => __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_handleOnAccountRemoved).call(this, account));
+    }
+    /**
+     * Starts the polling process.
+     */
+    start() {
+        __classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").start();
+    }
+    /**
+     * Stops the polling process.
+     */
+    stop() {
+        __classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").stop();
+    }
+    /**
+     * Updates the balances of one account. This method doesn't return
+     * anything, but it updates the state of the controller.
+     *
+     * @param accountId - The account ID.
+     */
+    async updateBalance(accountId) {
+        // NOTE: No need to track the account here, since we start tracking those when
+        // the "AccountsController:accountAdded" is fired.
+        await __classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").updateBalance(accountId);
+    }
+    /**
+     * Updates the balances of all supported accounts. This method doesn't return
+     * anything, but it updates the state of the controller.
+     */
+    async updateBalances() {
+        await __classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").updateBalances();
+    }
+}
+exports.MultichainBalancesController = MultichainBalancesController;
+_MultichainBalancesController_tracker = new WeakMap(), _MultichainBalancesController_networkAssetsMap = new WeakMap(), _MultichainBalancesController_instances = new WeakSet(), _MultichainBalancesController_trackAccount = function _MultichainBalancesController_trackAccount(account) {
+    if (!__classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_isNonEvmAccount).call(this, account)) {
+        // Nothing to do here for EVM accounts
+        return;
+    }
+    const updateTime = 
+    // @ts-expect-error - For the moment we are only tracking non-EVM accounts and this is
+    // checked with the method `#isNonEvmAccount`. We can ignore this error since
+    // eip155:eoa and eip155:erc4337 account type balances are not tracked here.
+    _1.BALANCE_UPDATE_INTERVALS[__classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_getAccount).call(this, account.id).type];
+    __classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").track(account.id, updateTime);
+}, _MultichainBalancesController_listMultichainAccounts = function _MultichainBalancesController_listMultichainAccounts() {
+    return this.messagingSystem.call('AccountsController:listMultichainAccounts');
+}, _MultichainBalancesController_listAccounts = function _MultichainBalancesController_listAccounts() {
+    const accounts = __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_listMultichainAccounts).call(this);
+    return accounts.filter((account) => __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_isNonEvmAccount).call(this, account));
+}, _MultichainBalancesController_getAccount = function _MultichainBalancesController_getAccount(accountId) {
+    const account = __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_listMultichainAccounts).call(this).find((multichainAccount) => multichainAccount.id === accountId);
+    if (!account) {
+        throw new Error(`Unknown account: ${accountId}`);
+    }
+    if (!__classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_isNonEvmAccount).call(this, account)) {
+        throw new Error(`Account is not a non-EVM account: ${accountId}`);
+    }
+    return account;
+}, _MultichainBalancesController_updateBalance = 
+/**
+ * Updates the balances of one account. This method doesn't return
+ * anything, but it updates the state of the controller.
+ *
+ * @param accountId - The account ID.
+ */
+async function _MultichainBalancesController_updateBalance(accountId) {
+    const account = __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_getAccount).call(this, accountId);
+    const partialState = { balances: {} };
+    const scope = (0, utils_1.getScopeForAddress)(account);
+    const assetsList = __classPrivateFieldGet(this, _MultichainBalancesController_networkAssetsMap, "f")[scope];
+    if (account.metadata.snap) {
+        partialState.balances[account.id] = await __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_getBalances).call(this, account.id, account.metadata.snap.id, assetsList);
+    }
+    this.update((state) => ({
+        ...state,
+        balances: {
+            ...state.balances,
+            ...partialState.balances,
+        },
+    }));
+}, _MultichainBalancesController_isNonEvmAccount = function _MultichainBalancesController_isNonEvmAccount(account) {
+    return (!(0, keyring_api_1.isEvmAccountType)(account.type) &&
+        // Non-EVM accounts are backed by a Snap for now
+        account.metadata.snap !== undefined);
+}, _MultichainBalancesController_handleOnAccountAdded = 
+/**
+ * Handles changes when a new account has been added.
+ *
+ * @param account - The new account being added.
+ */
+async function _MultichainBalancesController_handleOnAccountAdded(account) {
+    __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_trackAccount).call(this, account);
+    // NOTE: Unfortunately, we cannot update the balance right away here, because
+    // messenger's events are running synchronously and fetching the balance is
+    // asynchronous.
+    // Updating the balance here would resume at some point but the event emitter
+    // will not `await` this (so we have no real control "when" the balance will
+    // really be updated), see:
+    // - https://github.com/MetaMask/core/blob/v213.0.0/packages/accounts-controller/src/AccountsController.ts#L1036-L1039
+}, _MultichainBalancesController_handleOnAccountRemoved = 
+/**
+ * Handles changes when a new account has been removed.
+ *
+ * @param accountId - The account ID being removed.
+ */
+async function _MultichainBalancesController_handleOnAccountRemoved(accountId) {
+    if (__classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").isTracked(accountId)) {
+        __classPrivateFieldGet(this, _MultichainBalancesController_tracker, "f").untrack(accountId);
+    }
+    if (accountId in this.state.balances) {
+        this.update((state) => {
+            delete state.balances[accountId];
+            return state;
+        });
+    }
+}, _MultichainBalancesController_getBalances = 
+/**
+ * Get the balances for an account.
+ *
+ * @param accountId - ID of the account to get balances for.
+ * @param snapId - ID of the Snap which manages the account.
+ * @param assetTypes - Array of asset types to get balances for.
+ * @returns A map of asset types to balances.
+ */
+async function _MultichainBalancesController_getBalances(accountId, snapId, assetTypes) {
+    return await __classPrivateFieldGet(this, _MultichainBalancesController_instances, "m", _MultichainBalancesController_getClient).call(this, snapId).getAccountBalances(accountId, assetTypes);
+}, _MultichainBalancesController_getClient = function _MultichainBalancesController_getClient(snapId) {
+    return new keyring_api_1.KeyringClient({
+        send: async (request) => (await this.messagingSystem.call('SnapController:handleRequest', {
+            snapId: snapId,
+            origin: 'metamask',
+            handler: snaps_utils_1.HandlerType.OnKeyringRequest,
+            request,
+        })),
+    });
+};
+//# sourceMappingURL=MultichainBalancesController.cjs.map

package/dist/MultichainBalancesController/MultichainBalancesController.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"MultichainBalancesController.cjs","sourceRoot":"","sources":["../../src/MultichainBalancesController/MultichainBalancesController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAKA,+DAKmC;AACnC,uDAM+B;AAG/B,uDAAoD;AAIpD,kCAIW;AACX,uCAA6C;AAE7C,MAAM,cAAc,GAAG,8BAA8B,CAAC;AAgBtD;;GAEG;AACU,QAAA,YAAY,GAAsC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;AAmEhF;;;;;;GAMG;AACH,MAAM,oCAAoC,GAAG;IAC3C,QAAQ,EAAE;QACR,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAEF;;;GAGG;AACH,MAAa,4BAA6B,SAAQ,gCAIjD;IAQC,YAAY,EACV,SAAS,EACT,KAAK,GAIN;QACC,KAAK,CAAC;YACJ,SAAS;YACT,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,oCAAoC;YAC9C,KAAK,EAAE;gBACL,GAAG,oBAAY;gBACf,GAAG,KAAK;aACT;SACF,CAAC,CAAC;;QAtBI,wDAA0B;QAEnC,kEAAkE;QAClE,gEAAgE;QAChE,wEAAwE;QAC/D,yDAA8C,qBAAkB,EAAC;QAmBxE,uBAAA,IAAI,yCAAY,IAAI,kBAAe,CACjC,KAAK,EAAE,SAAiB,EAAE,EAAE,CAAC,MAAM,uBAAA,IAAI,4FAAe,MAAnB,IAAI,EAAgB,SAAS,CAAC,CAClE,MAAA,CAAC;QAEF,iDAAiD;QACjD,KAAK,MAAM,OAAO,IAAI,uBAAA,IAAI,2FAAc,MAAlB,IAAI,CAAgB,EAAE;YAC1C,uBAAA,IAAI,2FAAc,MAAlB,IAAI,EAAe,OAAO,CAAC,CAAC;SAC7B;QAED,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,CAAC,OAAO,EAAE,EAAE,CAAC,uBAAA,IAAI,mGAAsB,MAA1B,IAAI,EAAuB,OAAO,CAAC,CACjD,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,mCAAmC,EACnC,CAAC,OAAO,EAAE,EAAE,CAAC,uBAAA,IAAI,qGAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,CACnD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,uBAAA,IAAI,6CAAS,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,uBAAA,IAAI,6CAAS,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IAgGD;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,8EAA8E;QAC9E,kDAAkD;QAClD,MAAM,uBAAA,IAAI,6CAAS,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,uBAAA,IAAI,6CAAS,CAAC,cAAc,EAAE,CAAC;IACvC,CAAC;CAsFF;AApQD,oEAoQC;iRAjMe,OAAwB;IACpC,IAAI,CAAC,uBAAA,IAAI,8FAAiB,MAArB,IAAI,EAAkB,OAAO,CAAC,EAAE;QACnC,sCAAsC;QACtC,OAAO;KACR;IAED,MAAM,UAAU;IACd,sFAAsF;IACtF,6EAA6E;IAC7E,4EAA4E;IAC5E,2BAAwB,CAAC,uBAAA,IAAI,yFAAY,MAAhB,IAAI,EAAa,OAAO,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;IAC9D,uBAAA,IAAI,6CAAS,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;AAC9C,CAAC;IAQC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAC9B,2CAA2C,CAC5C,CAAC;AACJ,CAAC;IAUC,MAAM,QAAQ,GAAG,uBAAA,IAAI,qGAAwB,MAA5B,IAAI,CAA0B,CAAC;IAEhD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,uBAAA,IAAI,8FAAiB,MAArB,IAAI,EAAkB,OAAO,CAAC,CAAC,CAAC;AACtE,CAAC,+FAQW,SAAiB;IAC3B,MAAM,OAAO,GACX,uBAAA,IAAI,qGAAwB,MAA5B,IAAI,CAA0B,CAAC,IAAI,CACjC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,SAAS,CAC1D,CAAC;IAEJ,IAAI,CAAC,OAAO,EAAE;QACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAC;KAClD;IACD,IAAI,CAAC,uBAAA,IAAI,8FAAiB,MAArB,IAAI,EAAkB,OAAO,CAAC,EAAE;QACnC,MAAM,IAAI,KAAK,CAAC,qCAAqC,SAAS,EAAE,CAAC,CAAC;KACnE;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACH,KAAK,sDAAgB,SAAiB;IACpC,MAAM,OAAO,GAAG,uBAAA,IAAI,yFAAY,MAAhB,IAAI,EAAa,SAAS,CAAC,CAAC;IAC5C,MAAM,YAAY,GAAsC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAEzE,MAAM,KAAK,GAAG,IAAA,0BAAkB,EAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,UAAU,GAAG,uBAAA,IAAI,sDAAkB,CAAC,KAAK,CAAC,CAAC;IAEjD,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;QACzB,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG,MAAM,uBAAA,IAAI,0FAAa,MAAjB,IAAI,EAC5C,OAAO,CAAC,EAAE,EACV,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EACxB,UAAU,CACX,CAAC;KACH;IAED,IAAI,CAAC,MAAM,CAAC,CAAC,KAA+C,EAAE,EAAE,CAAC,CAAC;QAChE,GAAG,KAAK;QACR,QAAQ,EAAE;YACR,GAAG,KAAK,CAAC,QAAQ;YACjB,GAAG,YAAY,CAAC,QAAQ;SACzB;KACF,CAAC,CAAC,CAAC;AACN,CAAC,yGA4BgB,OAAwB;IACvC,OAAO,CACL,CAAC,IAAA,8BAAgB,EAAC,OAAO,CAAC,IAAI,CAAC;QAC/B,gDAAgD;QAChD,OAAO,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,CACpC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,KAAK,6DAAuB,OAAwB;IAClD,uBAAA,IAAI,2FAAc,MAAlB,IAAI,EAAe,OAAO,CAAC,CAAC;IAC5B,6EAA6E;IAC7E,2EAA2E;IAC3E,gBAAgB;IAChB,6EAA6E;IAC7E,4EAA4E;IAC5E,2BAA2B;IAC3B,sHAAsH;AACxH,CAAC;AAED;;;;GAIG;AACH,KAAK,+DAAyB,SAAiB;IAC7C,IAAI,uBAAA,IAAI,6CAAS,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE;QACtC,uBAAA,IAAI,6CAAS,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;KAClC;IAED,IAAI,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;QACpC,IAAI,CAAC,MAAM,CAAC,CAAC,KAA+C,EAAE,EAAE;YAC9D,OAAO,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;KACJ;AACH,CAAC;AAED;;;;;;;GAOG;AACH,KAAK,oDACH,SAAiB,EACjB,MAAc,EACd,UAA2B;IAE3B,OAAO,MAAM,uBAAA,IAAI,wFAAW,MAAf,IAAI,EAAY,MAAM,CAAC,CAAC,kBAAkB,CACrD,SAAS,EACT,UAAU,CACX,CAAC;AACJ,CAAC,6FAQU,MAAc;IACvB,OAAO,IAAI,2BAAa,CAAC;QACvB,IAAI,EAAE,KAAK,EAAE,OAAuB,EAAE,EAAE,CACtC,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAC/D,MAAM,EAAE,MAAgB;YACxB,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,yBAAW,CAAC,gBAAgB;YACrC,OAAO;SACR,CAAC,CAAkB;KACvB,CAAC,CAAC;AACL,CAAC","sourcesContent":["import type {\n  AccountsControllerAccountAddedEvent,\n  AccountsControllerAccountRemovedEvent,\n  AccountsControllerListMultichainAccountsAction,\n} from '@metamask/accounts-controller';\nimport {\n  BaseController,\n  type ControllerGetStateAction,\n  type ControllerStateChangeEvent,\n  type RestrictedControllerMessenger,\n} from '@metamask/base-controller';\nimport {\n  KeyringClient,\n  type Balance,\n  type CaipAssetType,\n  type InternalAccount,\n  isEvmAccountType,\n} from '@metamask/keyring-api';\nimport type { HandleSnapRequest } from '@metamask/snaps-controllers';\nimport type { SnapId } from '@metamask/snaps-sdk';\nimport { HandlerType } from '@metamask/snaps-utils';\nimport type { Json, JsonRpcRequest } from '@metamask/utils';\nimport type { Draft } from 'immer';\n\nimport {\n  BalancesTracker,\n  BALANCE_UPDATE_INTERVALS,\n  NETWORK_ASSETS_MAP,\n} from '.';\nimport { getScopeForAddress } from './utils';\n\nconst controllerName = 'MultichainBalancesController';\n\n/**\n * State used by the {@link MultichainBalancesController} to cache account balances.\n */\nexport type MultichainBalancesControllerState = {\n  balances: {\n    [account: string]: {\n      [asset: string]: {\n        amount: string;\n        unit: string;\n      };\n    };\n  };\n};\n\n/**\n * Default state of the {@link MultichainBalancesController}.\n */\nexport const defaultState: MultichainBalancesControllerState = { balances: {} };\n\n/**\n * Returns the state of the {@link MultichainBalancesController}.\n */\nexport type MultichainBalancesControllerGetStateAction =\n  ControllerGetStateAction<\n    typeof controllerName,\n    MultichainBalancesControllerState\n  >;\n\n/**\n * Updates the balances of all supported accounts in {@link MultichainBalancesController}.\n */\nexport type MultichainBalancesControllerUpdateBalancesAction = {\n  type: `${typeof controllerName}:updateBalances`;\n  handler: MultichainBalancesController['updateBalances'];\n};\n\n/**\n * Event emitted when the state of the {@link MultichainBalancesController} changes.\n */\nexport type MultichainBalancesControllerStateChange =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    MultichainBalancesControllerState\n  >;\n\n/**\n * Actions exposed by the {@link MultichainBalancesController}.\n */\nexport type MultichainBalancesControllerActions =\n  | MultichainBalancesControllerGetStateAction\n  | MultichainBalancesControllerUpdateBalancesAction;\n\n/**\n * Events emitted by {@link MultichainBalancesController}.\n */\nexport type MultichainBalancesControllerEvents =\n  MultichainBalancesControllerStateChange;\n\n/**\n * Actions that this controller is allowed to call.\n */\nexport type AllowedActions =\n  | HandleSnapRequest\n  | AccountsControllerListMultichainAccountsAction;\n\n/**\n * Events that this controller is allowed to subscribe.\n */\nexport type AllowedEvents =\n  | AccountsControllerAccountAddedEvent\n  | AccountsControllerAccountRemovedEvent;\n\n/**\n * Messenger type for the MultichainBalancesController.\n */\nexport type MultichainBalancesControllerMessenger =\n  RestrictedControllerMessenger<\n    typeof controllerName,\n    MultichainBalancesControllerActions | AllowedActions,\n    MultichainBalancesControllerEvents | AllowedEvents,\n    AllowedActions['type'],\n    AllowedEvents['type']\n  >;\n\n/**\n * {@link multichainBalancesController}'s metadata.\n *\n * This allows us to choose if fields of the state should be persisted or not\n * using the `persist` flag; and if they can be sent to Sentry or not, using\n * the `anonymous` flag.\n */\nconst multichainBalancesControllerMetadata = {\n  balances: {\n    persist: true,\n    anonymous: false,\n  },\n};\n\n/**\n * The MultichainBalancesController is responsible for fetching and caching account\n * balances.\n */\nexport class MultichainBalancesController extends BaseController<\n  typeof controllerName,\n  MultichainBalancesControllerState,\n  MultichainBalancesControllerMessenger\n> {\n  readonly #tracker: BalancesTracker;\n\n  // As a temporary solution, we are using a map to store the assets\n  // that are hardcoded in the module. In the future, this mapping\n  // should be dynamic to allow a client to register and unregister assets\n  readonly #networkAssetsMap: Record<string, string[]> = NETWORK_ASSETS_MAP;\n\n  constructor({\n    messenger,\n    state,\n  }: {\n    messenger: MultichainBalancesControllerMessenger;\n    state: MultichainBalancesControllerState;\n  }) {\n    super({\n      messenger,\n      name: controllerName,\n      metadata: multichainBalancesControllerMetadata,\n      state: {\n        ...defaultState,\n        ...state,\n      },\n    });\n\n    this.#tracker = new BalancesTracker(\n      async (accountId: string) => await this.#updateBalance(accountId),\n    );\n\n    // Register all non-EVM accounts into the tracker\n    for (const account of this.#listAccounts()) {\n      this.#trackAccount(account);\n    }\n\n    this.messagingSystem.subscribe(\n      'AccountsController:accountAdded',\n      (account) => this.#handleOnAccountAdded(account),\n    );\n    this.messagingSystem.subscribe(\n      'AccountsController:accountRemoved',\n      (account) => this.#handleOnAccountRemoved(account),\n    );\n  }\n\n  /**\n   * Starts the polling process.\n   */\n  start(): void {\n    this.#tracker.start();\n  }\n\n  /**\n   * Stops the polling process.\n   */\n  stop(): void {\n    this.#tracker.stop();\n  }\n\n  /**\n   * Tracks an account to get its balances.\n   *\n   * @param account - The account to track.\n   */\n  #trackAccount(account: InternalAccount): void {\n    if (!this.#isNonEvmAccount(account)) {\n      // Nothing to do here for EVM accounts\n      return;\n    }\n\n    const updateTime =\n      // @ts-expect-error - For the moment we are only tracking non-EVM accounts and this is\n      // checked with the method `#isNonEvmAccount`. We can ignore this error since\n      // eip155:eoa and eip155:erc4337 account type balances are not tracked here.\n      BALANCE_UPDATE_INTERVALS[this.#getAccount(account.id).type];\n    this.#tracker.track(account.id, updateTime);\n  }\n\n  /**\n   * Lists the multichain accounts coming from the `AccountsController`.\n   *\n   * @returns A list of multichain accounts.\n   */\n  #listMultichainAccounts(): InternalAccount[] {\n    return this.messagingSystem.call(\n      'AccountsController:listMultichainAccounts',\n    );\n  }\n\n  /**\n   * Lists the accounts that we should get balances for.\n   *\n   * Currently, we only get balances for non-EVM accounts.\n   *\n   * @returns A list of accounts that we should get balances for.\n   */\n  #listAccounts(): InternalAccount[] {\n    const accounts = this.#listMultichainAccounts();\n\n    return accounts.filter((account) => this.#isNonEvmAccount(account));\n  }\n\n  /**\n   * Get a non-EVM account from its ID.\n   *\n   * @param accountId - The account ID.\n   * @returns The non-EVM account.\n   */\n  #getAccount(accountId: string): InternalAccount {\n    const account: InternalAccount | undefined =\n      this.#listMultichainAccounts().find(\n        (multichainAccount) => multichainAccount.id === accountId,\n      );\n\n    if (!account) {\n      throw new Error(`Unknown account: ${accountId}`);\n    }\n    if (!this.#isNonEvmAccount(account)) {\n      throw new Error(`Account is not a non-EVM account: ${accountId}`);\n    }\n    return account;\n  }\n\n  /**\n   * Updates the balances of one account. This method doesn't return\n   * anything, but it updates the state of the controller.\n   *\n   * @param accountId - The account ID.\n   */\n  async #updateBalance(accountId: string) {\n    const account = this.#getAccount(accountId);\n    const partialState: MultichainBalancesControllerState = { balances: {} };\n\n    const scope = getScopeForAddress(account);\n    const assetsList = this.#networkAssetsMap[scope];\n\n    if (account.metadata.snap) {\n      partialState.balances[account.id] = await this.#getBalances(\n        account.id,\n        account.metadata.snap.id,\n        assetsList,\n      );\n    }\n\n    this.update((state: Draft<MultichainBalancesControllerState>) => ({\n      ...state,\n      balances: {\n        ...state.balances,\n        ...partialState.balances,\n      },\n    }));\n  }\n\n  /**\n   * Updates the balances of one account. This method doesn't return\n   * anything, but it updates the state of the controller.\n   *\n   * @param accountId - The account ID.\n   */\n  async updateBalance(accountId: string) {\n    // NOTE: No need to track the account here, since we start tracking those when\n    // the \"AccountsController:accountAdded\" is fired.\n    await this.#tracker.updateBalance(accountId);\n  }\n\n  /**\n   * Updates the balances of all supported accounts. This method doesn't return\n   * anything, but it updates the state of the controller.\n   */\n  async updateBalances() {\n    await this.#tracker.updateBalances();\n  }\n\n  /**\n   * Checks for non-EVM accounts.\n   *\n   * @param account - The new account to be checked.\n   * @returns True if the account is a non-EVM account, false otherwise.\n   */\n  #isNonEvmAccount(account: InternalAccount): boolean {\n    return (\n      !isEvmAccountType(account.type) &&\n      // Non-EVM accounts are backed by a Snap for now\n      account.metadata.snap !== undefined\n    );\n  }\n\n  /**\n   * Handles changes when a new account has been added.\n   *\n   * @param account - The new account being added.\n   */\n  async #handleOnAccountAdded(account: InternalAccount) {\n    this.#trackAccount(account);\n    // NOTE: Unfortunately, we cannot update the balance right away here, because\n    // messenger's events are running synchronously and fetching the balance is\n    // asynchronous.\n    // Updating the balance here would resume at some point but the event emitter\n    // will not `await` this (so we have no real control \"when\" the balance will\n    // really be updated), see:\n    // - https://github.com/MetaMask/core/blob/v213.0.0/packages/accounts-controller/src/AccountsController.ts#L1036-L1039\n  }\n\n  /**\n   * Handles changes when a new account has been removed.\n   *\n   * @param accountId - The account ID being removed.\n   */\n  async #handleOnAccountRemoved(accountId: string) {\n    if (this.#tracker.isTracked(accountId)) {\n      this.#tracker.untrack(accountId);\n    }\n\n    if (accountId in this.state.balances) {\n      this.update((state: Draft<MultichainBalancesControllerState>) => {\n        delete state.balances[accountId];\n        return state;\n      });\n    }\n  }\n\n  /**\n   * Get the balances for an account.\n   *\n   * @param accountId - ID of the account to get balances for.\n   * @param snapId - ID of the Snap which manages the account.\n   * @param assetTypes - Array of asset types to get balances for.\n   * @returns A map of asset types to balances.\n   */\n  async #getBalances(\n    accountId: string,\n    snapId: string,\n    assetTypes: CaipAssetType[],\n  ): Promise<Record<CaipAssetType, Balance>> {\n    return await this.#getClient(snapId).getAccountBalances(\n      accountId,\n      assetTypes,\n    );\n  }\n\n  /**\n   * Gets a `KeyringClient` for a Snap.\n   *\n   * @param snapId - ID of the Snap to get the client for.\n   * @returns A `KeyringClient` for the Snap.\n   */\n  #getClient(snapId: string): KeyringClient {\n    return new KeyringClient({\n      send: async (request: JsonRpcRequest) =>\n        (await this.messagingSystem.call('SnapController:handleRequest', {\n          snapId: snapId as SnapId,\n          origin: 'metamask',\n          handler: HandlerType.OnKeyringRequest,\n          request,\n        })) as Promise<Json>,\n    });\n  }\n}\n"]}