Instadapp/dsa-connectors
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors.git synced 2024-07-29 22:37:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

added referral parameter in deposit func

Browse Source
This commit is contained in:
bhavik-m 2022-03-07 00:39:38 +05:30
parent 0d7080b8b1
commit cb89771f7e
3 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
contracts/mainnet/connectors/lido_stETH/events.sol
View File

@ -1,5 +1,10 @@
pragma solidity ^0.7.0; pragma solidity ^0.7.0;
contract Events { contract Events {
event LogDeposit(uint256 Amt, uint256 getId, uint256 setId); event LogDeposit(
uint256 Amt,
address referral,
uint256 getId,
uint256 setId
);
} }

8
contracts/mainnet/connectors/lido_stETH/main.sol
View File

@ -15,11 +15,13 @@ abstract contract Resolver is Events, DSMath, Basic, Helpers {
* @dev deposit ETH into Lido. * @dev deposit ETH into Lido.
* @notice sends Eth into lido and in return you get equivalent of stEth tokens * @notice sends Eth into lido and in return you get equivalent of stEth tokens
* @param amt The amount of ETH to deposit. (For max: `uint256(-1)`) * @param amt The amount of ETH to deposit. (For max: `uint256(-1)`)
* @param referral optional referral parameter
* @param getId ID to retrieve amt. * @param getId ID to retrieve amt.
* @param setId ID stores the amount of ETH deposited. * @param setId ID stores the amount of ETH deposited.
*/ */
function deposit( function deposit(
uint256 amt, uint256 amt,
address referral,
uint256 getId, uint256 getId,
uint256 setId uint256 setId
) )
@ -30,11 +32,11 @@ abstract contract Resolver is Events, DSMath, Basic, Helpers {
uint256 _amt = getUint(getId, amt); uint256 _amt = getUint(getId, amt);
_amt = _amt == uint256(-1) ? address(this).balance : _amt; _amt = _amt == uint256(-1) ? address(this).balance : _amt;
lidoInterface.submit{ value: amt }(address(0)); lidoInterface.submit{ value: amt }(referral);
setUint(setId, _amt); setUint(setId, _amt);
_eventName = "LogDeposit(uint256,uint256,uint256)"; _eventName = "LogDeposit(uint256,address,uint256,uint256)";
_eventParam = abi.encode(_amt, getId, setId); _eventParam = abi.encode(_amt, referral, getId, setId);
} }
} }

2
test/mainnet/lido_stETH/lidoStEth.test.ts
View File

@ -85,7 +85,7 @@ describe("LidoStEth", function() {
{ {
connector: connectorName, connector: connectorName,
method: "deposit", method: "deposit",
args: [_amt,0,0] args: [_amt,"0x0000000000000000000000000000000000000000",0,0]
}, },
]; ];
const tx = await dsaWallet0 const tx = await dsaWallet0