From f0d5d76908892d8d8549522d65b53c682fbc004a Mon Sep 17 00:00:00 2001
From: Voyage <93342430+AlexanderMaxRanabel@users.noreply.github.com>
Date: Sun, 18 Jun 2023 19:57:00 +0300
Subject: [PATCH 1/2] Update README.md

---
 README.md | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/README.md b/README.md
index 19d0628..b5fc2a7 100644
--- a/README.md
+++ b/README.md
@@ -18,6 +18,10 @@ sudo pacman -S rustup git figlet
 ```bash
 sudo apt install rustup git figlet
 ```
+- NixOS
+```bash
+sudo nix-env -iA nixos.rustup nixos.git nixos.figlet
+```
 If you don't want to install figlet (for the ASCII logo) then you can remove "figlet" from the end of both commands. I'm working on removing figlet from the dependency list alltogether, and packaging the logos in the program, however that would require a lot of testing (likely involving VMs).
 
 ## 2. Clone the repository:

From 90fd10f202a17cce7b5050ba8ae4416b1cbe4633 Mon Sep 17 00:00:00 2001
From: Voyage <93342430+AlexanderMaxRanabel@users.noreply.github.com>
Date: Mon, 19 Jun 2023 22:28:31 +0300
Subject: [PATCH 2/2] Update main.rs

---
 src/main.rs | 38 ++++++++++++++++++++++++--------------
 1 file changed, 24 insertions(+), 14 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 048e4f0..0a96090 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -10,7 +10,7 @@ fn main() {
     // let user = Command::new("whoami")
     //     .output()
     //     .expect("Failed to get user");
-    
+
     // let user = String::from_utf8_lossy(&user.stdout);
 
     // let hostname = Command::new("uname")
@@ -22,14 +22,14 @@ fn main() {
 
    // let hostname = String::from_utf8_lossy(&hostname.stdout);
     let user = env::var("USER");
-    
+
     let mut hostname = String::new();
     {
         let hostname_file = File::open("/etc/hostname").expect("u forgor the /etc/hostname file u arch-using moronbox");
         let mut hostname_reader = BufReader::new(hostname_file);
         hostname_reader.read_line(&mut hostname).expect("Failed string conversion... EEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEE");
     }
-    
+
     // Read release file, AKA get OS name
     let mut release_distro = String::new();
     {
@@ -42,7 +42,7 @@ fn main() {
    let mut distro_name: String = release_distro[5..release_distro.len() - 1].to_string();
    distro_name = distro_name.replace('\"', "");
     // println!("OS: {}", Red.paint(distro_name.clone()));
-    
+
     let figlet = Command::new("figlet")
         .args(["-f", "smslant", &distro_name])
         .output();
@@ -59,7 +59,7 @@ fn main() {
     // I know it's terrible, but it works.
     let mut kernel = String::new();
     {
-        let kernel_file = File::open("/proc/version").expect("do u even linux bro???");
+        let kernel_file = File::open("/proc/version").expect("Read the README.md you dumbass");
         let mut kernel_reader = BufReader::new(kernel_file);
         kernel_reader.read_line(&mut kernel).expect("Failed string conversion");
     }
@@ -81,7 +81,7 @@ fn main() {
         // Vars
         let mut ramtotal: u32 = 0;
         let mut ramavail: u32 = 0;
-        
+
         // Read 1st & 2nd line
         if let Some(Ok(line)) = lines.next() {
             if let Some(idx) = line.find(char::is_whitespace) {
@@ -112,9 +112,9 @@ fn main() {
         let ramused = ramtotal - ramavail;
 
         println!("│ Mem: {}/{} GB ({} GB Available)", Green.paint(ramused.to_string()), Green.paint(ramtotal.to_string()), Green.paint(ramavail.to_string()));
- 
+
     }
-    
+
     {
         // This took me unusually long.
 
@@ -128,16 +128,26 @@ fn main() {
 
         // was never expecting rounding to be this difficult
         let uptimeint = uptime.parse::<f32>();
-        let roundeduptimeint: u32 = uptimeint.expect("phoque").round() as u32; 
+        let roundeduptimeint: u32 = uptimeint.expect("phoque").round() as u32;
         let uptimemins: u32 = roundeduptimeint / 60;
         println!("│ Uptime: {} minutes", Blue.paint(uptimemins.to_string()))
 
     }
-    
-    // Get shell
-    let shell = env::var("SHELL").expect("Could not read $SHELL variable");
-    println!("│ Shell: {}", Cyan.paint(shell));
 
+    // Get shell
+    let shell_raw = env::var("SHELL").expect("Could not read $SHELL variable");
+
+    // Split the path using '/' as the separator
+    // Thanks ChatGPT
+    let parts: Vec<&str> = shell_raw.rsplitn(2, '/').collect();
+
+    // Check if the path contains at least one '/'
+    if parts.len() > 1 {
+        let shell = parts[0];
+        println!("│ Shell: {}", Cyan.paint(shell));
+    } else {
+        println!("Invalid path format.");
+    }
     // Time for a challenge, Get CPU model!
     {
         let file = File::open("/proc/cpuinfo").expect("Could not read /proc/cpuinfo");
@@ -163,5 +173,5 @@ fn main() {
 
     // Colours
    println!("\n{} {} {} {} {} {}", Red.paint("◆"), Green.paint("◆"), Yellow.paint("◆"), Blue.paint("◆"), Purple.paint("◆"), Cyan.paint("◆"));
-    
+
 }