@pendle/core-v2 2.18.1 → 2.19.0-tradingbot

package/build/artifacts/contracts/router/swap-aggregator/kyberswap/{arbitrum/IMetaAggregationRouterV2.sol → IMetaAggregationRouterV2.sol}/IMetaAggregationRouterV2.json

@@ -1,7 +1,7 @@
 {
   "_format": "hh-sol-artifact-1",
   "contractName": "IMetaAggregationRouterV2",
-  "sourceName": "contracts/router/swap-aggregator/kyberswap/arbitrum/IMetaAggregationRouterV2.sol",
+  "sourceName": "contracts/router/swap-aggregator/kyberswap/IMetaAggregationRouterV2.sol",
   "abi": [
     {
       "inputs": [

package/build/artifacts/contracts/router/swap-aggregator/kyberswap/KyberInputScalingHelper.sol/KyberInputScalingHelper.dbg.json

@@ -0,0 +1,4 @@
+{
+  "_format": "hh-sol-dbg-1",
+  "buildInfo": "../../../../../build-info/036d718d54b0087f8ec1f7e236bdcbda.json"
+}

package/build/artifacts/contracts/router/swap-aggregator/kyberswap/{arbitrum/KyberHelperArbitrum.sol/KyberHelperArbitrum.json → KyberInputScalingHelper.sol/KyberInputScalingHelper.json}

@@ -1,7 +1,7 @@
 {
   "_format": "hh-sol-artifact-1",
-  "contractName": "KyberHelperArbitrum",
-  "sourceName": "contracts/router/swap-aggregator/kyberswap/arbitrum/KyberHelperArbitrum.sol",
+  "contractName": "KyberInputScalingHelper",
+  "sourceName": "contracts/router/swap-aggregator/kyberswap/KyberInputScalingHelper.sol",
   "abi": [],
   "bytecode": "0x",
   "deployedBytecode": "0x",

package/build/artifacts/contracts/router/swap-aggregator/kyberswap/ScalingDataLib.sol/ScalingDataLib.dbg.json

@@ -0,0 +1,4 @@
+{
+  "_format": "hh-sol-dbg-1",
+  "buildInfo": "../../../../../build-info/036d718d54b0087f8ec1f7e236bdcbda.json"
+}

package/build/artifacts/contracts/router/swap-aggregator/kyberswap/ScalingDataLib.sol/ScalingDataLib.json

@@ -0,0 +1,10 @@
+{
+  "_format": "hh-sol-artifact-1",
+  "contractName": "ScalingDataLib",
+  "sourceName": "contracts/router/swap-aggregator/kyberswap/ScalingDataLib.sol",
+  "abi": [],
+  "bytecode": "0x60566037600b82828239805160001a607314602a57634e487b7160e01b600052600060045260246000fd5b30600052607381538281f3fe73000000000000000000000000000000000000000030146080604052600080fdfea2646970667358221220cfba1793d0bf2005e9ec0047b2572c3062305c74344dcf37d39ad9fc7600e96b64736f6c63430008110033",
+  "deployedBytecode": "0x73000000000000000000000000000000000000000030146080604052600080fdfea2646970667358221220cfba1793d0bf2005e9ec0047b2572c3062305c74344dcf37d39ad9fc7600e96b64736f6c63430008110033",
+  "linkReferences": {},
+  "deployedLinkReferences": {}
+}

package/contracts/core/TradingBot/LongYield/LongYieldDecisionHelper.sol

@@ -0,0 +1,216 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+pragma solidity ^0.8.0;
+
+import "./../libraries/BotSimulationLib.sol";
+import "../../libraries/Errors.sol";
+import "../libraries/SpecSegmentLib.sol";
+import "../base/BotDecisionHelperBase.sol";
+import "../../../interfaces/TradingBot/IBotDecisionHelper.sol";
+import "../../../interfaces/TradingBot/ILongYieldTradingBot.sol";
+
+contract LongYieldDecisionHelper is BotDecisionHelperBase {
+    using MarketExtLib for MarketExtState;
+    using MarketExtLib for ApproxParams;
+    using MarketMathCore for MarketState;
+    using MarketApproxPtInLib for MarketState;
+    using BotSimulationLib for BotState;
+    using PYIndexLib for PYIndex;
+    using Math for uint256;
+    using Math for int256;
+    using LogExpMath for uint256;
+    using SpecSegmentLib for TradingSpecs;
+
+    // We dont care about AddLiqKeepYt here
+    function getActionDetails(address botAddress) public returns (ActionToTakeResult memory res) {
+        StrategyData memory strategyData = ILongYieldTradingBot(botAddress).readStrategyData();
+        uint256 iy = strategyData.marketExt.impliedYield();
+
+        res.currentBin = _getCurrentBin(strategyData.botState, strategyData.specs, iy);
+        if (res.currentBin == 0) {
+            res.action = ActionType.NONE;
+            return res;
+        }
+        res.targetIy = _getTargetImpliedYieldForBin(
+            strategyData.botState,
+            strategyData.specs,
+            res.currentBin
+        );
+
+        uint256 maxAmountForAction = INF;
+        uint256 rebalanceAdditionalCheck;
+
+        if (res.currentBin < 0) {
+            res.action = ActionType.RemoveLiqToYt;
+            maxAmountForAction = _calcMaxAmountForAction(
+                strategyData.botState.lpBalance,
+                res.currentBin,
+                strategyData.botState.buyBins,
+                strategyData.specs.numOfBins
+            );
+
+            if (maxAmountForAction == 0) {
+                return res;
+            }
+        } else {
+            res.action = ActionType.AddLiqFromYt;
+            uint256 maxYtToAddLiq = _searchMaxYtToAddLiq(strategyData);
+
+            rebalanceAdditionalCheck = _calcMaxAmountForAction(
+                maxYtToAddLiq,
+                res.currentBin,
+                strategyData.botState.buyBins,
+                strategyData.specs.numOfBins
+            );
+
+            if (rebalanceAdditionalCheck == 0) {
+                return res;
+            }
+        }
+
+        res.guessBotParams = searchForBotParam(
+            strategyData,
+            res.action,
+            res.targetIy,
+            rebalanceAdditionalCheck,
+            _getMockApproxParamsWithGuessMax(maxAmountForAction),
+            _getMockApproxParams()
+        );
+        (res.amountOut, res.guessIntParams) = _calcAmountOutAndIntParams(
+            strategyData.marketExt,
+            res.action,
+            res.guessBotParams
+        );
+    }
+
+    /**
+     * Given the rebalancing action and its `botParam`, binary search for the optimal internal
+     * parameter `intParam` for the internal action's approximation (e.g. for router's swap)
+     */
+    function _calcAmountOutAndIntParams(
+        MarketExtState memory marketExt,
+        ActionType action,
+        uint256 botParam
+    ) public pure returns (uint256 amountOut, uint256 intParams) {
+        if (action == ActionType.RemoveLiqToYt) {
+            (amountOut, intParams) = marketExt.clone().removeLiqToYt(
+                botParam,
+                _getMockApproxParams()
+            );
+        } else {
+            (amountOut, ) = marketExt.clone().addLiqFromYt(botParam);
+        }
+    }
+
+    function _searchMaxYtToAddLiq(
+        StrategyData memory strategyData
+    ) internal pure returns (uint256) {
+        (, , , uint256 ytPtRatioAfter) = _calcBotExtStats(
+            strategyData.botState,
+            strategyData.marketExt
+        );
+        if (ytPtRatioAfter < strategyData.specs.targetYtPtRatio) {
+            return 0;
+        }
+
+        ApproxParams memory guessParams = _getMockApproxParamsWithGuessMax(
+            strategyData.botState.ytBalance
+        );
+        ApproxParams memory mockSwapParams = _getMockApproxParams();
+        while (guessParams.guessMax > guessParams.guessMin + 1) {
+            uint256 guess = guessParams.guessMin +
+                (guessParams.guessMax - guessParams.guessMin) /
+                2;
+
+            (
+                bool success,
+                BotState memory botStateAfter,
+                MarketExtState memory marketExtAfter
+            ) = previewAction(strategyData, ActionType.AddLiqFromYt, guess, mockSwapParams);
+
+            if (!success) {
+                guessParams.guessMax = guess - 1;
+                continue;
+            }
+
+            (, , , ytPtRatioAfter) = _calcBotExtStats(botStateAfter, marketExtAfter);
+            if (ytPtRatioAfter >= strategyData.specs.targetYtPtRatio) {
+                guessParams.guessMin = guess;
+            } else {
+                guessParams.guessMax = guess - 1;
+            }
+        }
+        return guessParams.guessMin;
+    }
+
+    /// @dev Does not modify params, structs are cloned before simulation
+    function previewAction(
+        StrategyData memory strategyData,
+        ActionType action,
+        uint256 botParam,
+        ApproxParams memory intParams
+    ) public pure returns (bool success, BotState memory bot, MarketExtState memory marketExt) {
+        bot = strategyData.botState.clone();
+        marketExt = strategyData.marketExt.clone();
+
+        if (action == ActionType.AddLiqFromYt) {
+            success = bot.addLiqFromYt(marketExt, botParam);
+        } else if (action == ActionType.RemoveLiqToYt) {
+            success = bot.removeLiqToYt(marketExt, botParam, intParams);
+        } else {
+            success = false;
+        }
+    }
+
+    /// @dev Gives a weak upper bound so that MarketExtLib's simulation doesn't revert
+    function _weakUpperBound(
+        BotState memory bot,
+        MarketExtState memory marketExt,
+        ActionType action
+    ) internal pure override returns (uint256) {
+        if (action == ActionType.AddLiqFromYt) {
+            return marketExt.calcMaxPtOut();
+        } else {
+            return bot.lpBalance;
+        }
+    }
+
+    function _rebalanceResult(
+        RebalanceDetails memory rebalance,
+        uint256 botParam,
+        uint256 rebalanceAdditionalCheck,
+        uint256 eps,
+        ApproxParams memory intParams
+    ) internal pure override returns (RebalanceResult) {
+        (
+            bool success,
+            BotState memory newBotState,
+            MarketExtState memory marketExt
+        ) = previewAction(rebalance.strategyData, rebalance.action, botParam, intParams);
+
+        if (!success) return RebalanceResult.FAIL_OR_TARGET_EXCEED;
+        // return RebalanceResult.SUCCESS;
+
+        uint256 iy = marketExt.impliedYield();
+        if (rebalance.action == ActionType.AddLiqFromYt) {
+            uint256 netYtSwapped = rebalance.strategyData.botState.ytBalance -
+                newBotState.ytBalance;
+
+            if (netYtSwapped > rebalanceAdditionalCheck) {
+                return RebalanceResult.FAIL_OR_TARGET_EXCEED;
+            }
+            if (Math.isAGreaterApproxB(iy, rebalance.targetIy, eps)) {
+                return RebalanceResult.SUCCESS;
+            } else {
+                if (iy < rebalance.targetIy) return RebalanceResult.FAIL_OR_TARGET_EXCEED;
+                else return RebalanceResult.TARGET_NOT_REACHED;
+            }
+        } else {
+            if (Math.isASmallerApproxB(iy, rebalance.targetIy, eps)) {
+                return RebalanceResult.SUCCESS;
+            } else {
+                if (iy > rebalance.targetIy) return RebalanceResult.FAIL_OR_TARGET_EXCEED;
+                else return RebalanceResult.TARGET_NOT_REACHED;
+            }
+        }
+    }
+}

package/contracts/core/TradingBot/LongYield/LongYieldTradingBot.sol

@@ -0,0 +1,97 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+pragma solidity 0.8.17;
+
+import "../base/TradingBotBase.sol";
+import "../base/BotActionHelper.sol";
+import "../../../interfaces/TradingBot/ILongYieldTradingBot.sol";
+import "@openzeppelin/contracts/proxy/utils/UUPSUpgradeable.sol";
+
+contract LongYieldTradingBot is
+    TradingBotBase,
+    BotActionHelper,
+    ILongYieldTradingBot,
+    UUPSUpgradeable
+{
+    using MarketExtLib for MarketExtState;
+    using Math for uint256;
+    using Math for int256;
+
+    constructor(
+        address _market,
+        address _router,
+        address _PENDLE,
+        address _decisionHelper
+    ) TradingBotBase(_market, _router, _PENDLE, _decisionHelper) {}
+
+    function _authorizeUpgrade(address) internal override onlyOwner {}
+
+    function initialize(TradingSpecs memory _specs) external initializer {
+        _setSpecs(_specs);
+        __BoringOwnable_init();
+    }
+
+    function addLiqKeepYt(
+        uint256 netSyIn,
+        uint256 minLpOut,
+        uint256 minYtOut
+    ) external onlyOwner returns (uint256 netLpOut, uint256 netYtOut) {
+        (netLpOut, netYtOut) = _addLiqKeepYt(router, netSyIn, minLpOut, minYtOut);
+        emit AddLiqKeepYt(netSyIn, netLpOut, netYtOut);
+    }
+
+    function addLiqFromYt(
+        AddLiqFromYtInput calldata inputParams
+    ) external onlyOwner returns (uint256 netPtFromSwap, uint256 netLpOut, uint256 netYtIn) {
+        StrategyData memory strategyData = readStrategyData();
+
+        require(
+            IBotDecisionHelper(decisionHelper).getCurrentBin(strategyData) ==
+                inputParams.currentBin,
+            "addLiqFromYt: bin state expired"
+        );
+        netPtFromSwap = decisionHelper.searchForBotParam(
+            readStrategyData(),
+            ActionType.AddLiqFromYt,
+            inputParams.targetIy,
+            type(uint256).max,
+            inputParams.botParams,
+            inputParams.botParams // only placeholder, value doesn't matter
+        );
+        (netLpOut, netYtIn) = _addLiqFromYt(netPtFromSwap, inputParams.minLpOut);
+
+        _updateSellBinAfterTrade();
+
+        emit AddLiqFromYt(inputParams.botParams, netYtIn, netLpOut);
+    }
+
+    function removeLiqToYt(
+        RemoveLiqToYtInput calldata inputParams
+    ) external onlyOwner returns (uint256 netLpRemoved, uint256 netYtOut) {
+        StrategyData memory strategyData = readStrategyData();
+
+        require(
+            IBotDecisionHelper(decisionHelper).getCurrentBin(strategyData) ==
+                inputParams.currentBin,
+            "removeLiqToYt: bin state expired"
+        );
+
+        netLpRemoved = decisionHelper.searchForBotParam(
+            strategyData,
+            ActionType.RemoveLiqToYt,
+            inputParams.targetIy,
+            type(uint256).max,
+            inputParams.botParams,
+            inputParams.guessTotalPtToSwap // value DOES matter
+        );
+
+        (netYtOut, ) = _removeLiqToYt(
+            netLpRemoved,
+            inputParams.guessTotalPtToSwap,
+            inputParams.minYtOut
+        );
+
+        _updateSellBinAfterTrade();
+
+        emit RemoveLiqToYt(inputParams.botParams, netLpRemoved, netYtOut);
+    }
+}

package/contracts/core/TradingBot/README.md

@@ -0,0 +1,153 @@
+# Trading Bot
+
+## Structs and enums
+
+### MarketExtState
+
+Struct for external market state.
+
+- `state`: MarketState
+- `index`: YT's index
+- `blockTime`: timestamp
+
+### BotState
+
+Struct for bot's balances: `lpBalance`, `ytBalance`, `syBalance`.
+
+### TradingSpecs
+
+Struct for bot's trading limits:
+
+- `lowerIyLimit`: apeYtApy
+- `upperIyLimit`: sellYtApy
+- `floatingSyRatioLimit`: h_r - f/TVL limit, TVL calculated in number of SY.
+- `floatingSyRatioTarget`: l_r - f/TVL target.
+- `ytPtRatioLimit`: h_x - limit for YT/PT
+- `ytPtRatioTarget`: l_x - target for YT/PT
+
+### ActionType
+
+Enum for action.
+
+- 0: SwapSyToYt - Buy YT with SY
+- 1: AddLiqKeepYt - ZPI Zap
+- 2: AddLiqFromSy - Single sided LP
+- 3: AddLiqFromYt - Sell YT for LP
+- 4: RemoveLiqToYt - Zap out to YT
+- 5: NONE
+
+### StrategyFlag
+
+Enum for strategy state.
+
+- 0: GOOD
+- 1: TOO_LOW
+- 2: TOO_HIGH
+
+### StrategyState
+
+Struct for strategy state.
+
+- `iyFlag`: StrategyFlag for IY compared to ape/sellYtApy
+- `syRatioFlag`: StrategyFlag for f/TVL compared to given limit \(r\)
+
+## TradingBotBase.sol
+
+### 1. depositSy
+
+Deposit SY into the bot.
+- SY approval needed.
+
+### 2. depositToken
+
+Deposit funds into the bot, which are then compounded into SY.
+- `inp`: Token input data to swap deposited token to SY. 
+- If `inp.netTokenIn` is set to `2**256 - 1` then it will be set to `_selfBalance()` internally.`
+- Needs sender's approval to bot & bot's approval to router.
+
+### 3. withdrawFunds
+
+Withdraw funds back to sender (only owner).
+
+### 4. approveInf
+
+Infinite approve token to spender.
+
+### 5. claimAndCompound
+
+Claim rewards from SY, YT, market. Then compound the rewards into SY.
+
+- `inps`: Token input data to swap claimed rewards into SY
+- If `inp.netTokenIn` is set to `2**256 - 1` then it will be set to `_selfBalance()` internally.
+- Needs sender's approval to bot & bot's approval to router for tokens.
+
+### 6. claimWithoutCompound
+Intended to be static called from offchain. Claim rewards from SY, YT, market, then return amounts of tokens claimed.
+
+### 7. readMarketExtState
+Return the current `MarketExtState`: state of market, YT's index, current block timestamp.
+
+### 8. readBotState
+Return the current `BotState`: The bot's balance in LP, SY, YT.
+
+
+## LongYieldTradingBot.sol
+
+There are 5 functions for rebalancing actions.
+- For all of them, `botParams` is ApproxParams to find the optimal internal trade parameter.
+- For functions interacting with the router, approval is needed.
+
+### 1. setSpecs
+
+Set the bot's specs: apeYtApy, sellYtApy, h_r (f/TVL limit), l_r (f/TVL target), h_x (YT/PT ratio limit), l_x (YT/PT ratio target).
+
+### 2. swapSyToYt
+
+Rebalance by buying YT with SY. 
+
+### 3. addLiqKeepYt
+
+Rebalance by ZPI zap. 
+
+### 4. addLiqFromSy
+
+Rebalance by adding single sided LP from SY.
+
+### 5. addLiqFromYt
+
+Rebalance by selling YT to LP. 
+
+### 6. removeLiqToYt
+
+Rebalance by zapping out LP to YT.
+
+- `intParams`: ApproxParams for internal approx function (approxSwapExactPtForYt).
+- Zapping LP to YT can lose a few percent of the `tvlInSy`, the estimated TVL of bot in SY. Zapping out too much can cause a significant change in `tvlInSy`, which can unintentionally increase the estimated f/TVL.
+
+## BotDecisionHelper.sol
+
+Offchain helper for trading bot.
+
+### 1. searchForBotParam
+Binary search for the optimal `botParam` for the rebalancing action. To be used in bot's actual rebalancing actions.
+
+- Searches for the minimum `botParam` so that the bot trades just "target" is met.
+- Target to meet is calculated based on the `action` (desired f/TVL, YT/PT, IY).
+- `botParam`: The parameter for the rebalancing function (e.g. `netPtFromSwap` for addLiqFromSy).
+
+### 2. searchForIntParam
+(Only useful for `removeLiqToYt`) 
+Given `botParam`, binary search for the optimal `intParam` for the internal calculations (e.g. `approxSwapExactPtForYt`).
+
+### 3. strategyState
+Calculate the strategy state from BotState, MarketExtState, TradingSpecs.
+
+### 4. actionToTake
+Return optimal action to take based on the BotState and StrategyState.
+
+### 5. previewAction
+Simulate rebalancing action (swap/add/remove) from BotState and MarketExtState.
+
+### 6. botExtStats
+Calculate the external stats for the bot: Estimated TVL in equivalent SY amount, current IY, current f/TVL ratio, current YT/PT ratio
+