From 44929028777f7b15f6fd5627144735a23d360f81 Mon Sep 17 00:00:00 2001
From: El-BG-1970 <elouangros@hotmail.com>
Date: Tue, 26 Apr 2022 22:39:51 +0200
Subject: environment variables can now be accessed

---
 src/main.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index ec0b9ea..df292b5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -8,7 +8,12 @@ fn substitute_vars(word: &str) -> String {
 	match word {
 		"~" => env::var("HOME").unwrap(),
 		w if w.chars().nth(0).unwrap() == '~' => w.replace("~", env::var("HOME").unwrap().as_str()),
-		w if w.chars().nth(0).unwrap() == '$' => String::from_str(w).unwrap(),
+		w if w.chars().nth(0).unwrap() == '$' => {
+			match env::var(&w[1..]) {
+				Ok(val) => val,
+				Err(_) => String::from_str("").unwrap()
+			}
+		},
 		w => String::from_str(w).unwrap()
 	}
 }
-- 
cgit v1.2.3