{-# language BangPatterns #-}
{-# language DerivingStrategies #-}
{-# language DuplicateRecordFields #-}
{-# language NamedFieldPuns #-}

module Kafka.Produce.Response.V9
  ( Response(..)
  , Topic(..)
  , Partition(..)
  , Error(..)
  , parser
  , decode
  , decodeHeaded
  ) where

import Control.Applicative (liftA2)
import Data.Bytes (Bytes)
import Data.Bytes.Parser (Parser)
import Data.Int (Int32,Int64)
import Data.Primitive (SmallArray)
import Data.Text (Text)
import Kafka.ErrorCode (ErrorCode)
import Kafka.Parser.Context (Context)
import Kafka.TaggedField (TaggedField)

import qualified Data.Bytes.Parser as Parser
import qualified Kafka.Parser.Context as Ctx
import qualified Kafka.TaggedField as TaggedField
import qualified Kafka.Parser
import qualified Kafka.Header.Response.V1 as Header

data Response = Response
  { Response -> SmallArray Topic
topics :: !(SmallArray Topic)
  , Response -> Int32
throttleTimeMilliseconds :: !Int32
  , Response -> SmallArray TaggedField
taggedFields :: !(SmallArray TaggedField)
  } deriving stock (Int -> Response -> ShowS
[Response] -> ShowS
Response -> String
(Int -> Response -> ShowS)
-> (Response -> String) -> ([Response] -> ShowS) -> Show Response
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> Response -> ShowS
showsPrec :: Int -> Response -> ShowS
$cshow :: Response -> String
show :: Response -> String
$cshowList :: [Response] -> ShowS
showList :: [Response] -> ShowS
Show)

data Topic = Topic
  { Topic -> Text
name :: !Text
  , Topic -> SmallArray Partition
partitions :: !(SmallArray Partition)
  , Topic -> SmallArray TaggedField
taggedFields :: !(SmallArray TaggedField)
  } deriving stock (Int -> Topic -> ShowS
[Topic] -> ShowS
Topic -> String
(Int -> Topic -> ShowS)
-> (Topic -> String) -> ([Topic] -> ShowS) -> Show Topic
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> Topic -> ShowS
showsPrec :: Int -> Topic -> ShowS
$cshow :: Topic -> String
show :: Topic -> String
$cshowList :: [Topic] -> ShowS
showList :: [Topic] -> ShowS
Show)

data Partition = Partition
  { Partition -> Int32
index :: !Int32
  , Partition -> ErrorCode
errorCode :: !ErrorCode
  , Partition -> Int64
baseOffset :: !Int64
  , Partition -> Int64
logAppendTimeMilliseconds :: !Int64
  , Partition -> Int64
logStartOffset :: !Int64
  , Partition -> SmallArray Error
errors :: !(SmallArray Error)
  , Partition -> Text
errorMessage :: !Text
    -- ^ Kafka documentation describes this as: "The global error message
    -- summarizing the common root cause of the records that caused the
    -- batch to be dropped". Although kafka allows both NULL and the empty
    -- string as distinct values, we decode NULL by mapping it to the
    -- empty string.
  , Partition -> SmallArray TaggedField
taggedFields :: !(SmallArray TaggedField)
  } deriving stock (Int -> Partition -> ShowS
[Partition] -> ShowS
Partition -> String
(Int -> Partition -> ShowS)
-> (Partition -> String)
-> ([Partition] -> ShowS)
-> Show Partition
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> Partition -> ShowS
showsPrec :: Int -> Partition -> ShowS
$cshow :: Partition -> String
show :: Partition -> String
$cshowList :: [Partition] -> ShowS
showList :: [Partition] -> ShowS
Show)

data Error = Error
  { Error -> Int32
index :: !Int32
  , Error -> Text
message :: !Text
  , Error -> SmallArray TaggedField
taggedFields :: !(SmallArray TaggedField)
  } deriving stock (Int -> Error -> ShowS
[Error] -> ShowS
Error -> String
(Int -> Error -> ShowS)
-> (Error -> String) -> ([Error] -> ShowS) -> Show Error
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> Error -> ShowS
showsPrec :: Int -> Error -> ShowS
$cshow :: Error -> String
show :: Error -> String
$cshowList :: [Error] -> ShowS
showList :: [Error] -> ShowS
Show)

decodeHeaded :: Bytes -> Either Context (Header.Headed Response)
decodeHeaded :: Bytes -> Either Context (Headed Response)
decodeHeaded !Bytes
b = (forall s. Parser Context s (Headed Response))
-> Bytes -> Either Context (Headed Response)
forall e a. (forall s. Parser e s a) -> Bytes -> Either e a
Parser.parseBytesEither
  ((Header -> Response -> Headed Response)
-> Parser Context s Header
-> Parser Context s Response
-> Parser Context s (Headed Response)
forall a b c.
(a -> b -> c)
-> Parser Context s a -> Parser Context s b -> Parser Context s c
forall (f :: * -> *) a b c.
Applicative f =>
(a -> b -> c) -> f a -> f b -> f c
liftA2 Header -> Response -> Headed Response
forall a. Header -> a -> Headed a
Header.Headed
    (Context -> Parser Context s Header
forall s. Context -> Parser Context s Header
Header.parser Context
Ctx.Top)
    (Context -> Parser Context s Response
forall s. Context -> Parser Context s Response
parser Context
Ctx.Top Parser Context s Response
-> Parser Context s () -> Parser Context s Response
forall a b.
Parser Context s a -> Parser Context s b -> Parser Context s a
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Context -> Parser Context s ()
forall e s. e -> Parser e s ()
Parser.endOfInput Context
Ctx.End)
  ) Bytes
b

decode :: Bytes -> Either Context Response
decode :: Bytes -> Either Context Response
decode !Bytes
b = (forall s. Parser Context s Response)
-> Bytes -> Either Context Response
forall e a. (forall s. Parser e s a) -> Bytes -> Either e a
Parser.parseBytesEither (Context -> Parser Context s Response
forall s. Context -> Parser Context s Response
parser Context
Ctx.Top Parser Context s Response
-> Parser Context s () -> Parser Context s Response
forall a b.
Parser Context s a -> Parser Context s b -> Parser Context s a
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Context -> Parser Context s ()
forall e s. e -> Parser e s ()
Parser.endOfInput Context
Ctx.End) Bytes
b

parser :: Context -> Parser Context s Response
parser :: forall s. Context -> Parser Context s Response
parser Context
ctx = do
  SmallArray Topic
topics <- (Context -> Parser Context s Topic)
-> Context -> Parser Context s (SmallArray Topic)
forall s a.
(Context -> Parser Context s a)
-> Context -> Parser Context s (SmallArray a)
Kafka.Parser.compactArray Context -> Parser Context s Topic
forall s. Context -> Parser Context s Topic
parserTopic Context
ctx
  Int32
throttleTimeMilliseconds <- Context -> Parser Context s Int32
forall e s. e -> Parser e s Int32
Kafka.Parser.int32 (Field -> Context -> Context
Ctx.Field Field
Ctx.ThrottleTimeMilliseconds Context
ctx)
  SmallArray TaggedField
taggedFields <- Context -> Parser Context s (SmallArray TaggedField)
forall s. Context -> Parser Context s (SmallArray TaggedField)
TaggedField.parserMany (Field -> Context -> Context
Ctx.Field Field
Ctx.TagBuffer Context
ctx)
  Response -> Parser Context s Response
forall a. a -> Parser Context s a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Response{SmallArray Topic
$sel:topics:Response :: SmallArray Topic
topics :: SmallArray Topic
topics,Int32
$sel:throttleTimeMilliseconds:Response :: Int32
throttleTimeMilliseconds :: Int32
throttleTimeMilliseconds,SmallArray TaggedField
$sel:taggedFields:Response :: SmallArray TaggedField
taggedFields :: SmallArray TaggedField
taggedFields}

parserTopic :: Context -> Parser Context s Topic
parserTopic :: forall s. Context -> Parser Context s Topic
parserTopic Context
ctx = do
  Text
name <- Context -> Parser Context s Text
forall s. Context -> Parser Context s Text
Kafka.Parser.compactString (Field -> Context -> Context
Ctx.Field Field
Ctx.Name Context
ctx)
  SmallArray Partition
partitions <- (Context -> Parser Context s Partition)
-> Context -> Parser Context s (SmallArray Partition)
forall s a.
(Context -> Parser Context s a)
-> Context -> Parser Context s (SmallArray a)
Kafka.Parser.compactArray Context -> Parser Context s Partition
forall s. Context -> Parser Context s Partition
parserPartition
    (Field -> Context -> Context
Ctx.Field Field
Ctx.Partitions Context
ctx)
  SmallArray TaggedField
taggedFields <- Context -> Parser Context s (SmallArray TaggedField)
forall s. Context -> Parser Context s (SmallArray TaggedField)
TaggedField.parserMany (Field -> Context -> Context
Ctx.Field Field
Ctx.TagBuffer Context
ctx)
  Topic -> Parser Context s Topic
forall a. a -> Parser Context s a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Topic{Text
$sel:name:Topic :: Text
name :: Text
name,SmallArray Partition
$sel:partitions:Topic :: SmallArray Partition
partitions :: SmallArray Partition
partitions,SmallArray TaggedField
$sel:taggedFields:Topic :: SmallArray TaggedField
taggedFields :: SmallArray TaggedField
taggedFields}

parserPartition :: Context -> Parser Context s Partition
parserPartition :: forall s. Context -> Parser Context s Partition
parserPartition Context
ctx = do
  Int32
index <- Context -> Parser Context s Int32
forall e s. e -> Parser e s Int32
Kafka.Parser.int32 (Field -> Context -> Context
Ctx.Field Field
Ctx.Ix Context
ctx)
  ErrorCode
errorCode <- Context -> Parser Context s ErrorCode
forall e s. e -> Parser e s ErrorCode
Kafka.Parser.errorCode (Field -> Context -> Context
Ctx.Field Field
Ctx.ErrorCode Context
ctx)
  Int64
baseOffset <- Context -> Parser Context s Int64
forall e s. e -> Parser e s Int64
Kafka.Parser.int64 (Field -> Context -> Context
Ctx.Field Field
Ctx.BaseOffset Context
ctx)
  Int64
logAppendTimeMilliseconds <- Context -> Parser Context s Int64
forall e s. e -> Parser e s Int64
Kafka.Parser.int64 (Field -> Context -> Context
Ctx.Field Field
Ctx.LogAppendTimeMilliseconds Context
ctx)
  Int64
logStartOffset <- Context -> Parser Context s Int64
forall e s. e -> Parser e s Int64
Kafka.Parser.int64 (Field -> Context -> Context
Ctx.Field Field
Ctx.LogStartOffset Context
ctx)
  SmallArray Error
errors <- (Context -> Parser Context s Error)
-> Context -> Parser Context s (SmallArray Error)
forall s a.
(Context -> Parser Context s a)
-> Context -> Parser Context s (SmallArray a)
Kafka.Parser.compactArray Context -> Parser Context s Error
forall s. Context -> Parser Context s Error
parserError
    (Field -> Context -> Context
Ctx.Field Field
Ctx.Errors Context
ctx)
  Text
errorMessage <- Context -> Parser Context s Text
forall s. Context -> Parser Context s Text
Kafka.Parser.compactString (Field -> Context -> Context
Ctx.Field Field
Ctx.ErrorMessage Context
ctx)
  SmallArray TaggedField
taggedFields <- Context -> Parser Context s (SmallArray TaggedField)
forall s. Context -> Parser Context s (SmallArray TaggedField)
TaggedField.parserMany (Field -> Context -> Context
Ctx.Field Field
Ctx.TagBuffer Context
ctx)
  Partition -> Parser Context s Partition
forall a. a -> Parser Context s a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Partition
    { Int32
$sel:index:Partition :: Int32
index :: Int32
index
    , ErrorCode
$sel:errorCode:Partition :: ErrorCode
errorCode :: ErrorCode
errorCode
    , Int64
$sel:baseOffset:Partition :: Int64
baseOffset :: Int64
baseOffset
    , Int64
$sel:logAppendTimeMilliseconds:Partition :: Int64
logAppendTimeMilliseconds :: Int64
logAppendTimeMilliseconds
    , Int64
$sel:logStartOffset:Partition :: Int64
logStartOffset :: Int64
logStartOffset
    , SmallArray Error
$sel:errors:Partition :: SmallArray Error
errors :: SmallArray Error
errors
    , Text
$sel:errorMessage:Partition :: Text
errorMessage :: Text
errorMessage
    , SmallArray TaggedField
$sel:taggedFields:Partition :: SmallArray TaggedField
taggedFields :: SmallArray TaggedField
taggedFields
    }

parserError :: Context -> Parser Context s Error
parserError :: forall s. Context -> Parser Context s Error
parserError Context
ctx = do
  Int32
index <- Context -> Parser Context s Int32
forall e s. e -> Parser e s Int32
Kafka.Parser.int32 (Field -> Context -> Context
Ctx.Field Field
Ctx.Ix Context
ctx)
  Text
message <- Context -> Parser Context s Text
forall s. Context -> Parser Context s Text
Kafka.Parser.compactString (Field -> Context -> Context
Ctx.Field Field
Ctx.Message Context
ctx)
  SmallArray TaggedField
taggedFields <- Context -> Parser Context s (SmallArray TaggedField)
forall s. Context -> Parser Context s (SmallArray TaggedField)
TaggedField.parserMany (Field -> Context -> Context
Ctx.Field Field
Ctx.TagBuffer Context
ctx)
  Error -> Parser Context s Error
forall a. a -> Parser Context s a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Error{Int32
$sel:index:Error :: Int32
index :: Int32
index,Text
$sel:message:Error :: Text
message :: Text
message,SmallArray TaggedField
$sel:taggedFields:Error :: SmallArray TaggedField
taggedFields :: SmallArray TaggedField
taggedFields}