+{-# LANGUAGE CPP, ForeignFunctionInterface #-}
+
-----------------------------------------------------------------------------
-- |
-- Module : System.Environment
-----------------------------------------------------------------------------
module System.Environment
- (
- getArgs, -- :: IO [String]
+ (
+ getArgs, -- :: IO [String]
getProgName, -- :: IO String
getEnv, -- :: String -> IO String
#ifndef __NHC__
import Prelude
#ifdef __GLASGOW_HASKELL__
+import Data.List
import Foreign
import Foreign.C
-import Control.Exception ( bracket )
+import Control.Exception.Base ( bracket )
import Control.Monad
-import GHC.IOBase
-#include "ghcconfig.h"
+-- import GHC.IO
+import GHC.IO.Exception
#endif
#ifdef __HUGS__
#ifdef __GLASGOW_HASKELL__
getArgs :: IO [String]
-getArgs =
- alloca $ \ p_argc ->
+getArgs =
+ alloca $ \ p_argc ->
alloca $ \ p_argv -> do
getProgArgv p_argc p_argv
p <- fromIntegral `liftM` peek p_argc
argv <- peek p_argv
peekArray (p - 1) (advancePtr argv 1) >>= mapM peekCString
-
+
foreign import ccall unsafe "getProgArgv"
getProgArgv :: Ptr CInt -> Ptr (Ptr CString) -> IO ()
is probably really @FOO.EXE@, and that is what 'getProgName' will return.
-}
getProgName :: IO String
-getProgName =
+getProgName =
alloca $ \ p_argc ->
alloca $ \ p_argv -> do
getProgArgv p_argc p_argv
argv <- peek p_argv
unpackProgName argv
-
-unpackProgName :: Ptr (Ptr CChar) -> IO String -- argv[0]
-unpackProgName argv = do
+
+unpackProgName :: Ptr (Ptr CChar) -> IO String -- argv[0]
+unpackProgName argv = do
s <- peekElemOff argv 0 >>= peekCString
return (basename s)
where
basename f = go f f
where
go acc [] = acc
- go acc (x:xs)
+ go acc (x:xs)
| isPathSeparator x = go xs xs
| otherwise = go acc xs
withCString name $ \s -> do
litstring <- c_getenv s
if litstring /= nullPtr
- then peekCString litstring
+ then peekCString litstring
else ioException (IOError Nothing NoSuchThing "getEnv"
- "no environment variable" (Just name))
+ "no environment variable" Nothing (Just name))
foreign import ccall unsafe "getenv"
c_getenv :: CString -> IO (Ptr CChar)
withArgv new_args act = do
pName <- System.Environment.getProgName
existing_args <- System.Environment.getArgs
- bracket (setArgs new_args)
- (\argv -> do setArgs (pName:existing_args); freeArgv argv)
- (const act)
+ bracket (setArgs new_args)
+ (\argv -> do _ <- setArgs (pName:existing_args)
+ freeArgv argv)
+ (const act)
freeArgv :: Ptr CString -> IO ()
freeArgv argv = do
setArgs :: [String] -> IO (Ptr CString)
setArgs argv = do
vs <- mapM newCString argv >>= newArray0 nullPtr
- setArgsPrim (length argv) vs
+ setArgsPrim (genericLength argv) vs
return vs
foreign import ccall unsafe "setProgArgv"
- setArgsPrim :: Int -> Ptr CString -> IO ()
+ setArgsPrim :: CInt -> Ptr CString -> IO ()
-- |'getEnvironment' retrieves the entire environment as a
-- list of @(key,value)@ pairs.
stuff <- peekArray0 nullPtr pBlock >>= mapM peekCString
return (map divvy stuff)
where
- divvy str =
+ divvy str =
case break (=='=') str of
(xs,[]) -> (xs,[]) -- don't barf (like Posix.getEnvironment)
(name,_:value) -> (name,value)