From 9b0a26aff2338ab04248164599535521d68df965 Mon Sep 17 00:00:00 2001
From: Muhammad Nauman Raza <devraza.hazard643@slmail.me>
Date: Sun, 24 Mar 2024 13:04:37 +0000
Subject: [PATCH] feat: show terminal colours (optional, enabled by default)

---
 src/main.rs | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 052cb2e..f90bc7a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,9 +15,13 @@ struct Args {
     /// Show CPU architecture
     #[arg(short = 'x', long)]
     arch: bool,
+
+    /// Hide terminal colours 
+    #[arg(short = 'z', long, default_value_t = false)]
+    hide_colours: bool,
 }
 
-fn cpu_arch(args: Args) {
+fn cpu_arch(args: &Args) {
     if args.arch {
         let arch = arch();
         println!("{:>17} {}", "Arch".cyan().bold(), arch);
@@ -50,7 +54,7 @@ fn main() {
     println!();
     if !args.ascii_only {
         println!("{:>48}", combined);
-        cpu_arch(args);
+        cpu_arch(&args);
         println!("{:>8} {:>6} {}", ascii[0], "OS".blue().bold(), pretty);
         println!("{:>9} {:>9} {}", ascii[1], "Kernel".red().bold(), kernel);
         println!("{:>28} {:>4} {}", ascii[2], "WM".green().bold(), wm);
@@ -59,4 +63,14 @@ fn main() {
             println!("  {}", i);
         }
     }
+
+    if !args.hide_colours {
+        println!();
+        let colors = ["black", "red", "green", "yellow", "blue", "magenta", "cyan", "white"];
+        let mut color_string: String = "    ".to_owned();
+        for color in colors {
+            color_string.push_str(&format!("{:>3}", "●".color(color)));
+        }
+        println!("{}", color_string);
+    }
 }