{-# LANGUAGE OverloadedStrings #-}

module Test.Tasty.AutoCollect.Utils.Text (
  withoutPrefix,
  withoutSuffix,
  breakOnEnd,
  listify,
  quoted,
) where

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

withoutPrefix :: Text -> Text -> Text
withoutPrefix pre s = fromMaybe s $ Text.stripPrefix pre s

withoutSuffix :: Text -> Text -> Text
withoutSuffix post s = fromMaybe s $ Text.stripSuffix post s

-- | Same as 'Text.breakOnEnd', but omits the delimiter
breakOnEnd :: Text -> Text -> (Text, Text)
breakOnEnd delim s =
  let (a, b) = Text.breakOnEnd delim s
   in (fromMaybe a (Text.stripSuffix delim a), b)

-- | Convert a list @["a", "b"]@ to the text @"[\"a\", \"b\"]"@.
listify :: [Text] -> Text
listify xs = "[" <> Text.intercalate ", " xs <> "]"

quoted :: Text -> Text
quoted s = "\"" <> s <> "\""