diff options
author | 2014-06-09 15:24:05 -0400 | |
---|---|---|
committer | 2014-06-09 15:24:05 -0400 | |
commit | fb31e247cb0dea919824f5e3bd9086eef382e4bc (patch) | |
tree | 722c4dd581ae7d05083418b36c632ac841acef63 /Utility | |
parent | 6cd15cc11bf66dc0db15dcb339d61e9fc3628f89 (diff) |
Avoid leaving behind .tmp files when failing in some cases, including importing files to a disk that is full.
Diffstat (limited to 'Utility')
-rw-r--r-- | Utility/Tmp.hs | 21 |
1 files changed, 14 insertions, 7 deletions
diff --git a/Utility/Tmp.hs b/Utility/Tmp.hs index 0dc9f2c05..bed30bb4d 100644 --- a/Utility/Tmp.hs +++ b/Utility/Tmp.hs @@ -25,13 +25,20 @@ type Template = String - then moving it into place. The temp file is stored in the same - directory as the final file to avoid cross-device renames. -} viaTmp :: (FilePath -> String -> IO ()) -> FilePath -> String -> IO () -viaTmp a file content = do - let (dir, base) = splitFileName file - createDirectoryIfMissing True dir - (tmpfile, handle) <- openTempFile dir (base ++ ".tmp") - hClose handle - a tmpfile content - rename tmpfile file +viaTmp a file content = bracket setup cleanup use + where + (dir, base) = splitFileName file + template = base ++ ".tmp" + setup = do + createDirectoryIfMissing True dir + openTempFile dir template + cleanup (tmpfile, handle) = do + _ <- tryIO $ hClose handle + tryIO $ removeFile tmpfile + use (tmpfile, handle) = do + hClose handle + a tmpfile content + rename tmpfile file {- Runs an action with a tmp file located in the system's tmp directory - (or in "." if there is none) then removes the file. -} |