module Language.Haskell.Tools.Rewrite.Create.Binds where
import Data.String (IsString(..))
import Language.Haskell.Tools.AST
import Language.Haskell.Tools.PrettyPrint.Prepare
import Language.Haskell.Tools.Rewrite.Create.Patterns (mkVarPat)
import Language.Haskell.Tools.Rewrite.Create.Utils (mkAnn, mkAnnList, mkAnnMaybe)
import Language.Haskell.Tools.Rewrite.ElementTypes
mkSimpleBind' :: Name -> Expr -> ValueBind
mkSimpleBind' n e = mkSimpleBind (mkVarPat n) (mkUnguardedRhs e) Nothing
mkSimpleBind :: Pattern -> Rhs -> Maybe LocalBinds -> ValueBind
mkSimpleBind p r l = mkAnn (child <> child <> child) (USimpleBind p r (mkAnnMaybe opt l))
mkFunctionBind :: [Match] -> ValueBind
mkFunctionBind = mkAnn child . UFunBind . mkAnnList (indented list)
mkFunctionBind' :: Name -> [([Pattern], Expr)] -> ValueBind
mkFunctionBind' name matches = mkFunctionBind $ map (\(args, rhs) -> mkMatch (mkMatchLhs name args) (mkUnguardedRhs rhs) Nothing) matches
mkMatch :: MatchLhs -> Rhs -> Maybe LocalBinds -> Match
mkMatch lhs rhs locs
= mkAnn (child <> child <> child)
$ UMatch lhs rhs (mkAnnMaybe (after " " opt) locs)
mkMatchLhs :: Name -> [Pattern] -> MatchLhs
mkMatchLhs n pats = mkAnn (child <> child) $ UNormalLhs n (mkAnnList (after " " $ separatedBy " " list) pats)
mkInfixLhs :: Pattern -> Operator -> Pattern -> [Pattern] -> MatchLhs
mkInfixLhs lhs op rhs pats
= mkAnn (child <> child <> child <> child) $ UInfixLhs lhs op rhs (mkAnnList (after " " $ separatedBy " " list) pats)
mkLocalBinds :: [LocalBind] -> MaybeLocalBinds
mkLocalBinds = mkAnnMaybe (relativeIndented 2 $ after "\nwhere " opt)
. Just . mkAnn child . ULocalBinds . mkAnnList (indented list)
mkLocalBinds' :: [LocalBind] -> LocalBinds
mkLocalBinds' = mkAnn (" where " <> child) . ULocalBinds . mkAnnList (indented list)
mkLocalValBind :: ValueBind -> LocalBind
mkLocalValBind = mkAnn child . ULocalValBind
mkLocalTypeSig :: TypeSignature -> LocalBind
mkLocalTypeSig = mkAnn child . ULocalSignature
mkLocalFixity :: FixitySignature -> LocalBind
mkLocalFixity = mkAnn child . ULocalFixity
mkTypeSignature :: Name -> Type -> TypeSignature
mkTypeSignature n t = mkAnn (child <> " :: " <> child) (UTypeSignature (mkAnnList (separatedBy ", " list) [n]) t)
mkInfixL :: Int -> Operator -> FixitySignature
mkInfixL prec op = mkAnn (child <> " " <> child <> " " <> child)
$ UFixitySignature (mkAnn "infixl" AssocLeft) (mkAnnMaybe opt $ Just $ mkAnn (fromString (show prec)) (Precedence prec)) (mkAnnList (separatedBy ", " list) [op])
mkInfixR :: Int -> Operator -> FixitySignature
mkInfixR prec op = mkAnn (child <> " " <> child <> " " <> child)
$ UFixitySignature (mkAnn "infixr" AssocRight) (mkAnnMaybe opt $ Just $ mkAnn (fromString (show prec)) (Precedence prec)) (mkAnnList (separatedBy ", " list) [op])
mkInfix :: Int -> Operator -> FixitySignature
mkInfix prec op = mkAnn (child <> " " <> child <> " " <> child)
$ UFixitySignature (mkAnn "infix" AssocNone) (mkAnnMaybe opt $ Just $ mkAnn (fromString (show prec)) (Precedence prec)) (mkAnnList (separatedBy ", " list) [op])
mkUnguardedRhs :: Expr -> Rhs
mkUnguardedRhs = mkAnn (" = " <> child) . UUnguardedRhs
mkGuardedRhss :: [GuardedRhs] -> Rhs
mkGuardedRhss = mkAnn child . UGuardedRhss . mkAnnList (indented list)
mkGuardedRhs :: [RhsGuard] -> Expr -> GuardedRhs
mkGuardedRhs guards expr = mkAnn ("| " <> child <> " = " <> child) $ UGuardedRhs (mkAnnList (separatedBy ", " list) guards) expr
mkGuardBind :: Pattern -> Expr -> RhsGuard
mkGuardBind pat expr = mkAnn (child <> " <- " <> child) $ UGuardBind pat expr
mkGuardLet :: [LocalBind] -> RhsGuard
mkGuardLet = mkAnn ("let " <> child) . UGuardLet . mkAnnList (indented list)
mkGuardCheck :: Expr -> RhsGuard
mkGuardCheck = mkAnn child . UGuardCheck