{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ViewPatterns #-}

module Database.Persist.Postgresql.Internal
    ( P(..)
    , PgInterval(..)
    , getGetter
    ) where

import qualified Database.PostgreSQL.Simple as PG
import qualified Database.PostgreSQL.Simple.FromField as PGFF
import qualified Database.PostgreSQL.Simple.Internal as PG
import qualified Database.PostgreSQL.Simple.ToField as PGTF
import qualified Database.PostgreSQL.Simple.TypeInfo.Static as PS
import qualified Database.PostgreSQL.Simple.Types as PG

import qualified Blaze.ByteString.Builder.Char8 as BBB
import qualified Data.Attoparsec.ByteString.Char8 as P
import Data.Bits ((.&.))
import Data.ByteString (ByteString)
import qualified Data.ByteString.Builder as BB
import qualified Data.ByteString.Char8 as B8
import Data.Char (ord)
import Data.Data (Typeable)
import Data.Fixed (Fixed(..), Pico)
import Data.Int (Int64)
import qualified Data.IntMap as I
import Data.Maybe (fromMaybe)
import Data.String.Conversions.Monomorphic (toStrictByteString)
import Data.Text (Text)
import qualified Data.Text as T
import qualified Data.Text.Encoding as T
import Data.Time (NominalDiffTime, localTimeToUTC, utc)

import Database.Persist.Sql

-- | Newtype used to avoid orphan instances for @postgresql-simple@ classes.
--
-- @since 2.13.2.0
newtype P = P { P -> PersistValue
unP :: PersistValue }

instance PGTF.ToField P where
    toField :: P -> Action
toField (P (PersistText Text
t))        = forall a. ToField a => a -> Action
PGTF.toField Text
t
    toField (P (PersistByteString StrictByteString
bs)) = forall a. ToField a => a -> Action
PGTF.toField (forall a. a -> Binary a
PG.Binary StrictByteString
bs)
    toField (P (PersistInt64 Int64
i))       = forall a. ToField a => a -> Action
PGTF.toField Int64
i
    toField (P (PersistDouble Double
d))      = forall a. ToField a => a -> Action
PGTF.toField Double
d
    toField (P (PersistRational Rational
r))    = Builder -> Action
PGTF.Plain forall a b. (a -> b) -> a -> b
$
                                         String -> Builder
BBB.fromString forall a b. (a -> b) -> a -> b
$
                                         forall a. Show a => a -> String
show (forall a. Fractional a => Rational -> a
fromRational Rational
r :: Pico) --  FIXME: Too Ambigous, can not select precision without information about field
    toField (P (PersistBool Bool
b))        = forall a. ToField a => a -> Action
PGTF.toField Bool
b
    toField (P (PersistDay Day
d))         = forall a. ToField a => a -> Action
PGTF.toField Day
d
    toField (P (PersistTimeOfDay TimeOfDay
t))   = forall a. ToField a => a -> Action
PGTF.toField TimeOfDay
t
    toField (P (PersistUTCTime UTCTime
t))     = forall a. ToField a => a -> Action
PGTF.toField UTCTime
t
    toField (P PersistValue
PersistNull)            = forall a. ToField a => a -> Action
PGTF.toField Null
PG.Null
    toField (P (PersistList [PersistValue]
l))        = forall a. ToField a => a -> Action
PGTF.toField forall a b. (a -> b) -> a -> b
$ [PersistValue] -> Text
listToJSON [PersistValue]
l
    toField (P (PersistMap [(Text, PersistValue)]
m))         = forall a. ToField a => a -> Action
PGTF.toField forall a b. (a -> b) -> a -> b
$ [(Text, PersistValue)] -> Text
mapToJSON [(Text, PersistValue)]
m
    toField (P (PersistLiteral_ LiteralType
DbSpecific StrictByteString
s))  = forall a. ToField a => a -> Action
PGTF.toField (StrictByteString -> Unknown
Unknown StrictByteString
s)
    toField (P (PersistLiteral_ LiteralType
Unescaped StrictByteString
l))     = forall a. ToField a => a -> Action
PGTF.toField (StrictByteString -> UnknownLiteral
UnknownLiteral StrictByteString
l)
    toField (P (PersistLiteral_ LiteralType
Escaped StrictByteString
e)) = forall a. ToField a => a -> Action
PGTF.toField (StrictByteString -> Unknown
Unknown StrictByteString
e)
    toField (P (PersistArray [PersistValue]
a))       = forall a. ToField a => a -> Action
PGTF.toField forall a b. (a -> b) -> a -> b
$ forall a. [a] -> PGArray a
PG.PGArray forall a b. (a -> b) -> a -> b
$ PersistValue -> P
P forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [PersistValue]
a
    toField (P (PersistObjectId StrictByteString
_))    =
        forall a. HasCallStack => String -> a
error String
"Refusing to serialize a PersistObjectId to a PostgreSQL value"

instance PGFF.FromField P where
    fromField :: FieldParser P
fromField Field
field Maybe StrictByteString
mdata = forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap PersistValue -> P
P forall a b. (a -> b) -> a -> b
$ case Maybe StrictByteString
mdata of
      -- If we try to simply decode based on oid, we will hit unexpected null
      -- errors.
      Maybe StrictByteString
Nothing -> forall (f :: * -> *) a. Applicative f => a -> f a
pure PersistValue
PersistNull
      Maybe StrictByteString
data' -> Oid -> Getter PersistValue
getGetter (Field -> Oid
PGFF.typeOid Field
field) Field
field Maybe StrictByteString
data'

newtype Unknown = Unknown { Unknown -> StrictByteString
unUnknown :: ByteString }
  deriving (Unknown -> Unknown -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: Unknown -> Unknown -> Bool
$c/= :: Unknown -> Unknown -> Bool
== :: Unknown -> Unknown -> Bool
$c== :: Unknown -> Unknown -> Bool
Eq, Int -> Unknown -> ShowS
[Unknown] -> ShowS
Unknown -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [Unknown] -> ShowS
$cshowList :: [Unknown] -> ShowS
show :: Unknown -> String
$cshow :: Unknown -> String
showsPrec :: Int -> Unknown -> ShowS
$cshowsPrec :: Int -> Unknown -> ShowS
Show, ReadPrec [Unknown]
ReadPrec Unknown
Int -> ReadS Unknown
ReadS [Unknown]
forall a.
(Int -> ReadS a)
-> ReadS [a] -> ReadPrec a -> ReadPrec [a] -> Read a
readListPrec :: ReadPrec [Unknown]
$creadListPrec :: ReadPrec [Unknown]
readPrec :: ReadPrec Unknown
$creadPrec :: ReadPrec Unknown
readList :: ReadS [Unknown]
$creadList :: ReadS [Unknown]
readsPrec :: Int -> ReadS Unknown
$creadsPrec :: Int -> ReadS Unknown
Read, Eq Unknown
Unknown -> Unknown -> Bool
Unknown -> Unknown -> Ordering
Unknown -> Unknown -> Unknown
forall a.
Eq a
-> (a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
min :: Unknown -> Unknown -> Unknown
$cmin :: Unknown -> Unknown -> Unknown
max :: Unknown -> Unknown -> Unknown
$cmax :: Unknown -> Unknown -> Unknown
>= :: Unknown -> Unknown -> Bool
$c>= :: Unknown -> Unknown -> Bool
> :: Unknown -> Unknown -> Bool
$c> :: Unknown -> Unknown -> Bool
<= :: Unknown -> Unknown -> Bool
$c<= :: Unknown -> Unknown -> Bool
< :: Unknown -> Unknown -> Bool
$c< :: Unknown -> Unknown -> Bool
compare :: Unknown -> Unknown -> Ordering
$ccompare :: Unknown -> Unknown -> Ordering
Ord)

instance PGFF.FromField Unknown where
    fromField :: FieldParser Unknown
fromField Field
f Maybe StrictByteString
mdata =
      case Maybe StrictByteString
mdata of
        Maybe StrictByteString
Nothing  -> forall a err.
(Typeable a, Exception err) =>
(String -> Maybe Oid -> String -> String -> String -> err)
-> Field -> String -> Conversion a
PGFF.returnError String -> Maybe Oid -> String -> String -> String -> ResultError
PGFF.UnexpectedNull Field
f String
"Database.Persist.Postgresql/PGFF.FromField Unknown"
        Just StrictByteString
dat -> forall (m :: * -> *) a. Monad m => a -> m a
return (StrictByteString -> Unknown
Unknown StrictByteString
dat)

instance PGTF.ToField Unknown where
    toField :: Unknown -> Action
toField (Unknown StrictByteString
a) = StrictByteString -> Action
PGTF.Escape StrictByteString
a

newtype UnknownLiteral = UnknownLiteral { UnknownLiteral -> StrictByteString
unUnknownLiteral :: ByteString }
  deriving (UnknownLiteral -> UnknownLiteral -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: UnknownLiteral -> UnknownLiteral -> Bool
$c/= :: UnknownLiteral -> UnknownLiteral -> Bool
== :: UnknownLiteral -> UnknownLiteral -> Bool
$c== :: UnknownLiteral -> UnknownLiteral -> Bool
Eq, Int -> UnknownLiteral -> ShowS
[UnknownLiteral] -> ShowS
UnknownLiteral -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [UnknownLiteral] -> ShowS
$cshowList :: [UnknownLiteral] -> ShowS
show :: UnknownLiteral -> String
$cshow :: UnknownLiteral -> String
showsPrec :: Int -> UnknownLiteral -> ShowS
$cshowsPrec :: Int -> UnknownLiteral -> ShowS
Show, ReadPrec [UnknownLiteral]
ReadPrec UnknownLiteral
Int -> ReadS UnknownLiteral
ReadS [UnknownLiteral]
forall a.
(Int -> ReadS a)
-> ReadS [a] -> ReadPrec a -> ReadPrec [a] -> Read a
readListPrec :: ReadPrec [UnknownLiteral]
$creadListPrec :: ReadPrec [UnknownLiteral]
readPrec :: ReadPrec UnknownLiteral
$creadPrec :: ReadPrec UnknownLiteral
readList :: ReadS [UnknownLiteral]
$creadList :: ReadS [UnknownLiteral]
readsPrec :: Int -> ReadS UnknownLiteral
$creadsPrec :: Int -> ReadS UnknownLiteral
Read, Eq UnknownLiteral
UnknownLiteral -> UnknownLiteral -> Bool
UnknownLiteral -> UnknownLiteral -> Ordering
UnknownLiteral -> UnknownLiteral -> UnknownLiteral
forall a.
Eq a
-> (a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
min :: UnknownLiteral -> UnknownLiteral -> UnknownLiteral
$cmin :: UnknownLiteral -> UnknownLiteral -> UnknownLiteral
max :: UnknownLiteral -> UnknownLiteral -> UnknownLiteral
$cmax :: UnknownLiteral -> UnknownLiteral -> UnknownLiteral
>= :: UnknownLiteral -> UnknownLiteral -> Bool
$c>= :: UnknownLiteral -> UnknownLiteral -> Bool
> :: UnknownLiteral -> UnknownLiteral -> Bool
$c> :: UnknownLiteral -> UnknownLiteral -> Bool
<= :: UnknownLiteral -> UnknownLiteral -> Bool
$c<= :: UnknownLiteral -> UnknownLiteral -> Bool
< :: UnknownLiteral -> UnknownLiteral -> Bool
$c< :: UnknownLiteral -> UnknownLiteral -> Bool
compare :: UnknownLiteral -> UnknownLiteral -> Ordering
$ccompare :: UnknownLiteral -> UnknownLiteral -> Ordering
Ord, Typeable)

instance PGFF.FromField UnknownLiteral where
    fromField :: FieldParser UnknownLiteral
fromField Field
f Maybe StrictByteString
mdata =
      case Maybe StrictByteString
mdata of
        Maybe StrictByteString
Nothing  -> forall a err.
(Typeable a, Exception err) =>
(String -> Maybe Oid -> String -> String -> String -> err)
-> Field -> String -> Conversion a
PGFF.returnError String -> Maybe Oid -> String -> String -> String -> ResultError
PGFF.UnexpectedNull Field
f String
"Database.Persist.Postgresql/PGFF.FromField UnknownLiteral"
        Just StrictByteString
dat -> forall (m :: * -> *) a. Monad m => a -> m a
return (StrictByteString -> UnknownLiteral
UnknownLiteral StrictByteString
dat)

instance PGTF.ToField UnknownLiteral where
    toField :: UnknownLiteral -> Action
toField (UnknownLiteral StrictByteString
a) = Builder -> Action
PGTF.Plain forall a b. (a -> b) -> a -> b
$ StrictByteString -> Builder
BB.byteString StrictByteString
a

type Getter a = PGFF.FieldParser a

convertPV :: PGFF.FromField a => (a -> b) -> Getter b
convertPV :: forall a b. FromField a => (a -> b) -> Getter b
convertPV a -> b
f = (forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap a -> b
f forall b c a. (b -> c) -> (a -> b) -> a -> c
.) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. FromField a => FieldParser a
PGFF.fromField

builtinGetters :: I.IntMap (Getter PersistValue)
builtinGetters :: IntMap (Getter PersistValue)
builtinGetters = forall a. [(Int, a)] -> IntMap a
I.fromList
    [ (TypeInfo -> Int
k TypeInfo
PS.bool,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Bool -> PersistValue
PersistBool)
    , (TypeInfo -> Int
k TypeInfo
PS.bytea,       forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Binary a -> a
unBinary))
    , (TypeInfo -> Int
k TypeInfo
PS.char,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Text -> PersistValue
PersistText)
    , (TypeInfo -> Int
k TypeInfo
PS.name,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Text -> PersistValue
PersistText)
    , (TypeInfo -> Int
k TypeInfo
PS.int8,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Int64 -> PersistValue
PersistInt64)
    , (TypeInfo -> Int
k TypeInfo
PS.int2,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Int64 -> PersistValue
PersistInt64)
    , (TypeInfo -> Int
k TypeInfo
PS.int4,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Int64 -> PersistValue
PersistInt64)
    , (TypeInfo -> Int
k TypeInfo
PS.text,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Text -> PersistValue
PersistText)
    , (TypeInfo -> Int
k TypeInfo
PS.xml,         forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    , (TypeInfo -> Int
k TypeInfo
PS.float4,      forall a b. FromField a => (a -> b) -> Getter b
convertPV Double -> PersistValue
PersistDouble)
    , (TypeInfo -> Int
k TypeInfo
PS.float8,      forall a b. FromField a => (a -> b) -> Getter b
convertPV Double -> PersistValue
PersistDouble)
    , (TypeInfo -> Int
k TypeInfo
PS.money,       forall a b. FromField a => (a -> b) -> Getter b
convertPV Rational -> PersistValue
PersistRational)
    , (TypeInfo -> Int
k TypeInfo
PS.bpchar,      forall a b. FromField a => (a -> b) -> Getter b
convertPV Text -> PersistValue
PersistText)
    , (TypeInfo -> Int
k TypeInfo
PS.varchar,     forall a b. FromField a => (a -> b) -> Getter b
convertPV Text -> PersistValue
PersistText)
    , (TypeInfo -> Int
k TypeInfo
PS.date,        forall a b. FromField a => (a -> b) -> Getter b
convertPV Day -> PersistValue
PersistDay)
    , (TypeInfo -> Int
k TypeInfo
PS.time,        forall a b. FromField a => (a -> b) -> Getter b
convertPV TimeOfDay -> PersistValue
PersistTimeOfDay)
    , (TypeInfo -> Int
k TypeInfo
PS.timestamp,   forall a b. FromField a => (a -> b) -> Getter b
convertPV (UTCTime -> PersistValue
PersistUTCTimeforall b c a. (b -> c) -> (a -> b) -> a -> c
. TimeZone -> LocalTime -> UTCTime
localTimeToUTC TimeZone
utc))
    , (TypeInfo -> Int
k TypeInfo
PS.timestamptz, forall a b. FromField a => (a -> b) -> Getter b
convertPV UTCTime -> PersistValue
PersistUTCTime)
    , (TypeInfo -> Int
k TypeInfo
PS.interval,    forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistLiteralEscaped forall b c a. (b -> c) -> (a -> b) -> a -> c
. PgInterval -> StrictByteString
pgIntervalToBs))
    , (TypeInfo -> Int
k TypeInfo
PS.bit,         forall a b. FromField a => (a -> b) -> Getter b
convertPV Int64 -> PersistValue
PersistInt64)
    , (TypeInfo -> Int
k TypeInfo
PS.varbit,      forall a b. FromField a => (a -> b) -> Getter b
convertPV Int64 -> PersistValue
PersistInt64)
    , (TypeInfo -> Int
k TypeInfo
PS.numeric,     forall a b. FromField a => (a -> b) -> Getter b
convertPV Rational -> PersistValue
PersistRational)
    , (TypeInfo -> Int
k TypeInfo
PS.void,        \Field
_ Maybe StrictByteString
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return PersistValue
PersistNull)
    , (TypeInfo -> Int
k TypeInfo
PS.json,        forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    , (TypeInfo -> Int
k TypeInfo
PS.jsonb,       forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    , (TypeInfo -> Int
k TypeInfo
PS.unknown,     forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))

    -- Array types: same order as above.
    -- The OIDs were taken from pg_type.
    , (Int
1000,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Bool -> PersistValue
PersistBool)
    , (Int
1001,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Binary a -> a
unBinary))
    , (Int
1002,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Text -> PersistValue
PersistText)
    , (Int
1003,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Text -> PersistValue
PersistText)
    , (Int
1016,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Int64 -> PersistValue
PersistInt64)
    , (Int
1005,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Int64 -> PersistValue
PersistInt64)
    , (Int
1007,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Int64 -> PersistValue
PersistInt64)
    , (Int
1009,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Text -> PersistValue
PersistText)
    , (Int
143,              forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    , (Int
1021,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Double -> PersistValue
PersistDouble)
    , (Int
1022,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Double -> PersistValue
PersistDouble)
    , (Int
1023,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf UTCTime -> PersistValue
PersistUTCTime)
    , (Int
1024,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf UTCTime -> PersistValue
PersistUTCTime)
    , (Int
791,              forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Rational -> PersistValue
PersistRational)
    , (Int
1014,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Text -> PersistValue
PersistText)
    , (Int
1015,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Text -> PersistValue
PersistText)
    , (Int
1182,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Day -> PersistValue
PersistDay)
    , (Int
1183,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf TimeOfDay -> PersistValue
PersistTimeOfDay)
    , (Int
1115,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf UTCTime -> PersistValue
PersistUTCTime)
    , (Int
1185,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf UTCTime -> PersistValue
PersistUTCTime)
    , (Int
1187,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf (StrictByteString -> PersistValue
PersistLiteralEscaped forall b c a. (b -> c) -> (a -> b) -> a -> c
. PgInterval -> StrictByteString
pgIntervalToBs))
    , (Int
1561,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Int64 -> PersistValue
PersistInt64)
    , (Int
1563,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Int64 -> PersistValue
PersistInt64)
    , (Int
1231,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf Rational -> PersistValue
PersistRational)
    -- no array(void) type
    , (Int
2951,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf (StrictByteString -> PersistValue
PersistLiteralEscaped forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    , (Int
199,              forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    , (Int
3807,             forall {a}.
(FromField a, Typeable a) =>
(a -> PersistValue) -> Getter PersistValue
listOf (StrictByteString -> PersistValue
PersistByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown))
    -- no array(unknown) either
    ]
    where
        k :: TypeInfo -> Int
k (TypeInfo -> Oid
PGFF.typoid -> Oid
i) = Oid -> Int
PG.oid2int Oid
i
        -- A @listOf f@ will use a @PGArray (Maybe T)@ to convert
        -- the values to Haskell-land.  The @Maybe@ is important
        -- because the usual way of checking NULLs
        -- (c.f. withStmt') won't check for NULL inside
        -- arrays---or any other compound structure for that matter.
        listOf :: (a -> PersistValue) -> Getter PersistValue
listOf a -> PersistValue
f = forall a b. FromField a => (a -> b) -> Getter b
convertPV ([PersistValue] -> PersistValue
PersistList forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. (a -> b) -> [a] -> [b]
map (forall {a}. (a -> PersistValue) -> Maybe a -> PersistValue
nullable a -> PersistValue
f) forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. PGArray a -> [a]
PG.fromPGArray)
          where nullable :: (a -> PersistValue) -> Maybe a -> PersistValue
nullable = forall b a. b -> (a -> b) -> Maybe a -> b
maybe PersistValue
PersistNull

-- | Get the field parser corresponding to the given 'PG.Oid'.
--
-- For example, pass in the 'PG.Oid' of 'PS.bool', and you will get back a
-- field parser which parses boolean values in the table into 'PersistBool's.
--
-- @since 2.13.2.0
getGetter :: PG.Oid -> Getter PersistValue
getGetter :: Oid -> Getter PersistValue
getGetter Oid
oid
  = forall a. a -> Maybe a -> a
fromMaybe Getter PersistValue
defaultGetter forall a b. (a -> b) -> a -> b
$ forall a. Int -> IntMap a -> Maybe a
I.lookup (Oid -> Int
PG.oid2int Oid
oid) IntMap (Getter PersistValue)
builtinGetters
  where defaultGetter :: Getter PersistValue
defaultGetter = forall a b. FromField a => (a -> b) -> Getter b
convertPV (StrictByteString -> PersistValue
PersistLiteralEscaped forall b c a. (b -> c) -> (a -> b) -> a -> c
. Unknown -> StrictByteString
unUnknown)

unBinary :: PG.Binary a -> a
unBinary :: forall a. Binary a -> a
unBinary (PG.Binary a
x) = a
x

-- | Represent Postgres interval using NominalDiffTime
--
-- @since 2.11.0.0
newtype PgInterval = PgInterval { PgInterval -> NominalDiffTime
getPgInterval :: NominalDiffTime }
  deriving (PgInterval -> PgInterval -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: PgInterval -> PgInterval -> Bool
$c/= :: PgInterval -> PgInterval -> Bool
== :: PgInterval -> PgInterval -> Bool
$c== :: PgInterval -> PgInterval -> Bool
Eq, Int -> PgInterval -> ShowS
[PgInterval] -> ShowS
PgInterval -> String
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [PgInterval] -> ShowS
$cshowList :: [PgInterval] -> ShowS
show :: PgInterval -> String
$cshow :: PgInterval -> String
showsPrec :: Int -> PgInterval -> ShowS
$cshowsPrec :: Int -> PgInterval -> ShowS
Show)

pgIntervalToBs :: PgInterval -> ByteString
pgIntervalToBs :: PgInterval -> StrictByteString
pgIntervalToBs = forall a.
ConvertibleStrings a StrictByteString =>
a -> StrictByteString
toStrictByteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Show a => a -> String
show forall b c a. (b -> c) -> (a -> b) -> a -> c
. PgInterval -> NominalDiffTime
getPgInterval

instance PGTF.ToField PgInterval where
    toField :: PgInterval -> Action
toField (PgInterval NominalDiffTime
t) = forall a. ToField a => a -> Action
PGTF.toField NominalDiffTime
t

instance PGFF.FromField PgInterval where
    fromField :: FieldParser PgInterval
fromField Field
f Maybe StrictByteString
mdata =
      if Field -> Oid
PGFF.typeOid Field
f forall a. Eq a => a -> a -> Bool
/= TypeInfo -> Oid
PS.typoid TypeInfo
PS.interval
        then forall a err.
(Typeable a, Exception err) =>
(String -> Maybe Oid -> String -> String -> String -> err)
-> Field -> String -> Conversion a
PGFF.returnError String -> Maybe Oid -> String -> String -> String -> ResultError
PGFF.Incompatible Field
f String
""
        else case Maybe StrictByteString
mdata of
          Maybe StrictByteString
Nothing  -> forall a err.
(Typeable a, Exception err) =>
(String -> Maybe Oid -> String -> String -> String -> err)
-> Field -> String -> Conversion a
PGFF.returnError String -> Maybe Oid -> String -> String -> String -> ResultError
PGFF.UnexpectedNull Field
f String
""
          Just StrictByteString
dat -> case forall a. Parser a -> StrictByteString -> Either String a
P.parseOnly (Parser StrictByteString NominalDiffTime
nominalDiffTime forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall t. Chunk t => Parser t ()
P.endOfInput) StrictByteString
dat of
            Left String
msg  ->  forall a err.
(Typeable a, Exception err) =>
(String -> Maybe Oid -> String -> String -> String -> err)
-> Field -> String -> Conversion a
PGFF.returnError String -> Maybe Oid -> String -> String -> String -> ResultError
PGFF.ConversionFailed Field
f String
msg
            Right NominalDiffTime
t   -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ NominalDiffTime -> PgInterval
PgInterval NominalDiffTime
t

      where
        toPico :: Integer -> Pico
        toPico :: Integer -> Pico
toPico = forall k (a :: k). Integer -> Fixed a
MkFixed

        -- Taken from Database.PostgreSQL.Simple.Time.Internal.Parser
        twoDigits :: P.Parser Int
        twoDigits :: Parser Int
twoDigits = do
          Char
a <- Parser Char
P.digit
          Char
b <- Parser Char
P.digit
          let c2d :: Char -> Int
c2d Char
c = Char -> Int
ord Char
c forall a. Bits a => a -> a -> a
.&. Int
15
          forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! Char -> Int
c2d Char
a forall a. Num a => a -> a -> a
* Int
10 forall a. Num a => a -> a -> a
+ Char -> Int
c2d Char
b

        -- Taken from Database.PostgreSQL.Simple.Time.Internal.Parser
        seconds :: P.Parser Pico
        seconds :: Parser Pico
seconds = do
          Int
real <- Parser Int
twoDigits
          Maybe Char
mc <- Parser (Maybe Char)
P.peekChar
          case Maybe Char
mc of
            Just Char
'.' -> do
              StrictByteString
t <- Parser Char
P.anyChar forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> (Char -> Bool) -> Parser StrictByteString StrictByteString
P.takeWhile1 Char -> Bool
P.isDigit
              forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! Int64 -> StrictByteString -> Pico
parsePicos (forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
real) StrictByteString
t
            Maybe Char
_ -> forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$! forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
real
         where
          parsePicos :: Int64 -> B8.ByteString -> Pico
          parsePicos :: Int64 -> StrictByteString -> Pico
parsePicos Int64
a0 StrictByteString
t = Integer -> Pico
toPico (forall a b. (Integral a, Num b) => a -> b
fromIntegral (Int64
t' forall a. Num a => a -> a -> a
* Int64
10forall a b. (Num a, Integral b) => a -> b -> a
^Int
n))
            where n :: Int
n  = forall a. Ord a => a -> a -> a
max Int
0 (Int
12 forall a. Num a => a -> a -> a
- StrictByteString -> Int
B8.length StrictByteString
t)
                  t' :: Int64
t' = forall a. (a -> Char -> a) -> a -> StrictByteString -> a
B8.foldl' (\Int64
a Char
c -> Int64
10 forall a. Num a => a -> a -> a
* Int64
a forall a. Num a => a -> a -> a
+ forall a b. (Integral a, Num b) => a -> b
fromIntegral (Char -> Int
ord Char
c forall a. Bits a => a -> a -> a
.&. Int
15)) Int64
a0
                                 (Int -> StrictByteString -> StrictByteString
B8.take Int
12 StrictByteString
t)

        parseSign :: P.Parser Bool
        parseSign :: Parser Bool
parseSign = forall (f :: * -> *) a. Alternative f => [f a] -> f a
P.choice [Char -> Parser Char
P.char Char
'-' forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> forall (m :: * -> *) a. Monad m => a -> m a
return Bool
True, forall (m :: * -> *) a. Monad m => a -> m a
return Bool
False]

        -- Db stores it in [-]HHH:MM:SS.[SSSS]
        -- For example, nominalDay is stored as 24:00:00
        interval :: P.Parser (Bool, Int, Int, Pico)
        interval :: Parser (Bool, Int, Int, Pico)
interval = do
            Bool
s  <- Parser Bool
parseSign
            Int
h  <- forall a. Integral a => Parser a
P.decimal forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Char -> Parser Char
P.char Char
':'
            Int
m  <- Parser Int
twoDigits forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Char -> Parser Char
P.char Char
':'
            Pico
ss <- Parser Pico
seconds
            if Int
m forall a. Ord a => a -> a -> Bool
< Int
60 Bool -> Bool -> Bool
&& Pico
ss forall a. Ord a => a -> a -> Bool
<= Pico
60
                then forall (m :: * -> *) a. Monad m => a -> m a
return (Bool
s, Int
h, Int
m, Pico
ss)
                else forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"Invalid interval"

        nominalDiffTime :: P.Parser NominalDiffTime
        nominalDiffTime :: Parser StrictByteString NominalDiffTime
nominalDiffTime = do
          (Bool
s, Int
h, Int
m, Pico
ss) <- Parser (Bool, Int, Int, Pico)
interval
          let pico :: Pico
pico   = Pico
ss forall a. Num a => a -> a -> a
+ Pico
60 forall a. Num a => a -> a -> a
* (forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
m) forall a. Num a => a -> a -> a
+ Pico
60 forall a. Num a => a -> a -> a
* Pico
60 forall a. Num a => a -> a -> a
* (forall a b. (Integral a, Num b) => a -> b
fromIntegral (forall a. Num a => a -> a
abs Int
h))
          forall (m :: * -> *) a. Monad m => a -> m a
return forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Fractional a => Rational -> a
fromRational forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Real a => a -> Rational
toRational forall a b. (a -> b) -> a -> b
$ if Bool
s then (-Pico
pico) else Pico
pico

fromPersistValueError :: Text -- ^ Haskell type, should match Haskell name exactly, e.g. "Int64"
                      -> Text -- ^ Database type(s), should appear different from Haskell name, e.g. "integer" or "INT", not "Int".
                      -> PersistValue -- ^ Incorrect value
                      -> Text -- ^ Error message
fromPersistValueError :: Text -> Text -> PersistValue -> Text
fromPersistValueError Text
haskellType Text
databaseType PersistValue
received = [Text] -> Text
T.concat
    [ Text
"Failed to parse Haskell type `"
    , Text
haskellType
    , Text
"`; expected "
    , Text
databaseType
    , Text
" from database, but received: "
    , String -> Text
T.pack (forall a. Show a => a -> String
show PersistValue
received)
    , Text
". Potential solution: Check that your database schema matches your Persistent model definitions."
    ]

instance PersistField PgInterval where
    toPersistValue :: PgInterval -> PersistValue
toPersistValue = StrictByteString -> PersistValue
PersistLiteralEscaped forall b c a. (b -> c) -> (a -> b) -> a -> c
. PgInterval -> StrictByteString
pgIntervalToBs
    fromPersistValue :: PersistValue -> Either Text PgInterval
fromPersistValue (PersistLiteral_ LiteralType
DbSpecific StrictByteString
bs) =
        forall a. PersistField a => PersistValue -> Either Text a
fromPersistValue (StrictByteString -> PersistValue
PersistLiteralEscaped StrictByteString
bs)
    fromPersistValue x :: PersistValue
x@(PersistLiteral_ LiteralType
Escaped StrictByteString
bs) =
      case forall a. Parser a -> StrictByteString -> Either String a
P.parseOnly (forall a. Num a => Parser a -> Parser a
P.signed forall a. Fractional a => Parser a
P.rational forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Char -> Parser Char
P.char Char
's' forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* forall t. Chunk t => Parser t ()
P.endOfInput) StrictByteString
bs of
        Left String
_  -> forall a b. a -> Either a b
Left forall a b. (a -> b) -> a -> b
$ Text -> Text -> PersistValue -> Text
fromPersistValueError Text
"PgInterval" Text
"Interval" PersistValue
x
        Right NominalDiffTime
i -> forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ NominalDiffTime -> PgInterval
PgInterval NominalDiffTime
i
    fromPersistValue PersistValue
x = forall a b. a -> Either a b
Left forall a b. (a -> b) -> a -> b
$ Text -> Text -> PersistValue -> Text
fromPersistValueError Text
"PgInterval" Text
"Interval" PersistValue
x

instance PersistFieldSql PgInterval where
  sqlType :: Proxy PgInterval -> SqlType
sqlType Proxy PgInterval
_ = Text -> SqlType
SqlOther Text
"interval"