1 -----------------------------------------------------------------------------
2 -- $Id: TmpFiles.hs,v 1.6 2000/10/27 14:56:00 sewardj Exp $
4 -- Temporary file management
6 -- (c) The University of Glasgow 2000
8 -----------------------------------------------------------------------------
12 initTempFileStorage, -- :: IO ()
13 cleanTempFiles, -- :: IO ()
14 newTempName, -- :: Suffix -> IO FilePath
15 addFilesToClean -- :: [FilePath] -> IO ()
32 #include "HsVersions.h"
34 GLOBAL_VAR(v_FilesToClean, [], [String] )
35 GLOBAL_VAR(v_TmpDir, cDEFAULT_TMPDIR, String )
38 initTempFileStorage = do
39 -- check whether TMPDIR is set in the environment
40 IO.try (do dir <- getEnv "TMPDIR" -- fails if not set
41 writeIORef v_TmpDir dir)
44 cleanTempFiles :: Bool -> IO ()
45 cleanTempFiles verbose = do
46 fs <- readIORef v_FilesToClean
49 (do when verbose (hPutStrLn stderr ("removing: " ++ f))
50 if '*' `elem` f then system ("rm -f " ++ f) >> return ()
53 (\_ -> when verbose (hPutStrLn stderr
54 ("warning: can't remove tmp file" ++ f)))
59 -- find a temporary name that doesn't already exist.
60 newTempName :: Suffix -> IO FilePath
63 tmp_dir <- readIORef v_TmpDir
64 findTempName tmp_dir x
65 where findTempName tmp_dir x = do
66 let filename = tmp_dir ++ "/ghc" ++ show x ++ '.':extn
67 b <- doesFileExist filename
68 if b then findTempName tmp_dir (x+1)
69 else do add v_FilesToClean filename -- clean it up later
72 addFilesToClean :: [FilePath] -> IO ()
73 addFilesToClean files = mapM_ (add v_FilesToClean) files
75 add :: IORef [a] -> a -> IO ()