From 32fdee2fe7b33e58f44adbeae30cfb5e912c8057 Mon Sep 17 00:00:00 2001
From: Henil Dedania <dedaniahenil@gmail.com>
Date: Mon, 5 Dec 2022 18:40:03 +0530
Subject: [PATCH] feat(cli): Make configuration file optional and add a
 subcommand to generate a config file (#523)

---
 rumqttd/src/main.rs | 52 +++++++++++++++++++++++++++++++++++++--------
 1 file changed, 43 insertions(+), 9 deletions(-)

diff --git a/rumqttd/src/main.rs b/rumqttd/src/main.rs
index b60f168e8..260c274a3 100644
--- a/rumqttd/src/main.rs
+++ b/rumqttd/src/main.rs
@@ -1,7 +1,12 @@
-use rumqttd::{Broker, Config};
+use std::io::Write;
+
+use config::FileFormat;
+use rumqttd::Broker;
 
 use structopt::StructOpt;
 
+pub static RUMQTTD_DEFAULT_CONFIG: &str = include_str!("../demo.toml");
+
 #[derive(StructOpt)]
 #[structopt(name = "rumqttd")]
 #[structopt(about = "A high performance, lightweight and embeddable MQTT broker written in Rust.")]
@@ -21,15 +26,30 @@ struct CommandLine {
     commit_date: String,
     /// path to config file
     #[structopt(short, long)]
-    config: String,
+    config: Option<String>,
+    #[structopt(subcommand)]
+    command: Option<Command>,
     /// log level (v: info, vv: debug, vvv: trace)
     #[structopt(short = "v", long = "verbose", parse(from_occurrences))]
     verbose: u8,
 }
 
+#[derive(StructOpt)]
+enum Command {
+    /// Write default configuration file to stdout
+    GenerateConfig,
+}
+
 fn main() {
     let commandline: CommandLine = CommandLine::from_args();
 
+    if let Some(Command::GenerateConfig) = commandline.command {
+        std::io::stdout()
+            .write_all(RUMQTTD_DEFAULT_CONFIG.as_bytes())
+            .unwrap();
+        std::process::exit(0);
+    }
+
     banner(&commandline);
     let level = match commandline.verbose {
         0 => "rumqttd=warn",
@@ -54,17 +74,31 @@ fn main() {
         .try_init()
         .expect("initialized subscriber succesfully");
 
-    let config = config::Config::builder()
-        .add_source(config::File::with_name(&commandline.config))
-        .build()
-        .unwrap();
+    let mut configs: rumqttd::Config;
+    if let Some(config) = &commandline.config {
+        configs = config::Config::builder()
+            .add_source(config::File::with_name(config))
+            .build()
+            .unwrap()
+            .try_deserialize()
+            .unwrap();
+    } else {
+        configs = config::Config::builder()
+            .add_source(config::File::from_str(
+                RUMQTTD_DEFAULT_CONFIG,
+                FileFormat::Toml,
+            ))
+            .build()
+            .unwrap()
+            .try_deserialize()
+            .unwrap();
+    }
 
-    let mut config: Config = config.try_deserialize().unwrap();
-    config.console.set_filter_reload_handle(reload_handle);
+    configs.console.set_filter_reload_handle(reload_handle);
 
     // println!("{:#?}", config);
 
-    let mut broker = Broker::new(config);
+    let mut broker = Broker::new(configs);
     broker.start().unwrap();
 }