summaryrefslogtreecommitdiff
path: root/src/Hakyll/Web/Template.hs
blob: e23b532e98dded12fb07b127f75bda48e3874a19 (plain)
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
-- | This module provides means for reading and applying 'Template's.
--
-- Templates are tools to convert data (pages) into a string. They are
-- perfectly suited for laying out your site.
--
-- Let's look at an example template:
--
-- > <html>
-- >     <head>
-- >         <title>My crazy homepage - $title$</title>
-- >     </head>
-- >     <body>
-- >         <div id="header">
-- >             <h1>My crazy homepage - $title$</h1>
-- >         </div>
-- >         <div id="content">
-- >             $body$
-- >         </div>
-- >         <div id="footer">
-- >             By reading this you agree that I now own your soul
-- >         </div>
-- >     </body>
-- > </html>
--
-- We can use this template to render a 'Page' which has a body and a @$title$@
-- metadata field.
--
-- As you can see, the format is very simple -- @$key$@ is used to render the
-- @$key$@ field from the page, everything else is literally copied. If you want
-- to literally insert @\"$key$\"@ into your page (for example, when you're
-- writing a Hakyll tutorial) you can use
--
-- > <p>
-- >     A literal $$key$$.
-- > </p>
--
-- Because of it's simplicity, these templates can be used for more than HTML:
-- you could make, for example, CSS or JS templates as well.
--
-- In addition to the native format, Hakyll also supports hamlet templates. For
-- more information on hamlet templates, please refer to:
-- <http://hackage.haskell.org/package/hamlet>. Internally, hamlet templates are
-- converted to hakyll templates -- which means that you can only use variable
-- insertion (and not all hamlet's features).
--
-- This is an example of a valid hamlet template. You should place them in
-- files with a @.hamlet@ extension:
--
-- > !!!
-- > <html>
-- >     <head>
-- >         <meta charset="UTF-8">
-- >         <title> MyAweSomeCompany - #{title}
-- >     <body>
-- >         <h1> MyAweSomeCompany - #{title}
-- >         <div id="navigation">
-- >             <a href="/index.html"> Home
-- >             <a href="/about.html"> About
-- >             <a href="/code.html"> Code
-- >         #{body}
{-# LANGUAGE Arrows              #-}
{-# LANGUAGE ScopedTypeVariables #-}
module Hakyll.Web.Template
    ( Template
    , applyTemplate
    , templateCompiler
    , templateCompilerWith
    , applyTemplateCompiler
    ) where


--------------------------------------------------------------------------------
import           Control.Arrow
import           Control.Category             (id)
import qualified Data.Map                     as M
import           Data.Maybe                   (fromMaybe)
import           Data.Tuple                   (swap)
import           Prelude                      hiding (id)
import           System.FilePath              (takeExtension)
import           Text.Hamlet                  (HamletSettings,
                                               defaultHamletSettings)


--------------------------------------------------------------------------------
import           Hakyll.Core.Compiler
import           Hakyll.Core.Identifier
import           Hakyll.Core.Util.Arrow
import           Hakyll.Web.Page.Internal
import           Hakyll.Web.Template.Context
import           Hakyll.Web.Template.Internal
import           Hakyll.Web.Template.Read


--------------------------------------------------------------------------------
applyTemplate :: forall a b. (ArrowChoice a, ArrowMap a)
              => a (String, b) String
              -> a (Template, b) String
applyTemplate context =
    arr (\(tpl, x) -> [(e, x) | e <- unTemplate tpl]) >>>
    mapA applyElement >>^ concat
  where
    applyElement :: a (TemplateElement, b) String
    applyElement = unElement >>> (id ||| context)

    unElement :: a (TemplateElement, b) (Either String (String, b))
    unElement = arr $ \(e, x) -> case e of
        Chunk c -> Left c
        Escaped -> Left "$"
        Key k   -> Right (k, x)


--------------------------------------------------------------------------------
-- | Read a template. If the extension of the file we're compiling is
-- @.hml@ or @.hamlet@, it will be considered as a Hamlet template, and parsed
-- as such.
templateCompiler :: Compiler () Template
templateCompiler = templateCompilerWith defaultHamletSettings


--------------------------------------------------------------------------------
-- | Version of 'templateCompiler' that enables custom settings.
templateCompilerWith :: HamletSettings -> Compiler () Template
templateCompilerWith settings =
    cached "Hakyll.Web.Template.templateCompilerWith" $
        getIdentifier &&& getResourceString >>^ uncurry read'
  where
    read' identifier string =
        if takeExtension (toFilePath identifier) `elem` [".hml", ".hamlet"]
            -- Hamlet template
            then readHamletTemplateWith settings string
            -- Hakyll template
            else readTemplate string


--------------------------------------------------------------------------------
applyTemplateCompiler :: Identifier Template  -- ^ Template
                      -> Context Page         -- ^ Context
                      -> Compiler Page Page   -- ^ Compiler
applyTemplateCompiler identifier context = requireA identifier $
    arr swap >>> applyTemplate context'
  where
    context' = proc (k, x) -> do
        id' <- getIdentifier -< ()
        context -< (k, (id', x))