git-annex/Logs/FsckResults.hs
2021-06-30 16:42:08 -04:00

63 lines
1.7 KiB
Haskell

{- git-annex fsck results log files
-
- Copyright 2013 Joey Hess <id@joeyh.name>
-
- Licensed under the GNU AGPL version 3 or higher.
-}
module Logs.FsckResults (
writeFsckResults,
readFsckResults,
clearFsckResults,
) where
import Annex.Common
import Git.Fsck
import Git.Types
import Logs.File
import qualified Utility.RawFilePath as R
import qualified Data.Set as S
writeFsckResults :: UUID -> FsckResults -> Annex ()
writeFsckResults u fsckresults = do
logfile <- fromRepo $ gitAnnexFsckResultsLog u
case serializeFsckResults fsckresults of
Just s -> store s logfile
Nothing -> liftIO $
removeWhenExistsWith R.removeLink logfile
where
store s logfile = writeLogFile logfile s
serializeFsckResults :: FsckResults -> Maybe String
serializeFsckResults fsckresults = case fsckresults of
FsckFailed -> Just (serialize S.empty False)
FsckFoundMissing s t
| S.null s -> Nothing
| otherwise -> Just (serialize s t)
where
serialize s t =
let ls = map fromRef (S.toList s)
in if t
then unlines ("truncated":ls)
else unlines ls
readFsckResults :: UUID -> Annex FsckResults
readFsckResults u = do
logfile <- fromRepo $ gitAnnexFsckResultsLog u
liftIO $ catchDefaultIO (FsckFoundMissing S.empty False) $
deserializeFsckResults <$> readFile (fromRawFilePath logfile)
deserializeFsckResults :: String -> FsckResults
deserializeFsckResults = deserialize . lines
where
deserialize ("truncated":ls) = deserialize' ls True
deserialize ls = deserialize' ls False
deserialize' ls t =
let s = S.fromList $ map (Ref . encodeBS') ls
in if S.null s then FsckFailed else FsckFoundMissing s t
clearFsckResults :: UUID -> Annex ()
clearFsckResults = liftIO . removeWhenExistsWith R.removeLink
<=< fromRepo . gitAnnexFsckResultsLog