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

Addressing PR comments

Browse Source
This commit is contained in:
Tianjie Wei 2022-03-31 08:27:31 -07:00
parent 7534ddd4ad
commit 74a7effdb4
6 changed files with 953 additions and 953 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
contracts/mainnet/connectors/notional/events.sol
View File

@ -1,3 +1,4 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.7.6;
contract Events {

64
contracts/mainnet/connectors/notional/helpers.sol
View File

@ -1,3 +1,4 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.7.6;
pragma abicoder v2;
@ -11,55 +12,36 @@ abstract contract Helpers is DSMath, Basic {
uint8 internal constant BORROW_TRADE = 1;
uint256 internal constant INTERNAL_TOKEN_PRECISION = 1e8;
uint256 internal constant ETH_CURRENCY_ID = 1;
uint256 internal constant MAX_DEPOSIT = uint256(-1);
uint256 internal constant MAX_DEPOSIT = type(uint256).max;
/// @dev Contract address is different on Kovan: 0x0EAE7BAdEF8f95De91fDDb74a89A786cF891Eb0e
NotionalInterface internal constant notional =
NotionalInterface(0x1344A36A1B56144C3Bc62E7757377D288fDE0369);
/// @notice Returns the address of the underlying token for a given currency id,
function getUnderlyingToken(uint16 currencyId)
function getAssetOrUnderlyingToken(uint16 currencyId, bool underlying)
internal
view
returns (address)
{
// prettier-ignore
(/* assetToken */, Token memory underlyingToken) = notional.getCurrency(currencyId);
return underlyingToken.tokenAddress;
(Token memory assetToken, Token memory underlyingToken) = notional.getCurrency(currencyId);
return
underlying ? underlyingToken.tokenAddress : assetToken.tokenAddress;
}
/// @notice Returns the address of the asset token for a given currency id
function getAssetToken(uint16 currencyId) internal view returns (address) {
// prettier-ignore
(Token memory assetToken, /* underlyingToken */) = notional.getCurrency(currencyId);
return assetToken.tokenAddress;
}
function getCashBalance(uint16 currencyId)
internal
view
returns (int256 cashBalance)
{
// prettier-ignore
(
cashBalance,
/* int256 nTokenBalance */,
/* int256 lastClaimTime */
) = notional.getAccountBalance(currencyId, address(this));
}
function getNTokenBalance(uint16 currencyId)
function getCashOrNTokenBalance(uint16 currencyId, bool nToken)
internal
view
returns (uint256)
{
// prettier-ignore
(
/* int256 cashBalance */,
int256 cashBalance,
int256 nTokenBalance,
/* int256 lastClaimTime */
) = notional.getAccountBalance(currencyId, address(this));
return toUint(nTokenBalance);
return toUint(nToken ? nTokenBalance : cashBalance);
}
function getNTokenRedeemAmount(
@ -67,12 +49,22 @@ abstract contract Helpers is DSMath, Basic {
uint96 _tokensToRedeem,
uint256 getId
) internal returns (uint96 tokensToRedeem) {
tokensToRedeem = uint96(getUint(getId, _tokensToRedeem));
if (tokensToRedeem == uint96(-1)) {
tokensToRedeem = uint96(getNTokenBalance(currencyId));
tokensToRedeem = toUint96(getUint(getId, _tokensToRedeem));
if (tokensToRedeem == type(uint96).max) {
tokensToRedeem = toUint96(getCashOrNTokenBalance(currencyId, true));
}
}
function toUint96(uint256 value) internal pure returns (uint96) {
require(value <= type(uint96).max, "uint96 value overflow");
return uint96(value);
}
function toUint88(uint256 value) internal pure returns (uint88) {
require(value <= type(uint88).max, "uint88 value overflow");
return uint88(value);
}
function getMsgValue(
uint16 currencyId,
bool useUnderlying,
@ -140,9 +132,10 @@ abstract contract Helpers is DSMath, Basic {
: depositAmount;
}
address tokenAddress = useUnderlying
? getUnderlyingToken(currencyId)
: getAssetToken(currencyId);
address tokenAddress = getAssetOrUnderlyingToken(
currencyId,
useUnderlying
);
if (depositAmount == MAX_DEPOSIT) {
depositAmount = TokenInterface(tokenAddress).balanceOf(
@ -170,10 +163,7 @@ abstract contract Helpers is DSMath, Basic {
return ethAddr;
}
return
useUnderlying
? getUnderlyingToken(currencyId)
: getAssetToken(currencyId);
return getAssetOrUnderlyingToken(currencyId, useUnderlying);
}
/// @dev Executes a trade action and sets the balance change to setId

4
contracts/mainnet/connectors/notional/interface.sol
View File

@ -1,3 +1,4 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.7.6;
pragma abicoder v2;
@ -118,7 +119,8 @@ interface NotionalInterface {
address redeemer,
uint16 currencyId,
uint96 tokensToRedeem_,
bool sellTokenAssets
bool sellTokenAssets,
bool acceptResidualAssets
) external returns (int256);
function batchBalanceAction(

33
contracts/mainnet/connectors/notional/main.sol
View File

@ -1,3 +1,4 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.7.6;
pragma abicoder v2;
@ -96,11 +97,9 @@ abstract contract NotionalResolver is Events, Helpers {
returns (string memory _eventName, bytes memory _eventParam)
{
withdrawAmount = getUint(getId, withdrawAmount);
uint88 amountInternalPrecision = withdrawAmount == uint256(-1)
? uint88(getCashBalance(currencyId))
: uint88(
convertToInternal(currencyId, withdrawAmount)
);
uint88 amountInternalPrecision = withdrawAmount == type(uint256).max
? toUint88(getCashOrNTokenBalance(currencyId, false))
: toUint88(convertToInternal(currencyId, withdrawAmount));
uint256 amountWithdrawn = notional.withdraw(
currencyId,
@ -152,6 +151,7 @@ abstract contract NotionalResolver is Events, Helpers {
uint16 currencyId,
bool sellTokenAssets,
uint96 tokensToRedeem,
bool acceptResidualAssets,
uint256 getId,
uint256 setId
)
@ -169,7 +169,8 @@ abstract contract NotionalResolver is Events, Helpers {
address(this),
currencyId,
tokensToRedeem,
sellTokenAssets
sellTokenAssets,
acceptResidualAssets
);
// Floor asset cash change at zero in order to properly set the uint. If the asset cash change is negative
@ -225,7 +226,7 @@ abstract contract NotionalResolver is Events, Helpers {
action[0].currencyId = currencyId;
action[0].depositActionAmount = tokensToRedeem;
action[0].redeemToUnderlying = redeemToUnderlying;
if (amountToWithdraw == uint256(-1)) {
if (amountToWithdraw == type(uint256).max) {
// This setting will override the withdrawAmountInternalPrecision
action[0].withdrawEntireCashBalance = true;
} else {
@ -342,7 +343,7 @@ abstract contract NotionalResolver is Events, Helpers {
action[0].depositActionAmount = depositAmount;
// withdraw amount, withdraw cash and redeem to underlying are all 0 and false
uint256 nTokenBefore = getNTokenBalance(currencyId);
uint256 nTokenBefore = getCashOrNTokenBalance(currencyId, true);
uint256 msgValue = getMsgValue(
currencyId,
useUnderlying,
@ -351,7 +352,10 @@ abstract contract NotionalResolver is Events, Helpers {
notional.batchBalanceAction{ value: msgValue }(address(this), action);
uint256 nTokenBalanceChange = sub(getNTokenBalance(currencyId), nTokenBefore);
uint256 nTokenBalanceChange = sub(
getCashOrNTokenBalance(currencyId, true),
nTokenBefore
);
if (setId != 0) {
// Set the amount of nTokens minted
@ -388,8 +392,8 @@ abstract contract NotionalResolver is Events, Helpers {
returns (string memory _eventName, bytes memory _eventParam)
{
cashBalanceToMint = getUint(getId, cashBalanceToMint);
if (cashBalanceToMint == uint256(-1))
cashBalanceToMint = uint256(getCashBalance(currencyId));
if (cashBalanceToMint == type(uint256).max)
cashBalanceToMint = getCashOrNTokenBalance(currencyId, false);
BalanceAction[] memory action = new BalanceAction[](1);
action[0].actionType = DepositActionType.ConvertCashToNToken;
@ -397,11 +401,14 @@ abstract contract NotionalResolver is Events, Helpers {
action[0].depositActionAmount = cashBalanceToMint;
// NOTE: withdraw amount, withdraw cash and redeem to underlying are all 0 and false
uint256 nTokenBefore = getNTokenBalance(currencyId);
uint256 nTokenBefore = getCashOrNTokenBalance(currencyId, true);
notional.batchBalanceAction(address(this), action);
uint256 nTokenBalanceChange = sub(getNTokenBalance(currencyId), nTokenBefore);
uint256 nTokenBalanceChange = sub(
getCashOrNTokenBalance(currencyId, true),
nTokenBefore
);
if (setId != 0) {
// Set the amount of nTokens minted

2
test/mainnet/notional/notional.helpers.ts
View File

@ -95,7 +95,7 @@ const redeemNTokenRaw = async (
{
connector: "NOTIONAL-TEST-A",
method: "redeemNTokenRaw",
args: [currencyId, sellTokenAssets, tokensToRedeem, 0, 0]
args: [currencyId, sellTokenAssets, tokensToRedeem, false, 0, 0]
}
];

6
test/mainnet/notional/notional.test.ts
View File

@ -52,7 +52,7 @@ describe("Notional", function () {
forking: {
//@ts-ignore
jsonRpcUrl: hre.config.networks.hardhat.forking.url,
blockNumber: 13798624,
blockNumber: 14483893,
},
},
],
@ -240,7 +240,7 @@ describe("Notional", function () {
expect(
await daiToken.balanceOf(dsaWallet0.address),
"expect DSA wallet to contain borrowed balance minus fees"
).to.be.gte(ethers.utils.parseEther("990"));
).to.be.gte(ethers.utils.parseEther("985"));
});
it("test_deposit_ETH_and_borrow_DAI_asset", async function () {
@ -256,7 +256,7 @@ describe("Notional", function () {
expect(
await cdaiToken.balanceOf(dsaWallet0.address),
"expect DSA wallet to contain borrowed balance minus fees"
).to.be.gte(ethers.utils.parseUnits("4500000000000", 0));
).to.be.gte(ethers.utils.parseUnits("4490000000000", 0));
});
it("test_deposit_DAI_underlying_and_borrow_ETH", async function () {