{-# OPTIONS_GHC -fno-warn-orphans #-}
module Binrep.Put.Mason where
import Mason.Builder qualified as Mason
import Data.ByteString qualified as B
import Data.Word
import Data.Int
import Data.Void ( Void, absurd )
import GHC.Generics
import Generic.Data.Function.FoldMap
import Generic.Data.Function.Common
import Generic.Data.Rep.Assert
type Builder = Mason.BuilderFor Mason.StrictByteStringBackend
class Put a where
put :: a -> Builder
runPut :: Put a => a -> B.ByteString
runPut :: forall a. Put a => a -> ByteString
runPut = Builder -> ByteString
runBuilder (Builder -> ByteString) -> (a -> Builder) -> a -> ByteString
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> Builder
forall a. Put a => a -> Builder
put
runBuilder :: Builder -> B.ByteString
runBuilder :: Builder -> ByteString
runBuilder = Builder -> ByteString
Mason.toStrictByteString
instance GenericFoldMap Builder where
type GenericFoldMapC Builder a = Put a
genericFoldMapF :: forall a. GenericFoldMapC Builder a => a -> Builder
genericFoldMapF = a -> Builder
forall a. Put a => a -> Builder
put
putGenericNonSum
:: forall {cd} {f} {asserts} a
. ( Generic a, Rep a ~ D1 cd f, GFoldMapNonSum Builder f
, asserts ~ '[ 'NoEmpty, 'NoSum], ApplyGCAsserts asserts f)
=> a -> Builder
putGenericNonSum :: forall {cd :: Meta} {f :: Type -> Type} {asserts :: [GCAssert]} a.
(Generic a, Rep a ~ D1 cd f, GFoldMapNonSum Builder f,
asserts ~ '[ 'NoEmpty, 'NoSum], ApplyGCAsserts asserts f) =>
a -> Builder
putGenericNonSum = forall (asserts :: [GCAssert]) m a.
(Generic a, Rep a ~ D1 cd f, GFoldMapNonSum m f,
ApplyGCAsserts asserts f) =>
a -> m
forall {cd :: Meta} {f :: Type -> Type} (asserts :: [GCAssert]) m
a.
(Generic a, Rep a ~ D1 cd f, GFoldMapNonSum m f,
ApplyGCAsserts asserts f) =>
a -> m
genericFoldMapNonSum @asserts
putGenericSum
:: forall {cd} {f} {asserts} a
. (Generic a, Rep a ~ D1 cd f, GFoldMapSum 'SumOnly Builder f
, asserts ~ '[ 'NoEmpty, 'NeedSum], ApplyGCAsserts asserts f)
=> (String -> Builder) -> a -> Builder
putGenericSum :: forall {cd :: Meta} {f :: Type -> Type} {asserts :: [GCAssert]} a.
(Generic a, Rep a ~ D1 cd f, GFoldMapSum 'SumOnly Builder f,
asserts ~ '[ 'NoEmpty, 'NeedSum], ApplyGCAsserts asserts f) =>
(String -> Builder) -> a -> Builder
putGenericSum = forall {cd :: Meta} {f :: Type -> Type} (opts :: SumOpts)
(asserts :: [GCAssert]) m a.
(Generic a, Rep a ~ D1 cd f, GFoldMapSum opts m f,
ApplyGCAsserts asserts f) =>
(String -> m) -> a -> m
forall (opts :: SumOpts) (asserts :: [GCAssert]) m a.
(Generic a, Rep a ~ D1 cd f, GFoldMapSum opts m f,
ApplyGCAsserts asserts f) =>
(String -> m) -> a -> m
genericFoldMapSum @'SumOnly @asserts
instance Put Void where
put :: Void -> Builder
put = Void -> Builder
forall a. Void -> a
absurd
instance Put a => Put [a] where
put :: [a] -> Builder
put = [Builder] -> Builder
forall a. Monoid a => [a] -> a
mconcat ([Builder] -> Builder) -> ([a] -> [Builder]) -> [a] -> Builder
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (a -> Builder) -> [a] -> [Builder]
forall a b. (a -> b) -> [a] -> [b]
map a -> Builder
forall a. Put a => a -> Builder
put
instance (Put a, Put b) => Put (a, b) where
put :: (a, b) -> Builder
put (a
a, b
b) = a -> Builder
forall a. Put a => a -> Builder
put a
a Builder -> Builder -> Builder
forall a. Semigroup a => a -> a -> a
<> b -> Builder
forall a. Put a => a -> Builder
put b
b
instance Put B.ByteString where
put :: ByteString -> Builder
put = ByteString -> Builder
forall s. Buildable s => ByteString -> BuilderFor s
Mason.byteString
{-# INLINE put #-}
instance Put Word8 where
put :: Word8 -> Builder
put Word8
w = Word8 -> Builder
Mason.word8 Word8
w
{-# INLINE put #-}
instance Put Int8 where
put :: Int8 -> Builder
put Int8
w = Int8 -> Builder
Mason.int8 Int8
w
{-# INLINE put #-}
class PutWith r a where
putWith :: r -> a -> Either String Builder
default putWith :: Put a => r -> a -> Either String Builder
putWith r
_ = a -> Either String Builder
forall a. Put a => a -> Either String Builder
putWithout
putWithout :: Put a => a -> Either String Builder
putWithout :: forall a. Put a => a -> Either String Builder
putWithout = Builder -> Either String Builder
forall a b. b -> Either a b
Right (Builder -> Either String Builder)
-> (a -> Builder) -> a -> Either String Builder
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> Builder
forall a. Put a => a -> Builder
put
instance Put a => PutWith r [a]
runPutWith :: PutWith r a => r -> a -> Either String B.ByteString
runPutWith :: forall r a. PutWith r a => r -> a -> Either String ByteString
runPutWith r
r a
a = case r -> a -> Either String Builder
forall r a. PutWith r a => r -> a -> Either String Builder
putWith r
r a
a of Left String
e -> String -> Either String ByteString
forall a b. a -> Either a b
Left String
e
Right Builder
x -> ByteString -> Either String ByteString
forall a b. b -> Either a b
Right (ByteString -> Either String ByteString)
-> ByteString -> Either String ByteString
forall a b. (a -> b) -> a -> b
$ Builder -> ByteString
runBuilder Builder
x