def main():
deployer = accounts.at('0xB593d82d53e2c187dc49673709a6E9f806cdC835',
force=True)
# deployer = accounts.load('gh')
werc20 = WERC20.at('0xe28D9dF7718b0b5Ba69E01073fE82254a9eD2F98')
wmas = WMasterChef.at('0x373ae78a14577682591E088F2E78EF1417612c68')
wliq = WLiquidityGauge.at('0xfdB4f97953150e47C8606758C13e70b5a789a7ec')
wsindex = WStakingRewards.at('0x713df2DDDA9C7d7bDa98A9f8fCd82c06c50fbd90')
wsperp = WStakingRewards.at('0xC4635854480ffF80F742645da0310e9e59795c63')
uni_kp3r_oracle = ERC20KP3ROracle.at(
'0x8E2A3777AB22e1c5f6d1fF2BcC6C4aA6aB1DeA14')
sushi_kp3r_oracle = ERC20KP3ROracle.at(
'0xC331687FD71bb1D7f2e237091F8888dDcaD50C9a')
core_oracle = CoreOracle.at('0x1E5BDdd0cDF8839d6b27b34927869eF0AD7bf692')
uni_oracle = UniswapV2Oracle.at(
'0x910aD02D355c2966e8dD8E32C890cD50DB29C3B9')
bal_oracle = BalancerPairOracle.at(
'0x8F93B0AA2643bf74ab38d6a5910fA82D2da02134')
crv_oracle = CurveOracle.at('0x04DE0E42B3b0483248deafE86C4F5428613b76Ff')
proxy_oracle = ProxyOracle.at('0x43c425fB2b00a991A51b18c217D749E393bF1AB2')
# pair token with oracle
from .tokens import Tokens
# re-set oracles
oracle_params = [
# lp tokens
[Tokens.bal_perp_usdc, [14886, 6718, 10250]],
]
token_list_2, param_list = zip(*oracle_params)
proxy_oracle.setOracles(token_list_2, param_list, {'from': deployer})
proxy_oracle.unsetOracles([Tokens.bal_dai_weth], {'from': deployer})
print('DONE')
###########################################################
# test spells (UNCOMMENT TO TEST)
for token in [
Tokens.weth, Tokens.dai, Tokens.usdc, Tokens.usdt, Tokens.dpi,
Tokens.yfi, Tokens.snx, Tokens.susd
]:
token = interface.IERC20Ex(token)
deposit_safebox(token)
bank = HomoraBank.at('0x5f5Cd91070960D13ee549C9CC47e7a4Cd00457bb')
uniswap_spell = UniswapV2SpellV1.at(
'0xc671B7251a789de0835a2fa33c83c8D4afB39092')
sushiswap_spell = SushiswapSpellV1.at(
'0x21Fa95485f4571A3a0d0c396561cF4D8D13D445d')
balancer_spell = BalancerSpellV1.at(
'0x15B79c184A6a8E19a4CA1F637081270343E4D15D')
curve_spell = CurveSpellV1.at('0x42C750024E02816eE32EB2eB4DA79ff5BF343D30')
def main():
admin = accounts[0]
alice = accounts[1]
weth = interface.IERC20Ex('0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2')
dai = interface.IERC20Ex('0x6B175474E89094C44Da98b954EedeAC495271d0F')
lp = interface.IERC20Ex('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
# pool is lp for balancer
pool = interface.ICurvePool('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
werc20 = WERC20.deploy({'from': admin})
simple_oracle = SimpleOracle.deploy({'from': admin})
simple_oracle.setETHPx(
[weth, dai],
[5192296858534827628530496329220096, 8887571220661441971398610676149])
balancer_oracle = BalancerPairOracle.deploy(simple_oracle, {'from': admin})
oracle = ProxyOracle.deploy({'from': admin})
oracle.setWhitelistERC1155([werc20], True, {'from': admin})
oracle.setOracles(
[
'0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2', # WETH
'0x6B175474E89094C44Da98b954EedeAC495271d0F', # DAI
'0x6c3f90f043a72fa612cbac8115ee7e52bde6e490', # lp
],
[
[simple_oracle, 10000, 10000, 10000],
[simple_oracle, 10000, 10000, 10000],
[balancer_oracle, 10000, 10000, 10000],
],
{'from': admin},
)
#####################################################################################
print(
'========================================================================='
)
print('Case 1.')
lp_price = balancer_oracle.getETHPx(lp)
dai_price = simple_oracle.getETHPx(dai)
weth_price = simple_oracle.getETHPx(weth)
print('lp price', lp_price)
print('dai price', dai_price)
print('weth price', weth_price)
assert almostEqual(
lp_price,
weth.balanceOf(lp) * 5 // 4 * 2**112 // lp.totalSupply())
def main():
deployer = accounts.at('0xB593d82d53e2c187dc49673709a6E9f806cdC835',
force=True)
# deployer = accounts.load('gh')
werc20 = WERC20.at('0xe28D9dF7718b0b5Ba69E01073fE82254a9eD2F98')
wmas = WMasterChef.at('0x373ae78a14577682591E088F2E78EF1417612c68')
wliq = WLiquidityGauge.at('0xfdB4f97953150e47C8606758C13e70b5a789a7ec')
wsindex = WStakingRewards.at('0x713df2DDDA9C7d7bDa98A9f8fCd82c06c50fbd90')
wsperp = WStakingRewards.at('0xC4635854480ffF80F742645da0310e9e59795c63')
kp3r_oracle = ERC20KP3ROracle.at(
'0x8E2A3777AB22e1c5f6d1fF2BcC6C4aA6aB1DeA14')
core_oracle = CoreOracle.at('0x1E5BDdd0cDF8839d6b27b34927869eF0AD7bf692')
uni_oracle = UniswapV2Oracle.at(
'0x910aD02D355c2966e8dD8E32C890cD50DB29C3B9')
bal_oracle = BalancerPairOracle.at(
'0x8F93B0AA2643bf74ab38d6a5910fA82D2da02134')
crv_oracle = CurveOracle.at('0x04DE0E42B3b0483248deafE86C4F5428613b76Ff')
proxy_oracle = ProxyOracle.at('0x43c425fB2b00a991A51b18c217D749E393bF1AB2')
bank_impl = HomoraBank.at('0x95ab3Ca8133b47f16f47F6A79410680fE87a6139')
proxy_admin = Contract.from_explorer(
'0x090eCE252cEc5998Db765073D07fac77b8e60CB2')
bank = HomoraBank.at('0x5f5Cd91070960D13ee549C9CC47e7a4Cd00457bb')
proxy_admin.upgrade(bank, bank_impl, {'from': deployer})
bank.setOracle(proxy_oracle, {'from': deployer})
uniswap_spell = UniswapV2SpellV1.deploy(
bank,
werc20,
'0x7a250d5630B4cF539739dF2C5dAcb4c659F2488D',
{'from': deployer},
)
sushiswap_spell = SushiswapSpellV1.deploy(
bank,
werc20,
'0xd9e1cE17f2641f24aE83637ab66a2cca9C378B9F',
wmas,
{'from': deployer},
)
balancer_spell = BalancerSpellV1.at(
'0x15B79c184A6a8E19a4CA1F637081270343E4D15D')
curve_spell = CurveSpellV1.deploy(
bank,
werc20,
'0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2',
wliq,
{'from': deployer},
)
print('DONE')
def main():
deployer = accounts.at('0xB593d82d53e2c187dc49673709a6E9f806cdC835',
force=True)
# deployer = accounts.load('gh')
werc20 = WERC20.at('0xe28D9dF7718b0b5Ba69E01073fE82254a9eD2F98')
wmas = WMasterChef.at('0x373ae78a14577682591E088F2E78EF1417612c68')
wliq = WLiquidityGauge.at('0xfdB4f97953150e47C8606758C13e70b5a789a7ec')
wsindex = WStakingRewards.at('0x713df2DDDA9C7d7bDa98A9f8fCd82c06c50fbd90')
wsperp = WStakingRewards.at('0xC4635854480ffF80F742645da0310e9e59795c63')
uni_kp3r_oracle = ERC20KP3ROracle.at(
'0x8E2A3777AB22e1c5f6d1fF2BcC6C4aA6aB1DeA14')
sushi_kp3r_oracle = ERC20KP3ROracle.at(
'0xC331687FD71bb1D7f2e237091F8888dDcaD50C9a')
core_oracle = CoreOracle.at('0x1E5BDdd0cDF8839d6b27b34927869eF0AD7bf692')
uni_oracle = UniswapV2Oracle.at(
'0x910aD02D355c2966e8dD8E32C890cD50DB29C3B9')
bal_oracle = BalancerPairOracle.at(
'0x8F93B0AA2643bf74ab38d6a5910fA82D2da02134')
crv_oracle = CurveOracle.at('0x04DE0E42B3b0483248deafE86C4F5428613b76Ff')
proxy_oracle = ProxyOracle.at('0x43c425fB2b00a991A51b18c217D749E393bF1AB2')
# pair token with oracle
from .tokens import Tokens
# re-set oracles
oracle_params = [
# lp tokens
[Tokens.bal_dai_weth, [12616, 7927, 10250]],
]
token_list_2, param_list = zip(*oracle_params)
proxy_oracle.setOracles(token_list_2, param_list, {'from': deployer})
print('DONE')
def main():
admin = accounts[0]
alice = accounts[1]
weth = interface.IERC20Ex('0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2')
dai = interface.IERC20Ex('0x6B175474E89094C44Da98b954EedeAC495271d0F')
lp = interface.IERC20Ex('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
# pool is lp for balancer
pool = interface.ICurvePool('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
werc20 = WERC20.deploy({'from': admin})
uni_pair = interface.IUniswapV2Pair(
'0xa478c2975ab1ea89e8196811f51a7b7ade33eb11')
resA, resB, _ = uni_pair.getReserves()
if uni_pair.token0() == weth:
weth_dai_price = resB * 10**18 // resA
else:
weth_dai_price = resA * 10**18 // resB
print('weth dai price', weth_dai_price)
simple_oracle = SimpleOracle.deploy({'from': admin})
simple_oracle.setETHPx([weth, dai],
[2**112, 2**112 * 10**18 // weth_dai_price])
balancer_oracle = BalancerPairOracle.deploy(simple_oracle, {'from': admin})
core_oracle = CoreOracle.deploy({'from': admin})
oracle = ProxyOracle.deploy(core_oracle, {'from': admin})
oracle.setWhitelistERC1155([werc20], True, {'from': admin})
oracle.setOracles(
[weth, dai, lp],
[
[10000, 10000, 10000],
[10000, 10000, 10000],
[10000, 10000, 10000],
],
{'from': admin},
)
core_oracle.setRoute(
[weth, dai, lp],
[simple_oracle, simple_oracle, balancer_oracle],
{'from': admin},
)
#####################################################################################
print(
'========================================================================='
)
print('Case 1.')
lp_price = balancer_oracle.getETHPx(lp)
dai_price = simple_oracle.getETHPx(dai)
weth_price = simple_oracle.getETHPx(weth)
print('lp price', lp_price)
print('dai price', dai_price)
print('weth price', weth_price)
assert almostEqual(
lp_price,
weth.balanceOf(lp) * 5 // 4 * 2**112 // lp.totalSupply())
def main():
admin = accounts[0]
alice = accounts[1]
dai = interface.IERC20Ex('0x6B175474E89094C44Da98b954EedeAC495271d0F')
weth = interface.IERC20Ex('0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2')
lp = interface.IERC20Ex('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
# pool is lp for balancer
pool = interface.IBalancerPool('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
crdai = interface.ICErc20('0x92b767185fb3b04f881e3ac8e5b0662a027a1d9f')
werc20 = WERC20.deploy({'from': admin})
simple_oracle = SimpleOracle.deploy({'from': admin})
simple_oracle.setETHPx([weth, dai], [5192296858534827628530496329220096,
8887571220661441971398610676149])
balancer_oracle = BalancerPairOracle.deploy(simple_oracle, {'from': alice})
core_oracle = CoreOracle.deploy({'from': admin})
oracle = ProxyOracle.deploy(core_oracle, {'from': admin})
oracle.setWhitelistERC1155([werc20], True, {'from': admin})
core_oracle.setRoute(
[weth, dai, lp],
[simple_oracle, simple_oracle, balancer_oracle],
{'from': admin},
)
oracle.setOracles(
[weth, dai, lp],
[
[10000, 10000, 10000],
[10000, 10000, 10000],
[10000, 10000, 10000],
],
{'from': admin},
)
homora = HomoraBank.deploy({'from': admin})
homora.initialize(oracle, 1000, {'from': admin}) # 10% fee
setup_bank_hack(homora)
homora.addBank(dai, crdai, {'from': admin})
# setup initial funds to alice
mint_tokens(dai, alice)
mint_tokens(weth, alice)
# check alice's funds
print(f'Alice weth balance {weth.balanceOf(alice)}')
print(f'Alice dai balance {dai.balanceOf(alice)}')
# Steal some LP from the staking pool
mint_tokens(lp, alice)
# set approval
dai.approve(homora, 2**256-1, {'from': alice})
dai.approve(crdai, 2**256-1, {'from': alice})
weth.approve(homora, 2**256-1, {'from': alice})
lp.approve(homora, 2**256-1, {'from': alice})
balancer_spell = BalancerSpellV1.deploy(
homora, werc20, '0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2', {'from': admin})
# first time call to reduce gas
balancer_spell.getPair(lp, {'from': admin})
#####################################################################################
print('=========================================================================')
print('Case 1.')
prevABal = dai.balanceOf(alice)
prevBBal = weth.balanceOf(alice)
prevLPBal = lp.balanceOf(alice)
prevLPBal_bank = lp.balanceOf(homora)
prevLPBal_werc20 = lp.balanceOf(werc20)
prevARes = interface.IBalancerPool(lp).getBalance(dai)
prevBRes = interface.IBalancerPool(lp).getBalance(weth)
dai_amt = 40000 * 10**18
weth_amt = 10 ** 18
lp_amt = 1 * 10**16
borrow_dai_amt = 0
borrow_weth_amt = 0
# calculate slippage control
total_dai_amt = dai_amt + borrow_dai_amt
total_weth_amt = weth_amt + borrow_weth_amt
dai_weight = 0.2
weth_weight = 0.8
ratio = (((prevARes + total_dai_amt) / prevARes) ** dai_weight) * \
(((prevBRes + total_weth_amt) / prevBRes) ** weth_weight) - 1
lp_desired = lp_amt + int(interface.IERC20(lp).totalSupply() * ratio * 0.995)
print('lp desired', lp_desired)
tx = homora.execute(
0,
balancer_spell,
balancer_spell.addLiquidityWERC20.encode_input(
lp, # lp token
[dai_amt, # supply DAI
weth_amt, # supply WETH
lp_amt, # supply LP
borrow_dai_amt, # borrow DAI
borrow_weth_amt, # borrow WETH
0, # borrow LP tokens
lp_desired] # LP desired
),
{'from': alice}
)
curABal = dai.balanceOf(alice)
curBBal = weth.balanceOf(alice)
curLPBal = lp.balanceOf(alice)
curLPBal_bank = lp.balanceOf(homora)
curLPBal_werc20 = lp.balanceOf(werc20)
curARes = interface.IBalancerPool(lp).getBalance(dai)
curBRes = interface.IBalancerPool(lp).getBalance(weth)
print('spell lp balance', lp.balanceOf(balancer_spell))
print('Alice delta A balance', curABal - prevABal)
print('Alice delta B balance', curBBal - prevBBal)
print('add liquidity gas', tx.gas_used)
print('bank lp balance', curLPBal_bank)
_, _, _, daiDebt, daiShare = homora.getBankInfo(dai)
print('bank dai daiDebt', daiDebt)
print('bank dai daiShare', daiShare)
print('bank prev LP balance', prevLPBal_bank)
print('bank cur LP balance', curLPBal_bank)
print('werc20 prev LP balance', prevLPBal_werc20)
print('werc20 cur LP balance', curLPBal_werc20)
print('prev dai res', prevARes)
print('cur dai res', curARes)
print('prev weth res', prevBRes)
print('cur weth res', curBRes)
# alice
assert almostEqual(curABal - prevABal, -dai_amt), 'incorrect DAI amt'
assert almostEqual(curBBal - prevBBal, -weth_amt), 'incorrect WETH amt'
assert curLPBal - prevLPBal == -lp_amt, 'incorrect LP amt'
# spell
assert dai.balanceOf(balancer_spell) == 0, 'non-zero spell DAI balance'
assert weth.balanceOf(balancer_spell) == 0, 'non-zero spell WETH balance'
assert lp.balanceOf(balancer_spell) == 0, 'non-zero spell LP balance'
assert daiDebt == borrow_dai_amt
# check balance and pool reserves
assert almostEqual(curABal - prevABal - borrow_dai_amt, -
(curARes - prevARes)), 'not all DAI tokens go to LP pool'
assert almostEqual(curBBal - prevBBal - borrow_weth_amt, -
(curBRes - prevBRes)), 'not all WETH tokens go to LP pool'
#####################################################################################
print('=========================================================================')
print('Case 2.')
# remove liquidity from the same position
prevABal = dai.balanceOf(alice)
prevBBal = weth.balanceOf(alice)
prevETHBal = alice.balance()
prevLPBal = lp.balanceOf(alice)
prevLPBal_bank = lp.balanceOf(homora)
prevLPBal_werc20 = lp.balanceOf(werc20)
prevETHBal = alice.balance()
prevARes = interface.IBalancerPool(lp).getBalance(dai)
prevBRes = interface.IBalancerPool(lp).getBalance(weth)
lp_take_amt = 2**256-1 # max
lp_want = 1 * 10**15
dai_repay = 2**256-1 # max
weth_repay = 0
real_dai_repay = homora.borrowBalanceStored(1, dai)
_, _, _, real_lp_take_amt = homora.getPositionInfo(1)
tx = homora.execute(
1,
balancer_spell,
balancer_spell.removeLiquidityWERC20.encode_input(
lp, # LP token
[lp_take_amt, # take out LP tokens
lp_want, # withdraw LP tokens to wallet
dai_repay, # repay DAI
weth_repay, # repay WETH
0, # repay LP
0, # min DAI
0], # min WETH
),
{'from': alice}
)
# return tx
curABal = dai.balanceOf(alice)
curBBal = weth.balanceOf(alice)
curETHBal = alice.balance()
curLPBal = lp.balanceOf(alice)
curLPBal_bank = lp.balanceOf(homora)
curLPBal_werc20 = lp.balanceOf(werc20)
curETHBal = alice.balance()
curARes = interface.IBalancerPool(lp).getBalance(dai)
curBRes = interface.IBalancerPool(lp).getBalance(weth)
print('spell lp balance', lp.balanceOf(balancer_spell))
print('spell dai balance', dai.balanceOf(balancer_spell))
print('spell weth balance', weth.balanceOf(balancer_spell))
print('Alice delta A balance', curABal - prevABal)
print('Alice delta B balance', curBBal - prevBBal)
print('Alice delta ETH balance', curETHBal - prevETHBal)
print('Alice delta LP balance', curLPBal - prevLPBal)
print('remove liquidity gas', tx.gas_used)
print('bank delta lp balance', curLPBal_bank - prevLPBal_bank)
print('bank total lp balance', curLPBal_bank)
_, _, _, daiDebt, daiShare = homora.getBankInfo(dai)
print('bank dai totalDebt', daiDebt)
print('bank dai totalShare', daiShare)
print('LP want', lp_want)
print('bank delta LP amount', curLPBal_bank - prevLPBal_bank)
print('LP take amount', lp_take_amt)
print('prev werc20 LP balance', prevLPBal_werc20)
print('cur werc20 LP balance', curLPBal_werc20)
print('real dai repay', real_dai_repay)
# alice
assert almostEqual(curBBal - prevBBal, 0), 'incorrect WETH amt'
assert almostEqual(curLPBal - prevLPBal, lp_want), 'incorrect LP amt'
# werc20
assert almostEqual(curLPBal_werc20 - prevLPBal_werc20, -
real_lp_take_amt), 'incorrect werc20 LP amt'
# spell
assert dai.balanceOf(balancer_spell) == 0, 'non-zero spell DAI balance'
assert weth.balanceOf(balancer_spell) == 0, 'non-zero spell WETH balance'
assert lp.balanceOf(balancer_spell) == 0, 'non-zero spell LP balance'
# check balance and pool reserves
assert almostEqual(curABal - prevABal + real_dai_repay, -
(curARes - prevARes)), 'inconsistent DAI from withdraw'
assert almostEqual(curBBal - prevBBal,
0), 'inconsistent WETH from withdraw'
assert almostEqual(curETHBal - prevETHBal + weth_repay, -
(curBRes - prevBRes)), 'inconsistent ETH from withdraw'
return tx
def main():
deployer = accounts.at('0xB593d82d53e2c187dc49673709a6E9f806cdC835', force=True)
# deployer = accounts.load('gh')
werc20 = WERC20.at('0xe28D9dF7718b0b5Ba69E01073fE82254a9eD2F98')
wmas = WMasterChef.at('0x373ae78a14577682591E088F2E78EF1417612c68')
wliq = WLiquidityGauge.at('0xfdB4f97953150e47C8606758C13e70b5a789a7ec')
wsindex = WStakingRewards.at('0x713df2DDDA9C7d7bDa98A9f8fCd82c06c50fbd90')
wsperp = WStakingRewards.at('0xC4635854480ffF80F742645da0310e9e59795c63')
kp3r_oracle = ERC20KP3ROracle.at('0x8E2A3777AB22e1c5f6d1fF2BcC6C4aA6aB1DeA14')
core_oracle = CoreOracle.at('0x1E5BDdd0cDF8839d6b27b34927869eF0AD7bf692')
uni_oracle = UniswapV2Oracle.deploy(core_oracle, {'from': deployer})
bal_oracle = BalancerPairOracle.at('0x8F93B0AA2643bf74ab38d6a5910fA82D2da02134')
crv_oracle = CurveOracle.at('0x04DE0E42B3b0483248deafE86C4F5428613b76Ff')
proxy_oracle = ProxyOracle.deploy(core_oracle, {'from': deployer})
# re-route to new uniswap oracle
core_oracle.setRoute([
'0x0bc529c00C6401aEF6D220BE8C6Ea1667F6Ad93e', # YFI
'0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11', # UNI DAI-WETH
'0x0d4a11d5EEaaC28EC3F61d100daF4d40471f1852', # UNI WETH-USDT
'0xB4e16d0168e52d35CaCD2c6185b44281Ec28C9Dc', # UNI USDC-WETH
'0xBb2b8038a1640196FbE3e38816F3e67Cba72D940', # UNI WBTC-WETH
'0x2fDbAdf3C4D5A8666Bc06645B8358ab803996E28', # UNI WETH-YFI
'0x4d5ef58aAc27d99935E5b6B4A6778ff292059991', # UNI DPI-WETH
'0x06da0fd433C1A5d7a4faa01111c044910A184553', # SUSHI WETH-USDT
'0x088ee5007C98a9677165D78dD2109AE4a3D04d0C', # SUSHI WETH-YFI
'0x34b13F8CD184F55d0Bd4Dd1fe6C07D46f245c7eD', # SUSHI DPI-WETH
], [
kp3r_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
], {'from': deployer})
# re-set oracles
proxy_oracle.setOracles([
'0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2', # WETH
'0x6B175474E89094C44Da98b954EedeAC495271d0F', # DAI
'0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48', # USDC
'0xdAC17F958D2ee523a2206206994597C13D831ec7', # USDT
'0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599', # WBTC
'0x1494CA1F11D487c2bBe4543E90080AeBa4BA3C2b', # DPI
'0x0bc529c00C6401aEF6D220BE8C6Ea1667F6Ad93e', # YFI
'0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11', # UNI DAI-WETH
'0x0d4a11d5EEaaC28EC3F61d100daF4d40471f1852', # UNI WETH-USDT
'0xB4e16d0168e52d35CaCD2c6185b44281Ec28C9Dc', # UNI USDC-WETH
'0xBb2b8038a1640196FbE3e38816F3e67Cba72D940', # UNI WBTC-WETH
'0x2fDbAdf3C4D5A8666Bc06645B8358ab803996E28', # UNI WETH-YFI
'0x4d5ef58aAc27d99935E5b6B4A6778ff292059991', # UNI DPI-WETH
'0x06da0fd433C1A5d7a4faa01111c044910A184553', # SUSHI WETH-USDT
'0x088ee5007C98a9677165D78dD2109AE4a3D04d0C', # SUSHI WETH-YFI
'0x34b13F8CD184F55d0Bd4Dd1fe6C07D46f245c7eD', # SUSHI DPI-WETH
'0x8b6e6E7B5b3801FEd2CaFD4b22b8A16c2F2Db21a', # BAL WETH-DAI 80-20
'0xF54025aF2dc86809Be1153c1F20D77ADB7e8ecF4', # BAL PERP-USDC 80-20
'0x6c3F90f043a72FA612cbac8115EE7e52BDe6E490', # CRV 3-POOL
], [
[14350, 6970, 10250], # WETH
[10840, 9230, 10250], # DAI
[10840, 9230, 10250], # USDC
[10840, 9230, 10250], # USDT
[14350, 6970, 10250], # WBTC
[50000, 0, 10250], # DPI
[50000, 0, 10250], # YFI
[50000, 6970, 10250], # UNI DAI-WETH
[50000, 6970, 10250], # UNI WETH-USDT
[50000, 6970, 10250], # UNI USDC-WETH
[50000, 6970, 10250], # UNI WBTC-WETH
[50000, 6970, 10250], # UNI WETH-YFI
[50000, 6970, 10250], # UNI DPI-WETH
[50000, 6970, 10250], # SUSHI WETH-USDT
[50000, 6970, 10250], # SUSHI WETH-YFI
[50000, 6970, 10250], # SUSHI DPI-WETH
[50000, 6970, 10250], # BAL WETH-DAI 80-20
[50000, 0, 10250], # BAL PERP-USDC 80-20
[50000, 9230, 10250], # CRV 3-POOL
], {'from': deployer})
proxy_oracle.setWhitelistERC1155(
[werc20, wmas, wliq, wsindex, wsperp],
True,
{'from': deployer},
)
bank_impl = HomoraBank.deploy({'from': deployer})
proxy_admin = Contract.from_explorer('0x090eCE252cEc5998Db765073D07fac77b8e60CB2')
bank = Contract.from_explorer('0x5f5Cd91070960D13ee549C9CC47e7a4Cd00457bb')
proxy_admin.upgrade(bank, bank_impl, {'from': deployer})
bank.setOracle(proxy_oracle, {'from': deployer})
uniswap_spell = UniswapV2SpellV1.deploy(
bank, werc20, '0x7a250d5630B4cF539739dF2C5dAcb4c659F2488D',
{'from': deployer},
)
sushiswap_spell = SushiswapSpellV1.deploy(
bank, werc20, '0xd9e1cE17f2641f24aE83637ab66a2cca9C378B9F', wmas,
{'from': deployer},
)
balancer_spell = BalancerSpellV1.at('0x15B79c184A6a8E19a4CA1F637081270343E4D15D')
curve_spell = CurveSpellV1.deploy(
bank, werc20, '0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2', wliq,
{'from': deployer},
)
print('DONE')
def main():
admin = accounts[0]
alice = accounts[1]
bob = accounts[2]
dfd = interface.IERC20Ex('0x20c36f062a31865bED8a5B1e512D9a1A20AA333A')
dusd = interface.IERC20Ex('0x5BC25f649fc4e26069dDF4cF4010F9f706c23831')
lp = interface.IERC20Ex('0xd8e9690eff99e21a2de25e0b148ffaf47f47c972')
# pool is lp for balancer
pool = interface.IBalancerPool(
'0xd8e9690eff99e21a2de25e0b148ffaf47f47c972')
crdfd = MockCErc20.deploy(dfd, {'from': admin})
crdusd = MockCErc20.deploy(dusd, {'from': admin})
werc20 = WERC20.deploy({'from': admin})
staking = accounts.at('0xf068236ecad5fabb9883bbb26a6445d6c7c9a924',
force=True)
wstaking = WStakingRewards.deploy(staking, lp, dfd, {'from': admin})
simple_oracle = SimpleOracle.deploy({'from': admin})
simple_oracle.setETHPx([dfd, dusd],
[2**112 // 2 // 700, 2**112 * 2 // 700])
balancer_oracle = BalancerPairOracle.deploy(simple_oracle, {'from': alice})
core_oracle = CoreOracle.deploy({'from': admin})
oracle = ProxyOracle.deploy(core_oracle, {'from': admin})
oracle.setWhitelistERC1155([werc20, wstaking], True, {'from': admin})
core_oracle.setRoute(
[dfd, dusd, lp],
[simple_oracle, simple_oracle, balancer_oracle],
{'from': admin},
)
oracle.setOracles(
[dfd, dusd, lp],
[
[10000, 10000, 10000],
[10000, 10000, 10000],
[10000, 10000, 10000],
],
{'from': admin},
)
homora = HomoraBank.deploy({'from': admin})
homora.initialize(oracle, 1000, {'from': admin}) # 10% fee
setup_bank_hack(homora)
homora.addBank(dfd, crdfd, {'from': admin})
homora.addBank(dusd, crdusd, {'from': admin})
# setup initial funds to alice
mint_tokens(dfd, alice)
mint_tokens(dusd, alice)
# check alice's funds
print(f'Alice dusd balance {dusd.balanceOf(alice)}')
print(f'Alice dfd balance {dfd.balanceOf(alice)}')
# Steal some LP from the staking pool
mint_tokens(lp, alice)
mint_tokens(lp, bob)
# set approval
dfd.approve(homora, 2**256 - 1, {'from': alice})
dfd.approve(crdfd, 2**256 - 1, {'from': alice})
dusd.approve(homora, 2**256 - 1, {'from': alice})
dusd.approve(crdusd, 2**256 - 1, {'from': alice})
lp.approve(homora, 2**256 - 1, {'from': alice})
lp.approve(staking, 2**256 - 1, {'from': bob})
balancer_spell = BalancerSpellV1.deploy(
homora, werc20, '0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2',
{'from': admin})
# first time call to reduce gas
balancer_spell.getPair(lp, {'from': admin})
#####################################################################################
print(
'========================================================================='
)
print('Case 1.')
prevABal = dfd.balanceOf(alice)
prevBBal = dusd.balanceOf(alice)
prevLPBal = lp.balanceOf(alice)
prevLPBal_bank = lp.balanceOf(homora)
prevLPBal_staking = lp.balanceOf(staking)
prevARes = interface.IBalancerPool(lp).getBalance(dfd)
prevBRes = interface.IBalancerPool(lp).getBalance(dusd)
dfd_amt = 100 * 10**18
dusd_amt = 10**18
lp_amt = 0
borrow_dfd_amt = 0
borrow_dusd_amt = 0
# calculate slippage control
total_dfd_amt = dfd_amt + borrow_dfd_amt
total_dusd_amt = dusd_amt + borrow_dusd_amt
dfd_weight = 0.58
dusd_weight = 0.42
ratio = (((prevARes + total_dfd_amt) / prevARes) ** dfd_weight) * \
(((prevBRes + total_dusd_amt) / prevBRes) ** dusd_weight) - 1
lp_desired = lp_amt + int(
interface.IERC20(lp).totalSupply() * ratio * 0.995)
print('lp desired', lp_desired)
tx = homora.execute(
0,
balancer_spell,
balancer_spell.addLiquidityWStakingRewards.encode_input(
lp, # lp token
[
dfd_amt, # supply DFD
dusd_amt, # supply DUSD
lp_amt, # supply LP
borrow_dfd_amt, # borrow DFD
borrow_dusd_amt, # borrow DUSD
0, # borrow LP tokens
lp_desired
], # LP desired
wstaking),
{'from': alice})
curABal = dfd.balanceOf(alice)
curBBal = dusd.balanceOf(alice)
curLPBal = lp.balanceOf(alice)
curLPBal_bank = lp.balanceOf(homora)
curLPBal_staking = lp.balanceOf(staking)
curARes = interface.IBalancerPool(lp).getBalance(dfd)
curBRes = interface.IBalancerPool(lp).getBalance(dusd)
print('spell lp balance', lp.balanceOf(balancer_spell))
print('Alice delta A balance', curABal - prevABal)
print('Alice delta B balance', curBBal - prevBBal)
print('add liquidity gas', tx.gas_used)
print('bank lp balance', curLPBal_bank)
_, _, _, dfdDebt, dfdShare = homora.getBankInfo(dfd)
print('bank dfd dfdDebt', dfdDebt)
print('bank dfd dfdShare', dfdShare)
print('bank prev LP balance', prevLPBal_bank)
print('bank cur LP balance', curLPBal_bank)
print('staking prev LP balance', prevLPBal_staking)
print('staking cur LP balance', curLPBal_staking)
print('prev dfd res', prevARes)
print('cur dfd res', curARes)
print('prev dusd res', prevBRes)
print('cur dusd res', curBRes)
# alice
assert almostEqual(curABal - prevABal, -dfd_amt), 'incorrect DFD amt'
assert almostEqual(curBBal - prevBBal, -dusd_amt), 'incorrect DUSD amt'
assert curLPBal - prevLPBal == -lp_amt, 'incorrect LP amt'
# spell
assert dfd.balanceOf(balancer_spell) == 0, 'non-zero spell DFD balance'
assert dusd.balanceOf(balancer_spell) == 0, 'non-zero spell DUSD balance'
assert lp.balanceOf(balancer_spell) == 0, 'non-zero spell LP balance'
assert dfdDebt == borrow_dfd_amt
# check balance and pool reserves
assert almostEqual(
curABal - prevABal - borrow_dfd_amt,
-(curARes - prevARes)), 'not all DFD tokens go to LP pool'
assert almostEqual(
curBBal - prevBBal - borrow_dusd_amt,
-(curBRes - prevBRes)), 'not all DUSD tokens go to LP pool'
_, _, collId, collSize = homora.getPositionInfo(1)
print('collSize', collSize)
# staking directly
prevDfd = dfd.balanceOf(bob)
print('bob lp balance', interface.IERC20Ex(lp).balanceOf(bob))
tx = interface.IStakingRewards(staking).stake(collSize, {'from': bob})
chain.sleep(20000)
prevAliceDfdBalance = dfd.balanceOf(alice)
print('Alice dfd balance', prevAliceDfdBalance)
#####################################################################################
print(
'========================================================================='
)
print('Case 2. add liquidity the second time')
prevABal = dfd.balanceOf(alice)
prevBBal = dusd.balanceOf(alice)
prevLPBal = lp.balanceOf(alice)
prevLPBal_bank = lp.balanceOf(homora)
prevLPBal_staking = lp.balanceOf(staking)
prevARes = interface.IBalancerPool(lp).getBalance(dfd)
prevBRes = interface.IBalancerPool(lp).getBalance(dusd)
dfd_amt = 100 * 10**18
dusd_amt = 10**18
lp_amt = 0
borrow_dfd_amt = 0
borrow_dusd_amt = 0
# calculate slippage control
total_dfd_amt = dfd_amt + borrow_dfd_amt
total_dusd_amt = dusd_amt + borrow_dusd_amt
dfd_weight = 0.58
dusd_weight = 0.42
ratio = (((prevARes + total_dfd_amt) / prevARes) ** dfd_weight) * \
(((prevBRes + total_dusd_amt) / prevBRes) ** dusd_weight) - 1
lp_desired = lp_amt + int(
interface.IERC20(lp).totalSupply() * ratio * 0.995)
print('lp desired', lp_desired)
tx = homora.execute(
1,
balancer_spell,
balancer_spell.addLiquidityWStakingRewards.encode_input(
lp, # lp token
[
dfd_amt, # supply DFD
dusd_amt, # supply DUSD
lp_amt, # supply LP
borrow_dfd_amt, # borrow DFD
borrow_dusd_amt, # borrow DUSD
0, # borrow LP tokens
lp_desired
], # LP desired
wstaking),
{'from': alice})
curABal = dfd.balanceOf(alice)
curBBal = dusd.balanceOf(alice)
curLPBal = lp.balanceOf(alice)
curLPBal_bank = lp.balanceOf(homora)
curLPBal_staking = lp.balanceOf(staking)
curARes = interface.IBalancerPool(lp).getBalance(dfd)
curBRes = interface.IBalancerPool(lp).getBalance(dusd)
print('spell lp balance', lp.balanceOf(balancer_spell))
print('Alice delta A balance', curABal - prevABal)
print('Alice delta B balance', curBBal - prevBBal)
print('add liquidity gas', tx.gas_used)
print('bank lp balance', curLPBal_bank)
_, _, _, dfdDebt, dfdShare = homora.getBankInfo(dfd)
print('bank dfd dfdDebt', dfdDebt)
print('bank dfd dfdShare', dfdShare)
print('bank prev LP balance', prevLPBal_bank)
print('bank cur LP balance', curLPBal_bank)
print('staking prev LP balance', prevLPBal_staking)
print('staking cur LP balance', curLPBal_staking)
print('prev dfd res', prevARes)
print('cur dfd res', curARes)
print('prev dusd res', prevBRes)
print('cur dusd res', curBRes)
# alice
assert almostEqual(curABal - prevABal, -dfd_amt), 'incorrect DFD amt'
assert almostEqual(curBBal - prevBBal, -dusd_amt), 'incorrect DUSD amt'
assert curLPBal - prevLPBal == -lp_amt, 'incorrect LP amt'
# spell
assert dfd.balanceOf(balancer_spell) == 0, 'non-zero spell DFD balance'
assert dusd.balanceOf(balancer_spell) == 0, 'non-zero spell DUSD balance'
assert lp.balanceOf(balancer_spell) == 0, 'non-zero spell LP balance'
assert dfdDebt == borrow_dfd_amt
# check balance and pool reserves
assert almostEqual(
curABal - prevABal - borrow_dfd_amt,
-(curARes - prevARes)), 'not all DFD tokens go to LP pool'
assert almostEqual(
curBBal - prevBBal - borrow_dusd_amt,
-(curBRes - prevBRes)), 'not all DUSD tokens go to LP pool'
curAliceDfdBalance = dfd.balanceOf(alice)
print('Alice dfd balance', curAliceDfdBalance)
receivedDfd = curAliceDfdBalance - prevAliceDfdBalance + dfd_amt
print('received dfd', receivedDfd)
# check with staking directly
tx = interface.IStakingRewards(staking).getReward({'from': bob})
receivedDfdFromStaking = dfd.balanceOf(bob) - prevDfd
print('receivedDfdFromStaking', receivedDfdFromStaking)
assert almostEqual(receivedDfd, receivedDfdFromStaking)
return tx
def main():
admin = accounts[0]
alice = accounts[1]
bob = accounts[2]
dfd = interface.IERC20Ex('0x20c36f062a31865bED8a5B1e512D9a1A20AA333A')
dusd = interface.IERC20Ex('0x5BC25f649fc4e26069dDF4cF4010F9f706c23831')
weth = interface.IERC20Ex('0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2')
dai = interface.IERC20Ex('0x6B175474E89094C44Da98b954EedeAC495271d0F')
lp = interface.IERC20Ex('0xd8e9690eff99e21a2de25e0b148ffaf47f47c972')
# pool is lp for balancer
pool = interface.IBalancerPool(
'0xd8e9690eff99e21a2de25e0b148ffaf47f47c972')
lp_dai = interface.IERC20Ex('0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
pool_dai = interface.IBalancerPool(
'0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a')
crdfd = MockCErc20.deploy(dfd, {'from': admin})
crdusd = MockCErc20.deploy(dusd, {'from': admin})
crdai = MockCErc20.deploy(dai, {'from': admin})
crweth = MockCErc20.deploy(weth, {'from': admin})
werc20 = WERC20.deploy({'from': admin})
staking = accounts.at('0xf068236ecad5fabb9883bbb26a6445d6c7c9a924',
force=True)
wstaking = WStakingRewards.deploy(staking, lp, dfd, {'from': admin})
simple_oracle = SimpleOracle.deploy({'from': admin})
simple_oracle.setETHPx([dfd, dusd],
[2**112 // 2 // 700, 2**112 * 2 // 700])
balancer_oracle = BalancerPairOracle.deploy(simple_oracle, {'from': alice})
core_oracle = CoreOracle.deploy({'from': admin})
oracle = ProxyOracle.deploy(core_oracle, {'from': admin})
oracle.setWhitelistERC1155([werc20, wstaking], True, {'from': admin})
core_oracle.setRoute(
[dfd, dusd, lp],
[simple_oracle, simple_oracle, balancer_oracle],
{'from': admin},
)
oracle.setOracles(
[dfd, dusd, lp],
[
[10000, 10000, 10000],
[10000, 10000, 10000],
[10000, 10000, 10000],
],
{'from': admin},
)
homora = HomoraBank.deploy({'from': admin})
homora.initialize(oracle, 1000, {'from': admin}) # 10% fee
setup_bank_hack(homora)
homora.addBank(dfd, crdfd, {'from': admin})
homora.addBank(dusd, crdusd, {'from': admin})
# setup initial funds to alice
mint_tokens(dfd, alice)
mint_tokens(dusd, alice)
mint_tokens(weth, alice)
mint_tokens(dai, alice)
mint_tokens(dfd, crdfd)
# check alice's funds
print(f'Alice dusd balance {dusd.balanceOf(alice)}')
print(f'Alice dfd balance {dfd.balanceOf(alice)}')
# Steal some LP from the staking pool
mint_tokens(lp, alice)
mint_tokens(lp, bob)
# set approval
dfd.approve(homora, 2**256 - 1, {'from': alice})
dfd.approve(crdfd, 2**256 - 1, {'from': alice})
dusd.approve(homora, 2**256 - 1, {'from': alice})
dusd.approve(crdusd, 2**256 - 1, {'from': alice})
dai.approve(homora, 2**256 - 1, {'from': alice})
dai.approve(crdai, 2**256 - 1, {'from': alice})
weth.approve(homora, 2**256 - 1, {'from': alice})
weth.approve(crweth, 2**256 - 1, {'from': alice})
lp.approve(homora, 2**256 - 1, {'from': alice})
lp.approve(staking, 2**256 - 1, {'from': bob})
balancer_spell = BalancerSpellV1.deploy(
homora, werc20, '0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2',
{'from': admin})
# first time call to reduce gas
balancer_spell.getPair(lp, {'from': admin})
# whitelist spell in bank
homora.setWhitelistSpells([balancer_spell], [True], {'from': admin})
# whitelist token in bank
homora.setWhitelistTokens([dfd], [True], {'from': admin})
# whitelist lp in spell
balancer_spell.setWhitelistLPTokens([lp, lp_dai], [True, True],
{'from': admin})
#####################################################################################
print(
'========================================================================='
)
print('Case 1.')
prevABal = dfd.balanceOf(alice)
prevBBal = dusd.balanceOf(alice)
prevLPBal = lp.balanceOf(alice)
prevLPBal_bank = lp.balanceOf(homora)
prevLPBal_staking = lp.balanceOf(staking)
prevARes = interface.IBalancerPool(lp).getBalance(dfd)
prevBRes = interface.IBalancerPool(lp).getBalance(dusd)
dfd_amt = 100 * 10**18
dusd_amt = 10**18
lp_amt = 0
borrow_dfd_amt = 10**18
borrow_dusd_amt = 0
# calculate slippage control
total_dfd_amt = dfd_amt + borrow_dfd_amt
total_dusd_amt = dusd_amt + borrow_dusd_amt
dfd_weight = 0.58
dusd_weight = 0.42
ratio = (((prevARes + total_dfd_amt) / prevARes) ** dfd_weight) * \
(((prevBRes + total_dusd_amt) / prevBRes) ** dusd_weight) - 1
lp_desired = lp_amt + int(
interface.IERC20(lp).totalSupply() * ratio * 0.995)
print('lp desired', lp_desired)
tx = homora.execute(
0,
balancer_spell,
balancer_spell.addLiquidityWStakingRewards.encode_input(
lp, # lp token
[
dfd_amt, # supply DFD
dusd_amt, # supply DUSD
lp_amt, # supply LP
borrow_dfd_amt, # borrow DFD
borrow_dusd_amt, # borrow DUSD
0, # borrow LP tokens
lp_desired
], # LP desired
wstaking),
{'from': alice})
curABal = dfd.balanceOf(alice)
curBBal = dusd.balanceOf(alice)
curLPBal = lp.balanceOf(alice)
curLPBal_bank = lp.balanceOf(homora)
curLPBal_staking = lp.balanceOf(staking)
curARes = interface.IBalancerPool(lp).getBalance(dfd)
curBRes = interface.IBalancerPool(lp).getBalance(dusd)
print('spell lp balance', lp.balanceOf(balancer_spell))
print('Alice delta A balance', curABal - prevABal)
print('Alice delta B balance', curBBal - prevBBal)
print('add liquidity gas', tx.gas_used)
print('bank lp balance', curLPBal_bank)
_, _, _, dfdDebt, dfdShare = homora.getBankInfo(dfd)
print('bank dfd dfdDebt', dfdDebt)
print('bank dfd dfdShare', dfdShare)
print('bank prev LP balance', prevLPBal_bank)
print('bank cur LP balance', curLPBal_bank)
print('staking prev LP balance', prevLPBal_staking)
print('staking cur LP balance', curLPBal_staking)
print('prev dfd res', prevARes)
print('cur dfd res', curARes)
print('prev dusd res', prevBRes)
print('cur dusd res', curBRes)
# alice
assert almostEqual(curABal - prevABal, -dfd_amt), 'incorrect DFD amt'
assert almostEqual(curBBal - prevBBal, -dusd_amt), 'incorrect DUSD amt'
assert curLPBal - prevLPBal == -lp_amt, 'incorrect LP amt'
# spell
assert dfd.balanceOf(balancer_spell) == 0, 'non-zero spell DFD balance'
assert dusd.balanceOf(balancer_spell) == 0, 'non-zero spell DUSD balance'
assert lp.balanceOf(balancer_spell) == 0, 'non-zero spell LP balance'
assert dfdDebt == borrow_dfd_amt
# check balance and pool reserves
assert almostEqual(
curABal - prevABal - borrow_dfd_amt,
-(curARes - prevARes)), 'not all DFD tokens go to LP pool'
assert almostEqual(
curBBal - prevBBal - borrow_dusd_amt,
-(curBRes - prevBRes)), 'not all DUSD tokens go to LP pool'
_, _, collId, collSize = homora.getPositionInfo(1)
print('collSize', collSize)
# staking directly
prevDfd = dfd.balanceOf(bob)
print('bob lp balance', interface.IERC20Ex(lp).balanceOf(bob))
tx = interface.IStakingRewards(staking).stake(collSize, {'from': bob})
chain.sleep(20000)
prevAliceDfdBalance = dfd.balanceOf(alice)
print('Alice dfd balance', prevAliceDfdBalance)
#####################################################################################
print(
'========================================================================='
)
print('Case 2. add liquidity (failed tx desired)')
weth_amt = 100 * 10**18
dai_amt = 10**18
lp_amt = 0
borrow_weth_amt = 0
borrow_dai_amt = 0
# calculate slippage control
total_weth_amt = weth_amt + borrow_weth_amt
total_dai_amt = dai_amt + borrow_dai_amt
dfd_weight = 0.8
dusd_weight = 0.2
ratio = (((prevARes + total_weth_amt) / prevARes) ** dfd_weight) * \
(((prevBRes + total_dai_amt) / prevBRes) ** dusd_weight) - 1
lp_desired = lp_amt + int(
interface.IERC20(lp).totalSupply() * ratio * 0.995)
lp_desired = 0
print('lp desired', lp_desired)
try:
tx = homora.execute(
1,
balancer_spell,
balancer_spell.addLiquidityWStakingRewards.encode_input(
lp_dai, # lp token
[
weth_amt, # supply DFD
dai_amt, # supply DUSD
lp_amt, # supply LP
borrow_weth_amt, # borrow DFD
borrow_dai_amt, # borrow DUSD
0, # borrow LP tokens
lp_desired
], # LP desired
wstaking),
{'from': alice})
assert False, 'tx should fail'
except VirtualMachineError:
pass
#####################################################################################
print(
'========================================================================='
)
print('Case 3. remove liquidity (failed tx desired)')
lp_take_amt = 2**256 - 1 # max
lp_want = 0
weth_repay = 2**256 - 1 # max
dai_repay = 2**256 - 1 # max
real_weth_repay = homora.borrowBalanceStored(1, dfd)
_, _, _, real_lp_take_amt = homora.getPositionInfo(1)
expected_withdraw_dfd = collSize * prevARes // interface.IBalancerPool(
lp).totalSupply()
print('expected withdraw DFD', expected_withdraw_dfd)
try:
tx = homora.execute(
1,
balancer_spell,
balancer_spell.removeLiquidityWStakingRewards.encode_input(
lp_dai, # LP token
[
lp_take_amt, # take out LP tokens
lp_want, # withdraw LP tokens to wallet
weth_repay, # repay DFD
dai_repay, # repay DUSD
0, # repay LP
0, # min DFD
0
], # min DUSD
wstaking),
{'from': alice})
assert False, 'tx should fail'
except VirtualMachineError:
pass
#####################################################################################
print(
'========================================================================='
)
print('Case 4. remove liquidity')
# remove liquidity from the same position
prevABal = dfd.balanceOf(alice)
prevBBal = dusd.balanceOf(alice)
prevETHBal = alice.balance()
prevLPBal = lp.balanceOf(alice)
prevLPBal_bank = lp.balanceOf(homora)
prevLPBal_staking = lp.balanceOf(staking)
prevETHBal = alice.balance()
prevCrdfdBal = lp.balanceOf(crdfd)
prevARes = interface.IBalancerPool(lp).getBalance(dfd)
prevBRes = interface.IBalancerPool(lp).getBalance(dusd)
lp_take_amt = 2**256 - 1 # max
lp_want = 0
dfd_repay = 2**256 - 1 # max
dusd_repay = 0
real_dfd_repay = homora.borrowBalanceStored(1, dfd)
_, _, _, real_lp_take_amt = homora.getPositionInfo(1)
expected_withdraw_dfd = collSize * prevARes // interface.IBalancerPool(
lp).totalSupply()
print('expected withdraw DFD', expected_withdraw_dfd)
tx = homora.execute(
1,
balancer_spell,
balancer_spell.removeLiquidityWStakingRewards.encode_input(
lp, # LP token
[
lp_take_amt, # take out LP tokens
lp_want, # withdraw LP tokens to wallet
dfd_repay, # repay DFD
dusd_repay, # repay DUSD
0, # repay LP
0, # min DFD
0
], # min DUSD
wstaking),
{'from': alice})
curABal = dfd.balanceOf(alice)
curBBal = dusd.balanceOf(alice)
curETHBal = alice.balance()
curLPBal = lp.balanceOf(alice)
curLPBal_bank = lp.balanceOf(homora)
curLPBal_staking = lp.balanceOf(staking)
curETHBal = alice.balance()
curCrdfdBal = lp.balanceOf(crdfd)
curARes = interface.IBalancerPool(lp).getBalance(dfd)
curBRes = interface.IBalancerPool(lp).getBalance(dusd)
print('spell lp balance', lp.balanceOf(balancer_spell))
print('spell dfd balance', dfd.balanceOf(balancer_spell))
print('spell dusd balance', dusd.balanceOf(balancer_spell))
print('Alice delta A balance', curABal - prevABal)
print('Alice delta B balance', curBBal - prevBBal)
print('Alice delta ETH balance', curETHBal - prevETHBal)
print('Alice delta LP balance', curLPBal - prevLPBal)
print('remove liquidity gas', tx.gas_used)
print('bank delta lp balance', curLPBal_bank - prevLPBal_bank)
print('bank total lp balance', curLPBal_bank)
_, _, _, dfdDebt, dfdShare = homora.getBankInfo(dfd)
print('bank dfd totalDebt', dfdDebt)
print('bank dfd totalShare', dfdShare)
print('LP want', lp_want)
print('bank delta LP amount', curLPBal_bank - prevLPBal_bank)
print('LP take amount', lp_take_amt)
print('prev staking LP balance', prevLPBal_staking)
print('cur staking LP balance', curLPBal_staking)
print('real dfd repay', real_dfd_repay)
print('curCrdfdBal', curCrdfdBal)
print('delta crdfd', curCrdfdBal - prevCrdfdBal)
print('A res delta', prevARes - curARes)
print('B res delta', prevBRes - curBRes)
# alice
assert almostEqual(curLPBal - prevLPBal, lp_want), 'incorrect LP amt'
# staking
assert almostEqual(curLPBal_staking - prevLPBal_staking,
-real_lp_take_amt), 'incorrect staking LP amt'
# spell
assert dfd.balanceOf(balancer_spell) == 0, 'non-zero spell DFD balance'
assert dusd.balanceOf(balancer_spell) == 0, 'non-zero spell DUSD balance'
assert lp.balanceOf(balancer_spell) == 0, 'non-zero spell LP balance'
# check balance and pool reserves
assert almostEqual(curABal - prevABal + real_dfd_repay,
-(curARes - prevARes)), 'inconsistent DFD from withdraw'
assert almostEqual(
curBBal - prevBBal + dusd_repay,
-(curBRes - prevBRes)), 'inconsistent DUSD from withdraw'
curAliceDfdBalance = dfd.balanceOf(alice)
print('Alice dfd balance', curAliceDfdBalance)
receivedDfd = curAliceDfdBalance - prevAliceDfdBalance + real_dfd_repay - (
prevARes - curARes)
print('received dfd', receivedDfd)
# check with staking directly
tx = interface.IStakingRewards(staking).getReward({'from': bob})
receivedDfdFromStaking = dfd.balanceOf(bob) - prevDfd
print('receivedDfdFromStaking', receivedDfdFromStaking)
assert almostEqual(receivedDfd, receivedDfdFromStaking)
#####################################################################################
print(
'========================================================================='
)
print('Case 5. add & remove all LP')
lp_amt = 10 * 10**18
prevLPBal = lp.balanceOf(alice)
tx = homora.execute(
0,
balancer_spell,
balancer_spell.addLiquidityWStakingRewards.encode_input(
lp, # lp token
[
0, # supply DAI
0, # supply WETH
lp_amt, # supply LP
0, # borrow DAI
0, # borrow WETH
0, # borrow LP tokens
0
], # LP desired
wstaking),
{'from': alice})
tx = homora.execute(
2,
balancer_spell,
balancer_spell.removeLiquidityWStakingRewards.encode_input(
lp, # LP token
[
2**256 - 1, # take out LP tokens
lp_amt, # withdraw LP tokens to wallet
0, # repay DAI
0, # repay WETH
0, # repay LP
0, # min DAI
0
], # min WETH
wstaking),
{'from': alice})
curLPBal = lp.balanceOf(alice)
assert prevLPBal == curLPBal, 'incorrect LP Balance'
return tx
def main():
deployer = accounts.at('0xB593d82d53e2c187dc49673709a6E9f806cdC835', force=True)
# deployer = accounts.load('gh')
werc20 = WERC20.deploy({'from': deployer})
wmas = WMasterChef.deploy(MASTERCHEF, {'from': deployer})
wliq = WLiquidityGauge.deploy(CRV_REGISTRY, CRV_TOKEN, {'from': deployer})
wsindex = WStakingRewards.deploy(
'0xB93b505Ed567982E2b6756177ddD23ab5745f309',
'0x4d5ef58aAc27d99935E5b6B4A6778ff292059991', # UNI DPI-WETH
'0x0954906da0Bf32d5479e25f46056d22f08464cab', # INDEX
{'from': deployer},
)
wsperp = WStakingRewards.deploy(
'0xb9840a4a8a671f79de3df3b812feeb38047ce552',
'0xF54025aF2dc86809Be1153c1F20D77ADB7e8ecF4', # BAL PERP-USDC 80-20
'0xbC396689893D065F41bc2C6EcbeE5e0085233447', # PERP
{'from': deployer},
)
kp3r_oracle = ERC20KP3ROracle.deploy(KP3R_NETWORK, {'from': deployer})
core_oracle = CoreOracle.deploy({'from': deployer})
uni_oracle = UniswapV2Oracle.deploy(core_oracle, {'from': deployer})
bal_oracle = BalancerPairOracle.deploy(core_oracle, {'from': deployer})
crv_oracle = CurveOracle.deploy(core_oracle, CRV_REGISTRY, {'from': deployer})
proxy_oracle = ProxyOracle.deploy(core_oracle, {'from': deployer})
core_oracle.setRoute([
'0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE', # ETH
'0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2', # WETH
'0x6b175474e89094c44da98b954eedeac495271d0f', # DAI
'0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48', # USDC
'0xdac17f958d2ee523a2206206994597c13d831ec7', # USDT
'0x2260fac5e5542a773aa44fbcfedf7c193bc2c599', # WBTC
'0x1494ca1f11d487c2bbe4543e90080aeba4ba3c2b', # DPI
'0xbC396689893D065F41bc2C6EcbeE5e0085233447', # PERP
'0xc011a73ee8576fb46f5e1c5751ca3b9fe0af2a6f', # SNX
'0xa478c2975ab1ea89e8196811f51a7b7ade33eb11', # UNI DAI-WETH
'0x0d4a11d5eeaac28ec3f61d100daf4d40471f1852', # UNI WETH-USDT
'0xb4e16d0168e52d35cacd2c6185b44281ec28c9dc', # UNI USDC-WETH
'0xbb2b8038a1640196fbe3e38816f3e67cba72d940', # UNI WBTC-WETH
'0x4d5ef58aac27d99935e5b6b4a6778ff292059991', # UNI DPI-WETH
'0x06da0fd433c1a5d7a4faa01111c044910a184553', # SUSHI WETH-USDT
'0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a', # BAL WETH-DAI 80-20
'0xf54025af2dc86809be1153c1f20d77adb7e8ecf4', # BAL PERP-USDC 80-20
'0x6c3f90f043a72fa612cbac8115ee7e52bde6e490', # CRV 3-POOL
], [
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
kp3r_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
uni_oracle,
bal_oracle,
bal_oracle,
crv_oracle,
], {'from': deployer})
proxy_oracle.setOracles([
'0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2', # WETH
'0x6b175474e89094c44da98b954eedeac495271d0f', # DAI
'0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48', # USDC
'0xdac17f958d2ee523a2206206994597c13d831ec7', # USDT
'0x2260fac5e5542a773aa44fbcfedf7c193bc2c599', # WBTC
'0x1494ca1f11d487c2bbe4543e90080aeba4ba3c2b', # DPI
'0xbC396689893D065F41bc2C6EcbeE5e0085233447', # PERP
'0xc011a73ee8576fb46f5e1c5751ca3b9fe0af2a6f', # SNX
'0xa478c2975ab1ea89e8196811f51a7b7ade33eb11', # UNI DAI-WETH
'0x0d4a11d5eeaac28ec3f61d100daf4d40471f1852', # UNI WETH-USDT
'0xb4e16d0168e52d35cacd2c6185b44281ec28c9dc', # UNI USDC-WETH
'0xbb2b8038a1640196fbe3e38816f3e67cba72d940', # UNI WBTC-WETH
'0x4d5ef58aac27d99935e5b6b4a6778ff292059991', # UNI DPI-WETH
'0x06da0fd433c1a5d7a4faa01111c044910a184553', # SUSHI WETH-USDT
'0x8b6e6e7b5b3801fed2cafd4b22b8a16c2f2db21a', # BAL WETH-DAI 80-20
'0xf54025af2dc86809be1153c1f20d77adb7e8ecf4', # BAL PERP-USDC 80-20
'0x6c3f90f043a72fa612cbac8115ee7e52bde6e490', # CRV 3-POOL
], [
[12500, 8000, 10250],
[10500, 9500, 10250],
[10500, 9500, 10250],
[10500, 9500, 10250],
[12500, 8000, 10250],
[50000, 0, 10250],
[50000, 0, 10250],
[50000, 0, 10250],
[50000, 8000, 10250],
[50000, 8000, 10250],
[50000, 8000, 10250],
[50000, 8000, 10250],
[50000, 6000, 10250],
[50000, 8000, 10250],
[50000, 8000, 10250],
[50000, 0, 10250],
[50000, 9500, 10250],
], {'from': deployer})
proxy_oracle.setWhitelistERC1155(
[werc20, wmas, wliq, wsindex, wsperp],
True,
{'from': deployer},
)
bank = HomoraBank.at('0x5f5Cd91070960D13ee549C9CC47e7a4Cd00457bb')
bank.setOracle(proxy_oracle, {'from': deployer})
wliq.registerGauge(0, 0, {'from': deployer})
uniswap_spell = UniswapV2SpellV1.deploy(
bank, werc20, '0x7a250d5630B4cF539739dF2C5dAcb4c659F2488D',
{'from': deployer},
)
sushiswap_spell = SushiswapSpellV1.deploy(
bank, werc20, '0xd9e1ce17f2641f24ae83637ab66a2cca9c378b9f', wmas,
{'from': deployer},
)
balancer_spell = BalancerSpellV1.deploy(
bank, werc20, '0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2',
{'from': deployer},
)
curve_spell = CurveSpellV1.deploy(
bank, werc20, '0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2', wliq,
{'from': deployer},
)
# register 3pool
crv_oracle.registerPool('0x6c3f90f043a72fa612cbac8115ee7e52bde6e490')
print('DONE')
# pool is lp for balancer
pool = interface.IBalancerPool('0xd8e9690eff99e21a2de25e0b148ffaf47f47c972')
crdfd = MockCErc20.deploy(dfd, {'from': admin})
crdusd = MockCErc20.deploy(dusd, {'from': admin})
werc20 = WERC20.deploy({'from': admin})
staking = accounts.at('0xf068236ecad5fabb9883bbb26a6445d6c7c9a924', force=True)
wstaking = WStakingRewards.deploy(staking, lp, dfd, {'from': admin})
simple_oracle = SimpleOracle.deploy({'from': admin})
simple_oracle.setETHPx([dfd, dusd], [2**112 // 2 // 700, 2**112 * 2 // 700])
balancer_oracle = BalancerPairOracle.deploy(simple_oracle, {'from': alice})
core_oracle = CoreOracle.deploy({'from': admin})
oracle = ProxyOracle.deploy(core_oracle, {'from': admin})
oracle.setWhitelistERC1155([werc20, wstaking], True, {'from': admin})
core_oracle.setRoute(
[dfd, dusd, lp],
[simple_oracle, simple_oracle, balancer_oracle],
{'from': admin},
)
oracle.setOracles(
[dfd, dusd, lp],
[
[10000, 10000, 10000],
[10000, 10000, 10000],
[10000, 10000, 10000],
def main():
deployer = accounts.at('0xB593d82d53e2c187dc49673709a6E9f806cdC835', force=True)
# deployer = accounts.load('gh')
werc20 = WERC20.at('0xe28D9dF7718b0b5Ba69E01073fE82254a9eD2F98')
wmas = WMasterChef.at('0x373ae78a14577682591E088F2E78EF1417612c68')
wliq = WLiquidityGauge.at('0xfdB4f97953150e47C8606758C13e70b5a789a7ec')
wsindex = WStakingRewards.at('0x713df2DDDA9C7d7bDa98A9f8fCd82c06c50fbd90')
wsperp = WStakingRewards.at('0xC4635854480ffF80F742645da0310e9e59795c63')
uni_kp3r_oracle = ERC20KP3ROracle.at('0x8E2A3777AB22e1c5f6d1fF2BcC6C4aA6aB1DeA14')
sushi_kp3r_oracle = ERC20KP3ROracle.deploy(
'0xf67Ab1c914deE06Ba0F264031885Ea7B276a7cDa', {'from': deployer})
core_oracle = CoreOracle.at('0x1E5BDdd0cDF8839d6b27b34927869eF0AD7bf692')
uni_oracle = UniswapV2Oracle.at('0x910aD02D355c2966e8dD8E32C890cD50DB29C3B9')
bal_oracle = BalancerPairOracle.at('0x8F93B0AA2643bf74ab38d6a5910fA82D2da02134')
crv_oracle = CurveOracle.at('0x04DE0E42B3b0483248deafE86C4F5428613b76Ff')
proxy_oracle = ProxyOracle.at('0x43c425fB2b00a991A51b18c217D749E393bF1AB2')
# pair token with oracle
from .tokens import Tokens
token_oracle_settings = [
# base tokens
[Tokens.weth, uni_kp3r_oracle],
[Tokens.dai, uni_kp3r_oracle],
[Tokens.usdc, uni_kp3r_oracle],
[Tokens.usdt, uni_kp3r_oracle],
[Tokens.dpi, uni_kp3r_oracle],
[Tokens.yfi, sushi_kp3r_oracle],
[Tokens.perp, uni_kp3r_oracle],
[Tokens.snx, sushi_kp3r_oracle],
[Tokens.susd, sushi_kp3r_oracle],
[Tokens.uni, uni_kp3r_oracle],
[Tokens.sushi, sushi_kp3r_oracle],
# lp tokens
[Tokens.uni_eth_yfi, uni_oracle],
[Tokens.sushi_eth_yfi, uni_oracle],
[Tokens.uni_eth_dpi, uni_oracle],
[Tokens.crv_3pool, crv_oracle],
[Tokens.sushi_eth_dpi, uni_oracle],
[Tokens.uni_eth_snx, uni_oracle],
[Tokens.uni_eth_susd, uni_oracle],
[Tokens.sushi_eth_susd, uni_oracle],
[Tokens.sushi_eth_snx, uni_oracle],
[Tokens.crv_susd, crv_oracle],
[Tokens.sushi_eth_sushi, uni_oracle],
[Tokens.uni_eth_uni, uni_oracle],
[Tokens.bal_perp_usdc, bal_oracle],
]
token_list, oracle_list = zip(*token_oracle_settings)
# re-route to new uniswap oracle
core_oracle.setRoute(token_list, oracle_list, {'from': deployer})
# re-set oracles
oracle_params = [
# base tokens
[Tokens.weth, [17158, 5828, 12500]],
[Tokens.dai, [10790, 9268, 12500]],
[Tokens.usdc, [10651, 9389, 12500]],
[Tokens.usdt, [10743, 9308, 12500]],
[Tokens.dpi, [16195, 6175, 12500]],
[Tokens.yfi, [29155, 3430, 12500]],
[Tokens.perp, [20042, 4989, 12500]],
[Tokens.snx, [18162, 5506, 12500]],
[Tokens.susd, [12129, 8245, 12500]],
[Tokens.uni, [16022, 6241, 12500]],
[Tokens.sushi, [19418, 5150, 12500]],
# lp tokens
[Tokens.uni_eth_dpi, [17158, 5828, 12500]],
[Tokens.uni_eth_yfi, [29155, 3430, 12500]],
[Tokens.uni_eth_snx, [18162, 5506, 12500]],
[Tokens.uni_eth_susd, [17158, 5828, 12500]],
[Tokens.uni_eth_uni, [17158, 5828, 12500]],
[Tokens.sushi_eth_dpi, [17158, 5828, 12500]],
[Tokens.sushi_eth_yfi, [29155, 3430, 12500]],
[Tokens.sushi_eth_snx, [18162, 5506, 12500]],
[Tokens.sushi_eth_susd, [17158, 5828, 12500]],
[Tokens.sushi_eth_sushi, [19418, 5150, 12500]],
[Tokens.bal_perp_usdc, [20042, 4989, 12500]],
[Tokens.crv_3pool, [10790, 9268, 12500]],
[Tokens.crv_susd, [12129, 8245, 12500]],
]
token_list_2, param_list = zip(*oracle_params)
proxy_oracle.setOracles(token_list_2, param_list, {'from': deployer})
print('DONE')
###########################################################
# test spells (UNCOMMENT TO TEST)
bank = HomoraBank.at('0x5f5Cd91070960D13ee549C9CC47e7a4Cd00457bb')
uniswap_spell = UniswapV2SpellV1.at('0xc671B7251a789de0835a2fa33c83c8D4afB39092')
sushiswap_spell = SushiswapSpellV1.at('0x21Fa95485f4571A3a0d0c396561cF4D8D13D445d')
balancer_spell = BalancerSpellV1.at('0x15B79c184A6a8E19a4CA1F637081270343E4D15D')
curve_spell = CurveSpellV1.at('0x42C750024E02816eE32EB2eB4DA79ff5BF343D30')
