{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TemplateHaskell   #-}

--------------------------------------------------------------------------------
-- | A Module that allows easy rendering of RSS feeds.
--
-- The main rendering functions (@renderRss@, @renderAtom@) all assume that
-- you pass the list of items so that the most recent entry in the feed is the
-- first item in the list.
--
-- Also note that the context should have (at least) the following fields to
-- produce a correct feed:
--
-- - @$title$@: Title of the item
--
-- - @$description$@: Description to appear in the feed
--
-- - @$url$@: URL to the item - this is usually set automatically.
--
-- In addition, the posts should be named according to the rules for
-- 'Hakyll.Web.Template.Context.dateField'
module Hakyll.Web.Feed
    ( FeedConfiguration (..)
    , renderRss
    , renderAtom
    , renderRssWithTemplates
    , renderAtomWithTemplates
    ) where


--------------------------------------------------------------------------------
import           Hakyll.Core.Compiler
import           Hakyll.Core.Item
import           Hakyll.Core.Util.String     (replaceAll)
import           Hakyll.Web.Template
import           Hakyll.Web.Template.Context
import           Hakyll.Web.Template.List


--------------------------------------------------------------------------------
import           Data.FileEmbed              (makeRelativeToProject)
import           System.FilePath             ((</>))


--------------------------------------------------------------------------------
rssTemplate :: Template
rssTemplate :: Template
rssTemplate =
    $(makeRelativeToProject ("data" </> "templates" </> "rss.xml")
        >>= embedTemplate)

rssItemTemplate :: Template
rssItemTemplate :: Template
rssItemTemplate =
    $(makeRelativeToProject ("data" </> "templates" </> "rss-item.xml")
        >>= embedTemplate)

atomTemplate :: Template
atomTemplate :: Template
atomTemplate =
    $(makeRelativeToProject ("data" </> "templates" </> "atom.xml")
        >>= embedTemplate)

atomItemTemplate :: Template
atomItemTemplate :: Template
atomItemTemplate =
    $(makeRelativeToProject ("data" </> "templates" </> "atom-item.xml")
        >>= embedTemplate)


--------------------------------------------------------------------------------
-- | This is a data structure to keep the configuration of a feed.
data FeedConfiguration = FeedConfiguration
    { -- | Title of the feed.
      FeedConfiguration -> String
feedTitle       :: String
    , -- | Description of the feed.
      FeedConfiguration -> String
feedDescription :: String
    , -- | Name of the feed author.
      FeedConfiguration -> String
feedAuthorName  :: String
    , -- | Email of the feed author.
      FeedConfiguration -> String
feedAuthorEmail :: String
    , -- | Absolute root URL of the feed site (e.g. @http://jaspervdj.be@)
      FeedConfiguration -> String
feedRoot        :: String
    } deriving (Int -> FeedConfiguration -> ShowS
[FeedConfiguration] -> ShowS
FeedConfiguration -> String
(Int -> FeedConfiguration -> ShowS)
-> (FeedConfiguration -> String)
-> ([FeedConfiguration] -> ShowS)
-> Show FeedConfiguration
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [FeedConfiguration] -> ShowS
$cshowList :: [FeedConfiguration] -> ShowS
show :: FeedConfiguration -> String
$cshow :: FeedConfiguration -> String
showsPrec :: Int -> FeedConfiguration -> ShowS
$cshowsPrec :: Int -> FeedConfiguration -> ShowS
Show, FeedConfiguration -> FeedConfiguration -> Bool
(FeedConfiguration -> FeedConfiguration -> Bool)
-> (FeedConfiguration -> FeedConfiguration -> Bool)
-> Eq FeedConfiguration
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: FeedConfiguration -> FeedConfiguration -> Bool
$c/= :: FeedConfiguration -> FeedConfiguration -> Bool
== :: FeedConfiguration -> FeedConfiguration -> Bool
$c== :: FeedConfiguration -> FeedConfiguration -> Bool
Eq)


--------------------------------------------------------------------------------
-- | Abstract function to render any feed.
renderFeed :: Template                -- ^ Default feed template
           -> Template                -- ^ Default item template
           -> FeedConfiguration       -- ^ Feed configuration
           -> Context String          -- ^ Context for the items
           -> [Item String]           -- ^ Input items
           -> Compiler (Item String)  -- ^ Resulting item
renderFeed :: Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderFeed Template
feedTpl Template
itemTpl FeedConfiguration
config Context String
itemContext [Item String]
items = do
    [Item String]
protectedItems <- (Item String -> Compiler (Item String))
-> [Item String] -> Compiler [Item String]
forall (t :: * -> *) (m :: * -> *) a b.
(Traversable t, Monad m) =>
(a -> m b) -> t a -> m (t b)
mapM (ShowS -> Item String -> Compiler (Item String)
forall (m :: * -> *) (f :: * -> *).
(Monad m, Functor f) =>
ShowS -> f String -> m (f String)
applyFilter ShowS
protectCDATA) [Item String]
items
    Item String
body <- String -> Compiler (Item String)
forall a. a -> Compiler (Item a)
makeItem (String -> Compiler (Item String))
-> Compiler String -> Compiler (Item String)
forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< Template -> Context String -> [Item String] -> Compiler String
forall a. Template -> Context a -> [Item a] -> Compiler String
applyTemplateList Template
itemTpl Context String
itemContext' [Item String]
protectedItems
    Template -> Context String -> Item String -> Compiler (Item String)
forall a. Template -> Context a -> Item a -> Compiler (Item String)
applyTemplate Template
feedTpl Context String
feedContext Item String
body
  where
    applyFilter :: (Monad m,Functor f) => (String -> String) -> f String -> m (f String)
    applyFilter :: ShowS -> f String -> m (f String)
applyFilter ShowS
tr f String
str = f String -> m (f String)
forall (m :: * -> *) a. Monad m => a -> m a
return (f String -> m (f String)) -> f String -> m (f String)
forall a b. (a -> b) -> a -> b
$ ShowS -> f String -> f String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ShowS
tr f String
str
    protectCDATA :: String -> String
    protectCDATA :: ShowS
protectCDATA = String -> ShowS -> ShowS
replaceAll String
"]]>" (String -> ShowS
forall a b. a -> b -> a
const String
"]]&gt;")

    itemContext' :: Context String
itemContext' = [Context String] -> Context String
forall a. Monoid a => [a] -> a
mconcat
        [ Context String
itemContext
        , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"root" (FeedConfiguration -> String
feedRoot FeedConfiguration
config)
        , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"authorName"  (FeedConfiguration -> String
feedAuthorName FeedConfiguration
config)
        , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"authorEmail" (FeedConfiguration -> String
feedAuthorEmail FeedConfiguration
config)
        ]

    feedContext :: Context String
feedContext = [Context String] -> Context String
forall a. Monoid a => [a] -> a
mconcat
         [ String -> Context String
bodyField  String
"body"
         , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"title"       (FeedConfiguration -> String
feedTitle FeedConfiguration
config)
         , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"description" (FeedConfiguration -> String
feedDescription FeedConfiguration
config)
         , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"authorName"  (FeedConfiguration -> String
feedAuthorName FeedConfiguration
config)
         , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"authorEmail" (FeedConfiguration -> String
feedAuthorEmail FeedConfiguration
config)
         , String -> String -> Context String
forall a. String -> String -> Context a
constField String
"root"        (FeedConfiguration -> String
feedRoot FeedConfiguration
config)
         , String -> Context String
forall a. String -> Context a
urlField   String
"url"
         , Context String
forall a. Context a
updatedField
         , Context String
forall a. Context a
missingField
         ]

    -- Take the first "updated" field from all items -- this should be the most
    -- recent.
    updatedField :: Context a
updatedField = String -> (Item a -> Compiler String) -> Context a
forall a. String -> (Item a -> Compiler String) -> Context a
field String
"updated" ((Item a -> Compiler String) -> Context a)
-> (Item a -> Compiler String) -> Context a
forall a b. (a -> b) -> a -> b
$ \Item a
_ -> case [Item String]
items of
        []      -> String -> Compiler String
forall (m :: * -> *) a. Monad m => a -> m a
return String
"Unknown"
        (Item String
x : [Item String]
_) -> Context String
-> String -> [String] -> Item String -> Compiler ContextField
forall a.
Context a -> String -> [String] -> Item a -> Compiler ContextField
unContext Context String
itemContext' String
"updated" [] Item String
x Compiler ContextField
-> (ContextField -> Compiler String) -> Compiler String
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \ContextField
cf -> case ContextField
cf of
            StringField String
s -> String -> Compiler String
forall (m :: * -> *) a. Monad m => a -> m a
return String
s
            ContextField
_             -> String -> Compiler String
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"Hakyll.Web.Feed.renderFeed: Internal error"

--------------------------------------------------------------------------------
-- | Render an RSS feed using given templates with a number of items.
renderRssWithTemplates ::
       Template                -- ^ Feed template
    -> Template                -- ^ Item template
    -> FeedConfiguration       -- ^ Feed configuration
    -> Context String          -- ^ Item context
    -> [Item String]           -- ^ Feed items
    -> Compiler (Item String)  -- ^ Resulting feed
renderRssWithTemplates :: Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderRssWithTemplates Template
feedTemplate Template
itemTemplate FeedConfiguration
config Context String
context = Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderFeed
    Template
feedTemplate Template
itemTemplate FeedConfiguration
config
    (String -> Context String -> Context String
forall a. String -> Context a -> Context a
makeItemContext String
"%a, %d %b %Y %H:%M:%S UT" Context String
context)


--------------------------------------------------------------------------------
-- | Render an Atom feed using given templates with a number of items.
renderAtomWithTemplates ::
       Template                -- ^ Feed template
    -> Template                -- ^ Item template
    -> FeedConfiguration       -- ^ Feed configuration
    -> Context String          -- ^ Item context
    -> [Item String]           -- ^ Feed items
    -> Compiler (Item String)  -- ^ Resulting feed
renderAtomWithTemplates :: Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderAtomWithTemplates Template
feedTemplate Template
itemTemplate FeedConfiguration
config Context String
context = Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderFeed
    Template
feedTemplate Template
itemTemplate FeedConfiguration
config
    (String -> Context String -> Context String
forall a. String -> Context a -> Context a
makeItemContext String
"%Y-%m-%dT%H:%M:%SZ" Context String
context)


--------------------------------------------------------------------------------
-- | Render an RSS feed with a number of items.
renderRss :: FeedConfiguration       -- ^ Feed configuration
          -> Context String          -- ^ Item context
          -> [Item String]           -- ^ Feed items
          -> Compiler (Item String)  -- ^ Resulting feed
renderRss :: FeedConfiguration
-> Context String -> [Item String] -> Compiler (Item String)
renderRss = Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderRssWithTemplates Template
rssTemplate Template
rssItemTemplate


--------------------------------------------------------------------------------
-- | Render an Atom feed with a number of items.
renderAtom :: FeedConfiguration       -- ^ Feed configuration
           -> Context String          -- ^ Item context
           -> [Item String]           -- ^ Feed items
           -> Compiler (Item String)  -- ^ Resulting feed
renderAtom :: FeedConfiguration
-> Context String -> [Item String] -> Compiler (Item String)
renderAtom = Template
-> Template
-> FeedConfiguration
-> Context String
-> [Item String]
-> Compiler (Item String)
renderAtomWithTemplates Template
atomTemplate Template
atomItemTemplate


--------------------------------------------------------------------------------
-- | Copies @$updated$@ from @$published$@ if it is not already set.
makeItemContext :: String -> Context a -> Context a
makeItemContext :: String -> Context a -> Context a
makeItemContext String
fmt Context a
context = [Context a] -> Context a
forall a. Monoid a => [a] -> a
mconcat
    [Context a
context, String -> String -> Context a
forall a. String -> String -> Context a
dateField String
"published" String
fmt, String -> String -> Context a
forall a. String -> String -> Context a
dateField String
"updated" String
fmt]