withArgs,
withProgName,
#endif
+#ifdef __GLASGOW_HASKELL__
+ getEnvironment,
+#endif
) where
import Prelude
import Control.Exception ( bracket )
import Control.Monad
import GHC.IOBase
-#include "config.h"
#endif
#ifdef __HUGS__
isPathSeparator :: Char -> Bool
isPathSeparator '/' = True
-#ifdef mingw32_TARGET_OS
+#ifdef mingw32_HOST_OS
isPathSeparator '\\' = True
#endif
isPathSeparator _ = False
c_getenv :: CString -> IO (Ptr CChar)
{-|
-@withArgs args act@ - while executing action @act@, have 'System.getArgs'
+'withArgs' @args act@ - while executing action @act@, have 'getArgs'
return @args@.
-}
withArgs :: [String] -> IO a -> IO a
withArgv (p:xs) act
{-|
-@withProgName name act@ - while executing action @act@,
-have 'System.getProgName' return @name@.
+'withProgName' @name act@ - while executing action @act@,
+have 'getProgName' return @name@.
-}
withProgName :: String -> IO a -> IO a
withProgName nm act = do
foreign import ccall unsafe "setProgArgv"
setArgsPrim :: Int -> Ptr CString -> IO ()
+
+-- |'getEnvironment' retrieves the entire environment as a
+-- list of @(key,value)@ pairs.
+--
+-- If an environment entry does not contain an @\'=\'@ character,
+-- the @key@ is the whole entry and the @value@ is the empty string.
+
+getEnvironment :: IO [(String, String)]
+getEnvironment = do
+ pBlock <- getEnvBlock
+ if pBlock == nullPtr then return []
+ else do
+ stuff <- peekArray0 nullPtr pBlock >>= mapM peekCString
+ return (map divvy stuff)
+ where
+ divvy str =
+ case break (=='=') str of
+ (xs,[]) -> (xs,[]) -- don't barf (like Posix.getEnvironment)
+ (name,_:value) -> (name,value)
+
+foreign import ccall unsafe "__hscore_environ"
+ getEnvBlock :: IO (Ptr CString)
#endif /* __GLASGOW_HASKELL__ */