add custom url

zkLinkProtocol · Jun 16, 2024 · 0f49cb8 · 0f49cb8
1 parent e582ca6
commit 0f49cb8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/provider/src/network.rs b/provider/src/network.rs
@@ -3,7 +3,7 @@ use serde::{Deserialize, Serialize};
 use std::str::FromStr;
 
 /// Network to be used for a zklink client.
-#[derive(Debug, Serialize, Deserialize, Clone, Copy, PartialEq, Eq)]
+#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
 #[serde(rename_all = "camelCase")]
 pub enum Network {
     /// Mainnet.
@@ -12,6 +12,8 @@ pub enum Network {
     TestNet,
     /// Develop network
     DevNet,
+    /// Custom url
+    Custom(String),
 }
 
 impl Network {
@@ -20,6 +22,7 @@ impl Network {
             Network::MainNet => "https://api-v1.zk.link",
             Network::TestNet => "https://aws-gw-v2.zk.link",
             Network::DevNet => "https://dev-gw-v1.zk.link",
+            Network::Custom(s) => &s,
         }
     }
 }
@@ -32,7 +35,7 @@ impl FromStr for Network {
             "mainet" => Ok(Network::MainNet),
             "testnet" => Ok(Network::TestNet),
             "devnet" => Ok(Network::DevNet),
-            _ => Err(RpcError::InvalidNetwork),
+            _ => Ok(Network::Custom(s.to_string())),
         }
     }
 }