{-# LANGUAGE OverloadedStrings #-}

module Database.Bloodhound.Internal.Versions.Common.Types.Query.MoreLikeThisField
  ( MoreLikeThisFieldQuery (..),
  )
where

import Database.Bloodhound.Internal.Utils.Imports
import Database.Bloodhound.Internal.Versions.Common.Types.Newtypes
import Database.Bloodhound.Internal.Versions.Common.Types.Query.Commons
import GHC.Generics

data MoreLikeThisFieldQuery = MoreLikeThisFieldQuery
  { MoreLikeThisFieldQuery -> Text
moreLikeThisFieldText :: Text,
    MoreLikeThisFieldQuery -> FieldName
moreLikeThisFieldFields :: FieldName,
    -- default 0.3 (30%)
    MoreLikeThisFieldQuery -> Maybe PercentMatch
moreLikeThisFieldPercentMatch :: Maybe PercentMatch,
    MoreLikeThisFieldQuery -> Maybe MinimumTermFrequency
moreLikeThisFieldMinimumTermFreq :: Maybe MinimumTermFrequency,
    MoreLikeThisFieldQuery -> Maybe MaxQueryTerms
moreLikeThisFieldMaxQueryTerms :: Maybe MaxQueryTerms,
    MoreLikeThisFieldQuery -> Maybe (NonEmpty StopWord)
moreLikeThisFieldStopWords :: Maybe (NonEmpty StopWord),
    MoreLikeThisFieldQuery -> Maybe MinDocFrequency
moreLikeThisFieldMinDocFrequency :: Maybe MinDocFrequency,
    MoreLikeThisFieldQuery -> Maybe MaxDocFrequency
moreLikeThisFieldMaxDocFrequency :: Maybe MaxDocFrequency,
    MoreLikeThisFieldQuery -> Maybe MinWordLength
moreLikeThisFieldMinWordLength :: Maybe MinWordLength,
    MoreLikeThisFieldQuery -> Maybe MaxWordLength
moreLikeThisFieldMaxWordLength :: Maybe MaxWordLength,
    MoreLikeThisFieldQuery -> Maybe BoostTerms
moreLikeThisFieldBoostTerms :: Maybe BoostTerms,
    MoreLikeThisFieldQuery -> Maybe Boost
moreLikeThisFieldBoost :: Maybe Boost,
    MoreLikeThisFieldQuery -> Maybe Analyzer
moreLikeThisFieldAnalyzer :: Maybe Analyzer
  }
  deriving stock (MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool
(MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool)
-> (MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool)
-> Eq MoreLikeThisFieldQuery
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool
== :: MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool
$c/= :: MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool
/= :: MoreLikeThisFieldQuery -> MoreLikeThisFieldQuery -> Bool
Eq, Int -> MoreLikeThisFieldQuery -> ShowS
[MoreLikeThisFieldQuery] -> ShowS
MoreLikeThisFieldQuery -> String
(Int -> MoreLikeThisFieldQuery -> ShowS)
-> (MoreLikeThisFieldQuery -> String)
-> ([MoreLikeThisFieldQuery] -> ShowS)
-> Show MoreLikeThisFieldQuery
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> MoreLikeThisFieldQuery -> ShowS
showsPrec :: Int -> MoreLikeThisFieldQuery -> ShowS
$cshow :: MoreLikeThisFieldQuery -> String
show :: MoreLikeThisFieldQuery -> String
$cshowList :: [MoreLikeThisFieldQuery] -> ShowS
showList :: [MoreLikeThisFieldQuery] -> ShowS
Show, (forall x. MoreLikeThisFieldQuery -> Rep MoreLikeThisFieldQuery x)
-> (forall x.
    Rep MoreLikeThisFieldQuery x -> MoreLikeThisFieldQuery)
-> Generic MoreLikeThisFieldQuery
forall x. Rep MoreLikeThisFieldQuery x -> MoreLikeThisFieldQuery
forall x. MoreLikeThisFieldQuery -> Rep MoreLikeThisFieldQuery x
forall a.
(forall x. a -> Rep a x) -> (forall x. Rep a x -> a) -> Generic a
$cfrom :: forall x. MoreLikeThisFieldQuery -> Rep MoreLikeThisFieldQuery x
from :: forall x. MoreLikeThisFieldQuery -> Rep MoreLikeThisFieldQuery x
$cto :: forall x. Rep MoreLikeThisFieldQuery x -> MoreLikeThisFieldQuery
to :: forall x. Rep MoreLikeThisFieldQuery x -> MoreLikeThisFieldQuery
Generic)

instance ToJSON MoreLikeThisFieldQuery where
  toJSON :: MoreLikeThisFieldQuery -> Value
toJSON
    ( MoreLikeThisFieldQuery
        Text
text
        (FieldName Text
fieldName)
        Maybe PercentMatch
percent
        Maybe MinimumTermFrequency
mtf
        Maybe MaxQueryTerms
mqt
        Maybe (NonEmpty StopWord)
stopwords
        Maybe MinDocFrequency
mindf
        Maybe MaxDocFrequency
maxdf
        Maybe MinWordLength
minwl
        Maybe MaxWordLength
maxwl
        Maybe BoostTerms
boostTerms
        Maybe Boost
boost
        Maybe Analyzer
analyzer
      ) =
      [Pair] -> Value
object [Text -> Key
fromText Text
fieldName Key -> Value -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= [Pair] -> Value
omitNulls [Pair]
base]
      where
        base :: [Pair]
base =
          [ Key
"like_text" Key -> Text -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Text
text,
            Key
"percent_terms_to_match" Key -> Maybe PercentMatch -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe PercentMatch
percent,
            Key
"min_term_freq" Key -> Maybe MinimumTermFrequency -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe MinimumTermFrequency
mtf,
            Key
"max_query_terms" Key -> Maybe MaxQueryTerms -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe MaxQueryTerms
mqt,
            Key
"stop_words" Key -> Maybe (NonEmpty StopWord) -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe (NonEmpty StopWord)
stopwords,
            Key
"min_doc_freq" Key -> Maybe MinDocFrequency -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe MinDocFrequency
mindf,
            Key
"max_doc_freq" Key -> Maybe MaxDocFrequency -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe MaxDocFrequency
maxdf,
            Key
"min_word_length" Key -> Maybe MinWordLength -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe MinWordLength
minwl,
            Key
"max_word_length" Key -> Maybe MaxWordLength -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe MaxWordLength
maxwl,
            Key
"boost_terms" Key -> Maybe BoostTerms -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe BoostTerms
boostTerms,
            Key
"boost" Key -> Maybe Boost -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe Boost
boost,
            Key
"analyzer" Key -> Maybe Analyzer -> Pair
forall v. ToJSON v => Key -> v -> Pair
forall e kv v. (KeyValue e kv, ToJSON v) => Key -> v -> kv
.= Maybe Analyzer
analyzer
          ]

instance FromJSON MoreLikeThisFieldQuery where
  parseJSON :: Value -> Parser MoreLikeThisFieldQuery
parseJSON = String
-> (Object -> Parser MoreLikeThisFieldQuery)
-> Value
-> Parser MoreLikeThisFieldQuery
forall a. String -> (Object -> Parser a) -> Value -> Parser a
withObject String
"MoreLikeThisFieldQuery" Object -> Parser MoreLikeThisFieldQuery
parse
    where
      parse :: Object -> Parser MoreLikeThisFieldQuery
parse = (FieldName -> Object -> Parser MoreLikeThisFieldQuery)
-> Object -> Parser MoreLikeThisFieldQuery
forall (m :: * -> *) a.
(Monad m, MonadFail m) =>
(FieldName -> Object -> m a) -> Object -> m a
fieldTagged ((FieldName -> Object -> Parser MoreLikeThisFieldQuery)
 -> Object -> Parser MoreLikeThisFieldQuery)
-> (FieldName -> Object -> Parser MoreLikeThisFieldQuery)
-> Object
-> Parser MoreLikeThisFieldQuery
forall a b. (a -> b) -> a -> b
$ \FieldName
fn Object
o ->
        Text
-> FieldName
-> Maybe PercentMatch
-> Maybe MinimumTermFrequency
-> Maybe MaxQueryTerms
-> Maybe (NonEmpty StopWord)
-> Maybe MinDocFrequency
-> Maybe MaxDocFrequency
-> Maybe MinWordLength
-> Maybe MaxWordLength
-> Maybe BoostTerms
-> Maybe Boost
-> Maybe Analyzer
-> MoreLikeThisFieldQuery
MoreLikeThisFieldQuery
          (Text
 -> FieldName
 -> Maybe PercentMatch
 -> Maybe MinimumTermFrequency
 -> Maybe MaxQueryTerms
 -> Maybe (NonEmpty StopWord)
 -> Maybe MinDocFrequency
 -> Maybe MaxDocFrequency
 -> Maybe MinWordLength
 -> Maybe MaxWordLength
 -> Maybe BoostTerms
 -> Maybe Boost
 -> Maybe Analyzer
 -> MoreLikeThisFieldQuery)
-> Parser Text
-> Parser
     (FieldName
      -> Maybe PercentMatch
      -> Maybe MinimumTermFrequency
      -> Maybe MaxQueryTerms
      -> Maybe (NonEmpty StopWord)
      -> Maybe MinDocFrequency
      -> Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Object
o Object -> Key -> Parser Text
forall a. FromJSON a => Object -> Key -> Parser a
.: Key
"like_text"
          Parser
  (FieldName
   -> Maybe PercentMatch
   -> Maybe MinimumTermFrequency
   -> Maybe MaxQueryTerms
   -> Maybe (NonEmpty StopWord)
   -> Maybe MinDocFrequency
   -> Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser FieldName
-> Parser
     (Maybe PercentMatch
      -> Maybe MinimumTermFrequency
      -> Maybe MaxQueryTerms
      -> Maybe (NonEmpty StopWord)
      -> Maybe MinDocFrequency
      -> Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> FieldName -> Parser FieldName
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure FieldName
fn
          Parser
  (Maybe PercentMatch
   -> Maybe MinimumTermFrequency
   -> Maybe MaxQueryTerms
   -> Maybe (NonEmpty StopWord)
   -> Maybe MinDocFrequency
   -> Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe PercentMatch)
-> Parser
     (Maybe MinimumTermFrequency
      -> Maybe MaxQueryTerms
      -> Maybe (NonEmpty StopWord)
      -> Maybe MinDocFrequency
      -> Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe PercentMatch)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"percent_terms_to_match"
          Parser
  (Maybe MinimumTermFrequency
   -> Maybe MaxQueryTerms
   -> Maybe (NonEmpty StopWord)
   -> Maybe MinDocFrequency
   -> Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe MinimumTermFrequency)
-> Parser
     (Maybe MaxQueryTerms
      -> Maybe (NonEmpty StopWord)
      -> Maybe MinDocFrequency
      -> Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe MinimumTermFrequency)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"min_term_freq"
          Parser
  (Maybe MaxQueryTerms
   -> Maybe (NonEmpty StopWord)
   -> Maybe MinDocFrequency
   -> Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe MaxQueryTerms)
-> Parser
     (Maybe (NonEmpty StopWord)
      -> Maybe MinDocFrequency
      -> Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe MaxQueryTerms)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"max_query_terms"
          -- <*> (optionalNE =<< o .:? "stop_words")
          Parser
  (Maybe (NonEmpty StopWord)
   -> Maybe MinDocFrequency
   -> Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe (NonEmpty StopWord))
-> Parser
     (Maybe MinDocFrequency
      -> Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe (NonEmpty StopWord))
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"stop_words"
          Parser
  (Maybe MinDocFrequency
   -> Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe MinDocFrequency)
-> Parser
     (Maybe MaxDocFrequency
      -> Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe MinDocFrequency)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"min_doc_freq"
          Parser
  (Maybe MaxDocFrequency
   -> Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe MaxDocFrequency)
-> Parser
     (Maybe MinWordLength
      -> Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe MaxDocFrequency)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"max_doc_freq"
          Parser
  (Maybe MinWordLength
   -> Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe MinWordLength)
-> Parser
     (Maybe MaxWordLength
      -> Maybe BoostTerms
      -> Maybe Boost
      -> Maybe Analyzer
      -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe MinWordLength)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"min_word_length"
          Parser
  (Maybe MaxWordLength
   -> Maybe BoostTerms
   -> Maybe Boost
   -> Maybe Analyzer
   -> MoreLikeThisFieldQuery)
-> Parser (Maybe MaxWordLength)
-> Parser
     (Maybe BoostTerms
      -> Maybe Boost -> Maybe Analyzer -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe MaxWordLength)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"max_word_length"
          Parser
  (Maybe BoostTerms
   -> Maybe Boost -> Maybe Analyzer -> MoreLikeThisFieldQuery)
-> Parser (Maybe BoostTerms)
-> Parser (Maybe Boost -> Maybe Analyzer -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe BoostTerms)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"boost_terms"
          Parser (Maybe Boost -> Maybe Analyzer -> MoreLikeThisFieldQuery)
-> Parser (Maybe Boost)
-> Parser (Maybe Analyzer -> MoreLikeThisFieldQuery)
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe Boost)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"boost"
          Parser (Maybe Analyzer -> MoreLikeThisFieldQuery)
-> Parser (Maybe Analyzer) -> Parser MoreLikeThisFieldQuery
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> Object
o Object -> Key -> Parser (Maybe Analyzer)
forall a. FromJSON a => Object -> Key -> Parser (Maybe a)
.:? Key
"analyzer"