feat: add _postCheck internal function

axieinfinity · Dec 14, 2023 · b82b320 · b82b320
1 parent ed44275
commit b82b320
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 78 deletions.
diff --git a/script/extensions/ScriptExtended.s.sol b/script/extensions/ScriptExtended.s.sol
@@ -20,7 +20,7 @@ abstract contract ScriptExtended is Script, StdAssertions, IScriptExtended {
     console.log("> ", StdStyle.blue(fnName), "...");
     _;
   }
-  
+
   modifier onlyOn(TNetwork networkType) {
     require(network() == networkType, string.concat("ScriptExtended: Only allowed on ", CONFIG.getAlias(networkType)));
     _;
@@ -33,10 +33,13 @@ abstract contract ScriptExtended is Script, StdAssertions, IScriptExtended {
 
   function _configByteCode() internal virtual returns (bytes memory);
 
+  function _postCheck() internal virtual { }
+
   function run(bytes calldata callData, string calldata command) public virtual {
     CONFIG.resolveCommand(command);
     (bool success, bytes memory data) = address(this).delegatecall(callData);
     success.handleRevert(msg.sig, data);
+    _postCheck();
   }
 
   function network() public view virtual returns (TNetwork) {

diff --git a/src/WNT.sol b/src/WNT.sol
@@ -3,7 +3,7 @@ pragma solidity ^0.8.17;
 
 import { ERC20 } from "../lib/openzeppelin-contracts/contracts/token/ERC20/ERC20.sol";
 import { IWNT } from "./interfaces/IWNT.sol";
-import { LibNativeTransfer } from "./libraries/LibNativeTransfer.sol";
+import { LibNativeTransfer } from "contract-libs/transfers/LibNativeTransfer.sol";
 
 /// @notice Minimalist and modern Wrapped Ether implementation.
 /// @author Solmate

diff --git a/src/libraries/LibErrorHandler.sol b/src/libraries/LibErrorHandler.sol
diff --git a/src/libraries/LibNativeTransfer.sol b/src/libraries/LibNativeTransfer.sol