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

{-# LANGUAGE CPP #-}

module HIndent.Ast.Type.Strictness
  ( Strictness
  , mkStrictness
  ) where

import qualified GHC.Hs as GHC
import HIndent.Ast.NodeComments
import {-# SOURCE #-} HIndent.Pretty
import HIndent.Pretty.Combinators
import HIndent.Pretty.NodeComments

data Strictness
  = Lazy
  | Strict
  deriving (Eq)

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

instance Pretty Strictness where
  pretty' Lazy = string "~"
  pretty' Strict = string "!"

mkStrictness :: GHC.SrcStrictness -> Maybe Strictness
mkStrictness GHC.SrcLazy = Just Lazy
mkStrictness GHC.SrcStrict = Just Strict
mkStrictness GHC.NoSrcStrict = Nothing