src/Hakyll/Core/Store.hs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

-- | A store for stroing and retreiving items
--
module Hakyll.Core.Store
    ( Store
    , makeStore
    , storeSet
    , storeGet
    ) where

import Control.Applicative ((<$>))
import System.FilePath ((</>))
import System.Directory (doesFileExist)

import Data.Binary (Binary, encodeFile, decodeFile)

import Hakyll.Core.Identifier
import Hakyll.Core.Util.File

-- | Data structure used for the store
--
data Store = Store
    { storeDirectory :: FilePath
    }

-- | Initialize the store
--
makeStore :: FilePath -> IO Store
makeStore directory = return Store {storeDirectory = directory}

-- | Create a path
--
makePath :: Store -> String -> Identifier -> FilePath
makePath store name identifier =
    storeDirectory store </> name </> toFilePath identifier

-- | Store an item
--
storeSet :: Binary a => Store -> String -> Identifier -> a -> IO ()
storeSet store name identifier value = do
    makeDirectories path
    encodeFile path value
  where
    path = makePath store name identifier

-- | Load an item
--
storeGet :: Binary a => Store -> String -> Identifier -> IO (Maybe a)
storeGet store name identifier = do
    exists <- doesFileExist path
    if exists then Just <$> decodeFile path
              else return Nothing
  where
    path = makePath store name identifier