nanoreth/src/pseudo_peer/network.rs

use super::service::{BlockHashCache, BlockPoller};
use crate::{chainspec::HlChainSpec, node::network::HlNetworkPrimitives, HlPrimitives};
use reth_network::{
    config::{rng_secret_key, SecretKey},
    NetworkConfig, NetworkManager, PeersConfig,
};
use reth_network_peers::TrustedPeer;
use reth_provider::test_utils::NoopProvider;
use std::{str::FromStr, sync::Arc};
use tokio::sync::mpsc;

pub struct NetworkBuilder {
    secret: SecretKey,
    peer_config: PeersConfig,
    boot_nodes: Vec<TrustedPeer>,
    discovery_port: u16,
    listener_port: u16,
    chain_spec: HlChainSpec,
}

impl Default for NetworkBuilder {
    fn default() -> Self {
        Self {
            secret: rng_secret_key(),
            peer_config: PeersConfig::default().with_max_outbound(1).with_max_inbound(1),
            boot_nodes: vec![],
            discovery_port: 0,
            listener_port: 0,
            chain_spec: HlChainSpec::default(),
        }
    }
}

impl NetworkBuilder {
    pub fn with_boot_nodes(mut self, boot_nodes: Vec<TrustedPeer>) -> Self {
        self.boot_nodes = boot_nodes;
        self
    }

    pub fn with_ports(mut self, discovery_port: u16, listener_port: u16) -> Self {
        self.discovery_port = discovery_port;
        self.listener_port = listener_port;
        self
    }

    pub fn with_chain_spec(mut self, chain_spec: HlChainSpec) -> Self {
        self.chain_spec = chain_spec;
        self
    }

    pub async fn build<BS>(
        self,
        block_source: Arc<Box<dyn super::sources::BlockSource>>,
        blockhash_cache: BlockHashCache,
    ) -> eyre::Result<(NetworkManager<HlNetworkPrimitives>, mpsc::Sender<()>)> {
        let builder = NetworkConfig::<(), HlNetworkPrimitives>::builder(self.secret)
            .boot_nodes(self.boot_nodes)
            .peer_config(self.peer_config)
            .discovery_port(self.discovery_port)
            .listener_port(self.listener_port);
        let chain_id = self.chain_spec.inner.chain().id();

        let (block_poller, start_tx) =
            BlockPoller::new_suspended(chain_id, block_source, blockhash_cache);
        let config = builder.block_import(Box::new(block_poller)).build(Arc::new(NoopProvider::<
            HlChainSpec,
            HlPrimitives,
        >::new(
            self.chain_spec.into(),
        )));

        let network = NetworkManager::new(config).await.map_err(|e| eyre::eyre!(e))?;
        Ok((network, start_tx))
    }
}

pub async fn create_network_manager<BS>(
    chain_spec: HlChainSpec,
    destination_peer: String,
    block_source: Arc<Box<dyn super::sources::BlockSource>>,
    blockhash_cache: BlockHashCache,
) -> eyre::Result<(NetworkManager<HlNetworkPrimitives>, mpsc::Sender<()>)> {
    NetworkBuilder::default()
        .with_boot_nodes(vec![TrustedPeer::from_str(&destination_peer).unwrap()])
        .with_chain_spec(chain_spec)
        .build::<BS>(block_source, blockhash_cache)
        .await
}