@@ -8,6 +8,7 @@
8 , MagicHash
8 , MagicHash
9 , OverloadedStrings
9 , OverloadedStrings
10 , ScopedTypeVariables
10 , ScopedTypeVariables
11 , TypeFamilies
12 , TypeFamilies
12 , TypeOperators
13 , TypeOperators
13 , UndecidableInstances
14 , UndecidableInstances
@@ -15,7 +16,6 @@
16module Squeel.PostgreSQL.Statement where
17module Squeel.PostgreSQL.Statement where
18−import Control.Category
19import Data.Boolean
19import Data.Boolean
20import Data.Boolean.Numbers
20import Data.Boolean.Numbers
21import Data.ByteString (ByteString)
21import Data.ByteString (ByteString)
@@ -27,7 +27,7 @@ import Generics.SOP
27import GHC.Exts
27import GHC.Exts
28import GHC.OverloadedLabels
28import GHC.OverloadedLabels
29import GHC.TypeLits
29import GHC.TypeLits
30−import Prelude hiding (id,(.),RealFrac(..))
30+import Prelude hiding (RealFrac(..))
32import qualified Data.ByteString as ByteString
32import qualified Data.ByteString as ByteString
@@ -134,7 +134,11 @@ instance PGNum ty
134 (*) = unsafeBinaryOp "*"
134 (*) = unsafeBinaryOp "*"
135 abs = unsafeFunction "abs"
135 abs = unsafeFunction "abs"
136 signum = unsafeFunction "sign"
136 signum = unsafeFunction "sign"
137− fromInteger = UnsafeExpression . (<> ".") . fromString . show
138+ = UnsafeExpression
139+ . (<> if decimal (Proxy :: Proxy ty) then "." else "")
139instance PGFractional ty
143instance PGFractional ty
140 => Fractional (Expression params tables ('Required (nullity ty))) where
144 => Fractional (Expression params tables ('Required (nullity ty))) where
@@ -334,11 +338,6 @@ newtype Statement
334 = UnsafeStatement { renderStatement :: ByteString }
338 = UnsafeStatement { renderStatement :: ByteString }
335 deriving (Show,Eq)
339 deriving (Show,Eq)
337−instance Category (Statement '[] '[]) where
338− id = UnsafeStatement ";"
339− statement2 . statement1 = UnsafeStatement $
340− renderStatement statement1 <> " " <> renderStatement statement2
342newtype PreparedStatement
341newtype PreparedStatement
343 (params :: [ColumnType])
342 (params :: [ColumnType])
344 (columns :: [(Symbol,ColumnType)])
343 (columns :: [(Symbol,ColumnType)])
@@ -615,53 +614,53 @@ serial8 :: TypeExpression ('Optional ('NotNull 'PGInt8))
615serial8 = UnsafeTypeExpression "serial8"
614serial8 = UnsafeTypeExpression "serial8"
616bigserial :: TypeExpression ('Optional ('NotNull 'PGInt8))
615bigserial :: TypeExpression ('Optional ('NotNull 'PGInt8))
617bigserial = UnsafeTypeExpression "bigserial"
616bigserial = UnsafeTypeExpression "bigserial"
618−money :: TypeExpression ('Optional ('NotNull 'PGMoney))
617+money :: TypeExpression ('Required ('Null 'PGMoney))
619money = UnsafeTypeExpression "money"
618money = UnsafeTypeExpression "money"
620text :: TypeExpression ('Required ('Null 'PGText))
619text :: TypeExpression ('Required ('Null 'PGText))
621text = UnsafeTypeExpression "text"
620text = UnsafeTypeExpression "text"
623 :: KnownNat n
622 :: KnownNat n
624 => proxy n
623 => proxy n
625− -> TypeExpression ('Optional ('NotNull ('PGChar n)))
624+ -> TypeExpression ('Required ('Null ('PGChar n)))
626char (_ :: proxy n) = UnsafeTypeExpression $
625char (_ :: proxy n) = UnsafeTypeExpression $
627 "char(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
626 "char(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
629 :: KnownNat n
628 :: KnownNat n
630 => proxy n
629 => proxy n
631− -> TypeExpression ('Optional ('NotNull ('PGChar n)))
630+ -> TypeExpression ('Required ('Null ('PGChar n)))
632character (_ :: proxy n) = UnsafeTypeExpression $
631character (_ :: proxy n) = UnsafeTypeExpression $
633 "character(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
632 "character(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
635 :: KnownNat n
634 :: KnownNat n
636 => proxy n
635 => proxy n
637− -> TypeExpression ('Optional ('NotNull ('PGVarChar n)))
636+ -> TypeExpression ('Required ('Null ('PGVarChar n)))
638varchar (_ :: proxy n) = UnsafeTypeExpression $
637varchar (_ :: proxy n) = UnsafeTypeExpression $
639 "varchar(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
638 "varchar(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
640characterVarying
639characterVarying
641 :: KnownNat n
640 :: KnownNat n
642 => proxy n
641 => proxy n
643− -> TypeExpression ('Optional ('NotNull ('PGVarChar n)))
642+ -> TypeExpression ('Required ('Null ('PGVarChar n)))
644characterVarying (_ :: proxy n) = UnsafeTypeExpression $
643characterVarying (_ :: proxy n) = UnsafeTypeExpression $
645 "character varying(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
644 "character varying(" <> fromString (show (natVal' (proxy# :: Proxy# n))) <> ")"
646−bytea :: TypeExpression ('Optional ('NotNull ('PGBytea)))
645+bytea :: TypeExpression ('Required ('Null ('PGBytea)))
647bytea = UnsafeTypeExpression "bytea"
646bytea = UnsafeTypeExpression "bytea"
648−timestamp :: TypeExpression ('Optional ('NotNull ('PGTimestamp)))
647+timestamp :: TypeExpression ('Required ('Null ('PGTimestamp)))
649timestamp = UnsafeTypeExpression "timestamp"
648timestamp = UnsafeTypeExpression "timestamp"
650−timestampWithTimeZone :: TypeExpression ('Optional ('NotNull ('PGTimestampTZ)))
649+timestampWithTimeZone :: TypeExpression ('Required ('Null ('PGTimestampTZ)))
651timestampWithTimeZone = UnsafeTypeExpression "timestamp with time zone"
650timestampWithTimeZone = UnsafeTypeExpression "timestamp with time zone"
652−date :: TypeExpression ('Optional ('NotNull ('PGDate)))
651+date :: TypeExpression ('Required ('Null ('PGDate)))
653date = UnsafeTypeExpression "date"
652date = UnsafeTypeExpression "date"
654−time :: TypeExpression ('Optional ('NotNull ('PGTime)))
653+time :: TypeExpression ('Required ('Null ('PGTime)))
655time = UnsafeTypeExpression "time"
654time = UnsafeTypeExpression "time"
656−timeWithTimeZone :: TypeExpression ('Optional ('NotNull ('PGTimeTZ)))
655+timeWithTimeZone :: TypeExpression ('Required ('Null ('PGTimeTZ)))
657timeWithTimeZone = UnsafeTypeExpression "time with time zone"
656timeWithTimeZone = UnsafeTypeExpression "time with time zone"
658−interval :: TypeExpression ('Optional ('NotNull ('PGInterval)))
657+interval :: TypeExpression ('Required ('Null ('PGInterval)))
659interval = UnsafeTypeExpression "interval"
658interval = UnsafeTypeExpression "interval"
660−uuid :: TypeExpression ('Optional ('NotNull ('PGUuid)))
659+uuid :: TypeExpression ('Required ('Null ('PGUuid)))
661uuid = UnsafeTypeExpression "uuid"
660uuid = UnsafeTypeExpression "uuid"
662−json :: TypeExpression ('Optional ('NotNull ('PGJson)))
661+json :: TypeExpression ('Required ('Null ('PGJson)))
663json = UnsafeTypeExpression "json"
662json = UnsafeTypeExpression "json"
664−jsonb :: TypeExpression ('Optional ('NotNull ('PGJsonb)))
663+jsonb :: TypeExpression ('Required ('Null ('PGJsonb)))
665jsonb = UnsafeTypeExpression "jsonb"
664jsonb = UnsafeTypeExpression "jsonb"
@@ -677,11 +676,27 @@ default_ x ty = UnsafeTypeExpression $
678class PGTyped (ty :: PGType) where
677class PGTyped (ty :: PGType) where
679 pgtype :: TypeExpression ('Required ('Null ty))
678 pgtype :: TypeExpression ('Required ('Null ty))
679+instance PGTyped 'PGBool where pgtype = bool
680instance PGTyped 'PGInt2 where pgtype = int2
680instance PGTyped 'PGInt2 where pgtype = int2
681instance PGTyped 'PGInt4 where pgtype = int4
681instance PGTyped 'PGInt4 where pgtype = int4
682instance PGTyped 'PGInt8 where pgtype = int8
682instance PGTyped 'PGInt8 where pgtype = int8
683−instance PGTyped 'PGBool where pgtype = bool
683+instance PGTyped 'PGNumeric where pgtype = numeric
684+instance PGTyped 'PGFloat4 where pgtype = float4
685+instance PGTyped 'PGFloat8 where pgtype = float8
686+instance PGTyped 'PGMoney where pgtype = money
684instance PGTyped 'PGText where pgtype = text
687instance PGTyped 'PGText where pgtype = text
688+instance KnownNat n => PGTyped ('PGChar n) where pgtype = char (Proxy @n)
689+instance KnownNat n => PGTyped ('PGVarChar n) where pgtype = varchar (Proxy @n)
690+instance PGTyped 'PGBytea where pgtype = bytea
691+instance PGTyped 'PGTimestamp where pgtype = timestamp
692+instance PGTyped 'PGTimestampTZ where pgtype = timestampWithTimeZone
693+instance PGTyped 'PGDate where pgtype = date
694+instance PGTyped 'PGTime where pgtype = time
695+instance PGTyped 'PGTimeTZ where pgtype = timeWithTimeZone
696+instance PGTyped 'PGInterval where pgtype = interval
697+instance PGTyped 'PGUuid where pgtype = uuid
698+instance PGTyped 'PGJson where pgtype = json
699+instance PGTyped 'PGJsonb where pgtype = jsonb
686createTable
701createTable
687 :: (KnownSymbol table, SListI columns)
702 :: (KnownSymbol table, SListI columns)
@@ -711,8 +726,8 @@ class KnownSymbol table => DropTable table schema0 schema1
711 dropTable (Alias table) = UnsafeStatement $
726 dropTable (Alias table) = UnsafeStatement $
712 "DROP TABLE " <> fromString (symbolVal' table) <> ";"
727 "DROP TABLE " <> fromString (symbolVal' table) <> ";"
713instance {-# OVERLAPPING #-}
728instance {-# OVERLAPPING #-}
714− (KnownSymbol table, table ~ table')
729+ (KnownSymbol table, table ~ table', schema ~ schema')
715− => DropTable table ((table' ::: columns) ': schema) schema
730+ => DropTable table ((table' ::: columns) ': schema) schema'
716instance {-# OVERLAPPABLE #-}
731instance {-# OVERLAPPABLE #-}
717 DropTable table schema0 schema1
732 DropTable table schema0 schema1
718 => DropTable table (table' ': schema0) (table' ': schema1)
733 => DropTable table (table' ': schema0) (table' ': schema1)