108 lines
4.0 KiB
Haskell
108 lines
4.0 KiB
Haskell
module Propellor.Property.File where
|
||
|
||
import Propellor
|
||
import Utility.FileMode
|
||
|
||
import System.Posix.Files
|
||
import System.PosixCompat.Types
|
||
|
||
type Line = String
|
||
|
||
-- | Replaces all the content of a file.
|
||
hasContent :: FilePath -> [Line] -> Property NoInfo
|
||
f `hasContent` newcontent = fileProperty ("replace " ++ f)
|
||
(\_oldcontent -> newcontent) f
|
||
|
||
-- | Ensures a file has contents that comes from PrivData.
|
||
--
|
||
-- The file's permissions are preserved if the file already existed.
|
||
-- Otherwise, they're set to 600.
|
||
hasPrivContent :: IsContext c => FilePath -> c -> Property HasInfo
|
||
hasPrivContent f = hasPrivContentFrom (PrivDataSourceFile (PrivFile f) f) f
|
||
|
||
-- | Like hasPrivContent, but allows specifying a source
|
||
-- for PrivData, rather than using PrivDataSourceFile .
|
||
hasPrivContentFrom :: (IsContext c, IsPrivDataSource s) => s -> FilePath -> c -> Property HasInfo
|
||
hasPrivContentFrom = hasPrivContent' writeFileProtected
|
||
|
||
-- | Leaves the file at its default or current mode,
|
||
-- allowing "private" data to be read.
|
||
--
|
||
-- Use with caution!
|
||
hasPrivContentExposed :: IsContext c => FilePath -> c -> Property HasInfo
|
||
hasPrivContentExposed f = hasPrivContentExposedFrom (PrivDataSourceFile (PrivFile f) f) f
|
||
|
||
hasPrivContentExposedFrom :: (IsContext c, IsPrivDataSource s) => s -> FilePath -> c -> Property HasInfo
|
||
hasPrivContentExposedFrom = hasPrivContent' writeFile
|
||
|
||
hasPrivContent' :: (IsContext c, IsPrivDataSource s) => (String -> FilePath -> IO ()) -> s -> FilePath -> c -> Property HasInfo
|
||
hasPrivContent' writer source f context =
|
||
withPrivData source context $ \getcontent ->
|
||
property desc $ getcontent $ \privcontent ->
|
||
ensureProperty $ fileProperty' writer desc
|
||
(\_oldcontent -> lines privcontent) f
|
||
where
|
||
desc = "privcontent " ++ f
|
||
|
||
-- | Ensures that a line is present in a file, adding it to the end if not.
|
||
containsLine :: FilePath -> Line -> Property NoInfo
|
||
f `containsLine` l = f `containsLines` [l]
|
||
|
||
containsLines :: FilePath -> [Line] -> Property NoInfo
|
||
f `containsLines` ls = fileProperty (f ++ " contains:" ++ show ls) go f
|
||
where
|
||
go content = content ++ filter (`notElem` content) ls
|
||
|
||
-- | Ensures that a line is not present in a file.
|
||
-- Note that the file is ensured to exist, so if it doesn't, an empty
|
||
-- file will be written.
|
||
lacksLine :: FilePath -> Line -> Property NoInfo
|
||
f `lacksLine` l = fileProperty (f ++ " remove: " ++ l) (filter (/= l)) f
|
||
|
||
-- | Removes a file. Does not remove symlinks or non-plain-files.
|
||
notPresent :: FilePath -> Property NoInfo
|
||
notPresent f = check (doesFileExist f) $ property (f ++ " not present") $
|
||
makeChange $ nukeFile f
|
||
|
||
fileProperty :: Desc -> ([Line] -> [Line]) -> FilePath -> Property NoInfo
|
||
fileProperty = fileProperty' writeFile
|
||
fileProperty' :: (FilePath -> String -> IO ()) -> Desc -> ([Line] -> [Line]) -> FilePath -> Property NoInfo
|
||
fileProperty' writer desc a f = property desc $ go =<< liftIO (doesFileExist f)
|
||
where
|
||
go True = do
|
||
old <- liftIO $ readFile f
|
||
let new = unlines (a (lines old))
|
||
if old == new
|
||
then noChange
|
||
else makeChange $ viaTmp updatefile f new
|
||
go False = makeChange $ writer f (unlines $ a [])
|
||
|
||
-- viaTmp makes the temp file mode 600.
|
||
-- Replicate the original file's owner and mode.
|
||
updatefile f' content = do
|
||
writer f' content
|
||
s <- getFileStatus f
|
||
setFileMode f' (fileMode s)
|
||
setOwnerAndGroup f' (fileOwner s) (fileGroup s)
|
||
|
||
-- | Ensures a directory exists.
|
||
dirExists :: FilePath -> Property NoInfo
|
||
dirExists d = check (not <$> doesDirectoryExist d) $ property (d ++ " exists") $
|
||
makeChange $ createDirectoryIfMissing True d
|
||
|
||
-- | Ensures that a file/dir has the specified owner and group.
|
||
ownerGroup :: FilePath -> UserName -> GroupName -> Property NoInfo
|
||
ownerGroup f owner group = property (f ++ " owner " ++ og) $ do
|
||
r <- ensureProperty $ cmdProperty "chown" [og, f]
|
||
if r == FailedChange
|
||
then return r
|
||
else noChange
|
||
where
|
||
og = owner ++ ":" ++ group
|
||
|
||
-- | Ensures that a file/dir has the specfied mode.
|
||
mode :: FilePath -> FileMode -> Property NoInfo
|
||
mode f v = property (f ++ " mode " ++ show v) $ do
|
||
liftIO $ modifyFileMode f (\_old -> v)
|
||
noChange
|