Moved some config over to an actual config file, improved error handling

src/config.rs

@@ -0,0 +1,29 @@
+use std::{fs, error::Error};
+
+use log::debug;
+use serde::Deserialize;
+
+#[derive(Debug, Deserialize)]
+pub struct Config {
+    pub mqtt: MQTT,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct MQTT {
+    pub host: String,
+    pub port: u16,
+    pub username: String,
+    pub password: Option<String>,
+}
+
+impl Config {
+    pub fn build(filename: &str) -> Result<Self, Box<dyn Error>> {
+        debug!("Loading config: {filename}");
+        let file = fs::read_to_string(filename)?;
+        let mut config: Self = toml::from_str(&file)?;
+
+        config.mqtt.password = Some(std::env::var("MQTT_PASSWORD").or(config.mqtt.password.ok_or("MQTT password needs to be set in either config or the environment!"))?);
+
+        Ok(config)
+    }
+}

src/main.rs

@@ -1,34 +1,32 @@
-use std::{time::Duration, sync::{Arc, RwLock}, process::exit, net::SocketAddr};
+mod config;
 
+use std::{time::Duration, sync::{Arc, RwLock}, process, net::SocketAddr};
+
+use config::Config;
+use dotenv::dotenv;
 use warp::Filter;
 use rumqttc::{MqttOptions, Transport, AsyncClient};
-use dotenv::dotenv;
 use env_logger::Builder;
 use log::{error, info, LevelFilter};
 
 use automation::{devices::{Devices, IkeaOutlet, TestOutlet}, zigbee::Zigbee, mqtt::Notifier};
 use google_home::{GoogleHome, Request};
 
-fn get_required_env(name: &str) -> String {
-    match std::env::var(name) {
-        Ok(value) => value,
-        _ => {
-            error!("Environment variable ${name} is not set!");
-            exit(-1);
-        }
-    }
-}
-
 #[tokio::main]
 async fn main() {
+    dotenv().ok();
+
     // Setup logger
     Builder::new()
         .filter_module("automation", LevelFilter::Info)
         .parse_default_env()
         .init();
 
-    // Load dotfiles
-    dotenv().ok();
+    let config = std::env::var("AUTOMATION_CONFIG").unwrap_or("./config/config.toml".to_owned());
+    let config = Config::build(&config).unwrap_or_else(|err| {
+        error!("Failed to load config: {err}");
+        process::exit(1);
+    });
 
     info!("Starting automation_rs...");
 
@@ -37,8 +35,8 @@ async fn main() {
     let devices = Arc::new(RwLock::new(Devices::new()));
 
     // Setup MQTT
-    let mut mqttoptions = MqttOptions::new("rust-test", get_required_env("MQTT_HOST"), 8883);
-    mqttoptions.set_credentials(get_required_env("MQTT_USERNAME"), get_required_env("MQTT_PASSWORD"));
+    let mut mqttoptions = MqttOptions::new("rust-test", config.mqtt.host, config.mqtt.port);
+    mqttoptions.set_credentials(config.mqtt.username, config.mqtt.password.unwrap());
     mqttoptions.set_keep_alive(Duration::from_secs(5));
     mqttoptions.set_transport(Transport::tls_with_default_config());