fix(dips): allow specifying additional networks in config

Author: pcarranzav

crates/config/maximal-config-example.toml

@@ -165,3 +165,6 @@ price_per_entity = 1000
 [dips.price_per_epoch]
 mainnet = 100
 hardhat = 100
+
+[dips.additional_networks]
+eip155:1337 = "hardhat"

crates/config/src/config.rs

@@ -399,6 +399,7 @@ pub struct DipsConfig {
 
     pub price_per_entity: u64,
     pub price_per_epoch: BTreeMap<String, u64>,
+    pub additional_networks: HashMap<String, String>,
 }
 
 impl Default for DipsConfig {
@@ -409,6 +410,7 @@ impl Default for DipsConfig {
             allowed_payers: vec![],
             price_per_entity: 100,
             price_per_epoch: BTreeMap::new(),
+            additional_networks: HashMap::new(),
         }
     }
 }
@@ -485,6 +487,10 @@ mod tests {
                 ("mainnet".to_string(), 100),
                 ("hardhat".to_string(), 100),
             ]),
+            additional_networks: HashMap::from([(
+                "eip155:1337".to_string(),
+                "hardhat".to_string(),
+            )]),
             ..Default::default()
         });
 

crates/dips/src/lib.rs

@@ -312,6 +312,7 @@ pub async fn validate_and_create_agreement(
         price_calculator,
         signer_validator,
         registry,
+        additional_networks,
     } = ctx.as_ref();
     let decoded_voucher = SignedIndexingAgreementVoucher::abi_decode(voucher.as_ref(), true)
         .map_err(|e| DipsError::AbiDecoding(e.to_string()))?;
@@ -340,7 +341,10 @@ pub async fn validate_and_create_agreement(
 
     let network = match registry.get_network_by_id(&metadata.chainId) {
         Some(network) => network.id.clone(),
-        None => return Err(DipsError::UnsupportedChainId(metadata.chainId)),
+        None => match additional_networks.get(&metadata.chainId) {
+            Some(network) => network.clone(),
+            None => return Err(DipsError::UnsupportedChainId(metadata.chainId)),
+        },
     };
 
     let offered_epoch_price = metadata.basePricePerEpoch;

crates/dips/src/server.rs

@@ -1,7 +1,7 @@
 // Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
 // SPDX-License-Identifier: Apache-2.0
 
-use std::sync::Arc;
+use std::{collections::HashMap, sync::Arc};
 
 use async_trait::async_trait;
 use graph_networks_registry::NetworksRegistry;
@@ -31,6 +31,7 @@ pub struct DipsServerContext {
     pub price_calculator: PriceCalculator,
     pub signer_validator: Arc<dyn SignerValidator>,
     pub registry: Arc<NetworksRegistry>,
+    pub additional_networks: Arc<HashMap<String, String>>,
 }
 
 impl DipsServerContext {
@@ -48,6 +49,7 @@ impl DipsServerContext {
             price_calculator: PriceCalculator::for_testing(),
             signer_validator: Arc::new(signers::NoopSignerValidator),
             registry: Arc::new(test_registry()),
+            additional_networks: Arc::new(HashMap::new()),
         })
     }
 
@@ -61,6 +63,7 @@ impl DipsServerContext {
             price_calculator: PriceCalculator::for_testing(),
             signer_validator: Arc::new(signers::EscrowSignerValidator::mock(accounts).await),
             registry: Arc::new(crate::registry::test_registry()),
+            additional_networks: Arc::new(HashMap::new()),
         })
     }
 
@@ -76,6 +79,7 @@ impl DipsServerContext {
             price_calculator: PriceCalculator::for_testing(),
             signer_validator: Arc::new(signers::EscrowSignerValidator::mock(accounts).await),
             registry: Arc::new(test_registry()),
+            additional_networks: Arc::new(HashMap::new()),
         })
     }
 }

crates/service/src/service.rs

@@ -132,6 +132,7 @@ pub async fn run() -> anyhow::Result<()> {
             allowed_payers,
             price_per_entity,
             price_per_epoch,
+            additional_networks,
         } = dips;
 
         let addr = format!("{}:{}", host, port)
@@ -161,6 +162,7 @@ pub async fn run() -> anyhow::Result<()> {
             price_calculator: PriceCalculator::new(price_per_epoch.clone(), *price_per_entity),
             signer_validator: Arc::new(EscrowSignerValidator::new(watcher)),
             registry: Arc::new(registry),
+            additional_networks: Arc::new(additional_networks.clone()),
         };
 
         let dips = DipsServer {