added color printing of user, machine and location

bexxmodd · Jul 20, 2021 · 52c3fb2 · 52c3fb2
1 parent 3d72693
commit 52c3fb2
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
 /target
+/.idea
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -17,4 +17,4 @@ users = "0.11"
 sysinfo = "0.17"
 dirs = "3.0"
 chrono = "0.4"
-
+termion = "*"
diff --git a/src/shellname.rs b/src/shellname.rs
@@ -1,4 +1,5 @@
 use sysinfo::SystemExt;
+use termion::color;
 
 pub struct ShellName {
     name: String,
@@ -12,7 +13,14 @@ impl ShellName {
         ShellName {
             name: user.clone(),
             current_dir: current_dir.to_string(),
-            shell_name: user + ":" + current_dir + "§ ",
+            shell_name: format!(
+                "{}{}{}:{}§ {}",
+                color::Fg(color::Red),
+                user,
+                color::Fg(color::Cyan),
+                current_dir,
+                color::Fg(color::Reset),
+            ),
         }
     }
 
@@ -25,8 +33,14 @@ impl ShellName {
                 self.current_dir = dir.to_string();
             }
         }
-
-        self.shell_name = self.name.to_string() + ":" + &self.current_dir + "§ ";
+        self.shell_name = format!(
+                "{}{}{}:{}§ {}",
+                color::Fg(color::Red),
+                self.name.to_string(),
+                color::Fg(color::Cyan),
+                &self.current_dir,
+                color::Fg(color::Reset),
+        )
     }
 }