Add support for top-level finalizers
authorGeoffrey Mainland <mainland@apeiron.net>
Tue, 4 Jun 2013 12:58:52 +0000 (13:58 +0100)
committerGeoffrey Mainland <mainland@apeiron.net>
Fri, 4 Oct 2013 18:58:26 +0000 (14:58 -0400)
Language/Haskell/TH/Syntax.hs

index 79e2944..b62f15a 100644 (file)
@@ -64,6 +64,8 @@ class (Monad m, Applicative m) => Quasi m where
 
   qAddTopDecls :: [Dec] -> m ()
 
+  qAddModFinalizer :: Q () -> m ()
+
 -----------------------------------------------------
 --     The IO instance of Quasi
 --
@@ -91,6 +93,7 @@ instance Quasi IO where
   qRecover _ _               = badIO "recover" -- Maybe we could fix this?
   qAddDependentFile _ = badIO "addDependentFile"
   qAddTopDecls _      = badIO "addTopDecls"
+  qAddModFinalizer _  = badIO "addModFinalizer"
 
   qRunIO m = m
 
@@ -346,6 +349,11 @@ addDependentFile fp = Q (qAddDependentFile fp)
 addTopDecls :: [Dec] -> Q ()
 addTopDecls ds = Q (qAddTopDecls ds)
 
+-- | Add a finalizer that will run in the Q monad after the current module has
+-- been type checked. This only makes sense when run within a top-level splice.
+addModFinalizer :: Q () -> Q ()
+addModFinalizer act = Q (qAddModFinalizer (unQ act))
+
 instance Quasi Q where
   qNewName         = newName
   qReport          = report
@@ -358,6 +366,7 @@ instance Quasi Q where
   qRunIO           = runIO
   qAddDependentFile = addDependentFile
   qAddTopDecls      = addTopDecls
+  qAddModFinalizer  = addModFinalizer
 
 
 ----------------------------------------------------