{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TemplateHaskell #-}

-- | <http://strava.github.io/api/v3/activities/>
module Strive.Types.Activities
  ( ActivityDetailed (..),
    ActivitySummary (..),
    ActivityZoneDetailed (..),
    ActivityZoneDistributionBucket (..),
    ActivityLapSummary (..),
  )
where

import Control.Applicative (empty)
import Data.Aeson (FromJSON, Value (Object), parseJSON, (.:), (.:?))
import Data.Aeson.TH (deriveFromJSON)
import Data.Aeson.Types (Parser, withObject)
import Data.Text (Text)
import Data.Time.Clock (UTCTime)
import Strive.Enums (ActivityType, ActivityZoneType, ResourceState)
import Strive.Internal.TH (options)
import Strive.Types.Athletes (AthleteMeta)
import Strive.Types.Efforts (EffortDetailed)
import Strive.Types.Gear (GearSummary)
import Strive.Types.Polylines (PolylineDetailed, PolylineSummary)

parseLatlng :: Maybe [Double] -> Parser (Maybe (Double, Double))
parseLatlng = \case
  Nothing -> pure Nothing
  Just [] -> pure Nothing
  Just [lat, lng] -> pure $ Just (lat, lng)
  _ -> fail "Invalid array length when parsing a Latlng"

-- | <http://strava.github.io/api/v3/activities/#detailed>
data ActivityDetailed = ActivityDetailed
  { activityDetailed_achievementCount :: Integer,
    activityDetailed_athlete :: AthleteMeta,
    activityDetailed_athleteCount :: Integer,
    activityDetailed_averageSpeed :: Double,
    activityDetailed_averageWatts :: Maybe Double,
    activityDetailed_averageHeartrate :: Maybe Double,
    activityDetailed_calories :: Double,
    activityDetailed_commentCount :: Integer,
    activityDetailed_commute :: Bool,
    activityDetailed_description :: Maybe Text,
    activityDetailed_deviceWatts :: Maybe Bool,
    activityDetailed_distance :: Double,
    activityDetailed_elapsedTime :: Integer,
    activityDetailed_endLatlng :: Maybe (Double, Double),
    activityDetailed_externalId :: Maybe Text,
    activityDetailed_flagged :: Bool,
    activityDetailed_gear :: GearSummary,
    activityDetailed_gearId :: Maybe Text,
    activityDetailed_hasKudoed :: Bool,
    activityDetailed_id :: Integer,
    activityDetailed_instagramPrimaryPhoto :: Maybe Text,
    activityDetailed_kilojoules :: Maybe Double,
    activityDetailed_locationCity :: Maybe Text,
    activityDetailed_locationCountry :: Maybe Text,
    activityDetailed_locationState :: Maybe Text,
    activityDetailed_manual :: Bool,
    activityDetailed_map :: PolylineDetailed,
    activityDetailed_maxHeartrate :: Maybe Double,
    activityDetailed_maxSpeed :: Double,
    activityDetailed_movingTime :: Integer,
    activityDetailed_name :: Text,
    activityDetailed_photoCount :: Integer,
    activityDetailed_private :: Bool,
    activityDetailed_resourceState :: ResourceState,
    activityDetailed_segmentEfforts :: [EffortDetailed],
    activityDetailed_startDate :: UTCTime,
    activityDetailed_startDateLocal :: UTCTime,
    activityDetailed_startLatitude :: Double,
    activityDetailed_startLatlng :: Maybe (Double, Double),
    activityDetailed_startLongitude :: Double,
    activityDetailed_timezone :: Text,
    activityDetailed_totalElevationGain :: Double,
    activityDetailed_trainer :: Bool,
    activityDetailed_truncated :: Integer,
    activityDetailed_type :: ActivityType,
    activityDetailed_uploadId :: Maybe Integer,
    activityDetailed_weightedAverageWatts :: Maybe Integer
  }
  deriving (Show)

instance FromJSON ActivityDetailed where
  parseJSON = withObject "ActivityDetailed" $ \v ->
    ActivityDetailed
      <$> v
        .: "achievement_count"
      <*> v
        .: "athlete"
      <*> v
        .: "athlete_count"
      <*> v
        .: "average_speed"
      <*> v
        .:? "average_watts"
      <*> v
        .:? "average_heartrate"
      <*> v
        .: "calories"
      <*> v
        .: "comment_count"
      <*> v
        .: "commute"
      <*> v
        .:? "description"
      <*> v
        .:? "device_watts"
      <*> v
        .: "distance"
      <*> v
        .: "elapsed_time"
      <*> (v .:? "end_latlng" >>= parseLatlng)
      <*> v
        .:? "external_id"
      <*> v
        .: "flagged"
      <*> v
        .: "gear"
      <*> v
        .:? "gear_id"
      <*> v
        .: "has_kudoed"
      <*> v
        .: "id"
      <*> v
        .:? "instagram_primary_photo"
      <*> v
        .:? "kilojoules"
      <*> v
        .:? "location_city"
      <*> v
        .:? "location_country"
      <*> v
        .:? "location_state"
      <*> v
        .: "manual"
      <*> v
        .: "map"
      <*> v
        .:? "max_heartrate"
      <*> v
        .: "max_speed"
      <*> v
        .: "moving_time"
      <*> v
        .: "name"
      <*> v
        .: "photo_count"
      <*> v
        .: "private"
      <*> v
        .: "resource_state"
      <*> v
        .: "segment_efforts"
      <*> v
        .: "start_date"
      <*> v
        .: "start_date_local"
      <*> v
        .: "start_latitude"
      <*> (v .:? "start_latlng" >>= parseLatlng)
      <*> v
        .: "start_longitude"
      <*> v
        .: "timezone"
      <*> v
        .: "total_elevation_gain"
      <*> v
        .: "trainer"
      <*> v
        .: "truncated"
      <*> v
        .: "type"
      <*> v
        .:? "upload_id"
      <*> v
        .:? "weighted_average_watts"

-- | <http://strava.github.io/api/v3/activities/#summary>
data ActivitySummary = ActivitySummary
  { activitySummary_achievementCount :: Integer,
    activitySummary_athlete :: AthleteMeta,
    activitySummary_athleteCount :: Integer,
    activitySummary_averageSpeed :: Double,
    activitySummary_averageWatts :: Maybe Double,
    activitySummary_averageHeartrate :: Maybe Double,
    activitySummary_commentCount :: Integer,
    activitySummary_commute :: Bool,
    activitySummary_deviceWatts :: Maybe Bool,
    activitySummary_distance :: Double,
    activitySummary_elapsedTime :: Integer,
    activitySummary_endLatlng :: Maybe (Double, Double),
    activitySummary_externalId :: Maybe Text,
    activitySummary_flagged :: Bool,
    activitySummary_gearId :: Maybe Text,
    activitySummary_hasKudoed :: Bool,
    activitySummary_id :: Integer,
    activitySummary_kilojoules :: Maybe Double,
    activitySummary_kudosCount :: Integer,
    activitySummary_locationCity :: Maybe Text,
    activitySummary_locationCountry :: Maybe Text,
    activitySummary_locationState :: Maybe Text,
    activitySummary_manual :: Bool,
    activitySummary_map :: PolylineSummary,
    activitySummary_maxHeartrate :: Maybe Double,
    activitySummary_maxSpeed :: Double,
    activitySummary_movingTime :: Integer,
    activitySummary_name :: Text,
    activitySummary_photoCount :: Integer,
    activitySummary_private :: Bool,
    activitySummary_resourceState :: ResourceState,
    activitySummary_startDate :: UTCTime,
    activitySummary_startDateLocal :: UTCTime,
    activitySummary_startLatitude :: Double,
    activitySummary_startLatlng :: Maybe (Double, Double),
    activitySummary_startLongitude :: Double,
    activitySummary_timezone :: Text,
    activitySummary_totalElevationGain :: Double,
    activitySummary_trainer :: Bool,
    activitySummary_type :: ActivityType,
    activitySummary_uploadId :: Maybe Integer,
    activitySummary_weightedAverageWatts :: Maybe Integer
  }
  deriving (Show)

instance FromJSON ActivitySummary where
  parseJSON = withObject "ActivitySummary" $ \v ->
    ActivitySummary
      <$> v
        .: "achievement_count"
      <*> v
        .: "athlete"
      <*> v
        .: "athlete_count"
      <*> v
        .: "average_speed"
      <*> v
        .:? "average_watts"
      <*> v
        .:? "average_heartrate"
      <*> v
        .: "comment_count"
      <*> v
        .: "commute"
      <*> v
        .:? "device_watts"
      <*> v
        .: "distance"
      <*> v
        .: "elapsed_time"
      <*> (v .:? "end_latlng" >>= parseLatlng)
      <*> v
        .:? "external_id"
      <*> v
        .: "flagged"
      <*> v
        .:? "gear_id"
      <*> v
        .: "has_kudoed"
      <*> v
        .: "id"
      <*> v
        .:? "kilojoules"
      <*> v
        .: "kudos_count"
      <*> v
        .:? "location_city"
      <*> v
        .:? "location_country"
      <*> v
        .:? "location_state"
      <*> v
        .: "manual"
      <*> v
        .: "map"
      <*> v
        .:? "max_heartrate"
      <*> v
        .: "max_speed"
      <*> v
        .: "moving_time"
      <*> v
        .: "name"
      <*> v
        .: "photo_count"
      <*> v
        .: "private"
      <*> v
        .: "resource_state"
      <*> v
        .: "start_date"
      <*> v
        .: "start_date_local"
      <*> v
        .: "start_latitude"
      <*> (v .:? "start_latlng" >>= parseLatlng)
      <*> v
        .: "start_longitude"
      <*> v
        .: "timezone"
      <*> v
        .: "total_elevation_gain"
      <*> v
        .: "trainer"
      <*> v
        .: "type"
      <*> v
        .:? "upload_id"
      <*> v
        .:? "weighted_average_watts"

-- | <http://strava.github.io/api/v3/activities/#zones>
data ActivityZoneDistributionBucket = ActivityZoneDistributionBucket
  { activityZoneDistributionBucket_max :: Integer,
    activityZoneDistributionBucket_min :: Integer,
    activityZoneDistributionBucket_time :: Integer
  }
  deriving (Show)

$(deriveFromJSON options ''ActivityZoneDistributionBucket)

-- | <http://strava.github.io/api/v3/activities/#zones>
data ActivityZoneDetailed = ActivityZoneDetailed
  { activityZoneDetailed_distributionBuckets ::
      [ActivityZoneDistributionBucket],
    activityZoneDetailed_resourceState :: ResourceState,
    activityZoneDetailed_sensorBased :: Bool,
    activityZoneDetailed_type :: ActivityZoneType
  }
  deriving (Show)

$(deriveFromJSON options ''ActivityZoneDetailed)

-- | <http://strava.github.io/api/v3/activities/#laps>
data ActivityLapSummary = ActivityLapSummary
  { activityLapSummary_activityId :: Integer,
    activityLapSummary_athleteId :: Integer,
    activityLapSummary_averageSpeed :: Double,
    activityLapSummary_averageWatts :: Double,
    activityLapSummary_distance :: Double,
    activityLapSummary_elapsedTime :: Integer,
    activityLapSummary_endIndex :: Integer,
    activityLapSummary_id :: Integer,
    activityLapSummary_lapIndex :: Integer,
    activityLapSummary_maxSpeed :: Double,
    activityLapSummary_movingTime :: Double,
    activityLapSummary_name :: Text,
    activityLapSummary_resourceState :: ResourceState,
    activityLapSummary_startDate :: UTCTime,
    activityLapSummary_startDateLocal :: UTCTime,
    activityLapSummary_startIndex :: Integer,
    activityLapSummary_totalElevationGain :: Double
  }
  deriving (Show)

instance FromJSON ActivityLapSummary where
  parseJSON (Object o) =
    ActivityLapSummary
      <$> ((o .: "activity") >>= (.: "id"))
      <*> ((o .: "athlete") >>= (.: "id"))
      <*> o
        .: "average_speed"
      <*> o
        .: "average_watts"
      <*> o
        .: "distance"
      <*> o
        .: "elapsed_time"
      <*> o
        .: "end_index"
      <*> o
        .: "id"
      <*> o
        .: "lap_index"
      <*> o
        .: "max_speed"
      <*> o
        .: "moving_time"
      <*> o
        .: "name"
      <*> o
        .: "resource_state"
      <*> o
        .: "start_date"
      <*> o
        .: "start_date_local"
      <*> o
        .: "start_index"
      <*> o
        .: "total_elevation_gain"
  parseJSON _ = empty