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

updated the setFeeAmount bug

Browse Source
This commit is contained in:
Samyak Jain 2020-07-18 02:29:38 +10:00
parent c5078d6a71
commit fefbeba589
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
contracts/connectors/instapool.sol
View File

@ -154,12 +154,13 @@ contract LiquidityHelpers is Helpers {
return 0x06cB7C24990cBE6b9F99982f975f9147c000fec6; // TODO - change
}
function calculateTotalFeeAmt(uint amt) internal view returns (uint totalAmt) {
function calculateTotalFeeAmt(address token, uint amt) internal view returns (uint totalAmt) {
uint fee = InstaPoolFeeInterface(getInstaPoolFeeAddr()).fee();
if(fee == 0) {
uint flashAmt = liquidityContract.borrowedToken(token);
if (fee == 0) {
totalAmt = amt;
} else {
uint feeAmt = wmul(amt, fee);
uint feeAmt = wmul(flashAmt, fee);
totalAmt = add(amt, feeAmt);
}
}
@ -312,10 +313,10 @@ contract LiquidityAccessHelper is EventHelpers {
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/
function setFeeAmount(uint amt, uint getId, uint setId) external payable {
function setFeeAmount(address token, uint amt, uint getId, uint setId) external payable {
_amt = getUint(getId, amt);
require(_amt != 0, "amt-is-0");
uint totalFee = calculateTotalFeeAmt(_amt);
uint totalFee = calculateTotalFeeAmt(token, _amt);
setUint(setId, totalFee);
}