{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE TypeFamilies #-}
module Ormolu.Printer.Meat.Type
( p_hsType,
p_hsTypePostDoc,
hasDocStrings,
p_hsContext,
p_hsTyVarBndr,
ForAllVisibility (..),
p_forallBndrs,
p_conDeclFields,
p_lhsTypeArg,
p_hsSigType,
tyVarsToTyPats,
hsOuterTyVarBndrsToHsType,
lhsTypeToSigType,
)
where
import Control.Monad
import Data.Foldable (for_)
import GHC.Hs
import GHC.Types.Basic hiding (isPromoted)
import GHC.Types.SourceText
import GHC.Types.SrcLoc
import GHC.Types.Var
import Ormolu.Config
import Ormolu.Printer.Combinators
import Ormolu.Printer.Meat.Common
import {-# SOURCE #-} Ormolu.Printer.Meat.Declaration.OpTree (p_tyOpTree, tyOpTree)
import {-# SOURCE #-} Ormolu.Printer.Meat.Declaration.Value (p_hsSplice, p_stringLit)
import Ormolu.Printer.Operators
import Ormolu.Utils
p_hsType :: HsType GhcPs -> R ()
p_hsType :: HsType GhcPs -> R ()
p_hsType HsType GhcPs
t = Bool -> TypeDocStyle -> HsType GhcPs -> R ()
p_hsType' (HsType GhcPs -> Bool
hasDocStrings HsType GhcPs
t) TypeDocStyle
PipeStyle HsType GhcPs
t
p_hsTypePostDoc :: HsType GhcPs -> R ()
p_hsTypePostDoc :: HsType GhcPs -> R ()
p_hsTypePostDoc HsType GhcPs
t = Bool -> TypeDocStyle -> HsType GhcPs -> R ()
p_hsType' (HsType GhcPs -> Bool
hasDocStrings HsType GhcPs
t) TypeDocStyle
CaretStyle HsType GhcPs
t
data TypeDocStyle
= PipeStyle
| CaretStyle
p_hsType' :: Bool -> TypeDocStyle -> HsType GhcPs -> R ()
p_hsType' :: Bool -> TypeDocStyle -> HsType GhcPs -> R ()
p_hsType' Bool
multilineArgs TypeDocStyle
docStyle = \case
HsForAllTy XForAllTy GhcPs
_ HsForAllTelescope GhcPs
tele XRec GhcPs (HsType GhcPs)
t -> do
case HsForAllTelescope GhcPs
tele of
HsForAllInvis XHsForAllInvis GhcPs
_ [LHsTyVarBndr Specificity GhcPs]
bndrs -> forall a. ForAllVisibility -> (a -> R ()) -> [LocatedA a] -> R ()
p_forallBndrs ForAllVisibility
ForAllInvis forall flag.
IsInferredTyVarBndr flag =>
HsTyVarBndr flag GhcPs -> R ()
p_hsTyVarBndr [LHsTyVarBndr Specificity GhcPs]
bndrs
HsForAllVis XHsForAllVis GhcPs
_ [LHsTyVarBndr () GhcPs]
bndrs -> forall a. ForAllVisibility -> (a -> R ()) -> [LocatedA a] -> R ()
p_forallBndrs ForAllVisibility
ForAllVis forall flag.
IsInferredTyVarBndr flag =>
HsTyVarBndr flag GhcPs -> R ()
p_hsTyVarBndr [LHsTyVarBndr () GhcPs]
bndrs
R ()
interArgBreak
HsType GhcPs -> R ()
p_hsTypeR (forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
t)
HsQualTy XQualTy GhcPs
_ Maybe (LHsContext GhcPs)
qs' XRec GhcPs (HsType GhcPs)
t -> do
forall (t :: * -> *) (f :: * -> *) a b.
(Foldable t, Applicative f) =>
t a -> (a -> f b) -> f ()
for_ Maybe (LHsContext GhcPs)
qs' forall a b. (a -> b) -> a -> b
$ \GenLocated SrcSpanAnnC [GenLocated SrcSpanAnnA (HsType GhcPs)]
qs -> do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located GenLocated SrcSpanAnnC [GenLocated SrcSpanAnnA (HsType GhcPs)]
qs HsContext GhcPs -> R ()
p_hsContext
R ()
space
Text -> R ()
txt Text
"=>"
R ()
interArgBreak
case forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
t of
HsQualTy {} -> HsType GhcPs -> R ()
p_hsTypeR (forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
t)
HsFunTy {} -> HsType GhcPs -> R ()
p_hsTypeR (forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
t)
HsType GhcPs
_ -> forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsTypeR
HsTyVar XTyVar GhcPs
_ PromotionFlag
p LIdP GhcPs
n -> do
case PromotionFlag
p of
PromotionFlag
IsPromoted -> do
R ()
space
Text -> R ()
txt Text
"'"
case forall o. Outputable o => o -> String
showOutputable (forall l e. GenLocated l e -> e
unLoc LIdP GhcPs
n) of
Char
_ : Char
'\'' : String
_ -> R ()
space
String
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
PromotionFlag
NotPromoted -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
LocatedN RdrName -> R ()
p_rdrName LIdP GhcPs
n
HsAppTy XAppTy GhcPs
_ XRec GhcPs (HsType GhcPs)
f XRec GhcPs (HsType GhcPs)
x -> do
let
gatherArgs :: GenLocated l (HsType pass)
-> [XRec pass (HsType pass)]
-> (GenLocated l (HsType pass), [XRec pass (HsType pass)])
gatherArgs GenLocated l (HsType pass)
f' [XRec pass (HsType pass)]
knownArgs =
case GenLocated l (HsType pass)
f' of
L l
_ (HsAppTy XAppTy pass
_ XRec pass (HsType pass)
l XRec pass (HsType pass)
r) -> GenLocated l (HsType pass)
-> [XRec pass (HsType pass)]
-> (GenLocated l (HsType pass), [XRec pass (HsType pass)])
gatherArgs XRec pass (HsType pass)
l (XRec pass (HsType pass)
r forall a. a -> [a] -> [a]
: [XRec pass (HsType pass)]
knownArgs)
GenLocated l (HsType pass)
_ -> (GenLocated l (HsType pass)
f', [XRec pass (HsType pass)]
knownArgs)
(GenLocated SrcSpanAnnA (HsType GhcPs)
func, HsContext GhcPs
args) = forall {pass} {l}.
(XRec pass (HsType pass) ~ GenLocated l (HsType pass)) =>
GenLocated l (HsType pass)
-> [XRec pass (HsType pass)]
-> (GenLocated l (HsType pass), [XRec pass (HsType pass)])
gatherArgs XRec GhcPs (HsType GhcPs)
f [XRec GhcPs (HsType GhcPs)
x]
[SrcSpan] -> R () -> R ()
switchLayout (forall a e. GenLocated (SrcSpanAnn' a) e -> SrcSpan
getLocA XRec GhcPs (HsType GhcPs)
f forall a. a -> [a] -> [a]
: forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap forall a e. GenLocated (SrcSpanAnn' a) e -> SrcSpan
getLocA HsContext GhcPs
args) forall b c a. (b -> c) -> (a -> b) -> a -> c
. R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located GenLocated SrcSpanAnnA (HsType GhcPs)
func HsType GhcPs -> R ()
p_hsType
R ()
breakpoint
R () -> R ()
inci forall a b. (a -> b) -> a -> b
$
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
breakpoint (forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' HsType GhcPs -> R ()
p_hsType) HsContext GhcPs
args
HsAppKindTy XAppKindTy GhcPs
_ XRec GhcPs (HsType GhcPs)
ty XRec GhcPs (HsType GhcPs)
kd -> R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
ty HsType GhcPs -> R ()
p_hsType
R ()
breakpoint
R () -> R ()
inci forall a b. (a -> b) -> a -> b
$ do
Text -> R ()
txt Text
"@"
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
kd HsType GhcPs -> R ()
p_hsType
HsFunTy XFunTy GhcPs
_ HsArrow GhcPs
arrow XRec GhcPs (HsType GhcPs)
x y :: XRec GhcPs (HsType GhcPs)
y@(L SrcSpanAnnA
_ HsType GhcPs
y') -> do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
x HsType GhcPs -> R ()
p_hsType
R ()
space
case HsArrow GhcPs
arrow of
HsUnrestrictedArrow IsUnicodeSyntax
_ -> Text -> R ()
txt Text
"->"
HsLinearArrow IsUnicodeSyntax
_ Maybe AddEpAnn
_ -> Text -> R ()
txt Text
"%1 ->"
HsExplicitMult IsUnicodeSyntax
_ Maybe AddEpAnn
_ XRec GhcPs (HsType GhcPs)
mult -> do
Text -> R ()
txt Text
"%"
HsType GhcPs -> R ()
p_hsTypeR (forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
mult)
R ()
space
Text -> R ()
txt Text
"->"
R ()
interArgBreak
case HsType GhcPs
y' of
HsFunTy {} -> HsType GhcPs -> R ()
p_hsTypeR HsType GhcPs
y'
HsType GhcPs
_ -> forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
y HsType GhcPs -> R ()
p_hsTypeR
HsListTy XListTy GhcPs
_ XRec GhcPs (HsType GhcPs)
t ->
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t (BracketStyle -> R () -> R ()
brackets BracketStyle
N forall b c a. (b -> c) -> (a -> b) -> a -> c
. HsType GhcPs -> R ()
p_hsType)
HsTupleTy XTupleTy GhcPs
_ HsTupleSort
tsort HsContext GhcPs
xs ->
let parens' :: R () -> R ()
parens' =
case HsTupleSort
tsort of
HsTupleSort
HsUnboxedTuple -> BracketStyle -> R () -> R ()
parensHash BracketStyle
N
HsTupleSort
HsBoxedOrConstraintTuple -> BracketStyle -> R () -> R ()
parens BracketStyle
N
in R () -> R ()
parens' forall a b. (a -> b) -> a -> b
$ forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel (R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' HsType GhcPs -> R ()
p_hsType) HsContext GhcPs
xs
HsSumTy XSumTy GhcPs
_ HsContext GhcPs
xs ->
BracketStyle -> R () -> R ()
parensHash BracketStyle
N forall a b. (a -> b) -> a -> b
$
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep (R ()
space forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Text -> R ()
txt Text
"|" forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R ()
breakpoint) (R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' HsType GhcPs -> R ()
p_hsType) HsContext GhcPs
xs
HsOpTy XOpTy GhcPs
_ XRec GhcPs (HsType GhcPs)
x LIdP GhcPs
op XRec GhcPs (HsType GhcPs)
y -> do
FixityMap
fixityOverrides <- R FixityMap
askFixityOverrides
LazyFixityMap
fixityMap <- R LazyFixityMap
askFixityMap
let opTree :: OpTree (GenLocated SrcSpanAnnA (HsType GhcPs)) (LocatedN RdrName)
opTree = forall ty op. [OpTree ty op] -> [op] -> OpTree ty op
OpBranches [XRec GhcPs (HsType GhcPs)
-> OpTree (XRec GhcPs (HsType GhcPs)) (LocatedN RdrName)
tyOpTree XRec GhcPs (HsType GhcPs)
x, XRec GhcPs (HsType GhcPs)
-> OpTree (XRec GhcPs (HsType GhcPs)) (LocatedN RdrName)
tyOpTree XRec GhcPs (HsType GhcPs)
y] [LIdP GhcPs
op]
OpTree (XRec GhcPs (HsType GhcPs)) (OpInfo (LocatedN RdrName))
-> R ()
p_tyOpTree
(forall op ty.
(op -> Maybe RdrName)
-> FixityMap
-> LazyFixityMap
-> OpTree ty op
-> OpTree ty (OpInfo op)
reassociateOpTree (forall a. a -> Maybe a
Just forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l e. GenLocated l e -> e
unLoc) FixityMap
fixityOverrides LazyFixityMap
fixityMap OpTree (GenLocated SrcSpanAnnA (HsType GhcPs)) (LocatedN RdrName)
opTree)
HsParTy XParTy GhcPs
_ XRec GhcPs (HsType GhcPs)
t ->
BracketStyle -> R () -> R ()
parens BracketStyle
N forall a b. (a -> b) -> a -> b
$ R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsType
HsIParamTy XIParamTy GhcPs
_ XRec GhcPs HsIPName
n XRec GhcPs (HsType GhcPs)
t -> R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs HsIPName
n forall a. Outputable a => a -> R ()
atom
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
R () -> R ()
inci (forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsType)
HsStarTy XStarTy GhcPs
_ Bool
_ -> Text -> R ()
txt Text
"*"
HsKindSig XKindSig GhcPs
_ XRec GhcPs (HsType GhcPs)
t XRec GhcPs (HsType GhcPs)
k -> R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsType
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
R () -> R ()
inci (forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
k HsType GhcPs -> R ()
p_hsType)
HsSpliceTy XSpliceTy GhcPs
_ HsSplice GhcPs
splice -> HsSplice GhcPs -> R ()
p_hsSplice HsSplice GhcPs
splice
HsDocTy XDocTy GhcPs
_ XRec GhcPs (HsType GhcPs)
t LHsDocString
str ->
case TypeDocStyle
docStyle of
TypeDocStyle
PipeStyle -> do
HaddockStyle -> Bool -> LHsDocString -> R ()
p_hsDocString HaddockStyle
Pipe Bool
True LHsDocString
str
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsType
TypeDocStyle
CaretStyle -> do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsType
R ()
newline
HaddockStyle -> Bool -> LHsDocString -> R ()
p_hsDocString HaddockStyle
Caret Bool
False LHsDocString
str
HsBangTy XBangTy GhcPs
_ (HsSrcBang SourceText
_ SrcUnpackedness
u SrcStrictness
s) XRec GhcPs (HsType GhcPs)
t -> do
case SrcUnpackedness
u of
SrcUnpackedness
SrcUnpack -> Text -> R ()
txt Text
"{-# UNPACK #-}" forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R ()
space
SrcUnpackedness
SrcNoUnpack -> Text -> R ()
txt Text
"{-# NOUNPACK #-}" forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R ()
space
SrcUnpackedness
NoSrcUnpack -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
case SrcStrictness
s of
SrcStrictness
SrcLazy -> Text -> R ()
txt Text
"~"
SrcStrictness
SrcStrict -> Text -> R ()
txt Text
"!"
SrcStrictness
NoSrcStrict -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
t HsType GhcPs -> R ()
p_hsType
HsRecTy XRecTy GhcPs
_ [LConDeclField GhcPs]
fields ->
[LConDeclField GhcPs] -> R ()
p_conDeclFields [LConDeclField GhcPs]
fields
HsExplicitListTy XExplicitListTy GhcPs
_ PromotionFlag
p HsContext GhcPs
xs -> do
case PromotionFlag
p of
PromotionFlag
IsPromoted -> Text -> R ()
txt Text
"'"
PromotionFlag
NotPromoted -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
BracketStyle -> R () -> R ()
brackets BracketStyle
N forall a b. (a -> b) -> a -> b
$ do
case (PromotionFlag
p, HsContext GhcPs
xs) of
(PromotionFlag
IsPromoted, L SrcSpanAnnA
_ HsType GhcPs
t : [GenLocated SrcSpanAnnA (HsType GhcPs)]
_) | HsType GhcPs -> Bool
isPromoted HsType GhcPs
t -> R ()
space
(PromotionFlag, [GenLocated SrcSpanAnnA (HsType GhcPs)])
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel (R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' HsType GhcPs -> R ()
p_hsType) HsContext GhcPs
xs
HsExplicitTupleTy XExplicitTupleTy GhcPs
_ HsContext GhcPs
xs -> do
Text -> R ()
txt Text
"'"
BracketStyle -> R () -> R ()
parens BracketStyle
N forall a b. (a -> b) -> a -> b
$ do
case HsContext GhcPs
xs of
L SrcSpanAnnA
_ HsType GhcPs
t : HsContext GhcPs
_ | HsType GhcPs -> Bool
isPromoted HsType GhcPs
t -> R ()
space
HsContext GhcPs
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel (forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' HsType GhcPs -> R ()
p_hsType) HsContext GhcPs
xs
HsTyLit XTyLit GhcPs
_ HsTyLit
t ->
case HsTyLit
t of
HsStrTy (SourceText String
s) FastString
_ -> String -> R ()
p_stringLit String
s
HsTyLit
a -> forall a. Outputable a => a -> R ()
atom HsTyLit
a
HsWildCardTy XWildCardTy GhcPs
_ -> Text -> R ()
txt Text
"_"
XHsType XXType GhcPs
t -> forall a. Outputable a => a -> R ()
atom XXType GhcPs
t
where
isPromoted :: HsType GhcPs -> Bool
isPromoted = \case
HsAppTy XAppTy GhcPs
_ (L SrcSpanAnnA
_ HsType GhcPs
f) XRec GhcPs (HsType GhcPs)
_ -> HsType GhcPs -> Bool
isPromoted HsType GhcPs
f
HsTyVar XTyVar GhcPs
_ PromotionFlag
IsPromoted LIdP GhcPs
_ -> Bool
True
HsExplicitTupleTy {} -> Bool
True
HsExplicitListTy {} -> Bool
True
HsType GhcPs
_ -> Bool
False
interArgBreak :: R ()
interArgBreak =
if Bool
multilineArgs
then R ()
newline
else R ()
breakpoint
p_hsTypeR :: HsType GhcPs -> R ()
p_hsTypeR = Bool -> TypeDocStyle -> HsType GhcPs -> R ()
p_hsType' Bool
multilineArgs TypeDocStyle
docStyle
hasDocStrings :: HsType GhcPs -> Bool
hasDocStrings :: HsType GhcPs -> Bool
hasDocStrings = \case
HsDocTy {} -> Bool
True
HsFunTy XFunTy GhcPs
_ HsArrow GhcPs
_ (L SrcSpanAnnA
_ HsType GhcPs
x) (L SrcSpanAnnA
_ HsType GhcPs
y) -> HsType GhcPs -> Bool
hasDocStrings HsType GhcPs
x Bool -> Bool -> Bool
|| HsType GhcPs -> Bool
hasDocStrings HsType GhcPs
y
HsForAllTy XForAllTy GhcPs
_ HsForAllTelescope GhcPs
_ (L SrcSpanAnnA
_ HsType GhcPs
x) -> HsType GhcPs -> Bool
hasDocStrings HsType GhcPs
x
HsQualTy XQualTy GhcPs
_ Maybe (LHsContext GhcPs)
_ (L SrcSpanAnnA
_ HsType GhcPs
x) -> HsType GhcPs -> Bool
hasDocStrings HsType GhcPs
x
HsType GhcPs
_ -> Bool
False
p_hsContext :: HsContext GhcPs -> R ()
p_hsContext :: HsContext GhcPs -> R ()
p_hsContext = \case
[] -> Text -> R ()
txt Text
"()"
[XRec GhcPs (HsType GhcPs)
x] -> forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
x HsType GhcPs -> R ()
p_hsType
HsContext GhcPs
xs -> BracketStyle -> R () -> R ()
parens BracketStyle
N forall a b. (a -> b) -> a -> b
$ forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel (R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' HsType GhcPs -> R ()
p_hsType) HsContext GhcPs
xs
class IsInferredTyVarBndr flag where
isInferred :: flag -> Bool
instance IsInferredTyVarBndr () where
isInferred :: () -> Bool
isInferred () = Bool
False
instance IsInferredTyVarBndr Specificity where
isInferred :: Specificity -> Bool
isInferred = \case
Specificity
InferredSpec -> Bool
True
Specificity
SpecifiedSpec -> Bool
False
p_hsTyVarBndr :: IsInferredTyVarBndr flag => HsTyVarBndr flag GhcPs -> R ()
p_hsTyVarBndr :: forall flag.
IsInferredTyVarBndr flag =>
HsTyVarBndr flag GhcPs -> R ()
p_hsTyVarBndr = \case
UserTyVar XUserTyVar GhcPs
_ flag
flag LIdP GhcPs
x ->
(if forall flag. IsInferredTyVarBndr flag => flag -> Bool
isInferred flag
flag then BracketStyle -> R () -> R ()
braces BracketStyle
N else forall a. a -> a
id) forall a b. (a -> b) -> a -> b
$ LocatedN RdrName -> R ()
p_rdrName LIdP GhcPs
x
KindedTyVar XKindedTyVar GhcPs
_ flag
flag LIdP GhcPs
l XRec GhcPs (HsType GhcPs)
k -> (if forall flag. IsInferredTyVarBndr flag => flag -> Bool
isInferred flag
flag then BracketStyle -> R () -> R ()
braces else BracketStyle -> R () -> R ()
parens) BracketStyle
N forall b c a. (b -> c) -> (a -> b) -> a -> c
. R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ do
forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located LIdP GhcPs
l forall a. Outputable a => a -> R ()
atom
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
R () -> R ()
inci (forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
k HsType GhcPs -> R ()
p_hsType)
data ForAllVisibility = ForAllInvis | ForAllVis
p_forallBndrs :: ForAllVisibility -> (a -> R ()) -> [LocatedA a] -> R ()
p_forallBndrs :: forall a. ForAllVisibility -> (a -> R ()) -> [LocatedA a] -> R ()
p_forallBndrs ForAllVisibility
ForAllInvis a -> R ()
_ [] = Text -> R ()
txt Text
"forall."
p_forallBndrs ForAllVisibility
ForAllVis a -> R ()
_ [] = Text -> R ()
txt Text
"forall ->"
p_forallBndrs ForAllVisibility
vis a -> R ()
p [LocatedA a]
tyvars =
[SrcSpan] -> R () -> R ()
switchLayout (forall a e. GenLocated (SrcSpanAnn' a) e -> SrcSpan
getLocA forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [LocatedA a]
tyvars) forall a b. (a -> b) -> a -> b
$ do
Text -> R ()
txt Text
"forall"
R ()
breakpoint
R () -> R ()
inci forall a b. (a -> b) -> a -> b
$ do
R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$ forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
breakpoint (R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' a -> R ()
p) [LocatedA a]
tyvars
case ForAllVisibility
vis of
ForAllVisibility
ForAllInvis -> Text -> R ()
txt Text
"."
ForAllVisibility
ForAllVis -> R ()
space forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Text -> R ()
txt Text
"->"
p_conDeclFields :: [LConDeclField GhcPs] -> R ()
p_conDeclFields :: [LConDeclField GhcPs] -> R ()
p_conDeclFields [LConDeclField GhcPs]
xs =
BracketStyle -> R () -> R ()
braces BracketStyle
N forall a b. (a -> b) -> a -> b
$ forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel (R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' ConDeclField GhcPs -> R ()
p_conDeclField) [LConDeclField GhcPs]
xs
p_conDeclField :: ConDeclField GhcPs -> R ()
p_conDeclField :: ConDeclField GhcPs -> R ()
p_conDeclField ConDeclField {[LFieldOcc GhcPs]
Maybe LHsDocString
XRec GhcPs (HsType GhcPs)
XConDeclField GhcPs
cd_fld_ext :: forall pass. ConDeclField pass -> XConDeclField pass
cd_fld_names :: forall pass. ConDeclField pass -> [LFieldOcc pass]
cd_fld_type :: forall pass. ConDeclField pass -> LBangType pass
cd_fld_doc :: forall pass. ConDeclField pass -> Maybe LHsDocString
cd_fld_doc :: Maybe LHsDocString
cd_fld_type :: XRec GhcPs (HsType GhcPs)
cd_fld_names :: [LFieldOcc GhcPs]
cd_fld_ext :: XConDeclField GhcPs
..} = do
CommaStyle
commaStyle <- forall a. (forall (f :: * -> *). PrinterOpts f -> f a) -> R a
getPrinterOpt forall (f :: * -> *). PrinterOpts f -> f CommaStyle
poCommaStyle
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (CommaStyle
commaStyle forall a. Eq a => a -> a -> Bool
== CommaStyle
Trailing) forall a b. (a -> b) -> a -> b
$
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
(a -> m b) -> t a -> m ()
mapM_ (HaddockStyle -> Bool -> LHsDocString -> R ()
p_hsDocString HaddockStyle
Pipe Bool
True) Maybe LHsDocString
cd_fld_doc
R () -> R ()
sitcc forall a b. (a -> b) -> a -> b
$
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep
R ()
commaDel
(forall l a. HasSrcSpan l => (a -> R ()) -> GenLocated l a -> R ()
located' (LocatedN RdrName -> R ()
p_rdrName forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall pass. FieldOcc pass -> LocatedN RdrName
rdrNameFieldOcc))
[LFieldOcc GhcPs]
cd_fld_names
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
R () -> R ()
sitcc forall b c a. (b -> c) -> (a -> b) -> a -> c
. R () -> R ()
inci forall a b. (a -> b) -> a -> b
$ HsType GhcPs -> R ()
p_hsType (forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
cd_fld_type)
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (CommaStyle
commaStyle forall a. Eq a => a -> a -> Bool
== CommaStyle
Leading) forall a b. (a -> b) -> a -> b
$
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
(a -> m b) -> t a -> m ()
mapM_ (Int -> R () -> R ()
inciByFrac (-Int
1) forall b c a. (b -> c) -> (a -> b) -> a -> c
. (R ()
newline forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>) forall b c a. (b -> c) -> (a -> b) -> a -> c
. HaddockStyle -> Bool -> LHsDocString -> R ()
p_hsDocString HaddockStyle
Caret Bool
False) Maybe LHsDocString
cd_fld_doc
p_lhsTypeArg :: LHsTypeArg GhcPs -> R ()
p_lhsTypeArg :: LHsTypeArg GhcPs -> R ()
p_lhsTypeArg = \case
HsValArg XRec GhcPs (HsType GhcPs)
ty -> forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
ty HsType GhcPs -> R ()
p_hsType
HsTypeArg SrcSpan
_ XRec GhcPs (HsType GhcPs)
ty -> Text -> R ()
txt Text
"@" forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> forall l a. HasSrcSpan l => GenLocated l a -> (a -> R ()) -> R ()
located XRec GhcPs (HsType GhcPs)
ty HsType GhcPs -> R ()
p_hsType
HsArgPar SrcSpan
_ -> forall a. String -> a
notImplemented String
"HsArgPar"
p_hsSigType :: HsSigType GhcPs -> R ()
p_hsSigType :: HsSigType GhcPs -> R ()
p_hsSigType HsSig {HsOuterSigTyVarBndrs GhcPs
XRec GhcPs (HsType GhcPs)
XHsSig GhcPs
sig_ext :: forall pass. HsSigType pass -> XHsSig pass
sig_bndrs :: forall pass. HsSigType pass -> HsOuterSigTyVarBndrs pass
sig_body :: forall pass. HsSigType pass -> LHsType pass
sig_body :: XRec GhcPs (HsType GhcPs)
sig_bndrs :: HsOuterSigTyVarBndrs GhcPs
sig_ext :: XHsSig GhcPs
..} =
HsType GhcPs -> R ()
p_hsType forall a b. (a -> b) -> a -> b
$ HsOuterSigTyVarBndrs GhcPs
-> XRec GhcPs (HsType GhcPs) -> HsType GhcPs
hsOuterTyVarBndrsToHsType HsOuterSigTyVarBndrs GhcPs
sig_bndrs XRec GhcPs (HsType GhcPs)
sig_body
tyVarToType :: HsTyVarBndr () GhcPs -> HsType GhcPs
tyVarToType :: HsTyVarBndr () GhcPs -> HsType GhcPs
tyVarToType = \case
UserTyVar XUserTyVar GhcPs
_ () LIdP GhcPs
tvar -> forall pass.
XTyVar pass -> PromotionFlag -> LIdP pass -> HsType pass
HsTyVar forall ann. EpAnn ann
EpAnnNotUsed PromotionFlag
NotPromoted LIdP GhcPs
tvar
KindedTyVar XKindedTyVar GhcPs
_ () LIdP GhcPs
tvar XRec GhcPs (HsType GhcPs)
kind ->
forall pass. XParTy pass -> LHsType pass -> HsType pass
HsParTy forall ann. EpAnn ann
EpAnnNotUsed forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a an. a -> LocatedAn an a
noLocA forall a b. (a -> b) -> a -> b
$
forall pass.
XKindSig pass -> LHsType pass -> LHsType pass -> HsType pass
HsKindSig forall ann. EpAnn ann
EpAnnNotUsed (forall a an. a -> LocatedAn an a
noLocA (forall pass.
XTyVar pass -> PromotionFlag -> LIdP pass -> HsType pass
HsTyVar forall ann. EpAnn ann
EpAnnNotUsed PromotionFlag
NotPromoted LIdP GhcPs
tvar)) XRec GhcPs (HsType GhcPs)
kind
tyVarsToTyPats :: LHsQTyVars GhcPs -> HsTyPats GhcPs
tyVarsToTyPats :: LHsQTyVars GhcPs -> HsTyPats GhcPs
tyVarsToTyPats HsQTvs {[LHsTyVarBndr () GhcPs]
XHsQTvs GhcPs
hsq_ext :: forall pass. LHsQTyVars pass -> XHsQTvs pass
hsq_explicit :: forall pass. LHsQTyVars pass -> [LHsTyVarBndr () pass]
hsq_explicit :: [LHsTyVarBndr () GhcPs]
hsq_ext :: XHsQTvs GhcPs
..} = forall tm ty. tm -> HsArg tm ty
HsValArg forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap HsTyVarBndr () GhcPs -> HsType GhcPs
tyVarToType forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [LHsTyVarBndr () GhcPs]
hsq_explicit
hsOuterTyVarBndrsToHsType ::
HsOuterTyVarBndrs Specificity GhcPs ->
LHsType GhcPs ->
HsType GhcPs
hsOuterTyVarBndrsToHsType :: HsOuterSigTyVarBndrs GhcPs
-> XRec GhcPs (HsType GhcPs) -> HsType GhcPs
hsOuterTyVarBndrsToHsType HsOuterSigTyVarBndrs GhcPs
obndrs XRec GhcPs (HsType GhcPs)
ty = case HsOuterSigTyVarBndrs GhcPs
obndrs of
HsOuterImplicit NoExtField
XHsOuterImplicit GhcPs
NoExtField -> forall l e. GenLocated l e -> e
unLoc XRec GhcPs (HsType GhcPs)
ty
HsOuterExplicit XHsOuterExplicit GhcPs Specificity
_ [LHsTyVarBndr Specificity (NoGhcTc GhcPs)]
bndrs ->
forall pass.
XForAllTy pass
-> HsForAllTelescope pass -> LHsType pass -> HsType pass
HsForAllTy NoExtField
NoExtField (forall (p :: Pass).
EpAnnForallTy
-> [LHsTyVarBndr Specificity (GhcPass p)]
-> HsForAllTelescope (GhcPass p)
mkHsForAllInvisTele forall ann. EpAnn ann
EpAnnNotUsed [LHsTyVarBndr Specificity (NoGhcTc GhcPs)]
bndrs) XRec GhcPs (HsType GhcPs)
ty
lhsTypeToSigType :: LHsType GhcPs -> LHsSigType GhcPs
lhsTypeToSigType :: XRec GhcPs (HsType GhcPs) -> LHsSigType GhcPs
lhsTypeToSigType XRec GhcPs (HsType GhcPs)
ty =
forall e ann. Located e -> LocatedAn ann e
reLocA forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l e. l -> e -> GenLocated l e
L (forall a e. GenLocated (SrcSpanAnn' a) e -> SrcSpan
getLocA XRec GhcPs (HsType GhcPs)
ty) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall pass.
XHsSig pass
-> HsOuterSigTyVarBndrs pass -> LHsType pass -> HsSigType pass
HsSig NoExtField
NoExtField (forall flag pass.
XHsOuterImplicit pass -> HsOuterTyVarBndrs flag pass
HsOuterImplicit NoExtField
NoExtField) forall a b. (a -> b) -> a -> b
$ XRec GhcPs (HsType GhcPs)
ty