module Test.Hspec.JUnit.Config
  ( JUnitConfig

  -- * Construction
  , defaultJUnitConfig
  , setJUnitConfigOutputDirectory
  , setJUnitConfigOutputName
  , setJUnitConfigOutputFile

  -- * Use
  , getJUnitConfigOutputFile
  , getJUnitConfigSuiteName
  ) where

import Prelude

import Data.Maybe (fromMaybe)
import Data.Text (Text)
import System.FilePath ((</>))

data JUnitConfig = JUnitConfig
  { JUnitConfig -> FilePath
junitConfigOutputDirectory :: FilePath
  , JUnitConfig -> FilePath
junitConfigOutputName :: FilePath
  , JUnitConfig -> Maybe FilePath
junitConfigOutputFile :: Maybe FilePath
  , JUnitConfig -> Text
junitConfigSuiteName :: Text
  }

-- | Construct a 'JUnitConfig' given a suite name
--
-- See individual set functions for defaults.
--
defaultJUnitConfig :: Text -> JUnitConfig
defaultJUnitConfig :: Text -> JUnitConfig
defaultJUnitConfig Text
name = JUnitConfig :: FilePath -> FilePath -> Maybe FilePath -> Text -> JUnitConfig
JUnitConfig
  { junitConfigOutputDirectory :: FilePath
junitConfigOutputDirectory = FilePath
"."
  , junitConfigOutputName :: FilePath
junitConfigOutputName = FilePath
"junit.xml"
  , junitConfigOutputFile :: Maybe FilePath
junitConfigOutputFile = Maybe FilePath
forall a. Maybe a
Nothing
  , junitConfigSuiteName :: Text
junitConfigSuiteName = Text
name
  }

-- | Set the directory within which to generate the report
--
-- Default is current working directory.
--
setJUnitConfigOutputDirectory :: FilePath -> JUnitConfig -> JUnitConfig
setJUnitConfigOutputDirectory :: FilePath -> JUnitConfig -> JUnitConfig
setJUnitConfigOutputDirectory FilePath
x JUnitConfig
config =
  JUnitConfig
config { junitConfigOutputDirectory :: FilePath
junitConfigOutputDirectory = FilePath
x }

-- | Set the name for the generated report
--
-- Default is @junit.xml@.
--
setJUnitConfigOutputName :: FilePath -> JUnitConfig -> JUnitConfig
setJUnitConfigOutputName :: FilePath -> JUnitConfig -> JUnitConfig
setJUnitConfigOutputName FilePath
x JUnitConfig
config = JUnitConfig
config { junitConfigOutputName :: FilePath
junitConfigOutputName = FilePath
x }

-- | Set the full path to the generated report
--
-- If given, the directory and name configurations are ignored.
--
setJUnitConfigOutputFile :: FilePath -> JUnitConfig -> JUnitConfig
setJUnitConfigOutputFile :: FilePath -> JUnitConfig -> JUnitConfig
setJUnitConfigOutputFile FilePath
x JUnitConfig
config = JUnitConfig
config { junitConfigOutputFile :: Maybe FilePath
junitConfigOutputFile = FilePath -> Maybe FilePath
forall a. a -> Maybe a
Just FilePath
x }

-- | Retrieve the full path to the generated report
getJUnitConfigOutputFile :: JUnitConfig -> FilePath
getJUnitConfigOutputFile :: JUnitConfig -> FilePath
getJUnitConfigOutputFile JUnitConfig {FilePath
Maybe FilePath
Text
junitConfigSuiteName :: Text
junitConfigOutputFile :: Maybe FilePath
junitConfigOutputName :: FilePath
junitConfigOutputDirectory :: FilePath
junitConfigSuiteName :: JUnitConfig -> Text
junitConfigOutputFile :: JUnitConfig -> Maybe FilePath
junitConfigOutputName :: JUnitConfig -> FilePath
junitConfigOutputDirectory :: JUnitConfig -> FilePath
..} = FilePath -> Maybe FilePath -> FilePath
forall a. a -> Maybe a -> a
fromMaybe
  (FilePath
junitConfigOutputDirectory FilePath -> FilePath -> FilePath
</> FilePath
junitConfigOutputName)
  Maybe FilePath
junitConfigOutputFile

-- | Retrieve the suite name given on construction
getJUnitConfigSuiteName :: JUnitConfig -> Text
getJUnitConfigSuiteName :: JUnitConfig -> Text
getJUnitConfigSuiteName = JUnitConfig -> Text
junitConfigSuiteName