fix: infinifi minor fixes

Author: Van0k

contracts/adapters/infinifi/InfinifiGatewayAdapter.sol

@@ -39,6 +39,9 @@ contract InfinifiGatewayAdapter is AbstractAdapter, IInfinifiGatewayAdapter {
     /// @notice The mapping of unwinding epochs to locked tokens
     mapping(uint32 => address) public unwindingEpochToLockedToken;
 
+    /// @notice The mapping of locked tokens to unwinding epochs
+    mapping(address => uint32) public lockedTokenToUnwindingEpoch;
+
     /// @notice Constructor
     /// @param _creditManager Credit manager address
     /// @param _infinifiGateway Infinifi Gateway address
@@ -235,8 +238,10 @@ contract InfinifiGatewayAdapter is AbstractAdapter, IInfinifiGatewayAdapter {
                 _allowedLockedTokens.add(lockedTokens[i].lockedToken);
                 _getMaskOrRevert(lockedTokens[i].lockedToken);
                 unwindingEpochToLockedToken[lockedTokens[i].unwindingEpochs] = lockedTokens[i].lockedToken;
+                lockedTokenToUnwindingEpoch[lockedTokens[i].lockedToken] = lockedTokens[i].unwindingEpochs;
             } else {
                 _allowedLockedTokens.remove(lockedTokens[i].lockedToken);
+                delete lockedTokenToUnwindingEpoch[lockedTokens[i].lockedToken];
             }
 
             emit SetLockedTokenStatus(

contracts/adapters/infinifi/InfinifiUnwindingGatewayAdapter.sol

@@ -36,6 +36,9 @@ contract InfinifiUnwindingGatewayAdapter is AbstractAdapter, IInfinifiUnwindingG
     /// @notice The mapping of unwinding epochs to locked tokens
     mapping(uint32 => address) public unwindingEpochToLockedToken;
 
+    /// @notice The mapping of locked tokens to unwinding epochs
+    mapping(address => uint32) public lockedTokenToUnwindingEpoch;
+
     address public immutable infinifiUnwindingPhantomToken;
 
     constructor(address _creditManager, address _infinifiUnwindingGateway, address _infinifiUnwindingPhantomToken)
@@ -88,7 +91,7 @@ contract InfinifiUnwindingGatewayAdapter is AbstractAdapter, IInfinifiUnwindingG
 
     function setLockedTokenBatchStatus(LockedTokenStatus[] calldata lockedTokens) external configuratorOnly {
         uint256 len = lockedTokens.length;
-        address lockingController = IInfinifiGateway(targetContract).getAddress("lockingController");
+        address lockingController = IInfinifiUnwindingGateway(targetContract).lockingController();
         for (uint256 i; i < len; ++i) {
             if (
                 IInfinifiLockingController(lockingController).shareToken(lockedTokens[i].unwindingEpochs)
@@ -101,8 +104,10 @@ contract InfinifiUnwindingGatewayAdapter is AbstractAdapter, IInfinifiUnwindingG
                 _allowedLockedTokens.add(lockedTokens[i].lockedToken);
                 _getMaskOrRevert(lockedTokens[i].lockedToken);
                 unwindingEpochToLockedToken[lockedTokens[i].unwindingEpochs] = lockedTokens[i].lockedToken;
+                lockedTokenToUnwindingEpoch[lockedTokens[i].lockedToken] = lockedTokens[i].unwindingEpochs;
             } else {
                 _allowedLockedTokens.remove(lockedTokens[i].lockedToken);
+                delete lockedTokenToUnwindingEpoch[lockedTokens[i].lockedToken];
             }
 
             emit SetLockedTokenStatus(

contracts/helpers/infinifi/InfinifiUnwindingGateway.sol

@@ -36,8 +36,6 @@ contract InfinifiUnwindingGateway is IInfinifiUnwindingGateway {
 
     mapping(address => UserUnwindingData) public userToUnwindingData;
 
-    mapping(address => uint256) public userToTotalRedemptions;
-
     constructor(address _infinifiGateway) {
         infinifiGateway = _infinifiGateway;
         iUSD = IInfinifiGateway(infinifiGateway).getAddress("receiptToken");

contracts/integrations/infinifi/IInfinifiGateway.sol

@@ -31,12 +31,18 @@ interface IInfinifiGateway {
 
 interface IInfinifiMintController {
     function receiptToken() external view returns (address);
+
+    function assetToReceipt(uint256 _assetAmount) external view returns (uint256);
 }
 
 interface IInfinifiLockingController {
+    function getEnabledBuckets() external view returns (uint32[] memory);
+
     function shareToken(uint32 unwindingEpochs) external view returns (address);
 
     function unwindingModule() external view returns (address);
+
+    function exchangeRate(uint32 unwindingEpochs) external view returns (uint256);
 }
 
 interface IInfinifiUnwindingModule {
@@ -47,4 +53,10 @@ interface IInfinifiUnwindingModule {
 
 interface IInfinifiRedeemController {
     function receiptToAsset(uint256 amount) external view returns (uint256);
+
+    function accounting() external view returns (address);
+}
+
+interface IInfinifiAccounting {
+    function totalAssetsOf(address asset, uint256 farmType) external view returns (uint256);
 }

contracts/interfaces/infinifi/IInfinifiGatewayAdapter.sol

@@ -25,6 +25,12 @@ interface IInfinifiGatewayExceptions {
 
 /// @title Infinifi Gateway adapter interface
 interface IInfinifiGatewayAdapter is IAdapter, IInfinifiGatewayExceptions, IInfinifiGatewayEvents {
+    function usdc() external view returns (address);
+
+    function iusd() external view returns (address);
+
+    function siusd() external view returns (address);
+
     function mint(address to, uint256 amount) external returns (bool useSafePrices);
 
     function mintDiff(uint256 leftoverAmount) external returns (bool useSafePrices);
@@ -47,5 +53,7 @@ interface IInfinifiGatewayAdapter is IAdapter, IInfinifiGatewayExceptions, IInfi
 
     function claimRedemption() external returns (bool useSafePrices);
 
+    function getAllowedLockedTokens() external view returns (address[] memory);
+
     function setLockedTokenBatchStatus(LockedTokenStatus[] calldata lockedTokens) external;
 }