module Main ( main ) where

import Distribution.Simple
import Distribution.Simple.LocalBuildInfo ( withPrograms )
import Distribution.Simple.Program ( userSpecifyArgs )

main :: IO ()
main = defaultMainWithHooks $
         simpleUserHooks `modify_haddockHook` \oldHH pkg lbi hooks flags ->
           (\lbi' -> oldHH pkg lbi' hooks flags) $
             lbi `modify_withPrograms` \oldWP ->
               userSpecifyArgs "haddock" ["--optghc=-D__HADDOCK__"] oldWP
  where
    modify_haddockHook  hooks f = hooks { haddockHook  = f (haddockHook  hooks) }
    modify_withPrograms lbi   f = lbi   { withPrograms = f (withPrograms lbi)   }