lierdakil · July 20, 2017 13:39 · puterleat · Jul 20, 2017
diff --git a/index-filter.hs b/index-filter.hs
 import Text.Pandoc
 import Text.Pandoc.JSON
 import Text.Pandoc.Builder
 import Control.Monad.State
 import qualified Data.Map as M

 main :: IO ()
 main = toJSONFilter go
 where
  go (Just (Format "latex")) p                  = bottomUp runLatex p
  go _                       (Pandoc meta body) = Pandoc meta $ runOther body
  runLatex s@(Span (_id', cls, _props) val:xs)
    | "index" `elem` cls = concat
      [ [RawInline (Format "latex") "\\index{"]
      , val
      , [RawInline (Format "latex") "}"]
      , xs
      ]
    | otherwise = s
  runLatex x = x
  runOther :: [Block] -> [Block]
  runOther body = addIndex $ evalState (bottomUpM makeIds body) (1 :: Integer)
  addIndex :: [Block] -> [Block]
  addIndex bs = bs ++ toList idx
   where
    idx = header 1 (str "Index") <> definitionList (cat $ queryWith getIdx bs)
  makeIds s@(Span (_id', cls, props) val)
    | "index" `elem` cls = do
      curIdx <- get
      put $ curIdx + 1
      return $ Span ("index-id-no-" ++ show curIdx, cls, props) val
    | otherwise = return s
  makeIds x = return x
  getIdx :: Inline -> [(Inlines, Inlines)]
  getIdx (Span (id', cls, _props) val)
    | "index" `elem` cls
    = [(strong (fromList val), link ('#' : id') [] (str "idx"))]
    | otherwise
    = []
  getIdx _ = []
  cat = M.toList . M.map (return . plain) . M.fromListWith
    (\x acc -> acc <> str "," <> space <> x)
diff --git a/input.md b/input.md
diff --git a/output.md b/output.md
diff --git a/output.tex b/output.tex
 \index{Lorem} ipsum dolor sit amet, consectetur \index{adipiscing} elit.
 Donec in porttitor massa, \index{vitae} semper justo. \index{Praesent}
 aliquet \index{vitae} justo in malesuada. Praesent et sem at orci
 posuere fringilla. Pellentesque in lorem tincidunt, semper velit sed,
 luctus nulla. Cras lorem justo, bibendum in fringilla in, cursus vel
 odio. Phasellus vestibulum aliquam congue. Ut aliquet elit id nisl
 porta, suscipit euismod augue gravida. Duis maximus velit ut convallis
 mattis. Interdum et malesuada fames ac ante ipsum primis in faucibus.
 Duis porta, eros nec sodales malesuada, tellus eros porttitor justo,
 blandit viverra ante nibh vel ligula. Proin egestas turpis sit amet leo
 molestie, et venenatis purus maximus. Nunc ex enim, faucibus id metus a,
 venenatis venenatis neque. Phasellus in ultrices felis. Pellentesque
 iaculis aliquam odio eu finibus. Ut laoreet arcu id erat facilisis
 rhoncus. Donec eros mauris, porta id tortor ac, lobortis dictum risus.

 Aenean elementum \index{velit} \index{vitae} tristique varius. Praesent
 et lectus commodo, efficitur massa non, egestas nisl. Suspendisse
 bibendum, diam et euismod maximus, neque nulla dictum sem, sed
 vestibulum lectus enim quis augue. Nulla dignissim tristique posuere.
 Class aptent taciti sociosqu ad litora torquent per conubia nostra, per
 inceptos himenaeos. Cras varius nibh mattis tortor pharetra tincidunt.
 Sed id lacus varius, pellentesque risus a, tempus orci. Curabitur eget
 urna sapien.

 Aliquam congue imperdiet felis. Nullam gravida turpis a justo
 ullamcorper malesuada. Duis consectetur \index{cursus} enim. Maecenas
 enim lorem, elementum ac porttitor eu, cursus nec \index{orci}. Morbi
 sed rutrum ante. Mauris commodo, turpis a fringilla dignissim, libero
 nisl dictum ex, \index{vulputate} tristique augue lorem id enim. Aliquam
 ac tristique nulla. Nam nec posuere ex. Integer hendrerit cursus dictum.
 Class aptent taciti \index{sociosqu} ad litora torquent per conubia
 nostra, per inceptos himenaeos. Mauris maximus dapibus leo in consequat.
	import Text.Pandoc
	import Text.Pandoc.JSON
	import Text.Pandoc.Builder
	import Control.Monad.State
	import qualified Data.Map as M

	main :: IO ()
	main = toJSONFilter go
	where
	go (Just (Format "latex")) p = bottomUp runLatex p
	go _ (Pandoc meta body) = Pandoc meta $ runOther body
	runLatex s@(Span (_id', cls, _props) val:xs)
	\| "index" `elem` cls = concat
	[ [RawInline (Format "latex") "\\index{"]
	, val
	, [RawInline (Format "latex") "}"]
	, xs
	]
	\| otherwise = s
	runLatex x = x
	runOther :: [Block] -> [Block]
	runOther body = addIndex $ evalState (bottomUpM makeIds body) (1 :: Integer)
	addIndex :: [Block] -> [Block]
	addIndex bs = bs ++ toList idx
	where
	idx = header 1 (str "Index") <> definitionList (cat $ queryWith getIdx bs)
	makeIds s@(Span (_id', cls, props) val)
	\| "index" `elem` cls = do
	curIdx <- get
	put $ curIdx + 1
	return $ Span ("index-id-no-" ++ show curIdx, cls, props) val
	\| otherwise = return s
	makeIds x = return x
	getIdx :: Inline -> [(Inlines, Inlines)]
	getIdx (Span (id', cls, _props) val)
	\| "index" `elem` cls
	= [(strong (fromList val), link ('#' : id') [] (str "idx"))]
	\| otherwise
	= []
	getIdx _ = []
	cat = M.toList . M.map (return . plain) . M.fromListWith
	(\x acc -> acc <> str "," <> space <> x)
	\index{Lorem} ipsum dolor sit amet, consectetur \index{adipiscing} elit.
	Donec in porttitor massa, \index{vitae} semper justo. \index{Praesent}
	aliquet \index{vitae} justo in malesuada. Praesent et sem at orci
	posuere fringilla. Pellentesque in lorem tincidunt, semper velit sed,
	luctus nulla. Cras lorem justo, bibendum in fringilla in, cursus vel
	odio. Phasellus vestibulum aliquam congue. Ut aliquet elit id nisl
	porta, suscipit euismod augue gravida. Duis maximus velit ut convallis
	mattis. Interdum et malesuada fames ac ante ipsum primis in faucibus.
	Duis porta, eros nec sodales malesuada, tellus eros porttitor justo,
	blandit viverra ante nibh vel ligula. Proin egestas turpis sit amet leo
	molestie, et venenatis purus maximus. Nunc ex enim, faucibus id metus a,
	venenatis venenatis neque. Phasellus in ultrices felis. Pellentesque
	iaculis aliquam odio eu finibus. Ut laoreet arcu id erat facilisis
	rhoncus. Donec eros mauris, porta id tortor ac, lobortis dictum risus.

	Aenean elementum \index{velit} \index{vitae} tristique varius. Praesent
	et lectus commodo, efficitur massa non, egestas nisl. Suspendisse
	bibendum, diam et euismod maximus, neque nulla dictum sem, sed
	vestibulum lectus enim quis augue. Nulla dignissim tristique posuere.
	Class aptent taciti sociosqu ad litora torquent per conubia nostra, per
	inceptos himenaeos. Cras varius nibh mattis tortor pharetra tincidunt.
	Sed id lacus varius, pellentesque risus a, tempus orci. Curabitur eget
	urna sapien.

	Aliquam congue imperdiet felis. Nullam gravida turpis a justo
	ullamcorper malesuada. Duis consectetur \index{cursus} enim. Maecenas
	enim lorem, elementum ac porttitor eu, cursus nec \index{orci}. Morbi
	sed rutrum ante. Mauris commodo, turpis a fringilla dignissim, libero
	nisl dictum ex, \index{vulputate} tristique augue lorem id enim. Aliquam
	ac tristique nulla. Nam nec posuere ex. Integer hendrerit cursus dictum.
	Class aptent taciti \index{sociosqu} ad litora torquent per conubia
	nostra, per inceptos himenaeos. Mauris maximus dapibus leo in consequat.