101 lines
3.9 KiB
Haskell
101 lines
3.9 KiB
Haskell
--------------------------------------------------------------------------------
|
|
{-# LANGUAGE CPP #-}
|
|
module Hakyll.Core.Provider.Modified
|
|
( resourceModified
|
|
, resourceModificationTime
|
|
) where
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
import Control.Applicative ((<$>), (<*>))
|
|
import Control.Monad (when)
|
|
import qualified Crypto.Hash.MD5 as MD5
|
|
import qualified Data.ByteString as B
|
|
import qualified Data.ByteString.Lazy as BL
|
|
import Data.IORef
|
|
import qualified Data.Map as M
|
|
import Data.Time (UTCTime)
|
|
import System.Directory (getModificationTime)
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
#if !MIN_VERSION_directory(1,2,0)
|
|
import Data.Time (readTime)
|
|
import System.Locale (defaultTimeLocale)
|
|
import System.Time (formatCalendarTime,
|
|
toCalendarTime)
|
|
#endif
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
import Hakyll.Core.Identifier
|
|
import Hakyll.Core.Provider.Internal
|
|
import Hakyll.Core.Provider.MetadataCache
|
|
import Hakyll.Core.Store (Store)
|
|
import qualified Hakyll.Core.Store as Store
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | A resource is modified if it or its metadata has changed
|
|
resourceModified :: Provider -> Identifier -> IO Bool
|
|
resourceModified p r
|
|
| not exists = return False
|
|
| otherwise = do
|
|
cache <- readIORef cacheRef
|
|
case M.lookup normalized cache of
|
|
Just m -> return m
|
|
Nothing -> do
|
|
-- Check if the actual file was modified, and do a recursive
|
|
-- call to check if the metadata file was modified
|
|
m <- (||)
|
|
<$> fileDigestModified store filePath
|
|
<*> resourceModified p (resourceMetadataResource r)
|
|
modifyIORef cacheRef (M.insert normalized m)
|
|
|
|
-- Important! (But ugly)
|
|
when m $ resourceInvalidateMetadataCache p r
|
|
|
|
return m
|
|
where
|
|
normalized = setVersion Nothing r
|
|
exists = resourceExists p r
|
|
store = providerStore p
|
|
cacheRef = providerModifiedCache p
|
|
filePath = resourceFilePath p r
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | Utility: Check if a the digest of a file was modified
|
|
fileDigestModified :: Store -> FilePath -> IO Bool
|
|
fileDigestModified store fp = do
|
|
-- Get the latest seen digest from the store, and calculate the current
|
|
-- digest for the
|
|
lastDigest <- Store.get store key
|
|
newDigest <- fileDigest fp
|
|
if Store.Found newDigest == lastDigest
|
|
-- All is fine, not modified
|
|
then return False
|
|
-- Resource modified; store new digest
|
|
else do
|
|
Store.set store key newDigest
|
|
return True
|
|
where
|
|
key = ["Hakyll.Core.Resource.Provider.fileModified", fp]
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | Utility: Retrieve a digest for a given file
|
|
fileDigest :: FilePath -> IO B.ByteString
|
|
fileDigest = fmap MD5.hashlazy . BL.readFile
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
resourceModificationTime :: Provider -> Identifier -> IO UTCTime
|
|
resourceModificationTime p i = do
|
|
#if MIN_VERSION_directory(1,2,0)
|
|
getModificationTime $ resourceFilePath p i
|
|
#else
|
|
ct <- toCalendarTime =<< getModificationTime (resourceFilePath p i)
|
|
let str = formatCalendarTime defaultTimeLocale "%s" ct
|
|
return $ readTime defaultTimeLocale "%s" str
|
|
#endif
|