module Hakyll.Web.RelativizeUrls
( relativizeUrlsCompiler
, relativizeUrls
) where
import Prelude hiding (id)
import Control.Category (id)
import Control.Arrow ((&&&), (>>^))
import Data.List (isPrefixOf)
import qualified Data.Set as S
import Text.HTML.TagSoup
import Hakyll.Core.Compiler
import Hakyll.Web.Page
import Hakyll.Web.Util.Url
relativizeUrlsCompiler :: Compiler (Page String) (Page String)
relativizeUrlsCompiler = getRoute &&& id >>^ uncurry relativize
where
relativize Nothing = id
relativize (Just r) = fmap (relativizeUrls $ toSiteRoot r)
relativizeUrls :: String
-> String
-> String
relativizeUrls root = renderTags . map relativizeUrls' . parseTags
where
relativizeUrls' (TagOpen s a) = TagOpen s $ map (relativizeUrlsAttrs root) a
relativizeUrls' x = x
relativizeUrlsAttrs :: String
-> Attribute String
-> Attribute String
relativizeUrlsAttrs root (key, value)
| key `S.member` urls && "/" `isPrefixOf` value = (key, root ++ value)
| otherwise = (key, value)
where
urls = S.fromList ["src", "href"]