Merge pull request #154 from axieinfinity/feature/fix-callback-visibi…

…lity
axieinfinity · Jul 31, 2024 · ab9eed2 · ab9eed2
2 parents a77e54b + d06f76e
commit ab9eed2
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/script/BaseMigration.s.sol b/script/BaseMigration.s.sol
@@ -30,13 +30,13 @@ abstract contract BaseMigration is ScriptExtended {
     _injectDependencies();
   }
 
-  function upgradeCallback(
+  function _upgradeCallback(
     address, /* proxy */
     address, /* logic */
     uint256, /* callValue */
     bytes memory, /* callData */
     ProxyInterface /* proxyInterface */
-  ) external virtual { }
+  ) internal virtual { }
 
   function _sharedArguments() internal virtual returns (bytes memory rawSharedArgs);
 
@@ -271,7 +271,7 @@ abstract contract BaseMigration is ScriptExtended {
       callData: args,
       shouldPrompt: true,
       proxyInterface: ProxyInterface.Transparent,
-      upgradeCallback: this.upgradeCallback,
+      upgradeCallback: _upgradeCallback,
       shouldUseCallback: false
     }).upgrade();
   }

diff --git a/script/libraries/LibDeploy.sol b/script/libraries/LibDeploy.sol
@@ -32,7 +32,7 @@ struct UpgradeInfo {
   uint256 callValue;
   bytes callData;
   ProxyInterface proxyInterface;
-  function(address,address,uint256,bytes memory,ProxyInterface) external upgradeCallback;
+  function(address,address,uint256,bytes memory,ProxyInterface) internal upgradeCallback;
   bool shouldUseCallback;
   bool shouldPrompt;
 }
@@ -120,7 +120,7 @@ library LibDeploy {
     uint256 callValue,
     bytes memory callData,
     bool shouldPrompt,
-    function(address,address,uint256,bytes memory,ProxyInterface) external upgradeCallback,
+    function(address,address,uint256,bytes memory,ProxyInterface) internal upgradeCallback,
     bool shouldUseCallback
   ) internal validateUpgrade(proxy, logic, shouldPrompt) {
     if (shouldUseCallback) {

diff --git a/test/LibDeploy.t.sol b/test/LibDeploy.t.sol
@@ -31,7 +31,7 @@ contract LibDeployTest is Test {
       callValue: 0,
       callData: abi.encodeCall(MockERC721.initialize, ("Name", "Symbol")),
       proxyInterface: ProxyInterface.Transparent,
-      upgradeCallback: this.emptyFn,
+      upgradeCallback: emptyFn,
       shouldPrompt: false,
       shouldUseCallback: false
     });
@@ -56,7 +56,7 @@ contract LibDeployTest is Test {
       callValue: 0,
       callData: abi.encodeCall(MockERC20.initialize, ("Name", "Symbol", 18)),
       proxyInterface: ProxyInterface.Transparent,
-      upgradeCallback: this.emptyFn,
+      upgradeCallback: emptyFn,
       shouldPrompt: false,
       shouldUseCallback: false
     });
@@ -79,7 +79,7 @@ contract LibDeployTest is Test {
       callValue: 0,
       callData: abi.encodeCall(MockERC20.initialize, ("Name", "Symbol", 18)),
       proxyInterface: ProxyInterface.Transparent,
-      upgradeCallback: this.emptyFn,
+      upgradeCallback: emptyFn,
       shouldPrompt: false,
       shouldUseCallback: false
     });
@@ -108,7 +108,7 @@ contract LibDeployTest is Test {
       callValue: 0,
       callData: abi.encodeCall(MockERC20.initialize, ("Name", "Symbol", 18)),
       proxyInterface: ProxyInterface.Transparent,
-      upgradeCallback: this.emptyFn,
+      upgradeCallback: emptyFn,
       shouldPrompt: false,
       shouldUseCallback: false
     });
@@ -122,5 +122,5 @@ contract LibDeployTest is Test {
     uint256, /* callValue */
     bytes memory, /* callData */
     ProxyInterface /* proxyInterface */
-  ) external { }
+  ) internal { }
 }