From dda50b186cb5f269a798a8d5100f516134e7e6fb Mon Sep 17 00:00:00 2001
From: Will Jones <will@sacharissa.co.uk>
Date: Tue, 4 Jul 2017 15:09:23 +0100
Subject: [PATCH] Add support for stored scripts with IDs.

---
 src/Database/V5/Bloodhound/Types.hs | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/Database/V5/Bloodhound/Types.hs b/src/Database/V5/Bloodhound/Types.hs
index 2258ebf..24051d9 100644
--- a/src/Database/V5/Bloodhound/Types.hs
+++ b/src/Database/V5/Bloodhound/Types.hs
@@ -188,6 +188,7 @@ module Database.V5.Bloodhound.Types
        , FunctionScoreFunction(..)
        , FunctionScoreScript(..)
        , FunctionScoreScriptInline(..)
+       , FunctionScoreScriptId(..)
        , FunctionScoreScriptParams(..)
        , FunctionScoreScriptParamName
        , FunctionScoreScriptParamValue
@@ -1480,13 +1481,17 @@ data FunctionScoreFunction =
   deriving (Eq, Read, Show, Generic, Typeable)
 
 data FunctionScoreScript =
-  FunctionScoreScript { functionScoreScriptInline :: FunctionScoreScriptInline
+  FunctionScoreScript { functionScoreScriptInline :: Maybe FunctionScoreScriptInline
+                      , functionScoreScriptStored :: Maybe FunctionScoreScriptId
                       , functionScoreScriptParams :: Maybe FunctionScoreScriptParams
                       } deriving (Eq, Read, Show, Generic, Typeable)
 
 newtype FunctionScoreScriptInline =
   FunctionScoreScriptInline Text deriving (Eq, Read, Show, Generic, Typeable, ToJSON, FromJSON)
 
+newtype FunctionScoreScriptId =
+  FunctionScoreScriptId Text deriving (Eq, Read, Show, Generic, Typeable, ToJSON, FromJSON)
+
 newtype FunctionScoreScriptParams =
   FunctionScoreScriptParams (HM.HashMap FunctionScoreScriptParamName FunctionScoreScriptParamValue)
   deriving (Eq, Read, Show, Generic, Typeable)
@@ -2951,16 +2956,18 @@ parseFunctionScoreFunction o =
         singleFieldValueFactor = pure . FunctionScoreFunctionFieldValueFactor
 
 instance ToJSON FunctionScoreScript where
-  toJSON (FunctionScoreScript inline params) =
+  toJSON (FunctionScoreScript inline stored params) =
     object [ "script" .= omitNulls base ]
     where base = [ "inline" .= inline
+                 , "stored" .= stored
                  , "params" .= params ]
 
 instance FromJSON FunctionScoreScript where
   parseJSON = withObject "FunctionScoreScript" parse
     where parse o = o .: "script" >>= \o' ->
                       FunctionScoreScript
-                      <$> o' .: "inline"
+                      <$> o' .:? "inline"
+                      <*> o' .:? "stored"
                       <*> o' .:? "params"
 
 instance ToJSON FunctionScoreScriptParams where