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

Added fee taker function

Browse Source
This commit is contained in:
Thrilok Kumar 2020-07-29 19:21:14 +05:30
parent 991af63d50
commit 5ceaf7f7fc
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
contracts/connectors/1inch.sol
View File

@ -5,6 +5,9 @@ import { TokenInterface , MemoryInterface, EventInterface} from "../common/inter
import { Stores } from "../common/stores.sol"; import { Stores } from "../common/stores.sol";
import { DSMath } from "../common/math.sol"; import { DSMath } from "../common/math.sol";
import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
interface OneInchInterace { interface OneInchInterace {
function swap( function swap(
TokenInterface fromToken, TokenInterface fromToken,
@ -62,6 +65,7 @@ interface OneProtoInterface {
contract OneHelpers is Stores, DSMath { contract OneHelpers is Stores, DSMath {
using SafeERC20 for IERC20;
/** /**
* @dev Return 1Inch Address * @dev Return 1Inch Address
*/ */
@ -129,6 +133,30 @@ contract OneHelpers is Stores, DSMath {
} }
return _tokens; return _tokens;
} }
function _transfer(address payable to, IERC20 token, uint _amt) internal {
address(token) == getEthAddr() ?
to.transfer(_amt) :
token.safeTransfer(to, _amt);
}
function takeFee(
address token,
uint amount,
address feeCollector,
uint feePercent
) internal returns (uint leftAmt, uint feeAmount){
if (feeCollector != address(0)) {
feeAmount = wmul(amount, feePercent);
leftAmt = sub(amount, feeAmount);
uint feeCollectorAmt = wmul(feeAmount, 3 * 10 ** 17);
uint restAmt = sub(feeAmount, feeCollectorAmt);
_transfer(payable(feeCollector), IERC20(token), feeCollectorAmt);
_transfer(payable(getReferralAddr()), IERC20(token), restAmt); // TODO - change address
} else {
leftAmt = amount;
}
}
} }