Merge branch 'treasury' of https://github.com/CanzaApps/Baki-contracts …

…into treasury
CanzaApps · Apr 8, 2024 · b4d66f8 · b4d66f8
2 parents 2b6e163 + d8a7cdf
commit b4d66f8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/contracts/Treasury.sol b/contracts/Treasury.sol
@@ -23,7 +23,6 @@ contract Treasury is ITreasury, AccessControl, ReentrancyGuard {
         address indexed _token,
         uint256 _amount
     );
-    event UnauthorizedAccess(address indexed _user, bool status);
 
     bytes32 public constant VAULT_ROLE = keccak256("VAULT_ROLE");
 
@@ -62,8 +61,13 @@ contract Treasury is ITreasury, AccessControl, ReentrancyGuard {
         string memory _asset
     ) external isPaused nonReentrant {
         require(hasRole(VAULT_ROLE, msg.sender), " Not vault");
-
-        bool transferSuccess = IERC20(currencies[_asset]).transfer(
+        require(
+            IERC20(currencies[_asset]).balanceOf(address(this)) >= _amount,
+            "Insufficient balance"
+        );
+        IERC20(currencies[_asset]).approve(address(this), _amount);
+        bool transferSuccess = IERC20(currencies[_asset]).transferFrom(
+            address(this),
             _to,
             _amount
         );