feat: add clap for arg parsing

Author: Lloyd Lobo

Cargo.toml

@@ -6,9 +6,13 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+anyhow = "1.0.69"
+clap = { version = "4.1.4", features = ["derive", "cargo"] }
 # anyhow = "1.0.69"
 console = "0.15.5"
 image = "0.24.5"
 indicatif = "0.17.3"
+log = "0.4.17"
+pretty_env_logger = "0.4.0"
 # termion = "2.0.1"
 # tui = { version = "0.19.0", features = ["termion"] }

src/bin/mandelbrot.rs

@@ -1,43 +1,83 @@
-// use console::style;
+use clap::{arg, command, Arg, ArgAction, ArgMatches, Command};
+use console::Style;
 use indicatif::{ProgressBar, ProgressStyle};
+use log::info;
+use pretty_env_logger::env_logger::Builder;
 
 const ITERATIONS: u32 = 255;
 const WIDTH: u32 = 800;
 const HEIGHT: u32 = 800;
 const PATH: &str = "mandelbrot.png";
 
 fn main() {
-    let progress_bar = ProgressBar::new(WIDTH as u64 * HEIGHT as u64);
-    progress_bar.set_style(
-        ProgressStyle::default_bar()
-            .template("[{elapsed_precise}] [{bar:40.cyan/blue}] {percent}%")
-            .unwrap()
-            .progress_chars("##-"),
+    Builder::from_default_env().format_timestamp(None).init();
+
+    if let Err(e) = try_main() {
+        eprintln!("{}", e);
+        std::process::exit(1);
+    }
+}
+
+fn try_main() -> anyhow::Result<()> {
+    let matches: ArgMatches = command!()
+        .arg(
+            Arg::new("ascii")
+                .long("ascii")
+                .help("Generates the Mandelbrot set as ASCII art and print to terminal"),
+        )
+        .arg(
+            Arg::new("text")
+                .long("text")
+                .help("Generates the Mandelbrot set as ASCII art and saves to text file"),
+        )
+        .arg(
+            Arg::new("image")
+                .long("image")
+                .help("Generates the Mandelbrot set as an image and saves to file"),
+        )
+        .after_help(
+            "Longer explanation to appear after the options when displaying the help information \
+             from --help or -h",
+        )
+        .get_matches();
+
+    if let Some(ascii) = matches.get_one::<String>("ascii") {
+        dbg!(ascii);
+        info!("Rendering image Mandelbrot set as {}", Style::new().cyan().apply_to("ASCII"));
+        let pb = ProgressBar::new(WIDTH as u64 * HEIGHT as u64);
+        style_progress_bar(&pb);
+        let image = mandelbrot::mandelbrot_ascii::collect_ascii();
+        pb.finish();
+        mandelbrot::mandelbrot_ascii::print_ascii(image.clone());
+    }
+
+    info!(
+        "Rendering image Mandelbrot set as {}",
+        Style::new().cyan().apply_to("ASCII and saving to file")
     );
+    let pb = ProgressBar::new(WIDTH as u64 * HEIGHT as u64);
+    style_progress_bar(&pb);
     let image = mandelbrot::mandelbrot_ascii::collect_ascii();
-    progress_bar.finish();
-    mandelbrot::mandelbrot_ascii::print_ascii(image.clone());
+    mandelbrot::mandelbrot_ascii::write_ascii_to_file(image);
+    pb.finish_with_message("Wrote ascii to file");
 
-    let progress_bar = ProgressBar::new(WIDTH as u64 * HEIGHT as u64);
-    progress_bar.set_style(
-        ProgressStyle::default_bar()
-            .template("[{elapsed_precise}] [{bar:40.cyan/blue}] {percent}%")
-            .unwrap()
-            .progress_chars("##-"),
+    info!(
+        "Rendering image Mandelbrot set as {}",
+        Style::new().cyan().apply_to("image and saving to file")
     );
-    mandelbrot::mandelbrot_ascii::write_ascii_to_file(image);
-    progress_bar.finish_with_message("Wrote ascii to file");
+    let pb = ProgressBar::new(WIDTH as u64 * HEIGHT as u64);
+    style_progress_bar(&pb);
+    mandelbrot::mandelbrot_img::compose(WIDTH, HEIGHT, ITERATIONS).save(PATH)?;
+    pb.finish_with_message("Saved image to file");
+
+    Ok(())
+}
 
-    let progress_bar = ProgressBar::new(WIDTH as u64 * HEIGHT as u64);
-    progress_bar.set_style(
+fn style_progress_bar(pb: &ProgressBar) {
+    pb.set_style(
         ProgressStyle::default_bar()
             .template("[{elapsed_precise}] [{bar:40.cyan/blue}] {percent}%")
             .unwrap()
             .progress_chars("##-"),
     );
-    if let Err(e) = mandelbrot::mandelbrot_img::compose(WIDTH, HEIGHT, ITERATIONS).save(PATH) {
-        eprintln!("{}", e);
-        std::process::exit(1);
-    }
-    progress_bar.finish_with_message("Saved image to file");
 }