From 62d9d780208c69b38d90ce160478e29472e5a24d Mon Sep 17 00:00:00 2001 From: Joey Hess Date: Tue, 28 Apr 2015 14:58:29 -0400 Subject: merge generalization of writeFileProtected from propellor --- Utility/FileMode.hs | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'Utility') diff --git a/Utility/FileMode.hs b/Utility/FileMode.hs index f98e1bc87..fd0fe07b5 100644 --- a/Utility/FileMode.hs +++ b/Utility/FileMode.hs @@ -11,6 +11,7 @@ module Utility.FileMode where import System.IO import Control.Monad +import Control.Exception (bracket) import System.PosixCompat.Types import Utility.PosixFiles #ifndef mingw32_HOST_OS @@ -124,7 +125,7 @@ withUmask _ a = a #endif combineModes :: [FileMode] -> FileMode -combineModes [] = 0 +combineModes [] = undefined combineModes [m] = m combineModes (m:ms) = foldl unionFileModes m ms @@ -151,7 +152,11 @@ setSticky f = modifyFileMode f $ addModes [stickyMode] - as writeFile. -} writeFileProtected :: FilePath -> String -> IO () -writeFileProtected file content = withUmask 0o0077 $ +writeFileProtected file content = writeFileProtected' file + (\h -> hPutStr h content) + +writeFileProtected' :: FilePath -> (Handle -> IO ()) -> IO () +writeFileProtected' file writer = withUmask 0o0077 $ withFile file WriteMode $ \h -> do void $ tryIO $ modifyFileMode file $ removeModes otherGroupModes - hPutStr h content + writer h -- cgit v1.2.3