feat(cheatcodes): add computeCreateAddress cheatcodes (foundry-rs#6296)

* Add computeCreateAddress cheatcodes

* use Address methods

* ensure nonce can be u64

Author: qiweiii

crates/cheatcodes/assets/cheatcodes.json

@@ -1155,5 +1155,17 @@ interface Vm {
     /// Gets the label for the specified address.
     #[cheatcode(group = Utilities)]
     function getLabel(address account) external returns (string memory currentLabel);
+
+    /// Compute the address a contract will be deployed at for a given deployer address and nonce.
+    #[cheatcode(group = Utilities)]
+    function computeCreateAddress(address deployer, uint256 nonce) external pure returns (address);
+
+    /// Compute the address of a contract created with CREATE2 using the given CREATE2 deployer.
+    #[cheatcode(group = Utilities)]
+    function computeCreate2Address(bytes32 salt, bytes32 initCodeHash, address deployer) external pure returns (address);
+
+    /// Compute the address of a contract created with CREATE2 using the default CREATE2 deployer.
+    #[cheatcode(group = Utilities)]
+    function computeCreate2Address(bytes32 salt, bytes32 initCodeHash) external pure returns (address);
 }
 }

crates/cheatcodes/defs/src/vm.rs

@@ -15,6 +15,7 @@ use ethers_signers::{
     },
     LocalWallet, MnemonicBuilder, Signer,
 };
+use foundry_evm_core::constants::DEFAULT_CREATE2_DEPLOYER;
 use foundry_utils::types::{ToAlloy, ToEthers};
 
 /// The BIP32 default derivation path prefix.
@@ -111,6 +112,28 @@ impl Cheatcode for getLabelCall {
     }
 }
 
+impl Cheatcode for computeCreateAddressCall {
+    fn apply(&self, _state: &mut Cheatcodes) -> Result {
+        let Self { nonce, deployer } = self;
+        ensure!(*nonce <= U256::from(u64::MAX), "nonce must be less than 2^64 - 1");
+        Ok(deployer.create(nonce.to()).abi_encode())
+    }
+}
+
+impl Cheatcode for computeCreate2Address_0Call {
+    fn apply(&self, _state: &mut Cheatcodes) -> Result {
+        let Self { salt, initCodeHash, deployer } = self;
+        Ok(deployer.create2(salt, initCodeHash).abi_encode())
+    }
+}
+
+impl Cheatcode for computeCreate2Address_1Call {
+    fn apply(&self, _state: &mut Cheatcodes) -> Result {
+        let Self { salt, initCodeHash } = self;
+        Ok(DEFAULT_CREATE2_DEPLOYER.create2(salt, initCodeHash).abi_encode())
+    }
+}
+
 /// Using a given private key, return its public ETH address, its public key affine x and y
 /// coodinates, and its private key (see the 'Wallet' struct)
 ///