{-# LANGUAGE OverloadedStrings #-}
module Text.Pandoc.CrossRef.References.List (listOf) where
import Text.Pandoc.Definition
import Data.Accessor.Monad.Trans.State
import Control.Arrow
import Data.List
import qualified Data.Map as M
import qualified Data.Text as T
import Text.Pandoc.CrossRef.References.Types
import Text.Pandoc.CrossRef.Util.Util
import Text.Pandoc.CrossRef.Util.Options
listOf :: Options -> [Block] -> WS [Block]
listOf Options{outFormat=f} x | isLatexFormat f = return x
listOf opts (RawBlock fmt "\\listoffigures":xs)
| isLaTeXRawBlockFmt fmt
= get imgRefs >>= makeList opts lofTitle xs
listOf opts (RawBlock fmt "\\listoftables":xs)
| isLaTeXRawBlockFmt fmt
= get tblRefs >>= makeList opts lotTitle xs
listOf opts (RawBlock fmt "\\listoflistings":xs)
| isLaTeXRawBlockFmt fmt
= get lstRefs >>= makeList opts lolTitle xs
listOf _ x = return x
makeList :: Options -> (Options -> [Block]) -> [Block] -> M.Map T.Text RefRec -> WS [Block]
makeList opts titlef xs refs
= return $
titlef opts ++
(if chaptersDepth opts > 0
then Div ("", ["list"], []) (itemChap `map` refsSorted)
else OrderedList style (item `map` refsSorted))
: xs
where
refsSorted = sortBy compare' $ M.toList refs
compare' (_,RefRec{refIndex=i}) (_,RefRec{refIndex=j}) = compare i j
item = (:[]) . Plain . refTitle . snd
itemChap = Para . uncurry ((. (Space :)) . (++)) . (numWithChap . refIndex &&& refTitle) . snd
numWithChap = chapPrefix (chapDelim opts)
style = (1,DefaultStyle,DefaultDelim)