{-# LANGUAGE
AllowAmbiguousTypes
, DeriveFoldable
, DeriveFunctor
, DeriveGeneric
, DeriveTraversable
, FlexibleContexts
, FlexibleInstances
, GADTs
, LambdaCase
, MultiParamTypeClasses
, ScopedTypeVariables
, TypeApplications
, TypeInType
, TypeOperators
, UndecidableInstances
#-}
module Squeal.PostgreSQL.Binary
(
ToParam (..)
, ToColumnParam (..)
, ToParams (..)
, FromValue (..)
, FromColumnValue (..)
, FromRow (..)
, Only (..)
) where
import BinaryParser
import ByteString.StrictBuilder
import Data.Aeson hiding (Null)
import Data.Int
import Data.Kind
import Data.Monoid hiding (All)
import Data.Scientific
import Data.Time
import Data.UUID.Types
import Data.Vector (Vector)
import Data.Word
import Generics.SOP
import GHC.TypeLits
import Network.IP.Addr
import qualified Data.ByteString.Lazy as Lazy
import qualified Data.ByteString as Strict hiding (pack, unpack)
import qualified Data.Text.Lazy as Lazy
import qualified Data.Text as Strict
import qualified Data.Vector as Vector
import qualified GHC.Generics as GHC
import qualified PostgreSQL.Binary.Decoding as Decoding
import qualified PostgreSQL.Binary.Encoding as Encoding
import Squeal.PostgreSQL.Schema
class ToParam (x :: Type) (pg :: PGType) where
toParam :: x -> K Encoding.Encoding pg
instance ToParam Bool 'PGbool where toParam = K . Encoding.bool
instance ToParam Int16 'PGint2 where toParam = K . Encoding.int2_int16
instance ToParam Word16 'PGint2 where toParam = K . Encoding.int2_word16
instance ToParam Int32 'PGint4 where toParam = K . Encoding.int4_int32
instance ToParam Word32 'PGint4 where toParam = K . Encoding.int4_word32
instance ToParam Int64 'PGint8 where toParam = K . Encoding.int8_int64
instance ToParam Word64 'PGint8 where toParam = K . Encoding.int8_word64
instance ToParam Float 'PGfloat4 where toParam = K . Encoding.float4
instance ToParam Double 'PGfloat8 where toParam = K . Encoding.float8
instance ToParam Scientific 'PGnumeric where toParam = K . Encoding.numeric
instance ToParam UUID 'PGuuid where toParam = K . Encoding.uuid
instance ToParam (NetAddr IP) 'PGinet where toParam = K . Encoding.inet
instance ToParam Char ('PGchar 1) where toParam = K . Encoding.char_utf8
instance ToParam Strict.Text 'PGtext where toParam = K . Encoding.text_strict
instance ToParam Lazy.Text 'PGtext where toParam = K . Encoding.text_lazy
instance ToParam Strict.ByteString 'PGbytea where
toParam = K . Encoding.bytea_strict
instance ToParam Lazy.ByteString 'PGbytea where
toParam = K . Encoding.bytea_lazy
instance ToParam Day 'PGdate where toParam = K . Encoding.date
instance ToParam TimeOfDay 'PGtime where toParam = K . Encoding.time_int
instance ToParam (TimeOfDay, TimeZone) 'PGtimetz where
toParam = K . Encoding.timetz_int
instance ToParam LocalTime 'PGtimestamp where
toParam = K . Encoding.timestamp_int
instance ToParam UTCTime 'PGtimestamptz where
toParam = K . Encoding.timestamptz_int
instance ToParam DiffTime 'PGinterval where toParam = K . Encoding.interval_int
instance ToParam Value 'PGjson where toParam = K . Encoding.json_ast
instance ToParam Value 'PGjsonb where toParam = K . Encoding.jsonb_ast
instance (HasOid pg, ToParam x pg)
=> ToParam (Vector (Maybe x)) ('PGvararray pg) where
toParam = K . Encoding.nullableArray_vector
(oid @pg) (unK . toParam @x @pg)
instance
( IsEnumType x
, HasDatatypeInfo x
, LabelsFrom x ~ labels
) => ToParam x ('PGenum labels) where
toParam =
let
gshowConstructor :: NP ConstructorInfo xss -> SOP I xss -> String
gshowConstructor Nil _ = ""
gshowConstructor (constructor :* _) (SOP (Z _)) =
constructorName constructor
gshowConstructor (_ :* constructors) (SOP (S xs)) =
gshowConstructor constructors (SOP xs)
in
K . Encoding.text_strict
. Strict.pack
. gshowConstructor (constructorInfo (datatypeInfo (Proxy @x)))
. from
instance
( SListI fields
, MapMaybes xs
, IsProductType x (Maybes xs)
, AllZip ToAliasedParam xs fields
, FieldNamesFrom x ~ AliasesOf fields
, All HasAliasedOid fields
) => ToParam x ('PGcomposite fields) where
toParam =
let
encoders = htrans (Proxy @ToAliasedParam) toAliasedParam
composite
:: All HasAliasedOid row
=> NP (K (Maybe Encoding.Encoding)) row
-> K Encoding.Encoding ('PGcomposite row)
composite fields = K $
int32BE (fromIntegral (lengthSList (Proxy @xs))) <>
let
each
:: HasAliasedOid field
=> K (Maybe Encoding.Encoding) field
-> Encoding.Encoding
each (K field :: K (Maybe Encoding.Encoding) field) =
word32BE (aliasedOid @field)
<> case field of
Nothing -> int64BE (-1)
Just value ->
int32BE (fromIntegral (builderLength value))
<> value
in
hcfoldMap (Proxy @HasAliasedOid) each fields
in
composite . encoders . unMaybes . unZ . unSOP . from
class HasAliasedOid (field :: (Symbol, PGType)) where aliasedOid :: Word32
instance HasOid ty => HasAliasedOid (alias ::: ty) where aliasedOid = oid @ty
class ToAliasedParam (x :: Type) (field :: (Symbol, PGType)) where
toAliasedParam :: Maybe x -> K (Maybe Encoding.Encoding) field
instance ToParam x ty => ToAliasedParam x (alias ::: ty) where
toAliasedParam = \case
Nothing -> K Nothing
Just x -> K . Just . unK $ toParam @x @ty x
class ToColumnParam (x :: Type) (ty :: NullityType) where
toColumnParam :: x -> K (Maybe Strict.ByteString) ty
instance ToParam x pg => ToColumnParam x ('NotNull pg) where
toColumnParam = K . Just . Encoding.encodingBytes . unK . toParam @x @pg
instance ToParam x pg => ToColumnParam (Maybe x) ('Null pg) where
toColumnParam = K . fmap (Encoding.encodingBytes . unK . toParam @x @pg)
class SListI tys => ToParams (x :: Type) (tys :: [NullityType]) where
toParams :: x -> NP (K (Maybe Strict.ByteString)) tys
instance (SListI tys, IsProductType x xs, AllZip ToColumnParam xs tys)
=> ToParams x tys where
toParams
= htrans (Proxy @ToColumnParam) (toColumnParam . unI)
. unZ . unSOP . from
class FromValue (pg :: PGType) (y :: Type) where
fromValue :: proxy pg -> Decoding.Value y
instance FromValue 'PGbool Bool where fromValue _ = Decoding.bool
instance FromValue 'PGint2 Int16 where fromValue _ = Decoding.int
instance FromValue 'PGint4 Int32 where fromValue _ = Decoding.int
instance FromValue 'PGint8 Int64 where fromValue _ = Decoding.int
instance FromValue 'PGfloat4 Float where fromValue _ = Decoding.float4
instance FromValue 'PGfloat8 Double where fromValue _ = Decoding.float8
instance FromValue 'PGnumeric Scientific where fromValue _ = Decoding.numeric
instance FromValue 'PGuuid UUID where fromValue _ = Decoding.uuid
instance FromValue 'PGinet (NetAddr IP) where fromValue _ = Decoding.inet
instance FromValue ('PGchar 1) Char where fromValue _ = Decoding.char
instance FromValue 'PGtext Strict.Text where fromValue _ = Decoding.text_strict
instance FromValue 'PGtext Lazy.Text where fromValue _ = Decoding.text_lazy
instance FromValue 'PGbytea Strict.ByteString where
fromValue _ = Decoding.bytea_strict
instance FromValue 'PGbytea Lazy.ByteString where
fromValue _ = Decoding.bytea_lazy
instance FromValue 'PGdate Day where fromValue _ = Decoding.date
instance FromValue 'PGtime TimeOfDay where fromValue _ = Decoding.time_int
instance FromValue 'PGtimetz (TimeOfDay, TimeZone) where
fromValue _ = Decoding.timetz_int
instance FromValue 'PGtimestamp LocalTime where
fromValue _ = Decoding.timestamp_int
instance FromValue 'PGtimestamptz UTCTime where
fromValue _ = Decoding.timestamptz_int
instance FromValue 'PGinterval DiffTime where
fromValue _ = Decoding.interval_int
instance FromValue 'PGjson Value where fromValue _ = Decoding.json_ast
instance FromValue 'PGjsonb Value where fromValue _ = Decoding.jsonb_ast
instance FromValue pg y => FromValue ('PGvararray pg) (Vector (Maybe y)) where
fromValue _ = Decoding.array
(Decoding.dimensionArray Vector.replicateM
(Decoding.nullableValueArray (fromValue (Proxy @pg))))
instance FromValue pg y => FromValue ('PGfixarray n pg) (Vector (Maybe y)) where
fromValue _ = Decoding.array
(Decoding.dimensionArray Vector.replicateM
(Decoding.nullableValueArray (fromValue (Proxy @pg))))
instance
( IsEnumType y
, HasDatatypeInfo y
, LabelsFrom y ~ labels
) => FromValue ('PGenum labels) y where
fromValue _ =
let
greadConstructor
:: All ((~) '[]) xss
=> NP ConstructorInfo xss
-> String
-> Maybe (SOP I xss)
greadConstructor Nil _ = Nothing
greadConstructor (constructor :* constructors) name =
if name == constructorName constructor
then Just (SOP (Z Nil))
else SOP . S . unSOP <$> greadConstructor constructors name
in
Decoding.enum
$ fmap to
. greadConstructor (constructorInfo (datatypeInfo (Proxy @y)))
. Strict.unpack
instance
( SListI fields
, MapMaybes ys
, IsProductType y (Maybes ys)
, AllZip FromAliasedValue fields ys
, FieldNamesFrom y ~ AliasesOf fields
) => FromValue ('PGcomposite fields) y where
fromValue =
let
decoders
:: forall pgs zs proxy
. AllZip FromAliasedValue pgs zs
=> proxy ('PGcomposite pgs)
-> NP Decoding.Value zs
decoders _ = htrans (Proxy @FromAliasedValue) fromAliasedValue
(hpure Proxy :: NP Proxy pgs)
composite fields = do
unitOfSize 4
let
each field = do
unitOfSize 4
len <- sized 4 Decoding.int
if len == -1 then return Nothing else Just <$> sized len field
htraverse' each fields
in fmap (to . SOP . Z . maybes) . composite . decoders
class FromAliasedValue (pg :: (Symbol,PGType)) (y :: Type) where
fromAliasedValue :: proxy pg -> Decoding.Value y
instance FromValue pg y => FromAliasedValue (alias ::: pg) y where
fromAliasedValue _ = fromValue (Proxy @pg)
class FromColumnValue (colty :: (Symbol,NullityType)) (y :: Type) where
fromColumnValue :: K (Maybe Strict.ByteString) colty -> y
instance FromValue pg y
=> FromColumnValue (column ::: ('NotNull pg)) y where
fromColumnValue = \case
K Nothing -> error "fromColumnValue: saw NULL when expecting NOT NULL"
K (Just bs) ->
let
errOrValue =
Decoding.valueParser (fromValue @pg @y Proxy) bs
err str = error $ "fromColumnValue: " ++ Strict.unpack str
in
either err id errOrValue
instance FromValue pg y
=> FromColumnValue (column ::: ('Null pg)) (Maybe y) where
fromColumnValue (K nullOrBytes)
= either err id
. Decoding.valueParser (fromValue @pg @y Proxy)
<$> nullOrBytes
where
err str = error $ "fromColumnValue: " ++ Strict.unpack str
class SListI results => FromRow (results :: RelationType) y where
fromRow :: NP (K (Maybe Strict.ByteString)) results -> y
instance
( SListI results
, IsProductType y ys
, AllZip FromColumnValue results ys
, FieldNamesFrom y ~ AliasesOf results
) => FromRow results y where
fromRow
= to . SOP . Z . htrans (Proxy @FromColumnValue) (I . fromColumnValue)
newtype Only x = Only { fromOnly :: x }
deriving (Functor,Foldable,Traversable,Eq,Ord,Read,Show,GHC.Generic)
instance Generic (Only x)
instance HasDatatypeInfo (Only x)