chore: refactored _mint in DEX contract

jatZama · Mar 8, 2024 · bb0a6e3 · bb0a6e3
1 parent af70794
commit bb0a6e3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/contracts/EncryptedDEXPair.sol b/contracts/EncryptedDEXPair.sol
@@ -189,12 +189,16 @@ contract EncryptedDEXPair is EncryptedERC20 {
 
     function claimMint(uint256 tradingEpoch, address user) external {
         require(tradingEpoch < currentTradingEpoch, "tradingEpoch is not settled yet");
+        _mint(tradingEpoch, user);
+        pendingMints[tradingEpoch][user] = ZERO;
+    }
+
+    function _mint(uint256 tradingEpoch, address user) internal {
         if (tradingEpoch == 0) {
             balances[user] = TFHE.sub(balances[user] + pendingMints[tradingEpoch][user], MINIMIMUM_LIQUIDITY); // this could never underflow from the contract logic
         } else {
             balances[user] = balances[user] + pendingMints[tradingEpoch][user];
         }
-        pendingMints[tradingEpoch][user] = ZERO;
     }
 
     function claimBurn(uint256 tradingEpoch, address user) external {