module WWW.HW
    ( removeEnv
    , removeSolutions
    -- * LaTeX utils
    , LaTeX (..)
    , texmap
    -- * Hakyll utils
    , module Hakyll.TeX.HaTeX
    , module Hakyll.TeX.LaTeX
    ) where

-- import Prolog
import Hakyll.TeX.HaTeX
import Hakyll.TeX.LaTeX
import Text.LaTeX.Base.Syntax

removeEnv :: (String -> Bool) -> LaTeX -> LaTeX
removeEnv p = texmap (\case { TeXEnv s _ _ -> p s; _ -> False }) $ pure mempty

removeSolutions :: LaTeX -> LaTeX
removeSolutions = removeEnv (== "solution")