{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RecordWildCards #-}
module Ormolu.Printer.Meat.Declaration.Signature
( p_sigDecl,
p_typeAscription,
p_activation,
p_standaloneKindSig,
)
where
import Control.Monad
import GHC.Data.BooleanFormula
import GHC.Hs.Binds
import GHC.Hs.Decls
import GHC.Hs.Extension
import GHC.Hs.Type
import GHC.Types.Basic
import GHC.Types.Name.Reader
import GHC.Types.SrcLoc
import Ormolu.Printer.Combinators
import Ormolu.Printer.Meat.Common
import Ormolu.Printer.Meat.Type
import Ormolu.Utils
p_sigDecl :: Sig GhcPs -> R ()
p_sigDecl :: Sig GhcPs -> R ()
p_sigDecl = \case
TypeSig XTypeSig GhcPs
NoExtField [Located (IdP GhcPs)]
names LHsSigWcType GhcPs
hswc -> Bool -> [Located RdrName] -> LHsSigWcType GhcPs -> R ()
p_typeSig Bool
True [Located (IdP GhcPs)]
[Located RdrName]
names LHsSigWcType GhcPs
hswc
PatSynSig XPatSynSig GhcPs
NoExtField [Located (IdP GhcPs)]
names LHsSigType GhcPs
hsib -> [Located RdrName] -> LHsSigType GhcPs -> R ()
p_patSynSig [Located (IdP GhcPs)]
[Located RdrName]
names LHsSigType GhcPs
hsib
ClassOpSig XClassOpSig GhcPs
NoExtField Bool
def [Located (IdP GhcPs)]
names LHsSigType GhcPs
hsib -> Bool -> [Located RdrName] -> LHsSigType GhcPs -> R ()
p_classOpSig Bool
def [Located (IdP GhcPs)]
[Located RdrName]
names LHsSigType GhcPs
hsib
FixSig XFixSig GhcPs
NoExtField FixitySig GhcPs
sig -> FixitySig GhcPs -> R ()
p_fixSig FixitySig GhcPs
sig
InlineSig XInlineSig GhcPs
NoExtField Located (IdP GhcPs)
name InlinePragma
inlinePragma -> Located RdrName -> InlinePragma -> R ()
p_inlineSig Located (IdP GhcPs)
Located RdrName
name InlinePragma
inlinePragma
SpecSig XSpecSig GhcPs
NoExtField Located (IdP GhcPs)
name [LHsSigType GhcPs]
ts InlinePragma
inlinePragma -> Located RdrName -> [LHsSigType GhcPs] -> InlinePragma -> R ()
p_specSig Located (IdP GhcPs)
Located RdrName
name [LHsSigType GhcPs]
ts InlinePragma
inlinePragma
SpecInstSig XSpecInstSig GhcPs
NoExtField SourceText
_ LHsSigType GhcPs
hsib -> LHsSigType GhcPs -> R ()
p_specInstSig LHsSigType GhcPs
hsib
MinimalSig XMinimalSig GhcPs
NoExtField SourceText
_ LBooleanFormula (Located (IdP GhcPs))
booleanFormula -> LBooleanFormula (Located RdrName) -> R ()
p_minimalSig LBooleanFormula (Located (IdP GhcPs))
LBooleanFormula (Located RdrName)
booleanFormula
CompleteMatchSig XCompleteMatchSig GhcPs
NoExtField SourceText
_sourceText Located [Located (IdP GhcPs)]
cs Maybe (Located (IdP GhcPs))
ty -> Located [Located RdrName] -> Maybe (Located RdrName) -> R ()
p_completeSig Located [Located (IdP GhcPs)]
Located [Located RdrName]
cs Maybe (Located (IdP GhcPs))
Maybe (Located RdrName)
ty
SCCFunSig XSCCFunSig GhcPs
NoExtField SourceText
_ Located (IdP GhcPs)
name Maybe (Located StringLiteral)
literal -> Located (IdP GhcPs) -> Maybe (Located StringLiteral) -> R ()
p_sccSig Located (IdP GhcPs)
name Maybe (Located StringLiteral)
literal
Sig GhcPs
_ -> String -> R ()
forall a. String -> a
notImplemented String
"certain types of signature declarations"
p_typeSig ::
Bool ->
[Located RdrName] ->
LHsSigWcType GhcPs ->
R ()
p_typeSig :: Bool -> [Located RdrName] -> LHsSigWcType GhcPs -> R ()
p_typeSig Bool
_ [] LHsSigWcType GhcPs
_ = () -> R ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
p_typeSig Bool
indentTail (Located RdrName
n : [Located RdrName]
ns) LHsSigWcType GhcPs
hswc = do
Located RdrName -> R ()
p_rdrName Located RdrName
n
if [Located RdrName] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [Located RdrName]
ns
then LHsSigWcType GhcPs -> R ()
p_typeAscription LHsSigWcType GhcPs
hswc
else Bool -> R () -> R ()
inciIf Bool
indentTail (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
R ()
commaDel
R () -> (Located RdrName -> R ()) -> [Located RdrName] -> R ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel Located RdrName -> R ()
p_rdrName [Located RdrName]
ns
LHsSigWcType GhcPs -> R ()
p_typeAscription LHsSigWcType GhcPs
hswc
p_typeAscription ::
LHsSigWcType GhcPs ->
R ()
p_typeAscription :: LHsSigWcType GhcPs -> R ()
p_typeAscription HsWC {LHsSigType GhcPs
XHsWC GhcPs (LHsSigType GhcPs)
hswc_ext :: forall pass thing. HsWildCardBndrs pass thing -> XHsWC pass thing
hswc_body :: forall pass thing. HsWildCardBndrs pass thing -> thing
hswc_body :: LHsSigType GhcPs
hswc_ext :: XHsWC GhcPs (LHsSigType GhcPs)
..} = R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
R ()
space
Text -> R ()
txt Text
"::"
let t :: LHsType GhcPs
t = LHsSigType GhcPs -> LHsType GhcPs
forall pass thing. HsImplicitBndrs pass thing -> thing
hsib_body LHsSigType GhcPs
hswc_body
if HsType GhcPs -> Bool
hasDocStrings (LHsType GhcPs -> HsType GhcPs
forall l e. GenLocated l e -> e
unLoc LHsType GhcPs
t)
then R ()
newline
else R ()
breakpoint
LHsType GhcPs -> (HsType GhcPs -> R ()) -> R ()
forall a. Located a -> (a -> R ()) -> R ()
located LHsType GhcPs
t HsType GhcPs -> R ()
p_hsType
p_patSynSig ::
[Located RdrName] ->
HsImplicitBndrs GhcPs (LHsType GhcPs) ->
R ()
p_patSynSig :: [Located RdrName] -> LHsSigType GhcPs -> R ()
p_patSynSig [Located RdrName]
names LHsSigType GhcPs
hsib = do
Text -> R ()
txt Text
"pattern"
let body :: R ()
body =
Bool -> [Located RdrName] -> LHsSigWcType GhcPs -> R ()
p_typeSig
Bool
False
[Located RdrName]
names
HsWC :: forall pass thing.
XHsWC pass thing -> thing -> HsWildCardBndrs pass thing
HsWC {hswc_ext :: XHsWC GhcPs (LHsSigType GhcPs)
hswc_ext = NoExtField
XHsWC GhcPs (LHsSigType GhcPs)
NoExtField, hswc_body :: LHsSigType GhcPs
hswc_body = LHsSigType GhcPs
hsib}
if [Located RdrName] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Located RdrName]
names Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> Int
1
then R ()
breakpoint R () -> R () -> R ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R () -> R ()
inci R ()
body
else R ()
space R () -> R () -> R ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R ()
body
p_classOpSig ::
Bool ->
[Located RdrName] ->
HsImplicitBndrs GhcPs (LHsType GhcPs) ->
R ()
p_classOpSig :: Bool -> [Located RdrName] -> LHsSigType GhcPs -> R ()
p_classOpSig Bool
def [Located RdrName]
names LHsSigType GhcPs
hsib = do
Bool -> R () -> R ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when Bool
def (Text -> R ()
txt Text
"default" R () -> R () -> R ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R ()
space)
Bool -> [Located RdrName] -> LHsSigWcType GhcPs -> R ()
p_typeSig Bool
True [Located RdrName]
names HsWC :: forall pass thing.
XHsWC pass thing -> thing -> HsWildCardBndrs pass thing
HsWC {hswc_ext :: XHsWC GhcPs (LHsSigType GhcPs)
hswc_ext = NoExtField
XHsWC GhcPs (LHsSigType GhcPs)
NoExtField, hswc_body :: LHsSigType GhcPs
hswc_body = LHsSigType GhcPs
hsib}
p_fixSig ::
FixitySig GhcPs ->
R ()
p_fixSig :: FixitySig GhcPs -> R ()
p_fixSig = \case
FixitySig XFixitySig GhcPs
NoExtField [Located (IdP GhcPs)]
names (Fixity SourceText
_ Int
n FixityDirection
dir) -> do
Text -> R ()
txt (Text -> R ()) -> Text -> R ()
forall a b. (a -> b) -> a -> b
$ case FixityDirection
dir of
FixityDirection
InfixL -> Text
"infixl"
FixityDirection
InfixR -> Text
"infixr"
FixityDirection
InfixN -> Text
"infix"
R ()
space
Int -> R ()
forall a. Outputable a => a -> R ()
atom Int
n
R ()
space
R () -> R ()
sitcc (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ R () -> (Located RdrName -> R ()) -> [Located RdrName] -> R ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel Located RdrName -> R ()
p_rdrName [Located (IdP GhcPs)]
[Located RdrName]
names
p_inlineSig ::
Located RdrName ->
InlinePragma ->
R ()
p_inlineSig :: Located RdrName -> InlinePragma -> R ()
p_inlineSig Located RdrName
name InlinePragma {Maybe Int
SourceText
Activation
RuleMatchInfo
InlineSpec
inl_src :: InlinePragma -> SourceText
inl_inline :: InlinePragma -> InlineSpec
inl_sat :: InlinePragma -> Maybe Int
inl_act :: InlinePragma -> Activation
inl_rule :: InlinePragma -> RuleMatchInfo
inl_rule :: RuleMatchInfo
inl_act :: Activation
inl_sat :: Maybe Int
inl_inline :: InlineSpec
inl_src :: SourceText
..} = R () -> R ()
pragmaBraces (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
InlineSpec -> R ()
p_inlineSpec InlineSpec
inl_inline
R ()
space
case RuleMatchInfo
inl_rule of
RuleMatchInfo
ConLike -> Text -> R ()
txt Text
"CONLIKE"
RuleMatchInfo
FunLike -> () -> R ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
R ()
space
Activation -> R ()
p_activation Activation
inl_act
R ()
space
Located RdrName -> R ()
p_rdrName Located RdrName
name
p_specSig ::
Located RdrName ->
[LHsSigType GhcPs] ->
InlinePragma ->
R ()
p_specSig :: Located RdrName -> [LHsSigType GhcPs] -> InlinePragma -> R ()
p_specSig Located RdrName
name [LHsSigType GhcPs]
ts InlinePragma {Maybe Int
SourceText
Activation
RuleMatchInfo
InlineSpec
inl_rule :: RuleMatchInfo
inl_act :: Activation
inl_sat :: Maybe Int
inl_inline :: InlineSpec
inl_src :: SourceText
inl_src :: InlinePragma -> SourceText
inl_inline :: InlinePragma -> InlineSpec
inl_sat :: InlinePragma -> Maybe Int
inl_act :: InlinePragma -> Activation
inl_rule :: InlinePragma -> RuleMatchInfo
..} = R () -> R ()
pragmaBraces (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
Text -> R ()
txt Text
"SPECIALIZE"
R ()
space
InlineSpec -> R ()
p_inlineSpec InlineSpec
inl_inline
R ()
space
Activation -> R ()
p_activation Activation
inl_act
R ()
space
Located RdrName -> R ()
p_rdrName Located RdrName
name
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ R () -> (LHsSigType GhcPs -> R ()) -> [LHsSigType GhcPs] -> R ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel ((HsType GhcPs -> R ()) -> LHsType GhcPs -> R ()
forall a. (a -> R ()) -> Located a -> R ()
located' HsType GhcPs -> R ()
p_hsType (LHsType GhcPs -> R ())
-> (LHsSigType GhcPs -> LHsType GhcPs) -> LHsSigType GhcPs -> R ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsSigType GhcPs -> LHsType GhcPs
forall pass thing. HsImplicitBndrs pass thing -> thing
hsib_body) [LHsSigType GhcPs]
ts
p_inlineSpec :: InlineSpec -> R ()
p_inlineSpec :: InlineSpec -> R ()
p_inlineSpec = \case
InlineSpec
Inline -> Text -> R ()
txt Text
"INLINE"
InlineSpec
Inlinable -> Text -> R ()
txt Text
"INLINEABLE"
InlineSpec
NoInline -> Text -> R ()
txt Text
"NOINLINE"
InlineSpec
NoUserInline -> () -> R ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
p_activation :: Activation -> R ()
p_activation :: Activation -> R ()
p_activation = \case
Activation
NeverActive -> () -> R ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
Activation
AlwaysActive -> () -> R ()
forall (m :: * -> *) a. Monad m => a -> m a
return ()
ActiveBefore SourceText
_ Int
n -> do
Text -> R ()
txt Text
"[~"
Int -> R ()
forall a. Outputable a => a -> R ()
atom Int
n
Text -> R ()
txt Text
"]"
ActiveAfter SourceText
_ Int
n -> do
Text -> R ()
txt Text
"["
Int -> R ()
forall a. Outputable a => a -> R ()
atom Int
n
Text -> R ()
txt Text
"]"
Activation
FinalActive -> String -> R ()
forall a. String -> a
notImplemented String
"FinalActive"
p_specInstSig :: LHsSigType GhcPs -> R ()
p_specInstSig :: LHsSigType GhcPs -> R ()
p_specInstSig LHsSigType GhcPs
hsib =
Text -> R () -> R ()
pragma Text
"SPECIALIZE instance" (R () -> R ()) -> (R () -> R ()) -> R () -> R ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$
LHsType GhcPs -> (HsType GhcPs -> R ()) -> R ()
forall a. Located a -> (a -> R ()) -> R ()
located (LHsSigType GhcPs -> LHsType GhcPs
forall pass thing. HsImplicitBndrs pass thing -> thing
hsib_body LHsSigType GhcPs
hsib) HsType GhcPs -> R ()
p_hsType
p_minimalSig ::
LBooleanFormula (Located RdrName) ->
R ()
p_minimalSig :: LBooleanFormula (Located RdrName) -> R ()
p_minimalSig =
(BooleanFormula (Located RdrName) -> R ())
-> LBooleanFormula (Located RdrName) -> R ()
forall a. (a -> R ()) -> Located a -> R ()
located' ((BooleanFormula (Located RdrName) -> R ())
-> LBooleanFormula (Located RdrName) -> R ())
-> (BooleanFormula (Located RdrName) -> R ())
-> LBooleanFormula (Located RdrName)
-> R ()
forall a b. (a -> b) -> a -> b
$ \BooleanFormula (Located RdrName)
booleanFormula ->
Text -> R () -> R ()
pragma Text
"MINIMAL" (R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ BooleanFormula (Located RdrName) -> R ()
p_booleanFormula BooleanFormula (Located RdrName)
booleanFormula)
p_booleanFormula ::
BooleanFormula (Located RdrName) ->
R ()
p_booleanFormula :: BooleanFormula (Located RdrName) -> R ()
p_booleanFormula = \case
Var Located RdrName
name -> Located RdrName -> R ()
p_rdrName Located RdrName
name
And [LBooleanFormula (Located RdrName)]
xs ->
R () -> R ()
sitcc (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$
R ()
-> (LBooleanFormula (Located RdrName) -> R ())
-> [LBooleanFormula (Located RdrName)]
-> R ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep
R ()
commaDel
((BooleanFormula (Located RdrName) -> R ())
-> LBooleanFormula (Located RdrName) -> R ()
forall a. (a -> R ()) -> Located a -> R ()
located' BooleanFormula (Located RdrName) -> R ()
p_booleanFormula)
[LBooleanFormula (Located RdrName)]
xs
Or [LBooleanFormula (Located RdrName)]
xs ->
R () -> R ()
sitcc (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$
R ()
-> (LBooleanFormula (Located RdrName) -> R ())
-> [LBooleanFormula (Located RdrName)]
-> R ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep
(R ()
breakpoint R () -> R () -> R ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Text -> R ()
txt Text
"|" R () -> R () -> R ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> R ()
space)
((BooleanFormula (Located RdrName) -> R ())
-> LBooleanFormula (Located RdrName) -> R ()
forall a. (a -> R ()) -> Located a -> R ()
located' BooleanFormula (Located RdrName) -> R ()
p_booleanFormula)
[LBooleanFormula (Located RdrName)]
xs
Parens LBooleanFormula (Located RdrName)
l -> LBooleanFormula (Located RdrName)
-> (BooleanFormula (Located RdrName) -> R ()) -> R ()
forall a. Located a -> (a -> R ()) -> R ()
located LBooleanFormula (Located RdrName)
l (BracketStyle -> R () -> R ()
parens BracketStyle
N (R () -> R ())
-> (BooleanFormula (Located RdrName) -> R ())
-> BooleanFormula (Located RdrName)
-> R ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. BooleanFormula (Located RdrName) -> R ()
p_booleanFormula)
p_completeSig ::
Located [Located RdrName] ->
Maybe (Located RdrName) ->
R ()
p_completeSig :: Located [Located RdrName] -> Maybe (Located RdrName) -> R ()
p_completeSig Located [Located RdrName]
cs' Maybe (Located RdrName)
mty =
Located [Located RdrName] -> ([Located RdrName] -> R ()) -> R ()
forall a. Located a -> (a -> R ()) -> R ()
located Located [Located RdrName]
cs' (([Located RdrName] -> R ()) -> R ())
-> ([Located RdrName] -> R ()) -> R ()
forall a b. (a -> b) -> a -> b
$ \[Located RdrName]
cs ->
Text -> R () -> R ()
pragma Text
"COMPLETE" (R () -> R ()) -> (R () -> R ()) -> R () -> R ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
R () -> (Located RdrName -> R ()) -> [Located RdrName] -> R ()
forall a. R () -> (a -> R ()) -> [a] -> R ()
sep R ()
commaDel Located RdrName -> R ()
p_rdrName [Located RdrName]
cs
Maybe (Located RdrName) -> (Located RdrName -> R ()) -> R ()
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
t a -> (a -> m b) -> m ()
forM_ Maybe (Located RdrName)
mty ((Located RdrName -> R ()) -> R ())
-> (Located RdrName -> R ()) -> R ()
forall a b. (a -> b) -> a -> b
$ \Located RdrName
ty -> do
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
R () -> R ()
inci (Located RdrName -> R ()
p_rdrName Located RdrName
ty)
p_sccSig :: Located (IdP GhcPs) -> Maybe (Located StringLiteral) -> R ()
p_sccSig :: Located (IdP GhcPs) -> Maybe (Located StringLiteral) -> R ()
p_sccSig Located (IdP GhcPs)
loc Maybe (Located StringLiteral)
literal = Text -> R () -> R ()
pragma Text
"SCC" (R () -> R ()) -> (R () -> R ()) -> R () -> R ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
Located RdrName -> R ()
p_rdrName Located (IdP GhcPs)
Located RdrName
loc
Maybe (Located StringLiteral)
-> (Located StringLiteral -> R ()) -> R ()
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
t a -> (a -> m b) -> m ()
forM_ Maybe (Located StringLiteral)
literal ((Located StringLiteral -> R ()) -> R ())
-> (Located StringLiteral -> R ()) -> R ()
forall a b. (a -> b) -> a -> b
$ \Located StringLiteral
x -> do
R ()
breakpoint
Located StringLiteral -> R ()
forall a. Outputable a => a -> R ()
atom Located StringLiteral
x
p_standaloneKindSig :: StandaloneKindSig GhcPs -> R ()
p_standaloneKindSig :: StandaloneKindSig GhcPs -> R ()
p_standaloneKindSig (StandaloneKindSig XStandaloneKindSig GhcPs
NoExtField Located (IdP GhcPs)
name (HsIB XHsIB GhcPs (LHsType GhcPs)
NoExtField LHsType GhcPs
sig)) = do
Text -> R ()
txt Text
"type"
R () -> R ()
inci (R () -> R ()) -> R () -> R ()
forall a b. (a -> b) -> a -> b
$ do
R ()
space
Located RdrName -> R ()
p_rdrName Located (IdP GhcPs)
Located RdrName
name
R ()
space
Text -> R ()
txt Text
"::"
R ()
breakpoint
LHsType GhcPs -> (HsType GhcPs -> R ()) -> R ()
forall a. Located a -> (a -> R ()) -> R ()
located LHsType GhcPs
sig HsType GhcPs -> R ()
p_hsType