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
13pub fn local_config_gen_params(
17 peers: &[PeerId],
18 base_port: u16,
19) -> anyhow::Result<HashMap<PeerId, ConfigGenParams>> {
20 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 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}