{-# LANGUAGE NoMonomorphismRestriction #-}

module Text.HTML.Moe
(
    module Text.HTML.Moe.Element
  , module Text.HTML.Moe.Attribute
  , (/)
  , render
  , render'
)
where

import Control.Monad.Writer (execWriter)
import Prelude hiding ((/), (-), head, (>), (.), concat, concatMap, (+))
import qualified Prelude as P
import MPS.Light ((>), (.), times)
import Text.HTML.Moe.Type hiding (name, value)
import qualified Text.HTML.Moe.Type as T
import Text.HTML.Moe.Element
import Text.HTML.Moe.Attribute
import Text.HTML.Moe.Utils
import qualified Data.ByteString.Char8 as B
import Data.DList (toList)

(/) :: MoeUnit
(/) = return ()

render :: MoeUnit -> String
render = render' > B.unpack

render' :: MoeUnit -> B.ByteString
render' = execWriter > toList > map render_element > intercalate new_line > to_bs

_indent_space :: Int
_indent_space = 2

new_line :: Internal
new_line = pack "\n"

space :: Internal
space = pack " "

_indent :: Int -> Internal
_indent n = (n * _indent_space).times space.concat

render_element' :: Int -> Element -> Internal
render_element' _ (Raw x)   = x
render_element' _ (Pre x)   = x
render_element' n (Prim x)  = _indent n + x + new_line
render_element' n (Data x)  = _indent n + x + new_line
  
render_element' n x = 
  [ _indent n
  , "<".pack
  , x.T.name
  , x.attributes.map render_attribute. join_attribute
  , ">".pack
  , new_line
  , x.elements.map (render_element' (n P.+ 1)) .concat
  , _indent n
  , "</".pack
  , x.T.name
  , ">".pack
  , new_line
  ]
  .concat
  where
    join_attribute xs = xs.map (pack " " `append`) .concat


render_element :: Element -> Internal
render_element = render_element' 0

render_attribute :: Attribute -> Internal
render_attribute x = 
  [x.key, pack "=", pack "\"", x.T.value, pack "\""].concat