Skip to content

Commit

Permalink
feat: rebalanceReserves
Browse files Browse the repository at this point in the history
  • Loading branch information
@pegahcarter
pegahcarter committed Nov 13, 2024
1 parent bc36f7e commit c6a9c0a
Showing 1 changed file with 83 additions and 12 deletions.
95 changes: 83 additions & 12 deletions src/contracts/amos/FraxtalLZCurveAMO.sol
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,14 +28,19 @@ contract FraxtalLZCurveAMO is AccessControlUpgradeable, FraxtalConstants {
using OptionsBuilder for bytes;

bytes32 public constant EXCHANGE_ROLE = keccak256("EXCHANGE_ROLE");
bytes32 public constant SEND_ROLE = keccak256("SEND_ROLE");
uint256 public constant ONE_HUNDRED_PCT = 10000;

// keccak256(abi.encode(uint256(keccak256("frax.storage.LZCurveAmoStorage")) - 1));
bytes32 private constant LZCurveAmoStorageLocation =
0x34cfa87765bced8684ef975fad48f7c370ba6aca6fca817512efcf044977addf;
struct LZCurveAmoStorage {
address ethereumComposer;
address ethereumLzSenderAmo;
uint256 fraxPct;
uint256 sFraxPct;
uint256 sFrxEthPct;
uint256 fxsPct;
uint256 fpiPct;
}
function _getLZCurveAmoStorage() private pure returns (LZCurveAmoStorage storage $) {
assembly {
Expand All @@ -50,7 +55,6 @@ contract FraxtalLZCurveAMO is AccessControlUpgradeable, FraxtalConstants {
function initialize(address _owner) external initializer {
_grantRole(DEFAULT_ADMIN_ROLE, _owner);
_grantRole(EXCHANGE_ROLE, _owner);
_grantRole(SEND_ROLE, _owner);
}

function setStorage(address _ethereumComposer, address _ethereumLzSenderAmo) external onlyRole(DEFAULT_ADMIN_ROLE) {
Expand All @@ -77,18 +81,62 @@ contract FraxtalLZCurveAMO is AccessControlUpgradeable, FraxtalConstants {
IERC20(nToken).approve(curve, _amountIn);
amountOut = ICurve(curve).exchange({ i: int128(0), j: int128(1), _dx: _amountIn, _min_dy: _amountOutMin });
}
}

function rebalanceReserves() external {
_rebalanceReserves(FraxtalConstants.fraxOft);
_rebalanceReserves(FraxtalConstants.sFraxOft);
_rebalanceReserves(FraxtalConstants.sFrxEthOft);
_rebalanceReserves(FraxtalConstants.fxsOft);
_rebalanceReserves(FraxtalConstants.fpiOft);
}

function _rebalanceReserves(address _oApp) internal {
LZCurveAmoStorage storage $ = _getLZCurveAmoStorage();

(address nToken, ) = _getRespectiveTokens(_oApp);
uint256 pct;
if (nToken == FraxtalConstants.frax) {
pct = $.fraxPct;
} else if (nToken == FraxtalConstants.sFrax) {
pct = $.sFraxPct;
} else if (nToken == FraxtalConstants.sFrxEth) {
pct = $.sFrxEthPct;
} else if (nToken == FraxtalConstants.fxs) {
pct = $.fxsPct;
} else if (nToken == FraxtalConstants.fpi) {
pct = $.fpiPct;
}

// TODO: now what
uint256 balanceNative = IERC20(nToken).balanceOf(address(this));
uint256 balanceLz = IERC20(_oApp).balanceOf(address(this));
bool excessNative = balanceNative > balanceLz;
uint256 delta = excessNative ? balanceNative - balanceLz : balanceLz - balanceNative;
uint256 deltaPct = (ONE_HUNDRED_PCT * delta) / (balanceNative + balanceLz);

if (deltaPct > pct) {
// divide the difference by 2 to get the amount needed to equally weight the tokens
// For example, nToken/lzToken balance of 60/40 would have delta = 20, and require
// rebalancing 10 units back to 50/50
uint256 amount = delta / 2;
if (excessNative) {
_sendViaFerry({ _oApp: _oApp, _amount: amount });
} else {
_sendViaLz({ _oApp: _oApp, _amount: amount });
}
}
}

function sendToAdapterAndBridgeBackNatively(address _oApp, uint256 _amount) external onlyRole(SEND_ROLE) {
function _sendViaLz(address _oApp, uint256 _amount) internal {
bytes memory options = OptionsBuilder.newOptions().addExecutorLzComposeOption(0, 100_000, 0);
bytes memory composeMsg = abi.encode(uint256(0));
// Round down to avoid dust loss in send
uint256 amountRounded = (_amount / 1e13) * 1e13;
SendParam memory sendParam = SendParam({
dstEid: uint32(30101), // Ethereum
to: bytes32(uint256(uint160(ethereumComposer()))),
amountLD: _amount,
minAmountLD: 0,
amountLD: amountRounded,
minAmountLD: amountRounded,
extraOptions: options,
composeMsg: composeMsg,
oftCmd: ""
Expand All @@ -97,23 +145,46 @@ contract FraxtalLZCurveAMO is AccessControlUpgradeable, FraxtalConstants {
IOFT(_oApp).send{ value: fee.nativeFee }(sendParam, fee, payable(address(this)));
}

function sendToFerry(address _oApp, uint256 _amount) external onlyRole(SEND_ROLE) {
function _sendViaFerry(address _oApp, uint256 _amount) internal {
(address nToken, ) = _getRespectiveTokens(_oApp);
address ferry;
if (nToken == FraxtalL2.FRAX) {
if (nToken == FraxtalConstants.frax) {
ferry = FraxtalL2.FRAXFERRY_ETHEREUM_FRAX;
} else if (nToken == FraxtalL2.SFRAX) {
} else if (nToken == FraxtalConstants.sFrax) {
ferry = FraxtalL2.FRAXFERRY_ETHEREUM_SFRAX;
} else if (nToken == FraxtalL2.SFRXETH) {
} else if (nToken == FraxtalConstants.sFrxEth) {
ferry = FraxtalL2.FRAXFERRY_ETHEREUM_SFRXETH;
} else if (nToken == FraxtalL2.FXS) {
} else if (nToken == FraxtalConstants.fxs) {
ferry = FraxtalL2.FRAXFERRY_ETHEREUM_FXS;
} else if (nToken == FraxtalL2.FPI) {
} else if (nToken == FraxtalConstants.fpi) {
ferry = FraxtalL2.FRAXFERRY_ETHEREUM_FPI;
}
IFerry(ferry).embarkWithRecipient({ amount: _amount, recipient: ethereumLzSenderAmo() });
}

function setPcts(
uint256 _fraxPct,
uint256 _sFraxPct,
uint256 _sFrxEthPct,
uint256 _fxsPct,
uint256 _fpiPct
) external onlyRole(DEFAULT_ADMIN_ROLE) {
require(
_fraxPct < ONE_HUNDRED_PCT &&
_sFraxPct < ONE_HUNDRED_PCT &&
_sFrxEthPct < ONE_HUNDRED_PCT &&
_fxsPct < ONE_HUNDRED_PCT &&
_fpiPct < ONE_HUNDRED_PCT,
"Exceeds 100 pct"
);
LZCurveAmoStorage storage $ = _getLZCurveAmoStorage();
$.fraxPct = _fraxPct;
$.sFraxPct = _sFraxPct;
$.sFrxEthPct = _sFrxEthPct;
$.fxsPct = _fxsPct;
$.fpiPct = _fpiPct;
}

function ethereumComposer() public view returns (address) {
LZCurveAmoStorage storage $ = _getLZCurveAmoStorage();
return $.ethereumComposer;
Expand Down

0 comments on commit c6a9c0a

Please sign in to comment.