Add userModule

README.md

@@ -18,14 +18,14 @@ import Prompt
 
 main = do
   run $
-    path
-      ( segment Dull Blue currentDirectoryModule
-          <> ( gitBranchModule
-                 >>= ( \branch ->
-                         segment Dull Cyan (textModule ("\xe725 " ++ branch))
-                     )
-             )
-      )
+    path $
+      segment Dull Magenta userModule
+        <> ( currentDirectoryModule
+               >>= \dir -> segment Dull Blue $ textModule $ " \xf07b " ++ dir
+           )
+        <> ( gitBranchModule
+               >>= \branch -> segment Dull Cyan $ textModule $ " \xe725 " ++ branch
+           )
 ```
 
 ### Cloning from source

app/Prompt.hs

@@ -5,6 +5,7 @@ module Prompt
     textColor,
     currentDirectoryModule,
     gitBranchModule,
+    userModule,
     run,
     Segment (..),
     path,
@@ -23,6 +24,7 @@ import Data.Time (defaultTimeLocale, formatTime, getCurrentTime)
 import GHC.IO.Exception (ExitCode)
 import System.Console.ANSI.Codes (Color (..), ColorIntensity (..), ConsoleIntensity (BoldIntensity), ConsoleLayer (Background, Foreground), SGR (Reset, SetColor, SetConsoleIntensity), setSGRCode)
 import System.Directory (getCurrentDirectory)
+import System.Environment (getEnv)
 import System.Exit (ExitCode (ExitSuccess))
 import System.FilePath (takeFileName)
 import System.Process (readProcessWithExitCode)
@@ -78,6 +80,15 @@ gitBranchModule =
               else Nothing
    in Prompt f
 
+userModule :: Prompt String
+userModule = 
+  let f = do
+        result <- try (getEnv "USER") :: IO (Either SomeException String)
+        return $ case result of
+          Left _ -> Nothing
+          Right name -> Just name
+  in Prompt f
+
 run :: Prompt String -> IO ()
 run (Prompt f) = f >>= \s -> putStr (fromMaybe "" s)
 
@@ -101,7 +112,6 @@ timeModule =
         return $ Just timeStr
    in Prompt f
 
-
 data Segment = Segment ColorIntensity Color (Prompt String)
 
 segment :: ColorIntensity -> Color -> Prompt String -> Prompt [Segment]
@@ -129,7 +139,6 @@ pathSegment intensity color isFirst next m =
              Just (nextIntensity, nextColor) -> backgroundColor nextIntensity nextColor x
              Nothing -> x
 
-
 safeHead :: [a] -> Maybe a
 safeHead [] = Nothing
 safeHead (x : _) = Just x