diff options
author | 2012-10-25 18:17:32 -0400 | |
---|---|---|
committer | 2012-10-25 18:17:54 -0400 | |
commit | a0f689bcd87c63fe6efdc5ad112a44241d74259e (patch) | |
tree | e6fc62c9a058e7dd37ff440c4f9252f1f5a34ccb /Utility/UserInfo.hs | |
parent | 2ae218333249132c54360013c7c7f8c07594fa67 (diff) |
Use USER and HOME environment when set, and only fall back to getpwent, which doesn't work with LDAP or NIS.
Diffstat (limited to 'Utility/UserInfo.hs')
-rw-r--r-- | Utility/UserInfo.hs | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/Utility/UserInfo.hs b/Utility/UserInfo.hs new file mode 100644 index 000000000..6e757548a --- /dev/null +++ b/Utility/UserInfo.hs @@ -0,0 +1,32 @@ +{- user info + - + - Copyright 2012 Joey Hess <joey@kitenet.net> + - + - Licensed under the GNU GPL version 3 or higher. + -} + +module Utility.UserInfo ( + myHomeDir, + myUserName +) where + +import Control.Applicative +import System.Posix.User +import System.Posix.Env + +{- Current user's home directory. + - + - getpwent will fail on LDAP or NIS, so use HOME if set. -} +myHomeDir :: IO FilePath +myHomeDir = myVal ["HOME"] homeDirectory + +{- Current user's user name. -} +myUserName :: IO String +myUserName = myVal ["USER", "LOGNAME"] userName + +myVal :: [String] -> (UserEntry -> String) -> IO String +myVal envvars extract = maybe (extract <$> getpwent) return =<< check envvars + where + check [] = return Nothing + check (v:vs) = maybe (check vs) (return . Just) =<< getEnv v + getpwent = getUserEntryForID =<< getEffectiveUserID |