@metamask-previews/earn-controller 0.9.0-preview-62c1e797 → 0.10.0-preview-7ecf950

package/CHANGELOG.md

@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.10.0]
+
+### Changed
+
+- **BREAKING:** Updated `EarnController` methods (`refreshPooledStakingData`, `refreshPooledStakes`, and `refreshStakingEligibility`) to use an options bag parameter ([#5537](https://github.com/MetaMask/core/pull/5537))
+
 ## [0.9.0]
 
 ### Changed
@@ -69,7 +75,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Initial release ([#5271](https://github.com/MetaMask/core/pull/5271))
 
-[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.9.0...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.10.0...HEAD
+[0.10.0]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.9.0...@metamask/earn-controller@0.10.0
 [0.9.0]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.8.0...@metamask/earn-controller@0.9.0
 [0.8.0]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.7.0...@metamask/earn-controller@0.8.0
 [0.7.0]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.6.0...@metamask/earn-controller@0.7.0

package/dist/EarnController.cjs

@@ -122,9 +122,15 @@ class EarnController extends base_controller_1.BaseController {
             __classPrivateFieldSet(this, _EarnController_selectedNetworkClientId, networkControllerState.selectedNetworkClientId, "f");
         });
         // Listen for account changes
-        this.messagingSystem.subscribe('AccountsController:selectedAccountChange', () => {
-            this.refreshStakingEligibility().catch(console.error);
-            this.refreshPooledStakes().catch(console.error);
+        this.messagingSystem.subscribe('AccountsController:selectedAccountChange', (account) => {
+            const address = account?.address;
+            /**
+             * TEMP: There's a race condition where the account state isn't updated immediately.
+             * Until this has been fixed, we rely on the event payload for the latest account instead of #getCurrentAccount().
+             * Issue: https://github.com/MetaMask/accounts-planning/issues/887
+             */
+            this.refreshStakingEligibility({ address }).catch(console.error);
+            this.refreshPooledStakes({ address }).catch(console.error);
         });
     }
     /**
@@ -132,16 +138,18 @@ class EarnController extends base_controller_1.BaseController {
      * Fetches updated stake information including lifetime rewards, assets, and exit requests
      * from the staking API service and updates the state.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when the stakes data has been updated
      */
-    async refreshPooledStakes(resetCache = false) {
-        const currentAccount = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this);
-        if (!currentAccount?.address) {
+    async refreshPooledStakes({ resetCache = false, address, } = {}) {
+        const addressToUse = address ?? __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this)?.address;
+        if (!addressToUse) {
             return;
         }
         const chainId = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentChainId).call(this);
-        const { accounts, exchangeRate } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakes([currentAccount.address], chainId, resetCache);
+        const { accounts, exchangeRate } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakes([addressToUse], chainId, resetCache);
         this.update((state) => {
             state.pooled_staking.pooledStakes = accounts[0];
             state.pooled_staking.exchangeRate = exchangeRate;
@@ -151,15 +159,17 @@ class EarnController extends base_controller_1.BaseController {
      * Refreshes the staking eligibility status for the current account.
      * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.
      *
+     * @param options - Optional arguments
+     * @param [options.address] - Address to refresh staking eligibility for (optional).
      * @returns A promise that resolves when the eligibility status has been updated
      */
-    async refreshStakingEligibility() {
-        const currentAccount = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this);
-        if (!currentAccount?.address) {
+    async refreshStakingEligibility({ address, } = {}) {
+        const addressToCheck = address ?? __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this)?.address;
+        if (!addressToCheck) {
             return;
         }
         const { eligible: isEligible } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakingEligibility([
-            currentAccount.address,
+            addressToCheck,
         ]);
         this.update((state) => {
             state.pooled_staking.isEligible = isEligible;
@@ -212,17 +222,19 @@ class EarnController extends base_controller_1.BaseController {
      * This method allows partial success, meaning some data may update while other requests fail.
      * All errors are collected and thrown as a single error message.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when all possible data has been updated
      * @throws {Error} If any of the refresh operations fail, with concatenated error messages
      */
-    async refreshPooledStakingData(resetCache = false) {
+    async refreshPooledStakingData({ resetCache, address, } = {}) {
         const errors = [];
         await Promise.all([
-            this.refreshPooledStakes(resetCache).catch((error) => {
+            this.refreshPooledStakes({ resetCache, address }).catch((error) => {
                 errors.push(error);
             }),
-            this.refreshStakingEligibility().catch((error) => {
+            this.refreshStakingEligibility({ address }).catch((error) => {
                 errors.push(error);
             }),
             this.refreshPooledStakingVaultMetadata().catch((error) => {

package/dist/EarnController.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"EarnController.cjs","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,wDAAwD;AAWxD,+DAA2D;AAC3D,iEAAiE;AAMjE,mDAQ6B;AAEhB,QAAA,cAAc,GAAG,gBAAgB,CAAC;AA0B/C;;GAEG;AACH,MAAM,sBAAsB,GAAuC;IACjE,cAAc,EAAE;QACd,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,kBAAkB,EAAE;QAClB,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,WAAW,EAAE;QACX,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI;KAChB;CACF,CAAC;AASF,wBAAwB;AACxB,MAAM,wBAAwB,GAAoB;IAChD,MAAM,EAAE,EAAE;IACV,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,CAAC;IACV,YAAY,EAAE,EAAE;IAChB,YAAY,EAAE,EAAE;IAChB,UAAU,EAAE,GAAG;IACf,MAAM,EAAE,GAAG;IACX,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,MAAM,yCAAyC,GAAqB;IAClE,MAAM,EAAE,GAAG;IACX,OAAO,EAAE,GAAG;IACZ,QAAQ,EAAE,GAAG;IACb,WAAW,EAAE,GAAG;IAChB,SAAS,EAAE,GAAG;IACd,OAAO,EAAE,GAAG;CACb,CAAC;AAEF;;;;GAIG;AACH,SAAgB,6BAA6B;IAC3C,OAAO;QACL,cAAc,EAAE;YACd,YAAY,EAAE;gBACZ,OAAO,EAAE,EAAE;gBACX,eAAe,EAAE,GAAG;gBACpB,MAAM,EAAE,GAAG;gBACX,YAAY,EAAE,EAAE;aACjB;YACD,YAAY,EAAE,GAAG;YACjB,aAAa,EAAE;gBACb,GAAG,EAAE,GAAG;gBACR,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,4CAA4C;aAC3D;YACD,cAAc,EAAE,EAAE;YAClB,gBAAgB,EAAE,yCAAyC;YAC3D,UAAU,EAAE,KAAK;SAClB;QACD,kBAAkB,EAAE;YAClB,MAAM,EAAE,CAAC,wBAAwB,CAAC;SACnC;QACD,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AA1BD,sEA0BC;AAyDD,gCAAgC;AAEhC;;GAEG;AACH,MAAa,cAAe,SAAQ,gCAInC;IAOC,YAAY,EACV,SAAS,EACT,KAAK,GAAG,EAAE,GAIX;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,sBAAc;YACpB,QAAQ,EAAE,sBAAsB;YAChC,SAAS;YACT,KAAK,EAAE;gBACL,GAAG,6BAA6B,EAAE;gBAClC,GAAG,KAAK;aACT;SACF,CAAC,CAAC;;QArBL,mCAA6B,IAAI,EAAC;QAElC,0DAAkC;QAEzB,4CAAwC,IAAI,6BAAiB,EAAE,EAAC;QAmBvE,uBAAA,IAAI,gEAAe,MAAnB,IAAI,CAAiB,CAAC;QACtB,IAAI,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAErD,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;QACF,uBAAA,IAAI,2CAA4B,uBAAuB,MAAA,CAAC;QAExD,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,CAAC,sBAAsB,EAAE,EAAE;YACzB,IACE,sBAAsB,CAAC,uBAAuB;gBAC9C,uBAAA,IAAI,+CAAyB,EAC7B;gBACA,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,sBAAsB,CAAC,uBAAuB,CAAC,CAAC;gBACpE,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC9D,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC/D,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjE,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;aACjD;YACD,uBAAA,IAAI,2CACF,sBAAsB,CAAC,uBAAuB,MAAA,CAAC;QACnD,CAAC,CACF,CAAC;QAEF,6BAA6B;QAC7B,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,0CAA0C,EAC1C,GAAG,EAAE;YACH,IAAI,CAAC,yBAAyB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtD,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAClD,CAAC,CACF,CAAC;IACJ,CAAC;IA2DD;;;;;;;OAOG;IACH,KAAK,CAAC,mBAAmB,CAAC,UAAU,GAAG,KAAK;QAC1C,MAAM,cAAc,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QACjD,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;YAC5B,OAAO;SACR;QAED,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1C,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAC9B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,eAAe,CAC3C,CAAC,cAAc,CAAC,OAAO,CAAC,EACxB,OAAO,EACP,UAAU,CACX,CAAC;QAEJ,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAChD,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,YAAY,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,yBAAyB;QAC7B,MAAM,cAAc,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QACjD,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;YAC5B,OAAO;SACR;QAED,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAC5B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,2BAA2B,CAAC;YACxD,cAAc,CAAC,OAAO;SACvB,CAAC,CAAC;QAEL,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,iCAAiC;QACrC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE1E,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,aAAa,GAAG,aAAa,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,kCAAkC,CACtC,IAAI,GAAG,GAAG,EACV,QAAwB,MAAM;QAE9B,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,iBAAiB,CACpE,OAAO,EACP,IAAI,EACJ,KAAK,CACN,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,cAAc,GAAG,cAAc,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,oCAAoC;QACxC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,gBAAgB,GACpB,MAAM,uBAAA,IAAI,yCAAmB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAE7D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,wBAAwB,CAAC,UAAU,GAAG,KAAK;QAC/C,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACnD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,yBAAyB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC/C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACxD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;YACrB,MAAM,IAAI,KAAK,CACb,wCAAwC,MAAM;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;iBACrB,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;SACH;IACH,CAAC;CACF;AA1QD,wCA0QC;wQA1MgB,eAAwB;IACrC,MAAM,EAAE,uBAAuB,EAAE,GAAG,eAAe;QACjD,CAAC,CAAC,EAAE,uBAAuB,EAAE,eAAe,EAAE;QAC9C,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAE5D,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7C,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE;QAC5B,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,OAAO;KACR;IAED,MAAM,QAAQ,GAAG,IAAI,wBAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,aAAa,CAAC;IAEhD,oDAAoD;IACpD,MAAM,MAAM,GAAmB;QAC7B,OAAO,EAAE,IAAA,sCAAmB,EAAC,OAAO,CAAC;KACtC,CAAC;IAEF,IAAI;QACF,uBAAA,IAAI,4BAAa,oBAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAA,CAAC;QACzC,uBAAA,IAAI,gCAAU,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;KACxE;IAAC,OAAO,KAAK,EAAE;QACd,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,2DAA2D;QAC3D,IACE,CAAC,CACC,KAAK,YAAY,KAAK;YACtB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAC9C,EACD;YACA,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;SAC1D;KACF;AACH,CAAC;IAGC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;AAC5E,CAAC;IAGC,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;IACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,GAC3B,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IACF,OAAO,IAAA,sCAAmB,EAAC,OAAO,CAAC,CAAC;AACtC,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport { convertHexToDecimal } from '@metamask/controller-utils';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n} from '@metamask/network-controller';\nimport {\n  StakeSdk,\n  StakingApiService,\n  type PooledStake,\n  type StakeSdkConfig,\n  type VaultData,\n  type VaultDailyApy,\n  type VaultApyAverages,\n} from '@metamask/stake-sdk';\n\nexport const controllerName = 'EarnController';\n\nexport type PooledStakingState = {\n  pooledStakes: PooledStake;\n  exchangeRate: string;\n  vaultMetadata: VaultData;\n  vaultDailyApys: VaultDailyApy[];\n  vaultApyAverages: VaultApyAverages;\n  isEligible: boolean;\n};\n\nexport type StablecoinLendingState = {\n  vaults: StablecoinVault[];\n};\n\nexport type StablecoinVault = {\n  symbol: string;\n  name: string;\n  chainId: number;\n  tokenAddress: string;\n  vaultAddress: string;\n  currentAPY: string;\n  supply: string;\n  liquidity: string;\n};\n\n/**\n * Metadata for the EarnController.\n */\nconst earnControllerMetadata: StateMetadata<EarnControllerState> = {\n  pooled_staking: {\n    persist: true,\n    anonymous: false,\n  },\n  stablecoin_lending: {\n    persist: true,\n    anonymous: false,\n  },\n  lastUpdated: {\n    persist: false,\n    anonymous: true,\n  },\n};\n\n// === State Types ===\nexport type EarnControllerState = {\n  pooled_staking: PooledStakingState;\n  stablecoin_lending?: StablecoinLendingState;\n  lastUpdated: number;\n};\n\n// === Default State ===\nconst DEFAULT_STABLECOIN_VAULT: StablecoinVault = {\n  symbol: '',\n  name: '',\n  chainId: 0,\n  tokenAddress: '',\n  vaultAddress: '',\n  currentAPY: '0',\n  supply: '0',\n  liquidity: '0',\n};\n\nconst DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES: VaultApyAverages = {\n  oneDay: '0',\n  oneWeek: '0',\n  oneMonth: '0',\n  threeMonths: '0',\n  sixMonths: '0',\n  oneYear: '0',\n};\n\n/**\n * Gets the default state for the EarnController.\n *\n * @returns The default EarnController state.\n */\nexport function getDefaultEarnControllerState(): EarnControllerState {\n  return {\n    pooled_staking: {\n      pooledStakes: {\n        account: '',\n        lifetimeRewards: '0',\n        assets: '0',\n        exitRequests: [],\n      },\n      exchangeRate: '1',\n      vaultMetadata: {\n        apy: '0',\n        capacity: '0',\n        feePercent: 0,\n        totalAssets: '0',\n        vaultAddress: '0x0000000000000000000000000000000000000000',\n      },\n      vaultDailyApys: [],\n      vaultApyAverages: DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES,\n      isEligible: false,\n    },\n    stablecoin_lending: {\n      vaults: [DEFAULT_STABLECOIN_VAULT],\n    },\n    lastUpdated: 0,\n  };\n}\n\n// === MESSENGER ===\n\n/**\n * The action which can be used to retrieve the state of the EarnController.\n */\nexport type EarnControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All actions that EarnController registers, to be called externally.\n */\nexport type EarnControllerActions = EarnControllerGetStateAction;\n\n/**\n * All actions that EarnController calls internally.\n */\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction;\n\n/**\n * The event that EarnController publishes when updating state.\n */\nexport type EarnControllerStateChangeEvent = ControllerStateChangeEvent<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All events that EarnController publishes, to be subscribed to externally.\n */\nexport type EarnControllerEvents = EarnControllerStateChangeEvent;\n\n/**\n * All events that EarnController subscribes to internally.\n */\nexport type AllowedEvents =\n  | AccountsControllerSelectedAccountChangeEvent\n  | NetworkControllerStateChangeEvent;\n\n/**\n * The messenger which is restricted to actions and events accessed by\n * EarnController.\n */\nexport type EarnControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  EarnControllerActions | AllowedActions,\n  EarnControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n// === CONTROLLER DEFINITION ===\n\n/**\n * EarnController manages DeFi earning opportunities across different protocols and chains.\n */\nexport class EarnController extends BaseController<\n  typeof controllerName,\n  EarnControllerState,\n  EarnControllerMessenger\n> {\n  #stakeSDK: StakeSdk | null = null;\n\n  #selectedNetworkClientId?: string;\n\n  readonly #stakingApiService: StakingApiService = new StakingApiService();\n\n  constructor({\n    messenger,\n    state = {},\n  }: {\n    messenger: EarnControllerMessenger;\n    state?: Partial<EarnControllerState>;\n  }) {\n    super({\n      name: controllerName,\n      metadata: earnControllerMetadata,\n      messenger,\n      state: {\n        ...getDefaultEarnControllerState(),\n        ...state,\n      },\n    });\n\n    this.#initializeSDK();\n    this.refreshPooledStakingData().catch(console.error);\n\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    this.#selectedNetworkClientId = selectedNetworkClientId;\n\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      (networkControllerState) => {\n        if (\n          networkControllerState.selectedNetworkClientId !==\n          this.#selectedNetworkClientId\n        ) {\n          this.#initializeSDK(networkControllerState.selectedNetworkClientId);\n          this.refreshPooledStakingVaultMetadata().catch(console.error);\n          this.refreshPooledStakingVaultDailyApys().catch(console.error);\n          this.refreshPooledStakingVaultApyAverages().catch(console.error);\n          this.refreshPooledStakes().catch(console.error);\n        }\n        this.#selectedNetworkClientId =\n          networkControllerState.selectedNetworkClientId;\n      },\n    );\n\n    // Listen for account changes\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedAccountChange',\n      () => {\n        this.refreshStakingEligibility().catch(console.error);\n        this.refreshPooledStakes().catch(console.error);\n      },\n    );\n  }\n\n  #initializeSDK(networkClientId?: string) {\n    const { selectedNetworkClientId } = networkClientId\n      ? { selectedNetworkClientId: networkClientId }\n      : this.messagingSystem.call('NetworkController:getState');\n\n    const networkClient = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    if (!networkClient?.provider) {\n      this.#stakeSDK = null;\n      return;\n    }\n\n    const provider = new Web3Provider(networkClient.provider);\n    const { chainId } = networkClient.configuration;\n\n    // Initialize appropriate contracts based on chainId\n    const config: StakeSdkConfig = {\n      chainId: convertHexToDecimal(chainId),\n    };\n\n    try {\n      this.#stakeSDK = StakeSdk.create(config);\n      this.#stakeSDK.pooledStakingContract.connectSignerOrProvider(provider);\n    } catch (error) {\n      this.#stakeSDK = null;\n      // Only log unexpected errors, not unsupported chain errors\n      if (\n        !(\n          error instanceof Error &&\n          error.message.includes('Unsupported chainId')\n        )\n      ) {\n        console.error('Stake SDK initialization failed:', error);\n      }\n    }\n  }\n\n  #getCurrentAccount() {\n    return this.messagingSystem.call('AccountsController:getSelectedAccount');\n  }\n\n  #getCurrentChainId(): number {\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId },\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n    return convertHexToDecimal(chainId);\n  }\n\n  /**\n   * Refreshes the pooled stakes data for the current account.\n   * Fetches updated stake information including lifetime rewards, assets, and exit requests\n   * from the staking API service and updates the state.\n   *\n   * @param resetCache - Control whether the BE cache should be invalidated.\n   * @returns A promise that resolves when the stakes data has been updated\n   */\n  async refreshPooledStakes(resetCache = false): Promise<void> {\n    const currentAccount = this.#getCurrentAccount();\n    if (!currentAccount?.address) {\n      return;\n    }\n\n    const chainId = this.#getCurrentChainId();\n\n    const { accounts, exchangeRate } =\n      await this.#stakingApiService.getPooledStakes(\n        [currentAccount.address],\n        chainId,\n        resetCache,\n      );\n\n    this.update((state) => {\n      state.pooled_staking.pooledStakes = accounts[0];\n      state.pooled_staking.exchangeRate = exchangeRate;\n    });\n  }\n\n  /**\n   * Refreshes the staking eligibility status for the current account.\n   * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.\n   *\n   * @returns A promise that resolves when the eligibility status has been updated\n   */\n  async refreshStakingEligibility(): Promise<void> {\n    const currentAccount = this.#getCurrentAccount();\n    if (!currentAccount?.address) {\n      return;\n    }\n\n    const { eligible: isEligible } =\n      await this.#stakingApiService.getPooledStakingEligibility([\n        currentAccount.address,\n      ]);\n\n    this.update((state) => {\n      state.pooled_staking.isEligible = isEligible;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault metadata for the current chain.\n   * Updates the vault metadata in the controller state including APY, capacity,\n   * fee percentage, total assets, and vault address.\n   *\n   * @returns A promise that resolves when the vault metadata has been updated\n   */\n  async refreshPooledStakingVaultMetadata(): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultMetadata = await this.#stakingApiService.getVaultData(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultMetadata = vaultMetadata;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault daily apys for the current chain.\n   * Updates the pooled staking vault daily apys controller state.\n   *\n   * @param days - The number of days to fetch pooled staking vault daily apys for (defaults to 365).\n   * @param order - The order in which to fetch pooled staking vault daily apys. Descending order fetches the latest N days (latest working backwards). Ascending order fetches the oldest N days (oldest working forwards) (defaults to 'desc').\n   * @returns A promise that resolves when the pooled staking vault daily apys have been updated.\n   */\n  async refreshPooledStakingVaultDailyApys(\n    days = 365,\n    order: 'asc' | 'desc' = 'desc',\n  ): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultDailyApys = await this.#stakingApiService.getVaultDailyApys(\n      chainId,\n      days,\n      order,\n    );\n\n    this.update((state) => {\n      state.pooled_staking.vaultDailyApys = vaultDailyApys;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault apy averages for the current chain.\n   * Updates the pooled staking vault apy averages controller state.\n   *\n   * @returns A promise that resolves when the pooled staking vault apy averages have been updated.\n   */\n  async refreshPooledStakingVaultApyAverages() {\n    const chainId = this.#getCurrentChainId();\n    const vaultApyAverages =\n      await this.#stakingApiService.getVaultApyAverages(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultApyAverages = vaultApyAverages;\n    });\n  }\n\n  /**\n   * Refreshes all pooled staking related data including stakes, eligibility, and vault data.\n   * This method allows partial success, meaning some data may update while other requests fail.\n   * All errors are collected and thrown as a single error message.\n   *\n   * @param resetCache - Control whether the BE cache should be invalidated.\n   * @returns A promise that resolves when all possible data has been updated\n   * @throws {Error} If any of the refresh operations fail, with concatenated error messages\n   */\n  async refreshPooledStakingData(resetCache = false): Promise<void> {\n    const errors: Error[] = [];\n\n    await Promise.all([\n      this.refreshPooledStakes(resetCache).catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshStakingEligibility().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultMetadata().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultDailyApys().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultApyAverages().catch((error) => {\n        errors.push(error);\n      }),\n    ]);\n\n    if (errors.length > 0) {\n      throw new Error(\n        `Failed to refresh some staking data: ${errors\n          .map((e) => e.message)\n          .join(', ')}`,\n      );\n    }\n  }\n}\n"]}
+{"version":3,"file":"EarnController.cjs","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,wDAAwD;AAWxD,+DAA2D;AAC3D,iEAAiE;AAMjE,mDAQ6B;AAQhB,QAAA,cAAc,GAAG,gBAAgB,CAAC;AA0B/C;;GAEG;AACH,MAAM,sBAAsB,GAAuC;IACjE,cAAc,EAAE;QACd,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,kBAAkB,EAAE;QAClB,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,WAAW,EAAE;QACX,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI;KAChB;CACF,CAAC;AASF,wBAAwB;AACxB,MAAM,wBAAwB,GAAoB;IAChD,MAAM,EAAE,EAAE;IACV,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,CAAC;IACV,YAAY,EAAE,EAAE;IAChB,YAAY,EAAE,EAAE;IAChB,UAAU,EAAE,GAAG;IACf,MAAM,EAAE,GAAG;IACX,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,MAAM,yCAAyC,GAAqB;IAClE,MAAM,EAAE,GAAG;IACX,OAAO,EAAE,GAAG;IACZ,QAAQ,EAAE,GAAG;IACb,WAAW,EAAE,GAAG;IAChB,SAAS,EAAE,GAAG;IACd,OAAO,EAAE,GAAG;CACb,CAAC;AAEF;;;;GAIG;AACH,SAAgB,6BAA6B;IAC3C,OAAO;QACL,cAAc,EAAE;YACd,YAAY,EAAE;gBACZ,OAAO,EAAE,EAAE;gBACX,eAAe,EAAE,GAAG;gBACpB,MAAM,EAAE,GAAG;gBACX,YAAY,EAAE,EAAE;aACjB;YACD,YAAY,EAAE,GAAG;YACjB,aAAa,EAAE;gBACb,GAAG,EAAE,GAAG;gBACR,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,4CAA4C;aAC3D;YACD,cAAc,EAAE,EAAE;YAClB,gBAAgB,EAAE,yCAAyC;YAC3D,UAAU,EAAE,KAAK;SAClB;QACD,kBAAkB,EAAE;YAClB,MAAM,EAAE,CAAC,wBAAwB,CAAC;SACnC;QACD,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AA1BD,sEA0BC;AAyDD,gCAAgC;AAEhC;;GAEG;AACH,MAAa,cAAe,SAAQ,gCAInC;IAOC,YAAY,EACV,SAAS,EACT,KAAK,GAAG,EAAE,GAIX;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,sBAAc;YACpB,QAAQ,EAAE,sBAAsB;YAChC,SAAS;YACT,KAAK,EAAE;gBACL,GAAG,6BAA6B,EAAE;gBAClC,GAAG,KAAK;aACT;SACF,CAAC,CAAC;;QArBL,mCAA6B,IAAI,EAAC;QAElC,0DAAkC;QAEzB,4CAAwC,IAAI,6BAAiB,EAAE,EAAC;QAmBvE,uBAAA,IAAI,gEAAe,MAAnB,IAAI,CAAiB,CAAC;QACtB,IAAI,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAErD,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;QACF,uBAAA,IAAI,2CAA4B,uBAAuB,MAAA,CAAC;QAExD,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,CAAC,sBAAsB,EAAE,EAAE;YACzB,IACE,sBAAsB,CAAC,uBAAuB;gBAC9C,uBAAA,IAAI,+CAAyB,EAC7B;gBACA,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,sBAAsB,CAAC,uBAAuB,CAAC,CAAC;gBACpE,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC9D,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC/D,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjE,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;aACjD;YACD,uBAAA,IAAI,2CACF,sBAAsB,CAAC,uBAAuB,MAAA,CAAC;QACnD,CAAC,CACF,CAAC;QAEF,6BAA6B;QAC7B,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,0CAA0C,EAC1C,CAAC,OAAO,EAAE,EAAE;YACV,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC;YACjC;;;;eAIG;YACH,IAAI,CAAC,yBAAyB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACjE,IAAI,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC,CACF,CAAC;IACJ,CAAC;IA2DD;;;;;;;;;OASG;IACH,KAAK,CAAC,mBAAmB,CAAC,EACxB,UAAU,GAAG,KAAK,EAClB,OAAO,MACuB,EAAE;QAChC,MAAM,YAAY,GAAG,OAAO,IAAI,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,EAAE,OAAO,CAAC;QAEnE,IAAI,CAAC,YAAY,EAAE;YACjB,OAAO;SACR;QAED,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1C,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAC9B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,eAAe,CAC3C,CAAC,YAAY,CAAC,EACd,OAAO,EACP,UAAU,CACX,CAAC;QAEJ,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAChD,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,YAAY,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,yBAAyB,CAAC,EAC9B,OAAO,MAC6B,EAAE;QACtC,MAAM,cAAc,GAAG,OAAO,IAAI,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,EAAE,OAAO,CAAC;QAErE,IAAI,CAAC,cAAc,EAAE;YACnB,OAAO;SACR;QAED,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAC5B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,2BAA2B,CAAC;YACxD,cAAc;SACf,CAAC,CAAC;QAEL,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,iCAAiC;QACrC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE1E,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,aAAa,GAAG,aAAa,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,kCAAkC,CACtC,IAAI,GAAG,GAAG,EACV,QAAwB,MAAM;QAE9B,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,iBAAiB,CACpE,OAAO,EACP,IAAI,EACJ,KAAK,CACN,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,cAAc,GAAG,cAAc,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,oCAAoC;QACxC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,gBAAgB,GACpB,MAAM,uBAAA,IAAI,yCAAmB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAE7D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,wBAAwB,CAAC,EAC7B,UAAU,EACV,OAAO,MAC4B,EAAE;QACrC,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAChE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,yBAAyB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACxD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;YACrB,MAAM,IAAI,KAAK,CACb,wCAAwC,MAAM;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;iBACrB,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;SACH;IACH,CAAC;CACF;AAhSD,wCAgSC;wQA1NgB,eAAwB;IACrC,MAAM,EAAE,uBAAuB,EAAE,GAAG,eAAe;QACjD,CAAC,CAAC,EAAE,uBAAuB,EAAE,eAAe,EAAE;QAC9C,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAE5D,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7C,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE;QAC5B,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,OAAO;KACR;IAED,MAAM,QAAQ,GAAG,IAAI,wBAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,aAAa,CAAC;IAEhD,oDAAoD;IACpD,MAAM,MAAM,GAAmB;QAC7B,OAAO,EAAE,IAAA,sCAAmB,EAAC,OAAO,CAAC;KACtC,CAAC;IAEF,IAAI;QACF,uBAAA,IAAI,4BAAa,oBAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAA,CAAC;QACzC,uBAAA,IAAI,gCAAU,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;KACxE;IAAC,OAAO,KAAK,EAAE;QACd,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,2DAA2D;QAC3D,IACE,CAAC,CACC,KAAK,YAAY,KAAK;YACtB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAC9C,EACD;YACA,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;SAC1D;KACF;AACH,CAAC;IAGC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;AAC5E,CAAC;IAGC,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;IACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,GAC3B,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IACF,OAAO,IAAA,sCAAmB,EAAC,OAAO,CAAC,CAAC;AACtC,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport { convertHexToDecimal } from '@metamask/controller-utils';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n} from '@metamask/network-controller';\nimport {\n  StakeSdk,\n  StakingApiService,\n  type PooledStake,\n  type StakeSdkConfig,\n  type VaultData,\n  type VaultDailyApy,\n  type VaultApyAverages,\n} from '@metamask/stake-sdk';\n\nimport type {\n  RefreshPooledStakesOptions,\n  RefreshPooledStakingDataOptions,\n  RefreshStakingEligibilityOptions,\n} from './types';\n\nexport const controllerName = 'EarnController';\n\nexport type PooledStakingState = {\n  pooledStakes: PooledStake;\n  exchangeRate: string;\n  vaultMetadata: VaultData;\n  vaultDailyApys: VaultDailyApy[];\n  vaultApyAverages: VaultApyAverages;\n  isEligible: boolean;\n};\n\nexport type StablecoinLendingState = {\n  vaults: StablecoinVault[];\n};\n\nexport type StablecoinVault = {\n  symbol: string;\n  name: string;\n  chainId: number;\n  tokenAddress: string;\n  vaultAddress: string;\n  currentAPY: string;\n  supply: string;\n  liquidity: string;\n};\n\n/**\n * Metadata for the EarnController.\n */\nconst earnControllerMetadata: StateMetadata<EarnControllerState> = {\n  pooled_staking: {\n    persist: true,\n    anonymous: false,\n  },\n  stablecoin_lending: {\n    persist: true,\n    anonymous: false,\n  },\n  lastUpdated: {\n    persist: false,\n    anonymous: true,\n  },\n};\n\n// === State Types ===\nexport type EarnControllerState = {\n  pooled_staking: PooledStakingState;\n  stablecoin_lending?: StablecoinLendingState;\n  lastUpdated: number;\n};\n\n// === Default State ===\nconst DEFAULT_STABLECOIN_VAULT: StablecoinVault = {\n  symbol: '',\n  name: '',\n  chainId: 0,\n  tokenAddress: '',\n  vaultAddress: '',\n  currentAPY: '0',\n  supply: '0',\n  liquidity: '0',\n};\n\nconst DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES: VaultApyAverages = {\n  oneDay: '0',\n  oneWeek: '0',\n  oneMonth: '0',\n  threeMonths: '0',\n  sixMonths: '0',\n  oneYear: '0',\n};\n\n/**\n * Gets the default state for the EarnController.\n *\n * @returns The default EarnController state.\n */\nexport function getDefaultEarnControllerState(): EarnControllerState {\n  return {\n    pooled_staking: {\n      pooledStakes: {\n        account: '',\n        lifetimeRewards: '0',\n        assets: '0',\n        exitRequests: [],\n      },\n      exchangeRate: '1',\n      vaultMetadata: {\n        apy: '0',\n        capacity: '0',\n        feePercent: 0,\n        totalAssets: '0',\n        vaultAddress: '0x0000000000000000000000000000000000000000',\n      },\n      vaultDailyApys: [],\n      vaultApyAverages: DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES,\n      isEligible: false,\n    },\n    stablecoin_lending: {\n      vaults: [DEFAULT_STABLECOIN_VAULT],\n    },\n    lastUpdated: 0,\n  };\n}\n\n// === MESSENGER ===\n\n/**\n * The action which can be used to retrieve the state of the EarnController.\n */\nexport type EarnControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All actions that EarnController registers, to be called externally.\n */\nexport type EarnControllerActions = EarnControllerGetStateAction;\n\n/**\n * All actions that EarnController calls internally.\n */\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction;\n\n/**\n * The event that EarnController publishes when updating state.\n */\nexport type EarnControllerStateChangeEvent = ControllerStateChangeEvent<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All events that EarnController publishes, to be subscribed to externally.\n */\nexport type EarnControllerEvents = EarnControllerStateChangeEvent;\n\n/**\n * All events that EarnController subscribes to internally.\n */\nexport type AllowedEvents =\n  | AccountsControllerSelectedAccountChangeEvent\n  | NetworkControllerStateChangeEvent;\n\n/**\n * The messenger which is restricted to actions and events accessed by\n * EarnController.\n */\nexport type EarnControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  EarnControllerActions | AllowedActions,\n  EarnControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n// === CONTROLLER DEFINITION ===\n\n/**\n * EarnController manages DeFi earning opportunities across different protocols and chains.\n */\nexport class EarnController extends BaseController<\n  typeof controllerName,\n  EarnControllerState,\n  EarnControllerMessenger\n> {\n  #stakeSDK: StakeSdk | null = null;\n\n  #selectedNetworkClientId?: string;\n\n  readonly #stakingApiService: StakingApiService = new StakingApiService();\n\n  constructor({\n    messenger,\n    state = {},\n  }: {\n    messenger: EarnControllerMessenger;\n    state?: Partial<EarnControllerState>;\n  }) {\n    super({\n      name: controllerName,\n      metadata: earnControllerMetadata,\n      messenger,\n      state: {\n        ...getDefaultEarnControllerState(),\n        ...state,\n      },\n    });\n\n    this.#initializeSDK();\n    this.refreshPooledStakingData().catch(console.error);\n\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    this.#selectedNetworkClientId = selectedNetworkClientId;\n\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      (networkControllerState) => {\n        if (\n          networkControllerState.selectedNetworkClientId !==\n          this.#selectedNetworkClientId\n        ) {\n          this.#initializeSDK(networkControllerState.selectedNetworkClientId);\n          this.refreshPooledStakingVaultMetadata().catch(console.error);\n          this.refreshPooledStakingVaultDailyApys().catch(console.error);\n          this.refreshPooledStakingVaultApyAverages().catch(console.error);\n          this.refreshPooledStakes().catch(console.error);\n        }\n        this.#selectedNetworkClientId =\n          networkControllerState.selectedNetworkClientId;\n      },\n    );\n\n    // Listen for account changes\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedAccountChange',\n      (account) => {\n        const address = account?.address;\n        /**\n         * TEMP: There's a race condition where the account state isn't updated immediately.\n         * Until this has been fixed, we rely on the event payload for the latest account instead of #getCurrentAccount().\n         * Issue: https://github.com/MetaMask/accounts-planning/issues/887\n         */\n        this.refreshStakingEligibility({ address }).catch(console.error);\n        this.refreshPooledStakes({ address }).catch(console.error);\n      },\n    );\n  }\n\n  #initializeSDK(networkClientId?: string) {\n    const { selectedNetworkClientId } = networkClientId\n      ? { selectedNetworkClientId: networkClientId }\n      : this.messagingSystem.call('NetworkController:getState');\n\n    const networkClient = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    if (!networkClient?.provider) {\n      this.#stakeSDK = null;\n      return;\n    }\n\n    const provider = new Web3Provider(networkClient.provider);\n    const { chainId } = networkClient.configuration;\n\n    // Initialize appropriate contracts based on chainId\n    const config: StakeSdkConfig = {\n      chainId: convertHexToDecimal(chainId),\n    };\n\n    try {\n      this.#stakeSDK = StakeSdk.create(config);\n      this.#stakeSDK.pooledStakingContract.connectSignerOrProvider(provider);\n    } catch (error) {\n      this.#stakeSDK = null;\n      // Only log unexpected errors, not unsupported chain errors\n      if (\n        !(\n          error instanceof Error &&\n          error.message.includes('Unsupported chainId')\n        )\n      ) {\n        console.error('Stake SDK initialization failed:', error);\n      }\n    }\n  }\n\n  #getCurrentAccount() {\n    return this.messagingSystem.call('AccountsController:getSelectedAccount');\n  }\n\n  #getCurrentChainId(): number {\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId },\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n    return convertHexToDecimal(chainId);\n  }\n\n  /**\n   * Refreshes the pooled stakes data for the current account.\n   * Fetches updated stake information including lifetime rewards, assets, and exit requests\n   * from the staking API service and updates the state.\n   *\n   * @param options - Optional arguments\n   * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).\n   * @param [options.address] - The address to refresh pooled stakes for (optional).\n   * @returns A promise that resolves when the stakes data has been updated\n   */\n  async refreshPooledStakes({\n    resetCache = false,\n    address,\n  }: RefreshPooledStakesOptions = {}): Promise<void> {\n    const addressToUse = address ?? this.#getCurrentAccount()?.address;\n\n    if (!addressToUse) {\n      return;\n    }\n\n    const chainId = this.#getCurrentChainId();\n\n    const { accounts, exchangeRate } =\n      await this.#stakingApiService.getPooledStakes(\n        [addressToUse],\n        chainId,\n        resetCache,\n      );\n\n    this.update((state) => {\n      state.pooled_staking.pooledStakes = accounts[0];\n      state.pooled_staking.exchangeRate = exchangeRate;\n    });\n  }\n\n  /**\n   * Refreshes the staking eligibility status for the current account.\n   * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.\n   *\n   * @param options - Optional arguments\n   * @param [options.address] - Address to refresh staking eligibility for (optional).\n   * @returns A promise that resolves when the eligibility status has been updated\n   */\n  async refreshStakingEligibility({\n    address,\n  }: RefreshStakingEligibilityOptions = {}): Promise<void> {\n    const addressToCheck = address ?? this.#getCurrentAccount()?.address;\n\n    if (!addressToCheck) {\n      return;\n    }\n\n    const { eligible: isEligible } =\n      await this.#stakingApiService.getPooledStakingEligibility([\n        addressToCheck,\n      ]);\n\n    this.update((state) => {\n      state.pooled_staking.isEligible = isEligible;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault metadata for the current chain.\n   * Updates the vault metadata in the controller state including APY, capacity,\n   * fee percentage, total assets, and vault address.\n   *\n   * @returns A promise that resolves when the vault metadata has been updated\n   */\n  async refreshPooledStakingVaultMetadata(): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultMetadata = await this.#stakingApiService.getVaultData(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultMetadata = vaultMetadata;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault daily apys for the current chain.\n   * Updates the pooled staking vault daily apys controller state.\n   *\n   * @param days - The number of days to fetch pooled staking vault daily apys for (defaults to 365).\n   * @param order - The order in which to fetch pooled staking vault daily apys. Descending order fetches the latest N days (latest working backwards). Ascending order fetches the oldest N days (oldest working forwards) (defaults to 'desc').\n   * @returns A promise that resolves when the pooled staking vault daily apys have been updated.\n   */\n  async refreshPooledStakingVaultDailyApys(\n    days = 365,\n    order: 'asc' | 'desc' = 'desc',\n  ): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultDailyApys = await this.#stakingApiService.getVaultDailyApys(\n      chainId,\n      days,\n      order,\n    );\n\n    this.update((state) => {\n      state.pooled_staking.vaultDailyApys = vaultDailyApys;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault apy averages for the current chain.\n   * Updates the pooled staking vault apy averages controller state.\n   *\n   * @returns A promise that resolves when the pooled staking vault apy averages have been updated.\n   */\n  async refreshPooledStakingVaultApyAverages() {\n    const chainId = this.#getCurrentChainId();\n    const vaultApyAverages =\n      await this.#stakingApiService.getVaultApyAverages(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultApyAverages = vaultApyAverages;\n    });\n  }\n\n  /**\n   * Refreshes all pooled staking related data including stakes, eligibility, and vault data.\n   * This method allows partial success, meaning some data may update while other requests fail.\n   * All errors are collected and thrown as a single error message.\n   *\n   * @param options - Optional arguments\n   * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).\n   * @param [options.address] - The address to refresh pooled stakes for (optional).\n   * @returns A promise that resolves when all possible data has been updated\n   * @throws {Error} If any of the refresh operations fail, with concatenated error messages\n   */\n  async refreshPooledStakingData({\n    resetCache,\n    address,\n  }: RefreshPooledStakingDataOptions = {}): Promise<void> {\n    const errors: Error[] = [];\n\n    await Promise.all([\n      this.refreshPooledStakes({ resetCache, address }).catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshStakingEligibility({ address }).catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultMetadata().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultDailyApys().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultApyAverages().catch((error) => {\n        errors.push(error);\n      }),\n    ]);\n\n    if (errors.length > 0) {\n      throw new Error(\n        `Failed to refresh some staking data: ${errors\n          .map((e) => e.message)\n          .join(', ')}`,\n      );\n    }\n  }\n}\n"]}

package/dist/EarnController.d.cts

@@ -3,6 +3,7 @@ import type { ControllerGetStateAction, ControllerStateChangeEvent, RestrictedMe
 import { BaseController } from "@metamask/base-controller";
 import type { NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction, NetworkControllerStateChangeEvent } from "@metamask/network-controller";
 import { type PooledStake, type VaultData, type VaultDailyApy, type VaultApyAverages } from "@metamask/stake-sdk";
+import type { RefreshPooledStakesOptions, RefreshPooledStakingDataOptions, RefreshStakingEligibilityOptions } from "./types.cjs";
 export declare const controllerName = "EarnController";
 export type PooledStakingState = {
     pooledStakes: PooledStake;
@@ -79,17 +80,21 @@ export declare class EarnController extends BaseController<typeof controllerName
      * Fetches updated stake information including lifetime rewards, assets, and exit requests
      * from the staking API service and updates the state.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when the stakes data has been updated
      */
-    refreshPooledStakes(resetCache?: boolean): Promise<void>;
+    refreshPooledStakes({ resetCache, address, }?: RefreshPooledStakesOptions): Promise<void>;
     /**
      * Refreshes the staking eligibility status for the current account.
      * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.
      *
+     * @param options - Optional arguments
+     * @param [options.address] - Address to refresh staking eligibility for (optional).
      * @returns A promise that resolves when the eligibility status has been updated
      */
-    refreshStakingEligibility(): Promise<void>;
+    refreshStakingEligibility({ address, }?: RefreshStakingEligibilityOptions): Promise<void>;
     /**
      * Refreshes pooled staking vault metadata for the current chain.
      * Updates the vault metadata in the controller state including APY, capacity,
@@ -119,10 +124,12 @@ export declare class EarnController extends BaseController<typeof controllerName
      * This method allows partial success, meaning some data may update while other requests fail.
      * All errors are collected and thrown as a single error message.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when all possible data has been updated
      * @throws {Error} If any of the refresh operations fail, with concatenated error messages
      */
-    refreshPooledStakingData(resetCache?: boolean): Promise<void>;
+    refreshPooledStakingData({ resetCache, address, }?: RefreshPooledStakingDataOptions): Promise<void>;
 }
 //# sourceMappingURL=EarnController.d.cts.map

package/dist/EarnController.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"EarnController.d.cts","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAE3D,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAClC,qCAAqC;AACtC,OAAO,EAGL,KAAK,WAAW,EAEhB,KAAK,SAAS,EACd,KAAK,aAAa,EAClB,KAAK,gBAAgB,EACtB,4BAA4B;AAE7B,eAAO,MAAM,cAAc,mBAAmB,CAAC;AAE/C,MAAM,MAAM,kBAAkB,GAAG;IAC/B,YAAY,EAAE,WAAW,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,SAAS,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,EAAE,eAAe,EAAE,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAqBF,MAAM,MAAM,mBAAmB,GAAG;IAChC,cAAc,EAAE,kBAAkB,CAAC;IACnC,kBAAkB,CAAC,EAAE,sBAAsB,CAAC;IAC5C,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAuBF;;;;GAIG;AACH,wBAAgB,6BAA6B,IAAI,mBAAmB,CA0BnE;AAID;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,wBAAwB,CACjE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,4BAA4B,CAAC;AAEjE;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,0CAA0C,CAAC;AAE/C;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,0BAA0B,CACrE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,8BAA8B,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4CAA4C,GAC5C,iCAAiC,CAAC;AAEtC;;;GAGG;AACH,MAAM,MAAM,uBAAuB,GAAG,mBAAmB,CACvD,OAAO,cAAc,EACrB,qBAAqB,GAAG,cAAc,EACtC,oBAAoB,GAAG,aAAa,EACpC,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAIF;;GAEG;AACH,qBAAa,cAAe,SAAQ,cAAc,CAChD,OAAO,cAAc,EACrB,mBAAmB,EACnB,uBAAuB,CACxB;;gBAOa,EACV,SAAS,EACT,KAAU,GACX,EAAE;QACD,SAAS,EAAE,uBAAuB,CAAC;QACnC,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;KACtC;IAwGD;;;;;;;OAOG;IACG,mBAAmB,CAAC,UAAU,UAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB5D;;;;;OAKG;IACG,yBAAyB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBhD;;;;;;OAMG;IACG,iCAAiC,IAAI,OAAO,CAAC,IAAI,CAAC;IASxD;;;;;;;OAOG;IACG,kCAAkC,CACtC,IAAI,SAAM,EACV,KAAK,GAAE,KAAK,GAAG,MAAe,GAC7B,OAAO,CAAC,IAAI,CAAC;IAahB;;;;;OAKG;IACG,oCAAoC;IAU1C;;;;;;;;OAQG;IACG,wBAAwB,CAAC,UAAU,UAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;CA6BlE"}
+{"version":3,"file":"EarnController.d.cts","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAE3D,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAClC,qCAAqC;AACtC,OAAO,EAGL,KAAK,WAAW,EAEhB,KAAK,SAAS,EACd,KAAK,aAAa,EAClB,KAAK,gBAAgB,EACtB,4BAA4B;AAE7B,OAAO,KAAK,EACV,0BAA0B,EAC1B,+BAA+B,EAC/B,gCAAgC,EACjC,oBAAgB;AAEjB,eAAO,MAAM,cAAc,mBAAmB,CAAC;AAE/C,MAAM,MAAM,kBAAkB,GAAG;IAC/B,YAAY,EAAE,WAAW,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,SAAS,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,EAAE,eAAe,EAAE,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAqBF,MAAM,MAAM,mBAAmB,GAAG;IAChC,cAAc,EAAE,kBAAkB,CAAC;IACnC,kBAAkB,CAAC,EAAE,sBAAsB,CAAC;IAC5C,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAuBF;;;;GAIG;AACH,wBAAgB,6BAA6B,IAAI,mBAAmB,CA0BnE;AAID;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,wBAAwB,CACjE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,4BAA4B,CAAC;AAEjE;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,0CAA0C,CAAC;AAE/C;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,0BAA0B,CACrE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,8BAA8B,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4CAA4C,GAC5C,iCAAiC,CAAC;AAEtC;;;GAGG;AACH,MAAM,MAAM,uBAAuB,GAAG,mBAAmB,CACvD,OAAO,cAAc,EACrB,qBAAqB,GAAG,cAAc,EACtC,oBAAoB,GAAG,aAAa,EACpC,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAIF;;GAEG;AACH,qBAAa,cAAe,SAAQ,cAAc,CAChD,OAAO,cAAc,EACrB,mBAAmB,EACnB,uBAAuB,CACxB;;gBAOa,EACV,SAAS,EACT,KAAU,GACX,EAAE;QACD,SAAS,EAAE,uBAAuB,CAAC;QACnC,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;KACtC;IA8GD;;;;;;;;;OASG;IACG,mBAAmB,CAAC,EACxB,UAAkB,EAClB,OAAO,GACR,GAAE,0BAA+B,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBlD;;;;;;;OAOG;IACG,yBAAyB,CAAC,EAC9B,OAAO,GACR,GAAE,gCAAqC,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBxD;;;;;;OAMG;IACG,iCAAiC,IAAI,OAAO,CAAC,IAAI,CAAC;IASxD;;;;;;;OAOG;IACG,kCAAkC,CACtC,IAAI,SAAM,EACV,KAAK,GAAE,KAAK,GAAG,MAAe,GAC7B,OAAO,CAAC,IAAI,CAAC;IAahB;;;;;OAKG;IACG,oCAAoC;IAU1C;;;;;;;;;;OAUG;IACG,wBAAwB,CAAC,EAC7B,UAAU,EACV,OAAO,GACR,GAAE,+BAAoC,GAAG,OAAO,CAAC,IAAI,CAAC;CA6BxD"}

package/dist/EarnController.d.mts

@@ -3,6 +3,7 @@ import type { ControllerGetStateAction, ControllerStateChangeEvent, RestrictedMe
 import { BaseController } from "@metamask/base-controller";
 import type { NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction, NetworkControllerStateChangeEvent } from "@metamask/network-controller";
 import { type PooledStake, type VaultData, type VaultDailyApy, type VaultApyAverages } from "@metamask/stake-sdk";
+import type { RefreshPooledStakesOptions, RefreshPooledStakingDataOptions, RefreshStakingEligibilityOptions } from "./types.mjs";
 export declare const controllerName = "EarnController";
 export type PooledStakingState = {
     pooledStakes: PooledStake;
@@ -79,17 +80,21 @@ export declare class EarnController extends BaseController<typeof controllerName
      * Fetches updated stake information including lifetime rewards, assets, and exit requests
      * from the staking API service and updates the state.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when the stakes data has been updated
      */
-    refreshPooledStakes(resetCache?: boolean): Promise<void>;
+    refreshPooledStakes({ resetCache, address, }?: RefreshPooledStakesOptions): Promise<void>;
     /**
      * Refreshes the staking eligibility status for the current account.
      * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.
      *
+     * @param options - Optional arguments
+     * @param [options.address] - Address to refresh staking eligibility for (optional).
      * @returns A promise that resolves when the eligibility status has been updated
      */
-    refreshStakingEligibility(): Promise<void>;
+    refreshStakingEligibility({ address, }?: RefreshStakingEligibilityOptions): Promise<void>;
     /**
      * Refreshes pooled staking vault metadata for the current chain.
      * Updates the vault metadata in the controller state including APY, capacity,
@@ -119,10 +124,12 @@ export declare class EarnController extends BaseController<typeof controllerName
      * This method allows partial success, meaning some data may update while other requests fail.
      * All errors are collected and thrown as a single error message.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when all possible data has been updated
      * @throws {Error} If any of the refresh operations fail, with concatenated error messages
      */
-    refreshPooledStakingData(resetCache?: boolean): Promise<void>;
+    refreshPooledStakingData({ resetCache, address, }?: RefreshPooledStakingDataOptions): Promise<void>;
 }
 //# sourceMappingURL=EarnController.d.mts.map

package/dist/EarnController.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"EarnController.d.mts","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAE3D,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAClC,qCAAqC;AACtC,OAAO,EAGL,KAAK,WAAW,EAEhB,KAAK,SAAS,EACd,KAAK,aAAa,EAClB,KAAK,gBAAgB,EACtB,4BAA4B;AAE7B,eAAO,MAAM,cAAc,mBAAmB,CAAC;AAE/C,MAAM,MAAM,kBAAkB,GAAG;IAC/B,YAAY,EAAE,WAAW,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,SAAS,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,EAAE,eAAe,EAAE,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAqBF,MAAM,MAAM,mBAAmB,GAAG;IAChC,cAAc,EAAE,kBAAkB,CAAC;IACnC,kBAAkB,CAAC,EAAE,sBAAsB,CAAC;IAC5C,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAuBF;;;;GAIG;AACH,wBAAgB,6BAA6B,IAAI,mBAAmB,CA0BnE;AAID;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,wBAAwB,CACjE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,4BAA4B,CAAC;AAEjE;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,0CAA0C,CAAC;AAE/C;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,0BAA0B,CACrE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,8BAA8B,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4CAA4C,GAC5C,iCAAiC,CAAC;AAEtC;;;GAGG;AACH,MAAM,MAAM,uBAAuB,GAAG,mBAAmB,CACvD,OAAO,cAAc,EACrB,qBAAqB,GAAG,cAAc,EACtC,oBAAoB,GAAG,aAAa,EACpC,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAIF;;GAEG;AACH,qBAAa,cAAe,SAAQ,cAAc,CAChD,OAAO,cAAc,EACrB,mBAAmB,EACnB,uBAAuB,CACxB;;gBAOa,EACV,SAAS,EACT,KAAU,GACX,EAAE;QACD,SAAS,EAAE,uBAAuB,CAAC;QACnC,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;KACtC;IAwGD;;;;;;;OAOG;IACG,mBAAmB,CAAC,UAAU,UAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB5D;;;;;OAKG;IACG,yBAAyB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBhD;;;;;;OAMG;IACG,iCAAiC,IAAI,OAAO,CAAC,IAAI,CAAC;IASxD;;;;;;;OAOG;IACG,kCAAkC,CACtC,IAAI,SAAM,EACV,KAAK,GAAE,KAAK,GAAG,MAAe,GAC7B,OAAO,CAAC,IAAI,CAAC;IAahB;;;;;OAKG;IACG,oCAAoC;IAU1C;;;;;;;;OAQG;IACG,wBAAwB,CAAC,UAAU,UAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;CA6BlE"}
+{"version":3,"file":"EarnController.d.mts","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,0CAA0C,EAC1C,4CAA4C,EAC7C,sCAAsC;AACvC,OAAO,KAAK,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EAEpB,kCAAkC;AACnC,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAE3D,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAC/B,iCAAiC,EAClC,qCAAqC;AACtC,OAAO,EAGL,KAAK,WAAW,EAEhB,KAAK,SAAS,EACd,KAAK,aAAa,EAClB,KAAK,gBAAgB,EACtB,4BAA4B;AAE7B,OAAO,KAAK,EACV,0BAA0B,EAC1B,+BAA+B,EAC/B,gCAAgC,EACjC,oBAAgB;AAEjB,eAAO,MAAM,cAAc,mBAAmB,CAAC;AAE/C,MAAM,MAAM,kBAAkB,GAAG;IAC/B,YAAY,EAAE,WAAW,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,SAAS,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,EAAE,eAAe,EAAE,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAqBF,MAAM,MAAM,mBAAmB,GAAG;IAChC,cAAc,EAAE,kBAAkB,CAAC;IACnC,kBAAkB,CAAC,EAAE,sBAAsB,CAAC;IAC5C,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAuBF;;;;GAIG;AACH,wBAAgB,6BAA6B,IAAI,mBAAmB,CA0BnE;AAID;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,wBAAwB,CACjE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,4BAA4B,CAAC;AAEjE;;GAEG;AACH,MAAM,MAAM,cAAc,GACtB,2CAA2C,GAC3C,+BAA+B,GAC/B,0CAA0C,CAAC;AAE/C;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,0BAA0B,CACrE,OAAO,cAAc,EACrB,mBAAmB,CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,8BAA8B,CAAC;AAElE;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,4CAA4C,GAC5C,iCAAiC,CAAC;AAEtC;;;GAGG;AACH,MAAM,MAAM,uBAAuB,GAAG,mBAAmB,CACvD,OAAO,cAAc,EACrB,qBAAqB,GAAG,cAAc,EACtC,oBAAoB,GAAG,aAAa,EACpC,cAAc,CAAC,MAAM,CAAC,EACtB,aAAa,CAAC,MAAM,CAAC,CACtB,CAAC;AAIF;;GAEG;AACH,qBAAa,cAAe,SAAQ,cAAc,CAChD,OAAO,cAAc,EACrB,mBAAmB,EACnB,uBAAuB,CACxB;;gBAOa,EACV,SAAS,EACT,KAAU,GACX,EAAE;QACD,SAAS,EAAE,uBAAuB,CAAC;QACnC,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;KACtC;IA8GD;;;;;;;;;OASG;IACG,mBAAmB,CAAC,EACxB,UAAkB,EAClB,OAAO,GACR,GAAE,0BAA+B,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBlD;;;;;;;OAOG;IACG,yBAAyB,CAAC,EAC9B,OAAO,GACR,GAAE,gCAAqC,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBxD;;;;;;OAMG;IACG,iCAAiC,IAAI,OAAO,CAAC,IAAI,CAAC;IASxD;;;;;;;OAOG;IACG,kCAAkC,CACtC,IAAI,SAAM,EACV,KAAK,GAAE,KAAK,GAAG,MAAe,GAC7B,OAAO,CAAC,IAAI,CAAC;IAahB;;;;;OAKG;IACG,oCAAoC;IAU1C;;;;;;;;;;OAUG;IACG,wBAAwB,CAAC,EAC7B,UAAU,EACV,OAAO,GACR,GAAE,+BAAoC,GAAG,OAAO,CAAC,IAAI,CAAC;CA6BxD"}

package/dist/EarnController.mjs

@@ -118,9 +118,15 @@ export class EarnController extends BaseController {
             __classPrivateFieldSet(this, _EarnController_selectedNetworkClientId, networkControllerState.selectedNetworkClientId, "f");
         });
         // Listen for account changes
-        this.messagingSystem.subscribe('AccountsController:selectedAccountChange', () => {
-            this.refreshStakingEligibility().catch(console.error);
-            this.refreshPooledStakes().catch(console.error);
+        this.messagingSystem.subscribe('AccountsController:selectedAccountChange', (account) => {
+            const address = account?.address;
+            /**
+             * TEMP: There's a race condition where the account state isn't updated immediately.
+             * Until this has been fixed, we rely on the event payload for the latest account instead of #getCurrentAccount().
+             * Issue: https://github.com/MetaMask/accounts-planning/issues/887
+             */
+            this.refreshStakingEligibility({ address }).catch(console.error);
+            this.refreshPooledStakes({ address }).catch(console.error);
         });
     }
     /**
@@ -128,16 +134,18 @@ export class EarnController extends BaseController {
      * Fetches updated stake information including lifetime rewards, assets, and exit requests
      * from the staking API service and updates the state.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when the stakes data has been updated
      */
-    async refreshPooledStakes(resetCache = false) {
-        const currentAccount = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this);
-        if (!currentAccount?.address) {
+    async refreshPooledStakes({ resetCache = false, address, } = {}) {
+        const addressToUse = address ?? __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this)?.address;
+        if (!addressToUse) {
             return;
         }
         const chainId = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentChainId).call(this);
-        const { accounts, exchangeRate } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakes([currentAccount.address], chainId, resetCache);
+        const { accounts, exchangeRate } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakes([addressToUse], chainId, resetCache);
         this.update((state) => {
             state.pooled_staking.pooledStakes = accounts[0];
             state.pooled_staking.exchangeRate = exchangeRate;
@@ -147,15 +155,17 @@ export class EarnController extends BaseController {
      * Refreshes the staking eligibility status for the current account.
      * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.
      *
+     * @param options - Optional arguments
+     * @param [options.address] - Address to refresh staking eligibility for (optional).
      * @returns A promise that resolves when the eligibility status has been updated
      */
-    async refreshStakingEligibility() {
-        const currentAccount = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this);
-        if (!currentAccount?.address) {
+    async refreshStakingEligibility({ address, } = {}) {
+        const addressToCheck = address ?? __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this)?.address;
+        if (!addressToCheck) {
             return;
         }
         const { eligible: isEligible } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakingEligibility([
-            currentAccount.address,
+            addressToCheck,
         ]);
         this.update((state) => {
             state.pooled_staking.isEligible = isEligible;
@@ -208,17 +218,19 @@ export class EarnController extends BaseController {
      * This method allows partial success, meaning some data may update while other requests fail.
      * All errors are collected and thrown as a single error message.
      *
-     * @param resetCache - Control whether the BE cache should be invalidated.
+     * @param options - Optional arguments
+     * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).
+     * @param [options.address] - The address to refresh pooled stakes for (optional).
      * @returns A promise that resolves when all possible data has been updated
      * @throws {Error} If any of the refresh operations fail, with concatenated error messages
      */
-    async refreshPooledStakingData(resetCache = false) {
+    async refreshPooledStakingData({ resetCache, address, } = {}) {
         const errors = [];
         await Promise.all([
-            this.refreshPooledStakes(resetCache).catch((error) => {
+            this.refreshPooledStakes({ resetCache, address }).catch((error) => {
                 errors.push(error);
             }),
-            this.refreshStakingEligibility().catch((error) => {
+            this.refreshStakingEligibility({ address }).catch((error) => {
                 errors.push(error);
             }),
             this.refreshPooledStakingVaultMetadata().catch((error) => {

package/dist/EarnController.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"EarnController.mjs","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAWxD,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAC3D,OAAO,EAAE,mBAAmB,EAAE,mCAAmC;AAMjE,OAAO,EACL,QAAQ,EACR,iBAAiB,EAMlB,4BAA4B;AAE7B,MAAM,CAAC,MAAM,cAAc,GAAG,gBAAgB,CAAC;AA0B/C;;GAEG;AACH,MAAM,sBAAsB,GAAuC;IACjE,cAAc,EAAE;QACd,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,kBAAkB,EAAE;QAClB,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,WAAW,EAAE;QACX,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI;KAChB;CACF,CAAC;AASF,wBAAwB;AACxB,MAAM,wBAAwB,GAAoB;IAChD,MAAM,EAAE,EAAE;IACV,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,CAAC;IACV,YAAY,EAAE,EAAE;IAChB,YAAY,EAAE,EAAE;IAChB,UAAU,EAAE,GAAG;IACf,MAAM,EAAE,GAAG;IACX,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,MAAM,yCAAyC,GAAqB;IAClE,MAAM,EAAE,GAAG;IACX,OAAO,EAAE,GAAG;IACZ,QAAQ,EAAE,GAAG;IACb,WAAW,EAAE,GAAG;IAChB,SAAS,EAAE,GAAG;IACd,OAAO,EAAE,GAAG;CACb,CAAC;AAEF;;;;GAIG;AACH,MAAM,UAAU,6BAA6B;IAC3C,OAAO;QACL,cAAc,EAAE;YACd,YAAY,EAAE;gBACZ,OAAO,EAAE,EAAE;gBACX,eAAe,EAAE,GAAG;gBACpB,MAAM,EAAE,GAAG;gBACX,YAAY,EAAE,EAAE;aACjB;YACD,YAAY,EAAE,GAAG;YACjB,aAAa,EAAE;gBACb,GAAG,EAAE,GAAG;gBACR,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,4CAA4C;aAC3D;YACD,cAAc,EAAE,EAAE;YAClB,gBAAgB,EAAE,yCAAyC;YAC3D,UAAU,EAAE,KAAK;SAClB;QACD,kBAAkB,EAAE;YAClB,MAAM,EAAE,CAAC,wBAAwB,CAAC;SACnC;QACD,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AAyDD,gCAAgC;AAEhC;;GAEG;AACH,MAAM,OAAO,cAAe,SAAQ,cAInC;IAOC,YAAY,EACV,SAAS,EACT,KAAK,GAAG,EAAE,GAIX;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,sBAAsB;YAChC,SAAS;YACT,KAAK,EAAE;gBACL,GAAG,6BAA6B,EAAE;gBAClC,GAAG,KAAK;aACT;SACF,CAAC,CAAC;;QArBL,mCAA6B,IAAI,EAAC;QAElC,0DAAkC;QAEzB,4CAAwC,IAAI,iBAAiB,EAAE,EAAC;QAmBvE,uBAAA,IAAI,gEAAe,MAAnB,IAAI,CAAiB,CAAC;QACtB,IAAI,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAErD,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;QACF,uBAAA,IAAI,2CAA4B,uBAAuB,MAAA,CAAC;QAExD,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,CAAC,sBAAsB,EAAE,EAAE;YACzB,IACE,sBAAsB,CAAC,uBAAuB;gBAC9C,uBAAA,IAAI,+CAAyB,EAC7B;gBACA,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,sBAAsB,CAAC,uBAAuB,CAAC,CAAC;gBACpE,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC9D,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC/D,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjE,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;aACjD;YACD,uBAAA,IAAI,2CACF,sBAAsB,CAAC,uBAAuB,MAAA,CAAC;QACnD,CAAC,CACF,CAAC;QAEF,6BAA6B;QAC7B,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,0CAA0C,EAC1C,GAAG,EAAE;YACH,IAAI,CAAC,yBAAyB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtD,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAClD,CAAC,CACF,CAAC;IACJ,CAAC;IA2DD;;;;;;;OAOG;IACH,KAAK,CAAC,mBAAmB,CAAC,UAAU,GAAG,KAAK;QAC1C,MAAM,cAAc,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QACjD,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;YAC5B,OAAO;SACR;QAED,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1C,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAC9B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,eAAe,CAC3C,CAAC,cAAc,CAAC,OAAO,CAAC,EACxB,OAAO,EACP,UAAU,CACX,CAAC;QAEJ,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAChD,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,YAAY,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,yBAAyB;QAC7B,MAAM,cAAc,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QACjD,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;YAC5B,OAAO;SACR;QAED,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAC5B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,2BAA2B,CAAC;YACxD,cAAc,CAAC,OAAO;SACvB,CAAC,CAAC;QAEL,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,iCAAiC;QACrC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE1E,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,aAAa,GAAG,aAAa,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,kCAAkC,CACtC,IAAI,GAAG,GAAG,EACV,QAAwB,MAAM;QAE9B,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,iBAAiB,CACpE,OAAO,EACP,IAAI,EACJ,KAAK,CACN,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,cAAc,GAAG,cAAc,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,oCAAoC;QACxC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,gBAAgB,GACpB,MAAM,uBAAA,IAAI,yCAAmB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAE7D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,wBAAwB,CAAC,UAAU,GAAG,KAAK;QAC/C,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACnD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,yBAAyB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC/C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACxD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;YACrB,MAAM,IAAI,KAAK,CACb,wCAAwC,MAAM;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;iBACrB,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;SACH;IACH,CAAC;CACF;wQA1MgB,eAAwB;IACrC,MAAM,EAAE,uBAAuB,EAAE,GAAG,eAAe;QACjD,CAAC,CAAC,EAAE,uBAAuB,EAAE,eAAe,EAAE;QAC9C,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAE5D,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7C,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE;QAC5B,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,OAAO;KACR;IAED,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,aAAa,CAAC;IAEhD,oDAAoD;IACpD,MAAM,MAAM,GAAmB;QAC7B,OAAO,EAAE,mBAAmB,CAAC,OAAO,CAAC;KACtC,CAAC;IAEF,IAAI;QACF,uBAAA,IAAI,4BAAa,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAA,CAAC;QACzC,uBAAA,IAAI,gCAAU,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;KACxE;IAAC,OAAO,KAAK,EAAE;QACd,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,2DAA2D;QAC3D,IACE,CAAC,CACC,KAAK,YAAY,KAAK;YACtB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAC9C,EACD;YACA,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;SAC1D;KACF;AACH,CAAC;IAGC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;AAC5E,CAAC;IAGC,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;IACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,GAC3B,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IACF,OAAO,mBAAmB,CAAC,OAAO,CAAC,CAAC;AACtC,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport { convertHexToDecimal } from '@metamask/controller-utils';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n} from '@metamask/network-controller';\nimport {\n  StakeSdk,\n  StakingApiService,\n  type PooledStake,\n  type StakeSdkConfig,\n  type VaultData,\n  type VaultDailyApy,\n  type VaultApyAverages,\n} from '@metamask/stake-sdk';\n\nexport const controllerName = 'EarnController';\n\nexport type PooledStakingState = {\n  pooledStakes: PooledStake;\n  exchangeRate: string;\n  vaultMetadata: VaultData;\n  vaultDailyApys: VaultDailyApy[];\n  vaultApyAverages: VaultApyAverages;\n  isEligible: boolean;\n};\n\nexport type StablecoinLendingState = {\n  vaults: StablecoinVault[];\n};\n\nexport type StablecoinVault = {\n  symbol: string;\n  name: string;\n  chainId: number;\n  tokenAddress: string;\n  vaultAddress: string;\n  currentAPY: string;\n  supply: string;\n  liquidity: string;\n};\n\n/**\n * Metadata for the EarnController.\n */\nconst earnControllerMetadata: StateMetadata<EarnControllerState> = {\n  pooled_staking: {\n    persist: true,\n    anonymous: false,\n  },\n  stablecoin_lending: {\n    persist: true,\n    anonymous: false,\n  },\n  lastUpdated: {\n    persist: false,\n    anonymous: true,\n  },\n};\n\n// === State Types ===\nexport type EarnControllerState = {\n  pooled_staking: PooledStakingState;\n  stablecoin_lending?: StablecoinLendingState;\n  lastUpdated: number;\n};\n\n// === Default State ===\nconst DEFAULT_STABLECOIN_VAULT: StablecoinVault = {\n  symbol: '',\n  name: '',\n  chainId: 0,\n  tokenAddress: '',\n  vaultAddress: '',\n  currentAPY: '0',\n  supply: '0',\n  liquidity: '0',\n};\n\nconst DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES: VaultApyAverages = {\n  oneDay: '0',\n  oneWeek: '0',\n  oneMonth: '0',\n  threeMonths: '0',\n  sixMonths: '0',\n  oneYear: '0',\n};\n\n/**\n * Gets the default state for the EarnController.\n *\n * @returns The default EarnController state.\n */\nexport function getDefaultEarnControllerState(): EarnControllerState {\n  return {\n    pooled_staking: {\n      pooledStakes: {\n        account: '',\n        lifetimeRewards: '0',\n        assets: '0',\n        exitRequests: [],\n      },\n      exchangeRate: '1',\n      vaultMetadata: {\n        apy: '0',\n        capacity: '0',\n        feePercent: 0,\n        totalAssets: '0',\n        vaultAddress: '0x0000000000000000000000000000000000000000',\n      },\n      vaultDailyApys: [],\n      vaultApyAverages: DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES,\n      isEligible: false,\n    },\n    stablecoin_lending: {\n      vaults: [DEFAULT_STABLECOIN_VAULT],\n    },\n    lastUpdated: 0,\n  };\n}\n\n// === MESSENGER ===\n\n/**\n * The action which can be used to retrieve the state of the EarnController.\n */\nexport type EarnControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All actions that EarnController registers, to be called externally.\n */\nexport type EarnControllerActions = EarnControllerGetStateAction;\n\n/**\n * All actions that EarnController calls internally.\n */\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction;\n\n/**\n * The event that EarnController publishes when updating state.\n */\nexport type EarnControllerStateChangeEvent = ControllerStateChangeEvent<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All events that EarnController publishes, to be subscribed to externally.\n */\nexport type EarnControllerEvents = EarnControllerStateChangeEvent;\n\n/**\n * All events that EarnController subscribes to internally.\n */\nexport type AllowedEvents =\n  | AccountsControllerSelectedAccountChangeEvent\n  | NetworkControllerStateChangeEvent;\n\n/**\n * The messenger which is restricted to actions and events accessed by\n * EarnController.\n */\nexport type EarnControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  EarnControllerActions | AllowedActions,\n  EarnControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n// === CONTROLLER DEFINITION ===\n\n/**\n * EarnController manages DeFi earning opportunities across different protocols and chains.\n */\nexport class EarnController extends BaseController<\n  typeof controllerName,\n  EarnControllerState,\n  EarnControllerMessenger\n> {\n  #stakeSDK: StakeSdk | null = null;\n\n  #selectedNetworkClientId?: string;\n\n  readonly #stakingApiService: StakingApiService = new StakingApiService();\n\n  constructor({\n    messenger,\n    state = {},\n  }: {\n    messenger: EarnControllerMessenger;\n    state?: Partial<EarnControllerState>;\n  }) {\n    super({\n      name: controllerName,\n      metadata: earnControllerMetadata,\n      messenger,\n      state: {\n        ...getDefaultEarnControllerState(),\n        ...state,\n      },\n    });\n\n    this.#initializeSDK();\n    this.refreshPooledStakingData().catch(console.error);\n\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    this.#selectedNetworkClientId = selectedNetworkClientId;\n\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      (networkControllerState) => {\n        if (\n          networkControllerState.selectedNetworkClientId !==\n          this.#selectedNetworkClientId\n        ) {\n          this.#initializeSDK(networkControllerState.selectedNetworkClientId);\n          this.refreshPooledStakingVaultMetadata().catch(console.error);\n          this.refreshPooledStakingVaultDailyApys().catch(console.error);\n          this.refreshPooledStakingVaultApyAverages().catch(console.error);\n          this.refreshPooledStakes().catch(console.error);\n        }\n        this.#selectedNetworkClientId =\n          networkControllerState.selectedNetworkClientId;\n      },\n    );\n\n    // Listen for account changes\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedAccountChange',\n      () => {\n        this.refreshStakingEligibility().catch(console.error);\n        this.refreshPooledStakes().catch(console.error);\n      },\n    );\n  }\n\n  #initializeSDK(networkClientId?: string) {\n    const { selectedNetworkClientId } = networkClientId\n      ? { selectedNetworkClientId: networkClientId }\n      : this.messagingSystem.call('NetworkController:getState');\n\n    const networkClient = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    if (!networkClient?.provider) {\n      this.#stakeSDK = null;\n      return;\n    }\n\n    const provider = new Web3Provider(networkClient.provider);\n    const { chainId } = networkClient.configuration;\n\n    // Initialize appropriate contracts based on chainId\n    const config: StakeSdkConfig = {\n      chainId: convertHexToDecimal(chainId),\n    };\n\n    try {\n      this.#stakeSDK = StakeSdk.create(config);\n      this.#stakeSDK.pooledStakingContract.connectSignerOrProvider(provider);\n    } catch (error) {\n      this.#stakeSDK = null;\n      // Only log unexpected errors, not unsupported chain errors\n      if (\n        !(\n          error instanceof Error &&\n          error.message.includes('Unsupported chainId')\n        )\n      ) {\n        console.error('Stake SDK initialization failed:', error);\n      }\n    }\n  }\n\n  #getCurrentAccount() {\n    return this.messagingSystem.call('AccountsController:getSelectedAccount');\n  }\n\n  #getCurrentChainId(): number {\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId },\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n    return convertHexToDecimal(chainId);\n  }\n\n  /**\n   * Refreshes the pooled stakes data for the current account.\n   * Fetches updated stake information including lifetime rewards, assets, and exit requests\n   * from the staking API service and updates the state.\n   *\n   * @param resetCache - Control whether the BE cache should be invalidated.\n   * @returns A promise that resolves when the stakes data has been updated\n   */\n  async refreshPooledStakes(resetCache = false): Promise<void> {\n    const currentAccount = this.#getCurrentAccount();\n    if (!currentAccount?.address) {\n      return;\n    }\n\n    const chainId = this.#getCurrentChainId();\n\n    const { accounts, exchangeRate } =\n      await this.#stakingApiService.getPooledStakes(\n        [currentAccount.address],\n        chainId,\n        resetCache,\n      );\n\n    this.update((state) => {\n      state.pooled_staking.pooledStakes = accounts[0];\n      state.pooled_staking.exchangeRate = exchangeRate;\n    });\n  }\n\n  /**\n   * Refreshes the staking eligibility status for the current account.\n   * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.\n   *\n   * @returns A promise that resolves when the eligibility status has been updated\n   */\n  async refreshStakingEligibility(): Promise<void> {\n    const currentAccount = this.#getCurrentAccount();\n    if (!currentAccount?.address) {\n      return;\n    }\n\n    const { eligible: isEligible } =\n      await this.#stakingApiService.getPooledStakingEligibility([\n        currentAccount.address,\n      ]);\n\n    this.update((state) => {\n      state.pooled_staking.isEligible = isEligible;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault metadata for the current chain.\n   * Updates the vault metadata in the controller state including APY, capacity,\n   * fee percentage, total assets, and vault address.\n   *\n   * @returns A promise that resolves when the vault metadata has been updated\n   */\n  async refreshPooledStakingVaultMetadata(): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultMetadata = await this.#stakingApiService.getVaultData(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultMetadata = vaultMetadata;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault daily apys for the current chain.\n   * Updates the pooled staking vault daily apys controller state.\n   *\n   * @param days - The number of days to fetch pooled staking vault daily apys for (defaults to 365).\n   * @param order - The order in which to fetch pooled staking vault daily apys. Descending order fetches the latest N days (latest working backwards). Ascending order fetches the oldest N days (oldest working forwards) (defaults to 'desc').\n   * @returns A promise that resolves when the pooled staking vault daily apys have been updated.\n   */\n  async refreshPooledStakingVaultDailyApys(\n    days = 365,\n    order: 'asc' | 'desc' = 'desc',\n  ): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultDailyApys = await this.#stakingApiService.getVaultDailyApys(\n      chainId,\n      days,\n      order,\n    );\n\n    this.update((state) => {\n      state.pooled_staking.vaultDailyApys = vaultDailyApys;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault apy averages for the current chain.\n   * Updates the pooled staking vault apy averages controller state.\n   *\n   * @returns A promise that resolves when the pooled staking vault apy averages have been updated.\n   */\n  async refreshPooledStakingVaultApyAverages() {\n    const chainId = this.#getCurrentChainId();\n    const vaultApyAverages =\n      await this.#stakingApiService.getVaultApyAverages(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultApyAverages = vaultApyAverages;\n    });\n  }\n\n  /**\n   * Refreshes all pooled staking related data including stakes, eligibility, and vault data.\n   * This method allows partial success, meaning some data may update while other requests fail.\n   * All errors are collected and thrown as a single error message.\n   *\n   * @param resetCache - Control whether the BE cache should be invalidated.\n   * @returns A promise that resolves when all possible data has been updated\n   * @throws {Error} If any of the refresh operations fail, with concatenated error messages\n   */\n  async refreshPooledStakingData(resetCache = false): Promise<void> {\n    const errors: Error[] = [];\n\n    await Promise.all([\n      this.refreshPooledStakes(resetCache).catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshStakingEligibility().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultMetadata().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultDailyApys().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultApyAverages().catch((error) => {\n        errors.push(error);\n      }),\n    ]);\n\n    if (errors.length > 0) {\n      throw new Error(\n        `Failed to refresh some staking data: ${errors\n          .map((e) => e.message)\n          .join(', ')}`,\n      );\n    }\n  }\n}\n"]}
+{"version":3,"file":"EarnController.mjs","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAWxD,OAAO,EAAE,cAAc,EAAE,kCAAkC;AAC3D,OAAO,EAAE,mBAAmB,EAAE,mCAAmC;AAMjE,OAAO,EACL,QAAQ,EACR,iBAAiB,EAMlB,4BAA4B;AAQ7B,MAAM,CAAC,MAAM,cAAc,GAAG,gBAAgB,CAAC;AA0B/C;;GAEG;AACH,MAAM,sBAAsB,GAAuC;IACjE,cAAc,EAAE;QACd,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,kBAAkB,EAAE;QAClB,OAAO,EAAE,IAAI;QACb,SAAS,EAAE,KAAK;KACjB;IACD,WAAW,EAAE;QACX,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI;KAChB;CACF,CAAC;AASF,wBAAwB;AACxB,MAAM,wBAAwB,GAAoB;IAChD,MAAM,EAAE,EAAE;IACV,IAAI,EAAE,EAAE;IACR,OAAO,EAAE,CAAC;IACV,YAAY,EAAE,EAAE;IAChB,YAAY,EAAE,EAAE;IAChB,UAAU,EAAE,GAAG;IACf,MAAM,EAAE,GAAG;IACX,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,MAAM,yCAAyC,GAAqB;IAClE,MAAM,EAAE,GAAG;IACX,OAAO,EAAE,GAAG;IACZ,QAAQ,EAAE,GAAG;IACb,WAAW,EAAE,GAAG;IAChB,SAAS,EAAE,GAAG;IACd,OAAO,EAAE,GAAG;CACb,CAAC;AAEF;;;;GAIG;AACH,MAAM,UAAU,6BAA6B;IAC3C,OAAO;QACL,cAAc,EAAE;YACd,YAAY,EAAE;gBACZ,OAAO,EAAE,EAAE;gBACX,eAAe,EAAE,GAAG;gBACpB,MAAM,EAAE,GAAG;gBACX,YAAY,EAAE,EAAE;aACjB;YACD,YAAY,EAAE,GAAG;YACjB,aAAa,EAAE;gBACb,GAAG,EAAE,GAAG;gBACR,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,4CAA4C;aAC3D;YACD,cAAc,EAAE,EAAE;YAClB,gBAAgB,EAAE,yCAAyC;YAC3D,UAAU,EAAE,KAAK;SAClB;QACD,kBAAkB,EAAE;YAClB,MAAM,EAAE,CAAC,wBAAwB,CAAC;SACnC;QACD,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AAyDD,gCAAgC;AAEhC;;GAEG;AACH,MAAM,OAAO,cAAe,SAAQ,cAInC;IAOC,YAAY,EACV,SAAS,EACT,KAAK,GAAG,EAAE,GAIX;QACC,KAAK,CAAC;YACJ,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,sBAAsB;YAChC,SAAS;YACT,KAAK,EAAE;gBACL,GAAG,6BAA6B,EAAE;gBAClC,GAAG,KAAK;aACT;SACF,CAAC,CAAC;;QArBL,mCAA6B,IAAI,EAAC;QAElC,0DAAkC;QAEzB,4CAAwC,IAAI,iBAAiB,EAAE,EAAC;QAmBvE,uBAAA,IAAI,gEAAe,MAAnB,IAAI,CAAiB,CAAC;QACtB,IAAI,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAErD,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;QACF,uBAAA,IAAI,2CAA4B,uBAAuB,MAAA,CAAC;QAExD,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,+BAA+B,EAC/B,CAAC,sBAAsB,EAAE,EAAE;YACzB,IACE,sBAAsB,CAAC,uBAAuB;gBAC9C,uBAAA,IAAI,+CAAyB,EAC7B;gBACA,uBAAA,IAAI,gEAAe,MAAnB,IAAI,EAAgB,sBAAsB,CAAC,uBAAuB,CAAC,CAAC;gBACpE,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC9D,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC/D,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjE,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;aACjD;YACD,uBAAA,IAAI,2CACF,sBAAsB,CAAC,uBAAuB,MAAA,CAAC;QACnD,CAAC,CACF,CAAC;QAEF,6BAA6B;QAC7B,IAAI,CAAC,eAAe,CAAC,SAAS,CAC5B,0CAA0C,EAC1C,CAAC,OAAO,EAAE,EAAE;YACV,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,CAAC;YACjC;;;;eAIG;YACH,IAAI,CAAC,yBAAyB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACjE,IAAI,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC,CACF,CAAC;IACJ,CAAC;IA2DD;;;;;;;;;OASG;IACH,KAAK,CAAC,mBAAmB,CAAC,EACxB,UAAU,GAAG,KAAK,EAClB,OAAO,MACuB,EAAE;QAChC,MAAM,YAAY,GAAG,OAAO,IAAI,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,EAAE,OAAO,CAAC;QAEnE,IAAI,CAAC,YAAY,EAAE;YACjB,OAAO;SACR;QAED,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1C,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAC9B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,eAAe,CAC3C,CAAC,YAAY,CAAC,EACd,OAAO,EACP,UAAU,CACX,CAAC;QAEJ,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAChD,KAAK,CAAC,cAAc,CAAC,YAAY,GAAG,YAAY,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,yBAAyB,CAAC,EAC9B,OAAO,MAC6B,EAAE;QACtC,MAAM,cAAc,GAAG,OAAO,IAAI,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,EAAE,OAAO,CAAC;QAErE,IAAI,CAAC,cAAc,EAAE;YACnB,OAAO;SACR;QAED,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAC5B,MAAM,uBAAA,IAAI,yCAAmB,CAAC,2BAA2B,CAAC;YACxD,cAAc;SACf,CAAC,CAAC;QAEL,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,iCAAiC;QACrC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,aAAa,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE1E,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,aAAa,GAAG,aAAa,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,kCAAkC,CACtC,IAAI,GAAG,GAAG,EACV,QAAwB,MAAM;QAE9B,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,iBAAiB,CACpE,OAAO,EACP,IAAI,EACJ,KAAK,CACN,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,cAAc,GAAG,cAAc,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,oCAAoC;QACxC,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,gBAAgB,GACpB,MAAM,uBAAA,IAAI,yCAAmB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAE7D,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,wBAAwB,CAAC,EAC7B,UAAU,EACV,OAAO,MAC4B,EAAE;QACrC,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAChE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,yBAAyB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,iCAAiC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,kCAAkC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACxD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,IAAI,CAAC,oCAAoC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;YACrB,MAAM,IAAI,KAAK,CACb,wCAAwC,MAAM;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;iBACrB,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;SACH;IACH,CAAC;CACF;wQA1NgB,eAAwB;IACrC,MAAM,EAAE,uBAAuB,EAAE,GAAG,eAAe;QACjD,CAAC,CAAC,EAAE,uBAAuB,EAAE,eAAe,EAAE;QAC9C,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAE5D,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7C,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE;QAC5B,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,OAAO;KACR;IAED,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,aAAa,CAAC;IAEhD,oDAAoD;IACpD,MAAM,MAAM,GAAmB;QAC7B,OAAO,EAAE,mBAAmB,CAAC,OAAO,CAAC;KACtC,CAAC;IAEF,IAAI;QACF,uBAAA,IAAI,4BAAa,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAA,CAAC;QACzC,uBAAA,IAAI,gCAAU,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;KACxE;IAAC,OAAO,KAAK,EAAE;QACd,uBAAA,IAAI,4BAAa,IAAI,MAAA,CAAC;QACtB,2DAA2D;QAC3D,IACE,CAAC,CACC,KAAK,YAAY,KAAK;YACtB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAC9C,EACD;YACA,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;SAC1D;KACF;AACH,CAAC;IAGC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;AAC5E,CAAC;IAGC,MAAM,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3D,4BAA4B,CAC7B,CAAC;IACF,MAAM,EACJ,aAAa,EAAE,EAAE,OAAO,EAAE,GAC3B,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAC3B,wCAAwC,EACxC,uBAAuB,CACxB,CAAC;IACF,OAAO,mBAAmB,CAAC,OAAO,CAAC,CAAC;AACtC,CAAC","sourcesContent":["import { Web3Provider } from '@ethersproject/providers';\nimport type {\n  AccountsControllerGetSelectedAccountAction,\n  AccountsControllerSelectedAccountChangeEvent,\n} from '@metamask/accounts-controller';\nimport type {\n  ControllerGetStateAction,\n  ControllerStateChangeEvent,\n  RestrictedMessenger,\n  StateMetadata,\n} from '@metamask/base-controller';\nimport { BaseController } from '@metamask/base-controller';\nimport { convertHexToDecimal } from '@metamask/controller-utils';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n  NetworkControllerStateChangeEvent,\n} from '@metamask/network-controller';\nimport {\n  StakeSdk,\n  StakingApiService,\n  type PooledStake,\n  type StakeSdkConfig,\n  type VaultData,\n  type VaultDailyApy,\n  type VaultApyAverages,\n} from '@metamask/stake-sdk';\n\nimport type {\n  RefreshPooledStakesOptions,\n  RefreshPooledStakingDataOptions,\n  RefreshStakingEligibilityOptions,\n} from './types';\n\nexport const controllerName = 'EarnController';\n\nexport type PooledStakingState = {\n  pooledStakes: PooledStake;\n  exchangeRate: string;\n  vaultMetadata: VaultData;\n  vaultDailyApys: VaultDailyApy[];\n  vaultApyAverages: VaultApyAverages;\n  isEligible: boolean;\n};\n\nexport type StablecoinLendingState = {\n  vaults: StablecoinVault[];\n};\n\nexport type StablecoinVault = {\n  symbol: string;\n  name: string;\n  chainId: number;\n  tokenAddress: string;\n  vaultAddress: string;\n  currentAPY: string;\n  supply: string;\n  liquidity: string;\n};\n\n/**\n * Metadata for the EarnController.\n */\nconst earnControllerMetadata: StateMetadata<EarnControllerState> = {\n  pooled_staking: {\n    persist: true,\n    anonymous: false,\n  },\n  stablecoin_lending: {\n    persist: true,\n    anonymous: false,\n  },\n  lastUpdated: {\n    persist: false,\n    anonymous: true,\n  },\n};\n\n// === State Types ===\nexport type EarnControllerState = {\n  pooled_staking: PooledStakingState;\n  stablecoin_lending?: StablecoinLendingState;\n  lastUpdated: number;\n};\n\n// === Default State ===\nconst DEFAULT_STABLECOIN_VAULT: StablecoinVault = {\n  symbol: '',\n  name: '',\n  chainId: 0,\n  tokenAddress: '',\n  vaultAddress: '',\n  currentAPY: '0',\n  supply: '0',\n  liquidity: '0',\n};\n\nconst DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES: VaultApyAverages = {\n  oneDay: '0',\n  oneWeek: '0',\n  oneMonth: '0',\n  threeMonths: '0',\n  sixMonths: '0',\n  oneYear: '0',\n};\n\n/**\n * Gets the default state for the EarnController.\n *\n * @returns The default EarnController state.\n */\nexport function getDefaultEarnControllerState(): EarnControllerState {\n  return {\n    pooled_staking: {\n      pooledStakes: {\n        account: '',\n        lifetimeRewards: '0',\n        assets: '0',\n        exitRequests: [],\n      },\n      exchangeRate: '1',\n      vaultMetadata: {\n        apy: '0',\n        capacity: '0',\n        feePercent: 0,\n        totalAssets: '0',\n        vaultAddress: '0x0000000000000000000000000000000000000000',\n      },\n      vaultDailyApys: [],\n      vaultApyAverages: DEFAULT_POOLED_STAKING_VAULT_APY_AVERAGES,\n      isEligible: false,\n    },\n    stablecoin_lending: {\n      vaults: [DEFAULT_STABLECOIN_VAULT],\n    },\n    lastUpdated: 0,\n  };\n}\n\n// === MESSENGER ===\n\n/**\n * The action which can be used to retrieve the state of the EarnController.\n */\nexport type EarnControllerGetStateAction = ControllerGetStateAction<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All actions that EarnController registers, to be called externally.\n */\nexport type EarnControllerActions = EarnControllerGetStateAction;\n\n/**\n * All actions that EarnController calls internally.\n */\nexport type AllowedActions =\n  | NetworkControllerGetNetworkClientByIdAction\n  | NetworkControllerGetStateAction\n  | AccountsControllerGetSelectedAccountAction;\n\n/**\n * The event that EarnController publishes when updating state.\n */\nexport type EarnControllerStateChangeEvent = ControllerStateChangeEvent<\n  typeof controllerName,\n  EarnControllerState\n>;\n\n/**\n * All events that EarnController publishes, to be subscribed to externally.\n */\nexport type EarnControllerEvents = EarnControllerStateChangeEvent;\n\n/**\n * All events that EarnController subscribes to internally.\n */\nexport type AllowedEvents =\n  | AccountsControllerSelectedAccountChangeEvent\n  | NetworkControllerStateChangeEvent;\n\n/**\n * The messenger which is restricted to actions and events accessed by\n * EarnController.\n */\nexport type EarnControllerMessenger = RestrictedMessenger<\n  typeof controllerName,\n  EarnControllerActions | AllowedActions,\n  EarnControllerEvents | AllowedEvents,\n  AllowedActions['type'],\n  AllowedEvents['type']\n>;\n\n// === CONTROLLER DEFINITION ===\n\n/**\n * EarnController manages DeFi earning opportunities across different protocols and chains.\n */\nexport class EarnController extends BaseController<\n  typeof controllerName,\n  EarnControllerState,\n  EarnControllerMessenger\n> {\n  #stakeSDK: StakeSdk | null = null;\n\n  #selectedNetworkClientId?: string;\n\n  readonly #stakingApiService: StakingApiService = new StakingApiService();\n\n  constructor({\n    messenger,\n    state = {},\n  }: {\n    messenger: EarnControllerMessenger;\n    state?: Partial<EarnControllerState>;\n  }) {\n    super({\n      name: controllerName,\n      metadata: earnControllerMetadata,\n      messenger,\n      state: {\n        ...getDefaultEarnControllerState(),\n        ...state,\n      },\n    });\n\n    this.#initializeSDK();\n    this.refreshPooledStakingData().catch(console.error);\n\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    this.#selectedNetworkClientId = selectedNetworkClientId;\n\n    this.messagingSystem.subscribe(\n      'NetworkController:stateChange',\n      (networkControllerState) => {\n        if (\n          networkControllerState.selectedNetworkClientId !==\n          this.#selectedNetworkClientId\n        ) {\n          this.#initializeSDK(networkControllerState.selectedNetworkClientId);\n          this.refreshPooledStakingVaultMetadata().catch(console.error);\n          this.refreshPooledStakingVaultDailyApys().catch(console.error);\n          this.refreshPooledStakingVaultApyAverages().catch(console.error);\n          this.refreshPooledStakes().catch(console.error);\n        }\n        this.#selectedNetworkClientId =\n          networkControllerState.selectedNetworkClientId;\n      },\n    );\n\n    // Listen for account changes\n    this.messagingSystem.subscribe(\n      'AccountsController:selectedAccountChange',\n      (account) => {\n        const address = account?.address;\n        /**\n         * TEMP: There's a race condition where the account state isn't updated immediately.\n         * Until this has been fixed, we rely on the event payload for the latest account instead of #getCurrentAccount().\n         * Issue: https://github.com/MetaMask/accounts-planning/issues/887\n         */\n        this.refreshStakingEligibility({ address }).catch(console.error);\n        this.refreshPooledStakes({ address }).catch(console.error);\n      },\n    );\n  }\n\n  #initializeSDK(networkClientId?: string) {\n    const { selectedNetworkClientId } = networkClientId\n      ? { selectedNetworkClientId: networkClientId }\n      : this.messagingSystem.call('NetworkController:getState');\n\n    const networkClient = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n\n    if (!networkClient?.provider) {\n      this.#stakeSDK = null;\n      return;\n    }\n\n    const provider = new Web3Provider(networkClient.provider);\n    const { chainId } = networkClient.configuration;\n\n    // Initialize appropriate contracts based on chainId\n    const config: StakeSdkConfig = {\n      chainId: convertHexToDecimal(chainId),\n    };\n\n    try {\n      this.#stakeSDK = StakeSdk.create(config);\n      this.#stakeSDK.pooledStakingContract.connectSignerOrProvider(provider);\n    } catch (error) {\n      this.#stakeSDK = null;\n      // Only log unexpected errors, not unsupported chain errors\n      if (\n        !(\n          error instanceof Error &&\n          error.message.includes('Unsupported chainId')\n        )\n      ) {\n        console.error('Stake SDK initialization failed:', error);\n      }\n    }\n  }\n\n  #getCurrentAccount() {\n    return this.messagingSystem.call('AccountsController:getSelectedAccount');\n  }\n\n  #getCurrentChainId(): number {\n    const { selectedNetworkClientId } = this.messagingSystem.call(\n      'NetworkController:getState',\n    );\n    const {\n      configuration: { chainId },\n    } = this.messagingSystem.call(\n      'NetworkController:getNetworkClientById',\n      selectedNetworkClientId,\n    );\n    return convertHexToDecimal(chainId);\n  }\n\n  /**\n   * Refreshes the pooled stakes data for the current account.\n   * Fetches updated stake information including lifetime rewards, assets, and exit requests\n   * from the staking API service and updates the state.\n   *\n   * @param options - Optional arguments\n   * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).\n   * @param [options.address] - The address to refresh pooled stakes for (optional).\n   * @returns A promise that resolves when the stakes data has been updated\n   */\n  async refreshPooledStakes({\n    resetCache = false,\n    address,\n  }: RefreshPooledStakesOptions = {}): Promise<void> {\n    const addressToUse = address ?? this.#getCurrentAccount()?.address;\n\n    if (!addressToUse) {\n      return;\n    }\n\n    const chainId = this.#getCurrentChainId();\n\n    const { accounts, exchangeRate } =\n      await this.#stakingApiService.getPooledStakes(\n        [addressToUse],\n        chainId,\n        resetCache,\n      );\n\n    this.update((state) => {\n      state.pooled_staking.pooledStakes = accounts[0];\n      state.pooled_staking.exchangeRate = exchangeRate;\n    });\n  }\n\n  /**\n   * Refreshes the staking eligibility status for the current account.\n   * Updates the eligibility status in the controller state based on the location and address blocklist for compliance.\n   *\n   * @param options - Optional arguments\n   * @param [options.address] - Address to refresh staking eligibility for (optional).\n   * @returns A promise that resolves when the eligibility status has been updated\n   */\n  async refreshStakingEligibility({\n    address,\n  }: RefreshStakingEligibilityOptions = {}): Promise<void> {\n    const addressToCheck = address ?? this.#getCurrentAccount()?.address;\n\n    if (!addressToCheck) {\n      return;\n    }\n\n    const { eligible: isEligible } =\n      await this.#stakingApiService.getPooledStakingEligibility([\n        addressToCheck,\n      ]);\n\n    this.update((state) => {\n      state.pooled_staking.isEligible = isEligible;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault metadata for the current chain.\n   * Updates the vault metadata in the controller state including APY, capacity,\n   * fee percentage, total assets, and vault address.\n   *\n   * @returns A promise that resolves when the vault metadata has been updated\n   */\n  async refreshPooledStakingVaultMetadata(): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultMetadata = await this.#stakingApiService.getVaultData(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultMetadata = vaultMetadata;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault daily apys for the current chain.\n   * Updates the pooled staking vault daily apys controller state.\n   *\n   * @param days - The number of days to fetch pooled staking vault daily apys for (defaults to 365).\n   * @param order - The order in which to fetch pooled staking vault daily apys. Descending order fetches the latest N days (latest working backwards). Ascending order fetches the oldest N days (oldest working forwards) (defaults to 'desc').\n   * @returns A promise that resolves when the pooled staking vault daily apys have been updated.\n   */\n  async refreshPooledStakingVaultDailyApys(\n    days = 365,\n    order: 'asc' | 'desc' = 'desc',\n  ): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultDailyApys = await this.#stakingApiService.getVaultDailyApys(\n      chainId,\n      days,\n      order,\n    );\n\n    this.update((state) => {\n      state.pooled_staking.vaultDailyApys = vaultDailyApys;\n    });\n  }\n\n  /**\n   * Refreshes pooled staking vault apy averages for the current chain.\n   * Updates the pooled staking vault apy averages controller state.\n   *\n   * @returns A promise that resolves when the pooled staking vault apy averages have been updated.\n   */\n  async refreshPooledStakingVaultApyAverages() {\n    const chainId = this.#getCurrentChainId();\n    const vaultApyAverages =\n      await this.#stakingApiService.getVaultApyAverages(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultApyAverages = vaultApyAverages;\n    });\n  }\n\n  /**\n   * Refreshes all pooled staking related data including stakes, eligibility, and vault data.\n   * This method allows partial success, meaning some data may update while other requests fail.\n   * All errors are collected and thrown as a single error message.\n   *\n   * @param options - Optional arguments\n   * @param [options.resetCache] - Control whether the BE cache should be invalidated (optional).\n   * @param [options.address] - The address to refresh pooled stakes for (optional).\n   * @returns A promise that resolves when all possible data has been updated\n   * @throws {Error} If any of the refresh operations fail, with concatenated error messages\n   */\n  async refreshPooledStakingData({\n    resetCache,\n    address,\n  }: RefreshPooledStakingDataOptions = {}): Promise<void> {\n    const errors: Error[] = [];\n\n    await Promise.all([\n      this.refreshPooledStakes({ resetCache, address }).catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshStakingEligibility({ address }).catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultMetadata().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultDailyApys().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshPooledStakingVaultApyAverages().catch((error) => {\n        errors.push(error);\n      }),\n    ]);\n\n    if (errors.length > 0) {\n      throw new Error(\n        `Failed to refresh some staking data: ${errors\n          .map((e) => e.message)\n          .join(', ')}`,\n      );\n    }\n  }\n}\n"]}

package/dist/types.cjs

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

package/dist/types.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.cjs","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"","sourcesContent":["export type RefreshStakingEligibilityOptions = {\n  address?: string;\n};\n\nexport type RefreshPooledStakesOptions = {\n  resetCache?: boolean;\n  address?: string;\n};\n\nexport type RefreshPooledStakingDataOptions = {\n  resetCache?: boolean;\n  address?: string;\n};\n"]}

package/dist/types.d.cts

@@ -0,0 +1,12 @@
+export type RefreshStakingEligibilityOptions = {
+    address?: string;
+};
+export type RefreshPooledStakesOptions = {
+    resetCache?: boolean;
+    address?: string;
+};
+export type RefreshPooledStakingDataOptions = {
+    resetCache?: boolean;
+    address?: string;
+};
+//# sourceMappingURL=types.d.cts.map

package/dist/types.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.cts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,gCAAgC,GAAG;IAC7C,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACvC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,+BAA+B,GAAG;IAC5C,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC"}

package/dist/types.d.mts

@@ -0,0 +1,12 @@
+export type RefreshStakingEligibilityOptions = {
+    address?: string;
+};
+export type RefreshPooledStakesOptions = {
+    resetCache?: boolean;
+    address?: string;
+};
+export type RefreshPooledStakingDataOptions = {
+    resetCache?: boolean;
+    address?: string;
+};
+//# sourceMappingURL=types.d.mts.map

package/dist/types.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.mts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,gCAAgC,GAAG;IAC7C,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACvC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,+BAA+B,GAAG;IAC5C,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC"}

package/dist/types.mjs

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.mjs.map

package/dist/types.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.mjs","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"","sourcesContent":["export type RefreshStakingEligibilityOptions = {\n  address?: string;\n};\n\nexport type RefreshPooledStakesOptions = {\n  resetCache?: boolean;\n  address?: string;\n};\n\nexport type RefreshPooledStakingDataOptions = {\n  resetCache?: boolean;\n  address?: string;\n};\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@metamask-previews/earn-controller",
-  "version": "0.9.0-preview-62c1e797",
+  "version": "0.10.0-preview-7ecf950",
   "description": "Manages state for earning features and coordinates interactions between staking services, SDK integrations, and other controllers to enable users to participate in various earning opportunities",
   "keywords": [
     "MetaMask",