From b1e654558c29d20eadf711d78abee8e6295128ec Mon Sep 17 00:00:00 2001
From: Shriya Tyagi <shriya243@gmail.com>
Date: Sat, 20 Jan 2024 02:16:40 +0400
Subject: [PATCH] setup kyber aggregator on base

---
 .../kyber/kyber_aggregator/events.sol         | 12 ++++
 .../kyber/kyber_aggregator/helpers.sol        | 66 +++++++++++++++++++
 .../kyber/kyber_aggregator/main.sol           | 59 +++++++++++++++++
 3 files changed, 137 insertions(+)
 create mode 100644 contracts/base/connectors/kyber/kyber_aggregator/events.sol
 create mode 100644 contracts/base/connectors/kyber/kyber_aggregator/helpers.sol
 create mode 100644 contracts/base/connectors/kyber/kyber_aggregator/main.sol

diff --git a/contracts/base/connectors/kyber/kyber_aggregator/events.sol b/contracts/base/connectors/kyber/kyber_aggregator/events.sol
new file mode 100644
index 00000000..dcba616e
--- /dev/null
+++ b/contracts/base/connectors/kyber/kyber_aggregator/events.sol
@@ -0,0 +1,12 @@
+//SPDX-License-Identifier: MIT
+pragma solidity ^0.7.0;
+
+contract Events {
+    event LogSwap(
+        address buyToken,
+        address sellToken,
+        uint256 buyAmt,
+        uint256 sellAmt,
+        uint256 setId
+    );
+}
\ No newline at end of file
diff --git a/contracts/base/connectors/kyber/kyber_aggregator/helpers.sol b/contracts/base/connectors/kyber/kyber_aggregator/helpers.sol
new file mode 100644
index 00000000..1b429f2d
--- /dev/null
+++ b/contracts/base/connectors/kyber/kyber_aggregator/helpers.sol
@@ -0,0 +1,66 @@
+//SPDX-License-Identifier: MIT
+pragma solidity ^0.7.0;
+
+import {DSMath} from "../../../common/math.sol";
+import {Basic} from "../../../common/basic.sol";
+import {TokenInterface} from "../../../common/interfaces.sol";
+
+abstract contract Helpers is DSMath, Basic {
+    struct SwapData {
+        TokenInterface sellToken;
+        TokenInterface buyToken;
+        uint256 _sellAmt;
+        uint256 _buyAmt;
+        uint256 unitAmt;
+        bytes callData;
+    }
+
+    address internal constant kyberswap =
+        0x6131B5fae19EA4f9D964eAc0408E4408b66337b5;
+
+    function _swapHelper(SwapData memory swapData, uint256 ethAmt)
+        internal
+        returns (uint256 buyAmt)
+    {
+        TokenInterface buyToken = swapData.buyToken;
+        (uint256 _buyDec, uint256 _sellDec) = getTokensDec(
+            buyToken,
+            swapData.sellToken
+        );
+        uint256 _sellAmt18 = convertTo18(_sellDec, swapData._sellAmt);
+        uint256 _slippageAmt = convert18ToDec(
+            _buyDec,
+            wmul(swapData.unitAmt, _sellAmt18)
+        );
+
+        uint256 initalBal = getTokenBal(buyToken);
+        (bool success, ) = kyberswap.call{value: ethAmt}(swapData.callData);
+        if (!success) revert("kyberswap-failed");
+
+        uint256 finalBal = getTokenBal(buyToken);
+
+        buyAmt = sub(finalBal, initalBal);
+
+        require(_slippageAmt <= buyAmt, "Too much slippage");
+    }
+
+    function _swap(SwapData memory swapData, uint256 setId)
+        internal
+        returns (SwapData memory)
+    {
+        TokenInterface _sellAddr = swapData.sellToken;
+
+        uint256 ethAmt;
+
+        if (address(_sellAddr) == ethAddr) {
+            ethAmt = swapData._sellAmt;
+        } else {
+            approve(TokenInterface(_sellAddr), kyberswap, swapData._sellAmt);
+        }
+        swapData._buyAmt = _swapHelper(swapData, ethAmt);
+
+        setUint(setId, swapData._buyAmt);
+
+        return swapData;
+    }
+}
diff --git a/contracts/base/connectors/kyber/kyber_aggregator/main.sol b/contracts/base/connectors/kyber/kyber_aggregator/main.sol
new file mode 100644
index 00000000..8c85a562
--- /dev/null
+++ b/contracts/base/connectors/kyber/kyber_aggregator/main.sol
@@ -0,0 +1,59 @@
+//SPDX-License-Identifier: MIT
+pragma solidity ^0.7.0;
+
+/**
+ * @title Kyber.
+ * @dev DEX Aggregator.
+ */
+
+import {TokenInterface} from "../../../common/interfaces.sol";
+import {Stores} from "../../../common/stores.sol";
+import {Helpers} from "./helpers.sol";
+
+abstract contract KyberResolver is Helpers {
+    /**
+     * @dev Sell ETH/ERC20_Token using KyberSwap.
+     * @notice Swap tokens from exchanges like kyber, 0x etc, with calculation done off-chain.
+     * @param buyAddr The address of the token to buy.(For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
+     * @param sellAddr The address of the token to sell.(For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
+     * @param sellAmt The amount of the token to sell.
+     * @param unitAmt The amount of buyAmt/sellAmt with slippage.
+     * @param callData Data from kyberswap API.
+     * @param setId ID stores the amount of token brought.
+     */
+    function swap(
+        address buyAddr,
+        address sellAddr,
+        uint256 sellAmt,
+        uint256 unitAmt,
+        bytes calldata callData,
+        uint256 setId
+    )
+        external
+        payable
+        returns (string memory _eventName, bytes memory _eventParam)
+    {
+        Helpers.SwapData memory swapData = Helpers.SwapData({
+            buyToken: TokenInterface(buyAddr),
+            sellToken: TokenInterface(sellAddr),
+            unitAmt: unitAmt,
+            callData: callData,
+            _sellAmt: sellAmt,
+            _buyAmt: 0
+        });
+        swapData = _swap(swapData, setId);
+
+        _eventName = "LogSwap(address,address,uint256,uint256,uint256)";
+        _eventParam = abi.encode(
+            address(swapData.buyToken),
+            address(swapData.sellToken),
+            swapData._buyAmt,
+            swapData._sellAmt,
+            setId
+        );
+    }
+}
+
+contract ConnectV2KyberAggregatorBase is KyberResolver {
+    string public name = "Kyber-aggregator-v1.0";
+}