feat: use ActionAttributes lib

script/11_SetPeers.s.sol

@@ -2,7 +2,7 @@ pragma solidity ^0.8.19;
 
 import {Bootstrap} from "../src/core/Bootstrap.sol";
 import {ExocoreGateway} from "../src/core/ExocoreGateway.sol";
-import {GatewayStorage} from "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 
 import {BaseScript} from "./BaseScript.sol";
 import "forge-std/Script.sol";
@@ -51,8 +51,7 @@ contract SetPeersAndUpgrade is BaseScript {
 
         vm.selectFork(exocore);
         vm.startBroadcast(exocoreValidatorSet.privateKey);
-        uint256 nativeFee =
-            exocoreGateway.quote(clientChainId, abi.encodePacked(GatewayStorage.Action.REQUEST_MARK_BOOTSTRAP, ""));
+        uint256 nativeFee = exocoreGateway.quote(clientChainId, abi.encodePacked(Action.REQUEST_MARK_BOOTSTRAP, ""));
         exocoreGateway.markBootstrap{value: nativeFee}(clientChainId);
     }
 

script/13_DepositValidator.s.sol

@@ -5,7 +5,7 @@ import "../src/interfaces/IClientChainGateway.sol";
 import "../src/interfaces/IExocoreGateway.sol";
 import "../src/interfaces/IVault.sol";
 
-import "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 import "@beacon-oracle/contracts/src/EigenLayerBeaconOracle.sol";
 import "@layerzero-v2/protocol/contracts/interfaces/ILayerZeroEndpointV2.sol";
 import "@layerzero-v2/protocol/contracts/libs/AddressCast.sol";
@@ -74,7 +74,7 @@ contract DepositScript is BaseScript {
 
         vm.startBroadcast(depositor.privateKey);
         bytes memory msg_ = abi.encodePacked(
-            GatewayStorage.Action.REQUEST_DEPOSIT,
+            Action.REQUEST_DEPOSIT_LST,
             abi.encodePacked(bytes32(bytes20(VIRTUAL_STAKED_ETH_ADDRESS))),
             abi.encodePacked(bytes32(bytes20(depositor.addr))),
             uint256(_getEffectiveBalance(validatorContainer)) * GWEI_TO_WEI

script/3_Setup.s.sol

@@ -1,6 +1,6 @@
 pragma solidity ^0.8.19;
 
-import {GatewayStorage} from "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 
 import "../src/interfaces/IClientChainGateway.sol";
 import "../src/interfaces/IExocoreGateway.sol";
@@ -121,9 +121,7 @@ contract SetupScript is BaseScript {
         for (uint256 i = 0; i < whitelistTokensBytes32.length; i++) {
             nativeFee = exocoreGateway.quote(
                 clientChainId,
-                abi.encodePacked(
-                    GatewayStorage.Action.REQUEST_ADD_WHITELIST_TOKEN, abi.encodePacked(whitelistTokensBytes32[i])
-                )
+                abi.encodePacked(Action.REQUEST_ADD_WHITELIST_TOKEN, abi.encodePacked(whitelistTokensBytes32[i]))
             );
             exocoreGateway.addWhitelistToken{value: nativeFee}(
                 clientChainId,

script/4_Deposit.s.sol

@@ -4,7 +4,7 @@ import "../src/interfaces/IClientChainGateway.sol";
 
 import "../src/interfaces/IExocoreGateway.sol";
 import "../src/interfaces/IVault.sol";
-import "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 
 import {BaseScript} from "./BaseScript.sol";
 import "@layerzero-v2/protocol/contracts/interfaces/ILayerZeroEndpointV2.sol";
@@ -57,7 +57,7 @@ contract DepositScript is BaseScript {
 
     function run() public {
         bytes memory msg_ = abi.encodePacked(
-            GatewayStorage.Action.REQUEST_DEPOSIT,
+            Action.REQUEST_DEPOSIT_LST,
             abi.encodePacked(bytes32(bytes20(address(restakeToken)))),
             abi.encodePacked(bytes32(bytes20(depositor.addr))),
             uint256(DEPOSIT_AMOUNT)

script/5_Withdraw.s.sol

@@ -5,7 +5,7 @@ import "../src/interfaces/IClientChainGateway.sol";
 import "../src/interfaces/IExocoreGateway.sol";
 import "../src/interfaces/IVault.sol";
 
-import "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 
 import {BaseScript} from "./BaseScript.sol";
 import "@layerzero-v2/protocol/contracts/interfaces/ILayerZeroEndpointV2.sol";
@@ -57,7 +57,7 @@ contract DepositScript is BaseScript {
 
     function run() public {
         bytes memory msg_ = abi.encodePacked(
-            GatewayStorage.Action.REQUEST_WITHDRAW_PRINCIPAL_FROM_EXOCORE,
+            Action.REQUEST_WITHDRAW_LST,
             abi.encodePacked(bytes32(bytes20(address(restakeToken)))),
             abi.encodePacked(bytes32(bytes20(depositor.addr))),
             uint256(WITHDRAW_AMOUNT)

script/TestPrecompileErrorFixed.s.sol

@@ -8,7 +8,7 @@ import "../src/interfaces/IVault.sol";
 import "../src/interfaces/precompiles/IAssets.sol";
 import "../src/interfaces/precompiles/IClaimReward.sol";
 import "../src/interfaces/precompiles/IDelegation.sol";
-import "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 
 import {NonShortCircuitEndpointV2Mock} from "../test/mocks/NonShortCircuitEndpointV2Mock.sol";
 import {BaseScript} from "./BaseScript.sol";
@@ -73,7 +73,7 @@ contract DepositScript is BaseScript {
 
     function run() public {
         bytes memory depositMsg = abi.encodePacked(
-            GatewayStorage.Action.REQUEST_DEPOSIT,
+            Action.REQUEST_DEPOSIT_LST,
             abi.encodePacked(bytes32(bytes20(address(restakeToken)))),
             abi.encodePacked(bytes32(bytes20(depositor.addr))),
             uint256(TEST_DEPOSIT_AMOUNT)
@@ -94,7 +94,7 @@ contract DepositScript is BaseScript {
         vm.stopBroadcast();
 
         bytes memory withdrawMsg = abi.encodePacked(
-            GatewayStorage.Action.REQUEST_WITHDRAW_PRINCIPAL_FROM_EXOCORE,
+            Action.REQUEST_WITHDRAW_LST,
             abi.encodePacked(bytes32(bytes20(address(restakeToken)))),
             abi.encodePacked(bytes32(bytes20(depositor.addr))),
             uint256(TEST_WITHDRAWAL_AMOUNT)

script/TestPrecompileErrorFixed_Deploy.s.sol

@@ -5,7 +5,7 @@ import "../src/interfaces/IClientChainGateway.sol";
 import "../src/interfaces/IExocoreGateway.sol";
 import "../src/interfaces/IVault.sol";
 
-import "../src/storage/GatewayStorage.sol";
+import {Action, GatewayStorage} from "../src/storage/GatewayStorage.sol";
 
 import {NonShortCircuitEndpointV2Mock} from "../test/mocks/NonShortCircuitEndpointV2Mock.sol";
 import {BaseScript} from "./BaseScript.sol";

src/core/BaseRestakingController.sol

@@ -6,6 +6,7 @@ import {IExoCapsule} from "../interfaces/IExoCapsule.sol";
 import {IVault} from "../interfaces/IVault.sol";
 import {MessagingFee, MessagingReceipt, OAppSenderUpgradeable} from "../lzApp/OAppSenderUpgradeable.sol";
 import {ClientChainGatewayStorage} from "../storage/ClientChainGatewayStorage.sol";
+import {Action} from "../storage/GatewayStorage.sol";
 
 import {OptionsBuilder} from "@layerzero-v2/oapp/contracts/oapp/libs/OptionsBuilder.sol";
 
@@ -40,7 +41,7 @@ abstract contract BaseRestakingController is
         nonReentrant
     {
         require(recipient != address(0), "BaseRestakingController: recipient address cannot be empty or zero address");
-        if (token == VIRTUAL_STAKED_ETH_ADDRESS) {
+        if (token == VIRTUAL_NST_ADDRESS) {
             IExoCapsule capsule = _getCapsule(msg.sender);
             capsule.withdraw(amount, payable(recipient));
         } else {

src/core/Bootstrap.sol

@@ -22,7 +22,9 @@ import {ITokenWhitelister} from "../interfaces/ITokenWhitelister.sol";
 import {IVault} from "../interfaces/IVault.sol";
 
 import {Errors} from "../libraries/Errors.sol";
+
 import {BootstrapStorage} from "../storage/BootstrapStorage.sol";
+import {Action} from "../storage/GatewayStorage.sol";
 import {BootstrapLzReceiver} from "./BootstrapLzReceiver.sol";
 
 /// @title Bootstrap
@@ -213,7 +215,7 @@ contract Bootstrap is
             // whitelist, it means that it is missing a vault. we do not need to check for a
             // pre-existing vault. however, we still do ensure that the vault is not deployed
             // for restaking natively staked ETH.
-            if (token != VIRTUAL_STAKED_ETH_ADDRESS) {
+            if (token != VIRTUAL_NST_ADDRESS) {
                 // setting a tvlLimit higher than the supply is permitted.
                 // it allows for some margin for minting of the token, and lets us use
                 // a value of type(uint256).max to indicate no limit.
@@ -438,7 +440,7 @@ contract Bootstrap is
     /// @inheritdoc ILSTRestakingController
     /// @dev This is not yet supported.
     function withdrawRewardFromExocore(address, uint256) external payable override beforeLocked whenNotPaused {
-        revert NotYetSupported();
+        revert Errors.NotYetSupported();
     }
 
     /// @inheritdoc IBaseRestakingController

src/core/BootstrapLzReceiver.sol

@@ -2,8 +2,10 @@
 pragma solidity ^0.8.19;
 
 import {Errors} from "../libraries/Errors.sol";
+
 import {OAppReceiverUpgradeable, Origin} from "../lzApp/OAppReceiverUpgradeable.sol";
 import {BootstrapStorage} from "../storage/BootstrapStorage.sol";
+import {Action} from "../storage/GatewayStorage.sol";
 import {PausableUpgradeable} from "@openzeppelin/contracts-upgradeable/security/PausableUpgradeable.sol";
 
 /// @title BootstrapLzReceiver
@@ -24,7 +26,7 @@ abstract contract BootstrapLzReceiver is PausableUpgradeable, OAppReceiverUpgrad
     /// @inheritdoc OAppReceiverUpgradeable
     function _lzReceive(Origin calldata _origin, bytes calldata payload) internal virtual override {
         if (_origin.srcEid != EXOCORE_CHAIN_ID) {
-            revert UnexpectedSourceChain(_origin.srcEid);
+            revert Errors.UnexpectedSourceChain(_origin.srcEid);
         }
         _verifyAndUpdateNonce(_origin.srcEid, _origin.sender, _origin.nonce);
         Action act = Action(uint8(payload[0]));
@@ -33,11 +35,11 @@ abstract contract BootstrapLzReceiver is PausableUpgradeable, OAppReceiverUpgrad
         }
         bytes4 selector_ = _whiteListFunctionSelectors[act];
         if (selector_ == bytes4(0)) {
-            revert UnsupportedRequest(act);
+            revert Errors.UnsupportedRequest(act);
         }
         (bool success, bytes memory reason) = address(this).call(abi.encodePacked(selector_, abi.encode(payload[1:])));
         if (!success) {
-            revert RequestOrResponseExecuteFailed(act, _origin.nonce, reason);
+            revert Errors.RequestOrResponseExecuteFailed(act, _origin.nonce, reason);
         }
     }
 

src/core/ClientChainGateway.sol

@@ -15,6 +15,7 @@ import {LSTRestakingController} from "./LSTRestakingController.sol";
 import {NativeRestakingController} from "./NativeRestakingController.sol";
 
 import {Errors} from "../libraries/Errors.sol";
+import {Action} from "../storage/GatewayStorage.sol";
 import {IOAppCore} from "@layerzero-v2/oapp/contracts/oapp/interfaces/IOAppCore.sol";
 import {OptionsBuilder} from "@layerzero-v2/oapp/contracts/oapp/libs/OptionsBuilder.sol";
 import {OwnableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";