module Blaze.ByteString.Builder.Html.Word
(
fromHtmlEscapedByteString
, fromHtmlEscapedLazyByteString
) where
import qualified Data.ByteString as BS
import qualified Data.ByteString.Builder as B
import qualified Data.ByteString.Builder.Prim as P
import Data.ByteString.Internal (c2w)
import qualified Data.ByteString.Lazy as BSL
import Data.Word (Word8)
wordHtmlEscaped :: P.BoundedPrim Word8
wordHtmlEscaped =
P.condB (> c2w '>' ) (P.condB (== c2w '\DEL') P.emptyB $ P.liftFixedToBounded P.word8) $
P.condB (== c2w '<' ) (fixed4 ('&',('l',('t',';')))) $
P.condB (== c2w '>' ) (fixed4 ('&',('g',('t',';')))) $
P.condB (== c2w '&' ) (fixed5 ('&',('a',('m',('p',';'))))) $
P.condB (== c2w '"' ) (fixed6 ('&',('q',('u',('o',('t',';')))))) $
P.condB (== c2w '\'') (fixed5 ('&',('#',('3',('9',';'))))) $
P.condB (\c -> c >= c2w ' ' || c == c2w '\t' || c == c2w '\n' || c == c2w '\r')
(P.liftFixedToBounded P.word8) P.emptyB
where
fixed4 x = P.liftFixedToBounded $ const x P.>$<
P.char8 P.>*< P.char8 P.>*< P.char8 P.>*< P.char8
fixed5 x = P.liftFixedToBounded $ const x P.>$<
P.char8 P.>*< P.char8 P.>*< P.char8 P.>*< P.char8 P.>*< P.char8
fixed6 x = P.liftFixedToBounded $ const x P.>$<
P.char8 P.>*< P.char8 P.>*< P.char8 P.>*< P.char8 P.>*< P.char8 P.>*< P.char8
fromHtmlEscapedByteString :: BS.ByteString -> B.Builder
fromHtmlEscapedByteString = P.primMapByteStringBounded wordHtmlEscaped
fromHtmlEscapedLazyByteString :: BSL.ByteString -> B.Builder
fromHtmlEscapedLazyByteString = P.primMapLazyByteStringBounded wordHtmlEscaped