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

module Faker.Provider.Team where

import Config
import Control.Monad.Catch
import Control.Monad.IO.Class
import Data.Map.Strict (Map)
import Data.Monoid ((<>))
import Data.Text (Text)
import Data.Vector (Vector)
import Data.Yaml
import Faker
import Faker.Internal
import Faker.Provider.Address (resolveAddressText, stateProvider, resolveAddressField)
import Faker.Provider.TH
import Language.Haskell.TH


parseTeam :: FromJSON a => FakerSettings -> Value -> Parser a
parseTeam :: forall a. FromJSON a => FakerSettings -> Value -> Parser a
parseTeam FakerSettings
settings (Object Object
obj) = do
  Object
en <- Object
obj Object -> Key -> Parser Object
forall a. FromJSON a => Object -> Key -> Parser a
.: (FakerSettings -> Key
getLocaleKey FakerSettings
settings)
  Object
faker <- Object
en Object -> Key -> Parser Object
forall a. FromJSON a => Object -> Key -> Parser a
.: Key
"faker"
  a
team <- Object
faker Object -> Key -> Parser a
forall a. FromJSON a => Object -> Key -> Parser a
.: Key
"team"
  a -> Parser a
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure a
team
parseTeam FakerSettings
settings Value
val = String -> Parser a
forall a. String -> Parser a
forall (m :: * -> *) a. MonadFail m => String -> m a
fail (String -> Parser a) -> String -> Parser a
forall a b. (a -> b) -> a -> b
$ String
"expected Object, but got " String -> String -> String
forall a. Semigroup a => a -> a -> a
<> (Value -> String
forall a. Show a => a -> String
show Value
val)

parseTeamField ::
     (FromJSON a, Monoid a) => FakerSettings -> AesonKey -> Value -> Parser a
parseTeamField :: forall a.
(FromJSON a, Monoid a) =>
FakerSettings -> Key -> Value -> Parser a
parseTeamField FakerSettings
settings Key
txt Value
val = do
  Object
team <- FakerSettings -> Value -> Parser Object
forall a. FromJSON a => FakerSettings -> Value -> Parser a
parseTeam FakerSettings
settings Value
val
  a
field <- Object
team Object -> Key -> Parser (Maybe a)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
txt Parser (Maybe a) -> a -> Parser a
forall a. Parser (Maybe a) -> a -> Parser a
.!= a
forall a. Monoid a => a
mempty
  a -> Parser a
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure a
field

parseTeamFields ::
     (FromJSON a, Monoid a) => FakerSettings -> [AesonKey] -> Value -> Parser a
parseTeamFields :: forall a.
(FromJSON a, Monoid a) =>
FakerSettings -> [Key] -> Value -> Parser a
parseTeamFields FakerSettings
settings [Key]
txts Value
val = do
  Value
team <- FakerSettings -> Value -> Parser Value
forall a. FromJSON a => FakerSettings -> Value -> Parser a
parseTeam FakerSettings
settings Value
val
  Value -> [Key] -> Parser a
forall a. FromJSON a => Value -> [Key] -> Parser a
helper Value
team [Key]
txts
  where
    helper :: (FromJSON a) => Value -> [AesonKey] -> Parser a
    helper :: forall a. FromJSON a => Value -> [Key] -> Parser a
helper Value
a [] = Value -> Parser a
forall a. FromJSON a => Value -> Parser a
parseJSON Value
a
    helper (Object Object
a) (Key
x:[Key]
xs) = do
      Value
field <- Object
a Object -> Key -> Parser Value
forall a. FromJSON a => Object -> Key -> Parser a
.: Key
x
      Value -> [Key] -> Parser a
forall a. FromJSON a => Value -> [Key] -> Parser a
helper Value
field [Key]
xs
    helper Value
a (Key
x:[Key]
xs) = String -> Parser a
forall a. String -> Parser a
forall (m :: * -> *) a. MonadFail m => String -> m a
fail (String -> Parser a) -> String -> Parser a
forall a b. (a -> b) -> a -> b
$ String
"expect Object, but got " String -> String -> String
forall a. Semigroup a => a -> a -> a
<> (Value -> String
forall a. Show a => a -> String
show Value
a)

parseUnresolvedTeamField ::
     (FromJSON a, Monoid a)
  => FakerSettings
  -> AesonKey
  -> Value
  -> Parser (Unresolved a)
parseUnresolvedTeamField :: forall a.
(FromJSON a, Monoid a) =>
FakerSettings -> Key -> Value -> Parser (Unresolved a)
parseUnresolvedTeamField FakerSettings
settings Key
txt Value
val = do
  Object
team <- FakerSettings -> Value -> Parser Object
forall a. FromJSON a => FakerSettings -> Value -> Parser a
parseTeam FakerSettings
settings Value
val
  a
field <- Object
team Object -> Key -> Parser (Maybe a)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
txt Parser (Maybe a) -> a -> Parser a
forall a. Parser (Maybe a) -> a -> Parser a
.!= a
forall a. Monoid a => a
mempty
  Unresolved a -> Parser (Unresolved a)
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Unresolved a -> Parser (Unresolved a))
-> Unresolved a -> Parser (Unresolved a)
forall a b. (a -> b) -> a -> b
$ a -> Unresolved a
forall a. a -> Unresolved a
forall (f :: * -> *) a. Applicative f => a -> f a
pure a
field

$(genParser "team" "creature")

$(genProvider "team" "creature")

$(genParser "team" "sport")

$(genProvider "team" "sport")

$(genParser "team" "mascot")

$(genProvider "team" "mascot")

$(genParser "team" "main_teams")

$(genProvider "team" "main_teams")

$(genParser "team" "gentile")

$(genProvider "team" "gentile")

$(genParser "team" "prefix")

$(genProvider "team" "prefix")

$(genParser "team" "suffix")

$(genProvider "team" "suffix")

$(genParserUnresolved "team" "name")

$(genProviderUnresolved "team" "name")

-- For es-AR locale
$(genParser "team" "team_suffix")

$(genProvider "team" "team_suffix")

$(genParser "team" "team_prefix")

$(genProvider "team" "team_prefix")

resolveTeamText :: (MonadIO m, MonadThrow m) => FakerSettings -> AesonKey -> m Text
resolveTeamText :: forall (m :: * -> *).
(MonadIO m, MonadThrow m) =>
FakerSettings -> Key -> m Text
resolveTeamText = (FakerSettings -> Key -> m Text) -> FakerSettings -> Key -> m Text
forall (m :: * -> *).
(MonadIO m, MonadThrow m) =>
(FakerSettings -> Key -> m Text) -> FakerSettings -> Key -> m Text
genericResolver' FakerSettings -> Key -> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> Key -> m Text
resolveTeamField

resolveTeamField :: (MonadThrow m, MonadIO m) => FakerSettings -> AesonKey -> m Text
resolveTeamField :: forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> Key -> m Text
resolveTeamField FakerSettings
settings field :: Key
field@Key
"creature" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"team" Key
field FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamCreatureProvider FakerSettings
settings
resolveTeamField FakerSettings
settings Key
"Address.state" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
"state" FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
stateProvider FakerSettings
settings
resolveTeamField FakerSettings
settings Key
"Address.city" =
  FakerSettings -> Key -> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> Key -> m Text
resolveAddressField FakerSettings
settings Key
"city"
resolveTeamField FakerSettings
settings Key
"Team.main_teams" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
"main_teams" FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamMainTeamsProvider FakerSettings
settings
resolveTeamField FakerSettings
settings Key
"Team.prefix" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
"prefix" FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamPrefixProvider FakerSettings
settings
resolveTeamField FakerSettings
settings Key
"suffix" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
"suffix" FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamSuffixProvider FakerSettings
settings
resolveTeamField FakerSettings
settings Key
"Team.gentile" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
"gentile" FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamGentileProvider FakerSettings
settings
resolveTeamField FakerSettings
settings field :: Key
field@Key
"name" =
  Text
-> Key
-> (FakerSettings -> m (Unresolved (Vector Text)))
-> (FakerSettings -> Key -> m Text)
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Unresolved (Vector Text)))
-> (FakerSettings -> Key -> m Text)
-> FakerSettings
-> m Text
cachedRandomUnresolvedVec
    Text
"team"
    Key
field
    FakerSettings -> m (Unresolved (Vector Text))
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Unresolved (Vector Text))
teamNameProvider
    FakerSettings -> Key -> m Text
forall (m :: * -> *).
(MonadIO m, MonadThrow m) =>
FakerSettings -> Key -> m Text
resolveTeamText
    FakerSettings
settings
resolveTeamField FakerSettings
settings field :: Key
field@Key
"team_suffix" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
field FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamTeamSuffixProvider FakerSettings
settings
resolveTeamField FakerSettings
settings field :: Key
field@Key
"team_prefix" =
  Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Vector Text))
-> FakerSettings
-> m Text
cachedRandomVec Text
"address" Key
field FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
teamTeamPrefixProvider FakerSettings
settings
resolveTeamField FakerSettings
settings Key
str = FakerException -> m Text
forall e a. (HasCallStack, Exception e) => e -> m a
forall (m :: * -> *) e a.
(MonadThrow m, HasCallStack, Exception e) =>
e -> m a
throwM (FakerException -> m Text) -> FakerException -> m Text
forall a b. (a -> b) -> a -> b
$ String -> Key -> FakerException
InvalidField String
"team" Key
str