From d2066e85269496d9e1f1e32005a1269547e2488f Mon Sep 17 00:00:00 2001
From: Jordan Orelli <jordanorelli@gmail.com>
Date: Sat, 4 Mar 2023 23:28:00 -0600
Subject: [PATCH] wow it's actually a shell now

---
 src/prompt.rs | 25 ++++++++++++++++++-------
 1 file changed, 18 insertions(+), 7 deletions(-)

diff --git a/src/prompt.rs b/src/prompt.rs
index 4f6a159..54c6532 100644
--- a/src/prompt.rs
+++ b/src/prompt.rs
@@ -14,13 +14,24 @@ impl Prompt {
     }
 
     pub fn print(&self) -> Result<()> {
-        unsafe {
-            Error::check(Console::WriteConsoleA(
-                stdout_handle()?,
-                self.s.as_bytes(),
-                None,
-                None,
-            ))?;
+        match std::env::current_dir() {
+            Ok(d) => unsafe {
+                let text = d.to_str().unwrap().to_owned() + " " + &self.s;
+                Error::check(Console::WriteConsoleA(
+                    stdout_handle()?,
+                    text.as_bytes(),
+                    None,
+                    None,
+                ))?;
+            }
+            Err(_) => unsafe {
+                Error::check(Console::WriteConsoleA(
+                    stdout_handle()?,
+                    self.s.as_bytes(),
+                    None,
+                    None,
+                ))?;
+            },
         }
         Ok(())
     }