{-# LANGUAGE OverloadedStrings #-}

module Main
  ( main
  ) where

import Prelude hiding (lookup)

import qualified Data.HashMap.Strict as M
import Data.Maybe (fromJust)
import Data.Either (isLeft)
import Data.Text (Text)

import Test.Framework
import Test.Framework.Providers.HUnit
import Test.HUnit hiding (Test)

import Data.ConfigManager
import Data.ConfigManager.Types (Config(..))
import qualified Data.Text as T
import Data.Time.Clock (DiffTime)
import qualified Data.Time.Clock as Time

import Helper (forceGetConfig, getConfig, eitherToMaybe)

main :: IO ()
main = defaultMain tests

tests :: [Test]
tests =
  [ testCase "binding" bindingAssertion
  , testCase "lookupDefault" lookupDefaultAssertion
  , testCase "name" nameAssertion
  , testCase "value" valueAssertion
  , testCase "skip" skipAssertion
  , testCase "import" importAssertion
  , testCase "duration" durationAssertion
  ]

bindingAssertion :: Assertion
bindingAssertion = do
  empty <- forceGetConfig ""
  assertEqual "empty" (M.fromList []) (hashMap empty)

  oneBinding <- forceGetConfig "x = \"foo\""
  assertEqual "one binding present" (Right "foo") (lookup "x" oneBinding)
  assertBool "one binding missing" (isLeft (lookup "y" oneBinding :: Either Text Int))
  assertEqual "one binding count" 1 (M.size . hashMap $ oneBinding)

  multipleBindings <- forceGetConfig $ T.unlines
    [ "x = \"foo\""
    , "y = \"bar\""
    , "z = \"baz\""
    ]
  assertEqual "multiple bindings count" 3 (M.size . hashMap $ multipleBindings)
  assertEqual "multiple bindings last" (Right "baz") (lookup "z" multipleBindings)

  overlappingBindings <- forceGetConfig $ T.unlines
    [ "x = \"foo\""
    , "y = \"bar\""
    , "x = \"baz\""
    ]
  assertEqual "overlapping bindings count" 2 (M.size . hashMap $ overlappingBindings)
  assertEqual "overlapping bindings redefinition" (Right "baz") (lookup "x" overlappingBindings)

lookupDefaultAssertion :: Assertion
lookupDefaultAssertion = do
  config <- forceGetConfig "x = 5"
  assertEqual "x" 5 (lookupDefault 10 "x" config :: Int)
  assertEqual "y" 10 (lookupDefault 10 "y" config :: Int)

nameAssertion :: Assertion
nameAssertion = do
  validNames <- forceGetConfig $ T.unlines
    [ "validIdent = \"foo\" "
    , "valid_ident = \"foo\""
    , "valid-ident = \"foo\""
    ]
  assertEqual "validIdent" (Right "foo") (lookup "validIdent" validNames)
  assertEqual "valid_ident" (Right "foo") (lookup "valid_ident" validNames)
  assertEqual "valid-ident" (Right "foo") (lookup "valid-ident" validNames)

  invalid1 <- getConfig "-invalid_ident = \"foo\""
  assertEqual "-invalid" Nothing invalid1

  invalid2 <- getConfig "_invalid = \"foo\""
  assertEqual "_invalid" Nothing invalid2

valueAssertion :: Assertion
valueAssertion = do
  config <- forceGetConfig $ T.unlines
    [ "a = \"lorem ipsum sir dolor emet\""
    , "b = 4                             "
    , "c = 5.0                           "
    , "d = True                          "
    ]
  assertEqual "string" (Right "lorem ipsum sir dolor emet") (lookup "a" config)
  assertEqual "integer" (Right 4) (lookup "b" config)
  assertEqual "double 1" (Right 4.0) (lookup "b" config)
  assertEqual "double 2" (Right 5.0) (lookup "c" config)
  assertBool "integer fail" (isLeft $ (lookup "c" config :: Either Text Int))
  assertEqual "boolean" (Right True) (lookup "d" config)
  return ()

skipAssertion :: Assertion
skipAssertion = do
  config <- forceGetConfig $ T.unlines
    [ "                             "
    , "  # Comment                  "
    , "  x = \"foo\"                "
    , "                             "
    , "     ####                    "
    , "                             "
    , "  y = \"bar\" # Other comment"
    , "                             "
    ]
  assertEqual "bindings count" 2 (M.size . hashMap $ config)
  assertEqual "bindings x" (Right "foo") (lookup "x" config)
  assertEqual "bindings y" (Right "bar") (lookup "y" config)

importAssertion :: Assertion
importAssertion = do
  config <- fromJust . eitherToMaybe <$> readConfig "tests/resources/a.conf"
  assertEqual "a" (Right "foo") (lookup "a" config)
  assertEqual "b" (Right 15) (lookup "b" config)
  assertEqual "c" (Right "re baz") (lookup "c" config)
  assertEqual "d" (Right "zap") (lookup "d" config)
  assertEqual "e" (Right "re nam") (lookup "e" config)
  assertEqual "f" (Right 8.5) (lookup "f" config)

  missingConfig <- getConfig "import \"required.conf\""
  assertEqual "missing config" Nothing missingConfig

  missingOptionalConfig <- forceGetConfig $ T.unlines
    [ "importMaybe \"required.conf\""
    , "x = 4"
    ]
  assertEqual "missing optional config" (Right 4) (lookup "x" missingOptionalConfig)

durationAssertion :: Assertion
durationAssertion = do
  config <- forceGetConfig $ T.unlines
    [ "a = 1 second"
    , "b = 5 seconds"
    , "c = 1 minute"
    , "d = 10 minutes"
    , "e = 1 hour"
    , "f = 7 hours"
    , "g = 1 day"
    , "h = 2 days"
    , "i = 1 week"
    , "j = 9 weeks"
    , ""
    , "k = 1 minutes"
    , "l = 20  weeks"
    ]

  let second = 1
  let minute = 60 * second
  let hour = 60 * minute
  let day = 24 * hour
  let week = 7 * day

  assertEqual "a" (Right (Time.secondsToDiffTime $ 1 * second)) (lookup "a" config)
  assertEqual "b" (Right (Time.secondsToDiffTime $ 5 * second)) (lookup "b" config)
  assertEqual "c" (Right (Time.secondsToDiffTime $ 1 * minute)) (lookup "c" config)
  assertEqual "d" (Right (Time.secondsToDiffTime $ 10 * minute)) (lookup "d" config)
  assertEqual "e" (Right (Time.secondsToDiffTime $ 1 * hour)) (lookup "e" config)
  assertEqual "f" (Right (Time.secondsToDiffTime $ 7 * hour)) (lookup "f" config)
  assertEqual "g" (Right (Time.secondsToDiffTime $ 1 * day)) (lookup "g" config)
  assertEqual "h" (Right (Time.secondsToDiffTime $ 2 * day)) (lookup "h" config)
  assertEqual "i" (Right (Time.secondsToDiffTime $ 1 * week)) (lookup "i" config)
  assertEqual "j" (Right (Time.secondsToDiffTime $ 9 * week)) (lookup "j" config)
  assertBool "k" (isLeft (lookup "k" config :: Either Text DiffTime))
  assertBool "l" (isLeft (lookup "l" config :: Either Text DiffTime))