replace safe-hts association to IHRC

web3-nomad · Jun 11, 2024 · 46ab9b2 · 46ab9b2
1 parent d714b77
commit 46ab9b2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/contracts/common/hedera/IHRC.sol b/contracts/common/hedera/IHRC.sol
@@ -0,0 +1,8 @@
+// SPDX-License-Identifier: Apache-2.0
+pragma solidity ^0.8.7;
+
+interface IHRC {
+    function associate() external returns (uint256 responseCode);
+
+    function dissociate() external returns (uint256 responseCode);
+}
diff --git a/contracts/erc4626/Vault.sol b/contracts/erc4626/Vault.sol
@@ -4,6 +4,7 @@ pragma abicoder v2;
 
 import {ERC20} from "./ERC20.sol";
 import {IERC4626} from "./IERC4626.sol";
+import {IHRC} from "../common/hedera/IHRC.sol";
 
 import {FeeConfiguration} from "../common/FeeConfiguration.sol";
 
@@ -254,7 +255,7 @@ contract HederaVault is IERC4626, FeeConfiguration, Ownable, ReentrancyGuard {
             for (uint256 i; i < rewardTokensSize; i++) {
                 address token = rewardTokens[i];
                 userContribution[msg.sender].lastClaimedAmountT[token] = tokensRewardInfo[token].amount;
-                SafeHTS.safeAssociateToken(token, msg.sender);
+                IHRC(token).associate();
             }
             userContribution[msg.sender].sharesAmount = _amount;
             userContribution[msg.sender].exist = true;