refactor short_summary function in logs.rs to be less redundant

src/logs.rs

@@ -55,77 +55,56 @@ pub fn display_summary(
 
 /// Returns a string summary of the total number of failed and successful article downloads
 fn short_summary(download_count: DownloadCount) -> String {
-    // TODO: Refactor this
     if download_count.total
         != download_count.successful + download_count.failed + download_count.partial
     {
         panic!("initial_count must be equal to the sum of failed and successful count")
     }
     let get_noun = |count: usize| if count == 1 { "article" } else { "articles" };
-    if download_count.successful == download_count.total && download_count.successful == 1 {
+    let get_summary = |count, label, color: Color| {
-        "Article downloaded successfully".green().to_string()
+        if count == 0 {
-    } else if download_count.total == download_count.failed && download_count.failed == 1 {
+            return "".to_string();
-        "Article failed to download".red().to_string()
+        };
-    } else if download_count.total == download_count.partial && download_count.partial == 1 {
+
-        "Article partially failed to download".yellow().to_string()
-    } else if download_count.successful == download_count.total {
-        "All articles downloaded successfully".green().to_string()
-    } else if download_count.failed == download_count.total {
-        "All articles failed to download".red().to_string()
-    } else if download_count.partial == download_count.total {
-        "All articles partially failed to download"
-            .yellow()
-            .to_string()
-    } else if download_count.partial == 0 {
-        format!(
-            "{} {} downloaded successfully, {} {} failed",
-            download_count.successful,
-            get_noun(download_count.successful),
-            download_count.failed,
-            get_noun(download_count.failed)
-        )
-        .yellow()
-        .to_string()
-    } else if download_count.successful == 0
-        && download_count.partial > 0
-        && download_count.failed > 0
         {
-        format!(
+            if count == 1 && count == download_count.total {
-            "{} {} partially failed to download, {} {} failed",
+                "Article".to_string() + label
-            download_count.partial,
+            } else if count == download_count.total {
-            get_noun(download_count.partial),
+                "All ".to_string() + get_noun(count) + label
-            download_count.failed,
-            get_noun(download_count.failed)
-        )
-        .yellow()
-        .to_string()
-    } else if download_count.failed == 0
-        && download_count.successful > 0
-        && download_count.partial > 0
-    {
-        format!(
-            "{} {} downloaded successfully, {} {} partially failed to download",
-            download_count.successful,
-            get_noun(download_count.successful),
-            download_count.partial,
-            get_noun(download_count.partial)
-        )
-        .yellow()
-        .to_string()
             } else {
-        format!(
+                count.to_string() + " " + get_noun(count) + label
-            "{} {} downloaded successfully, {} {} partially failed to download, {} {} failed",
-            download_count.successful,
-            get_noun(download_count.successful),
-            download_count.partial,
-            get_noun(download_count.partial),
-            download_count.failed,
-            get_noun(download_count.failed)
-        )
-        .yellow()
-        .to_string()
             }
         }
+        .color(color)
+        .to_string()
+    };
+
+    let mut summary = get_summary(
+        download_count.successful,
+        " downloaded successfully",
+        Color::BrightGreen,
+    );
+
+    let partial_summary = get_summary(
+        download_count.partial,
+        " partially failed to download",
+        Color::Yellow,
+    );
+
+    if !summary.is_empty() && !partial_summary.is_empty() {
+        summary = summary + ", " + &partial_summary;
+    } else {
+        summary = summary + &partial_summary;
+    }
+
+    let failed_summary = get_summary(download_count.failed, " failed to download", Color::Red);
+    if !summary.is_empty() && !failed_summary.is_empty() {
+        summary = summary + ", " + &failed_summary;
+    } else {
+        summary = summary + &failed_summary;
+    }
+    summary
+}
 
 struct DownloadCount {
     total: usize,
@@ -192,7 +171,7 @@ mod tests {
     fn test_short_summary() {
         assert_eq!(
             short_summary(DownloadCount::new(1, 1, 0, 0)),
-            "Article downloaded successfully".green().to_string()
+            "Article downloaded successfully".bright_green().to_string()
         );
         assert_eq!(
             short_summary(DownloadCount::new(1, 0, 0, 1)),
@@ -200,7 +179,9 @@ mod tests {
         );
         assert_eq!(
             short_summary(DownloadCount::new(10, 10, 0, 0)),
-            "All articles downloaded successfully".green().to_string()
+            "All articles downloaded successfully"
+                .bright_green()
+                .to_string()
         );
         assert_eq!(
             short_summary(DownloadCount::new(10, 0, 0, 10)),
@@ -208,39 +189,52 @@ mod tests {
         );
         assert_eq!(
             short_summary(DownloadCount::new(10, 8, 0, 2)),
-            "8 articles downloaded successfully, 2 articles failed"
+            format!(
-                .yellow()
+                "{}, {}",
-                .to_string()
+                "8 articles downloaded successfully".bright_green(),
+                "2 articles failed to download".red()
+            )
         );
         assert_eq!(
             short_summary(DownloadCount::new(10, 1, 0, 9)),
-            "1 article downloaded successfully, 9 articles failed"
+            format!(
-                .yellow()
+                "{}, {}",
-                .to_string()
+                "1 article downloaded successfully".bright_green(),
+                "9 articles failed to download".red()
+            )
         );
         assert_eq!(
             short_summary(DownloadCount::new(7, 6, 0, 1)),
-            "6 articles downloaded successfully, 1 article failed"
+            format!(
-                .yellow()
+                "{}, {}",
-                .to_string()
+                "6 articles downloaded successfully".bright_green(),
+                "1 article failed to download".red()
+            )
         );
         assert_eq!(
             short_summary(DownloadCount::new(7, 4, 2, 1)),
-            "4 articles downloaded successfully, 2 articles partially failed to download, 1 article failed"
+            format!(
-                .yellow()
+                "{}, {}, {}",
-                .to_string()
+                "4 articles downloaded successfully".bright_green(),
+                "2 articles partially failed to download".yellow(),
+                "1 article failed to download".red()
+            )
         );
         assert_eq!(
             short_summary(DownloadCount::new(12, 6, 6, 0)),
-            "6 articles downloaded successfully, 6 articles partially failed to download"
+            format!(
-                .yellow()
+                "{}, {}",
-                .to_string()
+                "6 articles downloaded successfully".bright_green(),
+                "6 articles partially failed to download".yellow()
+            )
         );
         assert_eq!(
             short_summary(DownloadCount::new(5, 0, 4, 1)),
-            "4 articles partially failed to download, 1 article failed"
+            format!(
-                .yellow()
+                "{}, {}",
-                .to_string()
+                "4 articles partially failed to download".yellow(),
+                "1 article failed to download".red()
+            )
         );
         assert_eq!(
             short_summary(DownloadCount::new(4, 0, 4, 0)),