{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE UndecidableInstances #-}

module Database.Bloodhound.Client.Cluster
  ( module ReexportCompat,
    BH (..),
    BHEnv (..),
    MonadBH (..),
    emptyBody,
    mkBHEnv,
    performBHRequest,
    runBH,
    tryPerformBHRequest,
  )
where

import Control.Monad.Catch
import Control.Monad.Except
import qualified Data.ByteString.Lazy.Char8 as L
import qualified Data.Text as T
import Database.Bloodhound.Internal.Client.BHRequest
import Database.Bloodhound.Internal.Utils.Imports
import Database.Bloodhound.Internal.Versions.Common.Types.Bulk as ReexportCompat
import Database.Bloodhound.Internal.Versions.Common.Types.Count as ReexportCompat
import Database.Bloodhound.Internal.Versions.Common.Types.Indices as ReexportCompat
import Database.Bloodhound.Internal.Versions.Common.Types.Nodes as ReexportCompat
import Database.Bloodhound.Internal.Versions.Common.Types.Snapshots as ReexportCompat
import Database.Bloodhound.Internal.Versions.Common.Types.Units as ReexportCompat
import Network.HTTP.Client
import qualified Network.URI as URI

-- | Common environment for Elasticsearch calls. Connections will be
--   pipelined according to the provided HTTP connection manager.
data BHEnv = BHEnv
  { BHEnv -> Server
bhServer :: Server,
    BHEnv -> Manager
bhManager :: Manager,
    -- | Low-level hook that is run before every request is sent. Used to implement custom authentication strategies. Defaults to 'return' with 'mkBHEnv'.
    BHEnv -> Request -> IO Request
bhRequestHook :: Request -> IO Request
  }

-- | All API calls to Elasticsearch operate within
--   MonadBH
--   . The idea is that it can be easily embedded in your
--   own monad transformer stack. A default instance for a ReaderT and
--   alias 'BH' is provided for the simple case.
class (Functor m, Applicative m, MonadIO m, MonadCatch m) => MonadBH m where
  dispatch :: BHRequest contextualized body -> m (BHResponse contextualized body)
  tryEsError :: m a -> m (Either EsError a)
  throwEsError :: EsError -> m a

-- | Create a 'BHEnv' with all optional fields defaulted. HTTP hook
-- will be a noop. You can use the exported fields to customize
-- it further, e.g.:
--
-- >> (mkBHEnv myServer myManager) { bhRequestHook = customHook }
mkBHEnv :: Server -> Manager -> BHEnv
mkBHEnv :: Server -> Manager -> BHEnv
mkBHEnv Server
s Manager
m = Server -> Manager -> (Request -> IO Request) -> BHEnv
BHEnv Server
s Manager
m Request -> IO Request
forall a. a -> IO a
forall (m :: * -> *) a. Monad m => a -> m a
return

newtype BH m a = BH
  { forall (m :: * -> *) a.
BH m a -> ReaderT BHEnv (ExceptT EsError m) a
unBH :: ReaderT BHEnv (ExceptT EsError m) a
  }
  deriving newtype
    ( (forall a b. (a -> b) -> BH m a -> BH m b)
-> (forall a b. a -> BH m b -> BH m a) -> Functor (BH m)
forall a b. a -> BH m b -> BH m a
forall a b. (a -> b) -> BH m a -> BH m b
forall (m :: * -> *) a b. Functor m => a -> BH m b -> BH m a
forall (m :: * -> *) a b. Functor m => (a -> b) -> BH m a -> BH m b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
$cfmap :: forall (m :: * -> *) a b. Functor m => (a -> b) -> BH m a -> BH m b
fmap :: forall a b. (a -> b) -> BH m a -> BH m b
$c<$ :: forall (m :: * -> *) a b. Functor m => a -> BH m b -> BH m a
<$ :: forall a b. a -> BH m b -> BH m a
Functor,
      Functor (BH m)
Functor (BH m) =>
(forall a. a -> BH m a)
-> (forall a b. BH m (a -> b) -> BH m a -> BH m b)
-> (forall a b c. (a -> b -> c) -> BH m a -> BH m b -> BH m c)
-> (forall a b. BH m a -> BH m b -> BH m b)
-> (forall a b. BH m a -> BH m b -> BH m a)
-> Applicative (BH m)
forall a. a -> BH m a
forall a b. BH m a -> BH m b -> BH m a
forall a b. BH m a -> BH m b -> BH m b
forall a b. BH m (a -> b) -> BH m a -> BH m b
forall a b c. (a -> b -> c) -> BH m a -> BH m b -> BH m c
forall (m :: * -> *). Monad m => Functor (BH m)
forall (m :: * -> *) a. Monad m => a -> BH m a
forall (m :: * -> *) a b. Monad m => BH m a -> BH m b -> BH m a
forall (m :: * -> *) a b. Monad m => BH m a -> BH m b -> BH m b
forall (m :: * -> *) a b.
Monad m =>
BH m (a -> b) -> BH m a -> BH m b
forall (m :: * -> *) a b c.
Monad m =>
(a -> b -> c) -> BH m a -> BH m b -> BH m c
forall (f :: * -> *).
Functor f =>
(forall a. a -> f a)
-> (forall a b. f (a -> b) -> f a -> f b)
-> (forall a b c. (a -> b -> c) -> f a -> f b -> f c)
-> (forall a b. f a -> f b -> f b)
-> (forall a b. f a -> f b -> f a)
-> Applicative f
$cpure :: forall (m :: * -> *) a. Monad m => a -> BH m a
pure :: forall a. a -> BH m a
$c<*> :: forall (m :: * -> *) a b.
Monad m =>
BH m (a -> b) -> BH m a -> BH m b
<*> :: forall a b. BH m (a -> b) -> BH m a -> BH m b
$cliftA2 :: forall (m :: * -> *) a b c.
Monad m =>
(a -> b -> c) -> BH m a -> BH m b -> BH m c
liftA2 :: forall a b c. (a -> b -> c) -> BH m a -> BH m b -> BH m c
$c*> :: forall (m :: * -> *) a b. Monad m => BH m a -> BH m b -> BH m b
*> :: forall a b. BH m a -> BH m b -> BH m b
$c<* :: forall (m :: * -> *) a b. Monad m => BH m a -> BH m b -> BH m a
<* :: forall a b. BH m a -> BH m b -> BH m a
Applicative,
      Applicative (BH m)
Applicative (BH m) =>
(forall a b. BH m a -> (a -> BH m b) -> BH m b)
-> (forall a b. BH m a -> BH m b -> BH m b)
-> (forall a. a -> BH m a)
-> Monad (BH m)
forall a. a -> BH m a
forall a b. BH m a -> BH m b -> BH m b
forall a b. BH m a -> (a -> BH m b) -> BH m b
forall (m :: * -> *). Monad m => Applicative (BH m)
forall (m :: * -> *) a. Monad m => a -> BH m a
forall (m :: * -> *) a b. Monad m => BH m a -> BH m b -> BH m b
forall (m :: * -> *) a b.
Monad m =>
BH m a -> (a -> BH m b) -> BH m b
forall (m :: * -> *).
Applicative m =>
(forall a b. m a -> (a -> m b) -> m b)
-> (forall a b. m a -> m b -> m b)
-> (forall a. a -> m a)
-> Monad m
$c>>= :: forall (m :: * -> *) a b.
Monad m =>
BH m a -> (a -> BH m b) -> BH m b
>>= :: forall a b. BH m a -> (a -> BH m b) -> BH m b
$c>> :: forall (m :: * -> *) a b. Monad m => BH m a -> BH m b -> BH m b
>> :: forall a b. BH m a -> BH m b -> BH m b
$creturn :: forall (m :: * -> *) a. Monad m => a -> BH m a
return :: forall a. a -> BH m a
Monad,
      Monad (BH m)
Monad (BH m) => (forall a. IO a -> BH m a) -> MonadIO (BH m)
forall a. IO a -> BH m a
forall (m :: * -> *).
Monad m =>
(forall a. IO a -> m a) -> MonadIO m
forall (m :: * -> *). MonadIO m => Monad (BH m)
forall (m :: * -> *) a. MonadIO m => IO a -> BH m a
$cliftIO :: forall (m :: * -> *) a. MonadIO m => IO a -> BH m a
liftIO :: forall a. IO a -> BH m a
MonadIO,
      MonadState s,
      MonadWriter w,
      Applicative (BH m)
Applicative (BH m) =>
(forall a. BH m a)
-> (forall a. BH m a -> BH m a -> BH m a)
-> (forall a. BH m a -> BH m [a])
-> (forall a. BH m a -> BH m [a])
-> Alternative (BH m)
forall a. BH m a
forall a. BH m a -> BH m [a]
forall a. BH m a -> BH m a -> BH m a
forall (m :: * -> *). Monad m => Applicative (BH m)
forall (m :: * -> *) a. Monad m => BH m a
forall (m :: * -> *) a. Monad m => BH m a -> BH m [a]
forall (m :: * -> *) a. Monad m => BH m a -> BH m a -> BH m a
forall (f :: * -> *).
Applicative f =>
(forall a. f a)
-> (forall a. f a -> f a -> f a)
-> (forall a. f a -> f [a])
-> (forall a. f a -> f [a])
-> Alternative f
$cempty :: forall (m :: * -> *) a. Monad m => BH m a
empty :: forall a. BH m a
$c<|> :: forall (m :: * -> *) a. Monad m => BH m a -> BH m a -> BH m a
<|> :: forall a. BH m a -> BH m a -> BH m a
$csome :: forall (m :: * -> *) a. Monad m => BH m a -> BH m [a]
some :: forall a. BH m a -> BH m [a]
$cmany :: forall (m :: * -> *) a. Monad m => BH m a -> BH m [a]
many :: forall a. BH m a -> BH m [a]
Alternative,
      Monad (BH m)
Alternative (BH m)
(Alternative (BH m), Monad (BH m)) =>
(forall a. BH m a)
-> (forall a. BH m a -> BH m a -> BH m a) -> MonadPlus (BH m)
forall a. BH m a
forall a. BH m a -> BH m a -> BH m a
forall (m :: * -> *). Monad m => Monad (BH m)
forall (m :: * -> *). Monad m => Alternative (BH m)
forall (m :: * -> *) a. Monad m => BH m a
forall (m :: * -> *) a. Monad m => BH m a -> BH m a -> BH m a
forall (m :: * -> *).
(Alternative m, Monad m) =>
(forall a. m a) -> (forall a. m a -> m a -> m a) -> MonadPlus m
$cmzero :: forall (m :: * -> *) a. Monad m => BH m a
mzero :: forall a. BH m a
$cmplus :: forall (m :: * -> *) a. Monad m => BH m a -> BH m a -> BH m a
mplus :: forall a. BH m a -> BH m a -> BH m a
MonadPlus,
      Monad (BH m)
Monad (BH m) =>
(forall a. (a -> BH m a) -> BH m a) -> MonadFix (BH m)
forall a. (a -> BH m a) -> BH m a
forall (m :: * -> *).
Monad m =>
(forall a. (a -> m a) -> m a) -> MonadFix m
forall (m :: * -> *). MonadFix m => Monad (BH m)
forall (m :: * -> *) a. MonadFix m => (a -> BH m a) -> BH m a
$cmfix :: forall (m :: * -> *) a. MonadFix m => (a -> BH m a) -> BH m a
mfix :: forall a. (a -> BH m a) -> BH m a
MonadFix,
      Monad (BH m)
Monad (BH m) =>
(forall e a. (HasCallStack, Exception e) => e -> BH m a)
-> MonadThrow (BH m)
forall e a. (HasCallStack, Exception e) => e -> BH m a
forall (m :: * -> *).
Monad m =>
(forall e a. (HasCallStack, Exception e) => e -> m a)
-> MonadThrow m
forall (m :: * -> *). MonadThrow m => Monad (BH m)
forall (m :: * -> *) e a.
(MonadThrow m, HasCallStack, Exception e) =>
e -> BH m a
$cthrowM :: forall (m :: * -> *) e a.
(MonadThrow m, HasCallStack, Exception e) =>
e -> BH m a
throwM :: forall e a. (HasCallStack, Exception e) => e -> BH m a
MonadThrow,
      MonadThrow (BH m)
MonadThrow (BH m) =>
(forall e a.
 (HasCallStack, Exception e) =>
 BH m a -> (e -> BH m a) -> BH m a)
-> MonadCatch (BH m)
forall e a.
(HasCallStack, Exception e) =>
BH m a -> (e -> BH m a) -> BH m a
forall (m :: * -> *). MonadCatch m => MonadThrow (BH m)
forall (m :: * -> *) e a.
(MonadCatch m, HasCallStack, Exception e) =>
BH m a -> (e -> BH m a) -> BH m a
forall (m :: * -> *).
MonadThrow m =>
(forall e a.
 (HasCallStack, Exception e) =>
 m a -> (e -> m a) -> m a)
-> MonadCatch m
$ccatch :: forall (m :: * -> *) e a.
(MonadCatch m, HasCallStack, Exception e) =>
BH m a -> (e -> BH m a) -> BH m a
catch :: forall e a.
(HasCallStack, Exception e) =>
BH m a -> (e -> BH m a) -> BH m a
MonadCatch,
      Monad (BH m)
Monad (BH m) => (forall a. String -> BH m a) -> MonadFail (BH m)
forall a. String -> BH m a
forall (m :: * -> *).
Monad m =>
(forall a. String -> m a) -> MonadFail m
forall (m :: * -> *). MonadFail m => Monad (BH m)
forall (m :: * -> *) a. MonadFail m => String -> BH m a
$cfail :: forall (m :: * -> *) a. MonadFail m => String -> BH m a
fail :: forall a. String -> BH m a
MonadFail,
      MonadCatch (BH m)
MonadCatch (BH m) =>
(forall b.
 HasCallStack =>
 ((forall a. BH m a -> BH m a) -> BH m b) -> BH m b)
-> (forall b.
    HasCallStack =>
    ((forall a. BH m a -> BH m a) -> BH m b) -> BH m b)
-> (forall a b c.
    HasCallStack =>
    BH m a
    -> (a -> ExitCase b -> BH m c) -> (a -> BH m b) -> BH m (b, c))
-> MonadMask (BH m)
forall b.
HasCallStack =>
((forall a. BH m a -> BH m a) -> BH m b) -> BH m b
forall a b c.
HasCallStack =>
BH m a
-> (a -> ExitCase b -> BH m c) -> (a -> BH m b) -> BH m (b, c)
forall (m :: * -> *). MonadMask m => MonadCatch (BH m)
forall (m :: * -> *) b.
(MonadMask m, HasCallStack) =>
((forall a. BH m a -> BH m a) -> BH m b) -> BH m b
forall (m :: * -> *) a b c.
(MonadMask m, HasCallStack) =>
BH m a
-> (a -> ExitCase b -> BH m c) -> (a -> BH m b) -> BH m (b, c)
forall (m :: * -> *).
MonadCatch m =>
(forall b. HasCallStack => ((forall a. m a -> m a) -> m b) -> m b)
-> (forall b.
    HasCallStack =>
    ((forall a. m a -> m a) -> m b) -> m b)
-> (forall a b c.
    HasCallStack =>
    m a -> (a -> ExitCase b -> m c) -> (a -> m b) -> m (b, c))
-> MonadMask m
$cmask :: forall (m :: * -> *) b.
(MonadMask m, HasCallStack) =>
((forall a. BH m a -> BH m a) -> BH m b) -> BH m b
mask :: forall b.
HasCallStack =>
((forall a. BH m a -> BH m a) -> BH m b) -> BH m b
$cuninterruptibleMask :: forall (m :: * -> *) b.
(MonadMask m, HasCallStack) =>
((forall a. BH m a -> BH m a) -> BH m b) -> BH m b
uninterruptibleMask :: forall b.
HasCallStack =>
((forall a. BH m a -> BH m a) -> BH m b) -> BH m b
$cgeneralBracket :: forall (m :: * -> *) a b c.
(MonadMask m, HasCallStack) =>
BH m a
-> (a -> ExitCase b -> BH m c) -> (a -> BH m b) -> BH m (b, c)
generalBracket :: forall a b c.
HasCallStack =>
BH m a
-> (a -> ExitCase b -> BH m c) -> (a -> BH m b) -> BH m (b, c)
MonadMask
    )

instance MonadTrans BH where
  lift :: forall (m :: * -> *) a. Monad m => m a -> BH m a
lift = ReaderT BHEnv (ExceptT EsError m) a -> BH m a
forall (m :: * -> *) a.
ReaderT BHEnv (ExceptT EsError m) a -> BH m a
BH (ReaderT BHEnv (ExceptT EsError m) a -> BH m a)
-> (m a -> ReaderT BHEnv (ExceptT EsError m) a) -> m a -> BH m a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ExceptT EsError m a -> ReaderT BHEnv (ExceptT EsError m) a
forall (m :: * -> *) a. Monad m => m a -> ReaderT BHEnv m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (ExceptT EsError m a -> ReaderT BHEnv (ExceptT EsError m) a)
-> (m a -> ExceptT EsError m a)
-> m a
-> ReaderT BHEnv (ExceptT EsError m) a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. m a -> ExceptT EsError m a
forall (m :: * -> *) a. Monad m => m a -> ExceptT EsError m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift

instance (MonadReader r m) => MonadReader r (BH m) where
  ask :: BH m r
ask = m r -> BH m r
forall (m :: * -> *) a. Monad m => m a -> BH m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift m r
forall r (m :: * -> *). MonadReader r m => m r
ask
  local :: forall a. (r -> r) -> BH m a -> BH m a
local r -> r
f (BH (ReaderT BHEnv -> ExceptT EsError m a
m)) = ReaderT BHEnv (ExceptT EsError m) a -> BH m a
forall (m :: * -> *) a.
ReaderT BHEnv (ExceptT EsError m) a -> BH m a
BH (ReaderT BHEnv (ExceptT EsError m) a -> BH m a)
-> ReaderT BHEnv (ExceptT EsError m) a -> BH m a
forall a b. (a -> b) -> a -> b
$
    (BHEnv -> ExceptT EsError m a)
-> ReaderT BHEnv (ExceptT EsError m) a
forall r (m :: * -> *) a. (r -> m a) -> ReaderT r m a
ReaderT ((BHEnv -> ExceptT EsError m a)
 -> ReaderT BHEnv (ExceptT EsError m) a)
-> (BHEnv -> ExceptT EsError m a)
-> ReaderT BHEnv (ExceptT EsError m) a
forall a b. (a -> b) -> a -> b
$ \BHEnv
r ->
      (r -> r) -> ExceptT EsError m a -> ExceptT EsError m a
forall a. (r -> r) -> ExceptT EsError m a -> ExceptT EsError m a
forall r (m :: * -> *) a. MonadReader r m => (r -> r) -> m a -> m a
local r -> r
f (BHEnv -> ExceptT EsError m a
m BHEnv
r)

instance (Functor m, Applicative m, MonadIO m, MonadCatch m, MonadThrow m) => MonadBH (BH m) where
  dispatch :: forall contextualized body.
BHRequest contextualized body
-> BH m (BHResponse contextualized body)
dispatch BHRequest contextualized body
request = ReaderT BHEnv (ExceptT EsError m) (BHResponse contextualized body)
-> BH m (BHResponse contextualized body)
forall (m :: * -> *) a.
ReaderT BHEnv (ExceptT EsError m) a -> BH m a
BH (ReaderT BHEnv (ExceptT EsError m) (BHResponse contextualized body)
 -> BH m (BHResponse contextualized body))
-> ReaderT
     BHEnv (ExceptT EsError m) (BHResponse contextualized body)
-> BH m (BHResponse contextualized body)
forall a b. (a -> b) -> a -> b
$ do
    BHEnv
env <- forall r (m :: * -> *). MonadReader r m => m r
ask @BHEnv
    let url :: Text
url = Server -> Endpoint -> Text
getEndpoint (BHEnv -> Server
bhServer BHEnv
env) (BHRequest contextualized body -> Endpoint
forall parsingContext responseBody.
BHRequest parsingContext responseBody -> Endpoint
bhRequestEndpoint BHRequest contextualized body
request)
    Request
initReq <- IO Request -> ReaderT BHEnv (ExceptT EsError m) Request
forall a. IO a -> ReaderT BHEnv (ExceptT EsError m) a
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO Request -> ReaderT BHEnv (ExceptT EsError m) Request)
-> IO Request -> ReaderT BHEnv (ExceptT EsError m) Request
forall a b. (a -> b) -> a -> b
$ Text -> IO Request
forall (m :: * -> *). MonadThrow m => Text -> m Request
parseUrl' Text
url
    let reqHook :: Request -> IO Request
reqHook = BHEnv -> Request -> IO Request
bhRequestHook BHEnv
env
    let reqBody :: RequestBody
reqBody = ByteString -> RequestBody
RequestBodyLBS (ByteString -> RequestBody) -> ByteString -> RequestBody
forall a b. (a -> b) -> a -> b
$ ByteString -> Maybe ByteString -> ByteString
forall a. a -> Maybe a -> a
fromMaybe ByteString
emptyBody (Maybe ByteString -> ByteString) -> Maybe ByteString -> ByteString
forall a b. (a -> b) -> a -> b
$ BHRequest contextualized body -> Maybe ByteString
forall parsingContext responseBody.
BHRequest parsingContext responseBody -> Maybe ByteString
bhRequestBody BHRequest contextualized body
request
    Request
req <-
      IO Request -> ReaderT BHEnv (ExceptT EsError m) Request
forall a. IO a -> ReaderT BHEnv (ExceptT EsError m) a
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (IO Request -> ReaderT BHEnv (ExceptT EsError m) Request)
-> IO Request -> ReaderT BHEnv (ExceptT EsError m) Request
forall a b. (a -> b) -> a -> b
$
        Request -> IO Request
reqHook (Request -> IO Request) -> Request -> IO Request
forall a b. (a -> b) -> a -> b
$
          Request -> Request
setRequestIgnoreStatus (Request -> Request) -> Request -> Request
forall a b. (a -> b) -> a -> b
$
            Request
initReq
              { method = bhRequestMethod request,
                requestHeaders =
                  -- "application/x-ndjson" for bulk
                  ("Content-Type", "application/json") : requestHeaders initReq,
                requestBody = reqBody
              }
    -- req <- liftIO $ reqHook $ setRequestIgnoreStatus $ initReq { method = dMethod
    --                                                            , requestBody = reqBody }
    let mgr :: Manager
mgr = BHEnv -> Manager
bhManager BHEnv
env
    Response ByteString -> BHResponse contextualized body
forall parsingContext body.
Response ByteString -> BHResponse parsingContext body
BHResponse (Response ByteString -> BHResponse contextualized body)
-> ReaderT BHEnv (ExceptT EsError m) (Response ByteString)
-> ReaderT
     BHEnv (ExceptT EsError m) (BHResponse contextualized body)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> IO (Response ByteString)
-> ReaderT BHEnv (ExceptT EsError m) (Response ByteString)
forall a. IO a -> ReaderT BHEnv (ExceptT EsError m) a
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO (Request -> Manager -> IO (Response ByteString)
httpLbs Request
req Manager
mgr)
  tryEsError :: forall a. BH m a -> BH m (Either EsError a)
tryEsError = BH m a -> BH m (Either EsError a)
forall (m :: * -> *) e a.
(HasCallStack, MonadCatch m, Exception e) =>
m a -> m (Either e a)
try
  throwEsError :: forall a. EsError -> BH m a
throwEsError = EsError -> BH m a
forall e a. (HasCallStack, Exception e) => e -> BH m a
forall (m :: * -> *) e a.
(MonadThrow m, HasCallStack, Exception e) =>
e -> m a
throwM

tryPerformBHRequest ::
  (MonadBH m, MonadThrow m, ParseBHResponse contextualized) =>
  BHRequest contextualized a ->
  m (ParsedEsResponse a)
tryPerformBHRequest :: forall (m :: * -> *) contextualized a.
(MonadBH m, MonadThrow m, ParseBHResponse contextualized) =>
BHRequest contextualized a -> m (ParsedEsResponse a)
tryPerformBHRequest BHRequest contextualized a
req = BHRequest contextualized a -> m (BHResponse contextualized a)
forall contextualized body.
BHRequest contextualized body -> m (BHResponse contextualized body)
forall (m :: * -> *) contextualized body.
MonadBH m =>
BHRequest contextualized body -> m (BHResponse contextualized body)
dispatch BHRequest contextualized a
req m (BHResponse contextualized a)
-> (BHResponse contextualized a -> m (ParsedEsResponse a))
-> m (ParsedEsResponse a)
forall a b. m a -> (a -> m b) -> m b
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= (EsProtocolException -> m (ParsedEsResponse a))
-> (ParsedEsResponse a -> m (ParsedEsResponse a))
-> Either EsProtocolException (ParsedEsResponse a)
-> m (ParsedEsResponse a)
forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either EsProtocolException -> m (ParsedEsResponse a)
forall e a. (HasCallStack, Exception e) => e -> m a
forall (m :: * -> *) e a.
(MonadThrow m, HasCallStack, Exception e) =>
e -> m a
throwM ParsedEsResponse a -> m (ParsedEsResponse a)
forall a. a -> m a
forall (m :: * -> *) a. Monad m => a -> m a
return (Either EsProtocolException (ParsedEsResponse a)
 -> m (ParsedEsResponse a))
-> (BHResponse contextualized a
    -> Either EsProtocolException (ParsedEsResponse a))
-> BHResponse contextualized a
-> m (ParsedEsResponse a)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. BHRequest contextualized a
-> BHResponse contextualized a
-> Either EsProtocolException (ParsedEsResponse a)
forall parsingContext responseBody.
BHRequest parsingContext responseBody
-> BHResponse parsingContext responseBody
-> Either EsProtocolException (ParsedEsResponse responseBody)
bhRequestParser BHRequest contextualized a
req

performBHRequest ::
  (MonadBH m, MonadThrow m, ParseBHResponse contextualized) =>
  BHRequest contextualized a ->
  m a
performBHRequest :: forall (m :: * -> *) contextualized a.
(MonadBH m, MonadThrow m, ParseBHResponse contextualized) =>
BHRequest contextualized a -> m a
performBHRequest BHRequest contextualized a
req = BHRequest contextualized a -> m (ParsedEsResponse a)
forall (m :: * -> *) contextualized a.
(MonadBH m, MonadThrow m, ParseBHResponse contextualized) =>
BHRequest contextualized a -> m (ParsedEsResponse a)
tryPerformBHRequest BHRequest contextualized a
req m (ParsedEsResponse a) -> (ParsedEsResponse a -> m a) -> m a
forall a b. m a -> (a -> m b) -> m b
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= (EsError -> m a) -> (a -> m a) -> ParsedEsResponse a -> m a
forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either EsError -> m a
forall a. EsError -> m a
forall (m :: * -> *) a. MonadBH m => EsError -> m a
throwEsError a -> m a
forall a. a -> m a
forall (m :: * -> *) a. Monad m => a -> m a
return

emptyBody :: L.ByteString
emptyBody :: ByteString
emptyBody = String -> ByteString
L.pack String
""

parseUrl' :: (MonadThrow m) => Text -> m Request
parseUrl' :: forall (m :: * -> *). MonadThrow m => Text -> m Request
parseUrl' Text
t = String -> m Request
forall (m :: * -> *). MonadThrow m => String -> m Request
parseRequest ((Char -> Bool) -> String -> String
URI.escapeURIString Char -> Bool
URI.isAllowedInURI (Text -> String
T.unpack Text
t))

runBH :: BHEnv -> BH m a -> m (Either EsError a)
runBH :: forall (m :: * -> *) a. BHEnv -> BH m a -> m (Either EsError a)
runBH BHEnv
e BH m a
f = ExceptT EsError m a -> m (Either EsError a)
forall e (m :: * -> *) a. ExceptT e m a -> m (Either e a)
runExceptT (ExceptT EsError m a -> m (Either EsError a))
-> ExceptT EsError m a -> m (Either EsError a)
forall a b. (a -> b) -> a -> b
$ ReaderT BHEnv (ExceptT EsError m) a -> BHEnv -> ExceptT EsError m a
forall r (m :: * -> *) a. ReaderT r m a -> r -> m a
runReaderT (BH m a -> ReaderT BHEnv (ExceptT EsError m) a
forall (m :: * -> *) a.
BH m a -> ReaderT BHEnv (ExceptT EsError m) a
unBH BH m a
f) BHEnv
e