{-# LANGUAGE CPP #-}
{-# LANGUAGE TemplateHaskell #-}
module Language.Haskell.TH.Desugar.Sweeten (
expToTH, matchToTH, patToTH, decsToTH, decToTH,
letDecToTH, typeToTH,
conToTH, foreignToTH, pragmaToTH, ruleBndrToTH,
clauseToTH, tvbToTH, cxtToTH, predToTH, derivClauseToTH,
#if __GLASGOW_HASKELL__ >= 801
patSynDirToTH,
#endif
typeArgToTH
) where
import Prelude hiding (exp)
import Control.Arrow
import Language.Haskell.TH hiding (cxt)
import Language.Haskell.TH.Desugar.AST
import Language.Haskell.TH.Desugar.Core (DTypeArg(..))
import Language.Haskell.TH.Desugar.Util
import Data.Maybe ( maybeToList, mapMaybe )
expToTH :: DExp -> Exp
expToTH (DVarE n) = VarE n
expToTH (DConE n) = ConE n
expToTH (DLitE l) = LitE l
expToTH (DAppE e1 e2) = AppE (expToTH e1) (expToTH e2)
expToTH (DLamE names exp) = LamE (map VarP names) (expToTH exp)
expToTH (DCaseE exp matches) = CaseE (expToTH exp) (map matchToTH matches)
expToTH (DLetE decs exp) = LetE (mapMaybe letDecToTH decs) (expToTH exp)
expToTH (DSigE exp ty) = SigE (expToTH exp) (typeToTH ty)
#if __GLASGOW_HASKELL__ < 709
expToTH (DStaticE _) = error "Static expressions supported only in GHC 7.10+"
#else
expToTH (DStaticE exp) = StaticE (expToTH exp)
#endif
#if __GLASGOW_HASKELL__ >= 801
expToTH (DAppTypeE exp ty) = AppTypeE (expToTH exp) (typeToTH ty)
#else
expToTH (DAppTypeE exp _) = expToTH exp
#endif
matchToTH :: DMatch -> Match
matchToTH (DMatch pat exp) = Match (patToTH pat) (NormalB (expToTH exp)) []
patToTH :: DPat -> Pat
patToTH (DLitP lit) = LitP lit
patToTH (DVarP n) = VarP n
patToTH (DConP n pats) = ConP n (map patToTH pats)
patToTH (DTildeP pat) = TildeP (patToTH pat)
patToTH (DBangP pat) = BangP (patToTH pat)
patToTH (DSigP pat ty) = SigP (patToTH pat) (typeToTH ty)
patToTH DWildP = WildP
decsToTH :: [DDec] -> [Dec]
decsToTH = concatMap decToTH
decToTH :: DDec -> [Dec]
decToTH (DLetDec d) = maybeToList (letDecToTH d)
decToTH (DDataD Data cxt n tvbs _mk cons derivings) =
#if __GLASGOW_HASKELL__ > 710
[DataD (cxtToTH cxt) n (map tvbToTH tvbs) (fmap typeToTH _mk) (map conToTH cons)
(concatMap derivClauseToTH derivings)]
#else
[DataD (cxtToTH cxt) n (map tvbToTH tvbs) (map conToTH cons)
(map derivingToTH derivings)]
#endif
decToTH (DDataD Newtype cxt n tvbs _mk [con] derivings) =
#if __GLASGOW_HASKELL__ > 710
[NewtypeD (cxtToTH cxt) n (map tvbToTH tvbs) (fmap typeToTH _mk) (conToTH con)
(concatMap derivClauseToTH derivings)]
#else
[NewtypeD (cxtToTH cxt) n (map tvbToTH tvbs) (conToTH con)
(map derivingToTH derivings)]
#endif
decToTH (DTySynD n tvbs ty) = [TySynD n (map tvbToTH tvbs) (typeToTH ty)]
decToTH (DClassD cxt n tvbs fds decs) =
[ClassD (cxtToTH cxt) n (map tvbToTH tvbs) fds (decsToTH decs)]
decToTH (DInstanceD over mtvbs _cxt _ty decs) =
[instanceDToTH over cxt' ty' decs]
where
(cxt', ty') = case mtvbs of
Nothing -> (_cxt, _ty)
Just _tvbs ->
#if __GLASGOW_HASKELL__ < 800 || __GLASGOW_HASKELL__ >= 802
([], DForallT _tvbs _cxt _ty)
#else
error $ "Explicit foralls in instance declarations "
++ "are broken on GHC 8.0."
#endif
decToTH (DForeignD f) = [ForeignD (foreignToTH f)]
#if __GLASGOW_HASKELL__ > 710
decToTH (DOpenTypeFamilyD (DTypeFamilyHead n tvbs frs ann)) =
[OpenTypeFamilyD (TypeFamilyHead n (map tvbToTH tvbs) (frsToTH frs) ann)]
#else
decToTH (DOpenTypeFamilyD (DTypeFamilyHead n tvbs frs _ann)) =
[FamilyD TypeFam n (map tvbToTH tvbs) (frsToTH frs)]
#endif
decToTH (DDataFamilyD n tvbs mk) =
#if __GLASGOW_HASKELL__ > 710
[DataFamilyD n (map tvbToTH tvbs) (fmap typeToTH mk)]
#else
[FamilyD DataFam n (map tvbToTH tvbs) (fmap typeToTH mk)]
#endif
decToTH (DDataInstD nd cxt mtvbs lhs mk cons derivings) =
let ndc = case (nd, cons) of
(Newtype, [con]) -> DNewtypeCon con
(Newtype, _) -> error "Newtype that doesn't have only one constructor"
(Data, _) -> DDataCons cons
in dataInstDecToTH ndc cxt mtvbs lhs mk derivings
#if __GLASGOW_HASKELL__ >= 807
decToTH (DTySynInstD eqn) = [TySynInstD (snd $ tySynEqnToTH eqn)]
#else
decToTH (DTySynInstD eqn) =
let (n, eqn') = tySynEqnToTH eqn in
[TySynInstD n eqn']
#endif
#if __GLASGOW_HASKELL__ > 710
decToTH (DClosedTypeFamilyD (DTypeFamilyHead n tvbs frs ann) eqns) =
[ClosedTypeFamilyD (TypeFamilyHead n (map tvbToTH tvbs) (frsToTH frs) ann)
(map (snd . tySynEqnToTH) eqns)
]
#else
decToTH (DClosedTypeFamilyD (DTypeFamilyHead n tvbs frs _ann) eqns) =
[ClosedTypeFamilyD n (map tvbToTH tvbs) (frsToTH frs) (map (snd . tySynEqnToTH) eqns)]
#endif
decToTH (DRoleAnnotD n roles) = [RoleAnnotD n roles]
decToTH (DStandaloneDerivD mds mtvbs _cxt _ty) =
[standaloneDerivDToTH mds cxt' ty']
where
(cxt', ty') = case mtvbs of
Nothing -> (_cxt, _ty)
Just _tvbs ->
#if __GLASGOW_HASKELL__ < 710 || __GLASGOW_HASKELL__ >= 802
([], DForallT _tvbs _cxt _ty)
#else
error $ "Explicit foralls in standalone deriving declarations "
++ "are broken on GHC 7.10 and 8.0."
#endif
#if __GLASGOW_HASKELL__ < 709
decToTH (DDefaultSigD {}) =
error "Default method signatures supported only in GHC 7.10+"
#else
decToTH (DDefaultSigD n ty) = [DefaultSigD n (typeToTH ty)]
#endif
#if __GLASGOW_HASKELL__ >= 801
decToTH (DPatSynD n args dir pat) = [PatSynD n args (patSynDirToTH dir) (patToTH pat)]
decToTH (DPatSynSigD n ty) = [PatSynSigD n (typeToTH ty)]
#else
decToTH dec
| DPatSynD{} <- dec = patSynErr
| DPatSynSigD{} <- dec = patSynErr
where
patSynErr = error "Pattern synonyms supported only in GHC 8.2+"
#endif
decToTH _ = error "Newtype declaration without exactly 1 constructor."
data DNewOrDataCons
= DNewtypeCon DCon
| DDataCons [DCon]
dataInstDecToTH :: DNewOrDataCons -> DCxt -> Maybe [DTyVarBndr] -> DType
-> Maybe DKind -> [DDerivClause] -> [Dec]
dataInstDecToTH ndc cxt _mtvbs lhs _mk derivings =
case ndc of
DNewtypeCon con ->
#if __GLASGOW_HASKELL__ >= 807
[NewtypeInstD (cxtToTH cxt) (fmap (fmap tvbToTH) _mtvbs) (typeToTH lhs)
(fmap typeToTH _mk) (conToTH con)
(concatMap derivClauseToTH derivings)]
#elif __GLASGOW_HASKELL__ > 710
[NewtypeInstD (cxtToTH cxt) _n _lhs_args (fmap typeToTH _mk) (conToTH con)
(concatMap derivClauseToTH derivings)]
#else
[NewtypeInstD (cxtToTH cxt) _n _lhs_args (conToTH con)
(map derivingToTH derivings)]
#endif
DDataCons cons ->
#if __GLASGOW_HASKELL__ >= 807
[DataInstD (cxtToTH cxt) (fmap (fmap tvbToTH) _mtvbs) (typeToTH lhs)
(fmap typeToTH _mk) (map conToTH cons)
(concatMap derivClauseToTH derivings)]
#elif __GLASGOW_HASKELL__ > 710
[DataInstD (cxtToTH cxt) _n _lhs_args (fmap typeToTH _mk) (map conToTH cons)
(concatMap derivClauseToTH derivings)]
#else
[DataInstD (cxtToTH cxt) _n _lhs_args (map conToTH cons)
(map derivingToTH derivings)]
#endif
where
_lhs' = typeToTH lhs
(_n, _lhs_args) =
case unfoldType _lhs' of
(ConT n, lhs_args) -> (n, filterTANormals lhs_args)
(_, _) -> error $ "Illegal data instance LHS: " ++ pprint _lhs'
#if __GLASGOW_HASKELL__ > 710
frsToTH :: DFamilyResultSig -> FamilyResultSig
frsToTH DNoSig = NoSig
frsToTH (DKindSig k) = KindSig (typeToTH k)
frsToTH (DTyVarSig tvb) = TyVarSig (tvbToTH tvb)
#else
frsToTH :: DFamilyResultSig -> Maybe Kind
frsToTH DNoSig = Nothing
frsToTH (DKindSig k) = Just (typeToTH k)
frsToTH (DTyVarSig (DPlainTV _)) = Nothing
frsToTH (DTyVarSig (DKindedTV _ k)) = Just (typeToTH k)
#endif
#if __GLASGOW_HASKELL__ <= 710
derivingToTH :: DDerivClause -> Name
derivingToTH (DDerivClause _ [DConT nm]) = nm
derivingToTH p =
error ("Template Haskell in GHC < 8.0 only allows simple derivings: " ++ show p)
#endif
letDecToTH :: DLetDec -> Maybe Dec
letDecToTH (DFunD name clauses) = Just $ FunD name (map clauseToTH clauses)
letDecToTH (DValD pat exp) = Just $ ValD (patToTH pat) (NormalB (expToTH exp)) []
letDecToTH (DSigD name ty) = Just $ SigD name (typeToTH ty)
letDecToTH (DInfixD f name) = Just $ InfixD f name
letDecToTH (DPragmaD prag) = fmap PragmaD (pragmaToTH prag)
conToTH :: DCon -> Con
#if __GLASGOW_HASKELL__ > 710
conToTH (DCon [] [] n (DNormalC _ stys) rty) =
GadtC [n] (map (second typeToTH) stys) (typeToTH rty)
conToTH (DCon [] [] n (DRecC vstys) rty) =
RecGadtC [n] (map (thirdOf3 typeToTH) vstys) (typeToTH rty)
#else
conToTH (DCon [] [] n (DNormalC True [sty1, sty2]) _) =
InfixC ((bangToStrict *** typeToTH) sty1) n ((bangToStrict *** typeToTH) sty2)
conToTH (DCon [] [] n (DNormalC _ stys) _) =
NormalC n (map (bangToStrict *** typeToTH) stys)
conToTH (DCon [] [] n (DRecC vstys) _) =
RecC n (map (\(v,b,t) -> (v,bangToStrict b,typeToTH t)) vstys)
#endif
#if __GLASGOW_HASKELL__ > 710
conToTH (DCon tvbs cxt n fields rty) =
ForallC (map tvbToTH tvbs) (cxtToTH cxt) (conToTH $ DCon [] [] n fields rty)
#else
conToTH (DCon tvbs cxt n fields rty)
| null ex_tvbs && null cxt
= con'
| otherwise
= ForallC ex_tvbs (cxtToTH cxt) con'
where
ex_tvbs :: [TyVarBndr]
ex_tvbs = map tvbToTH $ drop num_univ_tvs tvbs
num_univ_tvs :: Int
num_univ_tvs = go rty
where
go :: DType -> Int
go (DAppT t1 t2) = go t1 + go t2
go (DSigT t _) = go t
go (DVarT {}) = 1
go (DConT {}) = 0
go DArrowT = 0
go (DLitT {}) = 0
go (DForallT {}) = error "`forall` type used in GADT return type"
go DWildCardT = 0
go (DAppKindT {}) = 0
con' :: Con
con' = conToTH $ DCon [] [] n fields rty
#endif
instanceDToTH :: Maybe Overlap -> DCxt -> DType -> [DDec] -> Dec
instanceDToTH _over cxt ty decs =
InstanceD
#if __GLASGOW_HASKELL__ >= 800
_over
#endif
(cxtToTH cxt) (typeToTH ty) (decsToTH decs)
standaloneDerivDToTH :: Maybe DDerivStrategy -> DCxt -> DType -> Dec
#if __GLASGOW_HASKELL__ >= 710
standaloneDerivDToTH _mds cxt ty =
StandaloneDerivD
#if __GLASGOW_HASKELL__ >= 802
(fmap derivStrategyToTH _mds)
#endif
(cxtToTH cxt) (typeToTH ty)
#else
standaloneDerivDToTH _ _ _ = error "Standalone deriving supported only in GHC 7.10+"
#endif
foreignToTH :: DForeign -> Foreign
foreignToTH (DImportF cc safety str n ty) =
ImportF cc safety str n (typeToTH ty)
foreignToTH (DExportF cc str n ty) = ExportF cc str n (typeToTH ty)
pragmaToTH :: DPragma -> Maybe Pragma
pragmaToTH (DInlineP n inl rm phases) = Just $ InlineP n inl rm phases
pragmaToTH (DSpecialiseP n ty m_inl phases) =
Just $ SpecialiseP n (typeToTH ty) m_inl phases
pragmaToTH (DSpecialiseInstP ty) = Just $ SpecialiseInstP (typeToTH ty)
#if __GLASGOW_HASKELL__ >= 807
pragmaToTH (DRuleP str mtvbs rbs lhs rhs phases) =
Just $ RuleP str (fmap (fmap tvbToTH) mtvbs) (map ruleBndrToTH rbs)
(expToTH lhs) (expToTH rhs) phases
#else
pragmaToTH (DRuleP str _ rbs lhs rhs phases) =
Just $ RuleP str (map ruleBndrToTH rbs) (expToTH lhs) (expToTH rhs) phases
#endif
pragmaToTH (DAnnP target exp) = Just $ AnnP target (expToTH exp)
#if __GLASGOW_HASKELL__ < 709
pragmaToTH (DLineP {}) = Nothing
#else
pragmaToTH (DLineP n str) = Just $ LineP n str
#endif
#if __GLASGOW_HASKELL__ < 801
pragmaToTH (DCompleteP {}) = Nothing
#else
pragmaToTH (DCompleteP cls mty) = Just $ CompleteP cls mty
#endif
ruleBndrToTH :: DRuleBndr -> RuleBndr
ruleBndrToTH (DRuleVar n) = RuleVar n
ruleBndrToTH (DTypedRuleVar n ty) = TypedRuleVar n (typeToTH ty)
#if __GLASGOW_HASKELL__ >= 807
tySynEqnToTH :: DTySynEqn -> (Name, TySynEqn)
tySynEqnToTH (DTySynEqn tvbs lhs rhs) =
let lhs' = typeToTH lhs in
case unfoldType lhs' of
(ConT n, _lhs_args) -> (n, TySynEqn (fmap (fmap tvbToTH) tvbs) lhs' (typeToTH rhs))
(_, _) -> error $ "Illegal type instance LHS: " ++ pprint lhs'
#else
tySynEqnToTH :: DTySynEqn -> (Name, TySynEqn)
tySynEqnToTH (DTySynEqn _ lhs rhs) =
let lhs' = typeToTH lhs in
case unfoldType lhs' of
(ConT n, lhs_args) -> (n, TySynEqn (filterTANormals lhs_args) (typeToTH rhs))
(_, _) -> error $ "Illegal type instance LHS: " ++ pprint lhs'
#endif
clauseToTH :: DClause -> Clause
clauseToTH (DClause pats exp) = Clause (map patToTH pats) (NormalB (expToTH exp)) []
typeToTH :: DType -> Type
typeToTH (DForallT tvbs cxt ty) = ForallT (map tvbToTH tvbs) (map predToTH cxt) (typeToTH ty)
typeToTH (DAppT t1 t2) = AppT (typeToTH t1) (typeToTH t2)
typeToTH (DSigT ty ki) = SigT (typeToTH ty) (typeToTH ki)
typeToTH (DVarT n) = VarT n
typeToTH (DConT n) = tyconToTH n
typeToTH DArrowT = ArrowT
typeToTH (DLitT lit) = LitT lit
#if __GLASGOW_HASKELL__ > 710
typeToTH DWildCardT = WildCardT
#else
typeToTH DWildCardT = error "Wildcards supported only in GHC 8.0+"
#endif
#if __GLASGOW_HASKELL__ >= 807
typeToTH (DAppKindT t k) = AppKindT (typeToTH t) (typeToTH k)
#else
typeToTH (DAppKindT t _) = typeToTH t
#endif
tvbToTH :: DTyVarBndr -> TyVarBndr
tvbToTH (DPlainTV n) = PlainTV n
tvbToTH (DKindedTV n k) = KindedTV n (typeToTH k)
cxtToTH :: DCxt -> Cxt
cxtToTH = map predToTH
#if __GLASGOW_HASKELL__ >= 801
derivClauseToTH :: DDerivClause -> [DerivClause]
derivClauseToTH (DDerivClause mds cxt) =
[DerivClause (fmap derivStrategyToTH mds) (cxtToTH cxt)]
#else
derivClauseToTH :: DDerivClause -> Cxt
derivClauseToTH (DDerivClause _ cxt) = cxtToTH cxt
#endif
#if __GLASGOW_HASKELL__ >= 801
derivStrategyToTH :: DDerivStrategy -> DerivStrategy
derivStrategyToTH DStockStrategy = StockStrategy
derivStrategyToTH DAnyclassStrategy = AnyclassStrategy
derivStrategyToTH DNewtypeStrategy = NewtypeStrategy
#if __GLASGOW_HASKELL__ >= 805
derivStrategyToTH (DViaStrategy ty) = ViaStrategy (typeToTH ty)
#else
derivStrategyToTH (DViaStrategy _) = error "DerivingVia supported only in GHC 8.6+"
#endif
#endif
#if __GLASGOW_HASKELL__ >= 801
patSynDirToTH :: DPatSynDir -> PatSynDir
patSynDirToTH DUnidir = Unidir
patSynDirToTH DImplBidir = ImplBidir
patSynDirToTH (DExplBidir clauses) = ExplBidir (map clauseToTH clauses)
#endif
predToTH :: DPred -> Pred
#if __GLASGOW_HASKELL__ < 709
predToTH = go []
where
go acc (DAppT p t) = go (typeToTH t : acc) p
go acc (DAppKindT t _) = go acc t
go acc (DSigT p _) = go acc p
go acc (DConT n)
| nameBase n == "~"
, [t1, t2] <- acc
= EqualP t1 t2
| otherwise
= ClassP n acc
go _ (DVarT _)
= error "Template Haskell in GHC <= 7.8 does not support variable constraints."
go _ DWildCardT
= error "Wildcards supported only in GHC 8.0+"
go _ (DForallT {})
= error "Quantified constraints supported only in GHC 8.6+"
go _ DArrowT
= error "(->) spotted at head of a constraint"
go _ (DLitT {})
= error "Type-level literal spotted at head of a constraint"
#else
predToTH (DAppT p t) = AppT (predToTH p) (typeToTH t)
predToTH (DSigT p k) = SigT (predToTH p) (typeToTH k)
predToTH (DVarT n) = VarT n
predToTH (DConT n) = typeToTH (DConT n)
predToTH DArrowT = ArrowT
predToTH (DLitT lit) = LitT lit
#if __GLASGOW_HASKELL__ > 710
predToTH DWildCardT = WildCardT
#else
predToTH DWildCardT = error "Wildcards supported only in GHC 8.0+"
#endif
#if __GLASGOW_HASKELL__ >= 805
predToTH (DForallT tvbs cxt p) =
ForallT (map tvbToTH tvbs) (map predToTH cxt) (predToTH p)
#else
predToTH (DForallT {}) = error "Quantified constraints supported only in GHC 8.6+"
#endif
#if __GLASGOW_HASKELL__ >= 807
predToTH (DAppKindT p k) = AppKindT (predToTH p) (typeToTH k)
#else
predToTH (DAppKindT p _) = predToTH p
#endif
#endif
tyconToTH :: Name -> Type
tyconToTH n
| n == ''(->) = ArrowT
| n == ''[] = ListT
#if __GLASGOW_HASKELL__ >= 709
| n == ''(~) = EqualityT
#endif
| n == '[] = PromotedNilT
| n == '(:) = PromotedConsT
| Just deg <- tupleNameDegree_maybe n
= if isDataName n
#if __GLASGOW_HASKELL__ >= 805
then PromotedTupleT deg
#else
then PromotedT n
#endif
else TupleT deg
| Just deg <- unboxedTupleNameDegree_maybe n = UnboxedTupleT deg
#if __GLASGOW_HASKELL__ >= 801
| Just deg <- unboxedSumNameDegree_maybe n = UnboxedSumT deg
#endif
| otherwise = ConT n
typeArgToTH :: DTypeArg -> TypeArg
typeArgToTH (DTANormal t) = TANormal (typeToTH t)
typeArgToTH (DTyArg k) = TyArg (typeToTH k)
#if __GLASGOW_HASKELL__ <= 710
bangToStrict :: Bang -> Strict
bangToStrict (Bang SourceUnpack _) = Unpacked
bangToStrict (Bang _ SourceStrict) = IsStrict
bangToStrict (Bang _ _) = NotStrict
#endif