feat: RpcUrls cheatcodes implementation

crates/era-cheatcodes/src/cheatcodes.rs

@@ -832,6 +832,33 @@ impl CheatcodeTracer {
                     &mut storage,
                 );
             }
+            rpcUrl(rpcUrlCall { rpcAlias }) => {
+                tracing::info!("👷 Getting rpc url of {}", rpcAlias);
+                let rpc_endpoints = &self.config.rpc_endpoints;
+                let rpc_url = match rpc_endpoints.get(&rpcAlias) {
+                    Some(Ok(url)) => Some(url.clone()),
+                    _ => None,
+                };
+                //this should revert but we don't have reverts yet
+                assert!(
+                    rpc_url.is_some(),
+                    "Failed to resolve env var `${rpcAlias}`: environment variable not found"
+                );
+                self.add_trimmed_return_data(rpc_url.unwrap().as_bytes());
+            }
+            rpcUrls(rpcUrlsCall {}) => {
+                tracing::info!("👷 Getting rpc urls");
+                let rpc_endpoints = &self.config.rpc_endpoints;
+                let rpc_urls: Vec<String> = rpc_endpoints
+                    .iter()
+                    .map(|(alias, url)| match url {
+                        Ok(url) => format!("{}:{}", alias, url),
+                        Err(_) => alias.clone(),
+                    })
+                    .collect();
+                let rpc_urls = rpc_urls.join(",");
+                self.add_trimmed_return_data(rpc_urls.as_bytes());
+            }
             serializeAddress_0(serializeAddress_0Call {
                 objectKey: object_key,
                 valueKey: value_key,

crates/era-cheatcodes/tests/src/cheatcodes/RpcUrls.t.sol

@@ -0,0 +1,48 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.13;
+
+import {Test, console2 as console} from "../../lib/forge-std/src/Test.sol";
+import {Constants} from "./Constants.sol";
+import {Utils} from "./Utils.sol";
+
+contract RpcUrlsTest is Test {
+    function testRpcUrl() public {
+        (bool success, bytes memory rawData) = Constants.CHEATCODE_ADDRESS.call(
+            abi.encodeWithSignature("rpcUrl(string)", "mainnet")
+        );
+
+        bytes memory return_data = Utils.trimReturnBytes(rawData);
+        string memory rpc_url = string(return_data);
+        console.log("rpc_url", rpc_url);
+        require(success, "rpcUrl() failed");
+        require(
+            keccak256(abi.encodePacked(rpc_url)) ==
+                keccak256(
+                    abi.encodePacked("https://mainnet.era.zksync.io:443")
+                ),
+            "rpc url retrieved does not match expected value"
+        );
+    }
+
+    function testRpcUrls() public {
+        (bool success, bytes memory rawData2) = Constants
+            .CHEATCODE_ADDRESS
+            .call(abi.encodeWithSignature("rpcUrls()"));
+
+        bytes memory return_data2 = Utils.trimReturnBytes(rawData2);
+        string memory rpc_urls = string(return_data2);
+
+        console.log("rpc_urls", rpc_urls);
+
+        require(success, "rpcUrls() failed");
+        require(
+            keccak256(abi.encodePacked(rpc_urls)) ==
+                keccak256(
+                    abi.encodePacked(
+                        "local,mainnet:https://mainnet.era.zksync.io:443,testnet:https://testnet.era.zksync.dev:443"
+                    )
+                ),
+            "rpc urls retrieved does not match expected value"
+        );
+    }
+}