access fix for the vault (#175)

PufferFinance · Feb 22, 2024 · 7d309a4 · 7d309a4
1 parent 4cd6dde
commit 7d309a4
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/script/SetupAccess.s.sol b/script/SetupAccess.s.sol
@@ -35,7 +35,7 @@ contract SetupAccess is BaseScript {
         bytes[] memory vaultMainnetAccess = _setupPufferVaultMainnetAccess();
         bytes[] memory pufferOracleAccess = _setupPufferOracleAccess();
 
-        bytes[] memory calldatas = new bytes[](17);
+        bytes[] memory calldatas = new bytes[](18);
         calldatas[0] = _setupGuardianModuleRoles();
         calldatas[1] = _setupEnclaveVerifierRoles();
         calldatas[2] = _setupUpgradeableBeacon();
@@ -55,9 +55,10 @@ contract SetupAccess is BaseScript {
 
         calldatas[13] = vaultMainnetAccess[0];
         calldatas[14] = vaultMainnetAccess[1];
+        calldatas[15] = vaultMainnetAccess[2];
 
-        calldatas[15] = pufferOracleAccess[0];
-        calldatas[16] = pufferOracleAccess[1];
+        calldatas[16] = pufferOracleAccess[0];
+        calldatas[17] = pufferOracleAccess[1];
 
         accessManager.multicall(calldatas);
 
@@ -95,7 +96,7 @@ contract SetupAccess is BaseScript {
     }
 
     function _setupPufferVaultMainnetAccess() internal view returns (bytes[] memory) {
-        bytes[] memory calldatas = new bytes[](2);
+        bytes[] memory calldatas = new bytes[](3);
 
         bytes4[] memory publicSelectors = new bytes4[](1);
         publicSelectors[0] = PufferVaultMainnet.burn.selector;
@@ -117,6 +118,16 @@ contract SetupAccess is BaseScript {
             ROLE_ID_OPERATIONS //@todo?
         );
 
+        bytes4[] memory protocolSelectors = new bytes4[](1);
+        protocolSelectors[0] = PufferVaultMainnet.transferETH.selector;
+
+        calldatas[2] = abi.encodeWithSelector(
+            AccessManager.setTargetFunctionRole.selector,
+            pufferDeployment.pufferVault,
+            protocolSelectors,
+            ROLE_ID_PUFFER_PROTOCOL
+        );
+
         return calldatas;
     }