-- | 'Strive.Actions.Segments'
module Strive.Options.Segments
  ( GetStarredSegmentsOptions,
    GetSegmentEffortsOptions (..),
    GetSegmentLeaderboardOptions (..),
    ExploreSegmentsOptions (..),
  )
where

import Data.Aeson (encode)
import Data.ByteString.Char8 (unpack)
import Data.ByteString.Lazy (toStrict)
import Data.Default (Default, def)
import Data.Time.Clock (UTCTime)
import Network.HTTP.Types (QueryLike, toQuery)
import Strive.Enums
  ( AgeGroup,
    Gender,
    SegmentActivityType (Riding),
    WeightClass,
  )
import Strive.Internal.Options (PaginationOptions)

-- | 'Strive.Actions.getStarredSegments'
type GetStarredSegmentsOptions = PaginationOptions

-- | 'Strive.Actions.getSegmentEfforts'
data GetSegmentEffortsOptions = GetSegmentEffortsOptions
  { getSegmentEffortsOptions_athleteId :: Maybe Integer,
    getSegmentEffortsOptions_range :: Maybe (UTCTime, UTCTime),
    getSegmentEffortsOptions_page :: Integer,
    getSegmentEffortsOptions_perPage :: Integer
  }
  deriving (Show)

instance Default GetSegmentEffortsOptions where
  def =
    GetSegmentEffortsOptions
      { getSegmentEffortsOptions_athleteId = Nothing,
        getSegmentEffortsOptions_range = Nothing,
        getSegmentEffortsOptions_page = 1,
        getSegmentEffortsOptions_perPage = 200
      }

instance QueryLike GetSegmentEffortsOptions where
  toQuery options =
    toQuery
      [ ("athlete_id", fmap show (getSegmentEffortsOptions_athleteId options)),
        ( "start_date_local",
          fmap
            (unpack . toStrict . encode . fst)
            (getSegmentEffortsOptions_range options)
        ),
        ( "end_date_local",
          fmap
            (unpack . toStrict . encode . snd)
            (getSegmentEffortsOptions_range options)
        ),
        ("page", Just (show (getSegmentEffortsOptions_page options))),
        ("per_page", Just (show (getSegmentEffortsOptions_perPage options)))
      ]

-- | 'Strive.Actions.getSegmentLeaderboard'
data GetSegmentLeaderboardOptions = GetSegmentLeaderboardOptions
  { getSegmentLeaderboardOptions_gender :: Maybe Gender,
    getSegmentLeaderboardOptions_ageGroup :: Maybe AgeGroup,
    getSegmentLeaderboardOptions_weightClass :: Maybe WeightClass,
    getSegmentLeaderboardOptions_following :: Maybe Bool,
    getSegmentLeaderboardOptions_clubId :: Maybe Integer,
    getSegmentLeaderboardOptions_dateRange :: Maybe String,
    getSegmentLeaderboardOptions_contextEntries :: Maybe Integer,
    getSegmentLeaderboardOptions_page :: Integer,
    getSegmentLeaderboardOptions_perPage :: Integer
  }
  deriving (Show)

instance Default GetSegmentLeaderboardOptions where
  def =
    GetSegmentLeaderboardOptions
      { getSegmentLeaderboardOptions_gender = Nothing,
        getSegmentLeaderboardOptions_ageGroup = Nothing,
        getSegmentLeaderboardOptions_weightClass = Nothing,
        getSegmentLeaderboardOptions_following = Nothing,
        getSegmentLeaderboardOptions_clubId = Nothing,
        getSegmentLeaderboardOptions_dateRange = Nothing,
        getSegmentLeaderboardOptions_contextEntries = Nothing,
        getSegmentLeaderboardOptions_page = 1,
        getSegmentLeaderboardOptions_perPage = 200
      }

instance QueryLike GetSegmentLeaderboardOptions where
  toQuery options =
    toQuery
      [ ("gender", fmap show (getSegmentLeaderboardOptions_gender options)),
        ("age_group", fmap show (getSegmentLeaderboardOptions_ageGroup options)),
        ( "weight_class",
          fmap show (getSegmentLeaderboardOptions_weightClass options)
        ),
        ( "following",
          fmap
            (unpack . toStrict . encode)
            (getSegmentLeaderboardOptions_following options)
        ),
        ("club_id", fmap show (getSegmentLeaderboardOptions_clubId options)),
        ("date_range", getSegmentLeaderboardOptions_dateRange options),
        ( "context_entries",
          fmap show (getSegmentLeaderboardOptions_contextEntries options)
        ),
        ("page", Just (show (getSegmentLeaderboardOptions_page options))),
        ("per_page", Just (show (getSegmentLeaderboardOptions_perPage options)))
      ]

-- | 'Strive.Actions.exploreSegments'
data ExploreSegmentsOptions = ExploreSegmentsOptions
  { exploreSegmentsOptions_activityType :: SegmentActivityType,
    exploreSegmentsOptions_minCat :: Integer,
    exploreSegmentsOptions_maxCat :: Integer
  }
  deriving (Show)

instance Default ExploreSegmentsOptions where
  def =
    ExploreSegmentsOptions
      { exploreSegmentsOptions_activityType = Riding,
        exploreSegmentsOptions_minCat = 0,
        exploreSegmentsOptions_maxCat = 5
      }

instance QueryLike ExploreSegmentsOptions where
  toQuery options =
    toQuery
      [ ("activity_type", show (exploreSegmentsOptions_activityType options)),
        ("min_cat", show (exploreSegmentsOptions_minCat options)),
        ("max_cat", show (exploreSegmentsOptions_maxCat options))
      ]