module WWW.HW
    ( removeEnv
    , removeSolutions
    ) where

import Prolog

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")