change prompt string in "normal" mode

1 files changed, 6 insertions, 1 deletions

diff --git a/Main.hs b/Main.hs
index ca76529..101ec89 100644
--- a/Main.hs
+++ b/Main.hs
@@ -292,7 +292,12 @@ renderInputLine m (lhs, rhs) = do
     moveCursorLeft (length (show m) - 1)
     putStr $ "\x1b[1;30m" ++ show m ++ "\x1b[m"
     unsaveCursor
-    putStr $ "> " ++ pp lhs ++ pp rhs
+
+    let promptString = case m of
+                        NormalMode -> "\x1b[33;1m@\x1b[m "
+                        InsertMode -> "> "
+
+    putStr $ promptString ++ pp lhs ++ pp rhs
     moveCursorLeft (length $ ppVis rhs)
   where
     pp = concat . map reform