From 03b5efce2b331d907de0bbf38b7e8b72b4e06b34 Mon Sep 17 00:00:00 2001
From: Shriya Tyagi <shriya243@gmail.com>
Date: Tue, 4 Jun 2024 20:01:29 +0530
Subject: [PATCH] feat: add repayOnBehalf

---
 .../mainnet/connectors/compound/v2/main.sol   | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/contracts/mainnet/connectors/compound/v2/main.sol b/contracts/mainnet/connectors/compound/v2/main.sol
index aea631ea..93e089f0 100644
--- a/contracts/mainnet/connectors/compound/v2/main.sol
+++ b/contracts/mainnet/connectors/compound/v2/main.sol
@@ -206,6 +206,46 @@ abstract contract CompoundResolver is Events, Helpers {
         _eventParam = abi.encode(token, cToken, _amt, getId, setId);
     }
 
+    /**
+     * @dev Payback borrowed ETH/ERC20_Token.
+     * @notice Payback debt owed.
+     * @param token The address of the token to payback. (For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
+     * @param cToken The address of the corresponding cToken.
+     * @param amt The amount of the token to payback. (For max: `uint256(-1)`)
+     * @param borrower The address whose debt needs to be repaid.
+     * @param getId ID to retrieve amt.
+     * @param setId ID stores the amount of tokens paid back.
+    */
+    function paybackRawOnBehalf(
+        address token,
+        address cToken,
+        uint256 amt,
+        address borrower,
+        uint256 getId,
+        uint256 setId
+    ) public payable returns (string memory _eventName, bytes memory _eventParam) {
+        uint _amt = getUint(getId, amt);
+
+        require(token != address(0) && cToken != address(0), "invalid token/ctoken address");
+
+        CTokenInterface cTokenContract = CTokenInterface(cToken);
+        _amt = _amt == uint(-1) ? cTokenContract.borrowBalanceCurrent(borrower) : _amt;
+
+        if (token == ethAddr) {
+            require(address(this).balance >= _amt, "not-enough-eth");
+            CETHInterface(cToken).repayBorrowBehalf{value: _amt}(borrower);
+        } else {
+            TokenInterface tokenContract = TokenInterface(token);
+            require(tokenContract.balanceOf(address(this)) >= _amt, "not-enough-token");
+            approve(tokenContract, cToken, _amt);
+            require(cTokenContract.repayBorrowBehalf(borrower, _amt) == 0, "repay-failed.");
+        }
+        setUint(setId, _amt);
+
+        _eventName = "LogPaybackOnBehalf(address,address,uint256,address,uint256,uint256)";
+        _eventParam = abi.encode(token, cToken, _amt, borrower, getId, setId);
+    }
+
     /**
      * @dev Payback borrowed ETH/ERC20_Token using the Mapping.
      * @notice Payback debt owed.
@@ -224,6 +264,26 @@ abstract contract CompoundResolver is Events, Helpers {
         (_eventName, _eventParam) = paybackRaw(token, cToken, amt, getId, setId);
     }
 
+    /**
+     * @dev Payback borrowed ETH/ERC20_Token using the Mapping.
+     * @notice Payback debt owed.
+     * @param tokenId The token id of the token to payback.(For eg: COMP-A)
+     * @param amt The amount of the token to payback. (For max: `uint256(-1)`)
+     * @param borrower The address whose debt needs to be repaid.
+     * @param getId ID to retrieve amt.
+     * @param setId ID stores the amount of tokens paid back.
+    */
+    function paybackOnBehalf(
+        string calldata tokenId,
+        uint256 amt,
+        address borrower,
+        uint256 getId,
+        uint256 setId
+    ) external payable returns (string memory _eventName, bytes memory _eventParam) {
+        (address token, address cToken) = compMapping.getMapping(tokenId);
+        (_eventName, _eventParam) = paybackRawOnBehalf(token, cToken, amt, borrower, getId, setId);
+    }
+
     /**
      * @dev Deposit ETH/ERC20_Token.
      * @notice Same as depositRaw. The only difference is this method stores cToken amount in set ID.