module Text.Blaze.Renderer.Utf8
( renderHtmlBuilder
, renderHtml
, renderHtmlToByteStringIO
) where
import Data.Monoid (mappend, mempty)
import Data.List (isInfixOf)
import qualified Data.ByteString.Lazy as L
import qualified Data.Text as T (isInfixOf)
import qualified Data.ByteString as S (ByteString, isInfixOf)
import Text.Blaze.Internal
import Blaze.ByteString.Builder (Builder)
import qualified Blaze.ByteString.Builder as B
import qualified Blaze.ByteString.Builder.Html.Utf8 as B
fromChoiceString :: ChoiceString
-> Builder
fromChoiceString (Static s) = B.copyByteString $ getUtf8ByteString s
fromChoiceString (String s) = B.fromHtmlEscapedString s
fromChoiceString (Text s) = B.fromHtmlEscapedText s
fromChoiceString (ByteString s) = B.fromByteString s
fromChoiceString (PreEscaped x) = case x of
String s -> B.fromString s
Text s -> B.fromText s
s -> fromChoiceString s
fromChoiceString (External x) = case x of
String s -> if "</" `isInfixOf` s then mempty else B.fromString s
Text s -> if "</" `T.isInfixOf` s then mempty else B.fromText s
ByteString s -> if "</" `S.isInfixOf` s then mempty else B.fromByteString s
s -> fromChoiceString s
fromChoiceString (AppendChoiceString x y) =
fromChoiceString x `mappend` fromChoiceString y
fromChoiceString EmptyChoiceString = mempty
renderHtmlBuilder :: Html
-> Builder
renderHtmlBuilder = go mempty
where
go :: Builder -> HtmlM b -> Builder
go attrs (Parent _ open close content) =
B.copyByteString (getUtf8ByteString open)
`mappend` attrs
`mappend` B.fromChar '>'
`mappend` go mempty content
`mappend` B.copyByteString (getUtf8ByteString close)
go attrs (Leaf _ begin end) =
B.copyByteString (getUtf8ByteString begin)
`mappend` attrs
`mappend` B.copyByteString (getUtf8ByteString end)
go attrs (AddAttribute _ key value h) =
go (B.copyByteString (getUtf8ByteString key)
`mappend` fromChoiceString value
`mappend` B.fromChar '"'
`mappend` attrs) h
go attrs (AddCustomAttribute _ key value h) =
go (fromChoiceString key
`mappend` fromChoiceString value
`mappend` B.fromChar '"'
`mappend` attrs) h
go _ (Content content) = fromChoiceString content
go attrs (Append h1 h2) = go attrs h1 `mappend` go attrs h2
go _ Empty = mempty
renderHtml :: Html
-> L.ByteString
renderHtml = B.toLazyByteString . renderHtmlBuilder
renderHtmlToByteStringIO :: (S.ByteString -> IO ())
-> Html
-> IO ()
renderHtmlToByteStringIO io = B.toByteStringIO io . renderHtmlBuilder