Fetch keys from SSV network

Author: ManuelBilbao

crates/common/src/config/mux.rs

@@ -149,6 +149,8 @@ pub enum MuxKeysLoader {
 pub enum NORegistry {
     #[serde(alias = "lido")]
     Lido,
+    #[serde(alias = "ssv")]
+    SSV,
 }
 
 impl MuxKeysLoader {
@@ -176,6 +178,9 @@ impl MuxKeysLoader {
 
                     fetch_lido_registry_keys(rpc_url, chain, U256::from(*node_operator_id)).await
                 }
+                NORegistry::SSV => {
+                    fetch_ssv_registry_keys(chain, U256::from(*node_operator_id)).await
+                }
             },
         }
     }
@@ -266,6 +271,71 @@ async fn fetch_lido_registry_keys(
     Ok(keys)
 }
 
+async fn fetch_ssv_registry_keys(
+    chain: Chain,
+    node_operator_id: U256,
+) -> eyre::Result<Vec<BlsPublicKey>> {
+    const MAX_PER_PAGE: usize = 100;
+
+    let chain_name = match chain {
+        Chain::Mainnet => "mainnet",
+        Chain::Holesky => "holesky",
+        _ => bail!("SSV network is not supported for chain: {chain:?}"),
+    };
+
+    let client = reqwest::Client::new();
+    let mut pubkeys: Vec<BlsPublicKey> = vec![];
+    let mut page = 1;
+
+    loop {
+        let response = client
+            .get(format!(
+                "https://api.ssv.network/api/v4/{}/validators/in_operator/{}?perPage={}&page={}",
+                chain_name, node_operator_id, MAX_PER_PAGE, page
+            ))
+            .send()
+            .await
+            .map_err(|e| eyre::eyre!("Error sending request to SSV network: {e}"))?
+            .json::<SSVResponse>()
+            .await?;
+
+        pubkeys.extend(response.validators.iter().map(|v| v.pubkey).collect::<Vec<BlsPublicKey>>());
+        page += 1;
+
+        if response.validators.is_empty() || response.validators.len() < MAX_PER_PAGE {
+            ensure!(
+                pubkeys.len() == response.pagination.total,
+                "expected {} keys, got {}",
+                response.pagination.total,
+                pubkeys.len()
+            );
+            break;
+        }
+    }
+
+    let unique = pubkeys.iter().collect::<HashSet<_>>();
+    ensure!(unique.len() == pubkeys.len(), "found duplicate keys in registry");
+
+    Ok(pubkeys)
+}
+
+#[derive(Deserialize)]
+struct SSVResponse {
+    validators: Vec<SSVValidator>,
+    pagination: SSVPagination,
+}
+
+#[derive(Deserialize)]
+struct SSVValidator {
+    #[serde(rename = "public_key")]
+    pubkey: BlsPublicKey,
+}
+
+#[derive(Deserialize)]
+struct SSVPagination {
+    total: usize,
+}
+
 #[cfg(test)]
 mod tests {
     use alloy::{primitives::U256, providers::ProviderBuilder};

examples/configs/pbs_mux.toml

@@ -19,7 +19,7 @@ id = "timing-mux"
 validator_pubkeys = [
     "0x80c7f782b2467c5898c5516a8b6595d75623960b4afc4f71ee07d40985d20e117ba35e7cd352a3e75fb85a8668a3b745",
 ]
-loader = "./mux_keys.example.json"
+loader = "./tests/data/mux_keys.example.json"
 timeout_get_header_ms = 900
 late_in_slot_time_ms = 1500
 
@@ -38,3 +38,11 @@ loader = { registry = "lido", node_operator_id = 8 }
 [[mux.relays]]
 id = "relay-3"
 url = "http://0x80c7f782b2467c5898c5516a8b6595d75623960b4afc4f71ee07d40985d20e117ba35e7cd352a3e75fb85a8668a3b745@fgh.xyz"
+
+[[mux]]
+id = "ssv-mux"
+loader = { registry = "ssv", node_operator_id = 200 }
+
+[[mux.relays]]
+id = "relay-4"
+url = "http://0x80c7f782b2467c5898c5516a8b6595d75623960b4afc4f71ee07d40985d20e117ba35e7cd352a3e75fb85a8668a3b745@fgh.xyz"