hindent-6.3.0: src/HIndent/Ast/Type/Bang.hs

{-# LANGUAGE CPP #-}

module HIndent.Ast.Type.Bang
  ( Bang
  , mkBang
  ) where

import Control.Monad
import Data.Maybe
import HIndent.Ast.NodeComments
import HIndent.Ast.Type.Strictness
import HIndent.Ast.Type.Unpackedness
import qualified HIndent.GhcLibParserWrapper.GHC.Hs as GHC
import {-# SOURCE #-} HIndent.Pretty
import HIndent.Pretty.Combinators
import HIndent.Pretty.NodeComments

data Bang =
  Bang (Maybe Unpackedness) (Maybe Strictness)

instance CommentExtraction Bang where
  nodeComments _ = NodeComments [] [] []

instance Pretty Bang where
  pretty' (Bang unpack strictness) = do
    maybe (pure ()) pretty' unpack
    unless (isNothing unpack) space
    maybe (pure ()) pretty' strictness
#if MIN_VERSION_ghc_lib_parser(9, 12, 1)
mkBang :: GHC.HsBang -> Bang
mkBang (GHC.HsBang unpack strictness) =
  Bang (mkUnpackedness unpack) (mkStrictness strictness)
#else
mkBang :: GHC.HsSrcBang -> Bang
mkBang (GHC.HsSrcBang _ unpack strictness) =
  Bang (mkUnpackedness unpack) (mkStrictness strictness)
#endif