@metamask/earn-controller 0.1.0

package/CHANGELOG.md

@@ -0,0 +1,17 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [Unreleased]
+
+## [0.1.0]
+
+### Added
+
+- Initial release ([#5271](https://github.com/MetaMask/core/pull/5271))
+
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/earn-controller@0.1.0...HEAD
+[0.1.0]: https://github.com/MetaMask/core/releases/tag/@metamask/earn-controller@0.1.0

package/LICENSE

@@ -0,0 +1,20 @@
+MIT License
+
+Copyright (c) 2025 MetaMask
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

package/README.md

@@ -0,0 +1,15 @@
+# `@metamask/earn-controller`
+
+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.
+
+## Installation
+
+`yarn add @metamask/earn-controller`
+
+or
+
+`npm install @metamask/earn-controller`
+
+## Contributing
+
+This package is part of a monorepo. Instructions for contributing can be found in the [monorepo README](https://github.com/MetaMask/core#readme).

package/dist/EarnController.cjs

@@ -0,0 +1,230 @@
+"use strict";
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var _EarnController_instances, _EarnController_stakeSDK, _EarnController_selectedNetworkClientId, _EarnController_stakingApiService, _EarnController_initializeSDK, _EarnController_getCurrentAccount, _EarnController_getCurrentChainId;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EarnController = exports.getDefaultEarnControllerState = exports.controllerName = void 0;
+const providers_1 = require("@ethersproject/providers");
+const base_controller_1 = require("@metamask/base-controller");
+const controller_utils_1 = require("@metamask/controller-utils");
+const stake_sdk_1 = require("@metamask/stake-sdk");
+exports.controllerName = 'EarnController';
+/**
+ * Metadata for the EarnController.
+ */
+const earnControllerMetadata = {
+    pooled_staking: {
+        persist: true,
+        anonymous: false,
+    },
+    stablecoin_lending: {
+        persist: true,
+        anonymous: false,
+    },
+    lastUpdated: {
+        persist: false,
+        anonymous: true,
+    },
+};
+// === Default State ===
+const DEFAULT_STABLECOIN_VAULT = {
+    symbol: '',
+    name: '',
+    chainId: 0,
+    tokenAddress: '',
+    vaultAddress: '',
+    currentAPY: '0',
+    supply: '0',
+    liquidity: '0',
+};
+/**
+ * Gets the default state for the EarnController.
+ *
+ * @returns The default EarnController state.
+ */
+function getDefaultEarnControllerState() {
+    return {
+        pooled_staking: {
+            pooledStakes: {
+                account: '',
+                lifetimeRewards: '0',
+                assets: '0',
+                exitRequests: [],
+            },
+            exchangeRate: '1',
+            vaultData: {
+                apy: '0',
+                capacity: '0',
+                feePercent: 0,
+                totalAssets: '0',
+                vaultAddress: '0x0000000000000000000000000000000000000000',
+            },
+            isEligible: false,
+        },
+        stablecoin_lending: {
+            vaults: [DEFAULT_STABLECOIN_VAULT],
+        },
+        lastUpdated: 0,
+    };
+}
+exports.getDefaultEarnControllerState = getDefaultEarnControllerState;
+// === CONTROLLER DEFINITION ===
+/**
+ * EarnController manages DeFi earning opportunities across different protocols and chains.
+ */
+class EarnController extends base_controller_1.BaseController {
+    constructor({ messenger, state = {}, }) {
+        super({
+            name: exports.controllerName,
+            metadata: earnControllerMetadata,
+            messenger,
+            state: {
+                ...getDefaultEarnControllerState(),
+                ...state,
+            },
+        });
+        _EarnController_instances.add(this);
+        _EarnController_stakeSDK.set(this, null);
+        _EarnController_selectedNetworkClientId.set(this, void 0);
+        _EarnController_stakingApiService.set(this, new stake_sdk_1.StakingApiService());
+        __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_initializeSDK).call(this);
+        this.refreshPooledStakingData().catch(console.error);
+        const { selectedNetworkClientId } = this.messagingSystem.call('NetworkController:getState');
+        __classPrivateFieldSet(this, _EarnController_selectedNetworkClientId, selectedNetworkClientId, "f");
+        this.messagingSystem.subscribe('NetworkController:stateChange', (networkControllerState) => {
+            if (networkControllerState.selectedNetworkClientId !==
+                __classPrivateFieldGet(this, _EarnController_selectedNetworkClientId, "f")) {
+                __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_initializeSDK).call(this, networkControllerState.selectedNetworkClientId);
+                this.refreshPooledStakingData().catch(console.error);
+            }
+            __classPrivateFieldSet(this, _EarnController_selectedNetworkClientId, networkControllerState.selectedNetworkClientId, "f");
+        });
+        // Listen for account changes
+        this.messagingSystem.subscribe('AccountsController:selectedAccountChange', () => {
+            this.refreshPooledStakingData().catch(console.error);
+        });
+    }
+    /**
+     * Refreshes the pooled stakes data for the current account.
+     * Fetches updated stake information including lifetime rewards, assets, and exit requests
+     * from the staking API service and updates the state.
+     *
+     * @returns A promise that resolves when the stakes data has been updated
+     */
+    async refreshPooledStakes() {
+        const currentAccount = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this);
+        if (!currentAccount?.address) {
+            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);
+        this.update((state) => {
+            state.pooled_staking.pooledStakes = accounts[0];
+            state.pooled_staking.exchangeRate = exchangeRate;
+        });
+    }
+    /**
+     * 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.
+     *
+     * @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) {
+            return;
+        }
+        const { eligible: isEligible } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakingEligibility([
+            currentAccount.address,
+        ]);
+        this.update((state) => {
+            state.pooled_staking.isEligible = isEligible;
+        });
+    }
+    /**
+     * Refreshes vault data for the current chain.
+     * Updates the vault data in the controller state including APY, capacity,
+     * fee percentage, total assets, and vault address.
+     *
+     * @returns A promise that resolves when the vault data has been updated
+     */
+    async refreshVaultData() {
+        const chainId = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentChainId).call(this);
+        const vaultData = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getVaultData(chainId);
+        this.update((state) => {
+            state.pooled_staking.vaultData = vaultData;
+        });
+    }
+    /**
+     * Refreshes all pooled staking related data including stakes, eligibility, and vault data.
+     * 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.
+     *
+     * @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() {
+        const errors = [];
+        await Promise.all([
+            this.refreshPooledStakes().catch((error) => {
+                errors.push(error);
+            }),
+            this.refreshStakingEligibility().catch((error) => {
+                errors.push(error);
+            }),
+            this.refreshVaultData().catch((error) => {
+                errors.push(error);
+            }),
+        ]);
+        if (errors.length > 0) {
+            throw new Error(`Failed to refresh some staking data: ${errors
+                .map((e) => e.message)
+                .join(', ')}`);
+        }
+    }
+}
+exports.EarnController = EarnController;
+_EarnController_stakeSDK = new WeakMap(), _EarnController_selectedNetworkClientId = new WeakMap(), _EarnController_stakingApiService = new WeakMap(), _EarnController_instances = new WeakSet(), _EarnController_initializeSDK = function _EarnController_initializeSDK(networkClientId) {
+    const { selectedNetworkClientId } = networkClientId
+        ? { selectedNetworkClientId: networkClientId }
+        : this.messagingSystem.call('NetworkController:getState');
+    const networkClient = this.messagingSystem.call('NetworkController:getNetworkClientById', selectedNetworkClientId);
+    if (!networkClient?.provider) {
+        __classPrivateFieldSet(this, _EarnController_stakeSDK, null, "f");
+        return;
+    }
+    const provider = new providers_1.Web3Provider(networkClient.provider);
+    const { chainId } = networkClient.configuration;
+    // Initialize appropriate contracts based on chainId
+    const config = {
+        chainId: (0, controller_utils_1.convertHexToDecimal)(chainId),
+    };
+    try {
+        __classPrivateFieldSet(this, _EarnController_stakeSDK, stake_sdk_1.StakeSdk.create(config), "f");
+        __classPrivateFieldGet(this, _EarnController_stakeSDK, "f").pooledStakingContract.connectSignerOrProvider(provider);
+    }
+    catch (error) {
+        __classPrivateFieldSet(this, _EarnController_stakeSDK, null, "f");
+        // Only log unexpected errors, not unsupported chain errors
+        if (!(error instanceof Error &&
+            error.message.includes('Unsupported chainId'))) {
+            console.error('Stake SDK initialization failed:', error);
+        }
+    }
+}, _EarnController_getCurrentAccount = function _EarnController_getCurrentAccount() {
+    return this.messagingSystem.call('AccountsController:getSelectedAccount');
+}, _EarnController_getCurrentChainId = function _EarnController_getCurrentChainId() {
+    const { selectedNetworkClientId } = this.messagingSystem.call('NetworkController:getState');
+    const { configuration: { chainId }, } = this.messagingSystem.call('NetworkController:getNetworkClientById', selectedNetworkClientId);
+    return (0, controller_utils_1.convertHexToDecimal)(chainId);
+};
+//# sourceMappingURL=EarnController.cjs.map

package/dist/EarnController.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"EarnController.cjs","sourceRoot":"","sources":["../src/EarnController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,wDAAwD;AAWxD,+DAA2D;AAC3D,iEAAiE;AAMjE,mDAM6B;AAEhB,QAAA,cAAc,GAAG,gBAAgB,CAAC;AAwB/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;;;;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,SAAS,EAAE;gBACT,GAAG,EAAE,GAAG;gBACR,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,4CAA4C;aAC3D;YACD,UAAU,EAAE,KAAK;SAClB;QACD,kBAAkB,EAAE;YAClB,MAAM,EAAE,CAAC,wBAAwB,CAAC;SACnC;QACD,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AAxBD,sEAwBC;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,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;aACtD;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,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACvD,CAAC,CACF,CAAC;IACJ,CAAC;IA2DD;;;;;;OAMG;IACH,KAAK,CAAC,mBAAmB;QACvB,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,CACR,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,gBAAgB;QACpB,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAEtE,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,wBAAwB;QAC5B,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzC,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,gBAAgB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACtC,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;AArND,wCAqNC;wQAzJgB,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 { NetworkControllerStateChangeEvent } from '@metamask/network-controller';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n} from '@metamask/network-controller';\nimport {\n  StakeSdk,\n  StakingApiService,\n  type PooledStake,\n  type StakeSdkConfig,\n  type VaultData,\n} from '@metamask/stake-sdk';\n\nexport const controllerName = 'EarnController';\n\nexport type PooledStakingState = {\n  pooledStakes: PooledStake;\n  exchangeRate: string;\n  vaultData: VaultData;\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\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      vaultData: {\n        apy: '0',\n        capacity: '0',\n        feePercent: 0,\n        totalAssets: '0',\n        vaultAddress: '0x0000000000000000000000000000000000000000',\n      },\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.refreshPooledStakingData().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.refreshPooledStakingData().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   * @returns A promise that resolves when the stakes data has been updated\n   */\n  async refreshPooledStakes(): 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      );\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 vault data for the current chain.\n   * Updates the vault data 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 data has been updated\n   */\n  async refreshVaultData(): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultData = await this.#stakingApiService.getVaultData(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultData = vaultData;\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   * @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(): Promise<void> {\n    const errors: Error[] = [];\n\n    await Promise.all([\n      this.refreshPooledStakes().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshStakingEligibility().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshVaultData().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

@@ -0,0 +1,109 @@
+import type { AccountsControllerGetSelectedAccountAction, AccountsControllerSelectedAccountChangeEvent } from "@metamask/accounts-controller";
+import type { ControllerGetStateAction, ControllerStateChangeEvent, RestrictedMessenger } from "@metamask/base-controller";
+import { BaseController } from "@metamask/base-controller";
+import type { NetworkControllerStateChangeEvent } from "@metamask/network-controller";
+import type { NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction } from "@metamask/network-controller";
+import { type PooledStake, type VaultData } from "@metamask/stake-sdk";
+export declare const controllerName = "EarnController";
+export type PooledStakingState = {
+    pooledStakes: PooledStake;
+    exchangeRate: string;
+    vaultData: VaultData;
+    isEligible: boolean;
+};
+export type StablecoinLendingState = {
+    vaults: StablecoinVault[];
+};
+export type StablecoinVault = {
+    symbol: string;
+    name: string;
+    chainId: number;
+    tokenAddress: string;
+    vaultAddress: string;
+    currentAPY: string;
+    supply: string;
+    liquidity: string;
+};
+export type EarnControllerState = {
+    pooled_staking: PooledStakingState;
+    stablecoin_lending?: StablecoinLendingState;
+    lastUpdated: number;
+};
+/**
+ * Gets the default state for the EarnController.
+ *
+ * @returns The default EarnController state.
+ */
+export declare function getDefaultEarnControllerState(): EarnControllerState;
+/**
+ * The action which can be used to retrieve the state of the EarnController.
+ */
+export type EarnControllerGetStateAction = ControllerGetStateAction<typeof controllerName, EarnControllerState>;
+/**
+ * All actions that EarnController registers, to be called externally.
+ */
+export type EarnControllerActions = EarnControllerGetStateAction;
+/**
+ * All actions that EarnController calls internally.
+ */
+export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | AccountsControllerGetSelectedAccountAction;
+/**
+ * The event that EarnController publishes when updating state.
+ */
+export type EarnControllerStateChangeEvent = ControllerStateChangeEvent<typeof controllerName, EarnControllerState>;
+/**
+ * All events that EarnController publishes, to be subscribed to externally.
+ */
+export type EarnControllerEvents = EarnControllerStateChangeEvent;
+/**
+ * All events that EarnController subscribes to internally.
+ */
+export type AllowedEvents = AccountsControllerSelectedAccountChangeEvent | NetworkControllerStateChangeEvent;
+/**
+ * The messenger which is restricted to actions and events accessed by
+ * EarnController.
+ */
+export type EarnControllerMessenger = RestrictedMessenger<typeof controllerName, EarnControllerActions | AllowedActions, EarnControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
+/**
+ * EarnController manages DeFi earning opportunities across different protocols and chains.
+ */
+export declare class EarnController extends BaseController<typeof controllerName, EarnControllerState, EarnControllerMessenger> {
+    #private;
+    constructor({ messenger, state, }: {
+        messenger: EarnControllerMessenger;
+        state?: Partial<EarnControllerState>;
+    });
+    /**
+     * Refreshes the pooled stakes data for the current account.
+     * Fetches updated stake information including lifetime rewards, assets, and exit requests
+     * from the staking API service and updates the state.
+     *
+     * @returns A promise that resolves when the stakes data has been updated
+     */
+    refreshPooledStakes(): 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.
+     *
+     * @returns A promise that resolves when the eligibility status has been updated
+     */
+    refreshStakingEligibility(): Promise<void>;
+    /**
+     * Refreshes vault data for the current chain.
+     * Updates the vault data in the controller state including APY, capacity,
+     * fee percentage, total assets, and vault address.
+     *
+     * @returns A promise that resolves when the vault data has been updated
+     */
+    refreshVaultData(): Promise<void>;
+    /**
+     * Refreshes all pooled staking related data including stakes, eligibility, and vault data.
+     * 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.
+     *
+     * @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(): Promise<void>;
+}
+//# sourceMappingURL=EarnController.d.cts.map

package/dist/EarnController.d.cts.map

@@ -0,0 +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,EAAE,iCAAiC,EAAE,qCAAqC;AACtF,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AACtC,OAAO,EAGL,KAAK,WAAW,EAEhB,KAAK,SAAS,EACf,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,SAAS,EAAE,SAAS,CAAC;IACrB,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;AAcF;;;;GAIG;AACH,wBAAgB,6BAA6B,IAAI,mBAAmB,CAwBnE;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;IAoGD;;;;;;OAMG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB1C;;;;;OAKG;IACG,yBAAyB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBhD;;;;;;OAMG;IACG,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IASvC;;;;;;;OAOG;IACG,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC;CAuBhD"}

package/dist/EarnController.d.mts

@@ -0,0 +1,109 @@
+import type { AccountsControllerGetSelectedAccountAction, AccountsControllerSelectedAccountChangeEvent } from "@metamask/accounts-controller";
+import type { ControllerGetStateAction, ControllerStateChangeEvent, RestrictedMessenger } from "@metamask/base-controller";
+import { BaseController } from "@metamask/base-controller";
+import type { NetworkControllerStateChangeEvent } from "@metamask/network-controller";
+import type { NetworkControllerGetNetworkClientByIdAction, NetworkControllerGetStateAction } from "@metamask/network-controller";
+import { type PooledStake, type VaultData } from "@metamask/stake-sdk";
+export declare const controllerName = "EarnController";
+export type PooledStakingState = {
+    pooledStakes: PooledStake;
+    exchangeRate: string;
+    vaultData: VaultData;
+    isEligible: boolean;
+};
+export type StablecoinLendingState = {
+    vaults: StablecoinVault[];
+};
+export type StablecoinVault = {
+    symbol: string;
+    name: string;
+    chainId: number;
+    tokenAddress: string;
+    vaultAddress: string;
+    currentAPY: string;
+    supply: string;
+    liquidity: string;
+};
+export type EarnControllerState = {
+    pooled_staking: PooledStakingState;
+    stablecoin_lending?: StablecoinLendingState;
+    lastUpdated: number;
+};
+/**
+ * Gets the default state for the EarnController.
+ *
+ * @returns The default EarnController state.
+ */
+export declare function getDefaultEarnControllerState(): EarnControllerState;
+/**
+ * The action which can be used to retrieve the state of the EarnController.
+ */
+export type EarnControllerGetStateAction = ControllerGetStateAction<typeof controllerName, EarnControllerState>;
+/**
+ * All actions that EarnController registers, to be called externally.
+ */
+export type EarnControllerActions = EarnControllerGetStateAction;
+/**
+ * All actions that EarnController calls internally.
+ */
+export type AllowedActions = NetworkControllerGetNetworkClientByIdAction | NetworkControllerGetStateAction | AccountsControllerGetSelectedAccountAction;
+/**
+ * The event that EarnController publishes when updating state.
+ */
+export type EarnControllerStateChangeEvent = ControllerStateChangeEvent<typeof controllerName, EarnControllerState>;
+/**
+ * All events that EarnController publishes, to be subscribed to externally.
+ */
+export type EarnControllerEvents = EarnControllerStateChangeEvent;
+/**
+ * All events that EarnController subscribes to internally.
+ */
+export type AllowedEvents = AccountsControllerSelectedAccountChangeEvent | NetworkControllerStateChangeEvent;
+/**
+ * The messenger which is restricted to actions and events accessed by
+ * EarnController.
+ */
+export type EarnControllerMessenger = RestrictedMessenger<typeof controllerName, EarnControllerActions | AllowedActions, EarnControllerEvents | AllowedEvents, AllowedActions['type'], AllowedEvents['type']>;
+/**
+ * EarnController manages DeFi earning opportunities across different protocols and chains.
+ */
+export declare class EarnController extends BaseController<typeof controllerName, EarnControllerState, EarnControllerMessenger> {
+    #private;
+    constructor({ messenger, state, }: {
+        messenger: EarnControllerMessenger;
+        state?: Partial<EarnControllerState>;
+    });
+    /**
+     * Refreshes the pooled stakes data for the current account.
+     * Fetches updated stake information including lifetime rewards, assets, and exit requests
+     * from the staking API service and updates the state.
+     *
+     * @returns A promise that resolves when the stakes data has been updated
+     */
+    refreshPooledStakes(): 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.
+     *
+     * @returns A promise that resolves when the eligibility status has been updated
+     */
+    refreshStakingEligibility(): Promise<void>;
+    /**
+     * Refreshes vault data for the current chain.
+     * Updates the vault data in the controller state including APY, capacity,
+     * fee percentage, total assets, and vault address.
+     *
+     * @returns A promise that resolves when the vault data has been updated
+     */
+    refreshVaultData(): Promise<void>;
+    /**
+     * Refreshes all pooled staking related data including stakes, eligibility, and vault data.
+     * 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.
+     *
+     * @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(): Promise<void>;
+}
+//# sourceMappingURL=EarnController.d.mts.map

package/dist/EarnController.d.mts.map

@@ -0,0 +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,EAAE,iCAAiC,EAAE,qCAAqC;AACtF,OAAO,KAAK,EACV,2CAA2C,EAC3C,+BAA+B,EAChC,qCAAqC;AACtC,OAAO,EAGL,KAAK,WAAW,EAEhB,KAAK,SAAS,EACf,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,SAAS,EAAE,SAAS,CAAC;IACrB,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;AAcF;;;;GAIG;AACH,wBAAgB,6BAA6B,IAAI,mBAAmB,CAwBnE;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;IAoGD;;;;;;OAMG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB1C;;;;;OAKG;IACG,yBAAyB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBhD;;;;;;OAMG;IACG,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IASvC;;;;;;;OAOG;IACG,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC;CAuBhD"}

package/dist/EarnController.mjs

@@ -0,0 +1,225 @@
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var _EarnController_instances, _EarnController_stakeSDK, _EarnController_selectedNetworkClientId, _EarnController_stakingApiService, _EarnController_initializeSDK, _EarnController_getCurrentAccount, _EarnController_getCurrentChainId;
+import { Web3Provider } from "@ethersproject/providers";
+import { BaseController } from "@metamask/base-controller";
+import { convertHexToDecimal } from "@metamask/controller-utils";
+import { StakeSdk, StakingApiService } from "@metamask/stake-sdk";
+export const controllerName = 'EarnController';
+/**
+ * Metadata for the EarnController.
+ */
+const earnControllerMetadata = {
+    pooled_staking: {
+        persist: true,
+        anonymous: false,
+    },
+    stablecoin_lending: {
+        persist: true,
+        anonymous: false,
+    },
+    lastUpdated: {
+        persist: false,
+        anonymous: true,
+    },
+};
+// === Default State ===
+const DEFAULT_STABLECOIN_VAULT = {
+    symbol: '',
+    name: '',
+    chainId: 0,
+    tokenAddress: '',
+    vaultAddress: '',
+    currentAPY: '0',
+    supply: '0',
+    liquidity: '0',
+};
+/**
+ * Gets the default state for the EarnController.
+ *
+ * @returns The default EarnController state.
+ */
+export function getDefaultEarnControllerState() {
+    return {
+        pooled_staking: {
+            pooledStakes: {
+                account: '',
+                lifetimeRewards: '0',
+                assets: '0',
+                exitRequests: [],
+            },
+            exchangeRate: '1',
+            vaultData: {
+                apy: '0',
+                capacity: '0',
+                feePercent: 0,
+                totalAssets: '0',
+                vaultAddress: '0x0000000000000000000000000000000000000000',
+            },
+            isEligible: false,
+        },
+        stablecoin_lending: {
+            vaults: [DEFAULT_STABLECOIN_VAULT],
+        },
+        lastUpdated: 0,
+    };
+}
+// === CONTROLLER DEFINITION ===
+/**
+ * EarnController manages DeFi earning opportunities across different protocols and chains.
+ */
+export class EarnController extends BaseController {
+    constructor({ messenger, state = {}, }) {
+        super({
+            name: controllerName,
+            metadata: earnControllerMetadata,
+            messenger,
+            state: {
+                ...getDefaultEarnControllerState(),
+                ...state,
+            },
+        });
+        _EarnController_instances.add(this);
+        _EarnController_stakeSDK.set(this, null);
+        _EarnController_selectedNetworkClientId.set(this, void 0);
+        _EarnController_stakingApiService.set(this, new StakingApiService());
+        __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_initializeSDK).call(this);
+        this.refreshPooledStakingData().catch(console.error);
+        const { selectedNetworkClientId } = this.messagingSystem.call('NetworkController:getState');
+        __classPrivateFieldSet(this, _EarnController_selectedNetworkClientId, selectedNetworkClientId, "f");
+        this.messagingSystem.subscribe('NetworkController:stateChange', (networkControllerState) => {
+            if (networkControllerState.selectedNetworkClientId !==
+                __classPrivateFieldGet(this, _EarnController_selectedNetworkClientId, "f")) {
+                __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_initializeSDK).call(this, networkControllerState.selectedNetworkClientId);
+                this.refreshPooledStakingData().catch(console.error);
+            }
+            __classPrivateFieldSet(this, _EarnController_selectedNetworkClientId, networkControllerState.selectedNetworkClientId, "f");
+        });
+        // Listen for account changes
+        this.messagingSystem.subscribe('AccountsController:selectedAccountChange', () => {
+            this.refreshPooledStakingData().catch(console.error);
+        });
+    }
+    /**
+     * Refreshes the pooled stakes data for the current account.
+     * Fetches updated stake information including lifetime rewards, assets, and exit requests
+     * from the staking API service and updates the state.
+     *
+     * @returns A promise that resolves when the stakes data has been updated
+     */
+    async refreshPooledStakes() {
+        const currentAccount = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentAccount).call(this);
+        if (!currentAccount?.address) {
+            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);
+        this.update((state) => {
+            state.pooled_staking.pooledStakes = accounts[0];
+            state.pooled_staking.exchangeRate = exchangeRate;
+        });
+    }
+    /**
+     * 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.
+     *
+     * @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) {
+            return;
+        }
+        const { eligible: isEligible } = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getPooledStakingEligibility([
+            currentAccount.address,
+        ]);
+        this.update((state) => {
+            state.pooled_staking.isEligible = isEligible;
+        });
+    }
+    /**
+     * Refreshes vault data for the current chain.
+     * Updates the vault data in the controller state including APY, capacity,
+     * fee percentage, total assets, and vault address.
+     *
+     * @returns A promise that resolves when the vault data has been updated
+     */
+    async refreshVaultData() {
+        const chainId = __classPrivateFieldGet(this, _EarnController_instances, "m", _EarnController_getCurrentChainId).call(this);
+        const vaultData = await __classPrivateFieldGet(this, _EarnController_stakingApiService, "f").getVaultData(chainId);
+        this.update((state) => {
+            state.pooled_staking.vaultData = vaultData;
+        });
+    }
+    /**
+     * Refreshes all pooled staking related data including stakes, eligibility, and vault data.
+     * 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.
+     *
+     * @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() {
+        const errors = [];
+        await Promise.all([
+            this.refreshPooledStakes().catch((error) => {
+                errors.push(error);
+            }),
+            this.refreshStakingEligibility().catch((error) => {
+                errors.push(error);
+            }),
+            this.refreshVaultData().catch((error) => {
+                errors.push(error);
+            }),
+        ]);
+        if (errors.length > 0) {
+            throw new Error(`Failed to refresh some staking data: ${errors
+                .map((e) => e.message)
+                .join(', ')}`);
+        }
+    }
+}
+_EarnController_stakeSDK = new WeakMap(), _EarnController_selectedNetworkClientId = new WeakMap(), _EarnController_stakingApiService = new WeakMap(), _EarnController_instances = new WeakSet(), _EarnController_initializeSDK = function _EarnController_initializeSDK(networkClientId) {
+    const { selectedNetworkClientId } = networkClientId
+        ? { selectedNetworkClientId: networkClientId }
+        : this.messagingSystem.call('NetworkController:getState');
+    const networkClient = this.messagingSystem.call('NetworkController:getNetworkClientById', selectedNetworkClientId);
+    if (!networkClient?.provider) {
+        __classPrivateFieldSet(this, _EarnController_stakeSDK, null, "f");
+        return;
+    }
+    const provider = new Web3Provider(networkClient.provider);
+    const { chainId } = networkClient.configuration;
+    // Initialize appropriate contracts based on chainId
+    const config = {
+        chainId: convertHexToDecimal(chainId),
+    };
+    try {
+        __classPrivateFieldSet(this, _EarnController_stakeSDK, StakeSdk.create(config), "f");
+        __classPrivateFieldGet(this, _EarnController_stakeSDK, "f").pooledStakingContract.connectSignerOrProvider(provider);
+    }
+    catch (error) {
+        __classPrivateFieldSet(this, _EarnController_stakeSDK, null, "f");
+        // Only log unexpected errors, not unsupported chain errors
+        if (!(error instanceof Error &&
+            error.message.includes('Unsupported chainId'))) {
+            console.error('Stake SDK initialization failed:', error);
+        }
+    }
+}, _EarnController_getCurrentAccount = function _EarnController_getCurrentAccount() {
+    return this.messagingSystem.call('AccountsController:getSelectedAccount');
+}, _EarnController_getCurrentChainId = function _EarnController_getCurrentChainId() {
+    const { selectedNetworkClientId } = this.messagingSystem.call('NetworkController:getState');
+    const { configuration: { chainId }, } = this.messagingSystem.call('NetworkController:getNetworkClientById', selectedNetworkClientId);
+    return convertHexToDecimal(chainId);
+};
+//# sourceMappingURL=EarnController.mjs.map

package/dist/EarnController.mjs.map

@@ -0,0 +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,EAIlB,4BAA4B;AAE7B,MAAM,CAAC,MAAM,cAAc,GAAG,gBAAgB,CAAC;AAwB/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;;;;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,SAAS,EAAE;gBACT,GAAG,EAAE,GAAG;gBACR,QAAQ,EAAE,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,4CAA4C;aAC3D;YACD,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,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;aACtD;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,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACvD,CAAC,CACF,CAAC;IACJ,CAAC;IA2DD;;;;;;OAMG;IACH,KAAK,CAAC,mBAAmB;QACvB,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,CACR,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,gBAAgB;QACpB,MAAM,OAAO,GAAG,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAC1C,MAAM,SAAS,GAAG,MAAM,uBAAA,IAAI,yCAAmB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAEtE,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACpB,KAAK,CAAC,cAAc,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,wBAAwB;QAC5B,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzC,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,gBAAgB,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACtC,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;wQAzJgB,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 { NetworkControllerStateChangeEvent } from '@metamask/network-controller';\nimport type {\n  NetworkControllerGetNetworkClientByIdAction,\n  NetworkControllerGetStateAction,\n} from '@metamask/network-controller';\nimport {\n  StakeSdk,\n  StakingApiService,\n  type PooledStake,\n  type StakeSdkConfig,\n  type VaultData,\n} from '@metamask/stake-sdk';\n\nexport const controllerName = 'EarnController';\n\nexport type PooledStakingState = {\n  pooledStakes: PooledStake;\n  exchangeRate: string;\n  vaultData: VaultData;\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\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      vaultData: {\n        apy: '0',\n        capacity: '0',\n        feePercent: 0,\n        totalAssets: '0',\n        vaultAddress: '0x0000000000000000000000000000000000000000',\n      },\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.refreshPooledStakingData().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.refreshPooledStakingData().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   * @returns A promise that resolves when the stakes data has been updated\n   */\n  async refreshPooledStakes(): 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      );\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 vault data for the current chain.\n   * Updates the vault data 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 data has been updated\n   */\n  async refreshVaultData(): Promise<void> {\n    const chainId = this.#getCurrentChainId();\n    const vaultData = await this.#stakingApiService.getVaultData(chainId);\n\n    this.update((state) => {\n      state.pooled_staking.vaultData = vaultData;\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   * @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(): Promise<void> {\n    const errors: Error[] = [];\n\n    await Promise.all([\n      this.refreshPooledStakes().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshStakingEligibility().catch((error) => {\n        errors.push(error);\n      }),\n      this.refreshVaultData().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/index.cjs

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EarnController = exports.getDefaultEarnControllerState = exports.controllerName = void 0;
+var EarnController_1 = require("./EarnController.cjs");
+Object.defineProperty(exports, "controllerName", { enumerable: true, get: function () { return EarnController_1.controllerName; } });
+Object.defineProperty(exports, "getDefaultEarnControllerState", { enumerable: true, get: function () { return EarnController_1.getDefaultEarnControllerState; } });
+Object.defineProperty(exports, "EarnController", { enumerable: true, get: function () { return EarnController_1.EarnController; } });
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAYA,uDAI0B;AAHxB,gHAAA,cAAc,OAAA;AACd,+HAAA,6BAA6B,OAAA;AAC7B,gHAAA,cAAc,OAAA","sourcesContent":["export type {\n  PooledStakingState,\n  StablecoinLendingState,\n  StablecoinVault,\n  EarnControllerState,\n  EarnControllerGetStateAction,\n  EarnControllerStateChangeEvent,\n  EarnControllerActions,\n  EarnControllerEvents,\n  EarnControllerMessenger,\n} from './EarnController';\n\nexport {\n  controllerName,\n  getDefaultEarnControllerState,\n  EarnController,\n} from './EarnController';\n"]}

package/dist/index.d.cts

@@ -0,0 +1,3 @@
+export type { PooledStakingState, StablecoinLendingState, StablecoinVault, EarnControllerState, EarnControllerGetStateAction, EarnControllerStateChangeEvent, EarnControllerActions, EarnControllerEvents, EarnControllerMessenger, } from "./EarnController.cjs";
+export { controllerName, getDefaultEarnControllerState, EarnController, } from "./EarnController.cjs";
+//# sourceMappingURL=index.d.cts.map

package/dist/index.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,kBAAkB,EAClB,sBAAsB,EACtB,eAAe,EACf,mBAAmB,EACnB,4BAA4B,EAC5B,8BAA8B,EAC9B,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,GACxB,6BAAyB;AAE1B,OAAO,EACL,cAAc,EACd,6BAA6B,EAC7B,cAAc,GACf,6BAAyB"}

package/dist/index.d.mts

@@ -0,0 +1,3 @@
+export type { PooledStakingState, StablecoinLendingState, StablecoinVault, EarnControllerState, EarnControllerGetStateAction, EarnControllerStateChangeEvent, EarnControllerActions, EarnControllerEvents, EarnControllerMessenger, } from "./EarnController.mjs";
+export { controllerName, getDefaultEarnControllerState, EarnController, } from "./EarnController.mjs";
+//# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,kBAAkB,EAClB,sBAAsB,EACtB,eAAe,EACf,mBAAmB,EACnB,4BAA4B,EAC5B,8BAA8B,EAC9B,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,GACxB,6BAAyB;AAE1B,OAAO,EACL,cAAc,EACd,6BAA6B,EAC7B,cAAc,GACf,6BAAyB"}

package/dist/index.mjs

@@ -0,0 +1,2 @@
+export { controllerName, getDefaultEarnControllerState, EarnController } from "./EarnController.mjs";
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAYA,OAAO,EACL,cAAc,EACd,6BAA6B,EAC7B,cAAc,EACf,6BAAyB","sourcesContent":["export type {\n  PooledStakingState,\n  StablecoinLendingState,\n  StablecoinVault,\n  EarnControllerState,\n  EarnControllerGetStateAction,\n  EarnControllerStateChangeEvent,\n  EarnControllerActions,\n  EarnControllerEvents,\n  EarnControllerMessenger,\n} from './EarnController';\n\nexport {\n  controllerName,\n  getDefaultEarnControllerState,\n  EarnController,\n} from './EarnController';\n"]}

package/package.json

@@ -0,0 +1,78 @@
+{
+  "name": "@metamask/earn-controller",
+  "version": "0.1.0",
+  "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",
+    "Ethereum"
+  ],
+  "homepage": "https://github.com/MetaMask/core/tree/main/packages/earn-controller#readme",
+  "bugs": {
+    "url": "https://github.com/MetaMask/core/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/MetaMask/core.git"
+  },
+  "license": "MIT",
+  "sideEffects": false,
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.mts",
+        "default": "./dist/index.mjs"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./package.json": "./package.json"
+  },
+  "main": "./dist/index.cjs",
+  "types": "./dist/index.d.cts",
+  "files": [
+    "dist/"
+  ],
+  "scripts": {
+    "build": "ts-bridge --project tsconfig.build.json --verbose --clean --no-references",
+    "build:docs": "typedoc",
+    "changelog:validate": "../../scripts/validate-changelog.sh @metamask/earn-controller",
+    "changelog:update": "../../scripts/update-changelog.sh @metamask/earn-controller",
+    "publish:preview": "yarn npm publish --tag preview",
+    "test": "NODE_OPTIONS=--experimental-vm-modules jest --reporters=jest-silent-reporter",
+    "test:clean": "NODE_OPTIONS=--experimental-vm-modules jest --clearCache",
+    "test:verbose": "NODE_OPTIONS=--experimental-vm-modules jest --verbose",
+    "test:watch": "NODE_OPTIONS=--experimental-vm-modules jest --watch",
+    "since-latest-release": "../../scripts/since-latest-release.sh"
+  },
+  "dependencies": {
+    "@ethersproject/providers": "^5.7.0",
+    "@metamask/base-controller": "^7.1.1",
+    "@metamask/controller-utils": "^11.4.5",
+    "@metamask/stake-sdk": "^1.0.0"
+  },
+  "devDependencies": {
+    "@metamask/accounts-controller": "^22.0.0",
+    "@metamask/auto-changelog": "^3.4.4",
+    "@metamask/network-controller": "^22.1.1",
+    "@types/jest": "^27.4.1",
+    "deepmerge": "^4.2.2",
+    "jest": "^27.5.1",
+    "ts-jest": "^27.1.4",
+    "typedoc": "^0.24.8",
+    "typedoc-plugin-missing-exports": "^2.0.0",
+    "typescript": "~5.2.2"
+  },
+  "peerDependencies": {
+    "@metamask/accounts-controller": "^22.0.0",
+    "@metamask/network-controller": "^22.1.1"
+  },
+  "engines": {
+    "node": "^18.18 || >=20"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org/"
+  }
+}