skeletest-0.4.0: src/Skeletest/Internal/Utils/Text.hs

{-# LANGUAGE OverloadedStrings #-}

module Skeletest.Internal.Utils.Text (
  showT,
  pluralize,
  parens,

  -- * Indentation
  IndentSize,
  IndentLevel,
  indent,
  indentWith,
) where

import Data.Text (Text)
import Data.Text qualified as Text

-- TODO: Remove when Text.show is available in the version of 'text' that's
-- shipped with the lowest version of GHC we support.
showT :: (Show a) => a -> Text
showT = Text.pack . show

pluralize :: (Num a, Eq a, Show a) => a -> Text -> Text
pluralize n item = showT n <> " " <> item <> (if n == 1 then "" else "s")

-- | Add parentheses if the given input contains spaces.
parens :: Text -> Text
parens s =
  if " " `Text.isInfixOf` s
    then "(" <> s <> ")"
    else s

indent :: Text -> Text
indent = indentWith 2 " " 1

type IndentSize = Int
type IndentLevel = Int

indentWith :: IndentSize -> Text -> IndentLevel -> Text -> Text
indentWith indentSize fill lvl =
  Text.intercalate "\n"
    . map (Text.replicate (lvl * indentSize) fill <>)
    . Text.splitOn "\n"