blob: 3752b8bed744730ad81b46a8210e1a758bb4eeac [file] [log] [blame]
{-# LANGUAGE OverloadedStrings #-}
-- Copyright (C) 2010-2011 John Millikin <jmillikin@gmail.com>
-- 2011 Dirk Laurie <dirk.laurie@gmail.com>
--
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- any later version.
--
-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
module Anansi.Loom.Markdown (loomMarkdown) where
import Control.Monad (forM_)
import Control.Monad.Reader (asks)
import Control.Monad.Writer (tell)
import Data.ByteString (ByteString)
import Data.Monoid (mconcat)
import qualified Data.Text
import Data.Text (Text)
import Data.Text.Encoding (encodeUtf8)
import Anansi.Types
-- | Generate Markdown. Modified from LaTeX.hs.
loomMarkdown :: Loom
loomMarkdown = mapM_ putBlock . documentBlocks where
putBlock b = case b of
BlockText text -> tell (encodeUtf8 text)
BlockFile path content -> do
tell "\n> **\xC2\xBB "
tell =<< escapeText path
tell "**\n\n"
putContent content
tell "\n"
BlockDefine name content -> do
tell "\n> **\xC2\xAB"
tell =<< escapeText name
tell "\xC2\xBB**\n\n"
putContent content
tell "\n"
putContent cs = forM_ cs $ \c -> case c of
ContentText _ text -> do
tell "> "
escapeCode text >>= tell
tell "\n"
ContentMacro _ indent name -> formatMacro indent name >>= tell
formatMacro indent name = do
escIndent <- escapeCode indent
escName <- escapeCode name
return (mconcat ["> ", escIndent, "\xC2\xAB", escName, "\xC2\xBB\n"])
escapeCode :: Text -> LoomM ByteString
escapeCode text = do
tabSize <- asks loomOptionTabSize
return $ encodeUtf8 $ Data.Text.concatMap (\c -> case c of
'\t' -> Data.Text.replicate (fromInteger tabSize) " "
_ -> Data.Text.singleton c) text
escapeText :: Text -> LoomM ByteString
escapeText text = do
tabSize <- asks loomOptionTabSize
return $ encodeUtf8 $ Data.Text.concatMap (\c -> case c of
'\t' -> Data.Text.replicate (fromInteger tabSize) " "
'\\' -> "\\\\"
'_' -> "\\_"
'*' -> "\\*"
'[' -> "\\["
']' -> "\\]"
'`' -> "\\`"
'&' -> "&amp;"
_ -> Data.Text.singleton c) text