@metamask-previews/assets-controllers 65.0.0-preview-de21a3fe → 66.0.0-preview-4ef4efd7

package/CHANGELOG.md

@@ -7,8 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [66.0.0]
+
 ### Added
 
+- Add optional parameter to track DeFi metrics when positions are being fetched ([#5868](https://github.com/MetaMask/core/pull/5868))
 - Add phishing protection for NFT metadata URLs in `NftController` ([#5598](https://github.com/MetaMask/core/pull/5598))
   - NFT metadata URLs are now scanned for malicious content using the `PhishingController`
   - Malicious URLs in NFT metadata fields (image, externalLink, etc.) are automatically sanitized
@@ -1674,7 +1677,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Use Ethers for AssetsContractController ([#845](https://github.com/MetaMask/core/pull/845))
 
-[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@65.0.0...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@66.0.0...HEAD
+[66.0.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@65.0.0...@metamask/assets-controllers@66.0.0
 [65.0.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@64.0.0...@metamask/assets-controllers@65.0.0
 [64.0.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@63.1.0...@metamask/assets-controllers@64.0.0
 [63.1.0]: https://github.com/MetaMask/core/compare/@metamask/assets-controllers@63.0.0...@metamask/assets-controllers@63.1.0

package/dist/DeFiPositionsController/DeFiPositionsController.cjs

@@ -10,10 +10,11 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     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 _DeFiPositionsController_instances, _DeFiPositionsController_fetchPositions, _DeFiPositionsController_isEnabled, _DeFiPositionsController_updateAccountPositions, _DeFiPositionsController_fetchAccountPositions;
+var _DeFiPositionsController_instances, _DeFiPositionsController_fetchPositions, _DeFiPositionsController_isEnabled, _DeFiPositionsController_trackEvent, _DeFiPositionsController_updateAccountPositions, _DeFiPositionsController_fetchAccountPositions, _DeFiPositionsController_updatePositionsCountMetrics;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DeFiPositionsController = exports.getDefaultDefiPositionsControllerState = void 0;
 const polling_controller_1 = require("@metamask/polling-controller");
+const calculate_defi_metrics_1 = require("./calculate-defi-metrics.cjs");
 const fetch_positions_1 = require("./fetch-positions.cjs");
 const group_defi_positions_1 = require("./group-defi-positions.cjs");
 const assetsUtil_1 = require("../assetsUtil.cjs");
@@ -25,10 +26,15 @@ const controllerMetadata = {
         persist: false,
         anonymous: false,
     },
+    allDeFiPositionsCount: {
+        persist: false,
+        anonymous: false,
+    },
 };
 const getDefaultDefiPositionsControllerState = () => {
     return {
         allDeFiPositions: {},
+        allDeFiPositionsCount: {},
     };
 };
 exports.getDefaultDefiPositionsControllerState = getDefaultDefiPositionsControllerState;
@@ -42,8 +48,9 @@ class DeFiPositionsController extends (0, polling_controller_1.StaticIntervalPol
      * @param options - Constructor options.
      * @param options.messenger - The controller messenger.
      * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)
+     * @param options.trackEvent - Function to track events. (default: undefined)
      */
-    constructor({ messenger, isEnabled = () => true, }) {
+    constructor({ messenger, isEnabled = () => true, trackEvent, }) {
         super({
             name: controllerName,
             metadata: controllerMetadata,
@@ -53,6 +60,7 @@ class DeFiPositionsController extends (0, polling_controller_1.StaticIntervalPol
         _DeFiPositionsController_instances.add(this);
         _DeFiPositionsController_fetchPositions.set(this, void 0);
         _DeFiPositionsController_isEnabled.set(this, void 0);
+        _DeFiPositionsController_trackEvent.set(this, void 0);
         this.setIntervalLength(TEN_MINUTES_IN_MS);
         __classPrivateFieldSet(this, _DeFiPositionsController_fetchPositions, (0, fetch_positions_1.buildPositionFetcher)(), "f");
         __classPrivateFieldSet(this, _DeFiPositionsController_isEnabled, isEnabled, "f");
@@ -74,6 +82,7 @@ class DeFiPositionsController extends (0, polling_controller_1.StaticIntervalPol
             }
             await __classPrivateFieldGet(this, _DeFiPositionsController_instances, "m", _DeFiPositionsController_updateAccountPositions).call(this, account.address);
         });
+        __classPrivateFieldSet(this, _DeFiPositionsController_trackEvent, trackEvent, "f");
     }
     async _executePoll() {
         if (!__classPrivateFieldGet(this, _DeFiPositionsController_isEnabled, "f").call(this)) {
@@ -109,7 +118,7 @@ class DeFiPositionsController extends (0, polling_controller_1.StaticIntervalPol
     }
 }
 exports.DeFiPositionsController = DeFiPositionsController;
-_DeFiPositionsController_fetchPositions = new WeakMap(), _DeFiPositionsController_isEnabled = new WeakMap(), _DeFiPositionsController_instances = new WeakSet(), _DeFiPositionsController_updateAccountPositions = async function _DeFiPositionsController_updateAccountPositions(accountAddress) {
+_DeFiPositionsController_fetchPositions = new WeakMap(), _DeFiPositionsController_isEnabled = new WeakMap(), _DeFiPositionsController_trackEvent = new WeakMap(), _DeFiPositionsController_instances = new WeakSet(), _DeFiPositionsController_updateAccountPositions = async function _DeFiPositionsController_updateAccountPositions(accountAddress) {
     const accountPositionsPerChain = await __classPrivateFieldGet(this, _DeFiPositionsController_instances, "m", _DeFiPositionsController_fetchAccountPositions).call(this, accountAddress);
     this.update((state) => {
         state.allDeFiPositions[accountAddress] = accountPositionsPerChain;
@@ -117,10 +126,30 @@ _DeFiPositionsController_fetchPositions = new WeakMap(), _DeFiPositionsControlle
 }, _DeFiPositionsController_fetchAccountPositions = async function _DeFiPositionsController_fetchAccountPositions(accountAddress) {
     try {
         const defiPositionsResponse = await __classPrivateFieldGet(this, _DeFiPositionsController_fetchPositions, "f").call(this, accountAddress);
-        return (0, group_defi_positions_1.groupDeFiPositions)(defiPositionsResponse);
+        const groupedDeFiPositions = (0, group_defi_positions_1.groupDeFiPositions)(defiPositionsResponse);
+        try {
+            __classPrivateFieldGet(this, _DeFiPositionsController_instances, "m", _DeFiPositionsController_updatePositionsCountMetrics).call(this, groupedDeFiPositions, accountAddress);
+        }
+        catch (error) {
+            console.error(`Failed to update positions count for account ${accountAddress}:`, error);
+        }
+        return groupedDeFiPositions;
     }
     catch {
         return null;
     }
+}, _DeFiPositionsController_updatePositionsCountMetrics = function _DeFiPositionsController_updatePositionsCountMetrics(groupedDeFiPositions, accountAddress) {
+    // If no track event passed then skip the metrics update
+    if (!__classPrivateFieldGet(this, _DeFiPositionsController_trackEvent, "f")) {
+        return;
+    }
+    const defiMetrics = (0, calculate_defi_metrics_1.calculateDeFiPositionMetrics)(groupedDeFiPositions);
+    const { totalPositions } = defiMetrics.properties;
+    if (totalPositions !== this.state.allDeFiPositionsCount[accountAddress]) {
+        this.update((state) => {
+            state.allDeFiPositionsCount[accountAddress] = totalPositions;
+        });
+        __classPrivateFieldGet(this, _DeFiPositionsController_trackEvent, "f")?.call(this, defiMetrics);
+    }
 };
 //# sourceMappingURL=DeFiPositionsController.cjs.map

package/dist/DeFiPositionsController/DeFiPositionsController.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"DeFiPositionsController.cjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAYA,qEAA+E;AAK/E,2DAAyD;AACzD,qEAGgC;AAChC,kDAAwD;AAExD,MAAM,iBAAiB,GAAG,KAAM,CAAC;AAEjC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAEtC,MAAM,cAAc,GAAG,yBAAyB,CAAC;AAejD,MAAM,kBAAkB,GAAgD;IACtE,gBAAgB,EAAE;QAChB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAEK,MAAM,sCAAsC,GACjD,GAAiC,EAAE;IACjC,OAAO;QACL,gBAAgB,EAAE,EAAE;KACrB,CAAC;AACJ,CAAC,CAAC;AALS,QAAA,sCAAsC,0CAK/C;AA4CJ;;GAEG;AACH,MAAa,uBAAwB,SAAQ,IAAA,oDAA+B,GAI3E;IAOC;;;;;;OAMG;IACH,YAAY,EACV,SAAS,EACT,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,GAIvB;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,kBAAkB;YAC5B,SAAS;YACT,KAAK,EAAE,IAAA,8CAAsC,GAAE;SAChD,CAAC,CAAC;;QAzBI,0DAE4B;QAE5B,qDAA0B;QAuBjC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1C,uBAAA,IAAI,2CAAmB,IAAA,sCAAoB,GAAE,MAAA,CAAC;QAC9C,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAE5B,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAC9D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,4CAA4C,EAC5C,KAAK,EAAE,eAAe,EAAE,EAAE;YACxB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;gBACtB,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,KAAK,EAAE,OAAO,EAAE,EAAE;YAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;gBAC7D,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,OAAO,CAAC,CAAC;QACtD,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;YACtB,OAAO;SACR;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACxC,iCAAiC,CAClC,CAAC;QAEF,MAAM,aAAa,GAGb,EAAE,CAAC;QAET,MAAM,OAAO,GAAG,MAAM,IAAA,oCAAuB,EAAC;YAC5C,aAAa;YACb,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,0BAA0B;YACrC,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,YAAY,GAAG,CACnB,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,EAAE;oBACpD,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;wBAC9B,MAAM,SAAS,GACb,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;wBAEpD,OAAO;4BACL,cAAc;4BACd,SAAS;yBACV,CAAC;qBACH;oBAED,OAAO,SAAS,CAAC;gBACnB,CAAC,CAAC,CACH,CACF,CAAC,MAAM,CAAC,OAAO,CAGb,CAAC;gBAEJ,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,OAAO,CAAC,MAAM,CACrC,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,EAAE,EAAE;YACrC,GAAG,CAAC,cAAc,CAAC,GAAG,SAAS,CAAC;YAChC,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAsD,CACvD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CAsBF;AAjJD,0DAiJC;mNApBC,KAAK,0DAAyB,cAAsB;IAClD,MAAM,wBAAwB,GAC5B,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;IAEpD,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,KAAK,CAAC,gBAAgB,CAAC,cAAc,CAAC,GAAG,wBAAwB,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,mDAED,KAAK,yDACH,cAAsB;IAEtB,IAAI;QACF,MAAM,qBAAqB,GAAG,MAAM,uBAAA,IAAI,+CAAgB,MAApB,IAAI,EAAiB,cAAc,CAAC,CAAC;QAEzE,OAAO,IAAA,yCAAkB,EAAC,qBAAqB,CAAC,CAAC;KAClD;IAAC,MAAM;QACN,OAAO,IAAI,CAAC;KACb;AACH,CAAC","sourcesContent":["import type {\n  AccountsControllerAccountAddedEvent,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport type { KeyringControllerUnlockEvent } from '@metamask/keyring-controller';\nimport type { KeyringControllerLockEvent } from '@metamask/keyring-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { TransactionControllerTransactionConfirmedEvent } from '@metamask/transaction-controller';\nimport type { Hex } from '@metamask/utils';\n\nimport type { DefiPositionResponse } from './fetch-positions';\nimport { buildPositionFetcher } from './fetch-positions';\nimport {\n  groupDeFiPositions,\n  type GroupedDeFiPositions,\n} from './group-defi-positions';\nimport { reduceInBatchesSerially } from '../assetsUtil';\n\nconst TEN_MINUTES_IN_MS = 60_000;\n\nconst FETCH_POSITIONS_BATCH_SIZE = 10;\n\nconst controllerName = 'DeFiPositionsController';\n\ntype GroupedDeFiPositionsPerChain = {\n  [chain: Hex]: GroupedDeFiPositions;\n};\n\nexport type DeFiPositionsControllerState = {\n  /**\n   * Object containing DeFi positions per account and network\n   */\n  allDeFiPositions: {\n    [accountAddress: string]: GroupedDeFiPositionsPerChain | null;\n  };\n};\n\nconst controllerMetadata: StateMetadata<DeFiPositionsControllerState> = {\n  allDeFiPositions: {\n    persist: false,\n    anonymous: false,\n  },\n};\n\nexport const getDefaultDefiPositionsControllerState =\n  (): DeFiPositionsControllerState => {\n    return {\n      allDeFiPositions: {},\n    };\n  };\n\nexport type DeFiPositionsControllerActions =\n  DeFiPositionsControllerGetStateAction;\n\nexport type DeFiPositionsControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  DeFiPositionsControllerState\n>;\n\nexport type DeFiPositionsControllerEvents =\n  DeFiPositionsControllerStateChangeEvent;\n\nexport type DeFiPositionsControllerStateChangeEvent =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    DeFiPositionsControllerState\n  >;\n\n/**\n * The external actions available to the {@link DeFiPositionsController}.\n */\nexport type AllowedActions = AccountsControllerListAccountsAction;\n\n/**\n * The external events available to the {@link DeFiPositionsController}.\n */\nexport type AllowedEvents =\n  | KeyringControllerUnlockEvent\n  | KeyringControllerLockEvent\n  | TransactionControllerTransactionConfirmedEvent\n  | AccountsControllerAccountAddedEvent;\n\n/**\n * The messenger of the {@link DeFiPositionsController}.\n */\nexport type DeFiPositionsControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  DeFiPositionsControllerActions | AllowedActions,\n  DeFiPositionsControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/**\n * Controller that stores assets and exposes convenience methods\n */\nexport class DeFiPositionsController extends StaticIntervalPollingController()<\n  typeof controllerName,\n  DeFiPositionsControllerState,\n  DeFiPositionsControllerMessenger\n> {\n  readonly #fetchPositions: (\n    accountAddress: string,\n  ) => Promise<DefiPositionResponse[]>;\n\n  readonly #isEnabled: () => boolean;\n\n  /**\n   * DeFiPositionsController constuctor\n   *\n   * @param options - Constructor options.\n   * @param options.messenger - The controller messenger.\n   * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)\n   */\n  constructor({\n    messenger,\n    isEnabled = () => true,\n  }: {\n    messenger: DeFiPositionsControllerMessenger;\n    isEnabled?: () => boolean;\n  }) {\n    super({\n      name: controllerName,\n      metadata: controllerMetadata,\n      messenger,\n      state: getDefaultDefiPositionsControllerState(),\n    });\n\n    this.setIntervalLength(TEN_MINUTES_IN_MS);\n\n    this.#fetchPositions = buildPositionFetcher();\n    this.#isEnabled = isEnabled;\n\n    this.messagingSystem.subscribe('KeyringController:unlock', () => {\n      this.startPolling(null);\n    });\n\n    this.messagingSystem.subscribe('KeyringController:lock', () => {\n      this.stopAllPolling();\n    });\n\n    this.messagingSystem.subscribe(\n      'TransactionController:transactionConfirmed',\n      async (transactionMeta) => {\n        if (!this.#isEnabled()) {\n          return;\n        }\n\n        await this.#updateAccountPositions(transactionMeta.txParams.from);\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'AccountsController:accountAdded',\n      async (account) => {\n        if (!this.#isEnabled() || !account.type.startsWith('eip155:')) {\n          return;\n        }\n\n        await this.#updateAccountPositions(account.address);\n      },\n    );\n  }\n\n  async _executePoll(): Promise<void> {\n    if (!this.#isEnabled()) {\n      return;\n    }\n\n    const accounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const initialResult: {\n      accountAddress: string;\n      positions: GroupedDeFiPositionsPerChain | null;\n    }[] = [];\n\n    const results = await reduceInBatchesSerially({\n      initialResult,\n      values: accounts,\n      batchSize: FETCH_POSITIONS_BATCH_SIZE,\n      eachBatch: async (workingResult, batch) => {\n        const batchResults = (\n          await Promise.all(\n            batch.map(async ({ address: accountAddress, type }) => {\n              if (type.startsWith('eip155:')) {\n                const positions =\n                  await this.#fetchAccountPositions(accountAddress);\n\n                return {\n                  accountAddress,\n                  positions,\n                };\n              }\n\n              return undefined;\n            }),\n          )\n        ).filter(Boolean) as {\n          accountAddress: string;\n          positions: GroupedDeFiPositionsPerChain | null;\n        }[];\n\n        return [...workingResult, ...batchResults];\n      },\n    });\n\n    const allDefiPositions = results.reduce(\n      (acc, { accountAddress, positions }) => {\n        acc[accountAddress] = positions;\n        return acc;\n      },\n      {} as DeFiPositionsControllerState['allDeFiPositions'],\n    );\n\n    this.update((state) => {\n      state.allDeFiPositions = allDefiPositions;\n    });\n  }\n\n  async #updateAccountPositions(accountAddress: string): Promise<void> {\n    const accountPositionsPerChain =\n      await this.#fetchAccountPositions(accountAddress);\n\n    this.update((state) => {\n      state.allDeFiPositions[accountAddress] = accountPositionsPerChain;\n    });\n  }\n\n  async #fetchAccountPositions(\n    accountAddress: string,\n  ): Promise<GroupedDeFiPositionsPerChain | null> {\n    try {\n      const defiPositionsResponse = await this.#fetchPositions(accountAddress);\n\n      return groupDeFiPositions(defiPositionsResponse);\n    } catch {\n      return null;\n    }\n  }\n}\n"]}
+{"version":3,"file":"DeFiPositionsController.cjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAYA,qEAA+E;AAI/E,yEAAwE;AAExE,2DAAyD;AACzD,qEAGgC;AAChC,kDAAwD;AAExD,MAAM,iBAAiB,GAAG,KAAM,CAAC;AAEjC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAEtC,MAAM,cAAc,GAAG,yBAAyB,CAAC;AAuCjD,MAAM,kBAAkB,GAAgD;IACtE,gBAAgB,EAAE;QAChB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,KAAK;KACjB;IACD,qBAAqB,EAAE;QACrB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAEK,MAAM,sCAAsC,GACjD,GAAiC,EAAE;IACjC,OAAO;QACL,gBAAgB,EAAE,EAAE;QACpB,qBAAqB,EAAE,EAAE;KAC1B,CAAC;AACJ,CAAC,CAAC;AANS,QAAA,sCAAsC,0CAM/C;AA4CJ;;GAEG;AACH,MAAa,uBAAwB,SAAQ,IAAA,oDAA+B,GAI3E;IASC;;;;;;;OAOG;IACH,YAAY,EACV,SAAS,EACT,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,EACtB,UAAU,GAKX;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,kBAAkB;YAC5B,SAAS;YACT,KAAK,EAAE,IAAA,8CAAsC,GAAE;SAChD,CAAC,CAAC;;QA9BI,0DAE4B;QAE5B,qDAA0B;QAE1B,sDAA6B;QA0BpC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1C,uBAAA,IAAI,2CAAmB,IAAA,sCAAoB,GAAE,MAAA,CAAC;QAC9C,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAE5B,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAC9D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,4CAA4C,EAC5C,KAAK,EAAE,eAAe,EAAE,EAAE;YACxB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;gBACtB,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,KAAK,EAAE,OAAO,EAAE,EAAE;YAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;gBAC7D,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,OAAO,CAAC,CAAC;QACtD,CAAC,CACF,CAAC;QAEF,uBAAA,IAAI,uCAAe,UAAU,MAAA,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;YACtB,OAAO;SACR;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACxC,iCAAiC,CAClC,CAAC;QAEF,MAAM,aAAa,GAGb,EAAE,CAAC;QAET,MAAM,OAAO,GAAG,MAAM,IAAA,oCAAuB,EAAC;YAC5C,aAAa;YACb,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,0BAA0B;YACrC,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,YAAY,GAAG,CACnB,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,EAAE;oBACpD,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;wBAC9B,MAAM,SAAS,GACb,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;wBAEpD,OAAO;4BACL,cAAc;4BACd,SAAS;yBACV,CAAC;qBACH;oBAED,OAAO,SAAS,CAAC;gBACnB,CAAC,CAAC,CACH,CACF,CAAC,MAAM,CAAC,OAAO,CAGb,CAAC;gBAEJ,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,OAAO,CAAC,MAAM,CACrC,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,EAAE,EAAE;YACrC,GAAG,CAAC,cAAc,CAAC,GAAG,SAAS,CAAC;YAChC,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAsD,CACvD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CAsDF;AAxLD,0DAwLC;wQApDC,KAAK,0DAAyB,cAAsB;IAClD,MAAM,wBAAwB,GAC5B,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;IAEpD,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,KAAK,CAAC,gBAAgB,CAAC,cAAc,CAAC,GAAG,wBAAwB,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,mDAED,KAAK,yDACH,cAAsB;IAEtB,IAAI;QACF,MAAM,qBAAqB,GAAG,MAAM,uBAAA,IAAI,+CAAgB,MAApB,IAAI,EAAiB,cAAc,CAAC,CAAC;QAEzE,MAAM,oBAAoB,GAAG,IAAA,yCAAkB,EAAC,qBAAqB,CAAC,CAAC;QAEvE,IAAI;YACF,uBAAA,IAAI,gGAA6B,MAAjC,IAAI,EAA8B,oBAAoB,EAAE,cAAc,CAAC,CAAC;SACzE;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,KAAK,CACX,gDAAgD,cAAc,GAAG,EACjE,KAAK,CACN,CAAC;SACH;QAED,OAAO,oBAAoB,CAAC;KAC7B;IAAC,MAAM;QACN,OAAO,IAAI,CAAC;KACb;AACH,CAAC,uHAGC,oBAAkD,EAClD,cAAsB;IAEtB,wDAAwD;IACxD,IAAI,CAAC,uBAAA,IAAI,2CAAY,EAAE;QACrB,OAAO;KACR;IAED,MAAM,WAAW,GAAG,IAAA,qDAA4B,EAAC,oBAAoB,CAAC,CAAC;IACvE,MAAM,EAAE,cAAc,EAAE,GAAG,WAAW,CAAC,UAAU,CAAC;IAElD,IAAI,cAAc,KAAK,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,cAAc,CAAC,EAAE;QACvE,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,qBAAqB,CAAC,cAAc,CAAC,GAAG,cAAc,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,uBAAA,IAAI,2CAAY,EAAE,KAAlB,IAAI,EAAe,WAAW,CAAC,CAAC;KACjC;AACH,CAAC","sourcesContent":["import type {\n  AccountsControllerAccountAddedEvent,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport type { KeyringControllerUnlockEvent } from '@metamask/keyring-controller';\nimport type { KeyringControllerLockEvent } from '@metamask/keyring-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { TransactionControllerTransactionConfirmedEvent } from '@metamask/transaction-controller';\nimport type { Hex } from '@metamask/utils';\n\nimport { calculateDeFiPositionMetrics } from './calculate-defi-metrics';\nimport type { DefiPositionResponse } from './fetch-positions';\nimport { buildPositionFetcher } from './fetch-positions';\nimport {\n  groupDeFiPositions,\n  type GroupedDeFiPositions,\n} from './group-defi-positions';\nimport { reduceInBatchesSerially } from '../assetsUtil';\n\nconst TEN_MINUTES_IN_MS = 60_000;\n\nconst FETCH_POSITIONS_BATCH_SIZE = 10;\n\nconst controllerName = 'DeFiPositionsController';\n\nexport type GroupedDeFiPositionsPerChain = {\n  [chain: Hex]: GroupedDeFiPositions;\n};\n\nexport type TrackingEventPayload = {\n  event: string;\n  category: string;\n  properties: {\n    totalPositions: number;\n    totalMarketValueUSD: number;\n    breakdown?: {\n      protocolId: string;\n      marketValueUSD: number;\n      chainId: Hex;\n      count: number;\n    }[];\n  };\n};\n\ntype TrackEventHook = (event: TrackingEventPayload) => void;\n\nexport type DeFiPositionsControllerState = {\n  /**\n   * Object containing DeFi positions per account and network\n   */\n  allDeFiPositions: {\n    [accountAddress: string]: GroupedDeFiPositionsPerChain | null;\n  };\n\n  /**\n   * Object containing DeFi positions count per account\n   */\n  allDeFiPositionsCount: {\n    [accountAddress: string]: number;\n  };\n};\n\nconst controllerMetadata: StateMetadata<DeFiPositionsControllerState> = {\n  allDeFiPositions: {\n    persist: false,\n    anonymous: false,\n  },\n  allDeFiPositionsCount: {\n    persist: false,\n    anonymous: false,\n  },\n};\n\nexport const getDefaultDefiPositionsControllerState =\n  (): DeFiPositionsControllerState => {\n    return {\n      allDeFiPositions: {},\n      allDeFiPositionsCount: {},\n    };\n  };\n\nexport type DeFiPositionsControllerActions =\n  DeFiPositionsControllerGetStateAction;\n\nexport type DeFiPositionsControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  DeFiPositionsControllerState\n>;\n\nexport type DeFiPositionsControllerEvents =\n  DeFiPositionsControllerStateChangeEvent;\n\nexport type DeFiPositionsControllerStateChangeEvent =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    DeFiPositionsControllerState\n  >;\n\n/**\n * The external actions available to the {@link DeFiPositionsController}.\n */\nexport type AllowedActions = AccountsControllerListAccountsAction;\n\n/**\n * The external events available to the {@link DeFiPositionsController}.\n */\nexport type AllowedEvents =\n  | KeyringControllerUnlockEvent\n  | KeyringControllerLockEvent\n  | TransactionControllerTransactionConfirmedEvent\n  | AccountsControllerAccountAddedEvent;\n\n/**\n * The messenger of the {@link DeFiPositionsController}.\n */\nexport type DeFiPositionsControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  DeFiPositionsControllerActions | AllowedActions,\n  DeFiPositionsControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/**\n * Controller that stores assets and exposes convenience methods\n */\nexport class DeFiPositionsController extends StaticIntervalPollingController()<\n  typeof controllerName,\n  DeFiPositionsControllerState,\n  DeFiPositionsControllerMessenger\n> {\n  readonly #fetchPositions: (\n    accountAddress: string,\n  ) => Promise<DefiPositionResponse[]>;\n\n  readonly #isEnabled: () => boolean;\n\n  readonly #trackEvent?: TrackEventHook;\n\n  /**\n   * DeFiPositionsController constuctor\n   *\n   * @param options - Constructor options.\n   * @param options.messenger - The controller messenger.\n   * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)\n   * @param options.trackEvent - Function to track events. (default: undefined)\n   */\n  constructor({\n    messenger,\n    isEnabled = () => true,\n    trackEvent,\n  }: {\n    messenger: DeFiPositionsControllerMessenger;\n    isEnabled?: () => boolean;\n    trackEvent?: TrackEventHook;\n  }) {\n    super({\n      name: controllerName,\n      metadata: controllerMetadata,\n      messenger,\n      state: getDefaultDefiPositionsControllerState(),\n    });\n\n    this.setIntervalLength(TEN_MINUTES_IN_MS);\n\n    this.#fetchPositions = buildPositionFetcher();\n    this.#isEnabled = isEnabled;\n\n    this.messagingSystem.subscribe('KeyringController:unlock', () => {\n      this.startPolling(null);\n    });\n\n    this.messagingSystem.subscribe('KeyringController:lock', () => {\n      this.stopAllPolling();\n    });\n\n    this.messagingSystem.subscribe(\n      'TransactionController:transactionConfirmed',\n      async (transactionMeta) => {\n        if (!this.#isEnabled()) {\n          return;\n        }\n\n        await this.#updateAccountPositions(transactionMeta.txParams.from);\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'AccountsController:accountAdded',\n      async (account) => {\n        if (!this.#isEnabled() || !account.type.startsWith('eip155:')) {\n          return;\n        }\n\n        await this.#updateAccountPositions(account.address);\n      },\n    );\n\n    this.#trackEvent = trackEvent;\n  }\n\n  async _executePoll(): Promise<void> {\n    if (!this.#isEnabled()) {\n      return;\n    }\n\n    const accounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const initialResult: {\n      accountAddress: string;\n      positions: GroupedDeFiPositionsPerChain | null;\n    }[] = [];\n\n    const results = await reduceInBatchesSerially({\n      initialResult,\n      values: accounts,\n      batchSize: FETCH_POSITIONS_BATCH_SIZE,\n      eachBatch: async (workingResult, batch) => {\n        const batchResults = (\n          await Promise.all(\n            batch.map(async ({ address: accountAddress, type }) => {\n              if (type.startsWith('eip155:')) {\n                const positions =\n                  await this.#fetchAccountPositions(accountAddress);\n\n                return {\n                  accountAddress,\n                  positions,\n                };\n              }\n\n              return undefined;\n            }),\n          )\n        ).filter(Boolean) as {\n          accountAddress: string;\n          positions: GroupedDeFiPositionsPerChain | null;\n        }[];\n\n        return [...workingResult, ...batchResults];\n      },\n    });\n\n    const allDefiPositions = results.reduce(\n      (acc, { accountAddress, positions }) => {\n        acc[accountAddress] = positions;\n        return acc;\n      },\n      {} as DeFiPositionsControllerState['allDeFiPositions'],\n    );\n\n    this.update((state) => {\n      state.allDeFiPositions = allDefiPositions;\n    });\n  }\n\n  async #updateAccountPositions(accountAddress: string): Promise<void> {\n    const accountPositionsPerChain =\n      await this.#fetchAccountPositions(accountAddress);\n\n    this.update((state) => {\n      state.allDeFiPositions[accountAddress] = accountPositionsPerChain;\n    });\n  }\n\n  async #fetchAccountPositions(\n    accountAddress: string,\n  ): Promise<GroupedDeFiPositionsPerChain | null> {\n    try {\n      const defiPositionsResponse = await this.#fetchPositions(accountAddress);\n\n      const groupedDeFiPositions = groupDeFiPositions(defiPositionsResponse);\n\n      try {\n        this.#updatePositionsCountMetrics(groupedDeFiPositions, accountAddress);\n      } catch (error) {\n        console.error(\n          `Failed to update positions count for account ${accountAddress}:`,\n          error,\n        );\n      }\n\n      return groupedDeFiPositions;\n    } catch {\n      return null;\n    }\n  }\n\n  #updatePositionsCountMetrics(\n    groupedDeFiPositions: GroupedDeFiPositionsPerChain,\n    accountAddress: string,\n  ) {\n    // If no track event passed then skip the metrics update\n    if (!this.#trackEvent) {\n      return;\n    }\n\n    const defiMetrics = calculateDeFiPositionMetrics(groupedDeFiPositions);\n    const { totalPositions } = defiMetrics.properties;\n\n    if (totalPositions !== this.state.allDeFiPositionsCount[accountAddress]) {\n      this.update((state) => {\n        state.allDeFiPositionsCount[accountAddress] = totalPositions;\n      });\n\n      this.#trackEvent?.(defiMetrics);\n    }\n  }\n}\n"]}

package/dist/DeFiPositionsController/DeFiPositionsController.d.cts

@@ -7,9 +7,24 @@ import type { TransactionControllerTransactionConfirmedEvent } from "@metamask/t
 import type { Hex } from "@metamask/utils";
 import { type GroupedDeFiPositions } from "./group-defi-positions.cjs";
 declare const controllerName = "DeFiPositionsController";
-type GroupedDeFiPositionsPerChain = {
+export type GroupedDeFiPositionsPerChain = {
     [chain: Hex]: GroupedDeFiPositions;
 };
+export type TrackingEventPayload = {
+    event: string;
+    category: string;
+    properties: {
+        totalPositions: number;
+        totalMarketValueUSD: number;
+        breakdown?: {
+            protocolId: string;
+            marketValueUSD: number;
+            chainId: Hex;
+            count: number;
+        }[];
+    };
+};
+type TrackEventHook = (event: TrackingEventPayload) => void;
 export type DeFiPositionsControllerState = {
     /**
      * Object containing DeFi positions per account and network
@@ -17,6 +32,12 @@ export type DeFiPositionsControllerState = {
     allDeFiPositions: {
         [accountAddress: string]: GroupedDeFiPositionsPerChain | null;
     };
+    /**
+     * Object containing DeFi positions count per account
+     */
+    allDeFiPositionsCount: {
+        [accountAddress: string]: number;
+    };
 };
 export declare const getDefaultDefiPositionsControllerState: () => DeFiPositionsControllerState;
 export type DeFiPositionsControllerActions = DeFiPositionsControllerGetStateAction;
@@ -61,10 +82,12 @@ export declare class DeFiPositionsController extends DeFiPositionsController_bas
      * @param options - Constructor options.
      * @param options.messenger - The controller messenger.
      * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)
+     * @param options.trackEvent - Function to track events. (default: undefined)
      */
-    constructor({ messenger, isEnabled, }: {
+    constructor({ messenger, isEnabled, trackEvent, }: {
         messenger: DeFiPositionsControllerMessenger;
         isEnabled?: () => boolean;
+        trackEvent?: TrackEventHook;
     });
     _executePoll(): Promise<void>;
 }

package/dist/DeFiPositionsController/DeFiPositionsController.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"DeFiPositionsController.d.cts","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EACV,mCAAmC,EACnC,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,KAAK,EAAE,4BAA4B,EAAE,qCAAqC;AACjF,OAAO,KAAK,EAAE,0BAA0B,EAAE,qCAAqC;AAE/E,OAAO,KAAK,EAAE,8CAA8C,EAAE,yCAAyC;AACvG,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAI3C,OAAO,EAEL,KAAK,oBAAoB,EAC1B,mCAA+B;AAOhC,QAAA,MAAM,cAAc,4BAA4B,CAAC;AAEjD,KAAK,4BAA4B,GAAG;IAClC,CAAC,KAAK,EAAE,GAAG,GAAG,oBAAoB,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC;;OAEG;IACH,gBAAgB,EAAE;QAChB,CAAC,cAAc,EAAE,MAAM,GAAG,4BAA4B,GAAG,IAAI,CAAC;KAC/D,CAAC;CACH,CAAC;AASF,eAAO,MAAM,sCAAsC,QAC7C,4BAIH,CAAC;AAEJ,MAAM,MAAM,8BAA8B,GACxC,qCAAqC,CAAC;AAExC,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACvC,uCAAuC,CAAC;AAE1C,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CACxB,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,oCAAoC,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4BAA4B,GAC5B,0BAA0B,GAC1B,8CAA8C,GAC9C,mCAAmC,CAAC;AAExC;;GAEG;AACH,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,cAAc,EACrB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,6BAC3C,OAAO,cAAc,EACrB,4BAA4B,EAC5B,gCAAgC,CACjC;;IAOC;;;;;;OAMG;gBACS,EACV,SAAS,EACT,SAAsB,GACvB,EAAE;QACD,SAAS,EAAE,gCAAgC,CAAC;QAC5C,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC;KAC3B;IA4CK,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;CA6EpC"}
+{"version":3,"file":"DeFiPositionsController.d.cts","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EACV,mCAAmC,EACnC,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,KAAK,EAAE,4BAA4B,EAAE,qCAAqC;AACjF,OAAO,KAAK,EAAE,0BAA0B,EAAE,qCAAqC;AAE/E,OAAO,KAAK,EAAE,8CAA8C,EAAE,yCAAyC;AACvG,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,EAEL,KAAK,oBAAoB,EAC1B,mCAA+B;AAOhC,QAAA,MAAM,cAAc,4BAA4B,CAAC;AAEjD,MAAM,MAAM,4BAA4B,GAAG;IACzC,CAAC,KAAK,EAAE,GAAG,GAAG,oBAAoB,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE;QACV,cAAc,EAAE,MAAM,CAAC;QACvB,mBAAmB,EAAE,MAAM,CAAC;QAC5B,SAAS,CAAC,EAAE;YACV,UAAU,EAAE,MAAM,CAAC;YACnB,cAAc,EAAE,MAAM,CAAC;YACvB,OAAO,EAAE,GAAG,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;SACf,EAAE,CAAC;KACL,CAAC;CACH,CAAC;AAEF,KAAK,cAAc,GAAG,CAAC,KAAK,EAAE,oBAAoB,KAAK,IAAI,CAAC;AAE5D,MAAM,MAAM,4BAA4B,GAAG;IACzC;;OAEG;IACH,gBAAgB,EAAE;QAChB,CAAC,cAAc,EAAE,MAAM,GAAG,4BAA4B,GAAG,IAAI,CAAC;KAC/D,CAAC;IAEF;;OAEG;IACH,qBAAqB,EAAE;QACrB,CAAC,cAAc,EAAE,MAAM,GAAG,MAAM,CAAC;KAClC,CAAC;CACH,CAAC;AAaF,eAAO,MAAM,sCAAsC,QAC7C,4BAKH,CAAC;AAEJ,MAAM,MAAM,8BAA8B,GACxC,qCAAqC,CAAC;AAExC,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACvC,uCAAuC,CAAC;AAE1C,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CACxB,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,oCAAoC,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4BAA4B,GAC5B,0BAA0B,GAC1B,8CAA8C,GAC9C,mCAAmC,CAAC;AAExC;;GAEG;AACH,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,cAAc,EACrB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,6BAC3C,OAAO,cAAc,EACrB,4BAA4B,EAC5B,gCAAgC,CACjC;;IASC;;;;;;;OAOG;gBACS,EACV,SAAS,EACT,SAAsB,EACtB,UAAU,GACX,EAAE;QACD,SAAS,EAAE,gCAAgC,CAAC;QAC5C,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC;QAC1B,UAAU,CAAC,EAAE,cAAc,CAAC;KAC7B;IA8CK,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;CA6GpC"}

package/dist/DeFiPositionsController/DeFiPositionsController.d.mts

@@ -7,9 +7,24 @@ import type { TransactionControllerTransactionConfirmedEvent } from "@metamask/t
 import type { Hex } from "@metamask/utils";
 import { type GroupedDeFiPositions } from "./group-defi-positions.mjs";
 declare const controllerName = "DeFiPositionsController";
-type GroupedDeFiPositionsPerChain = {
+export type GroupedDeFiPositionsPerChain = {
     [chain: Hex]: GroupedDeFiPositions;
 };
+export type TrackingEventPayload = {
+    event: string;
+    category: string;
+    properties: {
+        totalPositions: number;
+        totalMarketValueUSD: number;
+        breakdown?: {
+            protocolId: string;
+            marketValueUSD: number;
+            chainId: Hex;
+            count: number;
+        }[];
+    };
+};
+type TrackEventHook = (event: TrackingEventPayload) => void;
 export type DeFiPositionsControllerState = {
     /**
      * Object containing DeFi positions per account and network
@@ -17,6 +32,12 @@ export type DeFiPositionsControllerState = {
     allDeFiPositions: {
         [accountAddress: string]: GroupedDeFiPositionsPerChain | null;
     };
+    /**
+     * Object containing DeFi positions count per account
+     */
+    allDeFiPositionsCount: {
+        [accountAddress: string]: number;
+    };
 };
 export declare const getDefaultDefiPositionsControllerState: () => DeFiPositionsControllerState;
 export type DeFiPositionsControllerActions = DeFiPositionsControllerGetStateAction;
@@ -61,10 +82,12 @@ export declare class DeFiPositionsController extends DeFiPositionsController_bas
      * @param options - Constructor options.
      * @param options.messenger - The controller messenger.
      * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)
+     * @param options.trackEvent - Function to track events. (default: undefined)
      */
-    constructor({ messenger, isEnabled, }: {
+    constructor({ messenger, isEnabled, trackEvent, }: {
         messenger: DeFiPositionsControllerMessenger;
         isEnabled?: () => boolean;
+        trackEvent?: TrackEventHook;
     });
     _executePoll(): Promise<void>;
 }

package/dist/DeFiPositionsController/DeFiPositionsController.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"DeFiPositionsController.d.mts","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EACV,mCAAmC,EACnC,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,KAAK,EAAE,4BAA4B,EAAE,qCAAqC;AACjF,OAAO,KAAK,EAAE,0BAA0B,EAAE,qCAAqC;AAE/E,OAAO,KAAK,EAAE,8CAA8C,EAAE,yCAAyC;AACvG,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAI3C,OAAO,EAEL,KAAK,oBAAoB,EAC1B,mCAA+B;AAOhC,QAAA,MAAM,cAAc,4BAA4B,CAAC;AAEjD,KAAK,4BAA4B,GAAG;IAClC,CAAC,KAAK,EAAE,GAAG,GAAG,oBAAoB,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC;;OAEG;IACH,gBAAgB,EAAE;QAChB,CAAC,cAAc,EAAE,MAAM,GAAG,4BAA4B,GAAG,IAAI,CAAC;KAC/D,CAAC;CACH,CAAC;AASF,eAAO,MAAM,sCAAsC,QAC7C,4BAIH,CAAC;AAEJ,MAAM,MAAM,8BAA8B,GACxC,qCAAqC,CAAC;AAExC,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACvC,uCAAuC,CAAC;AAE1C,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CACxB,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,oCAAoC,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4BAA4B,GAC5B,0BAA0B,GAC1B,8CAA8C,GAC9C,mCAAmC,CAAC;AAExC;;GAEG;AACH,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,cAAc,EACrB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,6BAC3C,OAAO,cAAc,EACrB,4BAA4B,EAC5B,gCAAgC,CACjC;;IAOC;;;;;;OAMG;gBACS,EACV,SAAS,EACT,SAAsB,GACvB,EAAE;QACD,SAAS,EAAE,gCAAgC,CAAC;QAC5C,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC;KAC3B;IA4CK,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;CA6EpC"}
+{"version":3,"file":"DeFiPositionsController.d.mts","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EACV,mCAAmC,EACnC,oCAAoC,EACrC,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,KAAK,EAAE,4BAA4B,EAAE,qCAAqC;AACjF,OAAO,KAAK,EAAE,0BAA0B,EAAE,qCAAqC;AAE/E,OAAO,KAAK,EAAE,8CAA8C,EAAE,yCAAyC;AACvG,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAK3C,OAAO,EAEL,KAAK,oBAAoB,EAC1B,mCAA+B;AAOhC,QAAA,MAAM,cAAc,4BAA4B,CAAC;AAEjD,MAAM,MAAM,4BAA4B,GAAG;IACzC,CAAC,KAAK,EAAE,GAAG,GAAG,oBAAoB,CAAC;CACpC,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE;QACV,cAAc,EAAE,MAAM,CAAC;QACvB,mBAAmB,EAAE,MAAM,CAAC;QAC5B,SAAS,CAAC,EAAE;YACV,UAAU,EAAE,MAAM,CAAC;YACnB,cAAc,EAAE,MAAM,CAAC;YACvB,OAAO,EAAE,GAAG,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;SACf,EAAE,CAAC;KACL,CAAC;CACH,CAAC;AAEF,KAAK,cAAc,GAAG,CAAC,KAAK,EAAE,oBAAoB,KAAK,IAAI,CAAC;AAE5D,MAAM,MAAM,4BAA4B,GAAG;IACzC;;OAEG;IACH,gBAAgB,EAAE;QAChB,CAAC,cAAc,EAAE,MAAM,GAAG,4BAA4B,GAAG,IAAI,CAAC;KAC/D,CAAC;IAEF;;OAEG;IACH,qBAAqB,EAAE;QACrB,CAAC,cAAc,EAAE,MAAM,GAAG,MAAM,CAAC;KAClC,CAAC;CACH,CAAC;AAaF,eAAO,MAAM,sCAAsC,QAC7C,4BAKH,CAAC;AAEJ,MAAM,MAAM,8BAA8B,GACxC,qCAAqC,CAAC;AAExC,MAAM,MAAM,qCAAqC,GAAG,wBAAwB,CAC1E,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEF,MAAM,MAAM,6BAA6B,GACvC,uCAAuC,CAAC;AAE1C,MAAM,MAAM,uCAAuC,GACjD,0BAA0B,CACxB,OAAO,cAAc,EACrB,4BAA4B,CAC7B,CAAC;AAEJ;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,oCAAoC,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4BAA4B,GAC5B,0BAA0B,GAC1B,8CAA8C,GAC9C,mCAAmC,CAAC;AAExC;;GAEG;AACH,MAAM,MAAM,gCAAgC,GAAG,mBAAmB,CAChE,OAAO,cAAc,EACrB,8BAA8B,GAAG,cAAc,EAC/C,6BAA6B,GAAG,aAAa,EAC7C,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;;;;;;;;;;;;;;;;AAEF;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,6BAC3C,OAAO,cAAc,EACrB,4BAA4B,EAC5B,gCAAgC,CACjC;;IASC;;;;;;;OAOG;gBACS,EACV,SAAS,EACT,SAAsB,EACtB,UAAU,GACX,EAAE;QACD,SAAS,EAAE,gCAAgC,CAAC;QAC5C,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC;QAC1B,UAAU,CAAC,EAAE,cAAc,CAAC;KAC7B;IA8CK,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;CA6GpC"}

package/dist/DeFiPositionsController/DeFiPositionsController.mjs

@@ -9,8 +9,9 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     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 _DeFiPositionsController_instances, _DeFiPositionsController_fetchPositions, _DeFiPositionsController_isEnabled, _DeFiPositionsController_updateAccountPositions, _DeFiPositionsController_fetchAccountPositions;
+var _DeFiPositionsController_instances, _DeFiPositionsController_fetchPositions, _DeFiPositionsController_isEnabled, _DeFiPositionsController_trackEvent, _DeFiPositionsController_updateAccountPositions, _DeFiPositionsController_fetchAccountPositions, _DeFiPositionsController_updatePositionsCountMetrics;
 import { StaticIntervalPollingController } from "@metamask/polling-controller";
+import { calculateDeFiPositionMetrics } from "./calculate-defi-metrics.mjs";
 import { buildPositionFetcher } from "./fetch-positions.mjs";
 import { groupDeFiPositions } from "./group-defi-positions.mjs";
 import { reduceInBatchesSerially } from "../assetsUtil.mjs";
@@ -22,10 +23,15 @@ const controllerMetadata = {
         persist: false,
         anonymous: false,
     },
+    allDeFiPositionsCount: {
+        persist: false,
+        anonymous: false,
+    },
 };
 export const getDefaultDefiPositionsControllerState = () => {
     return {
         allDeFiPositions: {},
+        allDeFiPositionsCount: {},
     };
 };
 /**
@@ -38,8 +44,9 @@ export class DeFiPositionsController extends StaticIntervalPollingController() {
      * @param options - Constructor options.
      * @param options.messenger - The controller messenger.
      * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)
+     * @param options.trackEvent - Function to track events. (default: undefined)
      */
-    constructor({ messenger, isEnabled = () => true, }) {
+    constructor({ messenger, isEnabled = () => true, trackEvent, }) {
         super({
             name: controllerName,
             metadata: controllerMetadata,
@@ -49,6 +56,7 @@ export class DeFiPositionsController extends StaticIntervalPollingController() {
         _DeFiPositionsController_instances.add(this);
         _DeFiPositionsController_fetchPositions.set(this, void 0);
         _DeFiPositionsController_isEnabled.set(this, void 0);
+        _DeFiPositionsController_trackEvent.set(this, void 0);
         this.setIntervalLength(TEN_MINUTES_IN_MS);
         __classPrivateFieldSet(this, _DeFiPositionsController_fetchPositions, buildPositionFetcher(), "f");
         __classPrivateFieldSet(this, _DeFiPositionsController_isEnabled, isEnabled, "f");
@@ -70,6 +78,7 @@ export class DeFiPositionsController extends StaticIntervalPollingController() {
             }
             await __classPrivateFieldGet(this, _DeFiPositionsController_instances, "m", _DeFiPositionsController_updateAccountPositions).call(this, account.address);
         });
+        __classPrivateFieldSet(this, _DeFiPositionsController_trackEvent, trackEvent, "f");
     }
     async _executePoll() {
         if (!__classPrivateFieldGet(this, _DeFiPositionsController_isEnabled, "f").call(this)) {
@@ -104,7 +113,7 @@ export class DeFiPositionsController extends StaticIntervalPollingController() {
         });
     }
 }
-_DeFiPositionsController_fetchPositions = new WeakMap(), _DeFiPositionsController_isEnabled = new WeakMap(), _DeFiPositionsController_instances = new WeakSet(), _DeFiPositionsController_updateAccountPositions = async function _DeFiPositionsController_updateAccountPositions(accountAddress) {
+_DeFiPositionsController_fetchPositions = new WeakMap(), _DeFiPositionsController_isEnabled = new WeakMap(), _DeFiPositionsController_trackEvent = new WeakMap(), _DeFiPositionsController_instances = new WeakSet(), _DeFiPositionsController_updateAccountPositions = async function _DeFiPositionsController_updateAccountPositions(accountAddress) {
     const accountPositionsPerChain = await __classPrivateFieldGet(this, _DeFiPositionsController_instances, "m", _DeFiPositionsController_fetchAccountPositions).call(this, accountAddress);
     this.update((state) => {
         state.allDeFiPositions[accountAddress] = accountPositionsPerChain;
@@ -112,10 +121,30 @@ _DeFiPositionsController_fetchPositions = new WeakMap(), _DeFiPositionsControlle
 }, _DeFiPositionsController_fetchAccountPositions = async function _DeFiPositionsController_fetchAccountPositions(accountAddress) {
     try {
         const defiPositionsResponse = await __classPrivateFieldGet(this, _DeFiPositionsController_fetchPositions, "f").call(this, accountAddress);
-        return groupDeFiPositions(defiPositionsResponse);
+        const groupedDeFiPositions = groupDeFiPositions(defiPositionsResponse);
+        try {
+            __classPrivateFieldGet(this, _DeFiPositionsController_instances, "m", _DeFiPositionsController_updatePositionsCountMetrics).call(this, groupedDeFiPositions, accountAddress);
+        }
+        catch (error) {
+            console.error(`Failed to update positions count for account ${accountAddress}:`, error);
+        }
+        return groupedDeFiPositions;
     }
     catch {
         return null;
     }
+}, _DeFiPositionsController_updatePositionsCountMetrics = function _DeFiPositionsController_updatePositionsCountMetrics(groupedDeFiPositions, accountAddress) {
+    // If no track event passed then skip the metrics update
+    if (!__classPrivateFieldGet(this, _DeFiPositionsController_trackEvent, "f")) {
+        return;
+    }
+    const defiMetrics = calculateDeFiPositionMetrics(groupedDeFiPositions);
+    const { totalPositions } = defiMetrics.properties;
+    if (totalPositions !== this.state.allDeFiPositionsCount[accountAddress]) {
+        this.update((state) => {
+            state.allDeFiPositionsCount[accountAddress] = totalPositions;
+        });
+        __classPrivateFieldGet(this, _DeFiPositionsController_trackEvent, "f")?.call(this, defiMetrics);
+    }
 };
 //# sourceMappingURL=DeFiPositionsController.mjs.map

package/dist/DeFiPositionsController/DeFiPositionsController.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"DeFiPositionsController.mjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";;;;;;;;;;;;AAYA,OAAO,EAAE,+BAA+B,EAAE,qCAAqC;AAK/E,OAAO,EAAE,oBAAoB,EAAE,8BAA0B;AACzD,OAAO,EACL,kBAAkB,EAEnB,mCAA+B;AAChC,OAAO,EAAE,uBAAuB,EAAE,0BAAsB;AAExD,MAAM,iBAAiB,GAAG,KAAM,CAAC;AAEjC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAEtC,MAAM,cAAc,GAAG,yBAAyB,CAAC;AAejD,MAAM,kBAAkB,GAAgD;IACtE,gBAAgB,EAAE;QAChB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,sCAAsC,GACjD,GAAiC,EAAE;IACjC,OAAO;QACL,gBAAgB,EAAE,EAAE;KACrB,CAAC;AACJ,CAAC,CAAC;AA4CJ;;GAEG;AACH,MAAM,OAAO,uBAAwB,SAAQ,+BAA+B,EAI3E;IAOC;;;;;;OAMG;IACH,YAAY,EACV,SAAS,EACT,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,GAIvB;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,kBAAkB;YAC5B,SAAS;YACT,KAAK,EAAE,sCAAsC,EAAE;SAChD,CAAC,CAAC;;QAzBI,0DAE4B;QAE5B,qDAA0B;QAuBjC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1C,uBAAA,IAAI,2CAAmB,oBAAoB,EAAE,MAAA,CAAC;QAC9C,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAE5B,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAC9D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,4CAA4C,EAC5C,KAAK,EAAE,eAAe,EAAE,EAAE;YACxB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;gBACtB,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,KAAK,EAAE,OAAO,EAAE,EAAE;YAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;gBAC7D,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,OAAO,CAAC,CAAC;QACtD,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;YACtB,OAAO;SACR;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACxC,iCAAiC,CAClC,CAAC;QAEF,MAAM,aAAa,GAGb,EAAE,CAAC;QAET,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC;YAC5C,aAAa;YACb,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,0BAA0B;YACrC,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,YAAY,GAAG,CACnB,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,EAAE;oBACpD,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;wBAC9B,MAAM,SAAS,GACb,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;wBAEpD,OAAO;4BACL,cAAc;4BACd,SAAS;yBACV,CAAC;qBACH;oBAED,OAAO,SAAS,CAAC;gBACnB,CAAC,CAAC,CACH,CACF,CAAC,MAAM,CAAC,OAAO,CAGb,CAAC;gBAEJ,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,OAAO,CAAC,MAAM,CACrC,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,EAAE,EAAE;YACrC,GAAG,CAAC,cAAc,CAAC,GAAG,SAAS,CAAC;YAChC,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAsD,CACvD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CAsBF;mNApBC,KAAK,0DAAyB,cAAsB;IAClD,MAAM,wBAAwB,GAC5B,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;IAEpD,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,KAAK,CAAC,gBAAgB,CAAC,cAAc,CAAC,GAAG,wBAAwB,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,mDAED,KAAK,yDACH,cAAsB;IAEtB,IAAI;QACF,MAAM,qBAAqB,GAAG,MAAM,uBAAA,IAAI,+CAAgB,MAApB,IAAI,EAAiB,cAAc,CAAC,CAAC;QAEzE,OAAO,kBAAkB,CAAC,qBAAqB,CAAC,CAAC;KAClD;IAAC,MAAM;QACN,OAAO,IAAI,CAAC;KACb;AACH,CAAC","sourcesContent":["import type {\n  AccountsControllerAccountAddedEvent,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport type { KeyringControllerUnlockEvent } from '@metamask/keyring-controller';\nimport type { KeyringControllerLockEvent } from '@metamask/keyring-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { TransactionControllerTransactionConfirmedEvent } from '@metamask/transaction-controller';\nimport type { Hex } from '@metamask/utils';\n\nimport type { DefiPositionResponse } from './fetch-positions';\nimport { buildPositionFetcher } from './fetch-positions';\nimport {\n  groupDeFiPositions,\n  type GroupedDeFiPositions,\n} from './group-defi-positions';\nimport { reduceInBatchesSerially } from '../assetsUtil';\n\nconst TEN_MINUTES_IN_MS = 60_000;\n\nconst FETCH_POSITIONS_BATCH_SIZE = 10;\n\nconst controllerName = 'DeFiPositionsController';\n\ntype GroupedDeFiPositionsPerChain = {\n  [chain: Hex]: GroupedDeFiPositions;\n};\n\nexport type DeFiPositionsControllerState = {\n  /**\n   * Object containing DeFi positions per account and network\n   */\n  allDeFiPositions: {\n    [accountAddress: string]: GroupedDeFiPositionsPerChain | null;\n  };\n};\n\nconst controllerMetadata: StateMetadata<DeFiPositionsControllerState> = {\n  allDeFiPositions: {\n    persist: false,\n    anonymous: false,\n  },\n};\n\nexport const getDefaultDefiPositionsControllerState =\n  (): DeFiPositionsControllerState => {\n    return {\n      allDeFiPositions: {},\n    };\n  };\n\nexport type DeFiPositionsControllerActions =\n  DeFiPositionsControllerGetStateAction;\n\nexport type DeFiPositionsControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  DeFiPositionsControllerState\n>;\n\nexport type DeFiPositionsControllerEvents =\n  DeFiPositionsControllerStateChangeEvent;\n\nexport type DeFiPositionsControllerStateChangeEvent =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    DeFiPositionsControllerState\n  >;\n\n/**\n * The external actions available to the {@link DeFiPositionsController}.\n */\nexport type AllowedActions = AccountsControllerListAccountsAction;\n\n/**\n * The external events available to the {@link DeFiPositionsController}.\n */\nexport type AllowedEvents =\n  | KeyringControllerUnlockEvent\n  | KeyringControllerLockEvent\n  | TransactionControllerTransactionConfirmedEvent\n  | AccountsControllerAccountAddedEvent;\n\n/**\n * The messenger of the {@link DeFiPositionsController}.\n */\nexport type DeFiPositionsControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  DeFiPositionsControllerActions | AllowedActions,\n  DeFiPositionsControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/**\n * Controller that stores assets and exposes convenience methods\n */\nexport class DeFiPositionsController extends StaticIntervalPollingController()<\n  typeof controllerName,\n  DeFiPositionsControllerState,\n  DeFiPositionsControllerMessenger\n> {\n  readonly #fetchPositions: (\n    accountAddress: string,\n  ) => Promise<DefiPositionResponse[]>;\n\n  readonly #isEnabled: () => boolean;\n\n  /**\n   * DeFiPositionsController constuctor\n   *\n   * @param options - Constructor options.\n   * @param options.messenger - The controller messenger.\n   * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)\n   */\n  constructor({\n    messenger,\n    isEnabled = () => true,\n  }: {\n    messenger: DeFiPositionsControllerMessenger;\n    isEnabled?: () => boolean;\n  }) {\n    super({\n      name: controllerName,\n      metadata: controllerMetadata,\n      messenger,\n      state: getDefaultDefiPositionsControllerState(),\n    });\n\n    this.setIntervalLength(TEN_MINUTES_IN_MS);\n\n    this.#fetchPositions = buildPositionFetcher();\n    this.#isEnabled = isEnabled;\n\n    this.messagingSystem.subscribe('KeyringController:unlock', () => {\n      this.startPolling(null);\n    });\n\n    this.messagingSystem.subscribe('KeyringController:lock', () => {\n      this.stopAllPolling();\n    });\n\n    this.messagingSystem.subscribe(\n      'TransactionController:transactionConfirmed',\n      async (transactionMeta) => {\n        if (!this.#isEnabled()) {\n          return;\n        }\n\n        await this.#updateAccountPositions(transactionMeta.txParams.from);\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'AccountsController:accountAdded',\n      async (account) => {\n        if (!this.#isEnabled() || !account.type.startsWith('eip155:')) {\n          return;\n        }\n\n        await this.#updateAccountPositions(account.address);\n      },\n    );\n  }\n\n  async _executePoll(): Promise<void> {\n    if (!this.#isEnabled()) {\n      return;\n    }\n\n    const accounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const initialResult: {\n      accountAddress: string;\n      positions: GroupedDeFiPositionsPerChain | null;\n    }[] = [];\n\n    const results = await reduceInBatchesSerially({\n      initialResult,\n      values: accounts,\n      batchSize: FETCH_POSITIONS_BATCH_SIZE,\n      eachBatch: async (workingResult, batch) => {\n        const batchResults = (\n          await Promise.all(\n            batch.map(async ({ address: accountAddress, type }) => {\n              if (type.startsWith('eip155:')) {\n                const positions =\n                  await this.#fetchAccountPositions(accountAddress);\n\n                return {\n                  accountAddress,\n                  positions,\n                };\n              }\n\n              return undefined;\n            }),\n          )\n        ).filter(Boolean) as {\n          accountAddress: string;\n          positions: GroupedDeFiPositionsPerChain | null;\n        }[];\n\n        return [...workingResult, ...batchResults];\n      },\n    });\n\n    const allDefiPositions = results.reduce(\n      (acc, { accountAddress, positions }) => {\n        acc[accountAddress] = positions;\n        return acc;\n      },\n      {} as DeFiPositionsControllerState['allDeFiPositions'],\n    );\n\n    this.update((state) => {\n      state.allDeFiPositions = allDefiPositions;\n    });\n  }\n\n  async #updateAccountPositions(accountAddress: string): Promise<void> {\n    const accountPositionsPerChain =\n      await this.#fetchAccountPositions(accountAddress);\n\n    this.update((state) => {\n      state.allDeFiPositions[accountAddress] = accountPositionsPerChain;\n    });\n  }\n\n  async #fetchAccountPositions(\n    accountAddress: string,\n  ): Promise<GroupedDeFiPositionsPerChain | null> {\n    try {\n      const defiPositionsResponse = await this.#fetchPositions(accountAddress);\n\n      return groupDeFiPositions(defiPositionsResponse);\n    } catch {\n      return null;\n    }\n  }\n}\n"]}
+{"version":3,"file":"DeFiPositionsController.mjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/DeFiPositionsController.ts"],"names":[],"mappings":";;;;;;;;;;;;AAYA,OAAO,EAAE,+BAA+B,EAAE,qCAAqC;AAI/E,OAAO,EAAE,4BAA4B,EAAE,qCAAiC;AAExE,OAAO,EAAE,oBAAoB,EAAE,8BAA0B;AACzD,OAAO,EACL,kBAAkB,EAEnB,mCAA+B;AAChC,OAAO,EAAE,uBAAuB,EAAE,0BAAsB;AAExD,MAAM,iBAAiB,GAAG,KAAM,CAAC;AAEjC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAEtC,MAAM,cAAc,GAAG,yBAAyB,CAAC;AAuCjD,MAAM,kBAAkB,GAAgD;IACtE,gBAAgB,EAAE;QAChB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,KAAK;KACjB;IACD,qBAAqB,EAAE;QACrB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,KAAK;KACjB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,sCAAsC,GACjD,GAAiC,EAAE;IACjC,OAAO;QACL,gBAAgB,EAAE,EAAE;QACpB,qBAAqB,EAAE,EAAE;KAC1B,CAAC;AACJ,CAAC,CAAC;AA4CJ;;GAEG;AACH,MAAM,OAAO,uBAAwB,SAAQ,+BAA+B,EAI3E;IASC;;;;;;;OAOG;IACH,YAAY,EACV,SAAS,EACT,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,EACtB,UAAU,GAKX;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,kBAAkB;YAC5B,SAAS;YACT,KAAK,EAAE,sCAAsC,EAAE;SAChD,CAAC,CAAC;;QA9BI,0DAE4B;QAE5B,qDAA0B;QAE1B,sDAA6B;QA0BpC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAE1C,uBAAA,IAAI,2CAAmB,oBAAoB,EAAE,MAAA,CAAC;QAC9C,uBAAA,IAAI,sCAAc,SAAS,MAAA,CAAC;QAE5B,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAC9D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,4CAA4C,EAC5C,KAAK,EAAE,eAAe,EAAE,EAAE;YACxB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;gBACtB,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,iCAAiC,EACjC,KAAK,EAAE,OAAO,EAAE,EAAE;YAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;gBAC7D,OAAO;aACR;YAED,MAAM,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,OAAO,CAAC,CAAC;QACtD,CAAC,CACF,CAAC;QAEF,uBAAA,IAAI,uCAAe,UAAU,MAAA,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,uBAAA,IAAI,0CAAW,MAAf,IAAI,CAAa,EAAE;YACtB,OAAO;SACR;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CACxC,iCAAiC,CAClC,CAAC;QAEF,MAAM,aAAa,GAGb,EAAE,CAAC;QAET,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC;YAC5C,aAAa;YACb,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,0BAA0B;YACrC,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE;gBACxC,MAAM,YAAY,GAAG,CACnB,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,EAAE;oBACpD,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;wBAC9B,MAAM,SAAS,GACb,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;wBAEpD,OAAO;4BACL,cAAc;4BACd,SAAS;yBACV,CAAC;qBACH;oBAED,OAAO,SAAS,CAAC;gBACnB,CAAC,CAAC,CACH,CACF,CAAC,MAAM,CAAC,OAAO,CAGb,CAAC;gBAEJ,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,YAAY,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAAG,OAAO,CAAC,MAAM,CACrC,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,EAAE,EAAE;YACrC,GAAG,CAAC,cAAc,CAAC,GAAG,SAAS,CAAC;YAChC,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAsD,CACvD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CAsDF;wQApDC,KAAK,0DAAyB,cAAsB;IAClD,MAAM,wBAAwB,GAC5B,MAAM,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,cAAc,CAAC,CAAC;IAEpD,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QACpB,KAAK,CAAC,gBAAgB,CAAC,cAAc,CAAC,GAAG,wBAAwB,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,mDAED,KAAK,yDACH,cAAsB;IAEtB,IAAI;QACF,MAAM,qBAAqB,GAAG,MAAM,uBAAA,IAAI,+CAAgB,MAApB,IAAI,EAAiB,cAAc,CAAC,CAAC;QAEzE,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,qBAAqB,CAAC,CAAC;QAEvE,IAAI;YACF,uBAAA,IAAI,gGAA6B,MAAjC,IAAI,EAA8B,oBAAoB,EAAE,cAAc,CAAC,CAAC;SACzE;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,KAAK,CACX,gDAAgD,cAAc,GAAG,EACjE,KAAK,CACN,CAAC;SACH;QAED,OAAO,oBAAoB,CAAC;KAC7B;IAAC,MAAM;QACN,OAAO,IAAI,CAAC;KACb;AACH,CAAC,uHAGC,oBAAkD,EAClD,cAAsB;IAEtB,wDAAwD;IACxD,IAAI,CAAC,uBAAA,IAAI,2CAAY,EAAE;QACrB,OAAO;KACR;IAED,MAAM,WAAW,GAAG,4BAA4B,CAAC,oBAAoB,CAAC,CAAC;IACvE,MAAM,EAAE,cAAc,EAAE,GAAG,WAAW,CAAC,UAAU,CAAC;IAElD,IAAI,cAAc,KAAK,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,cAAc,CAAC,EAAE;QACvE,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,qBAAqB,CAAC,cAAc,CAAC,GAAG,cAAc,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,uBAAA,IAAI,2CAAY,EAAE,KAAlB,IAAI,EAAe,WAAW,CAAC,CAAC;KACjC;AACH,CAAC","sourcesContent":["import type {\n  AccountsControllerAccountAddedEvent,\n  AccountsControllerListAccountsAction,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport type { KeyringControllerUnlockEvent } from '@metamask/keyring-controller';\nimport type { KeyringControllerLockEvent } from '@metamask/keyring-controller';\nimport { StaticIntervalPollingController } from '@metamask/polling-controller';\nimport type { TransactionControllerTransactionConfirmedEvent } from '@metamask/transaction-controller';\nimport type { Hex } from '@metamask/utils';\n\nimport { calculateDeFiPositionMetrics } from './calculate-defi-metrics';\nimport type { DefiPositionResponse } from './fetch-positions';\nimport { buildPositionFetcher } from './fetch-positions';\nimport {\n  groupDeFiPositions,\n  type GroupedDeFiPositions,\n} from './group-defi-positions';\nimport { reduceInBatchesSerially } from '../assetsUtil';\n\nconst TEN_MINUTES_IN_MS = 60_000;\n\nconst FETCH_POSITIONS_BATCH_SIZE = 10;\n\nconst controllerName = 'DeFiPositionsController';\n\nexport type GroupedDeFiPositionsPerChain = {\n  [chain: Hex]: GroupedDeFiPositions;\n};\n\nexport type TrackingEventPayload = {\n  event: string;\n  category: string;\n  properties: {\n    totalPositions: number;\n    totalMarketValueUSD: number;\n    breakdown?: {\n      protocolId: string;\n      marketValueUSD: number;\n      chainId: Hex;\n      count: number;\n    }[];\n  };\n};\n\ntype TrackEventHook = (event: TrackingEventPayload) => void;\n\nexport type DeFiPositionsControllerState = {\n  /**\n   * Object containing DeFi positions per account and network\n   */\n  allDeFiPositions: {\n    [accountAddress: string]: GroupedDeFiPositionsPerChain | null;\n  };\n\n  /**\n   * Object containing DeFi positions count per account\n   */\n  allDeFiPositionsCount: {\n    [accountAddress: string]: number;\n  };\n};\n\nconst controllerMetadata: StateMetadata<DeFiPositionsControllerState> = {\n  allDeFiPositions: {\n    persist: false,\n    anonymous: false,\n  },\n  allDeFiPositionsCount: {\n    persist: false,\n    anonymous: false,\n  },\n};\n\nexport const getDefaultDefiPositionsControllerState =\n  (): DeFiPositionsControllerState => {\n    return {\n      allDeFiPositions: {},\n      allDeFiPositionsCount: {},\n    };\n  };\n\nexport type DeFiPositionsControllerActions =\n  DeFiPositionsControllerGetStateAction;\n\nexport type DeFiPositionsControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  DeFiPositionsControllerState\n>;\n\nexport type DeFiPositionsControllerEvents =\n  DeFiPositionsControllerStateChangeEvent;\n\nexport type DeFiPositionsControllerStateChangeEvent =\n  ControllerStateChangeEvent<\n    typeof controllerName,\n    DeFiPositionsControllerState\n  >;\n\n/**\n * The external actions available to the {@link DeFiPositionsController}.\n */\nexport type AllowedActions = AccountsControllerListAccountsAction;\n\n/**\n * The external events available to the {@link DeFiPositionsController}.\n */\nexport type AllowedEvents =\n  | KeyringControllerUnlockEvent\n  | KeyringControllerLockEvent\n  | TransactionControllerTransactionConfirmedEvent\n  | AccountsControllerAccountAddedEvent;\n\n/**\n * The messenger of the {@link DeFiPositionsController}.\n */\nexport type DeFiPositionsControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  DeFiPositionsControllerActions | AllowedActions,\n  DeFiPositionsControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n/**\n * Controller that stores assets and exposes convenience methods\n */\nexport class DeFiPositionsController extends StaticIntervalPollingController()<\n  typeof controllerName,\n  DeFiPositionsControllerState,\n  DeFiPositionsControllerMessenger\n> {\n  readonly #fetchPositions: (\n    accountAddress: string,\n  ) => Promise<DefiPositionResponse[]>;\n\n  readonly #isEnabled: () => boolean;\n\n  readonly #trackEvent?: TrackEventHook;\n\n  /**\n   * DeFiPositionsController constuctor\n   *\n   * @param options - Constructor options.\n   * @param options.messenger - The controller messenger.\n   * @param options.isEnabled - Function that returns whether the controller is enabled. (default: () => true)\n   * @param options.trackEvent - Function to track events. (default: undefined)\n   */\n  constructor({\n    messenger,\n    isEnabled = () => true,\n    trackEvent,\n  }: {\n    messenger: DeFiPositionsControllerMessenger;\n    isEnabled?: () => boolean;\n    trackEvent?: TrackEventHook;\n  }) {\n    super({\n      name: controllerName,\n      metadata: controllerMetadata,\n      messenger,\n      state: getDefaultDefiPositionsControllerState(),\n    });\n\n    this.setIntervalLength(TEN_MINUTES_IN_MS);\n\n    this.#fetchPositions = buildPositionFetcher();\n    this.#isEnabled = isEnabled;\n\n    this.messagingSystem.subscribe('KeyringController:unlock', () => {\n      this.startPolling(null);\n    });\n\n    this.messagingSystem.subscribe('KeyringController:lock', () => {\n      this.stopAllPolling();\n    });\n\n    this.messagingSystem.subscribe(\n      'TransactionController:transactionConfirmed',\n      async (transactionMeta) => {\n        if (!this.#isEnabled()) {\n          return;\n        }\n\n        await this.#updateAccountPositions(transactionMeta.txParams.from);\n      },\n    );\n\n    this.messagingSystem.subscribe(\n      'AccountsController:accountAdded',\n      async (account) => {\n        if (!this.#isEnabled() || !account.type.startsWith('eip155:')) {\n          return;\n        }\n\n        await this.#updateAccountPositions(account.address);\n      },\n    );\n\n    this.#trackEvent = trackEvent;\n  }\n\n  async _executePoll(): Promise<void> {\n    if (!this.#isEnabled()) {\n      return;\n    }\n\n    const accounts = this.messagingSystem.call(\n      'AccountsController:listAccounts',\n    );\n\n    const initialResult: {\n      accountAddress: string;\n      positions: GroupedDeFiPositionsPerChain | null;\n    }[] = [];\n\n    const results = await reduceInBatchesSerially({\n      initialResult,\n      values: accounts,\n      batchSize: FETCH_POSITIONS_BATCH_SIZE,\n      eachBatch: async (workingResult, batch) => {\n        const batchResults = (\n          await Promise.all(\n            batch.map(async ({ address: accountAddress, type }) => {\n              if (type.startsWith('eip155:')) {\n                const positions =\n                  await this.#fetchAccountPositions(accountAddress);\n\n                return {\n                  accountAddress,\n                  positions,\n                };\n              }\n\n              return undefined;\n            }),\n          )\n        ).filter(Boolean) as {\n          accountAddress: string;\n          positions: GroupedDeFiPositionsPerChain | null;\n        }[];\n\n        return [...workingResult, ...batchResults];\n      },\n    });\n\n    const allDefiPositions = results.reduce(\n      (acc, { accountAddress, positions }) => {\n        acc[accountAddress] = positions;\n        return acc;\n      },\n      {} as DeFiPositionsControllerState['allDeFiPositions'],\n    );\n\n    this.update((state) => {\n      state.allDeFiPositions = allDefiPositions;\n    });\n  }\n\n  async #updateAccountPositions(accountAddress: string): Promise<void> {\n    const accountPositionsPerChain =\n      await this.#fetchAccountPositions(accountAddress);\n\n    this.update((state) => {\n      state.allDeFiPositions[accountAddress] = accountPositionsPerChain;\n    });\n  }\n\n  async #fetchAccountPositions(\n    accountAddress: string,\n  ): Promise<GroupedDeFiPositionsPerChain | null> {\n    try {\n      const defiPositionsResponse = await this.#fetchPositions(accountAddress);\n\n      const groupedDeFiPositions = groupDeFiPositions(defiPositionsResponse);\n\n      try {\n        this.#updatePositionsCountMetrics(groupedDeFiPositions, accountAddress);\n      } catch (error) {\n        console.error(\n          `Failed to update positions count for account ${accountAddress}:`,\n          error,\n        );\n      }\n\n      return groupedDeFiPositions;\n    } catch {\n      return null;\n    }\n  }\n\n  #updatePositionsCountMetrics(\n    groupedDeFiPositions: GroupedDeFiPositionsPerChain,\n    accountAddress: string,\n  ) {\n    // If no track event passed then skip the metrics update\n    if (!this.#trackEvent) {\n      return;\n    }\n\n    const defiMetrics = calculateDeFiPositionMetrics(groupedDeFiPositions);\n    const { totalPositions } = defiMetrics.properties;\n\n    if (totalPositions !== this.state.allDeFiPositionsCount[accountAddress]) {\n      this.update((state) => {\n        state.allDeFiPositionsCount[accountAddress] = totalPositions;\n      });\n\n      this.#trackEvent?.(defiMetrics);\n    }\n  }\n}\n"]}

package/dist/DeFiPositionsController/calculate-defi-metrics.cjs

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.calculateDeFiPositionMetrics = void 0;
+/**
+ * Calculates the total market value and total positions for a given account
+ * and returns a breakdown of the market value per protocol.
+ *
+ * @param accountPositionsPerChain - The account positions per chain.
+ * @returns An object containing the total market value, total positions, and a breakdown of the market value per protocol.
+ */
+function calculateDeFiPositionMetrics(accountPositionsPerChain) {
+    let totalMarketValueUSD = 0;
+    let totalPositions = 0;
+    const breakdown = [];
+    Object.entries(accountPositionsPerChain).forEach(([chainId, chainPositions]) => {
+        const chainTotalMarketValueUSD = chainPositions.aggregatedMarketValue;
+        totalMarketValueUSD += chainTotalMarketValueUSD;
+        Object.entries(chainPositions.protocols).forEach(([protocolId, protocol]) => {
+            const protocolTotalMarketValueUSD = protocol.aggregatedMarketValue;
+            const protocolCount = Object.values(protocol.positionTypes).reduce((acc, positionType) => acc + (positionType?.positions?.flat().length || 0), 0);
+            totalPositions += protocolCount;
+            breakdown.push({
+                protocolId,
+                marketValueUSD: protocolTotalMarketValueUSD,
+                chainId: chainId,
+                count: protocolCount,
+            });
+        });
+    });
+    return {
+        category: 'DeFi',
+        event: 'DeFi Stats',
+        properties: {
+            totalMarketValueUSD,
+            totalPositions,
+            breakdown,
+        },
+    };
+}
+exports.calculateDeFiPositionMetrics = calculateDeFiPositionMetrics;
+//# sourceMappingURL=calculate-defi-metrics.cjs.map

package/dist/DeFiPositionsController/calculate-defi-metrics.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"calculate-defi-metrics.cjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/calculate-defi-metrics.ts"],"names":[],"mappings":";;;AAOA;;;;;;GAMG;AACH,SAAgB,4BAA4B,CAC1C,wBAAsD;IAEtD,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,MAAM,SAAS,GAKT,EAAE,CAAC;IAET,MAAM,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,OAAO,CAC9C,CAAC,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,EAAE;QAC5B,MAAM,wBAAwB,GAAG,cAAc,CAAC,qBAAqB,CAAC;QACtE,mBAAmB,IAAI,wBAAwB,CAAC;QAEhD,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,OAAO,CAC9C,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,EAAE;YACzB,MAAM,2BAA2B,GAAG,QAAQ,CAAC,qBAAqB,CAAC;YAEnE,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,MAAM,CAChE,CAAC,GAAG,EAAE,YAAY,EAAE,EAAE,CACpB,GAAG,GAAG,CAAC,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,EAErD,CAAC,CACF,CAAC;YAEF,cAAc,IAAI,aAAa,CAAC;YAEhC,SAAS,CAAC,IAAI,CAAC;gBACb,UAAU;gBACV,cAAc,EAAE,2BAA2B;gBAC3C,OAAO,EAAE,OAAc;gBACvB,KAAK,EAAE,aAAa;aACrB,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC,CACF,CAAC;IACF,OAAO;QACL,QAAQ,EAAE,MAAM;QAChB,KAAK,EAAE,YAAY;QACnB,UAAU,EAAE;YACV,mBAAmB;YACnB,cAAc;YACd,SAAS;SACV;KACF,CAAC;AACJ,CAAC;AAjDD,oEAiDC","sourcesContent":["import type { Hex } from '@metamask/utils';\n\nimport type {\n  GroupedDeFiPositionsPerChain,\n  TrackingEventPayload,\n} from './DeFiPositionsController';\n\n/**\n * Calculates the total market value and total positions for a given account\n * and returns a breakdown of the market value per protocol.\n *\n * @param accountPositionsPerChain - The account positions per chain.\n * @returns An object containing the total market value, total positions, and a breakdown of the market value per protocol.\n */\nexport function calculateDeFiPositionMetrics(\n  accountPositionsPerChain: GroupedDeFiPositionsPerChain,\n): TrackingEventPayload {\n  let totalMarketValueUSD = 0;\n  let totalPositions = 0;\n  const breakdown: {\n    protocolId: string;\n    marketValueUSD: number;\n    chainId: Hex;\n    count: number;\n  }[] = [];\n\n  Object.entries(accountPositionsPerChain).forEach(\n    ([chainId, chainPositions]) => {\n      const chainTotalMarketValueUSD = chainPositions.aggregatedMarketValue;\n      totalMarketValueUSD += chainTotalMarketValueUSD;\n\n      Object.entries(chainPositions.protocols).forEach(\n        ([protocolId, protocol]) => {\n          const protocolTotalMarketValueUSD = protocol.aggregatedMarketValue;\n\n          const protocolCount = Object.values(protocol.positionTypes).reduce(\n            (acc, positionType) =>\n              acc + (positionType?.positions?.flat().length || 0),\n\n            0,\n          );\n\n          totalPositions += protocolCount;\n\n          breakdown.push({\n            protocolId,\n            marketValueUSD: protocolTotalMarketValueUSD,\n            chainId: chainId as Hex,\n            count: protocolCount,\n          });\n        },\n      );\n    },\n  );\n  return {\n    category: 'DeFi',\n    event: 'DeFi Stats',\n    properties: {\n      totalMarketValueUSD,\n      totalPositions,\n      breakdown,\n    },\n  };\n}\n"]}

package/dist/DeFiPositionsController/calculate-defi-metrics.d.cts

@@ -0,0 +1,10 @@
+import type { GroupedDeFiPositionsPerChain, TrackingEventPayload } from "./DeFiPositionsController.cjs";
+/**
+ * Calculates the total market value and total positions for a given account
+ * and returns a breakdown of the market value per protocol.
+ *
+ * @param accountPositionsPerChain - The account positions per chain.
+ * @returns An object containing the total market value, total positions, and a breakdown of the market value per protocol.
+ */
+export declare function calculateDeFiPositionMetrics(accountPositionsPerChain: GroupedDeFiPositionsPerChain): TrackingEventPayload;
+//# sourceMappingURL=calculate-defi-metrics.d.cts.map

package/dist/DeFiPositionsController/calculate-defi-metrics.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"calculate-defi-metrics.d.cts","sourceRoot":"","sources":["../../src/DeFiPositionsController/calculate-defi-metrics.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,4BAA4B,EAC5B,oBAAoB,EACrB,sCAAkC;AAEnC;;;;;;GAMG;AACH,wBAAgB,4BAA4B,CAC1C,wBAAwB,EAAE,4BAA4B,GACrD,oBAAoB,CA+CtB"}

package/dist/DeFiPositionsController/calculate-defi-metrics.d.mts

@@ -0,0 +1,10 @@
+import type { GroupedDeFiPositionsPerChain, TrackingEventPayload } from "./DeFiPositionsController.mjs";
+/**
+ * Calculates the total market value and total positions for a given account
+ * and returns a breakdown of the market value per protocol.
+ *
+ * @param accountPositionsPerChain - The account positions per chain.
+ * @returns An object containing the total market value, total positions, and a breakdown of the market value per protocol.
+ */
+export declare function calculateDeFiPositionMetrics(accountPositionsPerChain: GroupedDeFiPositionsPerChain): TrackingEventPayload;
+//# sourceMappingURL=calculate-defi-metrics.d.mts.map

package/dist/DeFiPositionsController/calculate-defi-metrics.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"calculate-defi-metrics.d.mts","sourceRoot":"","sources":["../../src/DeFiPositionsController/calculate-defi-metrics.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,4BAA4B,EAC5B,oBAAoB,EACrB,sCAAkC;AAEnC;;;;;;GAMG;AACH,wBAAgB,4BAA4B,CAC1C,wBAAwB,EAAE,4BAA4B,GACrD,oBAAoB,CA+CtB"}

package/dist/DeFiPositionsController/calculate-defi-metrics.mjs

@@ -0,0 +1,37 @@
+/**
+ * Calculates the total market value and total positions for a given account
+ * and returns a breakdown of the market value per protocol.
+ *
+ * @param accountPositionsPerChain - The account positions per chain.
+ * @returns An object containing the total market value, total positions, and a breakdown of the market value per protocol.
+ */
+export function calculateDeFiPositionMetrics(accountPositionsPerChain) {
+    let totalMarketValueUSD = 0;
+    let totalPositions = 0;
+    const breakdown = [];
+    Object.entries(accountPositionsPerChain).forEach(([chainId, chainPositions]) => {
+        const chainTotalMarketValueUSD = chainPositions.aggregatedMarketValue;
+        totalMarketValueUSD += chainTotalMarketValueUSD;
+        Object.entries(chainPositions.protocols).forEach(([protocolId, protocol]) => {
+            const protocolTotalMarketValueUSD = protocol.aggregatedMarketValue;
+            const protocolCount = Object.values(protocol.positionTypes).reduce((acc, positionType) => acc + (positionType?.positions?.flat().length || 0), 0);
+            totalPositions += protocolCount;
+            breakdown.push({
+                protocolId,
+                marketValueUSD: protocolTotalMarketValueUSD,
+                chainId: chainId,
+                count: protocolCount,
+            });
+        });
+    });
+    return {
+        category: 'DeFi',
+        event: 'DeFi Stats',
+        properties: {
+            totalMarketValueUSD,
+            totalPositions,
+            breakdown,
+        },
+    };
+}
+//# sourceMappingURL=calculate-defi-metrics.mjs.map

package/dist/DeFiPositionsController/calculate-defi-metrics.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"calculate-defi-metrics.mjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/calculate-defi-metrics.ts"],"names":[],"mappings":"AAOA;;;;;;GAMG;AACH,MAAM,UAAU,4BAA4B,CAC1C,wBAAsD;IAEtD,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,MAAM,SAAS,GAKT,EAAE,CAAC;IAET,MAAM,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,OAAO,CAC9C,CAAC,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,EAAE;QAC5B,MAAM,wBAAwB,GAAG,cAAc,CAAC,qBAAqB,CAAC;QACtE,mBAAmB,IAAI,wBAAwB,CAAC;QAEhD,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,OAAO,CAC9C,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,EAAE;YACzB,MAAM,2BAA2B,GAAG,QAAQ,CAAC,qBAAqB,CAAC;YAEnE,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,MAAM,CAChE,CAAC,GAAG,EAAE,YAAY,EAAE,EAAE,CACpB,GAAG,GAAG,CAAC,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,CAAC,EAErD,CAAC,CACF,CAAC;YAEF,cAAc,IAAI,aAAa,CAAC;YAEhC,SAAS,CAAC,IAAI,CAAC;gBACb,UAAU;gBACV,cAAc,EAAE,2BAA2B;gBAC3C,OAAO,EAAE,OAAc;gBACvB,KAAK,EAAE,aAAa;aACrB,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC,CACF,CAAC;IACF,OAAO;QACL,QAAQ,EAAE,MAAM;QAChB,KAAK,EAAE,YAAY;QACnB,UAAU,EAAE;YACV,mBAAmB;YACnB,cAAc;YACd,SAAS;SACV;KACF,CAAC;AACJ,CAAC","sourcesContent":["import type { Hex } from '@metamask/utils';\n\nimport type {\n  GroupedDeFiPositionsPerChain,\n  TrackingEventPayload,\n} from './DeFiPositionsController';\n\n/**\n * Calculates the total market value and total positions for a given account\n * and returns a breakdown of the market value per protocol.\n *\n * @param accountPositionsPerChain - The account positions per chain.\n * @returns An object containing the total market value, total positions, and a breakdown of the market value per protocol.\n */\nexport function calculateDeFiPositionMetrics(\n  accountPositionsPerChain: GroupedDeFiPositionsPerChain,\n): TrackingEventPayload {\n  let totalMarketValueUSD = 0;\n  let totalPositions = 0;\n  const breakdown: {\n    protocolId: string;\n    marketValueUSD: number;\n    chainId: Hex;\n    count: number;\n  }[] = [];\n\n  Object.entries(accountPositionsPerChain).forEach(\n    ([chainId, chainPositions]) => {\n      const chainTotalMarketValueUSD = chainPositions.aggregatedMarketValue;\n      totalMarketValueUSD += chainTotalMarketValueUSD;\n\n      Object.entries(chainPositions.protocols).forEach(\n        ([protocolId, protocol]) => {\n          const protocolTotalMarketValueUSD = protocol.aggregatedMarketValue;\n\n          const protocolCount = Object.values(protocol.positionTypes).reduce(\n            (acc, positionType) =>\n              acc + (positionType?.positions?.flat().length || 0),\n\n            0,\n          );\n\n          totalPositions += protocolCount;\n\n          breakdown.push({\n            protocolId,\n            marketValueUSD: protocolTotalMarketValueUSD,\n            chainId: chainId as Hex,\n            count: protocolCount,\n          });\n        },\n      );\n    },\n  );\n  return {\n    category: 'DeFi',\n    event: 'DeFi Stats',\n    properties: {\n      totalMarketValueUSD,\n      totalPositions,\n      breakdown,\n    },\n  };\n}\n"]}

package/dist/DeFiPositionsController/group-defi-positions.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"group-defi-positions.cjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":";;;AAAA,iEAAmD;AAoCnD;;;;GAIG;AACH,SAAgB,kBAAkB,CAChC,qBAA6C;IAI7C,MAAM,oBAAoB,GAAyC,EAAE,CAAC;IAEtE,KAAK,MAAM,QAAQ,IAAI,qBAAqB,EAAE;QAC5C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE;YACrB,SAAS;SACV;QAED,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,GACvE,QAAQ,CAAC;QAEX,MAAM,KAAK,GAAG,IAAA,wBAAK,EAAC,OAAO,CAAC,CAAC;QAE7B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE;YAChC,oBAAoB,CAAC,KAAK,CAAC,GAAG;gBAC5B,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;SACH;QAED,MAAM,SAAS,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE;YACpC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG;gBAChC,eAAe,EAAE;oBACf,IAAI,EAAE,mBAAmB;oBACzB,OAAO;iBACR;gBACD,qBAAqB,EAAE,CAAC;gBACxB,aAAa,EAAE,EAAE;aAClB,CAAC;SACH;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAErD,IAAI,gBAAgB,GAAG,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,EAAE;YACrB,gBAAgB,GAAG;gBACjB,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;YACF,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,gBAAgB,CAAC;SAC7D;QAED,KAAK,MAAM,aAAa,IAAI,QAAQ,CAAC,MAAM,EAAE;YAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,aAAa,CAAiC,CAAC;YAE1E,qEAAqE;YACrE,IAAI,QAAQ,CAAC,QAAQ,EAAE,cAAc,EAAE;gBACrC,IAAI,gBAAgB,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC3C,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;iBAC1C;qBAAM;oBACL,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBAC3C;aACF;iBAAM;gBACL,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;aAC1C;YAED,IAAI,KAAK,CAAC,WAAW,EAAE;gBACrB,MAAM,UAAU,GAAG,QAAQ,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE/D,gBAAgB,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,CAAC;gBAC5D,YAAY,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;gBACrE,SAAS,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;aACnE;SACF;KACF;IAED,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAzED,gDAyEC;AAED;;;;GAIG;AACH,SAAS,YAAY,CACnB,YAAe;IAKf,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QACxB,OAAO;YACL,GAAG,YAAY;YACf,WAAW,EAAE,YAAY,CAAC,KAAK;gBAC7B,CAAC,CAAC,YAAY,CAAC,OAAO,GAAG,YAAY,CAAC,KAAK;gBAC3C,CAAC,CAAC,SAAS;SACd,CAAC;KACH;IAED,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;QAE/D,OAAO,uBAAuB,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CACxC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CACT,GAAG,KAAK,SAAS,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS;QAC9C,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,EACzB,CAAuB,CACxB,CAAC;IAEF,OAAO;QACL,GAAG,YAAY;QACf,WAAW;QACX,MAAM,EAAE,eAAe;KACxB,CAAC;AACJ,CAAC","sourcesContent":["import { toHex } from '@metamask/controller-utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n  DefiPositionResponse,\n  PositionType,\n  ProtocolToken,\n  Underlying,\n  Balance,\n} from './fetch-positions';\n\nexport type GroupedDeFiPositions = {\n  aggregatedMarketValue: number;\n  protocols: {\n    [protocolId: string]: {\n      protocolDetails: { name: string; iconUrl: string };\n      aggregatedMarketValue: number;\n      positionTypes: {\n        [key in PositionType]?: {\n          aggregatedMarketValue: number;\n          positions: ProtocolTokenWithMarketValue[][];\n        };\n      };\n    };\n  };\n};\n\nexport type ProtocolTokenWithMarketValue = Omit<ProtocolToken, 'tokens'> & {\n  marketValue?: number;\n  tokens: UnderlyingWithMarketValue[];\n};\n\nexport type UnderlyingWithMarketValue = Omit<Underlying, 'tokens'> & {\n  marketValue?: number;\n};\n\n/**\n *\n * @param defiPositionsResponse - The response from the defi positions API\n * @returns The grouped positions that get assigned to the state\n */\nexport function groupDeFiPositions(\n  defiPositionsResponse: DefiPositionResponse[],\n): {\n  [key: Hex]: GroupedDeFiPositions;\n} {\n  const groupedDeFiPositions: { [key: Hex]: GroupedDeFiPositions } = {};\n\n  for (const position of defiPositionsResponse) {\n    if (!position.success) {\n      continue;\n    }\n\n    const { chainId, protocolId, iconUrl, positionType, protocolDisplayName } =\n      position;\n\n    const chain = toHex(chainId);\n\n    if (!groupedDeFiPositions[chain]) {\n      groupedDeFiPositions[chain] = {\n        aggregatedMarketValue: 0,\n        protocols: {},\n      };\n    }\n\n    const chainData = groupedDeFiPositions[chain];\n\n    if (!chainData.protocols[protocolId]) {\n      chainData.protocols[protocolId] = {\n        protocolDetails: {\n          name: protocolDisplayName,\n          iconUrl,\n        },\n        aggregatedMarketValue: 0,\n        positionTypes: {},\n      };\n    }\n\n    const protocolData = chainData.protocols[protocolId];\n\n    let positionTypeData = protocolData.positionTypes[positionType];\n    if (!positionTypeData) {\n      positionTypeData = {\n        aggregatedMarketValue: 0,\n        positions: [],\n      };\n      protocolData.positionTypes[positionType] = positionTypeData;\n    }\n\n    for (const protocolToken of position.tokens) {\n      const token = processToken(protocolToken) as ProtocolTokenWithMarketValue;\n\n      // If groupPositions is true, we group all positions of the same type\n      if (position.metadata?.groupPositions) {\n        if (positionTypeData.positions.length === 0) {\n          positionTypeData.positions.push([token]);\n        } else {\n          positionTypeData.positions[0].push(token);\n        }\n      } else {\n        positionTypeData.positions.push([token]);\n      }\n\n      if (token.marketValue) {\n        const multiplier = position.positionType === 'borrow' ? -1 : 1;\n\n        positionTypeData.aggregatedMarketValue += token.marketValue;\n        protocolData.aggregatedMarketValue += token.marketValue * multiplier;\n        chainData.aggregatedMarketValue += token.marketValue * multiplier;\n      }\n    }\n  }\n\n  return groupedDeFiPositions;\n}\n\n/**\n *\n * @param tokenBalance - The token balance that is going to be processed\n * @returns The processed token balance\n */\nfunction processToken<T extends Balance>(\n  tokenBalance: T,\n): T & {\n  marketValue?: number;\n  tokens?: UnderlyingWithMarketValue[];\n} {\n  if (!tokenBalance.tokens) {\n    return {\n      ...tokenBalance,\n      marketValue: tokenBalance.price\n        ? tokenBalance.balance * tokenBalance.price\n        : undefined,\n    };\n  }\n\n  const processedTokens = tokenBalance.tokens.map((t) => {\n    const { tokens, ...tokenWithoutUnderlyings } = processToken(t);\n\n    return tokenWithoutUnderlyings;\n  });\n\n  const marketValue = processedTokens.reduce(\n    (acc, t) =>\n      acc === undefined || t.marketValue === undefined\n        ? undefined\n        : acc + t.marketValue,\n    0 as number | undefined,\n  );\n\n  return {\n    ...tokenBalance,\n    marketValue,\n    tokens: processedTokens,\n  };\n}\n"]}
+{"version":3,"file":"group-defi-positions.cjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":";;;AAAA,iEAAmD;AAuCnD;;;;GAIG;AACH,SAAgB,kBAAkB,CAChC,qBAA6C;IAI7C,MAAM,oBAAoB,GAAyC,EAAE,CAAC;IAEtE,KAAK,MAAM,QAAQ,IAAI,qBAAqB,EAAE;QAC5C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE;YACrB,SAAS;SACV;QAED,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,GACvE,QAAQ,CAAC;QAEX,MAAM,KAAK,GAAG,IAAA,wBAAK,EAAC,OAAO,CAAC,CAAC;QAE7B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE;YAChC,oBAAoB,CAAC,KAAK,CAAC,GAAG;gBAC5B,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;SACH;QAED,MAAM,SAAS,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE;YACpC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG;gBAChC,eAAe,EAAE;oBACf,IAAI,EAAE,mBAAmB;oBACzB,OAAO;iBACR;gBACD,qBAAqB,EAAE,CAAC;gBACxB,aAAa,EAAE,EAAE;aAClB,CAAC;SACH;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAErD,IAAI,gBAAgB,GAAG,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,EAAE;YACrB,gBAAgB,GAAG;gBACjB,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;YACF,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,gBAAgB,CAAC;SAC7D;QAED,KAAK,MAAM,aAAa,IAAI,QAAQ,CAAC,MAAM,EAAE;YAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,aAAa,CAAiC,CAAC;YAE1E,qEAAqE;YACrE,IAAI,QAAQ,CAAC,QAAQ,EAAE,cAAc,EAAE;gBACrC,IAAI,gBAAgB,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC3C,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;iBAC1C;qBAAM;oBACL,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBAC3C;aACF;iBAAM;gBACL,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;aAC1C;YAED,IAAI,KAAK,CAAC,WAAW,EAAE;gBACrB,MAAM,UAAU,GAAG,QAAQ,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE/D,gBAAgB,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,CAAC;gBAC5D,YAAY,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;gBACrE,SAAS,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;aACnE;SACF;KACF;IAED,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAzED,gDAyEC;AAED;;;;GAIG;AACH,SAAS,YAAY,CACnB,YAAe;IAKf,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QACxB,OAAO;YACL,GAAG,YAAY;YACf,WAAW,EAAE,YAAY,CAAC,KAAK;gBAC7B,CAAC,CAAC,YAAY,CAAC,OAAO,GAAG,YAAY,CAAC,KAAK;gBAC3C,CAAC,CAAC,SAAS;SACd,CAAC;KACH;IAED,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;QAE/D,OAAO,uBAAuB,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CACxC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CACT,GAAG,KAAK,SAAS,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS;QAC9C,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,EACzB,CAAuB,CACxB,CAAC;IAEF,OAAO;QACL,GAAG,YAAY;QACf,WAAW;QACX,MAAM,EAAE,eAAe;KACxB,CAAC;AACJ,CAAC","sourcesContent":["import { toHex } from '@metamask/controller-utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n  DefiPositionResponse,\n  PositionType,\n  ProtocolToken,\n  Underlying,\n  Balance,\n} from './fetch-positions';\n\nexport type GroupedDeFiPositions = {\n  aggregatedMarketValue: number;\n  protocols: {\n    [protocolId: string]: {\n      protocolDetails: {\n        name: string;\n        iconUrl: string;\n      };\n      aggregatedMarketValue: number;\n      positionTypes: {\n        [key in PositionType]?: {\n          aggregatedMarketValue: number;\n          positions: ProtocolTokenWithMarketValue[][];\n        };\n      };\n    };\n  };\n};\n\nexport type ProtocolTokenWithMarketValue = Omit<ProtocolToken, 'tokens'> & {\n  marketValue?: number;\n  tokens: UnderlyingWithMarketValue[];\n};\n\nexport type UnderlyingWithMarketValue = Omit<Underlying, 'tokens'> & {\n  marketValue?: number;\n};\n\n/**\n *\n * @param defiPositionsResponse - The response from the defi positions API\n * @returns The grouped positions that get assigned to the state\n */\nexport function groupDeFiPositions(\n  defiPositionsResponse: DefiPositionResponse[],\n): {\n  [key: Hex]: GroupedDeFiPositions;\n} {\n  const groupedDeFiPositions: { [key: Hex]: GroupedDeFiPositions } = {};\n\n  for (const position of defiPositionsResponse) {\n    if (!position.success) {\n      continue;\n    }\n\n    const { chainId, protocolId, iconUrl, positionType, protocolDisplayName } =\n      position;\n\n    const chain = toHex(chainId);\n\n    if (!groupedDeFiPositions[chain]) {\n      groupedDeFiPositions[chain] = {\n        aggregatedMarketValue: 0,\n        protocols: {},\n      };\n    }\n\n    const chainData = groupedDeFiPositions[chain];\n\n    if (!chainData.protocols[protocolId]) {\n      chainData.protocols[protocolId] = {\n        protocolDetails: {\n          name: protocolDisplayName,\n          iconUrl,\n        },\n        aggregatedMarketValue: 0,\n        positionTypes: {},\n      };\n    }\n\n    const protocolData = chainData.protocols[protocolId];\n\n    let positionTypeData = protocolData.positionTypes[positionType];\n    if (!positionTypeData) {\n      positionTypeData = {\n        aggregatedMarketValue: 0,\n        positions: [],\n      };\n      protocolData.positionTypes[positionType] = positionTypeData;\n    }\n\n    for (const protocolToken of position.tokens) {\n      const token = processToken(protocolToken) as ProtocolTokenWithMarketValue;\n\n      // If groupPositions is true, we group all positions of the same type\n      if (position.metadata?.groupPositions) {\n        if (positionTypeData.positions.length === 0) {\n          positionTypeData.positions.push([token]);\n        } else {\n          positionTypeData.positions[0].push(token);\n        }\n      } else {\n        positionTypeData.positions.push([token]);\n      }\n\n      if (token.marketValue) {\n        const multiplier = position.positionType === 'borrow' ? -1 : 1;\n\n        positionTypeData.aggregatedMarketValue += token.marketValue;\n        protocolData.aggregatedMarketValue += token.marketValue * multiplier;\n        chainData.aggregatedMarketValue += token.marketValue * multiplier;\n      }\n    }\n  }\n\n  return groupedDeFiPositions;\n}\n\n/**\n *\n * @param tokenBalance - The token balance that is going to be processed\n * @returns The processed token balance\n */\nfunction processToken<T extends Balance>(\n  tokenBalance: T,\n): T & {\n  marketValue?: number;\n  tokens?: UnderlyingWithMarketValue[];\n} {\n  if (!tokenBalance.tokens) {\n    return {\n      ...tokenBalance,\n      marketValue: tokenBalance.price\n        ? tokenBalance.balance * tokenBalance.price\n        : undefined,\n    };\n  }\n\n  const processedTokens = tokenBalance.tokens.map((t) => {\n    const { tokens, ...tokenWithoutUnderlyings } = processToken(t);\n\n    return tokenWithoutUnderlyings;\n  });\n\n  const marketValue = processedTokens.reduce(\n    (acc, t) =>\n      acc === undefined || t.marketValue === undefined\n        ? undefined\n        : acc + t.marketValue,\n    0 as number | undefined,\n  );\n\n  return {\n    ...tokenBalance,\n    marketValue,\n    tokens: processedTokens,\n  };\n}\n"]}

package/dist/DeFiPositionsController/group-defi-positions.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"group-defi-positions.d.cts","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAE3C,OAAO,KAAK,EACV,oBAAoB,EACpB,YAAY,EACZ,aAAa,EACb,UAAU,EAEX,8BAA0B;AAE3B,MAAM,MAAM,oBAAoB,GAAG;IACjC,qBAAqB,EAAE,MAAM,CAAC;IAC9B,SAAS,EAAE;QACT,CAAC,UAAU,EAAE,MAAM,GAAG;YACpB,eAAe,EAAE;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,OAAO,EAAE,MAAM,CAAA;aAAE,CAAC;YACnD,qBAAqB,EAAE,MAAM,CAAC;YAC9B,aAAa,EAAE;iBACZ,GAAG,IAAI,YAAY,CAAC,CAAC,EAAE;oBACtB,qBAAqB,EAAE,MAAM,CAAC;oBAC9B,SAAS,EAAE,4BAA4B,EAAE,EAAE,CAAC;iBAC7C;aACF,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,GAAG;IACzE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG;IACnE,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,qBAAqB,EAAE,oBAAoB,EAAE,GAC5C;IACD,CAAC,GAAG,EAAE,GAAG,GAAG,oBAAoB,CAAC;CAClC,CAqEA"}
+{"version":3,"file":"group-defi-positions.d.cts","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAE3C,OAAO,KAAK,EACV,oBAAoB,EACpB,YAAY,EACZ,aAAa,EACb,UAAU,EAEX,8BAA0B;AAE3B,MAAM,MAAM,oBAAoB,GAAG;IACjC,qBAAqB,EAAE,MAAM,CAAC;IAC9B,SAAS,EAAE;QACT,CAAC,UAAU,EAAE,MAAM,GAAG;YACpB,eAAe,EAAE;gBACf,IAAI,EAAE,MAAM,CAAC;gBACb,OAAO,EAAE,MAAM,CAAC;aACjB,CAAC;YACF,qBAAqB,EAAE,MAAM,CAAC;YAC9B,aAAa,EAAE;iBACZ,GAAG,IAAI,YAAY,CAAC,CAAC,EAAE;oBACtB,qBAAqB,EAAE,MAAM,CAAC;oBAC9B,SAAS,EAAE,4BAA4B,EAAE,EAAE,CAAC;iBAC7C;aACF,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,GAAG;IACzE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG;IACnE,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,qBAAqB,EAAE,oBAAoB,EAAE,GAC5C;IACD,CAAC,GAAG,EAAE,GAAG,GAAG,oBAAoB,CAAC;CAClC,CAqEA"}

package/dist/DeFiPositionsController/group-defi-positions.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"group-defi-positions.d.mts","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAE3C,OAAO,KAAK,EACV,oBAAoB,EACpB,YAAY,EACZ,aAAa,EACb,UAAU,EAEX,8BAA0B;AAE3B,MAAM,MAAM,oBAAoB,GAAG;IACjC,qBAAqB,EAAE,MAAM,CAAC;IAC9B,SAAS,EAAE;QACT,CAAC,UAAU,EAAE,MAAM,GAAG;YACpB,eAAe,EAAE;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,OAAO,EAAE,MAAM,CAAA;aAAE,CAAC;YACnD,qBAAqB,EAAE,MAAM,CAAC;YAC9B,aAAa,EAAE;iBACZ,GAAG,IAAI,YAAY,CAAC,CAAC,EAAE;oBACtB,qBAAqB,EAAE,MAAM,CAAC;oBAC9B,SAAS,EAAE,4BAA4B,EAAE,EAAE,CAAC;iBAC7C;aACF,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,GAAG;IACzE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG;IACnE,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,qBAAqB,EAAE,oBAAoB,EAAE,GAC5C;IACD,CAAC,GAAG,EAAE,GAAG,GAAG,oBAAoB,CAAC;CAClC,CAqEA"}
+{"version":3,"file":"group-defi-positions.d.mts","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAE3C,OAAO,KAAK,EACV,oBAAoB,EACpB,YAAY,EACZ,aAAa,EACb,UAAU,EAEX,8BAA0B;AAE3B,MAAM,MAAM,oBAAoB,GAAG;IACjC,qBAAqB,EAAE,MAAM,CAAC;IAC9B,SAAS,EAAE;QACT,CAAC,UAAU,EAAE,MAAM,GAAG;YACpB,eAAe,EAAE;gBACf,IAAI,EAAE,MAAM,CAAC;gBACb,OAAO,EAAE,MAAM,CAAC;aACjB,CAAC;YACF,qBAAqB,EAAE,MAAM,CAAC;YAC9B,aAAa,EAAE;iBACZ,GAAG,IAAI,YAAY,CAAC,CAAC,EAAE;oBACtB,qBAAqB,EAAE,MAAM,CAAC;oBAC9B,SAAS,EAAE,4BAA4B,EAAE,EAAE,CAAC;iBAC7C;aACF,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,GAAG;IACzE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG;IACnE,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,qBAAqB,EAAE,oBAAoB,EAAE,GAC5C;IACD,CAAC,GAAG,EAAE,GAAG,GAAG,oBAAoB,CAAC;CAClC,CAqEA"}

package/dist/DeFiPositionsController/group-defi-positions.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"group-defi-positions.mjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,mCAAmC;AAoCnD;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,qBAA6C;IAI7C,MAAM,oBAAoB,GAAyC,EAAE,CAAC;IAEtE,KAAK,MAAM,QAAQ,IAAI,qBAAqB,EAAE;QAC5C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE;YACrB,SAAS;SACV;QAED,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,GACvE,QAAQ,CAAC;QAEX,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;QAE7B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE;YAChC,oBAAoB,CAAC,KAAK,CAAC,GAAG;gBAC5B,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;SACH;QAED,MAAM,SAAS,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE;YACpC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG;gBAChC,eAAe,EAAE;oBACf,IAAI,EAAE,mBAAmB;oBACzB,OAAO;iBACR;gBACD,qBAAqB,EAAE,CAAC;gBACxB,aAAa,EAAE,EAAE;aAClB,CAAC;SACH;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAErD,IAAI,gBAAgB,GAAG,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,EAAE;YACrB,gBAAgB,GAAG;gBACjB,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;YACF,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,gBAAgB,CAAC;SAC7D;QAED,KAAK,MAAM,aAAa,IAAI,QAAQ,CAAC,MAAM,EAAE;YAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,aAAa,CAAiC,CAAC;YAE1E,qEAAqE;YACrE,IAAI,QAAQ,CAAC,QAAQ,EAAE,cAAc,EAAE;gBACrC,IAAI,gBAAgB,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC3C,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;iBAC1C;qBAAM;oBACL,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBAC3C;aACF;iBAAM;gBACL,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;aAC1C;YAED,IAAI,KAAK,CAAC,WAAW,EAAE;gBACrB,MAAM,UAAU,GAAG,QAAQ,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE/D,gBAAgB,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,CAAC;gBAC5D,YAAY,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;gBACrE,SAAS,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;aACnE;SACF;KACF;IAED,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED;;;;GAIG;AACH,SAAS,YAAY,CACnB,YAAe;IAKf,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QACxB,OAAO;YACL,GAAG,YAAY;YACf,WAAW,EAAE,YAAY,CAAC,KAAK;gBAC7B,CAAC,CAAC,YAAY,CAAC,OAAO,GAAG,YAAY,CAAC,KAAK;gBAC3C,CAAC,CAAC,SAAS;SACd,CAAC;KACH;IAED,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;QAE/D,OAAO,uBAAuB,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CACxC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CACT,GAAG,KAAK,SAAS,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS;QAC9C,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,EACzB,CAAuB,CACxB,CAAC;IAEF,OAAO;QACL,GAAG,YAAY;QACf,WAAW;QACX,MAAM,EAAE,eAAe;KACxB,CAAC;AACJ,CAAC","sourcesContent":["import { toHex } from '@metamask/controller-utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n  DefiPositionResponse,\n  PositionType,\n  ProtocolToken,\n  Underlying,\n  Balance,\n} from './fetch-positions';\n\nexport type GroupedDeFiPositions = {\n  aggregatedMarketValue: number;\n  protocols: {\n    [protocolId: string]: {\n      protocolDetails: { name: string; iconUrl: string };\n      aggregatedMarketValue: number;\n      positionTypes: {\n        [key in PositionType]?: {\n          aggregatedMarketValue: number;\n          positions: ProtocolTokenWithMarketValue[][];\n        };\n      };\n    };\n  };\n};\n\nexport type ProtocolTokenWithMarketValue = Omit<ProtocolToken, 'tokens'> & {\n  marketValue?: number;\n  tokens: UnderlyingWithMarketValue[];\n};\n\nexport type UnderlyingWithMarketValue = Omit<Underlying, 'tokens'> & {\n  marketValue?: number;\n};\n\n/**\n *\n * @param defiPositionsResponse - The response from the defi positions API\n * @returns The grouped positions that get assigned to the state\n */\nexport function groupDeFiPositions(\n  defiPositionsResponse: DefiPositionResponse[],\n): {\n  [key: Hex]: GroupedDeFiPositions;\n} {\n  const groupedDeFiPositions: { [key: Hex]: GroupedDeFiPositions } = {};\n\n  for (const position of defiPositionsResponse) {\n    if (!position.success) {\n      continue;\n    }\n\n    const { chainId, protocolId, iconUrl, positionType, protocolDisplayName } =\n      position;\n\n    const chain = toHex(chainId);\n\n    if (!groupedDeFiPositions[chain]) {\n      groupedDeFiPositions[chain] = {\n        aggregatedMarketValue: 0,\n        protocols: {},\n      };\n    }\n\n    const chainData = groupedDeFiPositions[chain];\n\n    if (!chainData.protocols[protocolId]) {\n      chainData.protocols[protocolId] = {\n        protocolDetails: {\n          name: protocolDisplayName,\n          iconUrl,\n        },\n        aggregatedMarketValue: 0,\n        positionTypes: {},\n      };\n    }\n\n    const protocolData = chainData.protocols[protocolId];\n\n    let positionTypeData = protocolData.positionTypes[positionType];\n    if (!positionTypeData) {\n      positionTypeData = {\n        aggregatedMarketValue: 0,\n        positions: [],\n      };\n      protocolData.positionTypes[positionType] = positionTypeData;\n    }\n\n    for (const protocolToken of position.tokens) {\n      const token = processToken(protocolToken) as ProtocolTokenWithMarketValue;\n\n      // If groupPositions is true, we group all positions of the same type\n      if (position.metadata?.groupPositions) {\n        if (positionTypeData.positions.length === 0) {\n          positionTypeData.positions.push([token]);\n        } else {\n          positionTypeData.positions[0].push(token);\n        }\n      } else {\n        positionTypeData.positions.push([token]);\n      }\n\n      if (token.marketValue) {\n        const multiplier = position.positionType === 'borrow' ? -1 : 1;\n\n        positionTypeData.aggregatedMarketValue += token.marketValue;\n        protocolData.aggregatedMarketValue += token.marketValue * multiplier;\n        chainData.aggregatedMarketValue += token.marketValue * multiplier;\n      }\n    }\n  }\n\n  return groupedDeFiPositions;\n}\n\n/**\n *\n * @param tokenBalance - The token balance that is going to be processed\n * @returns The processed token balance\n */\nfunction processToken<T extends Balance>(\n  tokenBalance: T,\n): T & {\n  marketValue?: number;\n  tokens?: UnderlyingWithMarketValue[];\n} {\n  if (!tokenBalance.tokens) {\n    return {\n      ...tokenBalance,\n      marketValue: tokenBalance.price\n        ? tokenBalance.balance * tokenBalance.price\n        : undefined,\n    };\n  }\n\n  const processedTokens = tokenBalance.tokens.map((t) => {\n    const { tokens, ...tokenWithoutUnderlyings } = processToken(t);\n\n    return tokenWithoutUnderlyings;\n  });\n\n  const marketValue = processedTokens.reduce(\n    (acc, t) =>\n      acc === undefined || t.marketValue === undefined\n        ? undefined\n        : acc + t.marketValue,\n    0 as number | undefined,\n  );\n\n  return {\n    ...tokenBalance,\n    marketValue,\n    tokens: processedTokens,\n  };\n}\n"]}
+{"version":3,"file":"group-defi-positions.mjs","sourceRoot":"","sources":["../../src/DeFiPositionsController/group-defi-positions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,mCAAmC;AAuCnD;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,qBAA6C;IAI7C,MAAM,oBAAoB,GAAyC,EAAE,CAAC;IAEtE,KAAK,MAAM,QAAQ,IAAI,qBAAqB,EAAE;QAC5C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE;YACrB,SAAS;SACV;QAED,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,GACvE,QAAQ,CAAC;QAEX,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;QAE7B,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE;YAChC,oBAAoB,CAAC,KAAK,CAAC,GAAG;gBAC5B,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;SACH;QAED,MAAM,SAAS,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE;YACpC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG;gBAChC,eAAe,EAAE;oBACf,IAAI,EAAE,mBAAmB;oBACzB,OAAO;iBACR;gBACD,qBAAqB,EAAE,CAAC;gBACxB,aAAa,EAAE,EAAE;aAClB,CAAC;SACH;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAErD,IAAI,gBAAgB,GAAG,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,EAAE;YACrB,gBAAgB,GAAG;gBACjB,qBAAqB,EAAE,CAAC;gBACxB,SAAS,EAAE,EAAE;aACd,CAAC;YACF,YAAY,CAAC,aAAa,CAAC,YAAY,CAAC,GAAG,gBAAgB,CAAC;SAC7D;QAED,KAAK,MAAM,aAAa,IAAI,QAAQ,CAAC,MAAM,EAAE;YAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,aAAa,CAAiC,CAAC;YAE1E,qEAAqE;YACrE,IAAI,QAAQ,CAAC,QAAQ,EAAE,cAAc,EAAE;gBACrC,IAAI,gBAAgB,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC3C,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;iBAC1C;qBAAM;oBACL,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBAC3C;aACF;iBAAM;gBACL,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;aAC1C;YAED,IAAI,KAAK,CAAC,WAAW,EAAE;gBACrB,MAAM,UAAU,GAAG,QAAQ,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE/D,gBAAgB,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,CAAC;gBAC5D,YAAY,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;gBACrE,SAAS,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,CAAC;aACnE;SACF;KACF;IAED,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED;;;;GAIG;AACH,SAAS,YAAY,CACnB,YAAe;IAKf,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QACxB,OAAO;YACL,GAAG,YAAY;YACf,WAAW,EAAE,YAAY,CAAC,KAAK;gBAC7B,CAAC,CAAC,YAAY,CAAC,OAAO,GAAG,YAAY,CAAC,KAAK;gBAC3C,CAAC,CAAC,SAAS;SACd,CAAC;KACH;IAED,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;QAE/D,OAAO,uBAAuB,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,eAAe,CAAC,MAAM,CACxC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CACT,GAAG,KAAK,SAAS,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS;QAC9C,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,WAAW,EACzB,CAAuB,CACxB,CAAC;IAEF,OAAO;QACL,GAAG,YAAY;QACf,WAAW;QACX,MAAM,EAAE,eAAe;KACxB,CAAC;AACJ,CAAC","sourcesContent":["import { toHex } from '@metamask/controller-utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n  DefiPositionResponse,\n  PositionType,\n  ProtocolToken,\n  Underlying,\n  Balance,\n} from './fetch-positions';\n\nexport type GroupedDeFiPositions = {\n  aggregatedMarketValue: number;\n  protocols: {\n    [protocolId: string]: {\n      protocolDetails: {\n        name: string;\n        iconUrl: string;\n      };\n      aggregatedMarketValue: number;\n      positionTypes: {\n        [key in PositionType]?: {\n          aggregatedMarketValue: number;\n          positions: ProtocolTokenWithMarketValue[][];\n        };\n      };\n    };\n  };\n};\n\nexport type ProtocolTokenWithMarketValue = Omit<ProtocolToken, 'tokens'> & {\n  marketValue?: number;\n  tokens: UnderlyingWithMarketValue[];\n};\n\nexport type UnderlyingWithMarketValue = Omit<Underlying, 'tokens'> & {\n  marketValue?: number;\n};\n\n/**\n *\n * @param defiPositionsResponse - The response from the defi positions API\n * @returns The grouped positions that get assigned to the state\n */\nexport function groupDeFiPositions(\n  defiPositionsResponse: DefiPositionResponse[],\n): {\n  [key: Hex]: GroupedDeFiPositions;\n} {\n  const groupedDeFiPositions: { [key: Hex]: GroupedDeFiPositions } = {};\n\n  for (const position of defiPositionsResponse) {\n    if (!position.success) {\n      continue;\n    }\n\n    const { chainId, protocolId, iconUrl, positionType, protocolDisplayName } =\n      position;\n\n    const chain = toHex(chainId);\n\n    if (!groupedDeFiPositions[chain]) {\n      groupedDeFiPositions[chain] = {\n        aggregatedMarketValue: 0,\n        protocols: {},\n      };\n    }\n\n    const chainData = groupedDeFiPositions[chain];\n\n    if (!chainData.protocols[protocolId]) {\n      chainData.protocols[protocolId] = {\n        protocolDetails: {\n          name: protocolDisplayName,\n          iconUrl,\n        },\n        aggregatedMarketValue: 0,\n        positionTypes: {},\n      };\n    }\n\n    const protocolData = chainData.protocols[protocolId];\n\n    let positionTypeData = protocolData.positionTypes[positionType];\n    if (!positionTypeData) {\n      positionTypeData = {\n        aggregatedMarketValue: 0,\n        positions: [],\n      };\n      protocolData.positionTypes[positionType] = positionTypeData;\n    }\n\n    for (const protocolToken of position.tokens) {\n      const token = processToken(protocolToken) as ProtocolTokenWithMarketValue;\n\n      // If groupPositions is true, we group all positions of the same type\n      if (position.metadata?.groupPositions) {\n        if (positionTypeData.positions.length === 0) {\n          positionTypeData.positions.push([token]);\n        } else {\n          positionTypeData.positions[0].push(token);\n        }\n      } else {\n        positionTypeData.positions.push([token]);\n      }\n\n      if (token.marketValue) {\n        const multiplier = position.positionType === 'borrow' ? -1 : 1;\n\n        positionTypeData.aggregatedMarketValue += token.marketValue;\n        protocolData.aggregatedMarketValue += token.marketValue * multiplier;\n        chainData.aggregatedMarketValue += token.marketValue * multiplier;\n      }\n    }\n  }\n\n  return groupedDeFiPositions;\n}\n\n/**\n *\n * @param tokenBalance - The token balance that is going to be processed\n * @returns The processed token balance\n */\nfunction processToken<T extends Balance>(\n  tokenBalance: T,\n): T & {\n  marketValue?: number;\n  tokens?: UnderlyingWithMarketValue[];\n} {\n  if (!tokenBalance.tokens) {\n    return {\n      ...tokenBalance,\n      marketValue: tokenBalance.price\n        ? tokenBalance.balance * tokenBalance.price\n        : undefined,\n    };\n  }\n\n  const processedTokens = tokenBalance.tokens.map((t) => {\n    const { tokens, ...tokenWithoutUnderlyings } = processToken(t);\n\n    return tokenWithoutUnderlyings;\n  });\n\n  const marketValue = processedTokens.reduce(\n    (acc, t) =>\n      acc === undefined || t.marketValue === undefined\n        ? undefined\n        : acc + t.marketValue,\n    0 as number | undefined,\n  );\n\n  return {\n    ...tokenBalance,\n    marketValue,\n    tokens: processedTokens,\n  };\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@metamask-previews/assets-controllers",
-  "version": "65.0.0-preview-de21a3fe",
+  "version": "66.0.0-preview-4ef4efd7",
   "description": "Controllers which manage interactions involving ERC-20, ERC-721, and ERC-1155 tokens (including NFTs)",
   "keywords": [
     "MetaMask",
@@ -84,14 +84,14 @@
     "@metamask/keyring-controller": "^22.0.0",
     "@metamask/keyring-internal-api": "^6.0.1",
     "@metamask/keyring-snap-client": "^4.1.0",
-    "@metamask/network-controller": "^23.5.0",
+    "@metamask/network-controller": "^23.5.1",
     "@metamask/permission-controller": "^11.0.6",
     "@metamask/phishing-controller": "^12.5.0",
-    "@metamask/preferences-controller": "^18.0.0",
+    "@metamask/preferences-controller": "^18.1.0",
     "@metamask/providers": "^21.0.0",
     "@metamask/snaps-controllers": "^11.2.1",
     "@metamask/snaps-sdk": "^6.22.0",
-    "@metamask/transaction-controller": "^56.2.0",
+    "@metamask/transaction-controller": "^56.3.0",
     "@types/jest": "^27.4.1",
     "@types/lodash": "^4.14.191",
     "@types/node": "^16.18.54",