Cleanup

broker/src/main.rs

@@ -10,10 +10,8 @@ use crate::chain_tracker::MqttSignerPort;
 use crate::mqtt::start_broker;
 use crate::unix_fd::SignerLoop;
 use crate::util::read_broker_config;
-use bitcoin::Network;
 use clap::{App, AppSettings, Arg};
 use std::env;
-use std::str::FromStr;
 use std::sync::Arc;
 use tokio::sync::{mpsc, oneshot};
 use url::Url;
@@ -75,11 +73,10 @@ fn main() -> anyhow::Result<()> {
     }
 
     let settings = read_broker_config(BROKER_CONFIG_PATH);
-    let network = Network::from_str(settings["network"].as_str().unwrap()).unwrap();
 
     let (tx, rx) = mpsc::channel(1000);
     let (status_tx, mut status_rx) = mpsc::channel(1000);
-    log::info!("=> start broker on network: {}", network);
+    log::info!("=> start broker on network: {}", settings.network);
     let runtime = start_broker(rx, status_tx, "sphinx-1", &settings);
     log::info!("=> wait for connected status");
     // wait for connection = true
@@ -93,7 +90,7 @@ fn main() -> anyhow::Result<()> {
         let frontend = Frontend::new(
             Arc::new(SignerPortFront {
                 signer_port: Box::new(signer_port),
-                network,
+                network: settings.network,
             }),
             Url::parse(&btc_url).expect("malformed btc rpc url"),
         );

broker/src/mqtt.rs

@@ -1,3 +1,4 @@
+use crate::util::Settings;
 use crate::{ChannelReply, ChannelRequest};
 use librumqttd::{
     async_locallink,
@@ -11,7 +12,6 @@ use std::thread;
 use std::time::Duration;
 use tokio::sync::mpsc;
 use tokio::time::timeout;
-use toml::Value;
 
 const SUB_TOPIC: &str = "sphinx-return";
 const PUB_TOPIC: &str = "sphinx";
@@ -32,7 +32,7 @@ pub fn start_broker(
     mut receiver: mpsc::Receiver<ChannelRequest>,
     status_sender: mpsc::Sender<bool>,
     expected_client_id: &str,
-    settings: &Value,
+    settings: &Settings,
 ) -> tokio::runtime::Runtime {
     let config = config(settings);
     let client_id = expected_client_id.to_string();
@@ -148,7 +148,7 @@ fn metrics_to_status(metrics: ConnectionMetrics, client_connected: bool) -> Opti
     }
 }
 
-fn config(settings: &Value) -> Config {
+fn config(settings: &Settings) -> Config {
     use librumqttd::rumqttlog::Config as RouterConfig;
     use librumqttd::{
         ConnectionLoginCredentials, ConnectionSettings, ConsoleSettings, ServerSettings,
@@ -169,11 +169,7 @@ fn config(settings: &Value) -> Config {
         id.to_string(),
         ServerSettings {
             cert: None,
-            listen: SocketAddrV4::new(
-                Ipv4Addr::new(0, 0, 0, 0),
-                settings["port"].as_integer().unwrap().try_into().unwrap(),
-            )
-            .into(),
+            listen: SocketAddrV4::new(Ipv4Addr::new(0, 0, 0, 0), settings.port).into(),
             next_connection_delay_ms: 1,
             connections: ConnectionSettings {
                 connection_timeout_ms: 5000,

broker/src/run_test.rs

@@ -1,9 +1,8 @@
 use crate::mqtt::start_broker;
+use crate::util::Settings;
 use crate::ChannelRequest;
 use sphinx_key_parser as parser;
 use tokio::sync::{mpsc, oneshot};
-use toml::map::Map;
-use toml::Value;
 use vls_protocol::serde_bolt::WireString;
 use vls_protocol::{msgs, msgs::Message};
 
@@ -15,10 +14,7 @@ pub fn run_test() {
     let mut id = 0u16;
     let mut sequence = 1;
 
-    let mut map = Map::new();
-    map.insert("port".to_string(), Value::Integer(1883));
-    map.insert("network".to_string(), Value::String("regtest".to_string()));
-    let settings = Value::Table(map);
+    let settings = Settings::default();
 
     let (tx, rx) = mpsc::channel(1000);
     let (status_tx, mut status_rx) = mpsc::channel(1000);

broker/src/util.rs

@@ -1,21 +1,40 @@
+use bitcoin::Network;
+use std::default::Default;
 use std::env;
 use std::fs;
 use std::str::FromStr;
-use toml::map::Map;
 use toml::Value;
 
-pub fn read_broker_config(config_path: &str) -> Value {
-    let mut ret = Value::Table(Map::new());
+pub struct Settings {
+    pub network: Network,
+    pub port: u16,
+}
+
+impl Default for Settings {
+    fn default() -> Self {
+        Settings {
+            network: Network::Regtest,
+            port: 1883,
+        }
+    }
+}
+
+pub fn read_broker_config(config_path: &str) -> Settings {
+    let mut settings = Settings::default();
     if let Ok(set) = fs::read_to_string(config_path) {
-        ret = Value::from_str(&set)
+        let table = Value::from_str(&set)
             .expect("Couldn't read broker.conf make sure it follows the toml format");
         log::info!("Read broker.conf");
+        if let Some(network) = read_network_setting(&table) {
+            settings.network = network
+        }
+        if let Some(port) = read_port_setting(&table) {
+            settings.port = port
+        }
     } else {
         log::info!("File broker.conf not found, using default settings");
     }
-    validate_network_setting(&mut ret);
-    validate_port_setting(&mut ret);
-    ret
+    settings
 }
 
 pub fn setup_logging(who: &str, level_arg: &str) {
@@ -50,36 +69,31 @@ pub fn setup_logging(who: &str, level_arg: &str) {
         .expect("log config");
 }
 
-fn validate_network_setting(settings: &mut Value) {
-    if let None = settings.get("network") {
+fn read_network_setting(table: &Value) -> Option<Network> {
+    if let None = table.get("network") {
         log::info!("Network not specified, setting to default regtest");
-        settings
-            .as_table_mut()
-            .unwrap()
-            .insert("network".to_string(), Value::String("regtest".to_string()));
+        None
     } else {
-        if !settings["network"].is_str()
-            || settings["network"].as_str().unwrap() != "bitcoin"
-                && settings["network"].as_str().unwrap() != "regtest"
+        if !table["network"].is_str()
+            || table["network"].as_str().unwrap() != "bitcoin"
+                && table["network"].as_str().unwrap() != "regtest"
         {
             panic!("The network must be set to either 'bitcoin' or 'regtest'");
         }
         log::info!(
             "Read network setting: {}",
-            settings["network"].as_str().unwrap()
+            table["network"].as_str().unwrap()
         );
+        Some(Network::from_str(table["network"].as_str().unwrap()).unwrap())
     }
 }
 
-fn validate_port_setting(settings: &mut Value) {
-    if let None = settings.get("port") {
+fn read_port_setting(table: &Value) -> Option<u16> {
+    if let None = table.get("port") {
         log::info!("Broker port not specified, setting to default 1883");
-        settings
-            .as_table_mut()
-            .unwrap()
-            .insert("port".to_string(), Value::Integer(1883));
+        None
     } else {
-        let temp = settings["port"]
+        let temp = table["port"]
             .as_integer()
             .expect("The port number is not an integer greater than 1023");
         if temp <= 1023 {
@@ -89,5 +103,6 @@ fn validate_port_setting(settings: &mut Value) {
             panic!("The port number is way too big!")
         }
         log::info!("Read broker port setting: {}", temp);
+        Some(temp.try_into().unwrap())
     }
 }