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

module Faker.Provider.Book 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.Name (nameNameProvider, resolveNameField, resolveNameText)
import Faker.Provider.TH
import Language.Haskell.TH


parseBook :: FromJSON a => FakerSettings -> Value -> Parser a
parseBook :: FakerSettings -> Value -> Parser a
parseBook 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
book <- Object
faker Object -> Key -> Parser a
forall a. FromJSON a => Object -> Key -> Parser a
.: Key
"book"
  a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure a
book
parseBook FakerSettings
settings Value
val = 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)

parseBookField ::
     (FromJSON a, Monoid a) => FakerSettings -> AesonKey -> Value -> Parser a
parseBookField :: FakerSettings -> Key -> Value -> Parser a
parseBookField FakerSettings
settings Key
txt Value
val = do
  Object
book <- FakerSettings -> Value -> Parser Object
forall a. FromJSON a => FakerSettings -> Value -> Parser a
parseBook FakerSettings
settings Value
val
  a
field <- Object
book 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 (f :: * -> *) a. Applicative f => a -> f a
pure a
field

parseUnresolvedBookField ::
     (FromJSON a, Monoid a)
  => FakerSettings
  -> AesonKey
  -> Value
  -> Parser (Unresolved a)
parseUnresolvedBookField :: FakerSettings -> Key -> Value -> Parser (Unresolved a)
parseUnresolvedBookField FakerSettings
settings Key
txt Value
val = do
  Object
book <- FakerSettings -> Value -> Parser Object
forall a. FromJSON a => FakerSettings -> Value -> Parser a
parseBook FakerSettings
settings Value
val
  a
field <- Object
book 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 (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 (f :: * -> *) a. Applicative f => a -> f a
pure a
field

$(genParser "book" "title")

$(genProvider "book" "title")

$(genParserUnresolved "book" "author")

bookAuthorProvider ::
     (MonadThrow m, MonadIO m) => FakerSettings -> m (Unresolved Text)
bookAuthorProvider :: FakerSettings -> m (Unresolved Text)
bookAuthorProvider FakerSettings
settings = FakerSettings
-> SourceData
-> (FakerSettings -> Value -> Parser (Unresolved Text))
-> m (Unresolved Text)
forall (m :: * -> *) a.
(MonadThrow m, MonadIO m) =>
FakerSettings
-> SourceData -> (FakerSettings -> Value -> Parser a) -> m a
fetchData FakerSettings
settings SourceData
Book FakerSettings -> Value -> Parser (Unresolved Text)
forall a.
(FromJSON a, Monoid a) =>
FakerSettings -> Value -> Parser (Unresolved a)
parseBookAuthorUnresolved

parseAuthorFieldHyLocale ::
     (FromJSON a, Monoid a) => FakerSettings -> Value -> Parser a
parseAuthorFieldHyLocale :: FakerSettings -> Value -> Parser a
parseAuthorFieldHyLocale FakerSettings
settings = FakerSettings -> Key -> Value -> Parser a
forall a.
(FromJSON a, Monoid a) =>
FakerSettings -> Key -> Value -> Parser a
parseBookField FakerSettings
settings Key
"author"

authorProviderHyLocale ::
     (MonadThrow m, MonadIO m) => FakerSettings -> m (Vector Text)
authorProviderHyLocale :: FakerSettings -> m (Vector Text)
authorProviderHyLocale FakerSettings
settings =
  FakerSettings
-> SourceData
-> (FakerSettings -> Value -> Parser (Vector Text))
-> m (Vector Text)
forall (m :: * -> *) a.
(MonadThrow m, MonadIO m) =>
FakerSettings
-> SourceData -> (FakerSettings -> Value -> Parser a) -> m a
fetchData FakerSettings
settings SourceData
Address FakerSettings -> Value -> Parser (Vector Text)
forall a.
(FromJSON a, Monoid a) =>
FakerSettings -> Value -> Parser a
parseAuthorFieldHyLocale

$(genParser "book" "publisher")

$(genProvider "book" "publisher")

$(genParser "book" "genre")

$(genProvider "book" "genre")

authorResolver :: (MonadIO m, MonadThrow m) => FakerSettings -> m Text
authorResolver :: FakerSettings -> m Text
authorResolver FakerSettings
settings =
  case (FakerSettings -> Text
getLocale FakerSettings
settings) Text -> [Text] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [Text
"hy"] of
    Bool
True -> (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
"book" Key
"author" FakerSettings -> m (Vector Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Vector Text)
authorProviderHyLocale FakerSettings
settings)
    Bool
False ->
      (Text
-> Key
-> (FakerSettings -> m (Unresolved Text))
-> (FakerSettings -> Key -> m Text)
-> FakerSettings
-> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
Text
-> Key
-> (FakerSettings -> m (Unresolved Text))
-> (FakerSettings -> Key -> m Text)
-> FakerSettings
-> m Text
cachedRandomUnresolvedVecWithoutVector
         Text
"book"
         Key
"author"
         FakerSettings -> m (Unresolved Text)
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Unresolved Text)
bookAuthorProvider
         FakerSettings -> Key -> m Text
forall (m :: * -> *).
(MonadIO m, MonadThrow m) =>
FakerSettings -> Key -> m Text
resolveBookText
         FakerSettings
settings)

resolveBookText :: (MonadIO m, MonadThrow m) => FakerSettings -> AesonKey -> m Text
resolveBookText :: FakerSettings -> Key -> m Text
resolveBookText = (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
resolveBookField

resolveBookField :: (MonadThrow m, MonadIO m) => FakerSettings -> AesonKey -> m Text
resolveBookField :: FakerSettings -> Key -> m Text
resolveBookField FakerSettings
settings Key
"Name.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
"name"
    Key
"name"
    FakerSettings -> m (Unresolved (Vector Text))
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> m (Unresolved (Vector Text))
nameNameProvider
    FakerSettings -> Key -> m Text
forall (m :: * -> *).
(MonadThrow m, MonadIO m) =>
FakerSettings -> Key -> m Text
resolveNameText
    FakerSettings
settings
resolveBookField FakerSettings
settings Key
str = FakerException -> m Text
forall (m :: * -> *) e a. (MonadThrow m, Exception e) => e -> m a
throwM (FakerException -> m Text) -> FakerException -> m Text
forall a b. (a -> b) -> a -> b
$ String -> Key -> FakerException
InvalidField String
"book" Key
str