Client.hs

-- | High-level client implementation for JetStream.
module JetStream.Client
  ( newJetStream
  , JetStream (..)
  , JetStreamOption
  , withDomain
  , withRequestTimeoutMicros
  ) where

import qualified API                        as Nats
import           JetStream.API              (JetStream (..))
import qualified JetStream.Consumer         as Consumer
import qualified JetStream.Message          as Message
import           JetStream.Options
    ( JetStreamOption
    , newJetStreamContext
    , withDomain
    , withRequestTimeoutMicros
    )
import qualified JetStream.Protocol.Request as Request
import qualified JetStream.Protocol.Subject as Subject
import qualified JetStream.Publish          as Publish
import qualified JetStream.Stream           as Stream

-- | Build a JetStream capability record from an existing NATS client.
newJetStream :: Nats.Client -> [JetStreamOption] -> JetStream
newJetStream client options =
  let ctx = newJetStreamContext client options
      consumerAPI = Consumer.consumerAPI ctx
  in JetStream
    { streams = Stream.streamAPI ctx
    , consumers = consumerAPI
    , publisher = Publish.publishAPI ctx
    , messages = Message.messageAPI ctx consumerAPI
    , accountInfo =
        Request.requestJSON ctx (Subject.accountInfoSubject ctx) Nothing
    }