summaryrefslogtreecommitdiff
path: root/testsuite/tests/plugins/hooks-plugin/Hooks/LogPlugin.hs
blob: 10591240f94cc5f181d24b41b284bd41073c9f14 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
module Hooks.LogPlugin (plugin) where

import GHC.Plugins
import GHC.Driver.Hooks
import GHC.Tc.Utils.Monad
import GHC.Utils.Logger
import GHC.Driver.Pipeline.Execute
import System.IO

plugin :: Plugin
plugin = defaultPlugin { driverPlugin = hooksP }

hooksP :: [CommandLineOption] -> HscEnv -> IO HscEnv
hooksP opts hsc_env = do
  hSetBuffering stdout NoBuffering
  let logger  = hsc_logger hsc_env
      logger' = pushLogHook logHook logger
      hsc_env' = hsc_env { hsc_logger = logger' }
  return hsc_env'

logHook :: LogAction -> LogAction
logHook action logFlags messageClass srcSpan msgDoc = do
  putStrLn "Log hook called"
  action logFlags messageClass srcSpan msgDoc