feat: added progress indicator at the start of create

eksup/Cargo.toml

@@ -31,6 +31,7 @@ aws-sdk-autoscaling = "0.25"
 aws-sdk-ec2 = "0.25"
 aws-sdk-eks = "0.25"
 aws-types = "0.55"
+indicatif = "0.16.2"
 clap = { version = "4.0", features = ["derive", "string"] }
 clap-verbosity-flag = "2.0"
 handlebars = { version = "4.3", features = ["rust-embed"] }

eksup/src/lib.rs

@@ -5,6 +5,7 @@ mod k8s;
 mod output;
 mod playbook;
 mod version;
+mod progress;
 
 use std::{env, process, str};
 
@@ -14,6 +15,7 @@ use aws_types::region::Region;
 use clap::{Args, Parser, Subcommand};
 use clap_verbosity_flag::Verbosity;
 use serde::{Deserialize, Serialize};
+use progress::ProgressTracker;
 
 #[derive(Parser, Debug)]
 #[command(author, about, version)]
@@ -98,6 +100,7 @@ pub async fn analyze(args: &Analysis) -> Result<()> {
   // All checks and validations on input should happen above/before running the analysis
   let results = analysis::analyze(&aws_config, &cluster).await?;
   output::output(&results, &args.format, &args.output).await?;
+  progress_tracker.set_progress(75);
 
   Ok(())
 }
@@ -118,6 +121,8 @@ async fn get_config(region: &Option<String>) -> Result<aws_config::SdkConfig> {
 pub async fn create(args: &Create) -> Result<()> {
   match &args.command {
     CreateCommands::Playbook(playbook) => {
+      // start the progress tracker
+      let progress_tracker = ProgressTracker::new();
       // Query Kubernetes first so that we can get AWS details that require them
       let aws_config = get_config(&playbook.region.to_owned()).await?;
       let region = aws_config.region().unwrap().to_string();
@@ -132,12 +137,15 @@ pub async fn create(args: &Create) -> Result<()> {
         return Ok(());
       }
 
+      progress_tracker.set_progress(25);
       let results = analysis::analyze(&aws_config, &cluster).await?;
+      progress_tracker.set_progress(50);
 
       if let Err(err) = playbook::create(playbook, region, &cluster, results) {
         eprintln!("{err}");
         process::exit(2);
       }
+      progress_tracker.finish();
     }
   }
 

eksup/src/progress.rs

@@ -0,0 +1,24 @@
+use indicatif::{ProgressBar, ProgressStyle};
+
+pub struct ProgressTracker {
+    progress_bar: ProgressBar,
+}
+
+impl ProgressTracker {
+    pub fn new() -> Self {
+        let progress_bar = ProgressBar::new(100);
+        progress_bar.set_style(ProgressStyle::default_bar()
+            .template("{spinner:.green} [{elapsed_precise}] [{bar:40.cyan/blue}] {percent}% ({eta})")
+            .progress_chars("=> "));
+
+        Self { progress_bar }
+    }
+
+    pub fn set_progress(&self, progress: u8) {
+        self.progress_bar.set_position(progress as u64);
+    }
+
+    pub fn finish(&self) {
+        self.progress_bar.finish();
+    }
+}