feat(cast): apply network precompiles for cast run and call (#11720)

feat(cast): apply precompiles for cast run and call

Author: grandizzy

Cargo.lock

@@ -592,6 +592,7 @@ impl NodeConfig {
     pub fn set_chain_id(&mut self, chain_id: Option<impl Into<u64>>) {
         self.chain_id = chain_id.map(Into::into);
         let chain_id = self.get_chain_id();
+        self.networks.with_chain_id(chain_id);
         self.genesis_accounts.iter_mut().for_each(|wallet| {
             *wallet = wallet.clone().with_chain_id(Some(chain_id));
         });

crates/anvil/src/config.rs

@@ -256,7 +256,7 @@ impl CallArgs {
 
             let create2_deployer = evm_opts.create2_deployer;
             let (mut env, fork, chain, networks) =
-                TracingExecutor::get_fork_material(&config, evm_opts).await?;
+                TracingExecutor::get_fork_material(&mut config, evm_opts).await?;
 
             // modify settings that usually set in eth_call
             env.evm_env.cfg_env.disable_block_gas_limit = true;

crates/cast/src/cmd/call.rs

@@ -158,7 +158,7 @@ impl RunArgs {
 
         let create2_deployer = evm_opts.create2_deployer;
         let (mut env, fork, chain, networks) =
-            TracingExecutor::get_fork_material(&config, evm_opts).await?;
+            TracingExecutor::get_fork_material(&mut config, evm_opts).await?;
         let mut evm_version = self.evm_version;
 
         env.evm_env.cfg_env.disable_block_gas_limit = self.disable_block_gas_limit;

crates/cast/src/cmd/run.rs

@@ -4062,3 +4062,25 @@ casttest!(abi_encode_event_dynamic_indexed, |_prj, cmd| {
 
 "#]]);
 });
+
+// Test cast run Celo transfer with precompiles.
+casttest!(run_celo_with_precompiles, |_prj, cmd| {
+    let rpc = next_rpc_endpoint(NamedChain::Celo);
+    cmd.args([
+        "run",
+        "0xa652b9f41bb1a617ea6b2835b3316e79f0f21b8264e7bcd20e57c4092a70a0f6",
+        "--rpc-url",
+        rpc.as_str(),
+    ])
+    .assert_success()
+    .stdout_eq(str![[r#"
+Executing previous transactions from the block.
+Traces:
+  [17776] 0x471EcE3750Da237f93B8E339c536989b8978a438::transfer(0xD2eB2d37d238Caeff39CFA36A013299C6DbAC56A, 138000000000000000 [1.38e17])
+    ├─ [12370] 0xFeA1B35f1D5f2A58532a70e7A32e6F2D3Bc4F7B1::transfer(0xD2eB2d37d238Caeff39CFA36A013299C6DbAC56A, 138000000000000000 [1.38e17]) [delegatecall]
+    │   ├─ [9000] CELO_TRANSFER_PRECOMPILE::00000000(00000000000000008106680ba7095cfd8f4351a8b7041da3060afb83000000000000000000000000d2eb2d37d238caeff39cfa36a013299c6dbac56a00000000000000000000000000000000000000000000000001ea4644d3010000)
+    │   │   └─ ← [Return]
+...
+
+"#]]);
+});

crates/cast/tests/cli/main.rs

@@ -77,7 +77,7 @@ impl TracingExecutor {
 
     /// uses the fork block number from the config
     pub async fn get_fork_material(
-        config: &Config,
+        config: &mut Config,
         mut evm_opts: EvmOpts,
     ) -> eyre::Result<(Env, Option<CreateFork>, Option<Chain>, NetworkConfigs)> {
         evm_opts.fork_url = Some(config.get_rpc_url_or_localhost_http()?.into_owned());
@@ -86,8 +86,10 @@ impl TracingExecutor {
         let env = evm_opts.evm_env().await?;
 
         let fork = evm_opts.get_fork(config, env.clone());
+        let networks = evm_opts.networks.with_chain_id(env.evm_env.cfg_env.chain_id);
+        config.labels.extend(networks.precompiles_label());
 
-        Ok((env, fork, evm_opts.get_remote_chain_id().await, evm_opts.networks))
+        Ok((env, fork, evm_opts.get_remote_chain_id().await, networks))
     }
 }
 

crates/evm/evm/src/executors/trace.rs

@@ -14,6 +14,7 @@ repository.workspace = true
 workspace = true
 
 [dependencies]
+alloy-chains.workspace = true
 alloy-evm.workspace = true
 alloy-primitives = { workspace = true, features = [
     "serde",

crates/evm/networks/Cargo.toml

@@ -17,6 +17,9 @@ use alloy_evm::precompiles::{DynPrecompile, PrecompileInput};
 use alloy_primitives::{Address, U256, address};
 use revm::precompile::{PrecompileError, PrecompileId, PrecompileOutput, PrecompileResult};
 
+/// Label of the Celo transfer precompile to display in traces.
+pub const CELO_TRANSFER_LABEL: &str = "CELO_TRANSFER_PRECOMPILE";
+
 /// Address of the Celo transfer precompile.
 pub const CELO_TRANSFER_ADDRESS: Address = address!("0x00000000000000000000000000000000000000fd");
 

crates/evm/networks/src/celo/transfer.rs

@@ -2,9 +2,12 @@
 //!
 //! Foundry EVM network custom precompiles.
 
-use crate::celo::transfer::{CELO_TRANSFER_ADDRESS, PRECOMPILE_ID_CELO_TRANSFER};
+use crate::celo::transfer::{
+    CELO_TRANSFER_ADDRESS, CELO_TRANSFER_LABEL, PRECOMPILE_ID_CELO_TRANSFER,
+};
+use alloy_chains::NamedChain;
 use alloy_evm::precompiles::PrecompilesMap;
-use alloy_primitives::Address;
+use alloy_primitives::{Address, map::AddressHashMap};
 use clap::Parser;
 use serde::{Deserialize, Serialize};
 use std::collections::BTreeMap;
@@ -33,6 +36,13 @@ impl NetworkConfigs {
         Self { optimism: true, ..Default::default() }
     }
 
+    pub fn with_chain_id(mut self, chain_id: u64) -> Self {
+        if let Ok(NamedChain::Celo | NamedChain::CeloSepolia) = NamedChain::try_from(chain_id) {
+            self.celo = true;
+        }
+        self
+    }
+
     /// Inject precompiles for configured networks.
     pub fn inject_precompiles(self, precompiles: &mut PrecompilesMap) {
         if self.celo {
@@ -42,6 +52,15 @@ impl NetworkConfigs {
         }
     }
 
+    /// Returns precompiles label for configured networks, to be used in traces.
+    pub fn precompiles_label(self) -> AddressHashMap<String> {
+        let mut labels = AddressHashMap::default();
+        if self.celo {
+            labels.insert(CELO_TRANSFER_ADDRESS, CELO_TRANSFER_LABEL.to_string());
+        }
+        labels
+    }
+
     /// Returns precompiles for configured networks.
     pub fn precompiles(self) -> BTreeMap<String, Address> {
         let mut precompiles = BTreeMap::new();

crates/evm/networks/src/lib.rs

@@ -2,7 +2,9 @@
 
 use foundry_config::{
     NamedChain,
-    NamedChain::{Arbitrum, Base, BinanceSmartChainTestnet, Mainnet, Optimism, Polygon, Sepolia},
+    NamedChain::{
+        Arbitrum, Base, BinanceSmartChainTestnet, Celo, Mainnet, Optimism, Polygon, Sepolia,
+    },
 };
 use rand::seq::SliceRandom;
 use std::sync::{
@@ -150,6 +152,10 @@ fn next_url(is_ws: bool, chain: NamedChain) -> String {
         return "https://bsc-testnet-rpc.publicnode.com".to_string();
     }
 
+    if matches!(chain, Celo) {
+        return "https://celo.drpc.org".to_string();
+    }
+
     let reth_works = true;
     let domain = if reth_works && matches!(chain, Mainnet) {
         *next(if is_ws { &WS_DOMAINS } else { &HTTP_DOMAINS })