{-# LANGUAGE RecordWildCards #-}

module TaskSpec (
  spec,
) where

import Control.Arrow (second)
import Prelude hiding (id)

import Data.Aeson
import qualified Data.Map.Strict as Map
import Data.Time
import Taskwarrior.Annotation
import Taskwarrior.Mask
import Taskwarrior.Priority
import Taskwarrior.RecurringChild
import Taskwarrior.Status
import Taskwarrior.Task
import Test.Hspec
import Test.QuickCheck
import Test.QuickCheck.Instances.Text ()
import Test.QuickCheck.Instances.UUID ()
import Test.QuickCheck.Instances.UnorderedContainers (

 )

prop_taskDeEncode :: Task -> Property
prop_taskDeEncode task = Just task === decode (encode task)

prop_taskReadShow :: Task -> Property
prop_taskReadShow task = task === (read . show $ task)

spec :: Spec
spec = do
  it "will be the same after read . show" $ property prop_taskReadShow
  it "will be the same after encoding to JSON and decoding" $
    property prop_taskDeEncode

instance Arbitrary MaskState where
  arbitrary = arbitraryBoundedEnum

instance Arbitrary Mask where
  arbitrary = Mask <$> (arbitrary :: Gen [MaskState])

instance Arbitrary Status where
  arbitrary =
    oneof
      [ pure Pending
      , Deleted <$> arbitrary
      , Completed <$> arbitrary
      , Recurring <$> arbitrary <*> arbitrary
      ]

instance Arbitrary RecurringChild where
  arbitrary = RecurringChild <$> arbitrary <*> arbitrary <*> arbitrary

instance Arbitrary UTCTime where
  arbitrary = do
    day <- ModifiedJulianDay <$> arbitrary
    dayTime <- secondsToDiffTime <$> choose (0, 86400)
    pure $ UTCTime day dayTime

instance Arbitrary Annotation where
  arbitrary = do
    entry <- arbitrary
    description <- arbitrary
    pure Annotation{..}

instance Arbitrary Priority where
  arbitrary = arbitraryBoundedEnum

instance Arbitrary Task where
  arbitrary = do
    status <- arbitrary
    recurringChild <- case status of
      Recurring{} -> pure Nothing -- A task cannot be both a parent and child recurrence
      _ -> arbitrary
    uuid <- arbitrary
    id <- arbitrary `suchThat` maybe True (>= 1) -- IDs can't be negative, and 0 is used as "not present"
    entry <- arbitrary
    description <- arbitrary
    start <- arbitrary
    wait <- arbitrary
    modified <- arbitrary
    due <- arbitrary
    until_ <- arbitrary
    annotations <- arbitrary
    scheduled <- arbitrary
    project <- arbitrary
    priority <- arbitrary
    depends <- arbitrary
    tags <- arbitrary
    urgency <- arbitrary
    uda <- Map.fromList . fmap (second String) <$> arbitrary
    pure Task{until = until_, ..}