From 5c0dd54f24ab7f630616e6ebbb1576530a800841 Mon Sep 17 00:00:00 2001 From: UniverseFinance <87629076+UniverseFinance@users.noreply.github.com> Date: Thu, 25 Nov 2021 13:20:41 +0800 Subject: [PATCH] Update helpers.sol add approve --- .../mainnet/connectors/universeFinance/helpers.sol | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/contracts/mainnet/connectors/universeFinance/helpers.sol b/contracts/mainnet/connectors/universeFinance/helpers.sol index 02161e7f..e9a9c6cc 100644 --- a/contracts/mainnet/connectors/universeFinance/helpers.sol +++ b/contracts/mainnet/connectors/universeFinance/helpers.sol @@ -27,4 +27,17 @@ abstract contract Helpers is DSMath, Basic { return IVaultV3(universeVault).withdraw(share0, share1); } + function _approve(address universeVault, uint256 amount0, uint256 amount1) internal { + IVaultV3 universe = IVaultV3(universeVault); + TokenInterface token; + if (amount0 > 0) { + token = universe.token0(); + token.approve(address(universeAdapter), amount0); + } + if (amount1 > 0) { + token = universe.token1(); + token.approve(address(universeAdapter), amount1); + } + } + }