-- Copyright (C) 2010 John Millikin <jmillikin@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/>.
-- 
{-# LANGUAGE OverloadedStrings #-}
module Anansi.Loom.NoWeb (loomNoWeb) where
import qualified Data.Text.Lazy as TL
import Control.Monad (forM_)
import Control.Monad.Writer (tell)
import Anansi.Types
import Anansi.Loom

loomNoWeb :: Loom
loomNoWeb = Loom "latex-noweb" $ mapM_ putBlock where
	putBlock b = case b of
		BlockText text -> tell text
		BlockFile path content -> do
			tell "\\nwbegincode{0}\\moddef{"
			tell $ escapeText path
			tell "}\\endmoddef\\nwstartdeflinemarkup\\nwenddeflinemarkup\n"
			putContent content
			tell "\\nwendcode{}\n"
			
		BlockDefine name content -> do
			tell "\\nwbegincode{0}\\moddef{"
			tell $ escapeText name
			tell "}\\endmoddef\\nwstartdeflinemarkup\\nwenddeflinemarkup\n"
			putContent content
			tell "\\nwendcode{}\n"
	
	putContent cs = forM_ cs $ \c -> case c of
		ContentText _ text -> tell . escapeCode $ TL.append text "\n"
		ContentMacro _ indent name -> tell $ formatMacro indent name

formatMacro :: TL.Text -> TL.Text -> TL.Text
formatMacro indent name = TL.concat [escapeCode indent, "\\LA{}", escapeText name, "\\RA{}\n"]

escapeCode :: TL.Text -> TL.Text
escapeCode = TL.concatMap $ \c -> case c of
	'\t' -> "        "
	'\\' -> "\\\\"
	'{' -> "\\{"
	'}' -> "\\}"
	'_' -> "\\_"
	_ -> TL.singleton c

escapeText :: TL.Text -> TL.Text
escapeText = TL.concatMap $ \c -> case c of
	'\t' -> "        "
	'\\' -> "\\\\"
	'{' -> "\\{"
	'}' -> "\\}"
	'_' -> "\\_"
	'<' -> "{$<$}"
	'>' -> "{$>$}"
	'$' -> "\\$"
	_ -> TL.singleton c