fedimint_testing_core/
config.rs

1use std::collections::{BTreeMap, HashMap};
2use std::sync::LazyLock;
3
4use fedimint_core::PeerId;
5use fedimint_core::module::ApiAuth;
6use fedimint_core::setup_code::{PeerEndpoints, PeerSetupCode};
7use fedimint_server::config::ConfigGenParams;
8use fedimint_server::net::p2p_connector::gen_cert_and_key;
9use tokio_rustls::rustls;
10
11pub static API_AUTH: LazyLock<ApiAuth> = LazyLock::new(|| ApiAuth("pass".to_string()));
12
13/// Creates the config gen params for each peer
14///
15/// Uses peers * 2 ports offset from `base_port`
16pub fn local_config_gen_params(
17    peers: &[PeerId],
18    base_port: u16,
19) -> anyhow::Result<HashMap<PeerId, ConfigGenParams>> {
20    // Generate TLS cert and private key
21    let tls_keys: HashMap<PeerId, (rustls::Certificate, rustls::PrivateKey)> = peers
22        .iter()
23        .map(|peer| {
24            (
25                *peer,
26                gen_cert_and_key(&format!("peer-{}", peer.to_usize())).unwrap(),
27            )
28        })
29        .collect();
30
31    // Generate the P2P and API URL on 2 different ports for each peer
32    let connections: BTreeMap<PeerId, PeerSetupCode> = peers
33        .iter()
34        .map(|peer| {
35            let peer_port = base_port + u16::from(*peer) * 3;
36
37            let p2p_url = format!("fedimint://127.0.0.1:{peer_port}");
38            let api_url = format!("ws://127.0.0.1:{}", peer_port + 1);
39
40            let params = PeerSetupCode {
41                name: format!("peer-{}", peer.to_usize()),
42                endpoints: PeerEndpoints::Tcp {
43                    api_url: api_url.parse().expect("Should parse"),
44                    p2p_url: p2p_url.parse().expect("Should parse"),
45                    cert: tls_keys[peer].0.clone().0,
46                },
47                federation_name: None,
48            };
49            (*peer, params)
50        })
51        .collect();
52
53    peers
54        .iter()
55        .map(|peer| {
56            let params = ConfigGenParams {
57                identity: *peer,
58                api_auth: API_AUTH.clone(),
59                tls_key: Some(tls_keys[peer].1.clone()),
60                iroh_api_sk: None,
61                iroh_p2p_sk: None,
62                peers: connections.clone(),
63                meta: BTreeMap::new(),
64            };
65            Ok((*peer, params))
66        })
67        .collect()
68}