adds withdraw

PufferFinance · Jun 13, 2024 · c8c1f26 · c8c1f26
1 parent 9460dcf
commit c8c1f26
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/mainnet-contracts/src/based-contracts/PufToken.sol b/mainnet-contracts/src/based-contracts/PufToken.sol
@@ -80,6 +80,18 @@ contract PufToken is ERC20, ERC20Permit {
 
         _mint(msg.sender, amount);
     }
+
+    function withdraw(uint256 amount) external onlyDepositor {
+        if (balanceOf(msg.sender) < amount) {
+            revert InsufficientBalance();
+        }
+
+        _burn(msg.sender, amount);
+
+        if (!IERC20(originalToken).transfer(msg.sender, amount)) {
+            revert TransferFailed();
+        }
+    }
 }
 
 interface IMigrator {