expose initialization data for verification

etherisc · Sep 2, 2024 · 1f14a3b · 1f14a3b
1 parent 9c4663e
commit 1f14a3b
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/contracts/upgradeability/UpgradableProxyWithAdmin.sol b/contracts/upgradeability/UpgradableProxyWithAdmin.sol
@@ -6,9 +6,15 @@ import {TransparentUpgradeableProxy} from "@openzeppelin/contracts/proxy/transpa
 
 contract UpgradableProxyWithAdmin is TransparentUpgradeableProxy {
 
+    bytes internal _initializationData;
+
     constructor(address implementation, address initialProxyAdminOwner, bytes memory data)
         TransparentUpgradeableProxy(implementation, initialProxyAdminOwner, data)
-    {}
+    {
+        _initializationData = data;
+    }
 
     function getProxyAdmin() external returns (ProxyAdmin) { return ProxyAdmin(_proxyAdmin()); }
+
+    function getInitializationData() external view returns (bytes memory) { return _initializationData; }
 }
diff --git a/scripts/libs/registry.ts b/scripts/libs/registry.ts
@@ -13,7 +13,8 @@ import {
     StakingReader,
     StakingStore,
     Staking__factory,
-    TokenRegistry
+    TokenRegistry,
+    UpgradableProxyWithAdmin__factory
 } from "../../typechain-types";
 import { logger } from "../logger";
 import { deployContract } from "./deployment";
@@ -282,11 +283,25 @@ export async function deployAndInitializeRegistry(owner: Signer, libraries: Libr
         await resolveAddress(owner));
 
     // verify staking implementation 
+    const stakingImplenentationAddress = await getImplementationAddress(hhEthers.provider, await stakingManager.getProxy());
     await prepareVerificationData(
         "Staking", 
-        await getImplementationAddress(hhEthers.provider, await stakingManager.getProxy()), 
+        stakingImplenentationAddress, 
         [], 
         undefined);
+
+    // verify staking proxy
+    const stakingProxyAddress = await stakingManager.getProxy();
+    const proxy = UpgradableProxyWithAdmin__factory.connect(stakingProxyAddress, owner);
+    await prepareVerificationData(
+        "UpgradableProxyWithAdmin", 
+        stakingProxyAddress, 
+        [
+            stakingImplenentationAddress,
+            stakingManagerAddress,
+            await proxy.getInitializationData(),
+        ], 
+        undefined);
 
     await prepareVerificationData(
         "TokenHandler",