#include "HsVersions.h"
import qualified GHC
+
import GHC ( TyThing(..), SrcLoc )
+import DataCon ( dataConResTys )
import Outputable
-- -----------------------------------------------------------------------------
pprTyThingHdr exts (AClass cls) = pprClassHdr exts cls
pprTyConHdr exts tyCon =
- ptext keyword <+> ppr_bndr tyCon <+> hsep (map ppr vars)
+ addFamily (ptext keyword) <+> ppr_bndr tyCon <+> hsep (map ppr vars)
where
vars | GHC.isPrimTyCon tyCon ||
GHC.isFunTyCon tyCon = take (GHC.tyConArity tyCon) GHC.alphaTyVars
| GHC.isNewTyCon tyCon = SLIT("newtype")
| otherwise = SLIT("data")
+ addFamily keytext
+ | GHC.isOpenTyCon tyCon = keytext <> ptext SLIT(" family")
+ | otherwise = keytext
+
pprDataConSig exts dataCon =
ppr_bndr dataCon <+> dcolon <+> pprType exts (GHC.dataConType dataCon)
pprTyCon exts tyCon
| GHC.isSynTyCon tyCon
- = let rhs_type = GHC.synTyConRhs tyCon
- in hang (pprTyConHdr exts tyCon <+> equals) 2 (pprType exts rhs_type)
+ = if GHC.isOpenTyCon tyCon
+ then pprTyConHdr exts tyCon <+> dcolon <+>
+ pprType exts (GHC.synTyConResKind tyCon)
+ else
+ let rhs_type = GHC.synTyConType tyCon
+ in hang (pprTyConHdr exts tyCon <+> equals) 2 (pprType exts rhs_type)
| otherwise
= pprAlgTyCon exts tyCon (const True) (const True)
| otherwise = ppr_bndr dataCon <+> dcolon <+>
sep [ ppr_tvs, GHC.pprThetaArrow theta, pp_tau ]
where
- (tyvars, theta, argTypes, tyCon, res_tys) = GHC.dataConSig dataCon
+ (tyvars, theta, argTypes) = GHC.dataConSig dataCon
+ tyCon = GHC.dataConTyCon dataCon
labels = GHC.dataConFieldLabels dataCon
+ res_tys = dataConResTys dataCon
qualVars = filter (flip notElem (GHC.tyConTyVars tyCon)) tyvars
stricts = GHC.dataConStrictMarks dataCon
tys_w_strs = zip stricts argTypes