module Generate.Pretty
  ( module Generate.Pretty
  ) where

import Protolude

import GHC.Natural (Natural)
import Numeric (showHex)
import Text.PrettyPrint.Leijen.Text

-------------------------------------------------------------------------------
-- Pretty
-------------------------------------------------------------------------------

prettyBreak :: Doc
prettyBreak
  = linebreak

prettyComment :: Doc -> Doc
prettyComment
  = enclose "-- " dot

prettyDocumentation :: Doc -> Doc
prettyDocumentation
  = enclose "-- | " dot

prettyInline :: Doc -> Doc
prettyInline
  = braces . enclose "-# INLINABLE " " #-"

prettyLine :: Doc
prettyLine
  = pretty (replicate 79 '-')

prettyNatural :: Natural -> Doc
prettyNatural
  = (<>) "0x" . pretty . flip showHex ""

prettySection :: Doc -> Doc
prettySection
  = enclose (prettyLine <$$> "-- ") (prettyBreak <> prettyLine <> prettyBreak)

prettyText :: Text -> Doc
prettyText
  = dquotes . pretty