Configure network

diff --git a/src/config.rs b/src/config.rs
index 444d54cfd10d2a18c5cf4faa65be4a0bac3ce50c..475d63a316e65adecbe274d4a574730e0b442ff6 100644 (file)
--- a/src/config.rs
+++ b/src/config.rs
@@ -5,6 +5,7 @@ use std::env;
 use std::io::Cursor;
 use std::net::{SocketAddr, ToSocketAddrs};
 
+use bitcoin::Network;
 use bitcoin::hashes::hex::FromHex;
 use bitcoin::secp256k1::PublicKey;
 use futures::stream::{FuturesUnordered, StreamExt};
@@ -17,6 +18,17 @@ pub(crate) const SCHEMA_VERSION: i32 = 8;
 pub(crate) const SNAPSHOT_CALCULATION_INTERVAL: u32 = 3600 * 24; // every 24 hours, in seconds
 pub(crate) const DOWNLOAD_NEW_GOSSIP: bool = true;
 
+pub(crate) fn network() -> Network {
+       let network = env::var("RAPID_GOSSIP_SYNC_SERVER_NETWORK").unwrap_or("Bitcoin".to_string());
+       match network.as_str() {
+               "Bitcoin" => Network::Bitcoin,
+               "Testnet" => Network::Testnet,
+               "Signet" => Network::Signet,
+               "Regtest" => Network::Regtest,
+               _ => panic!("Invalid network"),
+       }
+}
+
 pub(crate) fn network_graph_cache_path() -> &'static str {
        "./res/network_graph.bin"
 }

diff --git a/src/lib.rs b/src/lib.rs
index cc5a350847e2c4c42322d2bf20545d5de900059a..25ce1a2f5015c2a23dd76479fd12f831a854903d 100644 (file)
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -16,7 +16,6 @@ use std::io::BufReader;
 use std::sync::Arc;
 
 use bitcoin::blockdata::constants::genesis_block;
-use bitcoin::Network;
 use bitcoin::secp256k1::PublicKey;
 use lightning::routing::gossip::NetworkGraph;
 use lightning::util::ser::{ReadableArgs, Writeable};
@@ -54,6 +53,7 @@ pub struct SerializedResponse {
 
 impl RapidSyncProcessor {
        pub fn new() -> Self {
+               let network = config::network();
                let logger = TestLogger::new();
                let network_graph = if let Ok(file) = File::open(&config::network_graph_cache_path()) {
                        println!("Initializing from cached network graph…");
@@ -65,10 +65,10 @@ impl RapidSyncProcessor {
                                network_graph
                        } else {
                                println!("Initialization from cached network graph failed: {}", network_graph_result.err().unwrap());
-                               NetworkGraph::new(genesis_block(Network::Bitcoin).header.block_hash(), logger)
+                               NetworkGraph::new(genesis_block(network).header.block_hash(), logger)
                        }
                } else {
-                       NetworkGraph::new(genesis_block(Network::Bitcoin).header.block_hash(), logger)
+                       NetworkGraph::new(genesis_block(network).header.block_hash(), logger)
                };
                let arc_network_graph = Arc::new(network_graph);
                Self {