nanoreth/src/main.rs

use clap::Parser;
use reth::builder::NodeHandle;
use reth_hl::tx_forwarder::{self, EthForwarderApiServer};
use reth_hl::{
    chainspec::parser::HlChainSpecParser,
    node::{
        cli::{Cli, HlNodeArgs},
        storage::tables::Tables,
        HlNode,
    },
};
use tracing::info;

// We use jemalloc for performance reasons
#[cfg(all(feature = "jemalloc", unix))]
#[global_allocator]
static ALLOC: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;

fn main() -> eyre::Result<()> {
    reth_cli_util::sigsegv_handler::install();

    // Enable backtraces unless a RUST_BACKTRACE value has already been explicitly provided.
    if std::env::var_os("RUST_BACKTRACE").is_none() {
        std::env::set_var("RUST_BACKTRACE", "1");
    }

    Cli::<HlChainSpecParser, HlNodeArgs>::parse().run(|builder, ext| async move {
        builder.builder.database.create_tables_for::<Tables>()?;
        let (node, engine_handle_tx) = HlNode::new(ext.block_source_args.parse().await?);
        let NodeHandle { node, node_exit_future: exit_future } = builder
            .node(node)
            .extend_rpc_modules(|ctx| {
                let upstream_rpc_url = ext.upstream_rpc_url;
                if let Some(upstream_rpc_url) = upstream_rpc_url {
                    ctx.modules.replace_configured(
                        tx_forwarder::EthForwarderExt::new(upstream_rpc_url.clone()).into_rpc(),
                    )?;

                    info!("Transaction forwarding enabled");
                }
                Ok(())
            })
            .launch()
            .await?;

        engine_handle_tx.send(node.beacon_engine_handle.clone()).unwrap();

        exit_future.await
    })?;
    Ok(())
}