Added WSTETH connector

This commit is contained in:
Shriya Tyagi 2022-09-04 05:19:51 +05:30
parent 45d4cca1b9
commit b35e03d63a
4 changed files with 86 additions and 0 deletions

View File

@ -0,0 +1,7 @@
//SPDX-License-Identifier: MIT
pragma solidity ^0.7.0;
contract Events {
event LogDepositSTETH(uint256 tokenAmt, uint256 getId, uint256 setId);
event LogWithdrawWSTETH(uint256 tokenAmt, uint256 getId, uint256 setId);
}

View File

@ -0,0 +1,9 @@
//SPDX-License-Identifier: MIT
pragma solidity ^0.7.0;
import './interface.sol';
import { TokenInterface } from "../../common/interfaces.sol";
abstract contract Helpers {
IWSTETH internal constant wstethContract = IWSTETH(0x7f39C581F595B53c5cb19bD0b3f8dA6c935E2Ca0);
TokenInterface internal constant stethContract = TokenInterface(0xae7ab96520DE3A18E5e111B5EaAb095312D7fE84);
}

View File

@ -0,0 +1,8 @@
//SPDX-License-Identifier: MIT
pragma solidity ^0.7.0;
interface IWSTETH {
function balanceOf(address account) external view returns (uint256);
function wrap(uint256 _stETHAmount) external returns (uint256);
function unwrap(uint256 _wstETHAmount) external returns (uint256);
}

View File

@ -0,0 +1,62 @@
//SPDX-License-Identifier: MIT
pragma solidity ^0.7.0;
import { Basic } from "../../common/basic.sol";
import './helpers.sol';
/**
* @title WSTETH.
* @dev Wrap and Unwrap STETH.
*/
abstract contract WSTETHContract is Helpers, Basic {
/**
* @dev Deposit STETH into WSTETH.
* @notice Wrap STETH into WSTETH
* @param stethAmt The amount of STETH to deposit. (For max: `uint256(-1)`)
* @param getId ID to retrieve amt.
* @param setId ID stores the amount of STETH deposited.
*/
function deposit(
uint256 stethAmt,
uint256 getId,
uint256 setId
) external returns (string memory _eventName, bytes memory _eventParam) {
uint256 _amt = getUint(getId, stethAmt);
_amt = _amt == uint(-1) ? _amt = stethContract.balanceOf(address(this)) : _amt;
approve(stethContract, address(wstethContract), _amt);
setId = wstethContract.wrap(_amt);
_eventName = "LogDepositSTETH(uint256,uint256,uint256)";
_eventParam = abi.encode(_amt, getId, setId);
}
/**
* @dev Withdraw STETH from WSTETH from Smart Account
* @notice Unwrap STETH from WSTETH
* @param wstethAmt The amount of WSTETH to withdraw. (For max: `uint256(-1)`)
* @param getId ID to retrieve amt.
* @param setId ID stores the amount of ETH withdrawn.
*/
function withdraw(
uint256 wstethAmt,
uint256 getId,
uint256 setId
) external returns (string memory _eventName, bytes memory _eventParam) {
uint256 _amt = getUint(getId, wstethAmt);
_amt = _amt == uint(-1) ? wstethContract.balanceOf(address(this)) : _amt;
setId = wstethContract.unwrap(_amt);
_eventName = "LogWithdrawWSTETH(uint256,uint256,uint256)";
_eventParam = abi.encode(_amt, getId, setId);
}
}
contract ConnectV2WETH is WSTETHContract {
string constant public name = "WSTETH-v1.0";
}