Moved the Lido address map into a static var

Author: jclapis

Cargo.lock

@@ -26,6 +26,7 @@ ethereum_ssz_derive.workspace = true
 eyre.workspace = true
 futures.workspace = true
 jsonwebtoken.workspace = true
+lazy_static.workspace = true
 lh_eth2.workspace = true
 lh_eth2_keystore.workspace = true
 lh_types.workspace = true

crates/common/Cargo.toml

@@ -344,7 +344,6 @@ async fn fetch_lido_csm_registry_keys(
 ) -> eyre::Result<Vec<BlsPublicKey>> {
     let provider = ProviderBuilder::new().connect_client(rpc_client);
     let registry = get_lido_csm_registry(registry_address, provider);
-
     let total_keys = fetch_lido_csm_keys_total(&registry, node_operator_id).await?;
 
     collect_registry_keys(total_keys, |offset, limit| {

crates/common/src/config/mux.rs

@@ -2,6 +2,7 @@ use std::collections::HashMap;
 
 use alloy::primitives::{Address, Bytes, U256, address};
 use eyre::Context;
+use lazy_static::lazy_static;
 
 use crate::{
     interop::lido::types::{
@@ -11,76 +12,78 @@ use crate::{
     types::{Chain, HoleskyLidoModule, HoodiLidoModule, MainnetLidoModule},
 };
 
-pub fn lido_registry_addresses_by_module() -> HashMap<Chain, HashMap<u8, Address>> {
-    let mut map: HashMap<Chain, HashMap<u8, Address>> = HashMap::new();
-
-    // --- Mainnet ---
-    let mut mainnet = HashMap::new();
-    mainnet.insert(
-        MainnetLidoModule::Curated as u8,
-        address!("55032650b14df07b85bF18A3a3eC8E0Af2e028d5"),
-    );
-    mainnet.insert(
-        MainnetLidoModule::SimpleDVT as u8,
-        address!("aE7B191A31f627b4eB1d4DaC64eaB9976995b433"),
-    );
-    mainnet.insert(
-        MainnetLidoModule::CommunityStaking as u8,
-        address!("dA7dE2ECdDfccC6c3AF10108Db212ACBBf9EA83F"),
-    );
-    map.insert(Chain::Mainnet, mainnet);
-
-    // --- Holesky ---
-    let mut holesky = HashMap::new();
-    holesky.insert(
-        HoleskyLidoModule::Curated as u8,
-        address!("595F64Ddc3856a3b5Ff4f4CC1d1fb4B46cFd2bAC"),
-    );
-    holesky.insert(
-        HoleskyLidoModule::SimpleDVT as u8,
-        address!("11a93807078f8BB880c1BD0ee4C387537de4b4b6"),
-    );
-    holesky.insert(
-        HoleskyLidoModule::Sandbox as u8,
-        address!("D6C2ce3BB8bea2832496Ac8b5144819719f343AC"),
-    );
-    holesky.insert(
-        HoleskyLidoModule::CommunityStaking as u8,
-        address!("4562c3e63c2e586cD1651B958C22F88135aCAd4f"),
-    );
-    map.insert(Chain::Holesky, holesky);
-
-    // --- Hoodi ---
-    let mut hoodi = HashMap::new();
-    hoodi.insert(
-        HoodiLidoModule::Curated as u8,
-        address!("5cDbE1590c083b5A2A64427fAA63A7cfDB91FbB5"),
-    );
-    hoodi.insert(
-        HoodiLidoModule::SimpleDVT as u8,
-        address!("0B5236BECA68004DB89434462DfC3BB074d2c830"),
-    );
-    hoodi.insert(
-        HoodiLidoModule::Sandbox as u8,
-        address!("682E94d2630846a503BDeE8b6810DF71C9806891"),
-    );
-    hoodi.insert(
-        HoodiLidoModule::CommunityStaking as u8,
-        address!("79CEf36D84743222f37765204Bec41E92a93E59d"),
-    );
-    map.insert(Chain::Hoodi, hoodi);
-
-    // --- Sepolia --
-    let mut sepolia = HashMap::new();
-    sepolia.insert(1, address!("33d6E15047E8644F8DDf5CD05d202dfE587DA6E3"));
-    map.insert(Chain::Sepolia, sepolia);
-
-    map
+lazy_static! {
+    static ref LIDO_REGISTRY_ADDRESSES_BY_MODULE: HashMap<Chain, HashMap<u8, Address>> = {
+        let mut map: HashMap<Chain, HashMap<u8, Address>> = HashMap::new();
+
+        // --- Mainnet ---
+        let mut mainnet = HashMap::new();
+        mainnet.insert(
+            MainnetLidoModule::Curated as u8,
+            address!("55032650b14df07b85bF18A3a3eC8E0Af2e028d5"),
+        );
+        mainnet.insert(
+            MainnetLidoModule::SimpleDVT as u8,
+            address!("aE7B191A31f627b4eB1d4DaC64eaB9976995b433"),
+        );
+        mainnet.insert(
+            MainnetLidoModule::CommunityStaking as u8,
+            address!("dA7dE2ECdDfccC6c3AF10108Db212ACBBf9EA83F"),
+        );
+        map.insert(Chain::Mainnet, mainnet);
+
+        // --- Holesky ---
+        let mut holesky = HashMap::new();
+        holesky.insert(
+            HoleskyLidoModule::Curated as u8,
+            address!("595F64Ddc3856a3b5Ff4f4CC1d1fb4B46cFd2bAC"),
+        );
+        holesky.insert(
+            HoleskyLidoModule::SimpleDVT as u8,
+            address!("11a93807078f8BB880c1BD0ee4C387537de4b4b6"),
+        );
+        holesky.insert(
+            HoleskyLidoModule::Sandbox as u8,
+            address!("D6C2ce3BB8bea2832496Ac8b5144819719f343AC"),
+        );
+        holesky.insert(
+            HoleskyLidoModule::CommunityStaking as u8,
+            address!("4562c3e63c2e586cD1651B958C22F88135aCAd4f"),
+        );
+        map.insert(Chain::Holesky, holesky);
+
+        // --- Hoodi ---
+        let mut hoodi = HashMap::new();
+        hoodi.insert(
+            HoodiLidoModule::Curated as u8,
+            address!("5cDbE1590c083b5A2A64427fAA63A7cfDB91FbB5"),
+        );
+        hoodi.insert(
+            HoodiLidoModule::SimpleDVT as u8,
+            address!("0B5236BECA68004DB89434462DfC3BB074d2c830"),
+        );
+        hoodi.insert(
+            HoodiLidoModule::Sandbox as u8,
+            address!("682E94d2630846a503BDeE8b6810DF71C9806891"),
+        );
+        hoodi.insert(
+            HoodiLidoModule::CommunityStaking as u8,
+            address!("79CEf36D84743222f37765204Bec41E92a93E59d"),
+        );
+        map.insert(Chain::Hoodi, hoodi);
+
+        // --- Sepolia --
+        let mut sepolia = HashMap::new();
+        sepolia.insert(1, address!("33d6E15047E8644F8DDf5CD05d202dfE587DA6E3"));
+        map.insert(Chain::Sepolia, sepolia);
+
+        map
+    };
 }
 
 // Fetching appropiate registry address
 pub fn lido_registry_address(chain: Chain, lido_module_id: u8) -> eyre::Result<Address> {
-    lido_registry_addresses_by_module()
+    LIDO_REGISTRY_ADDRESSES_BY_MODULE
         .get(&chain)
         .ok_or_else(|| eyre::eyre!("Lido registry not supported for chain: {chain:?}"))?
         .get(&lido_module_id)