module Text.XHtmlCombinators.Render
( render, renderPretty
) where
import Control.Applicative hiding (empty)
import Data.Foldable
import qualified Data.Sequence as Seq
import Data.Text.Lazy (Text)
import qualified Data.Text.Lazy as T
import qualified Data.Text.Lazy.IO as T
import qualified Text.XML.Light as XML
import Text.XHtmlCombinators.Internal
lt = T.singleton '<'
gt = T.singleton '>'
space = T.singleton ' '
lts = T.pack ""
nl = T.singleton '\n'
renderAttrs :: Attrs -> Text
renderAttrs [] = T.empty
renderAttrs attrs = T.concat (space : fmap renderAttr attrs)
where renderAttr (Attr name val) =
T.concat [name, T.pack "=\"", val, T.pack "\""]
renderNode (TextNode t) = t
renderNode (Node name rattrs attrs c)
| Seq.null c = T.concat [lt, name, a, b, gt, lts, name, gt]
| otherwise = T.concat
[lt, name, a, b, gt, fold (fmap renderNode c), lts, name, gt]
where a = renderAttrs rattrs
b = renderAttrs attrs
-- | Quickly render a xhtml page to text.
--
-- This function will render the entire page on a single line, which
-- is somewhat unreadable. On the plus side, it's relatively fast.
render :: Content c => XHtml c -> Text
render page = fold $ renderNode . toContent <$> execXHtml page
-- | Renders a pretty xhtml page with readable indentation.
--
-- What we do is turn the document from 'Text' into a 'String',
-- parse that string with 'Text.XML.Light', then use 'Text.XML.Light''s
-- pretty printing function to render it, before finally packing it again.
-- It probably goes without saying, but this function is /incredibly/ inefficient!
--
-- Also, 'Text.XML.Light' will render the document as proper XML, which is
-- fine only if you're not trying to pass of your page as text/html.
renderPretty :: Content c => XHtml c -> Text
renderPretty = T.pack . unlines . fmap XML.ppContent . XML.parseXML . T.unpack . render