change balanceof and assetsOf functions - WIP

plumenetwork · Dec 9, 2024 · 6322211 · 6322211
1 parent 663410a
commit 6322211
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nest/src/token/BoringVaultAdapter.sol b/nest/src/token/BoringVaultAdapter.sol
@@ -448,8 +448,7 @@ abstract contract BoringVaultAdapter is
     function balanceOf(
         address account
     ) public view override(IERC20, ERC20Upgradeable) returns (uint256) {
-        BoringVaultAdapterStorage storage $ = _getBoringVaultAdapterStorage();
-        return $.boringVault.lens.balanceOf(account, $.boringVault.vault);
+        return super.balanceOf(account);
     }
 
     /**
@@ -460,8 +459,9 @@ abstract contract BoringVaultAdapter is
     function assetsOf(
         address account
     ) public view virtual override(ComponentToken) returns (uint256) {
-        BoringVaultAdapterStorage storage $ = _getBoringVaultAdapterStorage();
-        return $.boringVault.lens.balanceOfInAssets(account, $.boringVault.vault, $.boringVault.accountant);
+        // Convert the user's shares to assets using the conversion rate
+        uint256 shares = super.balanceOf(account);
+        return convertToAssets(shares);
     }
 
     // ========== METADATA OVERRIDES ==========