summaryrefslogtreecommitdiff
path: root/testsuite/tests/plugins/hooks-plugin/Hooks/PhasePlugin.hs
blob: 51cff77404ea710b17ff8c92a71ebc72e1df9891 (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
25
26
27
28
29
30
{-# LANGUAGE GADTs #-}
{-# OPTIONS_GHC -Wall #-}
module Hooks.PhasePlugin (plugin) where

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

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

hooksP :: [CommandLineOption] -> HscEnv -> IO HscEnv
hooksP opts hsc_env = do
  hSetBuffering stdout NoBuffering
  let hooks  = hsc_hooks hsc_env
      hooks' = hooks { runPhaseHook = Just fakeRunPhaseHook }
      hsc_env' = hsc_env { hsc_hooks = hooks' }
  return hsc_env'

fakeRunPhaseHook :: PhaseHook
fakeRunPhaseHook = PhaseHook $ \tPhase -> do
  liftIO $ case tPhase of
    T_Cpp{} -> putStrLn "Cpp hook fired"
    T_Hsc{} -> putStrLn "Hsc hook fired"
    T_FileArgs{} -> putStrLn "FileArgs hook fired"
    _ -> pure ()
  runPhase tPhase