TcSplice.hs 62 KB
Newer Older
Austin Seipp's avatar
Austin Seipp committed
1 2 3
{-
(c) The University of Glasgow 2006
(c) The GRASP/AQUA Project, Glasgow University, 1992-1998
4

5

Austin Seipp's avatar
Austin Seipp committed
6 7
TcSplice: Template Haskell splices
-}
8

9 10 11 12 13
{-# LANGUAGE CPP #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE MagicHash #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE InstanceSigs #-}
14
{-# OPTIONS_GHC -fno-warn-orphans #-}
15

16 17 18
module TcSplice(
     -- These functions are defined in stage1 and stage2
     -- The raise civilised errors in stage1
19
     tcSpliceExpr, tcTypedBracket, tcUntypedBracket,
20 21
--     runQuasiQuoteExpr, runQuasiQuotePat,
--     runQuasiQuoteDecl, runQuasiQuoteType,
22 23 24 25 26 27
     runAnnotation,

#ifdef GHCI
     -- These ones are defined only in stage2, and are
     -- called only in stage2 (ie GHCI is on)
     runMetaE, runMetaP, runMetaT, runMetaD, runQuasi,
28
     tcTopSpliceExpr, lookupThName_maybe,
Luite Stegeman's avatar
Luite Stegeman committed
29
     defaultRunMeta, runMeta'
30 31
#endif
      ) where
32 33 34

#include "HsVersions.h"

35 36 37 38 39 40
import HsSyn
import Annotations
import Name
import TcRnMonad
import TcType

41 42 43 44
import Outputable
import TcExpr
import SrcLoc
import FastString
45
import THNames
46 47 48
import TcUnify
import TcEnv

49
#ifdef GHCI
50
import HscMain
51 52
        -- These imports are the reason that TcSplice
        -- is very high up the module hierarchy
53 54
import RnSplice( traceSplice, SpliceInfo(..) )
import RdrName
55
import HscTypes
56 57 58 59 60 61
import Convert
import RnExpr
import RnEnv
import RnTypes
import TcHsSyn
import TcSimplify
batterseapower's avatar
batterseapower committed
62
import Type
63
import Kind
64
import NameSet
65 66 67 68
import TcMType
import TcHsType
import TcIface
import TypeRep
69 70
import FamInst
import FamInstEnv
71
import InstEnv
72
import NameEnv
73
import PrelNames
74
import OccName
75
import Hooks
76 77
import Var
import Module
78
import LoadIface
79
import Class
80
import Inst
81
import TyCon
82
import CoAxiom
83
import PatSyn ( patSynName )
cactus's avatar
cactus committed
84
import ConLike
85
import DataCon
86
import TcEvidence( TcEvBinds(..) )
87 88 89
import Id
import IdInfo
import DsExpr
90
import DsMonad
91
import Serialized
92
import ErrUtils
93
import Util
94
import Data.List        ( mapAccumL )
95
import Unique
96
import VarSet           ( isEmptyVarSet )
97
import Data.Maybe
98 99
import BasicTypes hiding( SuccessFlag(..) )
import Maybes( MaybeErr(..) )
100
import DynFlags
101
import Panic
102
import Lexeme
103

104 105 106
import qualified Language.Haskell.TH as TH
-- THSyntax gives access to internal functions and data types
import qualified Language.Haskell.TH.Syntax as TH
107

108 109
-- Because GHC.Desugar might not be in the base library of the bootstrapping compiler
import GHC.Desugar      ( AnnotationWrapper(..) )
110 111 112

import qualified Data.Map as Map
import Data.Dynamic  ( fromDynamic, toDyn )
113
import Data.Typeable ( typeOf, Typeable )
114
import Data.Data (Data)
115
import GHC.Exts         ( unsafeCoerce# )
116 117
#endif

Austin Seipp's avatar
Austin Seipp committed
118 119 120
{-
************************************************************************
*                                                                      *
121
\subsection{Main interface + stubs for the non-GHCI case
Austin Seipp's avatar
Austin Seipp committed
122 123 124
*                                                                      *
************************************************************************
-}
125

126 127
tcTypedBracket   :: HsBracket Name -> TcRhoType -> TcM (HsExpr TcId)
tcUntypedBracket :: HsBracket Name -> [PendingRnSplice] -> TcRhoType -> TcM (HsExpr TcId)
128
tcSpliceExpr     :: HsSplice Name  -> TcRhoType -> TcM (HsExpr TcId)
129 130
        -- None of these functions add constraints to the LIE

131 132 133 134
-- runQuasiQuoteExpr :: HsQuasiQuote RdrName -> RnM (LHsExpr RdrName)
-- runQuasiQuotePat  :: HsQuasiQuote RdrName -> RnM (LPat RdrName)
-- runQuasiQuoteType :: HsQuasiQuote RdrName -> RnM (LHsType RdrName)
-- runQuasiQuoteDecl :: HsQuasiQuote RdrName -> RnM [LHsDecl RdrName]
135 136

runAnnotation     :: CoreAnnTarget -> LHsExpr Name -> TcM Annotation
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
{-
************************************************************************
*                                                                      *
\subsection{Quoting an expression}
*                                                                      *
************************************************************************
-}

-- See Note [How brackets and nested splices are handled]
-- tcTypedBracket :: HsBracket Name -> TcRhoType -> TcM (HsExpr TcId)
tcTypedBracket brack@(TExpBr expr) res_ty
  = addErrCtxt (quotationCtxtDoc brack) $
    do { cur_stage <- getStage
       ; ps_ref <- newMutVar []
       ; lie_var <- getConstraintVar   -- Any constraints arising from nested splices
                                       -- should get thrown into the constraint set
                                       -- from outside the bracket

       -- Typecheck expr to make sure it is valid,
       -- Throw away the typechecked expression but return its type.
       -- We'll typecheck it again when we splice it in somewhere
       ; (_tc_expr, expr_ty) <- setStage (Brack cur_stage (TcPending ps_ref lie_var)) $
                                tcInferRhoNC expr
                                -- NC for no context; tcBracket does that

       ; meta_ty <- tcTExpTy expr_ty
       ; co <- unifyType meta_ty res_ty
       ; ps' <- readMutVar ps_ref
       ; texpco <- tcLookupId unsafeTExpCoerceName
       ; return (mkHsWrapCo co (unLoc (mkHsApp (nlHsTyApp texpco [expr_ty])
                                               (noLoc (HsTcBracketOut brack ps'))))) }
tcTypedBracket other_brack _
  = pprPanic "tcTypedBracket" (ppr other_brack)

-- tcUntypedBracket :: HsBracket Name -> [PendingRnSplice] -> TcRhoType -> TcM (HsExpr TcId)
tcUntypedBracket brack ps res_ty
  = do { traceTc "tc_bracket untyped" (ppr brack $$ ppr ps)
       ; ps' <- mapM tcPendingSplice ps
       ; meta_ty <- tcBrackTy brack
       ; co <- unifyType meta_ty res_ty
       ; traceTc "tc_bracket done untyped" (ppr meta_ty)
       ; return (mkHsWrapCo co (HsTcBracketOut brack ps'))  }

---------------
tcBrackTy :: HsBracket Name -> TcM TcType
tcBrackTy (VarBr _ _) = tcMetaTy nameTyConName  -- Result type is Var (not Q-monadic)
tcBrackTy (ExpBr _)   = tcMetaTy expQTyConName  -- Result type is ExpQ (= Q Exp)
tcBrackTy (TypBr _)   = tcMetaTy typeQTyConName -- Result type is Type (= Q Typ)
tcBrackTy (DecBrG _)  = tcMetaTy decsQTyConName -- Result type is Q [Dec]
tcBrackTy (PatBr _)   = tcMetaTy patQTyConName  -- Result type is PatQ (= Q Pat)
tcBrackTy (DecBrL _)  = panic "tcBrackTy: Unexpected DecBrL"
tcBrackTy (TExpBr _)  = panic "tcUntypedBracket: Unexpected TExpBr"

---------------
tcPendingSplice :: PendingRnSplice -> TcM PendingTcSplice
tcPendingSplice (PendingRnSplice flavour splice_name expr)
  = do { res_ty <- tcMetaTy meta_ty_name
       ; expr' <- tcMonoExpr expr res_ty
       ; return (PendingTcSplice splice_name expr') }
  where
     meta_ty_name = case flavour of
                       UntypedExpSplice  -> expQTyConName
                       UntypedPatSplice  -> patQTyConName
                       UntypedTypeSplice -> typeQTyConName
                       UntypedDeclSplice -> decsQTyConName

---------------
-- Takes a type tau and returns the type Q (TExp tau)
tcTExpTy :: TcType -> TcM TcType
tcTExpTy tau
  = do { q    <- tcLookupTyCon qTyConName
       ; texp <- tcLookupTyCon tExpTyConName
       ; return (mkTyConApp q [mkTyConApp texp [tau]]) }

quotationCtxtDoc :: HsBracket Name -> SDoc
quotationCtxtDoc br_body
  = hang (ptext (sLit "In the Template Haskell quotation"))
         2 (ppr br_body)

216 217

#ifndef GHCI
218
tcSpliceExpr  e _      = failTH e "Template Haskell splice"
219

220 221 222 223
-- runQuasiQuoteExpr q = failTH q "quasiquote"
-- runQuasiQuotePat  q = failTH q "pattern quasiquote"
-- runQuasiQuoteType q = failTH q "type quasiquote"
-- runQuasiQuoteDecl q = failTH q "declaration quasiquote"
224 225 226 227 228
runAnnotation   _ q = failTH q "annotation"

#else
  -- The whole of the rest of the file is the else-branch (ie stage2 only)

Austin Seipp's avatar
Austin Seipp committed
229
{-
230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
Note [How top-level splices are handled]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Top-level splices (those not inside a [| .. |] quotation bracket) are handled
very straightforwardly:

  1. tcTopSpliceExpr: typecheck the body e of the splice $(e)

  2. runMetaT: desugar, compile, run it, and convert result back to
     HsSyn RdrName (of the appropriate flavour, eg HsType RdrName,
     HsExpr RdrName etc)

  3. treat the result as if that's what you saw in the first place
     e.g for HsType, rename and kind-check
         for HsExpr, rename and type-check

Gabor Greif's avatar
typos  
Gabor Greif committed
245
     (The last step is different for decls, because they can *only* be
246 247 248 249
      top-level: we return the result of step 2.)

Note [How brackets and nested splices are handled]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
Nested splices (those inside a [| .. |] quotation bracket),
are treated quite differently.

Remember, there are two forms of bracket
         typed   [|| e ||]
   and untyped   [|  e  |]

The life cycle of a typed bracket:
   * Starts as HsBracket

   * When renaming:
        * Set the ThStage to (Brack s RnPendingTyped)
        * Rename the body
        * Result is still a HsBracket

   * When typechecking:
        * Set the ThStage to (Brack s (TcPending ps_var lie_var))
        * Typecheck the body, and throw away the elaborated result
        * Nested splices (which must be typed) are typechecked, and
          the results accumulated in ps_var; their constraints
          accumulate in lie_var
        * Result is a HsTcBracketOut rn_brack pending_splices
          where rn_brack is the incoming renamed bracket

The life cycle of a un-typed bracket:
   * Starts as HsBracket

   * When renaming:
        * Set the ThStage to (Brack s (RnPendingUntyped ps_var))
        * Rename the body
        * Nested splices (which must be untyped) are renamed, and the
          results accumulated in ps_var
        * Result is still (HsRnBracketOut rn_body pending_splices)

   * When typechecking a HsRnBracketOut
        * Typecheck the pending_splices individually
        * Ignore the body of the bracket; just check that the context
          expects a bracket of that type (e.g. a [p| pat |] bracket should
          be in a context needing a (Q Pat)
        * Result is a HsTcBracketOut rn_brack pending_splices
          where rn_brack is the incoming renamed bracket


In both cases, desugaring happens like this:
  * HsTcBracketOut is desugared by DsMeta.dsBracket.  It
295 296 297 298 299 300

      a) Extends the ds_meta environment with the PendingSplices
         attached to the bracket

      b) Converts the quoted (HsExpr Name) to a CoreExpr that, when
         run, will produce a suitable TH expression/type/decl.  This
301
         is why we leave the *renamed* expression attached to the bracket:
302 303 304 305 306 307 308 309 310 311 312 313
         the quoted expression should not be decorated with all the goop
         added by the type checker

  * Each splice carries a unique Name, called a "splice point", thus
    ${n}(e).  The name is initialised to an (Unqual "splice") when the
    splice is created; the renamer gives it a unique.

  * When DsMeta (used to desugar the body of the bracket) comes across
    a splice, it looks up the splice's Name, n, in the ds_meta envt,
    to find an (HsExpr Id) that should be substituted for the splice;
    it just desugars it to get a CoreExpr (DsMeta.repSplice).

314 315
Example:
    Source:       f = [| Just $(g 3) |]
316 317 318
      The [| |] part is a HsBracket

    Typechecked:  f = [| Just ${s7}(g 3) |]{s7 = g Int 3}
319 320 321 322
      The [| |] part is a HsBracketOut, containing *renamed*
        (not typechecked) expression
      The "s7" is the "splice point"; the (g Int 3) part
        is a typechecked expression
323

324 325
    Desugared:    f = do { s7 <- g Int 3
                         ; return (ConE "Data.Maybe.Just" s7) }
326 327 328 329 330


Note [Template Haskell state diagram]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Here are the ThStages, s, their corresponding level numbers
331
(the result of (thLevel s)), and their state transitions.
332
The top level of the program is stage Comp:
333

334 335 336
     Start here
         |
         V
337
      -----------     $      ------------   $
338
      |  Comp   | ---------> |  Splice  | -----|
339 340 341 342 343 344 345 346 347 348 349
      |   1     |            |    0     | <----|
      -----------            ------------
        ^     |                ^      |
      $ |     | [||]         $ |      | [||]
        |     v                |      v
   --------------          ----------------
   | Brack Comp |          | Brack Splice |
   |     2      |          |      1       |
   --------------          ----------------

* Normal top-level declarations start in state Comp
350 351 352 353
       (which has level 1).
  Annotations start in state Splice, since they are
       treated very like a splice (only without a '$')

354
* Code compiled in state Splice (and only such code)
355 356 357 358 359
  will be *run at compile time*, with the result replacing
  the splice

* The original paper used level -1 instead of 0, etc.

360 361
* The original paper did not allow a splice within a
  splice, but there is no reason not to. This is the
362 363
  $ transition in the top right.

364 365 366
Note [Template Haskell levels]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
* Imported things are impLevel (= 0)
367

368 369 370
* However things at level 0 are not *necessarily* imported.
      eg  $( \b -> ... )   here b is bound at level 0

371 372 373
* In GHCi, variables bound by a previous command are treated
  as impLevel, because we have bytecode for them.

374
* Variables are bound at the "current level"
375

376
* The current level starts off at outerLevel (= 1)
377

378
* The level is decremented by splicing $(..)
379 380
               incremented by brackets [| |]
               incremented by name-quoting 'f
381

382 383 384
When a variable is used, we compare
        bind:  binding level, and
        use:   current level at usage site
385 386

  Generally
387 388
        bind > use      Always error (bound later than used)
                        [| \x -> $(f x) |]
389

390 391 392 393 394 395
        bind = use      Always OK (bound same stage as used)
                        [| \x -> $(f [| x |]) |]

        bind < use      Inside brackets, it depends
                        Inside splice, OK
                        Inside neither, OK
396 397

  For (bind < use) inside brackets, there are three cases:
398 399 400 401
    - Imported things   OK      f = [| map |]
    - Top-level things  OK      g = [| f |]
    - Non-top-level     Only if there is a liftable instance
                                h = \(x:Int) -> [| x |]
402

403
  To track top-level-ness we use the ThBindEnv in TcLclEnv
404

405 406 407 408
  For example:
           f = ...
           g1 = $(map ...)         is OK
           g2 = $(f ...)           is not OK; because we havn't compiled f yet
409

Austin Seipp's avatar
Austin Seipp committed
410
-}
simonpj@microsoft.com's avatar
simonpj@microsoft.com committed
411

Austin Seipp's avatar
Austin Seipp committed
412 413 414
{-
************************************************************************
*                                                                      *
415
\subsection{Splicing an expression}
Austin Seipp's avatar
Austin Seipp committed
416 417 418
*                                                                      *
************************************************************************
-}
419

420
tcSpliceExpr splice@(HsTypedSplice name expr) res_ty
421
  = addErrCtxt (spliceCtxtDoc splice) $
422
    setSrcSpan (getLoc expr)    $ do
423
    { stage <- getStage
gmainland's avatar
gmainland committed
424
    ; case stage of
425 426 427
        Splice {}            -> tcTopSplice expr res_ty
        Comp                 -> tcTopSplice expr res_ty
        Brack pop_stage pend -> tcNestedSplice pop_stage pend name expr res_ty }
428 429
tcSpliceExpr splice _
  = pprPanic "tcSpliceExpr" (ppr splice)
430 431 432

tcNestedSplice :: ThStage -> PendingStuff -> Name
                -> LHsExpr Name -> TcRhoType -> TcM (HsExpr Id)
433 434
    -- See Note [How brackets and nested splices are handled]
    -- A splice inside brackets
435 436 437 438 439 440 441 442
tcNestedSplice pop_stage (TcPending ps_var lie_var) splice_name expr res_ty
  = do { meta_exp_ty <- tcTExpTy res_ty
       ; expr' <- setStage pop_stage $
                  setConstraintVar lie_var $
                  tcMonoExpr expr meta_exp_ty
       ; untypeq <- tcLookupId unTypeQName
       ; let expr'' = mkHsApp (nlHsTyApp untypeq [res_ty]) expr'
       ; ps <- readMutVar ps_var
443
       ; writeMutVar ps_var (PendingTcSplice splice_name expr'' : ps)
444 445 446 447 448 449

       -- The returned expression is ignored; it's in the pending splices
       ; return (panic "tcSpliceExpr") }

tcNestedSplice _ _ splice_name _ _
  = pprPanic "tcNestedSplice: rename stage found" (ppr splice_name)
450

gmainland's avatar
gmainland committed
451 452
tcTopSplice :: LHsExpr Name -> TcRhoType -> TcM (HsExpr Id)
tcTopSplice expr res_ty
453 454 455
  = do { -- Typecheck the expression,
         -- making sure it has type Q (T res_ty)
         meta_exp_ty <- tcTExpTy res_ty
gmainland's avatar
gmainland committed
456 457
       ; zonked_q_expr <- tcTopSpliceExpr True $
                          tcMonoExpr expr meta_exp_ty
458

459
         -- Run the expression
460
       ; expr2 <- runMetaE zonked_q_expr
461 462 463 464
       ; traceSplice (SpliceInfo { spliceDescription = "expression"
                                 , spliceIsDecl      = False
                                 , spliceSource      = Just expr
                                 , spliceGenerated   = ppr expr2 })
465

466 467 468
         -- Rename and typecheck the spliced-in expression,
         -- making sure it has type res_ty
         -- These steps should never fail; this is a *typed* splice
469
       ; addErrCtxt (spliceResultDoc expr) $ do
470
       { (exp3, _fvs) <- rnLExpr expr2
471
       ; exp4 <- tcMonoExpr exp3 res_ty
472
       ; return (unLoc exp4) } }
gmainland's avatar
gmainland committed
473

Austin Seipp's avatar
Austin Seipp committed
474 475 476
{-
************************************************************************
*                                                                      *
gmainland's avatar
gmainland committed
477
\subsection{Error messages}
Austin Seipp's avatar
Austin Seipp committed
478 479 480
*                                                                      *
************************************************************************
-}
481

482 483 484
spliceCtxtDoc :: HsSplice Name -> SDoc
spliceCtxtDoc splice
  = hang (ptext (sLit "In the Template Haskell splice"))
485
         2 (pprSplice splice)
486

487 488 489 490 491
spliceResultDoc :: LHsExpr Name -> SDoc
spliceResultDoc expr
  = sep [ ptext (sLit "In the result of the splice:")
        , nest 2 (char '$' <> pprParendExpr expr)
        , ptext (sLit "To see what the splice expanded to, use -ddump-splices")]
492

493
-------------------
494
tcTopSpliceExpr :: Bool -> TcM (LHsExpr Id) -> TcM (LHsExpr Id)
495
-- Note [How top-level splices are handled]
496 497
-- Type check an expression that is the body of a top-level splice
--   (the caller will compile and run it)
498 499
-- Note that set the level to Splice, regardless of the original level,
-- before typechecking the expression.  For example:
500 501
--      f x = $( ...$(g 3) ... )
-- The recursive call to tcMonoExpr will simply expand the
502 503
-- inner escape before dealing with the outer one

504
tcTopSpliceExpr isTypedSplice tc_action
505 506
  = checkNoErrs $  -- checkNoErrs: must not try to run the thing
                   -- if the type checker fails!
ian@well-typed.com's avatar
ian@well-typed.com committed
507
    unsetGOptM Opt_DeferTypeErrors $
508 509 510 511 512
                   -- Don't defer type errors.  Not only are we
                   -- going to run this code, but we do an unsafe
                   -- coerce, so we get a seg-fault if, say we
                   -- splice a type into a place where an expression
                   -- is expected (Trac #7276)
513
    setStage (Splice isTypedSplice) $
514
    do {    -- Typecheck the expression
515
         (expr', lie) <- captureConstraints tc_action
516 517 518 519

        -- Solve the constraints
        ; const_binds <- simplifyTop lie

520
          -- Zonk it and tie the knot of dictionary bindings
521
       ; zonkTopLExpr (mkHsDictLet (EvBinds const_binds) expr') }
522

Austin Seipp's avatar
Austin Seipp committed
523 524 525
{-
************************************************************************
*                                                                      *
526
        Annotations
Austin Seipp's avatar
Austin Seipp committed
527 528 529
*                                                                      *
************************************************************************
-}
530

531 532
runAnnotation target expr = do
    -- Find the classes we want instances for in order to call toAnnotationWrapper
533
    loc <- getSrcSpanM
534
    data_class <- tcLookupClass dataClassName
535
    to_annotation_wrapper_id <- tcLookupId toAnnotationWrapperName
536

537 538 539
    -- Check the instances we require live in another module (we want to execute it..)
    -- and check identifiers live in other modules using TH stage checks. tcSimplifyStagedExpr
    -- also resolves the LIE constraints to detect e.g. instance ambiguity
540
    zonked_wrapped_expr' <- tcTopSpliceExpr False $
541
           do { (expr', expr_ty) <- tcInferRhoNC expr
542 543 544
                -- We manually wrap the typechecked expression in a call to toAnnotationWrapper
                -- By instantiating the call >here< it gets registered in the
                -- LIE consulted by tcTopSpliceExpr
545
                -- and hence ensures the appropriate dictionary is bound by const_binds
546
              ; wrapper <- instCall AnnOrigin [expr_ty] [mkClassPred data_class [expr_ty]]
547
              ; let specialised_to_annotation_wrapper_expr
548 549
                      = L loc (HsWrap wrapper (HsVar to_annotation_wrapper_id))
              ; return (L loc (HsApp specialised_to_annotation_wrapper_expr expr')) }
550 551 552 553 554

    -- Run the appropriately wrapped expression to get the value of
    -- the annotation and its dictionaries. The return value is of
    -- type AnnotationWrapper by construction, so this conversion is
    -- safe
Luite Stegeman's avatar
Luite Stegeman committed
555 556 557 558 559 560 561 562
    serialized <- runMetaAW zonked_wrapped_expr'
    return Annotation {
               ann_target = target,
               ann_value = serialized
           }

convertAnnotationWrapper :: AnnotationWrapper -> Either MsgDoc Serialized
convertAnnotationWrapper  annotation_wrapper = Right $
563 564
        case annotation_wrapper of
            AnnotationWrapper value | let serialized = toSerialized serializeWithData value ->
565 566 567 568 569 570
                -- Got the value and dictionaries: build the serialized value and
                -- call it a day. We ensure that we seq the entire serialized value
                -- in order that any errors in the user-written code for the
                -- annotation are exposed at this point.  This is also why we are
                -- doing all this stuff inside the context of runMeta: it has the
                -- facilities to deal with user error in a meta-level expression
Luite Stegeman's avatar
Luite Stegeman committed
571 572
                seqSerialized serialized `seq` serialized

573

574

Austin Seipp's avatar
Austin Seipp committed
575 576 577
{-
************************************************************************
*                                                                      *
578
\subsection{Running an expression}
Austin Seipp's avatar
Austin Seipp committed
579 580 581
*                                                                      *
************************************************************************
-}
582

583 584 585
runQuasi :: TH.Q a -> TcM a
runQuasi act = TH.runQ act

Luite Stegeman's avatar
Luite Stegeman committed
586 587 588 589 590
runQResult :: (a -> String) -> (SrcSpan -> a -> b) -> SrcSpan -> TH.Q a -> TcM b
runQResult show_th f expr_span hval
  = do { th_result <- TH.runQ hval
       ; traceTc "Got TH result:" (text (show_th th_result))
       ; return (f expr_span th_result) }
591

Luite Stegeman's avatar
Luite Stegeman committed
592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610
-----------------
runMeta :: (MetaHook TcM -> LHsExpr Id -> TcM hs_syn)
        -> LHsExpr Id
        -> TcM hs_syn
runMeta unwrap e
  = do { h <- getHooked runMetaHook defaultRunMeta
       ; unwrap h e }

defaultRunMeta :: MetaHook TcM
defaultRunMeta (MetaE r)
  = fmap r . runMeta' True ppr (runQResult TH.pprint convertToHsExpr)
defaultRunMeta (MetaP r)
  = fmap r . runMeta' True ppr (runQResult TH.pprint convertToPat)
defaultRunMeta (MetaT r)
  = fmap r . runMeta' True ppr (runQResult TH.pprint convertToHsType)
defaultRunMeta (MetaD r)
  = fmap r . runMeta' True ppr (runQResult TH.pprint convertToHsDecls)
defaultRunMeta (MetaAW r)
  = fmap r . runMeta' False (const empty) (const (return . convertAnnotationWrapper))
611 612 613
    -- We turn off showing the code in meta-level exceptions because doing so exposes
    -- the toAnnotationWrapper function that we slap around the users code

Luite Stegeman's avatar
Luite Stegeman committed
614 615 616 617
----------------
runMetaAW :: LHsExpr Id         -- Of type AnnotationWrapper
          -> TcM Serialized
runMetaAW = runMeta metaRequestAW
618

619 620
runMetaE :: LHsExpr Id          -- Of type (Q Exp)
         -> TcM (LHsExpr RdrName)
Luite Stegeman's avatar
Luite Stegeman committed
621
runMetaE = runMeta metaRequestE
622

gmainland's avatar
gmainland committed
623 624
runMetaP :: LHsExpr Id          -- Of type (Q Pat)
         -> TcM (LPat RdrName)
Luite Stegeman's avatar
Luite Stegeman committed
625
runMetaP = runMeta metaRequestP
gmainland's avatar
gmainland committed
626

627 628
runMetaT :: LHsExpr Id          -- Of type (Q Type)
         -> TcM (LHsType RdrName)
Luite Stegeman's avatar
Luite Stegeman committed
629
runMetaT = runMeta metaRequestT
630

631 632
runMetaD :: LHsExpr Id          -- Of type Q [Dec]
         -> TcM [LHsDecl RdrName]
Luite Stegeman's avatar
Luite Stegeman committed
633
runMetaD = runMeta metaRequestD
634 635

---------------
Luite Stegeman's avatar
Luite Stegeman committed
636 637 638 639 640 641
runMeta' :: Bool                 -- Whether code should be printed in the exception message
         -> (hs_syn -> SDoc)                                    -- how to print the code
         -> (SrcSpan -> x -> TcM (Either MsgDoc hs_syn))        -- How to run x
         -> LHsExpr Id           -- Of type x; typically x = Q TH.Exp, or something like that
         -> TcM hs_syn           -- Of type t
runMeta' show_code ppr_hs run_and_convert expr
642
  = do  { traceTc "About to run" (ppr expr)
643 644
        ; recordThSpliceUse -- seems to be the best place to do this,
                            -- we catch all kinds of splices and annotations.
645

646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661
        -- Check that we've had no errors of any sort so far.
        -- For example, if we found an error in an earlier defn f, but
        -- recovered giving it type f :: forall a.a, it'd be very dodgy
        -- to carry ont.  Mind you, the staging restrictions mean we won't
        -- actually run f, but it still seems wrong. And, more concretely,
        -- see Trac #5358 for an example that fell over when trying to
        -- reify a function with a "?" kind in it.  (These don't occur
        -- in type-correct programs.
        ; failIfErrsM

        -- Desugar
        ; ds_expr <- initDsTc (dsLExpr expr)
        -- Compile and link it; might fail if linking fails
        ; hsc_env <- getTopEnv
        ; src_span <- getSrcSpanM
        ; traceTc "About to run (desugared)" (ppr ds_expr)
Simon Peyton Jones's avatar
Simon Peyton Jones committed
662
        ; either_hval <- tryM $ liftIO $
663 664
                         HscMain.hscCompileCoreExpr hsc_env src_span ds_expr
        ; case either_hval of {
665
            Left exn   -> fail_with_exn "compile and link" exn ;
666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
            Right hval -> do

        {       -- Coerce it to Q t, and run it

                -- Running might fail if it throws an exception of any kind (hence tryAllM)
                -- including, say, a pattern-match exception in the code we are running
                --
                -- We also do the TH -> HS syntax conversion inside the same
                -- exception-cacthing thing so that if there are any lurking
                -- exceptions in the data structure returned by hval, we'll
                -- encounter them inside the try
                --
                -- See Note [Exceptions in TH]
          let expr_span = getLoc expr
        ; either_tval <- tryAllM $
                         setSrcSpan expr_span $ -- Set the span so that qLocation can
                                                -- see where this splice is
             do { mb_result <- run_and_convert expr_span (unsafeCoerce# hval)
                ; case mb_result of
                    Left err     -> failWithTc err
Luite Stegeman's avatar
Luite Stegeman committed
686
                    Right result -> do { traceTc "Got HsSyn result:" (ppr_hs result)
687
                                       ; return $! result } }
688

689 690 691 692
        ; case either_tval of
            Right v -> return v
            Left se -> case fromException se of
                         Just IOEnvFailure -> failM -- Error already in Tc monad
693
                         _ -> fail_with_exn "run" se -- Exception
694
        }}}
695
  where
696 697 698 699 700
    -- see Note [Concealed TH exceptions]
    fail_with_exn phase exn = do
        exn_msg <- liftIO $ Panic.safeShowException exn
        let msg = vcat [text "Exception when trying to" <+> text phase <+> text "compile-time code:",
                        nest 2 (text exn_msg),
Simon Peyton Jones's avatar
Simon Peyton Jones committed
701
                        if show_code then text "Code:" <+> ppr expr else empty]
702
        failWithTc msg
703

Austin Seipp's avatar
Austin Seipp committed
704
{-
705 706
Note [Exceptions in TH]
~~~~~~~~~~~~~~~~~~~~~~~
707 708
Supppose we have something like this
        $( f 4 )
709
where
710 711 712
        f :: Int -> Q [Dec]
        f n | n>3       = fail "Too many declarations"
            | otherwise = ...
713 714 715 716 717 718

The 'fail' is a user-generated failure, and should be displayed as a
perfectly ordinary compiler error message, not a panic or anything
like that.  Here's how it's processed:

  * 'fail' is the monad fail.  The monad instance for Q in TH.Syntax
719 720
    effectively transforms (fail s) to
        qReport True s >> fail
721 722 723 724 725
    where 'qReport' comes from the Quasi class and fail from its monad
    superclass.

  * The TcM monad is an instance of Quasi (see TcSplice), and it implements
    (qReport True s) by using addErr to add an error message to the bag of errors.
726
    The 'fail' in TcM raises an IOEnvFailure exception
727

728 729 730 731 732 733
 * 'qReport' forces the message to ensure any exception hidden in unevaluated
   thunk doesn't get into the bag of errors. Otherwise the following splice
   will triger panic (Trac #8987):
        $(fail undefined)
   See also Note [Concealed TH exceptions]

734 735 736 737
  * So, when running a splice, we catch all exceptions; then for
        - an IOEnvFailure exception, we assume the error is already
                in the error-bag (above)
        - other errors, we add an error to the bag
738 739
    and then fail

740 741 742 743 744 745 746 747 748 749 750 751 752 753 754
Note [Concealed TH exceptions]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
When displaying the error message contained in an exception originated from TH
code, we need to make sure that the error message itself does not contain an
exception.  For example, when executing the following splice:

    $( error ("foo " ++ error "bar") )

the message for the outer exception is a thunk which will throw the inner
exception when evaluated.

For this reason, we display the message of a TH exception using the
'safeShowException' function, which recursively catches any exception thrown
when showing an error message.

755

756
To call runQ in the Tc monad, we need to make TcM an instance of Quasi:
Austin Seipp's avatar
Austin Seipp committed
757
-}
758

759
instance TH.Quasi (IOEnv (Env TcGblEnv TcLclEnv)) where
760 761 762
  qNewName s = do { u <- newUnique
                  ; let i = getKey u
                  ; return (TH.mkNameU s i) }
763

764 765 766 767
  -- 'msg' is forced to ensure exceptions don't escape,
  -- see Note [Exceptions in TH]
  qReport True msg  = seqList msg $ addErr  (text msg)
  qReport False msg = seqList msg $ addWarn (text msg)
768

769
  qLocation = do { m <- getModule
Ian Lynagh's avatar
Ian Lynagh committed
770 771 772 773 774 775 776
                 ; l <- getSrcSpanM
                 ; r <- case l of
                        UnhelpfulSpan _ -> pprPanic "qLocation: Unhelpful location"
                                                    (ppr l)
                        RealSrcSpan s -> return s
                 ; return (TH.Loc { TH.loc_filename = unpackFS (srcSpanFile r)
                                  , TH.loc_module   = moduleNameString (moduleName m)
777
                                  , TH.loc_package  = packageKeyString (modulePackageKey m)
Ian Lynagh's avatar
Ian Lynagh committed
778 779 780
                                  , TH.loc_start = (srcSpanStartLine r, srcSpanStartCol r)
                                  , TH.loc_end = (srcSpanEndLine   r, srcSpanEndCol   r) }) }

781 782 783 784 785
  qLookupName       = lookupName
  qReify            = reify
  qReifyInstances   = reifyInstances
  qReifyRoles       = reifyRoles
  qReifyAnnotations = reifyAnnotations
786
  qReifyModule      = reifyModule
787

788 789 790
        -- For qRecover, discard error messages if
        -- the recovery action is chosen.  Otherwise
        -- we'll only fail higher up.  c.f. tryTcLIE_
791
  qRecover recover main = do { (msgs, mb_res) <- tryTcErrs main
792 793 794 795 796
                             ; case mb_res of
                                 Just val -> do { addMessages msgs      -- There might be warnings
                                                ; return val }
                                 Nothing  -> recover                    -- Discard all msgs
                          }
797

twanvl's avatar
twanvl committed
798
  qRunIO io = liftIO io
GregWeber's avatar
GregWeber committed
799 800 801 802 803

  qAddDependentFile fp = do
    ref <- fmap tcg_dependent_files getGblEnv
    dep_files <- readTcRef ref
    writeTcRef ref (fp:dep_files)
804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823

  qAddTopDecls thds = do
      l <- getSrcSpanM
      let either_hval = convertToHsDecls l thds
      ds <- case either_hval of
              Left exn -> pprPanic "qAddTopDecls: can't convert top-level declarations" exn
              Right ds -> return ds
      mapM_ (checkTopDecl . unLoc) ds
      th_topdecls_var <- fmap tcg_th_topdecls getGblEnv
      updTcRef th_topdecls_var (\topds -> ds ++ topds)
    where
      checkTopDecl :: HsDecl RdrName -> TcM ()
      checkTopDecl (ValD binds)
        = mapM_ bindName (collectHsBindBinders binds)
      checkTopDecl (SigD _)
        = return ()
      checkTopDecl (ForD (ForeignImport (L _ name) _ _ _))
        = bindName name
      checkTopDecl _
        = addErr $ text "Only function, value, and foreign import declarations may be added with addTopDecl"
824

825 826 827
      bindName :: RdrName -> TcM ()
      bindName (Exact n)
        = do { th_topnames_var <- fmap tcg_th_topnames getGblEnv
828
             ; updTcRef th_topnames_var (\ns -> extendNameSet ns n)
829 830 831 832 833 834
             }

      bindName name =
          addErr $
          hang (ptext (sLit "The binder") <+> quotes (ppr name) <+> ptext (sLit "is not a NameU."))
             2 (text "Probable cause: you used mkName instead of newName to generate a binding.")
835 836 837 838

  qAddModFinalizer fin = do
      th_modfinalizers_var <- fmap tcg_th_modfinalizers getGblEnv
      updTcRef th_modfinalizers_var (\fins -> fin:fins)
839

840
  qGetQ :: forall a. Typeable a => IOEnv (Env TcGblEnv TcLclEnv) (Maybe a)
841 842 843
  qGetQ = do
      th_state_var <- fmap tcg_th_state getGblEnv
      th_state <- readTcRef th_state_var
844 845 846 847
      -- See #10596 for why we use a scoped type variable here.
      -- ToDo: convert @undefined :: a@ to @proxy :: Proxy a@ when
      -- we drop support for GHC 7.6.
      return (Map.lookup (typeOf (undefined :: a)) th_state >>= fromDynamic)
848 849 850 851

  qPutQ x = do
      th_state_var <- fmap tcg_th_state getGblEnv
      updTcRef th_state_var (\m -> Map.insert (typeOf x) (toDyn x) m)
852

853

Austin Seipp's avatar
Austin Seipp committed
854 855 856
{-
************************************************************************
*                                                                      *
857
            Instance Testing
Austin Seipp's avatar
Austin Seipp committed
858 859 860
*                                                                      *
************************************************************************
-}
861

862 863
reifyInstances :: TH.Name -> [TH.Type] -> TcM [TH.Dec]
reifyInstances th_nm th_tys
864
   = addErrCtxt (ptext (sLit "In the argument of reifyInstances:")
865
                 <+> ppr_th th_nm <+> sep (map ppr_th th_tys)) $
866 867
     do { loc <- getSrcSpanM
        ; rdr_ty <- cvt loc (mkThAppTs (TH.ConT th_nm) th_tys)
868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883
          -- #9262 says to bring vars into scope, like in HsForAllTy case
          -- of rnHsTyKi
        ; let (kvs, tvs) = extractHsTyRdrTyVars rdr_ty
              tv_bndrs   = userHsTyVarBndrs loc tvs
              hs_tvbs    = mkHsQTvs tv_bndrs
          -- Rename  to HsType Name
        ; ((rn_tvbs, rn_ty), _fvs)
            <- bindHsTyVars doc Nothing kvs hs_tvbs $ \ rn_tvbs ->
               do { (rn_ty, fvs) <- rnLHsType doc rdr_ty
                  ; return ((rn_tvbs, rn_ty), fvs) }
        ; (ty, _kind) <- tcHsTyVarBndrs rn_tvbs $ \ _tvs ->
                         tcLHsType rn_ty
        ; ty <- zonkTcTypeToType emptyZonkEnv ty
                -- Substitute out the meta type variables
                -- In particular, the type might have kind
                -- variables inside it (Trac #7477)
884

885
        ; traceTc "reifyInstances" (ppr ty $$ ppr (typeKind ty))
886 887 888 889
        ; case splitTyConApp_maybe ty of   -- This expands any type synonyms
            Just (tc, tys)                 -- See Trac #7910
               | Just cls <- tyConClass_maybe tc
               -> do { inst_envs <- tcGetInstEnvs
890
                     ; let (matches, unifies, _) = lookupInstEnv False inst_envs cls tys
891
                     ; traceTc "reifyInstances1" (ppr matches)
892
                     ; reifyClassInstances cls (map fst matches ++ unifies) }
893
               | isOpenFamilyTyCon tc
894 895
               -> do { inst_envs <- tcGetFamInstEnvs
                     ; let matches = lookupFamInstEnv inst_envs tc tys
896
                     ; traceTc "reifyInstances2" (ppr matches)
897
                     ; reifyFamilyInstances tc (map fim_instance matches) }
898
            _  -> bale_out (hang (ptext (sLit "reifyInstances:") <+> quotes (ppr ty))
899
                               2 (ptext (sLit "is not a class constraint or type family application"))) }
900
  where
dreixel's avatar
dreixel committed
901
    doc = ClassInstanceCtx
902 903 904 905 906 907
    bale_out msg = failWithTc msg

    cvt :: SrcSpan -> TH.Type -> TcM (LHsType RdrName)
    cvt loc th_ty = case convertToHsType loc th_ty of
                      Left msg -> failWithTc msg
                      Right ty -> return ty
908

Austin Seipp's avatar
Austin Seipp committed
909 910 911
{-
************************************************************************
*                                                                      *
912
                        Reification
Austin Seipp's avatar
Austin Seipp committed
913 914 915
*                                                                      *
************************************************************************
-}
916

917 918
lookupName :: Bool      -- True  <=> type namespace
                        -- False <=> value namespace
919 920 921 922 923 924 925 926
           -> String -> TcM (Maybe TH.Name)
lookupName is_type_name s
  = do { lcl_env <- getLocalRdrEnv
       ; case lookupLocalRdrEnv lcl_env rdr_name of
           Just n  -> return (Just (reifyName n))
           Nothing -> do { mb_nm <- lookupGlobalOccRn_maybe rdr_name
                         ; return (fmap reifyName mb_nm) } }
  where
927
    th_name = TH.mkName s       -- Parses M.x into a base of 'x' and a module of 'M'
928 929 930 931 932

    occ_fs :: FastString
    occ_fs = mkFastString (TH.nameBase th_name)

    occ :: OccName
933 934
    occ | is_type_name
        = if isLexCon occ_fs then mkTcOccFS    occ_fs
935
                             else mkTyVarOccFS occ_fs
936
        | otherwise
937 938 939 940 941 942 943 944
        = if isLexCon occ_fs then mkDataOccFS occ_fs
                             else mkVarOccFS  occ_fs

    rdr_name = case TH.nameModule th_name of
                 Nothing  -> mkRdrUnqual occ
                 Just mod -> mkRdrQual (mkModuleName mod) occ

getThing :: TH.Name -> TcM TcTyThing
945 946 947 948 949 950
getThing th_name
  = do  { name <- lookupThName th_name
        ; traceIf (text "reify" <+> text (show th_name) <+> brackets (ppr_ns th_name) <+> ppr name)
        ; tcLookupTh name }
        -- ToDo: this tcLookup could fail, which would give a
        --       rather unhelpful error message
951
  where
952
    ppr_ns (TH.Name _ (TH.NameG TH.DataName  _pkg _mod)) = text "data"
Simon Marlow's avatar
Simon Marlow committed
953
    ppr_ns (TH.Name _ (TH.NameG TH.TcClsName _pkg _mod)) = text "tc"
954
    ppr_ns (TH.Name _ (TH.NameG TH.VarName   _pkg _mod)) = text "var"
955
    ppr_ns _ = panic "reify/ppr_ns"
956

957 958
reify :: TH.Name -> TcM TH.Info
reify th_name
959 960 961
  = do  { traceTc "reify 1" (text (TH.showName th_name))
        ; thing <- getThing th_name
        ; traceTc "reify 2" (ppr thing)
962
        ; reifyThing thing }
963

964
lookupThName :: TH.Name -> TcM Name
965 966 967 968 969 970
lookupThName th_name = do
    mb_name <- lookupThName_maybe th_name
    case mb_name of
        Nothing   -> failWithTc (notInScope th_name)
        Just name -> return name

971
lookupThName_maybe :: TH.Name -> TcM (Maybe Name)
972 973 974
lookupThName_maybe th_name
  =  do { names <- mapMaybeM lookup (thRdrNameGuesses th_name)
          -- Pick the first that works
975 976
          -- E.g. reify (mkName "A") will pick the class A in preference to the data constructor A
        ; return (listToMaybe names) }
977
  where
978
    lookup rdr_name
979 980 981 982 983 984
        = do {  -- Repeat much of lookupOccRn, becase we want
                -- to report errors in a TH-relevant way
             ; rdr_env <- getLocalRdrEnv
             ; case lookupLocalRdrEnv rdr_env rdr_name of
                 Just name -> return (Just name)
                 Nothing   -> lookupGlobalOccRn_maybe rdr_name }
985

986 987 988 989 990
tcLookupTh :: Name -> TcM TcTyThing
-- This is a specialised version of TcEnv.tcLookup; specialised mainly in that
-- it gives a reify-related error message on failure, whereas in the normal
-- tcLookup, failure is a bug.
tcLookupTh name
991 992 993
  = do  { (gbl_env, lcl_env) <- getEnvs
        ; case lookupNameEnv (tcl_env lcl_env) name of {
                Just thing -> return thing;
994 995 996 997 998 999 1000
                Nothing    ->

          case lookupNameEnv (tcg_type_env gbl_env) name of {
                Just thing -> return (AGlobal thing);
                Nothing    ->

          if nameIsLocalOrFrom (tcg_mod gbl_env) name
1001
          then  -- It's defined in this module
1002
                failWithTc (notInEnv name)
1003

1004 1005
          else
     do { mb_thing <- tcLookupImported_maybe name
1006 1007 1008
        ; case mb_thing of
            Succeeded thing -> return (AGlobal thing)
            Failed msg      -> failWithTc msg
1009
    }}}}
1010

1011
notInScope :: TH.Name -> SDoc
1012 1013 1014
notInScope th_name = quotes (text (TH.pprint th_name)) <+>
                     ptext (sLit "is not in scope at a reify")
        -- Ugh! Rather an indirect way to display the name
1015

1016
notInEnv :: Name -> SDoc
1017 1018
notInEnv name = quotes (ppr name) <+>
                     ptext (sLit "is not in the type environment at a reify")
1019

1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032
------------------------------
reifyRoles :: TH.Name -> TcM [TH.Role]
reifyRoles th_name
  = do { thing <- getThing th_name
       ; case thing of
           AGlobal (ATyCon tc) -> return (map reify_role (tyConRoles tc))
           _ -> failWithTc (ptext (sLit "No roles associated with") <+> (ppr thing))
       }
  where
    reify_role Nominal          = TH.NominalR
    reify_role Representational = TH.RepresentationalR
    reify_role Phantom          = TH.PhantomR

1033 1034 1035 1036 1037 1038 1039
------------------------------
reifyThing :: TcTyThing -> TcM TH.Info
-- The only reason this is monadic is for error reporting,
-- which in turn is mainly for the case when TH can't express
-- some random GHC extension

reifyThing (AGlobal (AnId id))
1040 1041 1042 1043 1044 1045
  = do  { ty <- reifyType (idType id)
        ; fix <- reifyFixity (idName id)
        ; let v = reifyName id
        ; case idDetails id of
            ClassOpId cls -> return (TH.ClassOpI v ty (reifyName cls) fix)
            _             -> return (TH.VarI     v ty Nothing fix)
1046 1047
    }

1048
reifyThing (AGlobal (ATyCon tc))   = reifyTyCon tc
cactus's avatar
cactus committed
1049
reifyThing (AGlobal (AConLike (RealDataCon dc)))
1050 1051 1052 1053 1054
  = do  { let name = dataConName dc
        ; ty <- reifyType (idType (dataConWrapId dc))
        ; fix <- reifyFixity name
        ; return (TH.DataConI (reifyName name) ty
                              (reifyName (dataConOrigTyCon dc)) fix)
1055
        }
1056
reifyThing (AGlobal (AConLike (PatSynCon ps)))
1057
  = noTH (sLit "pattern synonyms") (ppr $ patSynName ps)
1058

1059 1060 1061 1062 1063 1064
reifyThing (ATcId {tct_id = id})
  = do  { ty1 <- zonkTcType (idType id) -- Make use of all the info we have, even
                                        -- though it may be incomplete
        ; ty2 <- reifyType ty1
        ; fix <- reifyFixity (idName id)
        ; return (TH.VarI (reifyName id) ty2 Nothing fix) }
1065

1066 1067 1068 1069
reifyThing (ATyVar tv tv1)
  = do { ty1 <- zonkTcTyVar tv1
       ; ty2 <- reifyType ty1
       ; return (TH.TyVarI (reifyName tv) ty2) }
1070

1071
reifyThing thing = pprPanic "reifyThing" (pprTcTyThingCategory thing)
1072

1073
-------------------------------------------
1074 1075
reifyAxBranch :: CoAxBranch -> TcM TH.TySynEqn
reifyAxBranch (CoAxBranch { cab_lhs = args, cab_rhs = rhs })
eir@cis.upenn.edu's avatar
eir@cis.upenn.edu committed
1076 1077
            -- remove kind patterns (#8884)
  = do { args' <- mapM reifyType (filter (not . isKind) args)
1078
       ; rhs'  <- reifyType rhs
1079
       ; return (TH.TySynEqn args' rhs') }
1080

1081
reifyTyCon :: TyCon -> TcM TH.Info
1082
reifyTyCon tc
batterseapower's avatar
batterseapower committed
1083 1084 1085
  | Just cls <- tyConClass_maybe tc
  = reifyClass cls

1086 1087
  | isFunTyCon tc
  = return (TH.PrimTyConI (reifyName tc) 2                False)
1088

1089
  | isPrimTyCon tc
1090
  = return (TH.PrimTyConI (reifyName tc) (tyConArity tc) (isUnLiftedTyCon tc))
1091

1092
  | isFamilyTyCon tc
eir@cis.upenn.edu's avatar
eir@cis.upenn.edu committed
1093 1094 1095 1096 1097 1098 1099 1100 1101
  = do { let tvs      = tyConTyVars tc
             kind     = tyConKind tc

             -- we need the *result kind* (see #8884)
             (kvs, mono_kind) = splitForAllTys kind
                                -- tyConArity includes *kind* params
             (_, res_kind)    = splitKindFunTysN (tyConArity tc - length kvs)
                                                 mono_kind
       ; kind' <- fmap Just (reifyKind res_kind)
1102

1103
       ; tvs' <- reifyTyVars tvs
1104 1105 1106 1107
       ; flav' <- reifyFamFlavour tc
       ; case flav' of
         { Left flav ->  -- open type/data family
             do { fam_envs <- tcGetFamInstEnvs
1108 1109
                ; instances <- reifyFamilyInstances tc
                                 (familyInstances fam_envs tc)
1110 1111 1112 1113 1114 1115 1116
                ; return (TH.FamilyI
                            (TH.FamilyD flav (reifyName tc) tvs' kind')
                            instances) }
         ; Right eqns -> -- closed type family
             return (TH.FamilyI
                      (TH.ClosedTypeFamilyD (reifyName tc) tvs' kind' eqns)
                      []) } }
1117

1118 1119
  | Just (tvs, rhs) <- synTyConDefn_maybe tc  -- Vanilla type synonym
  = do { rhs' <- reifyType rhs
1120
       ; tvs' <- reifyTyVars tvs
1121
       ; return (TH.TyConI
1122
                   (TH.TySynD (reifyName tc) tvs' rhs'))
1123
       }
1124

1125
  | otherwise
1126 1127 1128
  = do  { cxt <- reifyCxt (tyConStupidTheta tc)
        ; let tvs = tyConTyVars tc
        ; cons <- mapM (reifyDataCon (mkTyVarTys tvs)) (tyConDataCons tc)
1129
        ; r_tvs <- reifyTyVars tvs
1130 1131 1132 1133 1134
        ; let name = reifyName tc
              deriv = []        -- Don't know about deriving
              decl | isNewTyCon tc = TH.NewtypeD cxt name r_tvs (head cons) deriv
                   | otherwise     = TH.DataD    cxt name r_tvs cons        deriv
        ; return (TH.TyConI decl) }
1135

1136
reifyDataCon :: [Type] -> DataCon -> TcM TH.Con
1137
-- For GADTs etc, see Note [Reifying data constructors]
1138
reifyDataCon tys dc
1139
  = do { let (tvs, theta, arg_tys, _) = dataConSig dc
1140
             subst             = mkTopTvSubst (tvs `zip` tys)   -- Dicard ex_tvs
1141 1142 1143
             (subst', ex_tvs') = mapAccumL substTyVarBndr subst (dropList tys tvs)
             theta'   = substTheta subst' theta
             arg_tys' = substTys subst' arg_tys
Simon Peyton Jones's avatar
Simon Peyton Jones committed
1144
             stricts  = map reifyStrict (dataConSrcBangs dc)
1145 1146
             fields   = dataConFieldLabels dc
             name     = reifyName dc
1147 1148 1149

       ; r_arg_tys <- reifyTypes arg_tys'

1150
       ; let main_con | not (null fields)
1151 1152 1153
                      = TH.RecC name (zip3 (map reifyName fields) stricts r_arg_tys)
                      | dataConIsInfix dc
                      = ASSERT( length arg_tys == 2 )
1154
                        TH.InfixC (s1,r_a1) name (s2,r_a2)
simonpj@microsoft.com's avatar