on dev help sign in
35 files changed+2280−1674
since splitsnapshots
unifiedsplit
README.md+6−6 RELEASE NOTES.md+2−2 squeal-postgresql/exe/Example.hs+2−2 squeal-postgresql/squeal-postgresql.cabal+8−1 squeal-postgresql/src/Squeal/PostgreSQL.hs+8−1 squeal-postgresql/src/Squeal/PostgreSQL/Binary.hs+130−40 squeal-postgresql/src/Squeal/PostgreSQL/Definition.hs+1−1389 squeal-postgresql/src/Squeal/PostgreSQL/Definition/Function.hs+238 squeal-postgresql/src/Squeal/PostgreSQL/Definition/Index.hs+159 squeal-postgresql/src/Squeal/PostgreSQL/Definition/Schema.hs+96 squeal-postgresql/src/Squeal/PostgreSQL/Definition/Table.hs+500 squeal-postgresql/src/Squeal/PostgreSQL/Definition/Table/Constraint.hs+339 squeal-postgresql/src/Squeal/PostgreSQL/Definition/Type.hs+229 squeal-postgresql/src/Squeal/PostgreSQL/Definition/View.hs+104 squeal-postgresql/src/Squeal/PostgreSQL/Expression.hs+10−10 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Collection.hs+5−5 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Comparison.hs+6−6 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Json.hs+40−47 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Logic.hs+1−1 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Math.hs+4−6 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Null.hs+4−4 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Range.hs+8−9 squeal-postgresql/src/Squeal/PostgreSQL/Expression/SetOf.hs → squeal-postgresql/src/Squeal/PostgreSQL/Expression/Set.hs+52−40 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Text.hs+3−3 squeal-postgresql/src/Squeal/PostgreSQL/Expression/TextSearch.hs+19−20 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Time.hs+7−12 squeal-postgresql/src/Squeal/PostgreSQL/Expression/Type.hs+108−3 squeal-postgresql/src/Squeal/PostgreSQL/List.hs+11−2 squeal-postgresql/src/Squeal/PostgreSQL/Migration.hs+2 squeal-postgresql/src/Squeal/PostgreSQL/PG.hs+22−12 squeal-postgresql/src/Squeal/PostgreSQL/PQ.hs+41−17 squeal-postgresql/src/Squeal/PostgreSQL/Query.hs+33−25 squeal-postgresql/src/Squeal/PostgreSQL/Schema.hs+69−9 squeal-postgresql/test/Property.hs+2−2 squeal-postgresql/test/Spec.hs+11
README.md+6−6
@@ -121,7 +121,7 @@ type Schemas = Public Schema
121121
122122Notice the use of type operators.
123123
124−`:::` is used to pair an alias `GHC.TypeLits.Symbol` with a `SchemumType`,
124+`:::` is used to pair an alias `Symbol` with a `SchemumType`,
125125a `TableConstraint` or a `ColumnType`. It is intended to connote Haskell's `::`
126126operator.
127127
@@ -180,8 +180,8 @@ DROP TABLE "emails";
180180DROP TABLE "users";
181181```
182182
183−We'll need a Haskell type for @User@s. We give the type `Generics.SOP.Generic` and
184−`Generics.SOP.HasDatatypeInfo` instances so that we can encode and decode @User@s.
183+We'll need a Haskell type for `User`s. We give the type `SOP.Generic` and
184+`SOP.HasDatatypeInfo` instances so that we can encode and decode `User`s.
185185
186186```Haskell
187187>>> data User = User { userName :: Text, userEmail :: Maybe Text } deriving (Show, GHC.Generic)
@@ -189,12 +189,12 @@ We'll need a Haskell type for @User@s. We give the type `Generics.SOP.Generic` a
189189>>> instance SOP.HasDatatypeInfo User
190190```
191191
192−Next, we'll write `Manipulation_`s to insert @User@s into our two tables.
192+Next, we'll write `Manipulation_`s to insert `User`s into our two tables.
193193A `Manipulation_` like `insertInto`, `update` or `deleteFrom`
194194has three type parameters, the schemas it refers to, input parameters
195195and an output row. When
196−we insert into the users table, we will need a parameter for the @name@
197−field but not for the @id@ field. Since it's serial, we can use a default
196+we insert into the users table, we will need a parameter for the `name`
197+field but not for the `id` field. Since it's serial, we can use a default
198198value. However, since the emails table refers to the users table, we will
199199need to retrieve the user id that the insert generates and insert it into
200200the emails table. We can do this in a single `Manipulation_` by using a
RELEASE NOTES.md+2−2
@@ -138,11 +138,11 @@ type Expr x
138138 . Expression outer commons grp schemas params from x
139139```
140140
141−There is also a function type `(:-->)`, which is a subtype of the usual Haskell function
141+There is also a function type `(-->)`, which is a subtype of the usual Haskell function
142142type `(->)`.
143143
144144```Haskell
145−type (:-->) x y
145+type (-->) x y
146146 = forall outer commons grp schemas params from
147147 . Expression outer commons grp schemas params from x
148148 -> Expression outer commons grp schemas params from y
squeal-postgresql/exe/Example.hs+2−2
@@ -53,8 +53,8 @@ setup =
5353 >>>
5454 createTable #emails
5555 ( serial `as` #id :*
56− (int & notNullable) `as` #user_id :*
57− (text & nullable) `as` #email )
56+ hask @Int32 `as` #user_id :*
57+ hask @(Maybe Text) `as` #email )
5858 ( primaryKey #id `as` #pk_emails :*
5959 foreignKey #user_id #users #id
6060 OnDeleteCascade OnUpdateCascade `as` #fk_user_id )
squeal-postgresql/squeal-postgresql.cabal+8−1
@@ -25,6 +25,13 @@ library
2525 Squeal.PostgreSQL.Alias
2626 Squeal.PostgreSQL.Binary
2727 Squeal.PostgreSQL.Definition
28+ Squeal.PostgreSQL.Definition.Function
29+ Squeal.PostgreSQL.Definition.Index
30+ Squeal.PostgreSQL.Definition.Table
31+ Squeal.PostgreSQL.Definition.Table.Constraint
32+ Squeal.PostgreSQL.Definition.Type
33+ Squeal.PostgreSQL.Definition.Schema
34+ Squeal.PostgreSQL.Definition.View
2835 Squeal.PostgreSQL.Expression
2936 Squeal.PostgreSQL.Expression.Aggregate
3037 Squeal.PostgreSQL.Expression.Collection
@@ -36,7 +43,7 @@ library
3643 Squeal.PostgreSQL.Expression.Null
3744 Squeal.PostgreSQL.Expression.Parameter
3845 Squeal.PostgreSQL.Expression.Range
39− Squeal.PostgreSQL.Expression.SetOf
46+ Squeal.PostgreSQL.Expression.Set
4047 Squeal.PostgreSQL.Expression.Sort
4148 Squeal.PostgreSQL.Expression.Subquery
4249 Squeal.PostgreSQL.Expression.Text
squeal-postgresql/src/Squeal/PostgreSQL.hs+8−1
@@ -195,6 +195,13 @@ module Squeal.PostgreSQL (module X, RenderSQL(..), printSQL) where
195195import Squeal.PostgreSQL.Alias as X
196196import Squeal.PostgreSQL.Binary as X
197197import Squeal.PostgreSQL.Definition as X
198+import Squeal.PostgreSQL.Definition.Function as X
199+import Squeal.PostgreSQL.Definition.Index as X
200+import Squeal.PostgreSQL.Definition.Schema as X
201+import Squeal.PostgreSQL.Definition.Table as X
202+import Squeal.PostgreSQL.Definition.Table.Constraint as X
203+import Squeal.PostgreSQL.Definition.Type as X
204+import Squeal.PostgreSQL.Definition.View as X
198205import Squeal.PostgreSQL.Expression as X
199206import Squeal.PostgreSQL.Expression.Aggregate as X
200207import Squeal.PostgreSQL.Expression.Collection as X
@@ -206,7 +213,7 @@ import Squeal.PostgreSQL.Expression.Math as X
206213import Squeal.PostgreSQL.Expression.Null as X
207214import Squeal.PostgreSQL.Expression.Parameter as X
208215import Squeal.PostgreSQL.Expression.Range as X
209−import Squeal.PostgreSQL.Expression.SetOf as X
216+import Squeal.PostgreSQL.Expression.Set as X
210217import Squeal.PostgreSQL.Expression.Sort as X
211218import Squeal.PostgreSQL.Expression.Subquery as X
212219import Squeal.PostgreSQL.Expression.Text as X
squeal-postgresql/src/Squeal/PostgreSQL/Binary.hs+130−40
@@ -216,8 +216,9 @@ module Squeal.PostgreSQL.Binary
216216 -- * Only
217217 , Only (..)
218218 -- * Oid
219− , HasOid (..)
220− , HasAliasedOid (..)
219+ , OidOf (..)
220+ , OidOfParam (..)
221+ , OidOfField (..)
221222 ) where
222223
223224import BinaryParser
@@ -231,6 +232,7 @@ import Data.Time
231232import Data.UUID.Types
232233import Data.Vector (Vector)
233234import Data.Word
235+import Foreign.C.Types (CUInt(CUInt))
234236import Generics.SOP
235237import Generics.SOP.Record
236238import GHC.TypeLits
@@ -244,6 +246,7 @@ import qualified Data.Text.Lazy as Lazy (Text)
244246import qualified Data.Text as Strict (Text)
245247import qualified Data.Text as Strict.Text
246248import qualified Data.Vector as Vector
249+import qualified Database.PostgreSQL.LibPQ as LibPQ
247250import qualified GHC.Generics as GHC
248251import qualified PostgreSQL.Binary.Decoding as Decoding
249252import qualified PostgreSQL.Binary.Encoding as Encoding
@@ -309,24 +312,24 @@ instance Aeson.ToJSON x => ToParam (Json x) 'PGjson where
309312instance Aeson.ToJSON x => ToParam (Jsonb x) 'PGjsonb where
310313 toParam = K . Encoding.jsonb_bytes
311314 . Lazy.ByteString.toStrict . Aeson.encode . getJsonb
312−instance (ToNullityParam x ty, ty ~ nullity pg, HasOid pg)
315+instance (ToNullityParam x ty, ty ~ nullity pg, OidOf pg)
313316 => ToParam (VarArray [x]) ('PGvararray ty) where
314317 toParam = K
315318 . Encoding.array_foldable
316319 (getOid (oidOf @pg)) (unK . toNullityParam @x @ty)
317320 . getVarArray
318−instance (ToParam x pg, HasOid pg)
321+instance (ToParam x pg, OidOf pg)
319322 => ToParam (VarArray (Vector x)) ('PGvararray ('NotNull pg)) where
320323 toParam = K
321324 . Encoding.array_vector (getOid (oidOf @pg)) (unK . toParam @x @pg)
322325 . getVarArray
323−instance (ToParam x pg, HasOid pg)
326+instance (ToParam x pg, OidOf pg)
324327 => ToParam (VarArray (Vector (Maybe x))) ('PGvararray ('Null pg)) where
325328 toParam = K
326329 . Encoding.nullableArray_vector
327330 (getOid (oidOf @pg)) (unK . toParam @x @pg)
328331 . getVarArray
329−instance (ToFixArray x dims ty, ty ~ nullity pg, HasOid pg)
332+instance (ToFixArray x dims ty, ty ~ nullity pg, OidOf pg)
330333 => ToParam (FixArray x) ('PGfixarray dims ty) where
331334 toParam = K . Encoding.array (getOid (oidOf @pg))
332335 . unK . unK . toFixArray @x @dims @ty . getFixArray
@@ -353,7 +356,7 @@ instance
353356 ( SListI fields
354357 , IsRecord x xs
355358 , AllZip ToField xs fields
356− , All HasAliasedOid fields
359+ , All OidOfField fields
357360 ) => ToParam (Composite x) ('PGcomposite fields) where
358361 toParam =
359362 let
@@ -361,7 +364,7 @@ instance
361364 encoders = htrans (Proxy @ToField) toField
362365
363366 composite
364− :: All HasAliasedOid row
367+ :: All OidOfField row
365368 => NP (K (Maybe Encoding.Encoding)) row
366369 -> K Encoding.Encoding ('PGcomposite row)
367370 composite fields = K $
@@ -378,18 +381,18 @@ instance
378381 int32BE (fromIntegral (lengthSList (Proxy @xs))) <>
379382 let
380383 each
381− :: HasAliasedOid field
384+ :: OidOfField field
382385 => K (Maybe Encoding.Encoding) field
383386 -> Encoding.Encoding
384387 each (K field :: K (Maybe Encoding.Encoding) field) =
385− word32BE (getOid (aliasedOid @field))
388+ word32BE (getOid (oidOfField @field))
386389 <> case field of
387390 Nothing -> int64BE (-1)
388391 Just value ->
389392 int32BE (fromIntegral (builderLength value))
390393 <> value
391394 in
392− hcfoldMap (Proxy @HasAliasedOid) each fields
395+ hcfoldMap (Proxy @OidOfField) each fields
393396
394397 in
395398 composite . encoders . toRecord . getComposite
@@ -398,35 +401,119 @@ instance
398401--
399402-- >>> :set -XTypeApplications
400403-- >>> oidOf @'PGbool
401−-- Oid {getOid = 16}
402−class HasOid (ty :: PGType) where oidOf :: Oid
403−instance HasOid 'PGbool where oidOf = Oid 16
404−instance HasOid 'PGint2 where oidOf = Oid 21
405−instance HasOid 'PGint4 where oidOf = Oid 23
406−instance HasOid 'PGint8 where oidOf = Oid 20
407−instance HasOid 'PGnumeric where oidOf = Oid 1700
408−instance HasOid 'PGfloat4 where oidOf = Oid 700
409−instance HasOid 'PGfloat8 where oidOf = Oid 701
410−instance HasOid ('PGchar n) where oidOf = Oid 18
411−instance HasOid ('PGvarchar n) where oidOf = Oid 1043
412−instance HasOid 'PGtext where oidOf = Oid 25
413−instance HasOid 'PGbytea where oidOf = Oid 17
414−instance HasOid 'PGtimestamp where oidOf = Oid 1114
415−instance HasOid 'PGtimestamptz where oidOf = Oid 1184
416−instance HasOid 'PGdate where oidOf = Oid 1082
417−instance HasOid 'PGtime where oidOf = Oid 1083
418−instance HasOid 'PGtimetz where oidOf = Oid 1266
419−instance HasOid 'PGinterval where oidOf = Oid 1186
420−instance HasOid 'PGuuid where oidOf = Oid 2950
421−instance HasOid 'PGinet where oidOf = Oid 869
422−instance HasOid 'PGjson where oidOf = Oid 114
423−instance HasOid 'PGjsonb where oidOf = Oid 3802
424−
425−-- | Lifts a `HasOid` constraint to a field.
426−class HasAliasedOid (field :: (Symbol, NullityType)) where
427− aliasedOid :: Oid
428−instance HasOid ty => HasAliasedOid (alias ::: nullity ty) where
429− aliasedOid = oidOf @ty
404+-- Oid 16
405+class OidOf (ty :: PGType) where oidOf :: LibPQ.Oid
406+instance OidOf 'PGbool where oidOf = LibPQ.Oid 16
407+instance OidOf ('PGfixarray ns (null 'PGbool)) where oidOf = LibPQ.Oid 1000
408+instance OidOf ('PGvararray (null 'PGbool)) where oidOf = LibPQ.Oid 1000
409+instance OidOf 'PGint2 where oidOf = LibPQ.Oid 21
410+instance OidOf ('PGfixarray ns (null 'PGint2)) where oidOf = LibPQ.Oid 1005
411+instance OidOf ('PGvararray (null 'PGint2)) where oidOf = LibPQ.Oid 1005
412+instance OidOf 'PGint4 where oidOf = LibPQ.Oid 23
413+instance OidOf ('PGfixarray ns (null 'PGint4)) where oidOf = LibPQ.Oid 1007
414+instance OidOf ('PGvararray (null 'PGint4)) where oidOf = LibPQ.Oid 1007
415+instance OidOf 'PGint8 where oidOf = LibPQ.Oid 20
416+instance OidOf ('PGfixarray ns (null 'PGint8)) where oidOf = LibPQ.Oid 1016
417+instance OidOf ('PGvararray (null 'PGint8)) where oidOf = LibPQ.Oid 1016
418+instance OidOf 'PGnumeric where oidOf = LibPQ.Oid 1700
419+instance OidOf ('PGfixarray ns (null 'PGnumeric)) where oidOf = LibPQ.Oid 1231
420+instance OidOf ('PGvararray (null 'PGnumeric)) where oidOf = LibPQ.Oid 1231
421+instance OidOf 'PGfloat4 where oidOf = LibPQ.Oid 700
422+instance OidOf ('PGfixarray ns (null 'PGfloat4)) where oidOf = LibPQ.Oid 1021
423+instance OidOf ('PGvararray (null 'PGfloat4)) where oidOf = LibPQ.Oid 1021
424+instance OidOf 'PGfloat8 where oidOf = LibPQ.Oid 701
425+instance OidOf ('PGfixarray ns (null 'PGfloat8)) where oidOf = LibPQ.Oid 1022
426+instance OidOf ('PGvararray (null 'PGfloat8)) where oidOf = LibPQ.Oid 1022
427+instance OidOf 'PGmoney where oidOf = LibPQ.Oid 790
428+instance OidOf ('PGfixarray ns (null 'PGmoney)) where oidOf = LibPQ.Oid 791
429+instance OidOf ('PGvararray (null 'PGmoney)) where oidOf = LibPQ.Oid 791
430+instance OidOf ('PGchar n) where oidOf = LibPQ.Oid 18
431+instance OidOf ('PGfixarray ns (null ('PGchar n))) where oidOf = LibPQ.Oid 1002
432+instance OidOf ('PGvararray (null ('PGchar n))) where oidOf = LibPQ.Oid 1002
433+instance OidOf ('PGvarchar n) where oidOf = LibPQ.Oid 1043
434+instance OidOf ('PGfixarray ns (null ('PGvarchar n))) where oidOf = LibPQ.Oid 1015
435+instance OidOf ('PGvararray (null ('PGvarchar n))) where oidOf = LibPQ.Oid 1015
436+instance OidOf 'PGtext where oidOf = LibPQ.Oid 25
437+instance OidOf ('PGfixarray ns (null 'PGtext)) where oidOf = LibPQ.Oid 1009
438+instance OidOf ('PGvararray (null 'PGtext)) where oidOf = LibPQ.Oid 1009
439+instance OidOf 'PGbytea where oidOf = LibPQ.Oid 17
440+instance OidOf ('PGfixarray ns (null 'PGbytea)) where oidOf = LibPQ.Oid 1001
441+instance OidOf ('PGvararray (null 'PGbytea)) where oidOf = LibPQ.Oid 1001
442+instance OidOf 'PGtimestamp where oidOf = LibPQ.Oid 1114
443+instance OidOf ('PGfixarray ns (null 'PGtimestamp)) where oidOf = LibPQ.Oid 1115
444+instance OidOf ('PGvararray (null 'PGtimestamp)) where oidOf = LibPQ.Oid 1115
445+instance OidOf 'PGtimestamptz where oidOf = LibPQ.Oid 1184
446+instance OidOf ('PGfixarray ns (null 'PGtimestamptz)) where oidOf = LibPQ.Oid 1185
447+instance OidOf ('PGvararray (null 'PGtimestamptz)) where oidOf = LibPQ.Oid 1185
448+instance OidOf 'PGdate where oidOf = LibPQ.Oid 1082
449+instance OidOf ('PGfixarray ns (null 'PGdate)) where oidOf = LibPQ.Oid 1182
450+instance OidOf ('PGvararray (null 'PGdate)) where oidOf = LibPQ.Oid 1182
451+instance OidOf 'PGtime where oidOf = LibPQ.Oid 1083
452+instance OidOf ('PGfixarray ns (null 'PGtime)) where oidOf = LibPQ.Oid 1183
453+instance OidOf ('PGvararray (null 'PGtime)) where oidOf = LibPQ.Oid 1183
454+instance OidOf 'PGtimetz where oidOf = LibPQ.Oid 1266
455+instance OidOf ('PGfixarray ns (null 'PGtimetz)) where oidOf = LibPQ.Oid 1270
456+instance OidOf ('PGvararray (null 'PGtimetz)) where oidOf = LibPQ.Oid 1270
457+instance OidOf 'PGinterval where oidOf = LibPQ.Oid 1186
458+instance OidOf ('PGfixarray ns (null 'PGinterval)) where oidOf = LibPQ.Oid 1187
459+instance OidOf ('PGvararray (null 'PGinterval)) where oidOf = LibPQ.Oid 1187
460+instance OidOf 'PGuuid where oidOf = LibPQ.Oid 2950
461+instance OidOf ('PGfixarray ns (null 'PGuuid)) where oidOf = LibPQ.Oid 2951
462+instance OidOf ('PGvararray (null 'PGuuid)) where oidOf = LibPQ.Oid 2951
463+instance OidOf 'PGinet where oidOf = LibPQ.Oid 869
464+instance OidOf ('PGfixarray ns (null 'PGinet)) where oidOf = LibPQ.Oid 1041
465+instance OidOf ('PGvararray (null 'PGinet)) where oidOf = LibPQ.Oid 1041
466+instance OidOf 'PGjson where oidOf = LibPQ.Oid 114
467+instance OidOf ('PGfixarray ns (null 'PGjson)) where oidOf = LibPQ.Oid 199
468+instance OidOf ('PGvararray (null 'PGjson)) where oidOf = LibPQ.Oid 199
469+instance OidOf 'PGjsonb where oidOf = LibPQ.Oid 3802
470+instance OidOf ('PGfixarray ns (null 'PGjsonb)) where oidOf = LibPQ.Oid 3807
471+instance OidOf ('PGvararray (null 'PGjsonb)) where oidOf = LibPQ.Oid 3807
472+instance OidOf 'PGtsvector where oidOf = LibPQ.Oid 3614
473+instance OidOf ('PGfixarray ns (null 'PGtsvector)) where oidOf = LibPQ.Oid 3643
474+instance OidOf ('PGvararray (null 'PGtsvector)) where oidOf = LibPQ.Oid 3643
475+instance OidOf 'PGtsquery where oidOf = LibPQ.Oid 3615
476+instance OidOf ('PGfixarray ns (null 'PGtsquery)) where oidOf = LibPQ.Oid 3645
477+instance OidOf ('PGvararray (null 'PGtsquery)) where oidOf = LibPQ.Oid 3645
478+instance OidOf 'PGoid where oidOf = LibPQ.Oid 26
479+instance OidOf ('PGfixarray ns (null 'PGoid)) where oidOf = LibPQ.Oid 1028
480+instance OidOf ('PGvararray (null 'PGoid)) where oidOf = LibPQ.Oid 1028
481+instance OidOf ('PGrange 'PGint4) where oidOf = LibPQ.Oid 3904
482+instance OidOf ('PGfixarray ns (null ('PGrange 'PGint4))) where oidOf = LibPQ.Oid 3905
483+instance OidOf ('PGvararray (null ('PGrange 'PGint4))) where oidOf = LibPQ.Oid 3905
484+instance OidOf ('PGrange 'PGint8) where oidOf = LibPQ.Oid 3926
485+instance OidOf ('PGfixarray ns (null ('PGrange 'PGint8))) where oidOf = LibPQ.Oid 3927
486+instance OidOf ('PGvararray (null ('PGrange 'PGint8))) where oidOf = LibPQ.Oid 3927
487+instance OidOf ('PGrange 'PGnumeric) where oidOf = LibPQ.Oid 3906
488+instance OidOf ('PGfixarray ns (null ('PGrange 'PGnumeric))) where oidOf = LibPQ.Oid 3907
489+instance OidOf ('PGvararray (null ('PGrange 'PGnumeric))) where oidOf = LibPQ.Oid 3907
490+instance OidOf ('PGrange 'PGtimestamp) where oidOf = LibPQ.Oid 3908
491+instance OidOf ('PGfixarray ns (null ('PGrange 'PGtimestamp))) where oidOf = LibPQ.Oid 3909
492+instance OidOf ('PGvararray (null ('PGrange 'PGtimestamp))) where oidOf = LibPQ.Oid 3909
493+instance OidOf ('PGrange 'PGtimestamptz) where oidOf = LibPQ.Oid 3910
494+instance OidOf ('PGfixarray ns (null ('PGrange 'PGtimestamptz))) where oidOf = LibPQ.Oid 3911
495+instance OidOf ('PGvararray (null ('PGrange 'PGtimestamptz))) where oidOf = LibPQ.Oid 3911
496+instance OidOf ('PGrange 'PGdate) where oidOf = LibPQ.Oid 3912
497+instance OidOf ('PGfixarray ns (null ('PGrange 'PGdate))) where oidOf = LibPQ.Oid 3913
498+instance OidOf ('PGvararray (null ('PGrange 'PGdate))) where oidOf = LibPQ.Oid 3913
499+instance {-# OVERLAPPABLE #-} OidOf ('PGrange ty) where oidOf = LibPQ.invalidOid
500+instance {-# OVERLAPPABLE #-} OidOf ('PGfixarray ns (null ('PGrange ty))) where oidOf = LibPQ.invalidOid
501+instance {-# OVERLAPPABLE #-} OidOf ('PGvararray (null ('PGrange ty))) where oidOf = LibPQ.invalidOid
502+instance {-# OVERLAPPABLE #-} OidOf ('PGcomposite row) where oidOf = LibPQ.invalidOid
503+instance {-# OVERLAPPABLE #-} OidOf ('PGfixarray ns (null ('PGcomposite row))) where oidOf = LibPQ.invalidOid
504+instance {-# OVERLAPPABLE #-} OidOf ('PGvararray (null ('PGcomposite row))) where oidOf = LibPQ.invalidOid
505+instance {-# OVERLAPPABLE #-} OidOf ('PGenum labels) where oidOf = LibPQ.invalidOid
506+instance {-# OVERLAPPABLE #-} OidOf ('PGfixarray ns (null ('PGenum labels))) where oidOf = LibPQ.invalidOid
507+instance {-# OVERLAPPABLE #-} OidOf ('PGvararray (null ('PGenum labels))) where oidOf = LibPQ.invalidOid
508+
509+class OidOfParam (ty :: NullityType) where oidOfParam :: Oid
510+instance OidOf ty => OidOfParam (null ty) where oidOfParam = oidOf @ty
511+
512+-- | Lifts a `OidOf` constraint to a field.
513+class OidOfField (field :: (Symbol, NullityType)) where
514+ oidOfField :: Oid
515+instance OidOf ty => OidOfField (alias ::: nullity ty) where
516+ oidOfField = oidOf @ty
430517
431518-- | A `ToNullityParam` constraint gives an encoding of a Haskell `Type` into
432519-- into the binary format of a PostgreSQL `NullityType`.
@@ -756,3 +843,6 @@ replicateMN
756843 => m x -> m (NP I xs)
757844replicateMN mx = hsequence' $
758845 hcpure (Proxy :: Proxy ((~) x)) (Comp (I <$> mx))
846+
847+getOid :: LibPQ.Oid -> Word32
848+getOid (LibPQ.Oid (CUInt oid)) = oid
squeal-postgresql/src/Squeal/PostgreSQL/Definition.hs+1−1389
@@ -27,6 +27,7 @@ Squeal data definition language.
2727 , TypeInType
2828 , TypeOperators
2929 , UndecidableSuperClasses
30+ , UndecidableInstances
3031#-}
3132
3233module Squeal.PostgreSQL.Definition
@@ -34,105 +35,16 @@ module Squeal.PostgreSQL.Definition
3435 Definition (..)
3536 , (>>>)
3637 , manipDefinition
37− -- * Tables
38− -- ** Create
39− , createSchema
40− , createSchemaIfNotExists
41− , createTable
42− , createTableIfNotExists
43− , createView
44− , createOrReplaceView
45− , createTypeEnum
46− , createTypeEnumFrom
47− , createTypeComposite
48− , createTypeCompositeFrom
49− , createTypeRange
50− , createDomain
51− , FieldTyped (..)
52− , createIndex
53− , createIndexIfNotExists
54− , IndexMethod (..)
55− , btree, hash, gist, spgist, gin, brin
56− , createFunction
57− , createOrReplaceFunction
58− , createSetFunction
59− , createOrReplaceSetFunction
60− , createBinaryOp
61− , createLeftOp
62− , createRightOp
63− , TableConstraintExpression (..)
64− , check
65− , unique
66− , primaryKey
67− , foreignKey
68− , ForeignKeyed
69− , OnDeleteClause (..)
70− , OnUpdateClause (..)
71− , FunctionDefinition(..)
72− , languageSqlExpr
73− , languageSqlQuery
74− -- ** Drop
75− , dropSchema
76− , dropSchemaIfExists
77− , dropTable
78− , dropTableIfExists
79− , dropView
80− , dropViewIfExists
81− , dropType
82− , dropTypeIfExists
83− , dropIndex
84− , dropIndexIfExists
85− , dropFunction
86− , dropFunctionIfExists
87− , dropOperator
88− , dropOperatorIfExists
89− -- ** Alter
90− , alterTable
91− , alterTableRename
92− , AlterTable (..)
93− , addConstraint
94− , dropConstraint
95− , AddColumn (..)
96− , dropColumn
97− , renameColumn
98− , alterColumn
99− , AlterColumn (..)
100− , setDefault
101− , dropDefault
102− , setNotNull
103− , dropNotNull
104− , alterType
105− -- * Columns
106− , ColumnTypeExpression (..)
107− , nullable
108− , notNullable
109− , default_
110− , serial2
111− , smallserial
112− , serial4
113− , serial
114− , serial8
115− , bigserial
11638 ) where
11739
11840import Control.Category
11941import Control.DeepSeq
12042import Data.ByteString
12143import Data.Monoid
122−import GHC.TypeLits
12344import Prelude hiding ((.), id)
12445
125−import qualified Generics.SOP as SOP
12646import qualified GHC.Generics as GHC
12747
128−import Squeal.PostgreSQL.Alias
129−import Squeal.PostgreSQL.Expression
130−import Squeal.PostgreSQL.Expression.Logic
131−import Squeal.PostgreSQL.Expression.Sort
132−import Squeal.PostgreSQL.Expression.Type
133−import Squeal.PostgreSQL.List
134−import Squeal.PostgreSQL.PG
135−import Squeal.PostgreSQL.Query
13648import Squeal.PostgreSQL.Manipulation
13749import Squeal.PostgreSQL.Render
13850import Squeal.PostgreSQL.Schema
@@ -168,1303 +80,3 @@ manipDefinition
16880 -- ^ no input or output
16981 -> Definition schemas schemas
17082manipDefinition = UnsafeDefinition . (<> ";") . renderSQL
171−
172−{-----------------------------------------
173−CREATE statements
174−-----------------------------------------}
175−
176−{- |
177−`createSchema` enters a new schema into the current database.
178−The schema name must be distinct from the name of any existing schema
179−in the current database.
180−
181−A schema is essentially a namespace: it contains named objects
182−(tables, data types, functions, and operators) whose names
183−can duplicate those of other objects existing in other schemas.
184−Named objects are accessed by `QualifiedAlias`es with the schema
185−name as a prefix.
186−-}
187−createSchema
188− :: KnownSymbol sch
189− => Alias sch
190− -> Definition schemas (Create sch '[] schemas)
191−createSchema sch = UnsafeDefinition $
192− "CREATE" <+> "SCHEMA" <+> renderSQL sch <> ";"
193−
194−{- | Idempotent version of `createSchema`. -}
195−createSchemaIfNotExists
196− :: (KnownSymbol sch, Has sch schemas schema)
197− => Alias sch
198− -> Definition schemas (CreateIfNotExists sch '[] schemas)
199−createSchemaIfNotExists sch = UnsafeDefinition $
200− "CREATE" <+> "SCHEMA" <+> "IF" <+> "NOT" <+> "EXISTS"
201− <+> renderSQL sch <> ";"
202−
203−{- | `createTable` adds a table to the schema.
204−
205−>>> :set -XOverloadedLabels
206−>>> :{
207−type Table = '[] :=>
208− '[ "a" ::: 'NoDef :=> 'Null 'PGint4
209− , "b" ::: 'NoDef :=> 'Null 'PGfloat4 ]
210−:}
211−
212−>>> :{
213−let
214− setup :: Definition (Public '[]) (Public '["tab" ::: 'Table Table])
215− setup = createTable #tab
216− (nullable int `as` #a :* nullable real `as` #b) Nil
217−in printSQL setup
218−:}
219−CREATE TABLE "tab" ("a" int NULL, "b" real NULL);
220−-}
221−createTable
222− :: ( KnownSymbol sch
223− , KnownSymbol tab
224− , columns ~ (col ': cols)
225− , SOP.SListI columns
226− , SOP.SListI constraints
227− , Has sch schemas0 schema0
228− , schemas1 ~ Alter sch (Create tab ('Table (constraints :=> columns)) schema0) schemas0 )
229− => QualifiedAlias sch tab -- ^ the name of the table to add
230− -> NP (Aliased (ColumnTypeExpression schemas0)) columns
231− -- ^ the names and datatype of each column
232− -> NP (Aliased (TableConstraintExpression sch tab schemas1)) constraints
233− -- ^ constraints that must hold for the table
234− -> Definition schemas0 schemas1
235−createTable tab columns constraints = UnsafeDefinition $
236− "CREATE TABLE" <+> renderCreation tab columns constraints
237−
238−{-| `createTableIfNotExists` creates a table if it doesn't exist, but does not add it to the schema.
239−Instead, the schema already has the table so if the table did not yet exist, the schema was wrong.
240−`createTableIfNotExists` fixes this. Interestingly, this property makes it an idempotent in
241−the `Category` of `Definition`s.
242−
243−>>> :set -XOverloadedLabels -XTypeApplications
244−>>> :{
245−type Table = '[] :=>
246− '[ "a" ::: 'NoDef :=> 'Null 'PGint4
247− , "b" ::: 'NoDef :=> 'Null 'PGfloat4 ]
248−:}
249−
250−>>> type Schemas = Public '["tab" ::: 'Table Table]
251−
252−>>> :{
253−let
254− setup :: Definition Schemas Schemas
255− setup = createTableIfNotExists #tab
256− (nullable int `as` #a :* nullable real `as` #b) Nil
257−in printSQL setup
258−:}
259−CREATE TABLE IF NOT EXISTS "tab" ("a" int NULL, "b" real NULL);
260−-}
261−createTableIfNotExists
262− :: ( KnownSymbol sch
263− , KnownSymbol tab
264− , columns ~ (col ': cols)
265− , SOP.SListI columns
266− , SOP.SListI constraints
267− , Has sch schemas0 schema0
268− , schemas1 ~ Alter sch (CreateIfNotExists tab ('Table (constraints :=> columns)) schema0) schemas0 )
269− => QualifiedAlias sch tab -- ^ the name of the table to add
270− -> NP (Aliased (ColumnTypeExpression schemas0)) columns
271− -- ^ the names and datatype of each column
272− -> NP (Aliased (TableConstraintExpression sch tab schemas1)) constraints
273− -- ^ constraints that must hold for the table
274− -> Definition schemas0 schemas1
275−createTableIfNotExists tab columns constraints = UnsafeDefinition $
276− "CREATE TABLE IF NOT EXISTS"
277− <+> renderCreation tab columns constraints
278−
279−-- helper function for `createTable` and `createTableIfNotExists`
280−renderCreation
281− :: ( KnownSymbol sch
282− , KnownSymbol tab
283− , SOP.SListI columns
284− , SOP.SListI constraints )
285− => QualifiedAlias sch tab -- ^ the name of the table to add
286− -> NP (Aliased (ColumnTypeExpression schemas0)) columns
287− -- ^ the names and datatype of each column
288− -> NP (Aliased (TableConstraintExpression sch tab schemas1)) constraints
289− -- ^ constraints that must hold for the table
290− -> ByteString
291−renderCreation tab columns constraints = renderSQL tab
292− <+> parenthesized
293− ( renderCommaSeparated renderColumnDef columns
294− <> ( case constraints of
295− Nil -> ""
296− _ -> ", " <>
297− renderCommaSeparated renderConstraint constraints ) )
298− <> ";"
299− where
300− renderColumnDef :: Aliased (ColumnTypeExpression schemas) x -> ByteString
301− renderColumnDef (ty `As` column) =
302− renderSQL column <+> renderColumnTypeExpression ty
303− renderConstraint
304− :: Aliased (TableConstraintExpression sch tab schemas) constraint
305− -> ByteString
306− renderConstraint (constraint `As` alias) =
307− "CONSTRAINT" <+> renderSQL alias <+> renderSQL constraint
308−
309−-- | Data types are a way to limit the kind of data that can be stored in a
310−-- table. For many applications, however, the constraint they provide is
311−-- too coarse. For example, a column containing a product price should
312−-- probably only accept positive values. But there is no standard data type
313−-- that accepts only positive numbers. Another issue is that you might want
314−-- to constrain column data with respect to other columns or rows.
315−-- For example, in a table containing product information,
316−-- there should be only one row for each product number.
317−-- `TableConstraint`s give you as much control over the data in your tables
318−-- as you wish. If a user attempts to store data in a column that would
319−-- violate a constraint, an error is raised. This applies
320−-- even if the value came from the default value definition.
321−newtype TableConstraintExpression
322− (sch :: Symbol)
323− (tab :: Symbol)
324− (schemas :: SchemasType)
325− (constraint :: TableConstraint)
326− = UnsafeTableConstraintExpression
327− { renderTableConstraintExpression :: ByteString }
328− deriving (GHC.Generic,Show,Eq,Ord,NFData)
329−instance RenderSQL
330− (TableConstraintExpression sch tab schemas constraint) where
331− renderSQL = renderTableConstraintExpression
332−
333−{-| A `check` constraint is the most generic `TableConstraint` type.
334−It allows you to specify that the value in a certain column must satisfy
335−a Boolean (truth-value) expression.
336−
337−>>> :{
338−type Schema = '[
339− "tab" ::: 'Table ('[ "inequality" ::: 'Check '["a","b"]] :=> '[
340− "a" ::: 'NoDef :=> 'NotNull 'PGint4,
341− "b" ::: 'NoDef :=> 'NotNull 'PGint4
342− ])]
343−:}
344−
345−>>> :{
346−let
347− definition :: Definition (Public '[]) (Public Schema)
348− definition = createTable #tab
349− ( (int & notNullable) `as` #a :*
350− (int & notNullable) `as` #b )
351− ( check (#a :* #b) (#a .> #b) `as` #inequality )
352−:}
353−
354−>>> printSQL definition
355−CREATE TABLE "tab" ("a" int NOT NULL, "b" int NOT NULL, CONSTRAINT "inequality" CHECK (("a" > "b")));
356−-}
357−check
358− :: ( Has sch schemas schema
359− , Has tab schema ('Table table)
360− , HasAll aliases (TableToRow table) subcolumns )
361− => NP Alias aliases
362− -- ^ specify the subcolumns which are getting checked
363− -> (forall t. Condition '[] '[] 'Ungrouped schemas '[] '[t ::: subcolumns])
364− -- ^ a closed `Condition` on those subcolumns
365− -> TableConstraintExpression sch tab schemas ('Check aliases)
366−check _cols condition = UnsafeTableConstraintExpression $
367− "CHECK" <+> parenthesized (renderSQL condition)
368−
369−{-| A `unique` constraint ensure that the data contained in a column,
370−or a group of columns, is unique among all the rows in the table.
371−
372−>>> :{
373−type Schema = '[
374− "tab" ::: 'Table( '[ "uq_a_b" ::: 'Unique '["a","b"]] :=> '[
375− "a" ::: 'NoDef :=> 'Null 'PGint4,
376− "b" ::: 'NoDef :=> 'Null 'PGint4
377− ])]
378−:}
379−
380−>>> :{
381−let
382− definition :: Definition (Public '[]) (Public Schema)
383− definition = createTable #tab
384− ( (int & nullable) `as` #a :*
385− (int & nullable) `as` #b )
386− ( unique (#a :* #b) `as` #uq_a_b )
387−:}
388−
389−>>> printSQL definition
390−CREATE TABLE "tab" ("a" int NULL, "b" int NULL, CONSTRAINT "uq_a_b" UNIQUE ("a", "b"));
391−-}
392−unique
393− :: ( Has sch schemas schema
394− , Has tab schema('Table table)
395− , HasAll aliases (TableToRow table) subcolumns )
396− => NP Alias aliases
397− -- ^ specify subcolumns which together are unique for each row
398− -> TableConstraintExpression sch tab schemas ('Unique aliases)
399−unique columns = UnsafeTableConstraintExpression $
400− "UNIQUE" <+> parenthesized (renderSQL columns)
401−
402−{-| A `primaryKey` constraint indicates that a column, or group of columns,
403−can be used as a unique identifier for rows in the table.
404−This requires that the values be both unique and not null.
405−
406−>>> :{
407−type Schema = '[
408− "tab" ::: 'Table ('[ "pk_id" ::: 'PrimaryKey '["id"]] :=> '[
409− "id" ::: 'Def :=> 'NotNull 'PGint4,
410− "name" ::: 'NoDef :=> 'NotNull 'PGtext
411− ])]
412−:}
413−
414−>>> :{
415−let
416− definition :: Definition (Public '[]) (Public Schema)
417− definition = createTable #tab
418− ( serial `as` #id :*
419− (text & notNullable) `as` #name )
420− ( primaryKey #id `as` #pk_id )
421−:}
422−
423−>>> printSQL definition
424−CREATE TABLE "tab" ("id" serial, "name" text NOT NULL, CONSTRAINT "pk_id" PRIMARY KEY ("id"));
425−-}
426−primaryKey
427− :: ( Has sch schemas schema
428− , Has tab schema ('Table table)
429− , HasAll aliases (TableToColumns table) subcolumns
430− , AllNotNull subcolumns )
431− => NP Alias aliases
432− -- ^ specify the subcolumns which together form a primary key.
433− -> TableConstraintExpression sch tab schemas ('PrimaryKey aliases)
434−primaryKey columns = UnsafeTableConstraintExpression $
435− "PRIMARY KEY" <+> parenthesized (renderSQL columns)
436−
437−{-| A `foreignKey` specifies that the values in a column
438−(or a group of columns) must match the values appearing in some row of
439−another table. We say this maintains the referential integrity
440−between two related tables.
441−
442−>>> :{
443−type Schema =
444− '[ "users" ::: 'Table (
445− '[ "pk_users" ::: 'PrimaryKey '["id"] ] :=>
446− '[ "id" ::: 'Def :=> 'NotNull 'PGint4
447− , "name" ::: 'NoDef :=> 'NotNull 'PGtext
448− ])
449− , "emails" ::: 'Table (
450− '[ "pk_emails" ::: 'PrimaryKey '["id"]
451− , "fk_user_id" ::: 'ForeignKey '["user_id"] "users" '["id"]
452− ] :=>
453− '[ "id" ::: 'Def :=> 'NotNull 'PGint4
454− , "user_id" ::: 'NoDef :=> 'NotNull 'PGint4
455− , "email" ::: 'NoDef :=> 'Null 'PGtext
456− ])
457− ]
458−:}
459−
460−>>> :{
461−let
462− setup :: Definition (Public '[]) (Public Schema)
463− setup =
464− createTable #users
465− ( serial `as` #id :*
466− (text & notNullable) `as` #name )
467− ( primaryKey #id `as` #pk_users ) >>>
468− createTable #emails
469− ( serial `as` #id :*
470− (int & notNullable) `as` #user_id :*
471− (text & nullable) `as` #email )
472− ( primaryKey #id `as` #pk_emails :*
473− foreignKey #user_id #users #id
474− OnDeleteCascade OnUpdateCascade `as` #fk_user_id )
475−in printSQL setup
476−:}
477−CREATE TABLE "users" ("id" serial, "name" text NOT NULL, CONSTRAINT "pk_users" PRIMARY KEY ("id"));
478−CREATE TABLE "emails" ("id" serial, "user_id" int NOT NULL, "email" text NULL, CONSTRAINT "pk_emails" PRIMARY KEY ("id"), CONSTRAINT "fk_user_id" FOREIGN KEY ("user_id") REFERENCES "users" ("id") ON DELETE CASCADE ON UPDATE CASCADE);
479−
480−A `foreignKey` can even be a table self-reference.
481−
482−>>> :{
483−type Schema =
484− '[ "employees" ::: 'Table (
485− '[ "employees_pk" ::: 'PrimaryKey '["id"]
486− , "employees_employer_fk" ::: 'ForeignKey '["employer_id"] "employees" '["id"]
487− ] :=>
488− '[ "id" ::: 'Def :=> 'NotNull 'PGint4
489− , "name" ::: 'NoDef :=> 'NotNull 'PGtext
490− , "employer_id" ::: 'NoDef :=> 'Null 'PGint4
491− ])
492− ]
493−:}
494−
495−>>> :{
496−let
497− setup :: Definition (Public '[]) (Public Schema)
498− setup =
499− createTable #employees
500− ( serial `as` #id :*
501− (text & notNullable) `as` #name :*
502− (integer & nullable) `as` #employer_id )
503− ( primaryKey #id `as` #employees_pk :*
504− foreignKey #employer_id #employees #id
505− OnDeleteCascade OnUpdateCascade `as` #employees_employer_fk )
506−in printSQL setup
507−:}
508−CREATE TABLE "employees" ("id" serial, "name" text NOT NULL, "employer_id" integer NULL, CONSTRAINT "employees_pk" PRIMARY KEY ("id"), CONSTRAINT "employees_employer_fk" FOREIGN KEY ("employer_id") REFERENCES "employees" ("id") ON DELETE CASCADE ON UPDATE CASCADE);
509−-}
510−foreignKey
511− :: (ForeignKeyed schemas sch schema child parent
512− table reftable
513− columns refcolumns
514− constraints cols
515− reftys tys )
516− => NP Alias columns
517− -- ^ column or columns in the table
518− -> Alias parent
519− -- ^ reference table
520− -> NP Alias refcolumns
521− -- ^ reference column or columns in the reference table
522− -> OnDeleteClause
523− -- ^ what to do when reference is deleted
524− -> OnUpdateClause
525− -- ^ what to do when reference is updated
526− -> TableConstraintExpression sch child schemas
527− ('ForeignKey columns parent refcolumns)
528−foreignKey keys parent refs ondel onupd = UnsafeTableConstraintExpression $
529− "FOREIGN KEY" <+> parenthesized (renderSQL keys)
530− <+> "REFERENCES" <+> renderSQL parent
531− <+> parenthesized (renderSQL refs)
532− <+> renderSQL ondel
533− <+> renderSQL onupd
534−
535−-- | A constraint synonym between types involved in a foreign key constraint.
536−type ForeignKeyed schemas
537− sch
538− schema
539− child parent
540− table reftable
541− columns refcolumns
542− constraints cols
543− reftys tys =
544− ( Has sch schemas schema
545− , Has child schema ('Table table)
546− , Has parent schema ('Table reftable)
547− , HasAll columns (TableToColumns table) tys
548− , reftable ~ (constraints :=> cols)
549− , HasAll refcolumns cols reftys
550− , SOP.AllZip SamePGType tys reftys
551− , Uniquely refcolumns constraints )
552−
553−-- | `OnDeleteClause` indicates what to do with rows that reference a deleted row.
554−data OnDeleteClause
555− = OnDeleteNoAction
556− -- ^ if any referencing rows still exist when the constraint is checked,
557− -- an error is raised
558− | OnDeleteRestrict -- ^ prevents deletion of a referenced row
559− | OnDeleteCascade
560− -- ^ specifies that when a referenced row is deleted,
561− -- row(s) referencing it should be automatically deleted as well
562− deriving (GHC.Generic,Show,Eq,Ord)
563−instance NFData OnDeleteClause
564−-- | Render `OnDeleteClause`.
565−instance RenderSQL OnDeleteClause where
566− renderSQL = \case
567− OnDeleteNoAction -> "ON DELETE NO ACTION"
568− OnDeleteRestrict -> "ON DELETE RESTRICT"
569− OnDeleteCascade -> "ON DELETE CASCADE"
570−
571−-- | Analagous to `OnDeleteClause` there is also `OnUpdateClause` which is invoked
572−-- when a referenced column is changed (updated).
573−data OnUpdateClause
574− = OnUpdateNoAction
575− -- ^ if any referencing rows has not changed when the constraint is checked,
576− -- an error is raised
577− | OnUpdateRestrict -- ^ prevents update of a referenced row
578− | OnUpdateCascade
579− -- ^ the updated values of the referenced column(s) should be copied
580− -- into the referencing row(s)
581− deriving (GHC.Generic,Show,Eq,Ord)
582−instance NFData OnUpdateClause
583−
584−-- | Render `OnUpdateClause`.
585−instance RenderSQL OnUpdateClause where
586− renderSQL = \case
587− OnUpdateNoAction -> "ON UPDATE NO ACTION"
588− OnUpdateRestrict -> "ON UPDATE RESTRICT"
589− OnUpdateCascade -> "ON UPDATE CASCADE"
590−
591−{-----------------------------------------
592−DROP statements
593−-----------------------------------------}
594−
595−-- | >>> :{
596−-- let
597−-- definition :: Definition '["muh_schema" ::: schema, "public" ::: public] '["public" ::: public]
598−-- definition = dropSchema #muh_schema
599−-- :}
600−--
601−-- >>> printSQL definition
602−-- DROP SCHEMA "muh_schema";
603−dropSchema
604− :: KnownSymbol sch
605− => Alias sch
606− -- ^ user defined schema
607− -> Definition schemas (Drop sch schemas)
608−dropSchema sch = UnsafeDefinition $ "DROP SCHEMA" <+> renderSQL sch <> ";"
609−
610−dropSchemaIfExists
611− :: KnownSymbol sch
612− => Alias sch
613− -- ^ user defined schema
614− -> Definition schemas (DropIfExists sch schemas)
615−dropSchemaIfExists sch = UnsafeDefinition $ "DROP SCHEMA" <+> renderSQL sch <> ";"
616−
617−-- | `dropTable` removes a table from the schema.
618−--
619−-- >>> :{
620−-- let
621−-- definition :: Definition '["public" ::: '["muh_table" ::: 'Table t]] (Public '[])
622−-- definition = dropTable #muh_table
623−-- :}
624−--
625−-- >>> printSQL definition
626−-- DROP TABLE "muh_table";
627−dropTable
628− :: ( Has sch schemas schema
629− , KnownSymbol tab )
630− => QualifiedAlias sch tab -- ^ table to remove
631− -> Definition schemas (Alter sch (DropSchemum tab 'Table schema) schemas)
632−dropTable tab = UnsafeDefinition $ "DROP TABLE" <+> renderSQL tab <> ";"
633−
634−dropTableIfExists
635− :: ( Has sch schemas schema
636− , Has tab schema ('Table table))
637− => QualifiedAlias sch tab -- ^ table to remove
638− -> Definition schemas (Alter sch (DropIfExists tab schema) schemas)
639−dropTableIfExists tab = UnsafeDefinition $ "DROP TABLE IF EXISTS" <+> renderSQL tab <> ";"
640−
641−{-----------------------------------------
642−ALTER statements
643−-----------------------------------------}
644−
645−-- | `alterTable` changes the definition of a table from the schema.
646−alterTable
647− :: (Has sch schemas schema, Has tab schema ('Table table0))
648− => QualifiedAlias sch tab -- ^ table to alter
649− -> AlterTable sch tab schemas table1 -- ^ alteration to perform
650− -> Definition schemas (Alter sch (Alter tab ('Table table1) schema) schemas)
651−alterTable tab alteration = UnsafeDefinition $
652− "ALTER TABLE"
653− <+> renderSQL tab
654− <+> renderAlterTable alteration
655− <> ";"
656−
657−-- | `alterTableRename` changes the name of a table from the schema.
658−--
659−-- >>> printSQL $ alterTableRename #foo #bar
660−-- ALTER TABLE "foo" RENAME TO "bar";
661−alterTableRename
662− :: (KnownSymbol table0, KnownSymbol table1)
663− => Alias table0 -- ^ table to rename
664− -> Alias table1 -- ^ what to rename it
665− -> Definition schema (Rename table0 table1 schema)
666−alterTableRename table0 table1 = UnsafeDefinition $
667− "ALTER TABLE" <+> renderSQL table0
668− <+> "RENAME TO" <+> renderSQL table1 <> ";"
669−
670−-- | An `AlterTable` describes the alteration to perform on the columns
671−-- of a table.
672−newtype AlterTable
673− (sch :: Symbol)
674− (tab :: Symbol)
675− (schemas :: SchemasType)
676− (table :: TableType) =
677− UnsafeAlterTable {renderAlterTable :: ByteString}
678− deriving (GHC.Generic,Show,Eq,Ord,NFData)
679−
680−-- | An `addConstraint` adds a table constraint.
681−--
682−-- >>> :{
683−-- let
684−-- definition :: Definition
685−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
686−-- '["public" ::: '["tab" ::: 'Table ('["positive" ::: 'Check '["col"]] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
687−-- definition = alterTable #tab (addConstraint #positive (check #col (#col .> 0)))
688−-- in printSQL definition
689−-- :}
690−-- ALTER TABLE "tab" ADD CONSTRAINT "positive" CHECK (("col" > 0));
691−addConstraint
692− :: ( KnownSymbol alias
693− , Has sch schemas schema
694− , Has tab schema ('Table table0)
695− , table0 ~ (constraints :=> columns)
696− , table1 ~ (Create alias constraint constraints :=> columns) )
697− => Alias alias
698− -> TableConstraintExpression sch tab schemas constraint
699− -- ^ constraint to add
700− -> AlterTable sch tab schemas table1
701−addConstraint alias constraint = UnsafeAlterTable $
702− "ADD" <+> "CONSTRAINT" <+> renderSQL alias
703− <+> renderSQL constraint
704−
705−-- | A `dropConstraint` drops a table constraint.
706−--
707−-- >>> :{
708−-- let
709−-- definition :: Definition
710−-- '["public" ::: '["tab" ::: 'Table ('["positive" ::: Check '["col"]] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
711−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
712−-- definition = alterTable #tab (dropConstraint #positive)
713−-- in printSQL definition
714−-- :}
715−-- ALTER TABLE "tab" DROP CONSTRAINT "positive";
716−dropConstraint
717− :: ( KnownSymbol constraint
718− , Has sch schemas schema
719− , Has tab schema ('Table table0)
720− , table0 ~ (constraints :=> columns)
721− , table1 ~ (Drop constraint constraints :=> columns) )
722− => Alias constraint
723− -- ^ constraint to drop
724− -> AlterTable sch tab schemas table1
725−dropConstraint constraint = UnsafeAlterTable $
726− "DROP" <+> "CONSTRAINT" <+> renderSQL constraint
727−
728−-- | An `AddColumn` is either @NULL@ or has @DEFAULT@.
729−class AddColumn ty where
730− -- | `addColumn` adds a new column, initially filled with whatever
731− -- default value is given or with @NULL@.
732− --
733− -- >>> :{
734− -- let
735− -- definition :: Definition
736− -- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col1" ::: 'NoDef :=> 'Null 'PGint4])]]
737− -- '["public" ::: '["tab" ::: 'Table ('[] :=>
738− -- '[ "col1" ::: 'NoDef :=> 'Null 'PGint4
739− -- , "col2" ::: 'Def :=> 'Null 'PGtext ])]]
740− -- definition = alterTable #tab (addColumn #col2 (text & nullable & default_ "foo"))
741− -- in printSQL definition
742− -- :}
743− -- ALTER TABLE "tab" ADD COLUMN "col2" text NULL DEFAULT E'foo';
744− --
745− -- >>> :{
746− -- let
747− -- definition :: Definition
748− -- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col1" ::: 'NoDef :=> 'Null 'PGint4])]]
749− -- '["public" ::: '["tab" ::: 'Table ('[] :=>
750− -- '[ "col1" ::: 'NoDef :=> 'Null 'PGint4
751− -- , "col2" ::: 'NoDef :=> 'Null 'PGtext ])]]
752− -- definition = alterTable #tab (addColumn #col2 (text & nullable))
753− -- in printSQL definition
754− -- :}
755− -- ALTER TABLE "tab" ADD COLUMN "col2" text NULL;
756− addColumn
757− :: ( KnownSymbol column
758− , Has sch schemas schema
759− , Has tab schema ('Table table0)
760− , table0 ~ (constraints :=> columns) )
761− => Alias column -- ^ column to add
762− -> ColumnTypeExpression schemas ty -- ^ type of the new column
763− -> AlterTable sch tab schemas (constraints :=> Create column ty columns)
764− addColumn column ty = UnsafeAlterTable $
765− "ADD COLUMN" <+> renderSQL column <+> renderColumnTypeExpression ty
766−instance {-# OVERLAPPING #-} AddColumn ('Def :=> ty)
767−instance {-# OVERLAPPABLE #-} AddColumn ('NoDef :=> 'Null ty)
768−
769−-- | A `dropColumn` removes a column. Whatever data was in the column
770−-- disappears. Table constraints involving the column are dropped, too.
771−-- However, if the column is referenced by a foreign key constraint of
772−-- another table, PostgreSQL will not silently drop that constraint.
773−--
774−-- >>> :{
775−-- let
776−-- definition :: Definition
777−-- '["public" ::: '["tab" ::: 'Table ('[] :=>
778−-- '[ "col1" ::: 'NoDef :=> 'Null 'PGint4
779−-- , "col2" ::: 'NoDef :=> 'Null 'PGtext ])]]
780−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col1" ::: 'NoDef :=> 'Null 'PGint4])]]
781−-- definition = alterTable #tab (dropColumn #col2)
782−-- in printSQL definition
783−-- :}
784−-- ALTER TABLE "tab" DROP COLUMN "col2";
785−dropColumn
786− :: ( KnownSymbol column
787− , Has sch schemas schema
788− , Has tab schema ('Table table0)
789− , table0 ~ (constraints :=> columns)
790− , table1 ~ (constraints :=> Drop column columns) )
791− => Alias column -- ^ column to remove
792− -> AlterTable sch tab schemas table1
793−dropColumn column = UnsafeAlterTable $
794− "DROP COLUMN" <+> renderSQL column
795−
796−-- | A `renameColumn` renames a column.
797−--
798−-- >>> :{
799−-- let
800−-- definition :: Definition
801−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["foo" ::: 'NoDef :=> 'Null 'PGint4])]]
802−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["bar" ::: 'NoDef :=> 'Null 'PGint4])]]
803−-- definition = alterTable #tab (renameColumn #foo #bar)
804−-- in printSQL definition
805−-- :}
806−-- ALTER TABLE "tab" RENAME COLUMN "foo" TO "bar";
807−renameColumn
808− :: ( KnownSymbol column0
809− , KnownSymbol column1
810− , Has sch schemas schema
811− , Has tab schema ('Table table0)
812− , table0 ~ (constraints :=> columns)
813− , table1 ~ (constraints :=> Rename column0 column1 columns) )
814− => Alias column0 -- ^ column to rename
815− -> Alias column1 -- ^ what to rename the column
816− -> AlterTable sch tab schemas table1
817−renameColumn column0 column1 = UnsafeAlterTable $
818− "RENAME COLUMN" <+> renderSQL column0 <+> "TO" <+> renderSQL column1
819−
820−-- | An `alterColumn` alters a single column.
821−alterColumn
822− :: ( KnownSymbol column
823− , Has sch schemas schema
824− , Has tab schema ('Table table0)
825− , table0 ~ (constraints :=> columns)
826− , Has column columns ty0
827− , table1 ~ (constraints :=> Alter column ty1 columns))
828− => Alias column -- ^ column to alter
829− -> AlterColumn schemas ty0 ty1 -- ^ alteration to perform
830− -> AlterTable sch tab schemas table1
831−alterColumn column alteration = UnsafeAlterTable $
832− "ALTER COLUMN" <+> renderSQL column <+> renderAlterColumn alteration
833−
834−-- | An `AlterColumn` describes the alteration to perform on a single column.
835−newtype AlterColumn (schemas :: SchemasType) (ty0 :: ColumnType) (ty1 :: ColumnType) =
836− UnsafeAlterColumn {renderAlterColumn :: ByteString}
837− deriving (GHC.Generic,Show,Eq,Ord,NFData)
838−
839−-- | A `setDefault` sets a new default for a column. Note that this doesn't
840−-- affect any existing rows in the table, it just changes the default for
841−-- future insert and update commands.
842−--
843−-- >>> :{
844−-- let
845−-- definition :: Definition
846−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
847−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'Def :=> 'Null 'PGint4])]]
848−-- definition = alterTable #tab (alterColumn #col (setDefault 5))
849−-- in printSQL definition
850−-- :}
851−-- ALTER TABLE "tab" ALTER COLUMN "col" SET DEFAULT 5;
852−setDefault
853− :: Expression '[] '[] 'Ungrouped schemas '[] '[] ty -- ^ default value to set
854− -> AlterColumn schemas (constraint :=> ty) ('Def :=> ty)
855−setDefault expression = UnsafeAlterColumn $
856− "SET DEFAULT" <+> renderExpression expression
857−
858−-- | A `dropDefault` removes any default value for a column.
859−--
860−-- >>> :{
861−-- let
862−-- definition :: Definition
863−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'Def :=> 'Null 'PGint4])]]
864−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
865−-- definition = alterTable #tab (alterColumn #col dropDefault)
866−-- in printSQL definition
867−-- :}
868−-- ALTER TABLE "tab" ALTER COLUMN "col" DROP DEFAULT;
869−dropDefault :: AlterColumn schemas ('Def :=> ty) ('NoDef :=> ty)
870−dropDefault = UnsafeAlterColumn $ "DROP DEFAULT"
871−
872−-- | A `setNotNull` adds a @NOT NULL@ constraint to a column.
873−-- The constraint will be checked immediately, so the table data must satisfy
874−-- the constraint before it can be added.
875−--
876−-- >>> :{
877−-- let
878−-- definition :: Definition
879−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
880−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
881−-- definition = alterTable #tab (alterColumn #col setNotNull)
882−-- in printSQL definition
883−-- :}
884−-- ALTER TABLE "tab" ALTER COLUMN "col" SET NOT NULL;
885−setNotNull
886− :: AlterColumn schemas (constraint :=> 'Null ty) (constraint :=> 'NotNull ty)
887−setNotNull = UnsafeAlterColumn $ "SET NOT NULL"
888−
889−-- | A `dropNotNull` drops a @NOT NULL@ constraint from a column.
890−--
891−-- >>> :{
892−-- let
893−-- definition :: Definition
894−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
895−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
896−-- definition = alterTable #tab (alterColumn #col dropNotNull)
897−-- in printSQL definition
898−-- :}
899−-- ALTER TABLE "tab" ALTER COLUMN "col" DROP NOT NULL;
900−dropNotNull
901− :: AlterColumn schemas (constraint :=> 'NotNull ty) (constraint :=> 'Null ty)
902−dropNotNull = UnsafeAlterColumn $ "DROP NOT NULL"
903−
904−-- | An `alterType` converts a column to a different data type.
905−-- This will succeed only if each existing entry in the column can be
906−-- converted to the new type by an implicit cast.
907−--
908−-- >>> :{
909−-- let
910−-- definition :: Definition
911−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
912−-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGnumeric])]]
913−-- definition =
914−-- alterTable #tab (alterColumn #col (alterType (numeric & notNullable)))
915−-- in printSQL definition
916−-- :}
917−-- ALTER TABLE "tab" ALTER COLUMN "col" TYPE numeric NOT NULL;
918−alterType :: ColumnTypeExpression schemas ty -> AlterColumn schemas ty0 ty
919−alterType ty = UnsafeAlterColumn $ "TYPE" <+> renderColumnTypeExpression ty
920−
921−{- | Create a view.
922−>>> type ABC = '["a" ::: 'NoDef :=> 'Null 'PGint4, "b" ::: 'NoDef :=> 'Null 'PGint4, "c" ::: 'NoDef :=> 'Null 'PGint4]
923−>>> type BC = '["b" ::: 'Null 'PGint4, "c" ::: 'Null 'PGint4]
924−>>> :{
925−let
926− definition :: Definition
927− '[ "public" ::: '["abc" ::: 'Table ('[] :=> ABC)]]
928− '[ "public" ::: '["abc" ::: 'Table ('[] :=> ABC), "bc" ::: 'View BC]]
929− definition =
930− createView #bc (select_ (#b :* #c) (from (table #abc)))
931−in printSQL definition
932−:}
933−CREATE VIEW "bc" AS SELECT "b" AS "b", "c" AS "c" FROM "abc" AS "abc";
934−-}
935−createView
936− :: (Has sch schemas schema, KnownSymbol vw)
937− => QualifiedAlias sch vw -- ^ the name of the view to add
938− -> Query '[] '[] schemas '[] view -- ^ query
939− -> Definition schemas (Alter sch (Create vw ('View view) schema) schemas)
940−createView alias query = UnsafeDefinition $
941− "CREATE" <+> "VIEW" <+> renderSQL alias <+> "AS"
942− <+> renderQuery query <> ";"
943−
944−createOrReplaceView
945− :: (Has sch schemas schema, KnownSymbol vw)
946− => QualifiedAlias sch vw -- ^ the name of the view to add
947− -> Query '[] '[] schemas '[] view -- ^ query
948− -> Definition schemas (Alter sch (CreateOrReplace vw ('View view) schema) schemas)
949−createOrReplaceView alias query = UnsafeDefinition $
950− "CREATE OR REPLACE VIEW" <+> renderSQL alias <+> "AS"
951− <+> renderQuery query <> ";"
952−
953−-- | Drop a view.
954−--
955−-- >>> :{
956−-- let
957−-- definition :: Definition
958−-- '[ "public" ::: '["abc" ::: 'Table ('[] :=> '["a" ::: 'NoDef :=> 'Null 'PGint4, "b" ::: 'NoDef :=> 'Null 'PGint4, "c" ::: 'NoDef :=> 'Null 'PGint4])
959−-- , "bc" ::: 'View ('["b" ::: 'Null 'PGint4, "c" ::: 'Null 'PGint4])]]
960−-- '[ "public" ::: '["abc" ::: 'Table ('[] :=> '["a" ::: 'NoDef :=> 'Null 'PGint4, "b" ::: 'NoDef :=> 'Null 'PGint4, "c" ::: 'NoDef :=> 'Null 'PGint4])]]
961−-- definition = dropView #bc
962−-- in printSQL definition
963−-- :}
964−-- DROP VIEW "bc";
965−dropView
966− :: (Has sch schemas schema, KnownSymbol vw)
967− => QualifiedAlias sch vw -- ^ view to remove
968− -> Definition schemas (Alter sch (DropSchemum vw 'View schema) schemas)
969−dropView vw = UnsafeDefinition $ "DROP VIEW" <+> renderSQL vw <> ";"
970−
971−dropViewIfExists
972− :: (Has sch schemas schema, KnownSymbol vw)
973− => QualifiedAlias sch vw -- ^ view to remove
974− -> Definition schemas (Alter sch (DropIfExists vw schema) schemas)
975−dropViewIfExists vw = UnsafeDefinition $ "DROP VIEW IF EXISTS" <+> renderSQL vw <> ";"
976−
977−-- | Enumerated types are created using the `createTypeEnum` command, for example
978−--
979−-- >>> printSQL $ (createTypeEnum #mood (label @"sad" :* label @"ok" :* label @"happy") :: Definition (Public '[]) '["public" ::: '["mood" ::: 'Typedef ('PGenum '["sad","ok","happy"])]])
980−-- CREATE TYPE "mood" AS ENUM ('sad', 'ok', 'happy');
981−createTypeEnum
982− :: (KnownSymbol enum, Has sch schemas schema, SOP.All KnownSymbol labels)
983− => QualifiedAlias sch enum
984− -- ^ name of the user defined enumerated type
985− -> NP PGlabel labels
986− -- ^ labels of the enumerated type
987− -> Definition schemas (Alter sch (Create enum ('Typedef ('PGenum labels)) schema) schemas)
988−createTypeEnum enum labels = UnsafeDefinition $
989− "CREATE" <+> "TYPE" <+> renderSQL enum <+> "AS" <+> "ENUM" <+>
990− parenthesized (renderSQL labels) <> ";"
991−
992−-- | Enumerated types can also be generated from a Haskell type, for example
993−--
994−-- >>> data Schwarma = Beef | Lamb | Chicken deriving GHC.Generic
995−-- >>> instance SOP.Generic Schwarma
996−-- >>> instance SOP.HasDatatypeInfo Schwarma
997−-- >>> :{
998−-- let
999−-- createSchwarma :: Definition (Public '[]) '["public" ::: '["schwarma" ::: 'Typedef (PG (Enumerated Schwarma))]]
1000−-- createSchwarma = createTypeEnumFrom @Schwarma #schwarma
1001−-- in
1002−-- printSQL createSchwarma
1003−-- :}
1004−-- CREATE TYPE "schwarma" AS ENUM ('Beef', 'Lamb', 'Chicken');
1005−createTypeEnumFrom
1006− :: forall hask sch enum schemas schema.
1007− ( SOP.Generic hask
1008− , SOP.All KnownSymbol (LabelsPG hask)
1009− , KnownSymbol enum
1010− , Has sch schemas schema )
1011− => QualifiedAlias sch enum
1012− -- ^ name of the user defined enumerated type
1013− -> Definition schemas (Alter sch (Create enum ('Typedef (PG (Enumerated hask))) schema) schemas)
1014−createTypeEnumFrom enum = createTypeEnum enum
1015− (SOP.hpure label :: NP PGlabel (LabelsPG hask))
1016−
1017−{- | `createTypeComposite` creates a composite type. The composite type is
1018−specified by a list of attribute names and data types.
1019−
1020−>>> :{
1021−type PGcomplex = 'PGcomposite
1022− '[ "real" ::: 'NotNull 'PGfloat8
1023− , "imaginary" ::: 'NotNull 'PGfloat8 ]
1024−:}
1025−
1026−>>> :{
1027−let
1028− setup :: Definition (Public '[]) '["public" ::: '["complex" ::: 'Typedef PGcomplex]]
1029− setup = createTypeComposite #complex
1030− (float8 `as` #real :* float8 `as` #imaginary)
1031−in printSQL setup
1032−:}
1033−CREATE TYPE "complex" AS ("real" float8, "imaginary" float8);
1034−-}
1035−createTypeComposite
1036− :: (KnownSymbol ty, Has sch schemas schema, SOP.SListI fields)
1037− => QualifiedAlias sch ty
1038− -- ^ name of the user defined composite type
1039− -> NP (Aliased (TypeExpression schemas)) fields
1040− -- ^ list of attribute names and data types
1041− -> Definition schemas (Alter sch (Create ty ('Typedef ('PGcomposite fields)) schema) schemas)
1042−createTypeComposite ty fields = UnsafeDefinition $
1043− "CREATE" <+> "TYPE" <+> renderSQL ty <+> "AS" <+> parenthesized
1044− (renderCommaSeparated renderField fields) <> ";"
1045− where
1046− renderField :: Aliased (TypeExpression schemas) x -> ByteString
1047− renderField (typ `As` alias) =
1048− renderSQL alias <+> renderSQL typ
1049−
1050−-- | Composite types can also be generated from a Haskell type, for example
1051−--
1052−-- >>> data Complex = Complex {real :: Double, imaginary :: Double} deriving GHC.Generic
1053−-- >>> instance SOP.Generic Complex
1054−-- >>> instance SOP.HasDatatypeInfo Complex
1055−-- >>> type Schema = '["complex" ::: 'Typedef (PG (Composite Complex))]
1056−-- >>> :{
1057−-- let
1058−-- createComplex :: Definition (Public '[]) (Public Schema)
1059−-- createComplex = createTypeCompositeFrom @Complex #complex
1060−-- in
1061−-- printSQL createComplex
1062−-- :}
1063−-- CREATE TYPE "complex" AS ("real" float8, "imaginary" float8);
1064−createTypeCompositeFrom
1065− :: forall hask sch ty schemas schema.
1066− ( SOP.All (FieldTyped schemas) (RowPG hask)
1067− , KnownSymbol ty
1068− , Has sch schemas schema )
1069− => QualifiedAlias sch ty
1070− -- ^ name of the user defined composite type
1071− -> Definition schemas (Alter sch (Create ty ( 'Typedef (PG (Composite hask))) schema) schemas)
1072−createTypeCompositeFrom ty = createTypeComposite ty
1073− (SOP.hcpure (SOP.Proxy :: SOP.Proxy (FieldTyped schemas)) fieldtype
1074− :: NP (Aliased (TypeExpression schemas)) (RowPG hask))
1075−
1076−{-|
1077−`createDomain` creates a new domain. A domain is essentially a data type
1078−with constraints (restrictions on the allowed set of values).
1079−
1080−Domains are useful for abstracting common constraints on fields
1081−into a single location for maintenance. For example, several tables might
1082−contain email address columns, all requiring the same `check` constraint
1083−to verify the address syntax. Define a domain rather than setting up
1084−each table's constraint individually.
1085−
1086−>>> :{
1087−let
1088− createPositive :: Definition (Public '[]) (Public '["positive" ::: 'Typedef 'PGfloat4])
1089− createPositive = createDomain #positive real (#value .> 0 .&& (#value & isNotNull))
1090−in printSQL createPositive
1091−:}
1092−CREATE DOMAIN "positive" AS real CHECK ((("value" > 0) AND "value" IS NOT NULL));
1093−-}
1094−createDomain
1095− :: (Has sch schemas schema, KnownSymbol dom)
1096− => QualifiedAlias sch dom
1097− -> (forall null. TypeExpression schemas (null ty))
1098− -> (forall tab. Condition '[] '[] 'Ungrouped schemas '[] '[tab ::: '["value" ::: 'Null ty]])
1099− -> Definition schemas (Alter sch (Create dom ('Typedef ty) schema) schemas)
1100−createDomain dom ty condition =
1101− UnsafeDefinition $ "CREATE DOMAIN" <+> renderSQL dom
1102− <+> "AS" <+> renderTypeExpression ty
1103− <+> "CHECK" <+> parenthesized (renderSQL condition) <> ";"
1104−
1105−{- |
1106−>>> :{
1107−let
1108− createSmallIntRange :: Definition (Public '[]) (Public '["int2range" ::: 'Typedef ('PGrange 'PGint2)])
1109− createSmallIntRange = createTypeRange #int2range int2
1110−in printSQL createSmallIntRange
1111−:}
1112−CREATE TYPE "int2range" AS RANGE (subtype = int2);
1113−-}
1114−createTypeRange
1115− :: (Has sch schemas schema, KnownSymbol range)
1116− => QualifiedAlias sch range
1117− -> (forall null. TypeExpression schemas (null ty))
1118− -> Definition schemas (Alter sch (Create range ('Typedef ('PGrange ty)) schema) schemas)
1119−createTypeRange range ty = UnsafeDefinition $
1120− "CREATE" <+> "TYPE" <+> renderSQL range <+> "AS" <+> "RANGE" <+>
1121− parenthesized ("subtype" <+> "=" <+> renderTypeExpression ty) <> ";"
1122−
1123−{- |
1124−>>> :{
1125−type Table = '[] :=>
1126− '[ "a" ::: 'NoDef :=> 'Null 'PGint4
1127− , "b" ::: 'NoDef :=> 'Null 'PGfloat4 ]
1128−:}
1129−
1130−>>> :{
1131−let
1132− setup :: Definition (Public '[]) (Public '["tab" ::: 'Table Table, "ix" ::: 'Index 'Btree])
1133− setup =
1134− createTable #tab (nullable int `as` #a :* nullable real `as` #b) Nil >>>
1135− createIndex #ix #tab btree [#a & AscNullsFirst, #b & AscNullsLast]
1136−in printSQL setup
1137−:}
1138−CREATE TABLE "tab" ("a" int NULL, "b" real NULL);
1139−CREATE INDEX "ix" ON "tab" USING btree (("a") ASC NULLS FIRST, ("b") ASC NULLS LAST);
1140−-}
1141−createIndex
1142− :: (Has sch schemas schema, Has tab schema ('Table table), KnownSymbol ix)
1143− => Alias ix
1144− -> QualifiedAlias sch tab
1145− -> IndexMethod method
1146− -> [SortExpression '[] '[] 'Ungrouped schemas '[] '[tab ::: TableToRow table]]
1147− -> Definition schemas (Alter sch (Create ix ('Index method) schema) schemas)
1148−createIndex ix tab method cols = UnsafeDefinition $
1149− "CREATE" <+> "INDEX" <+> renderSQL ix <+> "ON" <+> renderSQL tab
1150− <+> "USING" <+> renderSQL method
1151− <+> parenthesized (commaSeparated (renderIndex <$> cols))
1152− <> ";"
1153− where
1154− renderIndex = \case
1155− Asc expression -> parenthesized (renderSQL expression) <+> "ASC"
1156− Desc expression -> parenthesized (renderSQL expression) <+> "DESC"
1157− AscNullsFirst expression -> parenthesized (renderSQL expression)
1158− <+> "ASC NULLS FIRST"
1159− DescNullsFirst expression -> parenthesized (renderSQL expression)
1160− <+> "DESC NULLS FIRST"
1161− AscNullsLast expression -> parenthesized (renderSQL expression)
1162− <+> "ASC NULLS LAST"
1163− DescNullsLast expression -> parenthesized (renderSQL expression)
1164− <+> "DESC NULLS LAST"
1165−
1166−createIndexIfNotExists
1167− :: (Has sch schemas schema, Has tab schema ('Table table), KnownSymbol ix)
1168− => Alias ix
1169− -> QualifiedAlias sch tab
1170− -> IndexMethod method
1171− -> [SortExpression '[] '[] 'Ungrouped schemas '[] '[tab ::: TableToRow table]]
1172− -> Definition schemas (Alter sch (CreateIfNotExists ix ('Index method) schema) schemas)
1173−createIndexIfNotExists ix tab method cols = UnsafeDefinition $
1174− "CREATE INDEX IF NOT EXISTS" <+> renderSQL ix <+> "ON" <+> renderSQL tab
1175− <+> "USING" <+> renderSQL method
1176− <+> parenthesized (commaSeparated (renderIndex <$> cols))
1177− <> ";"
1178− where
1179− renderIndex = \case
1180− Asc expression -> parenthesized (renderSQL expression) <+> "ASC"
1181− Desc expression -> parenthesized (renderSQL expression) <+> "DESC"
1182− AscNullsFirst expression -> parenthesized (renderSQL expression)
1183− <+> "ASC NULLS FIRST"
1184− DescNullsFirst expression -> parenthesized (renderSQL expression)
1185− <+> "DESC NULLS FIRST"
1186− AscNullsLast expression -> parenthesized (renderSQL expression)
1187− <+> "ASC NULLS LAST"
1188− DescNullsLast expression -> parenthesized (renderSQL expression)
1189− <+> "DESC NULLS LAST"
1190−
1191−newtype IndexMethod ty = UnsafeIndexMethod {renderIndexMethod :: ByteString}
1192− deriving stock (Eq, Ord, Show, GHC.Generic)
1193−instance RenderSQL (IndexMethod ty) where renderSQL = renderIndexMethod
1194−btree :: IndexMethod 'Btree
1195−btree = UnsafeIndexMethod "btree"
1196−hash :: IndexMethod 'Hash
1197−hash = UnsafeIndexMethod "hash"
1198−gist :: IndexMethod 'Gist
1199−gist = UnsafeIndexMethod "gist"
1200−spgist :: IndexMethod 'Spgist
1201−spgist = UnsafeIndexMethod "spgist"
1202−gin :: IndexMethod 'Gin
1203−gin = UnsafeIndexMethod "gin"
1204−brin :: IndexMethod 'Brin
1205−brin = UnsafeIndexMethod "brin"
1206−
1207−createFunction
1208− :: ( Has sch schemas schema
1209− , KnownSymbol fun
1210− , SOP.SListI args )
1211− => QualifiedAlias sch fun
1212− -> NP (TypeExpression schemas) args
1213− -> TypeExpression schemas ret
1214− -> FunctionDefinition schemas args ('Returns ret)
1215− -> Definition schemas (Alter sch (Create fun ('Function (args '::--> 'Returns ret)) schema) schemas)
1216−createFunction fun args ret fundef = UnsafeDefinition $
1217− "CREATE" <+> "FUNCTION" <+> renderSQL fun
1218− <+> parenthesized (renderCommaSeparated renderSQL args)
1219− <+> "RETURNS" <+> renderSQL ret <+> renderSQL fundef <> ";"
1220−
1221−createOrReplaceFunction
1222− :: ( Has sch schemas schema
1223− , KnownSymbol fun
1224− , SOP.SListI args )
1225− => QualifiedAlias sch fun
1226− -> NP (TypeExpression schemas) args
1227− -> TypeExpression schemas ret
1228− -> FunctionDefinition schemas args ('Returns ret)
1229− -> Definition schemas (Alter sch (CreateOrReplace fun ('Function (args '::--> 'Returns ret)) schema) schemas)
1230−createOrReplaceFunction fun args ret fundef = UnsafeDefinition $
1231− "CREATE" <+> "OR" <+> "REPLACE" <+> "FUNCTION" <+> renderSQL fun
1232− <+> parenthesized (renderCommaSeparated renderSQL args)
1233− <+> "RETURNS" <+> renderSQL ret <+> renderSQL fundef <> ";"
1234−
1235−languageSqlExpr
1236− :: Expression '[] '[] 'Ungrouped schemas args '[] ret
1237− -> FunctionDefinition schemas args ('Returns ret)
1238−languageSqlExpr expr = UnsafeFunctionDefinition $
1239− "language sql as"
1240− <+> "$$" <+> renderSQL (values_ (expr `as` #ret)) <+> "$$"
1241−
1242−languageSqlQuery
1243− :: Query '[] '[] schemas args rets
1244− -> FunctionDefinition schemas args ('ReturnsTable rets)
1245−languageSqlQuery qry = UnsafeFunctionDefinition $
1246− "language sql as" <+> "$$" <+> renderSQL qry <+> "$$"
1247−
1248−createSetFunction
1249− :: ( Has sch schemas schema
1250− , KnownSymbol fun
1251− , SOP.SListI args
1252− , SOP.SListI rets )
1253− => QualifiedAlias sch fun
1254− -> NP (TypeExpression schemas) args
1255− -> NP (Aliased (TypeExpression schemas)) rets
1256− -> FunctionDefinition schemas args ('ReturnsTable rets)
1257− -> Definition schemas (Alter sch (Create fun ('Function (args '::--> 'ReturnsTable rets)) schema) schemas)
1258−createSetFunction fun args rets fundef = UnsafeDefinition $
1259− "CREATE" <+> "FUNCTION" <+> renderSQL fun
1260− <+> parenthesized (renderCommaSeparated renderSQL args)
1261− <+> "RETURNS" <+> "TABLE"
1262− <+> parenthesized (renderCommaSeparated renderRet rets)
1263− <+> renderSQL fundef <> ";"
1264− where
1265− renderRet :: Aliased (TypeExpression s) r -> ByteString
1266− renderRet (ty `As` col) = renderSQL col <+> renderSQL ty
1267−
1268−createOrReplaceSetFunction
1269− :: ( Has sch schemas schema
1270− , Has fun schema ('Function (args0 '::--> ret0))
1271− , SOP.SListI args
1272− , SOP.SListI rets )
1273− => QualifiedAlias sch fun
1274− -> NP (TypeExpression schemas) args
1275− -> NP (Aliased (TypeExpression schemas)) rets
1276− -> FunctionDefinition schemas args ('ReturnsTable rets)
1277− -> Definition schemas (Alter sch (CreateOrReplace fun ('Function (args '::--> 'ReturnsTable rets)) schema) schemas)
1278−createOrReplaceSetFunction fun args rets fundef = UnsafeDefinition $
1279− "CREATE" <+> "OR" <+> "REPLACE" <+> "FUNCTION" <+> renderSQL fun
1280− <+> parenthesized (renderCommaSeparated renderSQL args)
1281− <+> "RETURNS" <+> "TABLE"
1282− <+> parenthesized (renderCommaSeparated renderRet rets)
1283− <+> renderSQL fundef <> ";"
1284− where
1285− renderRet :: Aliased (TypeExpression s) r -> ByteString
1286− renderRet (ty `As` col) = renderSQL col <+> renderSQL ty
1287−
1288−createBinaryOp
1289− :: forall op fun sch schemas schema x y z.
1290− ( Has sch schemas schema
1291− , Has fun schema ('Function ('[x,y] '::--> 'Returns z))
1292− , KnownSymbol op )
1293− => Alias fun
1294− -> TypeExpression schemas x
1295− -> TypeExpression schemas y
1296− -> Definition schemas
1297− (Alter sch (Create op ('Operator ('BinaryOp x y z)) schema) schemas)
1298−createBinaryOp fun x y = UnsafeDefinition $
1299− "CREATE" <+> "OPERATOR" <+> renderSymbol @op
1300− <+> parenthesized (commaSeparated opdef)
1301− where
1302− opdef =
1303− [ "FUNCTION" <+> "=" <+> renderSQL fun
1304− , "LEFTARG" <+> "=" <+> renderSQL x
1305− , "RIGHTARG" <+> "=" <+> renderSQL y ]
1306−
1307−createLeftOp
1308− :: forall op fun sch schemas schema x y.
1309− ( Has sch schemas schema
1310− , Has fun schema ('Function ('[x] '::--> 'Returns y))
1311− , KnownSymbol op )
1312− => Alias fun
1313− -> TypeExpression schemas x
1314− -> Definition schemas
1315− (Alter sch (Create op ('Operator ('LeftOp x y)) schema) schemas)
1316−createLeftOp fun x = UnsafeDefinition $
1317− "CREATE" <+> "OPERATOR" <+> renderSymbol @op
1318− <+> parenthesized (commaSeparated opdef)
1319− where
1320− opdef =
1321− [ "FUNCTION" <+> "=" <+> renderSQL fun
1322− , "RIGHTARG" <+> "=" <+> renderSQL x ]
1323−
1324−createRightOp
1325− :: forall op fun sch schemas schema x y.
1326− ( Has sch schemas schema
1327− , Has fun schema ('Function ('[x] '::--> 'Returns y))
1328− , KnownSymbol op )
1329− => Alias fun
1330− -> TypeExpression schemas x
1331− -> Definition schemas
1332− (Alter sch (Create op ('Operator ('RightOp x y)) schema) schemas)
1333−createRightOp fun x = UnsafeDefinition $
1334− "CREATE" <+> "OPERATOR" <+> renderSymbol @op
1335− <+> parenthesized (commaSeparated opdef)
1336− where
1337− opdef =
1338− [ "FUNCTION" <+> "=" <+> renderSQL fun
1339− , "LEFTARG" <+> "=" <+> renderSQL x ]
1340−
1341−dropFunction
1342− :: (Has sch schemas schema, KnownSymbol fun)
1343− => QualifiedAlias sch fun
1344− -- ^ name of the user defined function
1345− -> Definition schemas (Alter sch (DropSchemum fun 'Function schema) schemas)
1346−dropFunction fun = UnsafeDefinition $
1347− "DROP" <+> "FUNCTION" <+> renderSQL fun <> ";"
1348−
1349−dropFunctionIfExists
1350− :: (Has sch schemas schema, KnownSymbol fun)
1351− => QualifiedAlias sch fun
1352− -- ^ name of the user defined function
1353− -> Definition schemas (Alter sch (DropSchemumIfExists fun 'Function schema) schemas)
1354−dropFunctionIfExists fun = UnsafeDefinition $
1355− "DROP FUNCTION IF EXISTS" <+> renderSQL fun <> ";"
1356−
1357−dropOperator
1358− :: (Has sch schemas schema, KnownSymbol op)
1359− => QualifiedAlias sch op
1360− -- ^ name of the user defined operator
1361− -> Definition schemas (Alter sch (DropSchemum op 'Operator schema) schemas)
1362−dropOperator op = UnsafeDefinition $
1363− "DROP" <+> "OPERATOR" <+> renderSQL op <> ";"
1364−
1365−dropOperatorIfExists
1366− :: (Has sch schemas schema, KnownSymbol op)
1367− => QualifiedAlias sch op
1368− -- ^ name of the user defined operator
1369− -> Definition schemas (Alter sch (DropSchemumIfExists op 'Operator schema) schemas)
1370−dropOperatorIfExists op = UnsafeDefinition $
1371− "DROP OPERATOR IF EXISTS" <+> renderSQL op <> ";"
1372−
1373−newtype FunctionDefinition schemas args ret = UnsafeFunctionDefinition
1374− { renderFunctionDefinition :: ByteString }
1375− deriving (Eq,Show,GHC.Generic,NFData)
1376−instance RenderSQL (FunctionDefinition schemas args ret) where
1377− renderSQL = renderFunctionDefinition
1378−
1379−-- |
1380−-- >>> printSQL (dropIndex #ix :: Definition (Public '["ix" ::: 'Index 'Btree]) (Public '[]))
1381−-- DROP INDEX "ix";
1382−dropIndex
1383− :: (Has sch schemas schema, KnownSymbol ix)
1384− => QualifiedAlias sch ix
1385− -- ^ name of the user defined index
1386− -> Definition schemas (Alter sch (DropSchemum ix 'Index schema) schemas)
1387−dropIndex ix = UnsafeDefinition $ "DROP" <+> "INDEX" <+> renderSQL ix <> ";"
1388−
1389−dropIndexIfExists
1390− :: (Has sch schemas schema, KnownSymbol ix)
1391− => QualifiedAlias sch ix
1392− -- ^ name of the user defined index
1393− -> Definition schemas (Alter sch (DropSchemumIfExists ix 'Index schema) schemas)
1394−dropIndexIfExists ix = UnsafeDefinition $ "DROP INDEX IF EXISTS" <+> renderSQL ix <> ";"
1395−
1396−-- | Lift `PGTyped` to a field
1397−class FieldTyped schemas ty where
1398− fieldtype :: Aliased (TypeExpression schemas) ty
1399−instance (KnownSymbol alias, PGTyped schemas ty)
1400− => FieldTyped schemas (alias ::: ty) where
1401− fieldtype = pgtype `As` Alias
1402−
1403−-- | Drop a type.
1404−--
1405−-- >>> data Schwarma = Beef | Lamb | Chicken deriving GHC.Generic
1406−-- >>> instance SOP.Generic Schwarma
1407−-- >>> instance SOP.HasDatatypeInfo Schwarma
1408−-- >>> printSQL (dropType #schwarma :: Definition '["public" ::: '["schwarma" ::: 'Typedef (PG (Enumerated Schwarma))]] (Public '[]))
1409−-- DROP TYPE "schwarma";
1410−dropType
1411− :: (Has sch schemas schema, KnownSymbol td)
1412− => QualifiedAlias sch td
1413− -- ^ name of the user defined type
1414− -> Definition schemas (Alter sch (DropSchemum td 'Typedef schema) schemas)
1415−dropType tydef = UnsafeDefinition $ "DROP" <+> "TYPE" <+> renderSQL tydef <> ";"
1416−
1417−dropTypeIfExists
1418− :: (Has sch schemas schema, KnownSymbol td)
1419− => QualifiedAlias sch td
1420− -- ^ name of the user defined type
1421− -> Definition schemas (Alter sch (DropSchemumIfExists td 'Typedef schema) schemas)
1422−dropTypeIfExists tydef = UnsafeDefinition $ "DROP" <+> "TYPE" <+> renderSQL tydef <> ";"
1423−
1424−-- | `ColumnTypeExpression`s are used in `createTable` commands.
1425−newtype ColumnTypeExpression (schemas :: SchemasType) (ty :: ColumnType)
1426− = UnsafeColumnTypeExpression { renderColumnTypeExpression :: ByteString }
1427− deriving (GHC.Generic,Show,Eq,Ord,NFData)
1428−instance RenderSQL (ColumnTypeExpression schemas ty) where
1429− renderSQL = renderColumnTypeExpression
1430−
1431−-- | used in `createTable` commands as a column constraint to note that
1432−-- @NULL@ may be present in a column
1433−nullable
1434− :: TypeExpression schemas (nullity ty)
1435− -> ColumnTypeExpression schemas ('NoDef :=> 'Null ty)
1436−nullable ty = UnsafeColumnTypeExpression $ renderSQL ty <+> "NULL"
1437−
1438−-- | used in `createTable` commands as a column constraint to ensure
1439−-- @NULL@ is not present in a column
1440−notNullable
1441− :: TypeExpression schemas (nullity ty)
1442− -> ColumnTypeExpression schemas ('NoDef :=> 'NotNull ty)
1443−notNullable ty = UnsafeColumnTypeExpression $ renderSQL ty <+> "NOT NULL"
1444−
1445−-- | used in `createTable` commands as a column constraint to give a default
1446−default_
1447− :: Expression '[] '[] 'Ungrouped schemas '[] '[] ty
1448− -> ColumnTypeExpression schemas ('NoDef :=> ty)
1449− -> ColumnTypeExpression schemas ('Def :=> ty)
1450−default_ x ty = UnsafeColumnTypeExpression $
1451− renderSQL ty <+> "DEFAULT" <+> renderExpression x
1452−
1453−-- | not a true type, but merely a notational convenience for creating
1454−-- unique identifier columns with type `PGint2`
1455−serial2, smallserial
1456− :: ColumnTypeExpression schemas ('Def :=> 'NotNull 'PGint2)
1457−serial2 = UnsafeColumnTypeExpression "serial2"
1458−smallserial = UnsafeColumnTypeExpression "smallserial"
1459−-- | not a true type, but merely a notational convenience for creating
1460−-- unique identifier columns with type `PGint4`
1461−serial4, serial
1462− :: ColumnTypeExpression schemas ('Def :=> 'NotNull 'PGint4)
1463−serial4 = UnsafeColumnTypeExpression "serial4"
1464−serial = UnsafeColumnTypeExpression "serial"
1465−-- | not a true type, but merely a notational convenience for creating
1466−-- unique identifier columns with type `PGint8`
1467−serial8, bigserial
1468− :: ColumnTypeExpression schemas ('Def :=> 'NotNull 'PGint8)
1469−serial8 = UnsafeColumnTypeExpression "serial8"
1470−bigserial = UnsafeColumnTypeExpression "bigserial"
squeal-postgresql/src/Squeal/PostgreSQL/Definition/Function.hsnew file+238
@@ -0,0 +1,238 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.Function
3+Description: Create and drop function and operator definitions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Create and drop function and operator definitions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.Function
33+ ( createFunction
34+ , createOrReplaceFunction
35+ , createSetFunction
36+ , createOrReplaceSetFunction
37+ , createBinaryOp
38+ , createLeftOp
39+ , createRightOp
40+ , dropFunction
41+ , dropFunctionIfExists
42+ , dropOperator
43+ , dropOperatorIfExists
44+ , FunctionDefinition(..)
45+ , languageSqlExpr
46+ , languageSqlQuery
47+ ) where
48+
49+import Control.DeepSeq
50+import Data.ByteString
51+import GHC.TypeLits
52+
53+import qualified Generics.SOP as SOP
54+import qualified GHC.Generics as GHC
55+
56+import Squeal.PostgreSQL.Alias
57+import Squeal.PostgreSQL.Definition
58+import Squeal.PostgreSQL.Expression
59+import Squeal.PostgreSQL.Expression.Type
60+import Squeal.PostgreSQL.List
61+import Squeal.PostgreSQL.Query
62+import Squeal.PostgreSQL.Render
63+import Squeal.PostgreSQL.Schema
64+
65+-- $setup
66+-- >>> import Squeal.PostgreSQL
67+
68+createFunction
69+ :: ( Has sch schemas schema
70+ , KnownSymbol fun
71+ , SOP.SListI args )
72+ => QualifiedAlias sch fun
73+ -> NP (TypeExpression schemas) args
74+ -> TypeExpression schemas ret
75+ -> FunctionDefinition schemas args ('Returns ret)
76+ -> Definition schemas (Alter sch (Create fun ('Function (args :=> 'Returns ret)) schema) schemas)
77+createFunction fun args ret fundef = UnsafeDefinition $
78+ "CREATE" <+> "FUNCTION" <+> renderSQL fun
79+ <+> parenthesized (renderCommaSeparated renderSQL args)
80+ <+> "RETURNS" <+> renderSQL ret <+> renderSQL fundef <> ";"
81+
82+createOrReplaceFunction
83+ :: ( Has sch schemas schema
84+ , KnownSymbol fun
85+ , SOP.SListI args )
86+ => QualifiedAlias sch fun
87+ -> NP (TypeExpression schemas) args
88+ -> TypeExpression schemas ret
89+ -> FunctionDefinition schemas args ('Returns ret)
90+ -> Definition schemas (Alter sch (CreateOrReplace fun ('Function (args :=> 'Returns ret)) schema) schemas)
91+createOrReplaceFunction fun args ret fundef = UnsafeDefinition $
92+ "CREATE" <+> "OR" <+> "REPLACE" <+> "FUNCTION" <+> renderSQL fun
93+ <+> parenthesized (renderCommaSeparated renderSQL args)
94+ <+> "RETURNS" <+> renderSQL ret <+> renderSQL fundef <> ";"
95+
96+languageSqlExpr
97+ :: Expression '[] '[] 'Ungrouped schemas args '[] ret
98+ -> FunctionDefinition schemas args ('Returns ret)
99+languageSqlExpr expr = UnsafeFunctionDefinition $
100+ "language sql as"
101+ <+> "$$" <+> renderSQL (values_ (expr `as` #ret)) <+> "$$"
102+
103+languageSqlQuery
104+ :: Query '[] '[] schemas args rets
105+ -> FunctionDefinition schemas args ('ReturnsTable rets)
106+languageSqlQuery qry = UnsafeFunctionDefinition $
107+ "language sql as" <+> "$$" <+> renderSQL qry <+> "$$"
108+
109+createSetFunction
110+ :: ( Has sch schemas schema
111+ , KnownSymbol fun
112+ , SOP.SListI args
113+ , SOP.SListI rets )
114+ => QualifiedAlias sch fun
115+ -> NP (TypeExpression schemas) args
116+ -> NP (Aliased (TypeExpression schemas)) rets
117+ -> FunctionDefinition schemas args ('ReturnsTable rets)
118+ -> Definition schemas (Alter sch (Create fun ('Function (args :=> 'ReturnsTable rets)) schema) schemas)
119+createSetFunction fun args rets fundef = UnsafeDefinition $
120+ "CREATE" <+> "FUNCTION" <+> renderSQL fun
121+ <+> parenthesized (renderCommaSeparated renderSQL args)
122+ <+> "RETURNS" <+> "TABLE"
123+ <+> parenthesized (renderCommaSeparated renderRet rets)
124+ <+> renderSQL fundef <> ";"
125+ where
126+ renderRet :: Aliased (TypeExpression s) r -> ByteString
127+ renderRet (ty `As` col) = renderSQL col <+> renderSQL ty
128+
129+createOrReplaceSetFunction
130+ :: ( Has sch schemas schema
131+ , KnownSymbol fun
132+ , SOP.SListI args
133+ , SOP.SListI rets )
134+ => QualifiedAlias sch fun
135+ -> NP (TypeExpression schemas) args
136+ -> NP (Aliased (TypeExpression schemas)) rets
137+ -> FunctionDefinition schemas args ('ReturnsTable rets)
138+ -> Definition schemas (Alter sch (CreateOrReplace fun ('Function (args :=> 'ReturnsTable rets)) schema) schemas)
139+createOrReplaceSetFunction fun args rets fundef = UnsafeDefinition $
140+ "CREATE" <+> "OR" <+> "REPLACE" <+> "FUNCTION" <+> renderSQL fun
141+ <+> parenthesized (renderCommaSeparated renderSQL args)
142+ <+> "RETURNS" <+> "TABLE"
143+ <+> parenthesized (renderCommaSeparated renderRet rets)
144+ <+> renderSQL fundef <> ";"
145+ where
146+ renderRet :: Aliased (TypeExpression s) r -> ByteString
147+ renderRet (ty `As` col) = renderSQL col <+> renderSQL ty
148+
149+createBinaryOp
150+ :: forall op fun sch schemas schema x y z.
151+ ( Has sch schemas schema
152+ , Has fun schema ('Function ('[x,y] :=> 'Returns z))
153+ , KnownSymbol op )
154+ => QualifiedAlias sch fun
155+ -> TypeExpression schemas x
156+ -> TypeExpression schemas y
157+ -> Definition schemas
158+ (Alter sch (Create op ('Operator ('BinaryOp x y z)) schema) schemas)
159+createBinaryOp fun x y = UnsafeDefinition $
160+ "CREATE" <+> "OPERATOR" <+> renderSymbol @op
161+ <+> parenthesized (commaSeparated opdef)
162+ where
163+ opdef =
164+ [ "FUNCTION" <+> "=" <+> renderSQL fun
165+ , "LEFTARG" <+> "=" <+> renderSQL x
166+ , "RIGHTARG" <+> "=" <+> renderSQL y ]
167+
168+createLeftOp
169+ :: forall op fun sch schemas schema x y.
170+ ( Has sch schemas schema
171+ , Has fun schema ('Function ('[x] :=> 'Returns y))
172+ , KnownSymbol op )
173+ => QualifiedAlias sch fun
174+ -> TypeExpression schemas x
175+ -> Definition schemas
176+ (Alter sch (Create op ('Operator ('LeftOp x y)) schema) schemas)
177+createLeftOp fun x = UnsafeDefinition $
178+ "CREATE" <+> "OPERATOR" <+> renderSymbol @op
179+ <+> parenthesized (commaSeparated opdef)
180+ where
181+ opdef =
182+ [ "FUNCTION" <+> "=" <+> renderSQL fun
183+ , "RIGHTARG" <+> "=" <+> renderSQL x ]
184+
185+createRightOp
186+ :: forall op fun sch schemas schema x y.
187+ ( Has sch schemas schema
188+ , Has fun schema ('Function ('[x] :=> 'Returns y))
189+ , KnownSymbol op )
190+ => QualifiedAlias sch fun
191+ -> TypeExpression schemas x
192+ -> Definition schemas
193+ (Alter sch (Create op ('Operator ('RightOp x y)) schema) schemas)
194+createRightOp fun x = UnsafeDefinition $
195+ "CREATE" <+> "OPERATOR" <+> renderSymbol @op
196+ <+> parenthesized (commaSeparated opdef)
197+ where
198+ opdef =
199+ [ "FUNCTION" <+> "=" <+> renderSQL fun
200+ , "LEFTARG" <+> "=" <+> renderSQL x ]
201+
202+dropFunction
203+ :: (Has sch schemas schema, KnownSymbol fun)
204+ => QualifiedAlias sch fun
205+ -- ^ name of the user defined function
206+ -> Definition schemas (Alter sch (DropSchemum fun 'Function schema) schemas)
207+dropFunction fun = UnsafeDefinition $
208+ "DROP" <+> "FUNCTION" <+> renderSQL fun <> ";"
209+
210+dropFunctionIfExists
211+ :: (Has sch schemas schema, KnownSymbol fun)
212+ => QualifiedAlias sch fun
213+ -- ^ name of the user defined function
214+ -> Definition schemas (Alter sch (DropSchemumIfExists fun 'Function schema) schemas)
215+dropFunctionIfExists fun = UnsafeDefinition $
216+ "DROP FUNCTION IF EXISTS" <+> renderSQL fun <> ";"
217+
218+dropOperator
219+ :: (Has sch schemas schema, KnownSymbol op)
220+ => QualifiedAlias sch op
221+ -- ^ name of the user defined operator
222+ -> Definition schemas (Alter sch (DropSchemum op 'Operator schema) schemas)
223+dropOperator op = UnsafeDefinition $
224+ "DROP" <+> "OPERATOR" <+> renderSQL op <> ";"
225+
226+dropOperatorIfExists
227+ :: (Has sch schemas schema, KnownSymbol op)
228+ => QualifiedAlias sch op
229+ -- ^ name of the user defined operator
230+ -> Definition schemas (Alter sch (DropSchemumIfExists op 'Operator schema) schemas)
231+dropOperatorIfExists op = UnsafeDefinition $
232+ "DROP OPERATOR IF EXISTS" <+> renderSQL op <> ";"
233+
234+newtype FunctionDefinition schemas args ret = UnsafeFunctionDefinition
235+ { renderFunctionDefinition :: ByteString }
236+ deriving (Eq,Show,GHC.Generic,NFData)
237+instance RenderSQL (FunctionDefinition schemas args ret) where
238+ renderSQL = renderFunctionDefinition
squeal-postgresql/src/Squeal/PostgreSQL/Definition/Index.hsnew file+159
@@ -0,0 +1,159 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.Index
3+Description: Create and drop index definitions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Create and drop index definitions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.Index
33+ ( createIndex
34+ , createIndexIfNotExists
35+ , dropIndex
36+ , dropIndexIfExists
37+ , IndexMethod (..)
38+ , btree
39+ , hash
40+ , gist
41+ , spgist
42+ , gin
43+ , brin
44+ ) where
45+
46+import Data.ByteString
47+import GHC.TypeLits
48+
49+import qualified GHC.Generics as GHC
50+
51+import Squeal.PostgreSQL.Alias
52+import Squeal.PostgreSQL.Definition
53+import Squeal.PostgreSQL.Expression.Sort
54+import Squeal.PostgreSQL.Render
55+import Squeal.PostgreSQL.Schema
56+
57+-- $setup
58+-- >>> import Squeal.PostgreSQL
59+
60+{- |
61+>>> :{
62+type Table = '[] :=>
63+ '[ "a" ::: 'NoDef :=> 'Null 'PGint4
64+ , "b" ::: 'NoDef :=> 'Null 'PGfloat4 ]
65+:}
66+
67+>>> :{
68+let
69+ setup :: Definition (Public '[]) (Public '["tab" ::: 'Table Table, "ix" ::: 'Index 'Btree])
70+ setup =
71+ createTable #tab (nullable int `as` #a :* nullable real `as` #b) Nil >>>
72+ createIndex #ix #tab btree [#a & AscNullsFirst, #b & AscNullsLast]
73+in printSQL setup
74+:}
75+CREATE TABLE "tab" ("a" int NULL, "b" real NULL);
76+CREATE INDEX "ix" ON "tab" USING btree (("a") ASC NULLS FIRST, ("b") ASC NULLS LAST);
77+-}
78+createIndex
79+ :: (Has sch schemas schema, Has tab schema ('Table table), KnownSymbol ix)
80+ => Alias ix
81+ -> QualifiedAlias sch tab
82+ -> IndexMethod method
83+ -> [SortExpression '[] '[] 'Ungrouped schemas '[] '[tab ::: TableToRow table]]
84+ -> Definition schemas (Alter sch (Create ix ('Index method) schema) schemas)
85+createIndex ix tab method cols = UnsafeDefinition $
86+ "CREATE" <+> "INDEX" <+> renderSQL ix <+> "ON" <+> renderSQL tab
87+ <+> "USING" <+> renderSQL method
88+ <+> parenthesized (commaSeparated (renderIndex <$> cols))
89+ <> ";"
90+ where
91+ renderIndex = \case
92+ Asc expression -> parenthesized (renderSQL expression) <+> "ASC"
93+ Desc expression -> parenthesized (renderSQL expression) <+> "DESC"
94+ AscNullsFirst expression -> parenthesized (renderSQL expression)
95+ <+> "ASC NULLS FIRST"
96+ DescNullsFirst expression -> parenthesized (renderSQL expression)
97+ <+> "DESC NULLS FIRST"
98+ AscNullsLast expression -> parenthesized (renderSQL expression)
99+ <+> "ASC NULLS LAST"
100+ DescNullsLast expression -> parenthesized (renderSQL expression)
101+ <+> "DESC NULLS LAST"
102+
103+createIndexIfNotExists
104+ :: (Has sch schemas schema, Has tab schema ('Table table), KnownSymbol ix)
105+ => Alias ix
106+ -> QualifiedAlias sch tab
107+ -> IndexMethod method
108+ -> [SortExpression '[] '[] 'Ungrouped schemas '[] '[tab ::: TableToRow table]]
109+ -> Definition schemas (Alter sch (CreateIfNotExists ix ('Index method) schema) schemas)
110+createIndexIfNotExists ix tab method cols = UnsafeDefinition $
111+ "CREATE INDEX IF NOT EXISTS" <+> renderSQL ix <+> "ON" <+> renderSQL tab
112+ <+> "USING" <+> renderSQL method
113+ <+> parenthesized (commaSeparated (renderIndex <$> cols))
114+ <> ";"
115+ where
116+ renderIndex = \case
117+ Asc expression -> parenthesized (renderSQL expression) <+> "ASC"
118+ Desc expression -> parenthesized (renderSQL expression) <+> "DESC"
119+ AscNullsFirst expression -> parenthesized (renderSQL expression)
120+ <+> "ASC NULLS FIRST"
121+ DescNullsFirst expression -> parenthesized (renderSQL expression)
122+ <+> "DESC NULLS FIRST"
123+ AscNullsLast expression -> parenthesized (renderSQL expression)
124+ <+> "ASC NULLS LAST"
125+ DescNullsLast expression -> parenthesized (renderSQL expression)
126+ <+> "DESC NULLS LAST"
127+
128+newtype IndexMethod ty = UnsafeIndexMethod {renderIndexMethod :: ByteString}
129+ deriving stock (Eq, Ord, Show, GHC.Generic)
130+instance RenderSQL (IndexMethod ty) where renderSQL = renderIndexMethod
131+btree :: IndexMethod 'Btree
132+btree = UnsafeIndexMethod "btree"
133+hash :: IndexMethod 'Hash
134+hash = UnsafeIndexMethod "hash"
135+gist :: IndexMethod 'Gist
136+gist = UnsafeIndexMethod "gist"
137+spgist :: IndexMethod 'Spgist
138+spgist = UnsafeIndexMethod "spgist"
139+gin :: IndexMethod 'Gin
140+gin = UnsafeIndexMethod "gin"
141+brin :: IndexMethod 'Brin
142+brin = UnsafeIndexMethod "brin"
143+
144+-- |
145+-- >>> printSQL (dropIndex #ix :: Definition (Public '["ix" ::: 'Index 'Btree]) (Public '[]))
146+-- DROP INDEX "ix";
147+dropIndex
148+ :: (Has sch schemas schema, KnownSymbol ix)
149+ => QualifiedAlias sch ix
150+ -- ^ name of the user defined index
151+ -> Definition schemas (Alter sch (DropSchemum ix 'Index schema) schemas)
152+dropIndex ix = UnsafeDefinition $ "DROP" <+> "INDEX" <+> renderSQL ix <> ";"
153+
154+dropIndexIfExists
155+ :: (Has sch schemas schema, KnownSymbol ix)
156+ => QualifiedAlias sch ix
157+ -- ^ name of the user defined index
158+ -> Definition schemas (Alter sch (DropSchemumIfExists ix 'Index schema) schemas)
159+dropIndexIfExists ix = UnsafeDefinition $ "DROP INDEX IF EXISTS" <+> renderSQL ix <> ";"
squeal-postgresql/src/Squeal/PostgreSQL/Definition/Schema.hsnew file+96
@@ -0,0 +1,96 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.Schema
3+Description: Create and drop schema definitions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Create and drop schema definitions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.Schema
33+ ( createSchema
34+ , createSchemaIfNotExists
35+ , dropSchema
36+ , dropSchemaIfExists
37+ ) where
38+
39+import GHC.TypeLits
40+
41+import Squeal.PostgreSQL.Alias
42+import Squeal.PostgreSQL.Definition
43+import Squeal.PostgreSQL.Render
44+import Squeal.PostgreSQL.Schema
45+
46+-- $setup
47+-- >>> import Squeal.PostgreSQL
48+
49+{- |
50+`createSchema` enters a new schema into the current database.
51+The schema name must be distinct from the name of any existing schema
52+in the current database.
53+
54+A schema is essentially a namespace: it contains named objects
55+(tables, data types, functions, and operators) whose names
56+can duplicate those of other objects existing in other schemas.
57+Named objects are accessed by `QualifiedAlias`es with the schema
58+name as a prefix.
59+-}
60+createSchema
61+ :: KnownSymbol sch
62+ => Alias sch
63+ -> Definition schemas (Create sch '[] schemas)
64+createSchema sch = UnsafeDefinition $
65+ "CREATE" <+> "SCHEMA" <+> renderSQL sch <> ";"
66+
67+{- | Idempotent version of `createSchema`. -}
68+createSchemaIfNotExists
69+ :: (KnownSymbol sch, Has sch schemas schema)
70+ => Alias sch
71+ -> Definition schemas (CreateIfNotExists sch '[] schemas)
72+createSchemaIfNotExists sch = UnsafeDefinition $
73+ "CREATE" <+> "SCHEMA" <+> "IF" <+> "NOT" <+> "EXISTS"
74+ <+> renderSQL sch <> ";"
75+
76+-- | >>> :{
77+-- let
78+-- definition :: Definition '["muh_schema" ::: schema, "public" ::: public] '["public" ::: public]
79+-- definition = dropSchema #muh_schema
80+-- :}
81+--
82+-- >>> printSQL definition
83+-- DROP SCHEMA "muh_schema";
84+dropSchema
85+ :: KnownSymbol sch
86+ => Alias sch
87+ -- ^ user defined schema
88+ -> Definition schemas (Drop sch schemas)
89+dropSchema sch = UnsafeDefinition $ "DROP SCHEMA" <+> renderSQL sch <> ";"
90+
91+dropSchemaIfExists
92+ :: KnownSymbol sch
93+ => Alias sch
94+ -- ^ user defined schema
95+ -> Definition schemas (DropIfExists sch schemas)
96+dropSchemaIfExists sch = UnsafeDefinition $ "DROP SCHEMA" <+> renderSQL sch <> ";"
squeal-postgresql/src/Squeal/PostgreSQL/Definition/Table.hsnew file+500
@@ -0,0 +1,500 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.Table
3+Description: Create, drop and alter table definitions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Create, drop and alter table definitions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.Table
33+ ( createTable
34+ , createTableIfNotExists
35+ , dropTable
36+ , dropTableIfExists
37+ , alterTable
38+ , alterTableIfExists
39+ , alterTableRename
40+ , alterTableIfExistsRename
41+ , AlterTable (..)
42+ , addConstraint
43+ , dropConstraint
44+ , AddColumn (..)
45+ , dropColumn
46+ , renameColumn
47+ , alterColumn
48+ , AlterColumn (..)
49+ , setDefault
50+ , dropDefault
51+ , setNotNull
52+ , dropNotNull
53+ , alterType
54+ ) where
55+
56+import Control.DeepSeq
57+import Data.ByteString
58+import GHC.TypeLits
59+
60+import qualified Generics.SOP as SOP
61+import qualified GHC.Generics as GHC
62+
63+import Squeal.PostgreSQL.Alias
64+import Squeal.PostgreSQL.Definition
65+import Squeal.PostgreSQL.Definition.Table.Constraint
66+import Squeal.PostgreSQL.Expression
67+import Squeal.PostgreSQL.Expression.Type
68+import Squeal.PostgreSQL.List
69+import Squeal.PostgreSQL.Render
70+import Squeal.PostgreSQL.Schema
71+
72+-- $setup
73+-- >>> import Squeal.PostgreSQL
74+
75+{- | `createTable` adds a table to the schema.
76+
77+>>> :set -XOverloadedLabels
78+>>> :{
79+type Table = '[] :=>
80+ '[ "a" ::: 'NoDef :=> 'Null 'PGint4
81+ , "b" ::: 'NoDef :=> 'Null 'PGfloat4 ]
82+:}
83+
84+>>> :{
85+let
86+ setup :: Definition (Public '[]) (Public '["tab" ::: 'Table Table])
87+ setup = createTable #tab
88+ (nullable int `as` #a :* nullable real `as` #b) Nil
89+in printSQL setup
90+:}
91+CREATE TABLE "tab" ("a" int NULL, "b" real NULL);
92+-}
93+createTable
94+ :: ( KnownSymbol sch
95+ , KnownSymbol tab
96+ , columns ~ (col ': cols)
97+ , SOP.SListI columns
98+ , SOP.SListI constraints
99+ , Has sch schemas0 schema0
100+ , schemas1 ~ Alter sch (Create tab ('Table (constraints :=> columns)) schema0) schemas0 )
101+ => QualifiedAlias sch tab -- ^ the name of the table to add
102+ -> NP (Aliased (ColumnTypeExpression schemas0)) columns
103+ -- ^ the names and datatype of each column
104+ -> NP (Aliased (TableConstraintExpression sch tab schemas1)) constraints
105+ -- ^ constraints that must hold for the table
106+ -> Definition schemas0 schemas1
107+createTable tab columns constraints = UnsafeDefinition $
108+ "CREATE TABLE" <+> renderCreation tab columns constraints
109+
110+{-| `createTableIfNotExists` creates a table if it doesn't exist, but does not add it to the schema.
111+Instead, the schema already has the table so if the table did not yet exist, the schema was wrong.
112+`createTableIfNotExists` fixes this. Interestingly, this property makes it an idempotent in
113+the `Category` of `Definition`s.
114+
115+>>> :set -XOverloadedLabels -XTypeApplications
116+>>> :{
117+type Table = '[] :=>
118+ '[ "a" ::: 'NoDef :=> 'Null 'PGint4
119+ , "b" ::: 'NoDef :=> 'Null 'PGfloat4 ]
120+:}
121+
122+>>> type Schemas = Public '["tab" ::: 'Table Table]
123+
124+>>> :{
125+let
126+ setup :: Definition Schemas Schemas
127+ setup = createTableIfNotExists #tab
128+ (nullable int `as` #a :* nullable real `as` #b) Nil
129+in printSQL setup
130+:}
131+CREATE TABLE IF NOT EXISTS "tab" ("a" int NULL, "b" real NULL);
132+-}
133+createTableIfNotExists
134+ :: ( KnownSymbol sch
135+ , KnownSymbol tab
136+ , columns ~ (col ': cols)
137+ , SOP.SListI columns
138+ , SOP.SListI constraints
139+ , Has sch schemas0 schema0
140+ , schemas1 ~ Alter sch (CreateIfNotExists tab ('Table (constraints :=> columns)) schema0) schemas0 )
141+ => QualifiedAlias sch tab -- ^ the name of the table to add
142+ -> NP (Aliased (ColumnTypeExpression schemas0)) columns
143+ -- ^ the names and datatype of each column
144+ -> NP (Aliased (TableConstraintExpression sch tab schemas1)) constraints
145+ -- ^ constraints that must hold for the table
146+ -> Definition schemas0 schemas1
147+createTableIfNotExists tab columns constraints = UnsafeDefinition $
148+ "CREATE TABLE IF NOT EXISTS"
149+ <+> renderCreation tab columns constraints
150+
151+-- helper function for `createTable` and `createTableIfNotExists`
152+renderCreation
153+ :: ( KnownSymbol sch
154+ , KnownSymbol tab
155+ , SOP.SListI columns
156+ , SOP.SListI constraints )
157+ => QualifiedAlias sch tab -- ^ the name of the table to add
158+ -> NP (Aliased (ColumnTypeExpression schemas0)) columns
159+ -- ^ the names and datatype of each column
160+ -> NP (Aliased (TableConstraintExpression sch tab schemas1)) constraints
161+ -- ^ constraints that must hold for the table
162+ -> ByteString
163+renderCreation tab columns constraints = renderSQL tab
164+ <+> parenthesized
165+ ( renderCommaSeparated renderColumnDef columns
166+ <> ( case constraints of
167+ Nil -> ""
168+ _ -> ", " <>
169+ renderCommaSeparated renderConstraint constraints ) )
170+ <> ";"
171+ where
172+ renderColumnDef :: Aliased (ColumnTypeExpression schemas) x -> ByteString
173+ renderColumnDef (ty `As` column) =
174+ renderSQL column <+> renderColumnTypeExpression ty
175+ renderConstraint
176+ :: Aliased (TableConstraintExpression sch tab schemas) constraint
177+ -> ByteString
178+ renderConstraint (constraint `As` alias) =
179+ "CONSTRAINT" <+> renderSQL alias <+> renderSQL constraint
180+
181+-- | `dropTable` removes a table from the schema.
182+--
183+-- >>> :{
184+-- let
185+-- definition :: Definition '["public" ::: '["muh_table" ::: 'Table t]] (Public '[])
186+-- definition = dropTable #muh_table
187+-- :}
188+--
189+-- >>> printSQL definition
190+-- DROP TABLE "muh_table";
191+dropTable
192+ :: ( Has sch schemas schema
193+ , KnownSymbol tab )
194+ => QualifiedAlias sch tab -- ^ table to remove
195+ -> Definition schemas (Alter sch (DropSchemum tab 'Table schema) schemas)
196+dropTable tab = UnsafeDefinition $ "DROP TABLE" <+> renderSQL tab <> ";"
197+
198+dropTableIfExists
199+ :: ( Has sch schemas schema
200+ , KnownSymbol tab)
201+ => QualifiedAlias sch tab -- ^ table to remove
202+ -> Definition schemas (Alter sch (DropSchemumIfExists tab 'Table schema) schemas)
203+dropTableIfExists tab = UnsafeDefinition $ "DROP TABLE IF EXISTS" <+> renderSQL tab <> ";"
204+
205+-- | `alterTable` changes the definition of a table from the schema.
206+alterTable
207+ :: (Has sch schemas schema, KnownSymbol tab)
208+ => QualifiedAlias sch tab -- ^ table to alter
209+ -> AlterTable sch tab schemas table -- ^ alteration to perform
210+ -> Definition schemas (Alter sch (Alter tab ('Table table) schema) schemas)
211+alterTable tab alteration = UnsafeDefinition $
212+ "ALTER TABLE"
213+ <+> renderSQL tab
214+ <+> renderAlterTable alteration
215+ <> ";"
216+
217+-- | `alterTable` changes the definition of a table from the schema.
218+alterTableIfExists
219+ :: (Has sch schemas schema, KnownSymbol tab)
220+ => QualifiedAlias sch tab -- ^ table to alter
221+ -> AlterTable sch tab schemas table -- ^ alteration to perform
222+ -> Definition schemas (Alter sch (AlterIfExists tab ('Table table) schema) schemas)
223+alterTableIfExists tab alteration = UnsafeDefinition $
224+ "ALTER TABLE IF EXISTS"
225+ <+> renderSQL tab
226+ <+> renderAlterTable alteration
227+ <> ";"
228+
229+-- | `alterTableRename` changes the name of a table from the schema.
230+--
231+-- >>> printSQL $ alterTableRename #foo #bar
232+-- ALTER TABLE "foo" RENAME TO "bar";
233+alterTableRename
234+ :: (KnownSymbol tab0, KnownSymbol tab1)
235+ => Alias tab0 -- ^ table to rename
236+ -> Alias tab1 -- ^ what to rename it
237+ -> Definition schema (Rename tab0 tab1 schema)
238+alterTableRename tab0 tab1 = UnsafeDefinition $
239+ "ALTER TABLE" <+> renderSQL tab0
240+ <+> "RENAME TO" <+> renderSQL tab1 <> ";"
241+
242+alterTableIfExistsRename
243+ :: (KnownSymbol tab0, KnownSymbol tab1)
244+ => Alias tab0 -- ^ table to rename
245+ -> Alias tab1 -- ^ what to rename it
246+ -> Definition schema (RenameIfExists tab0 tab1 schema)
247+alterTableIfExistsRename tab0 tab1 = UnsafeDefinition $
248+ "ALTER TABLE IF EXISTS" <+> renderSQL tab0
249+ <+> "RENAME TO" <+> renderSQL tab1 <> ";"
250+
251+-- | An `AlterTable` describes the alteration to perform on the columns
252+-- of a table.
253+newtype AlterTable
254+ (sch :: Symbol)
255+ (tab :: Symbol)
256+ (schemas :: SchemasType)
257+ (table :: TableType) =
258+ UnsafeAlterTable {renderAlterTable :: ByteString}
259+ deriving (GHC.Generic,Show,Eq,Ord,NFData)
260+
261+-- | An `addConstraint` adds a table constraint.
262+--
263+-- >>> :{
264+-- let
265+-- definition :: Definition
266+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
267+-- '["public" ::: '["tab" ::: 'Table ('["positive" ::: 'Check '["col"]] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
268+-- definition = alterTable #tab (addConstraint #positive (check #col (#col .> 0)))
269+-- in printSQL definition
270+-- :}
271+-- ALTER TABLE "tab" ADD CONSTRAINT "positive" CHECK (("col" > 0));
272+addConstraint
273+ :: ( KnownSymbol alias
274+ , Has sch schemas schema
275+ , Has tab schema ('Table table0)
276+ , table0 ~ (constraints :=> columns)
277+ , table1 ~ (Create alias constraint constraints :=> columns) )
278+ => Alias alias
279+ -> TableConstraintExpression sch tab schemas constraint
280+ -- ^ constraint to add
281+ -> AlterTable sch tab schemas table1
282+addConstraint alias constraint = UnsafeAlterTable $
283+ "ADD" <+> "CONSTRAINT" <+> renderSQL alias
284+ <+> renderSQL constraint
285+
286+-- | A `dropConstraint` drops a table constraint.
287+--
288+-- >>> :{
289+-- let
290+-- definition :: Definition
291+-- '["public" ::: '["tab" ::: 'Table ('["positive" ::: Check '["col"]] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
292+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
293+-- definition = alterTable #tab (dropConstraint #positive)
294+-- in printSQL definition
295+-- :}
296+-- ALTER TABLE "tab" DROP CONSTRAINT "positive";
297+dropConstraint
298+ :: ( KnownSymbol constraint
299+ , Has sch schemas schema
300+ , Has tab schema ('Table table0)
301+ , table0 ~ (constraints :=> columns)
302+ , table1 ~ (Drop constraint constraints :=> columns) )
303+ => Alias constraint
304+ -- ^ constraint to drop
305+ -> AlterTable sch tab schemas table1
306+dropConstraint constraint = UnsafeAlterTable $
307+ "DROP" <+> "CONSTRAINT" <+> renderSQL constraint
308+
309+-- | An `AddColumn` is either @NULL@ or has @DEFAULT@.
310+class AddColumn ty where
311+ -- | `addColumn` adds a new column, initially filled with whatever
312+ -- default value is given or with @NULL@.
313+ --
314+ -- >>> :{
315+ -- let
316+ -- definition :: Definition
317+ -- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col1" ::: 'NoDef :=> 'Null 'PGint4])]]
318+ -- '["public" ::: '["tab" ::: 'Table ('[] :=>
319+ -- '[ "col1" ::: 'NoDef :=> 'Null 'PGint4
320+ -- , "col2" ::: 'Def :=> 'Null 'PGtext ])]]
321+ -- definition = alterTable #tab (addColumn #col2 (text & nullable & default_ "foo"))
322+ -- in printSQL definition
323+ -- :}
324+ -- ALTER TABLE "tab" ADD COLUMN "col2" text NULL DEFAULT E'foo';
325+ --
326+ -- >>> :{
327+ -- let
328+ -- definition :: Definition
329+ -- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col1" ::: 'NoDef :=> 'Null 'PGint4])]]
330+ -- '["public" ::: '["tab" ::: 'Table ('[] :=>
331+ -- '[ "col1" ::: 'NoDef :=> 'Null 'PGint4
332+ -- , "col2" ::: 'NoDef :=> 'Null 'PGtext ])]]
333+ -- definition = alterTable #tab (addColumn #col2 (text & nullable))
334+ -- in printSQL definition
335+ -- :}
336+ -- ALTER TABLE "tab" ADD COLUMN "col2" text NULL;
337+ addColumn
338+ :: ( KnownSymbol column
339+ , Has sch schemas schema
340+ , Has tab schema ('Table table0)
341+ , table0 ~ (constraints :=> columns) )
342+ => Alias column -- ^ column to add
343+ -> ColumnTypeExpression schemas ty -- ^ type of the new column
344+ -> AlterTable sch tab schemas (constraints :=> Create column ty columns)
345+ addColumn column ty = UnsafeAlterTable $
346+ "ADD COLUMN" <+> renderSQL column <+> renderColumnTypeExpression ty
347+instance {-# OVERLAPPING #-} AddColumn ('Def :=> ty)
348+instance {-# OVERLAPPABLE #-} AddColumn ('NoDef :=> 'Null ty)
349+
350+-- | A `dropColumn` removes a column. Whatever data was in the column
351+-- disappears. Table constraints involving the column are dropped, too.
352+-- However, if the column is referenced by a foreign key constraint of
353+-- another table, PostgreSQL will not silently drop that constraint.
354+--
355+-- >>> :{
356+-- let
357+-- definition :: Definition
358+-- '["public" ::: '["tab" ::: 'Table ('[] :=>
359+-- '[ "col1" ::: 'NoDef :=> 'Null 'PGint4
360+-- , "col2" ::: 'NoDef :=> 'Null 'PGtext ])]]
361+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col1" ::: 'NoDef :=> 'Null 'PGint4])]]
362+-- definition = alterTable #tab (dropColumn #col2)
363+-- in printSQL definition
364+-- :}
365+-- ALTER TABLE "tab" DROP COLUMN "col2";
366+dropColumn
367+ :: ( KnownSymbol column
368+ , Has sch schemas schema
369+ , Has tab schema ('Table table0)
370+ , table0 ~ (constraints :=> columns)
371+ , table1 ~ (constraints :=> Drop column columns) )
372+ => Alias column -- ^ column to remove
373+ -> AlterTable sch tab schemas table1
374+dropColumn column = UnsafeAlterTable $
375+ "DROP COLUMN" <+> renderSQL column
376+
377+-- | A `renameColumn` renames a column.
378+--
379+-- >>> :{
380+-- let
381+-- definition :: Definition
382+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["foo" ::: 'NoDef :=> 'Null 'PGint4])]]
383+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["bar" ::: 'NoDef :=> 'Null 'PGint4])]]
384+-- definition = alterTable #tab (renameColumn #foo #bar)
385+-- in printSQL definition
386+-- :}
387+-- ALTER TABLE "tab" RENAME COLUMN "foo" TO "bar";
388+renameColumn
389+ :: ( KnownSymbol column0
390+ , KnownSymbol column1
391+ , Has sch schemas schema
392+ , Has tab schema ('Table table0)
393+ , table0 ~ (constraints :=> columns)
394+ , table1 ~ (constraints :=> Rename column0 column1 columns) )
395+ => Alias column0 -- ^ column to rename
396+ -> Alias column1 -- ^ what to rename the column
397+ -> AlterTable sch tab schemas table1
398+renameColumn column0 column1 = UnsafeAlterTable $
399+ "RENAME COLUMN" <+> renderSQL column0 <+> "TO" <+> renderSQL column1
400+
401+-- | An `alterColumn` alters a single column.
402+alterColumn
403+ :: ( KnownSymbol column
404+ , Has sch schemas schema
405+ , Has tab schema ('Table table0)
406+ , table0 ~ (constraints :=> columns)
407+ , Has column columns ty0
408+ , table1 ~ (constraints :=> Alter column ty1 columns))
409+ => Alias column -- ^ column to alter
410+ -> AlterColumn schemas ty0 ty1 -- ^ alteration to perform
411+ -> AlterTable sch tab schemas table1
412+alterColumn column alteration = UnsafeAlterTable $
413+ "ALTER COLUMN" <+> renderSQL column <+> renderAlterColumn alteration
414+
415+-- | An `AlterColumn` describes the alteration to perform on a single column.
416+newtype AlterColumn (schemas :: SchemasType) (ty0 :: ColumnType) (ty1 :: ColumnType) =
417+ UnsafeAlterColumn {renderAlterColumn :: ByteString}
418+ deriving (GHC.Generic,Show,Eq,Ord,NFData)
419+
420+-- | A `setDefault` sets a new default for a column. Note that this doesn't
421+-- affect any existing rows in the table, it just changes the default for
422+-- future insert and update commands.
423+--
424+-- >>> :{
425+-- let
426+-- definition :: Definition
427+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
428+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'Def :=> 'Null 'PGint4])]]
429+-- definition = alterTable #tab (alterColumn #col (setDefault 5))
430+-- in printSQL definition
431+-- :}
432+-- ALTER TABLE "tab" ALTER COLUMN "col" SET DEFAULT 5;
433+setDefault
434+ :: Expression '[] '[] 'Ungrouped schemas '[] '[] ty -- ^ default value to set
435+ -> AlterColumn schemas (constraint :=> ty) ('Def :=> ty)
436+setDefault expression = UnsafeAlterColumn $
437+ "SET DEFAULT" <+> renderExpression expression
438+
439+-- | A `dropDefault` removes any default value for a column.
440+--
441+-- >>> :{
442+-- let
443+-- definition :: Definition
444+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'Def :=> 'Null 'PGint4])]]
445+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
446+-- definition = alterTable #tab (alterColumn #col dropDefault)
447+-- in printSQL definition
448+-- :}
449+-- ALTER TABLE "tab" ALTER COLUMN "col" DROP DEFAULT;
450+dropDefault :: AlterColumn schemas ('Def :=> ty) ('NoDef :=> ty)
451+dropDefault = UnsafeAlterColumn $ "DROP DEFAULT"
452+
453+-- | A `setNotNull` adds a @NOT NULL@ constraint to a column.
454+-- The constraint will be checked immediately, so the table data must satisfy
455+-- the constraint before it can be added.
456+--
457+-- >>> :{
458+-- let
459+-- definition :: Definition
460+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
461+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
462+-- definition = alterTable #tab (alterColumn #col setNotNull)
463+-- in printSQL definition
464+-- :}
465+-- ALTER TABLE "tab" ALTER COLUMN "col" SET NOT NULL;
466+setNotNull
467+ :: AlterColumn schemas (constraint :=> 'Null ty) (constraint :=> 'NotNull ty)
468+setNotNull = UnsafeAlterColumn $ "SET NOT NULL"
469+
470+-- | A `dropNotNull` drops a @NOT NULL@ constraint from a column.
471+--
472+-- >>> :{
473+-- let
474+-- definition :: Definition
475+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
476+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'Null 'PGint4])]]
477+-- definition = alterTable #tab (alterColumn #col dropNotNull)
478+-- in printSQL definition
479+-- :}
480+-- ALTER TABLE "tab" ALTER COLUMN "col" DROP NOT NULL;
481+dropNotNull
482+ :: AlterColumn schemas (constraint :=> 'NotNull ty) (constraint :=> 'Null ty)
483+dropNotNull = UnsafeAlterColumn $ "DROP NOT NULL"
484+
485+-- | An `alterType` converts a column to a different data type.
486+-- This will succeed only if each existing entry in the column can be
487+-- converted to the new type by an implicit cast.
488+--
489+-- >>> :{
490+-- let
491+-- definition :: Definition
492+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGint4])]]
493+-- '["public" ::: '["tab" ::: 'Table ('[] :=> '["col" ::: 'NoDef :=> 'NotNull 'PGnumeric])]]
494+-- definition =
495+-- alterTable #tab (alterColumn #col (alterType (numeric & notNullable)))
496+-- in printSQL definition
497+-- :}
498+-- ALTER TABLE "tab" ALTER COLUMN "col" TYPE numeric NOT NULL;
499+alterType :: ColumnTypeExpression schemas ty -> AlterColumn schemas ty0 ty
500+alterType ty = UnsafeAlterColumn $ "TYPE" <+> renderColumnTypeExpression ty
squeal-postgresql/src/Squeal/PostgreSQL/Definition/Table/Constraint.hsnew file+339
@@ -0,0 +1,339 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.Table.Constraint
3+Description: Table constraint expressions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Table constraint expressions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.Table.Constraint
33+ ( TableConstraintExpression (..)
34+ , check
35+ , unique
36+ , primaryKey
37+ , foreignKey
38+ , ForeignKeyed
39+ , OnDeleteClause (..)
40+ , OnUpdateClause (..)
41+ ) where
42+
43+import Control.DeepSeq
44+import Data.ByteString
45+import GHC.TypeLits
46+
47+import qualified Generics.SOP as SOP
48+import qualified GHC.Generics as GHC
49+
50+import Squeal.PostgreSQL.Alias
51+import Squeal.PostgreSQL.Expression.Logic
52+import Squeal.PostgreSQL.List
53+import Squeal.PostgreSQL.Render
54+import Squeal.PostgreSQL.Schema
55+
56+-- $setup
57+-- >>> import Squeal.PostgreSQL
58+
59+-- | Data types are a way to limit the kind of data that can be stored in a
60+-- table. For many applications, however, the constraint they provide is
61+-- too coarse. For example, a column containing a product price should
62+-- probably only accept positive values. But there is no standard data type
63+-- that accepts only positive numbers. Another issue is that you might want
64+-- to constrain column data with respect to other columns or rows.
65+-- For example, in a table containing product information,
66+-- there should be only one row for each product number.
67+-- `TableConstraint`s give you as much control over the data in your tables
68+-- as you wish. If a user attempts to store data in a column that would
69+-- violate a constraint, an error is raised. This applies
70+-- even if the value came from the default value definition.
71+newtype TableConstraintExpression
72+ (sch :: Symbol)
73+ (tab :: Symbol)
74+ (schemas :: SchemasType)
75+ (constraint :: TableConstraint)
76+ = UnsafeTableConstraintExpression
77+ { renderTableConstraintExpression :: ByteString }
78+ deriving (GHC.Generic,Show,Eq,Ord,NFData)
79+instance RenderSQL
80+ (TableConstraintExpression sch tab schemas constraint) where
81+ renderSQL = renderTableConstraintExpression
82+
83+{-| A `check` constraint is the most generic `TableConstraint` type.
84+It allows you to specify that the value in a certain column must satisfy
85+a Boolean (truth-value) expression.
86+
87+>>> :{
88+type Schema = '[
89+ "tab" ::: 'Table ('[ "inequality" ::: 'Check '["a","b"]] :=> '[
90+ "a" ::: 'NoDef :=> 'NotNull 'PGint4,
91+ "b" ::: 'NoDef :=> 'NotNull 'PGint4
92+ ])]
93+:}
94+
95+>>> :{
96+let
97+ definition :: Definition (Public '[]) (Public Schema)
98+ definition = createTable #tab
99+ ( (int & notNullable) `as` #a :*
100+ (int & notNullable) `as` #b )
101+ ( check (#a :* #b) (#a .> #b) `as` #inequality )
102+:}
103+
104+>>> printSQL definition
105+CREATE TABLE "tab" ("a" int NOT NULL, "b" int NOT NULL, CONSTRAINT "inequality" CHECK (("a" > "b")));
106+-}
107+check
108+ :: ( Has sch schemas schema
109+ , Has tab schema ('Table table)
110+ , HasAll aliases (TableToRow table) subcolumns )
111+ => NP Alias aliases
112+ -- ^ specify the subcolumns which are getting checked
113+ -> (forall t. Condition '[] '[] 'Ungrouped schemas '[] '[t ::: subcolumns])
114+ -- ^ a closed `Condition` on those subcolumns
115+ -> TableConstraintExpression sch tab schemas ('Check aliases)
116+check _cols condition = UnsafeTableConstraintExpression $
117+ "CHECK" <+> parenthesized (renderSQL condition)
118+
119+{-| A `unique` constraint ensure that the data contained in a column,
120+or a group of columns, is unique among all the rows in the table.
121+
122+>>> :{
123+type Schema = '[
124+ "tab" ::: 'Table( '[ "uq_a_b" ::: 'Unique '["a","b"]] :=> '[
125+ "a" ::: 'NoDef :=> 'Null 'PGint4,
126+ "b" ::: 'NoDef :=> 'Null 'PGint4
127+ ])]
128+:}
129+
130+>>> :{
131+let
132+ definition :: Definition (Public '[]) (Public Schema)
133+ definition = createTable #tab
134+ ( (int & nullable) `as` #a :*
135+ (int & nullable) `as` #b )
136+ ( unique (#a :* #b) `as` #uq_a_b )
137+:}
138+
139+>>> printSQL definition
140+CREATE TABLE "tab" ("a" int NULL, "b" int NULL, CONSTRAINT "uq_a_b" UNIQUE ("a", "b"));
141+-}
142+unique
143+ :: ( Has sch schemas schema
144+ , Has tab schema('Table table)
145+ , HasAll aliases (TableToRow table) subcolumns )
146+ => NP Alias aliases
147+ -- ^ specify subcolumns which together are unique for each row
148+ -> TableConstraintExpression sch tab schemas ('Unique aliases)
149+unique columns = UnsafeTableConstraintExpression $
150+ "UNIQUE" <+> parenthesized (renderSQL columns)
151+
152+{-| A `primaryKey` constraint indicates that a column, or group of columns,
153+can be used as a unique identifier for rows in the table.
154+This requires that the values be both unique and not null.
155+
156+>>> :{
157+type Schema = '[
158+ "tab" ::: 'Table ('[ "pk_id" ::: 'PrimaryKey '["id"]] :=> '[
159+ "id" ::: 'Def :=> 'NotNull 'PGint4,
160+ "name" ::: 'NoDef :=> 'NotNull 'PGtext
161+ ])]
162+:}
163+
164+>>> :{
165+let
166+ definition :: Definition (Public '[]) (Public Schema)
167+ definition = createTable #tab
168+ ( serial `as` #id :*
169+ (text & notNullable) `as` #name )
170+ ( primaryKey #id `as` #pk_id )
171+:}
172+
173+>>> printSQL definition
174+CREATE TABLE "tab" ("id" serial, "name" text NOT NULL, CONSTRAINT "pk_id" PRIMARY KEY ("id"));
175+-}
176+primaryKey
177+ :: ( Has sch schemas schema
178+ , Has tab schema ('Table table)
179+ , HasAll aliases (TableToColumns table) subcolumns
180+ , AllNotNull subcolumns )
181+ => NP Alias aliases
182+ -- ^ specify the subcolumns which together form a primary key.
183+ -> TableConstraintExpression sch tab schemas ('PrimaryKey aliases)
184+primaryKey columns = UnsafeTableConstraintExpression $
185+ "PRIMARY KEY" <+> parenthesized (renderSQL columns)
186+
187+{-| A `foreignKey` specifies that the values in a column
188+(or a group of columns) must match the values appearing in some row of
189+another table. We say this maintains the referential integrity
190+between two related tables.
191+
192+>>> :{
193+type Schema =
194+ '[ "users" ::: 'Table (
195+ '[ "pk_users" ::: 'PrimaryKey '["id"] ] :=>
196+ '[ "id" ::: 'Def :=> 'NotNull 'PGint4
197+ , "name" ::: 'NoDef :=> 'NotNull 'PGtext
198+ ])
199+ , "emails" ::: 'Table (
200+ '[ "pk_emails" ::: 'PrimaryKey '["id"]
201+ , "fk_user_id" ::: 'ForeignKey '["user_id"] "users" '["id"]
202+ ] :=>
203+ '[ "id" ::: 'Def :=> 'NotNull 'PGint4
204+ , "user_id" ::: 'NoDef :=> 'NotNull 'PGint4
205+ , "email" ::: 'NoDef :=> 'Null 'PGtext
206+ ])
207+ ]
208+:}
209+
210+>>> :{
211+let
212+ setup :: Definition (Public '[]) (Public Schema)
213+ setup =
214+ createTable #users
215+ ( serial `as` #id :*
216+ (text & notNullable) `as` #name )
217+ ( primaryKey #id `as` #pk_users ) >>>
218+ createTable #emails
219+ ( serial `as` #id :*
220+ (int & notNullable) `as` #user_id :*
221+ (text & nullable) `as` #email )
222+ ( primaryKey #id `as` #pk_emails :*
223+ foreignKey #user_id #users #id
224+ OnDeleteCascade OnUpdateCascade `as` #fk_user_id )
225+in printSQL setup
226+:}
227+CREATE TABLE "users" ("id" serial, "name" text NOT NULL, CONSTRAINT "pk_users" PRIMARY KEY ("id"));
228+CREATE TABLE "emails" ("id" serial, "user_id" int NOT NULL, "email" text NULL, CONSTRAINT "pk_emails" PRIMARY KEY ("id"), CONSTRAINT "fk_user_id" FOREIGN KEY ("user_id") REFERENCES "users" ("id") ON DELETE CASCADE ON UPDATE CASCADE);
229+
230+A `foreignKey` can even be a table self-reference.
231+
232+>>> :{
233+type Schema =
234+ '[ "employees" ::: 'Table (
235+ '[ "employees_pk" ::: 'PrimaryKey '["id"]
236+ , "employees_employer_fk" ::: 'ForeignKey '["employer_id"] "employees" '["id"]
237+ ] :=>
238+ '[ "id" ::: 'Def :=> 'NotNull 'PGint4
239+ , "name" ::: 'NoDef :=> 'NotNull 'PGtext
240+ , "employer_id" ::: 'NoDef :=> 'Null 'PGint4
241+ ])
242+ ]
243+:}
244+
245+>>> :{
246+let
247+ setup :: Definition (Public '[]) (Public Schema)
248+ setup =
249+ createTable #employees
250+ ( serial `as` #id :*
251+ (text & notNullable) `as` #name :*
252+ (integer & nullable) `as` #employer_id )
253+ ( primaryKey #id `as` #employees_pk :*
254+ foreignKey #employer_id #employees #id
255+ OnDeleteCascade OnUpdateCascade `as` #employees_employer_fk )
256+in printSQL setup
257+:}
258+CREATE TABLE "employees" ("id" serial, "name" text NOT NULL, "employer_id" integer NULL, CONSTRAINT "employees_pk" PRIMARY KEY ("id"), CONSTRAINT "employees_employer_fk" FOREIGN KEY ("employer_id") REFERENCES "employees" ("id") ON DELETE CASCADE ON UPDATE CASCADE);
259+-}
260+foreignKey
261+ :: (ForeignKeyed schemas sch schema child parent
262+ table reftable
263+ columns refcolumns
264+ constraints cols
265+ reftys tys )
266+ => NP Alias columns
267+ -- ^ column or columns in the table
268+ -> Alias parent
269+ -- ^ reference table
270+ -> NP Alias refcolumns
271+ -- ^ reference column or columns in the reference table
272+ -> OnDeleteClause
273+ -- ^ what to do when reference is deleted
274+ -> OnUpdateClause
275+ -- ^ what to do when reference is updated
276+ -> TableConstraintExpression sch child schemas
277+ ('ForeignKey columns parent refcolumns)
278+foreignKey keys parent refs ondel onupd = UnsafeTableConstraintExpression $
279+ "FOREIGN KEY" <+> parenthesized (renderSQL keys)
280+ <+> "REFERENCES" <+> renderSQL parent
281+ <+> parenthesized (renderSQL refs)
282+ <+> renderSQL ondel
283+ <+> renderSQL onupd
284+
285+-- | A constraint synonym between types involved in a foreign key constraint.
286+type ForeignKeyed schemas
287+ sch
288+ schema
289+ child parent
290+ table reftable
291+ columns refcolumns
292+ constraints cols
293+ reftys tys =
294+ ( Has sch schemas schema
295+ , Has child schema ('Table table)
296+ , Has parent schema ('Table reftable)
297+ , HasAll columns (TableToColumns table) tys
298+ , reftable ~ (constraints :=> cols)
299+ , HasAll refcolumns cols reftys
300+ , SOP.AllZip SamePGType tys reftys
301+ , Uniquely refcolumns constraints )
302+
303+-- | `OnDeleteClause` indicates what to do with rows that reference a deleted row.
304+data OnDeleteClause
305+ = OnDeleteNoAction
306+ -- ^ if any referencing rows still exist when the constraint is checked,
307+ -- an error is raised
308+ | OnDeleteRestrict -- ^ prevents deletion of a referenced row
309+ | OnDeleteCascade
310+ -- ^ specifies that when a referenced row is deleted,
311+ -- row(s) referencing it should be automatically deleted as well
312+ deriving (GHC.Generic,Show,Eq,Ord)
313+instance NFData OnDeleteClause
314+-- | Render `OnDeleteClause`.
315+instance RenderSQL OnDeleteClause where
316+ renderSQL = \case
317+ OnDeleteNoAction -> "ON DELETE NO ACTION"
318+ OnDeleteRestrict -> "ON DELETE RESTRICT"
319+ OnDeleteCascade -> "ON DELETE CASCADE"
320+
321+-- | Analagous to `OnDeleteClause` there is also `OnUpdateClause` which is invoked
322+-- when a referenced column is changed (updated).
323+data OnUpdateClause
324+ = OnUpdateNoAction
325+ -- ^ if any referencing rows has not changed when the constraint is checked,
326+ -- an error is raised
327+ | OnUpdateRestrict -- ^ prevents update of a referenced row
328+ | OnUpdateCascade
329+ -- ^ the updated values of the referenced column(s) should be copied
330+ -- into the referencing row(s)
331+ deriving (GHC.Generic,Show,Eq,Ord)
332+instance NFData OnUpdateClause
333+
334+-- | Render `OnUpdateClause`.
335+instance RenderSQL OnUpdateClause where
336+ renderSQL = \case
337+ OnUpdateNoAction -> "ON UPDATE NO ACTION"
338+ OnUpdateRestrict -> "ON UPDATE RESTRICT"
339+ OnUpdateCascade -> "ON UPDATE CASCADE"
squeal-postgresql/src/Squeal/PostgreSQL/Definition/Type.hsnew file+229
@@ -0,0 +1,229 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.Type
3+Description: Create and drop type definitions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Create and drop type definitions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.Type
33+ ( createTypeEnum
34+ , createTypeEnumFrom
35+ , createTypeComposite
36+ , createTypeCompositeFrom
37+ , createTypeRange
38+ , createDomain
39+ , dropType
40+ , dropTypeIfExists
41+ ) where
42+
43+import Data.ByteString
44+import Data.Monoid
45+import GHC.TypeLits
46+import Prelude hiding ((.), id)
47+
48+import qualified Generics.SOP as SOP
49+
50+import Squeal.PostgreSQL.Alias
51+import Squeal.PostgreSQL.Expression.Logic
52+import Squeal.PostgreSQL.Expression.Type
53+import Squeal.PostgreSQL.Definition
54+import Squeal.PostgreSQL.List
55+import Squeal.PostgreSQL.PG
56+import Squeal.PostgreSQL.Render
57+import Squeal.PostgreSQL.Schema
58+
59+-- $setup
60+-- >>> import Squeal.PostgreSQL
61+-- >>> import qualified GHC.Generics as GHC
62+-- >>> import qualified Generics.SOP as SOP
63+
64+-- | Enumerated types are created using the `createTypeEnum` command, for example
65+--
66+-- >>> printSQL $ (createTypeEnum #mood (label @"sad" :* label @"ok" :* label @"happy") :: Definition (Public '[]) '["public" ::: '["mood" ::: 'Typedef ('PGenum '["sad","ok","happy"])]])
67+-- CREATE TYPE "mood" AS ENUM ('sad', 'ok', 'happy');
68+createTypeEnum
69+ :: (KnownSymbol enum, Has sch schemas schema, SOP.All KnownSymbol labels)
70+ => QualifiedAlias sch enum
71+ -- ^ name of the user defined enumerated type
72+ -> NP PGlabel labels
73+ -- ^ labels of the enumerated type
74+ -> Definition schemas (Alter sch (Create enum ('Typedef ('PGenum labels)) schema) schemas)
75+createTypeEnum enum labels = UnsafeDefinition $
76+ "CREATE" <+> "TYPE" <+> renderSQL enum <+> "AS" <+> "ENUM" <+>
77+ parenthesized (renderSQL labels) <> ";"
78+
79+-- | Enumerated types can also be generated from a Haskell type, for example
80+--
81+-- >>> data Schwarma = Beef | Lamb | Chicken deriving GHC.Generic
82+-- >>> instance SOP.Generic Schwarma
83+-- >>> instance SOP.HasDatatypeInfo Schwarma
84+-- >>> :{
85+-- let
86+-- createSchwarma :: Definition (Public '[]) '["public" ::: '["schwarma" ::: 'Typedef (PG (Enumerated Schwarma))]]
87+-- createSchwarma = createTypeEnumFrom @Schwarma #schwarma
88+-- in
89+-- printSQL createSchwarma
90+-- :}
91+-- CREATE TYPE "schwarma" AS ENUM ('Beef', 'Lamb', 'Chicken');
92+createTypeEnumFrom
93+ :: forall hask sch enum schemas schema.
94+ ( SOP.Generic hask
95+ , SOP.All KnownSymbol (LabelsPG hask)
96+ , KnownSymbol enum
97+ , Has sch schemas schema )
98+ => QualifiedAlias sch enum
99+ -- ^ name of the user defined enumerated type
100+ -> Definition schemas (Alter sch (Create enum ('Typedef (PG (Enumerated hask))) schema) schemas)
101+createTypeEnumFrom enum = createTypeEnum enum
102+ (SOP.hpure label :: NP PGlabel (LabelsPG hask))
103+
104+{- | `createTypeComposite` creates a composite type. The composite type is
105+specified by a list of attribute names and data types.
106+
107+>>> :{
108+type PGcomplex = 'PGcomposite
109+ '[ "real" ::: 'NotNull 'PGfloat8
110+ , "imaginary" ::: 'NotNull 'PGfloat8 ]
111+:}
112+
113+>>> :{
114+let
115+ setup :: Definition (Public '[]) '["public" ::: '["complex" ::: 'Typedef PGcomplex]]
116+ setup = createTypeComposite #complex
117+ (float8 `as` #real :* float8 `as` #imaginary)
118+in printSQL setup
119+:}
120+CREATE TYPE "complex" AS ("real" float8, "imaginary" float8);
121+-}
122+createTypeComposite
123+ :: (KnownSymbol ty, Has sch schemas schema, SOP.SListI fields)
124+ => QualifiedAlias sch ty
125+ -- ^ name of the user defined composite type
126+ -> NP (Aliased (TypeExpression schemas)) fields
127+ -- ^ list of attribute names and data types
128+ -> Definition schemas (Alter sch (Create ty ('Typedef ('PGcomposite fields)) schema) schemas)
129+createTypeComposite ty fields = UnsafeDefinition $
130+ "CREATE" <+> "TYPE" <+> renderSQL ty <+> "AS" <+> parenthesized
131+ (renderCommaSeparated renderField fields) <> ";"
132+ where
133+ renderField :: Aliased (TypeExpression schemas) x -> ByteString
134+ renderField (typ `As` alias) =
135+ renderSQL alias <+> renderSQL typ
136+
137+-- | Composite types can also be generated from a Haskell type, for example
138+--
139+-- >>> data Complex = Complex {real :: Double, imaginary :: Double} deriving GHC.Generic
140+-- >>> instance SOP.Generic Complex
141+-- >>> instance SOP.HasDatatypeInfo Complex
142+-- >>> type Schema = '["complex" ::: 'Typedef (PG (Composite Complex))]
143+-- >>> :{
144+-- let
145+-- createComplex :: Definition (Public '[]) (Public Schema)
146+-- createComplex = createTypeCompositeFrom @Complex #complex
147+-- in
148+-- printSQL createComplex
149+-- :}
150+-- CREATE TYPE "complex" AS ("real" float8, "imaginary" float8);
151+createTypeCompositeFrom
152+ :: forall hask sch ty schemas schema.
153+ ( SOP.All (FieldTyped schemas) (RowPG hask)
154+ , KnownSymbol ty
155+ , Has sch schemas schema )
156+ => QualifiedAlias sch ty
157+ -- ^ name of the user defined composite type
158+ -> Definition schemas (Alter sch (Create ty ( 'Typedef (PG (Composite hask))) schema) schemas)
159+createTypeCompositeFrom ty = createTypeComposite ty
160+ (SOP.hcpure (SOP.Proxy :: SOP.Proxy (FieldTyped schemas)) fieldtype
161+ :: NP (Aliased (TypeExpression schemas)) (RowPG hask))
162+
163+{-|
164+`createDomain` creates a new domain. A domain is essentially a data type
165+with constraints (restrictions on the allowed set of values).
166+
167+Domains are useful for abstracting common constraints on fields
168+into a single location for maintenance. For example, several tables might
169+contain email address columns, all requiring the same `check` constraint
170+to verify the address syntax. Define a domain rather than setting up
171+each table's constraint individually.
172+
173+>>> :{
174+let
175+ createPositive :: Definition (Public '[]) (Public '["positive" ::: 'Typedef 'PGfloat4])
176+ createPositive = createDomain #positive real (#value .> 0 .&& (#value & isNotNull))
177+in printSQL createPositive
178+:}
179+CREATE DOMAIN "positive" AS real CHECK ((("value" > 0) AND "value" IS NOT NULL));
180+-}
181+createDomain
182+ :: (Has sch schemas schema, KnownSymbol dom)
183+ => QualifiedAlias sch dom
184+ -> (forall null. TypeExpression schemas (null ty))
185+ -> (forall tab. Condition '[] '[] 'Ungrouped schemas '[] '[tab ::: '["value" ::: 'Null ty]])
186+ -> Definition schemas (Alter sch (Create dom ('Typedef ty) schema) schemas)
187+createDomain dom ty condition =
188+ UnsafeDefinition $ "CREATE DOMAIN" <+> renderSQL dom
189+ <+> "AS" <+> renderTypeExpression ty
190+ <+> "CHECK" <+> parenthesized (renderSQL condition) <> ";"
191+
192+{- |
193+>>> :{
194+let
195+ createSmallIntRange :: Definition (Public '[]) (Public '["int2range" ::: 'Typedef ('PGrange 'PGint2)])
196+ createSmallIntRange = createTypeRange #int2range int2
197+in printSQL createSmallIntRange
198+:}
199+CREATE TYPE "int2range" AS RANGE (subtype = int2);
200+-}
201+createTypeRange
202+ :: (Has sch schemas schema, KnownSymbol range)
203+ => QualifiedAlias sch range
204+ -> (forall null. TypeExpression schemas (null ty))
205+ -> Definition schemas (Alter sch (Create range ('Typedef ('PGrange ty)) schema) schemas)
206+createTypeRange range ty = UnsafeDefinition $
207+ "CREATE" <+> "TYPE" <+> renderSQL range <+> "AS" <+> "RANGE" <+>
208+ parenthesized ("subtype" <+> "=" <+> renderTypeExpression ty) <> ";"
209+
210+-- | Drop a type.
211+--
212+-- >>> data Schwarma = Beef | Lamb | Chicken deriving GHC.Generic
213+-- >>> instance SOP.Generic Schwarma
214+-- >>> instance SOP.HasDatatypeInfo Schwarma
215+-- >>> printSQL (dropType #schwarma :: Definition '["public" ::: '["schwarma" ::: 'Typedef (PG (Enumerated Schwarma))]] (Public '[]))
216+-- DROP TYPE "schwarma";
217+dropType
218+ :: (Has sch schemas schema, KnownSymbol td)
219+ => QualifiedAlias sch td
220+ -- ^ name of the user defined type
221+ -> Definition schemas (Alter sch (DropSchemum td 'Typedef schema) schemas)
222+dropType tydef = UnsafeDefinition $ "DROP" <+> "TYPE" <+> renderSQL tydef <> ";"
223+
224+dropTypeIfExists
225+ :: (Has sch schemas schema, KnownSymbol td)
226+ => QualifiedAlias sch td
227+ -- ^ name of the user defined type
228+ -> Definition schemas (Alter sch (DropSchemumIfExists td 'Typedef schema) schemas)
229+dropTypeIfExists tydef = UnsafeDefinition $ "DROP" <+> "TYPE" <+> renderSQL tydef <> ";"
squeal-postgresql/src/Squeal/PostgreSQL/Definition/View.hsnew file+104
@@ -0,0 +1,104 @@
1+{-|
2+Module: Squeal.PostgreSQL.Definition.View
3+Description: Create and drop view definitions
4+Copyright: (c) Eitan Chatav, 2017
5+Maintainer: eitan@morphism.tech
6+Stability: experimental
7+
8+Create and drop view definitions.
9+-}
10+
11+{-# LANGUAGE
12+ AllowAmbiguousTypes
13+ , ConstraintKinds
14+ , DeriveAnyClass
15+ , DeriveGeneric
16+ , DerivingStrategies
17+ , FlexibleContexts
18+ , FlexibleInstances
19+ , GADTs
20+ , LambdaCase
21+ , MultiParamTypeClasses
22+ , OverloadedLabels
23+ , OverloadedStrings
24+ , RankNTypes
25+ , ScopedTypeVariables
26+ , TypeApplications
27+ , TypeInType
28+ , TypeOperators
29+ , UndecidableSuperClasses
30+#-}
31+
32+module Squeal.PostgreSQL.Definition.View
33+ ( createView
34+ , createOrReplaceView
35+ , dropView
36+ , dropViewIfExists
37+ ) where
38+
39+import GHC.TypeLits
40+
41+import Squeal.PostgreSQL.Alias
42+import Squeal.PostgreSQL.Definition
43+import Squeal.PostgreSQL.Query
44+import Squeal.PostgreSQL.Render
45+import Squeal.PostgreSQL.Schema
46+
47+-- $setup
48+-- >>> import Squeal.PostgreSQL
49+
50+{- | Create a view.
51+>>> type ABC = '["a" ::: 'NoDef :=> 'Null 'PGint4, "b" ::: 'NoDef :=> 'Null 'PGint4, "c" ::: 'NoDef :=> 'Null 'PGint4]
52+>>> type BC = '["b" ::: 'Null 'PGint4, "c" ::: 'Null 'PGint4]
53+>>> :{
54+let
55+ definition :: Definition
56+ '[ "public" ::: '["abc" ::: 'Table ('[] :=> ABC)]]
57+ '[ "public" ::: '["abc" ::: 'Table ('[] :=> ABC), "bc" ::: 'View BC]]
58+ definition =
59+ createView #bc (select_ (#b :* #c) (from (table #abc)))
60+in printSQL definition
61+:}
62+CREATE VIEW "bc" AS SELECT "b" AS "b", "c" AS "c" FROM "abc" AS "abc";
63+-}
64+createView
65+ :: (Has sch schemas schema, KnownSymbol vw)
66+ => QualifiedAlias sch vw -- ^ the name of the view to add
67+ -> Query '[] '[] schemas '[] view -- ^ query
68+ -> Definition schemas (Alter sch (Create vw ('View view) schema) schemas)
69+createView alias query = UnsafeDefinition $
70+ "CREATE" <+> "VIEW" <+> renderSQL alias <+> "AS"
71+ <+> renderQuery query <> ";"
72+
73+createOrReplaceView
74+ :: (Has sch schemas schema, KnownSymbol vw)
75+ => QualifiedAlias sch vw -- ^ the name of the view to add
76+ -> Query '[] '[] schemas '[] view -- ^ query
77+ -> Definition schemas (Alter sch (CreateOrReplace vw ('View view) schema) schemas)
78+createOrReplaceView alias query = UnsafeDefinition $
79+ "CREATE OR REPLACE VIEW" <+> renderSQL alias <+> "AS"
80+ <+> renderQuery query <> ";"
81+
82+-- | Drop a view.
83+--
84+-- >>> :{
85+-- let
86+-- definition :: Definition
87+-- '[ "public" ::: '["abc" ::: 'Table ('[] :=> '["a" ::: 'NoDef :=> 'Null 'PGint4, "b" ::: 'NoDef :=> 'Null 'PGint4, "c" ::: 'NoDef :=> 'Null 'PGint4])
88+-- , "bc" ::: 'View ('["b" ::: 'Null 'PGint4, "c" ::: 'Null 'PGint4])]]
89+-- '[ "public" ::: '["abc" ::: 'Table ('[] :=> '["a" ::: 'NoDef :=> 'Null 'PGint4, "b" ::: 'NoDef :=> 'Null 'PGint4, "c" ::: 'NoDef :=> 'Null 'PGint4])]]
90+-- definition = dropView #bc
91+-- in printSQL definition
92+-- :}
93+-- DROP VIEW "bc";
94+dropView
95+ :: (Has sch schemas schema, KnownSymbol vw)
96+ => QualifiedAlias sch vw -- ^ view to remove
97+ -> Definition schemas (Alter sch (DropSchemum vw 'View schema) schemas)
98+dropView vw = UnsafeDefinition $ "DROP VIEW" <+> renderSQL vw <> ";"
99+
100+dropViewIfExists
101+ :: (Has sch schemas schema, KnownSymbol vw)
102+ => QualifiedAlias sch vw -- ^ view to remove
103+ -> Definition schemas (Alter sch (DropIfExists vw schema) schemas)
104+dropViewIfExists vw = UnsafeDefinition $ "DROP VIEW IF EXISTS" <+> renderSQL vw <> ";"
squeal-postgresql/src/Squeal/PostgreSQL/Expression.hs+10−10
@@ -34,7 +34,7 @@ module Squeal.PostgreSQL.Expression
3434 ( -- * Expression
3535 Expression (..)
3636 , Expr
37− , (:-->)
37+ , type (-->)
3838 , FunctionDB
3939 , unsafeFunction
4040 , function
@@ -48,7 +48,7 @@ module Squeal.PostgreSQL.Expression
4848 , binaryOp
4949 , FunctionVar
5050 , unsafeFunctionVar
51− , FunctionN
51+ , type (--->)
5252 , FunctionNDB
5353 , unsafeFunctionN
5454 , functionN
@@ -159,7 +159,7 @@ type OperatorDB schemas x1 x2 y
159159-- This is a subtype of the usual Haskell function type `Prelude.->`,
160160-- indeed a subcategory as it is closed under the usual
161161-- `Prelude..` and `Prelude.id`.
162−type (:-->) x y
162+type (-->) x y
163163 = forall outer commons grp schemas params from
164164 . Expression outer commons grp schemas params from x
165165 -- ^ input
@@ -179,7 +179,7 @@ Use the `*:` operator to end your argument lists, like so.
179179>>> printSQL (unsafeFunctionN "fun" (true :* false :* localTime *: true))
180180fun(TRUE, FALSE, LOCALTIME, TRUE)
181181-}
182−type FunctionN xs y
182+type (--->) xs y
183183 = forall outer commons grp schemas params from
184184 . NP (Expression outer commons grp schemas params from) xs
185185 -- ^ inputs
@@ -327,7 +327,7 @@ binaryOp = unsafeBinaryOp $ renderSymbol @op
327327
328328-- | >>> printSQL $ unsafeLeftOp "NOT" true
329329-- (NOT TRUE)
330−unsafeLeftOp :: ByteString -> x :--> y
330+unsafeLeftOp :: ByteString -> x --> y
331331unsafeLeftOp op x = UnsafeExpression $ parenthesized $ op <+> renderSQL x
332332
333333leftOp
@@ -339,7 +339,7 @@ leftOp = unsafeLeftOp $ renderSymbol @op
339339
340340-- | >>> printSQL $ true & unsafeRightOp "IS NOT TRUE"
341341-- (TRUE IS NOT TRUE)
342−unsafeRightOp :: ByteString -> x :--> y
342+unsafeRightOp :: ByteString -> x --> y
343343unsafeRightOp op x = UnsafeExpression $ parenthesized $ renderSQL x <+> op
344344
345345rightOp
@@ -351,25 +351,25 @@ rightOp = unsafeRightOp $ renderSymbol @op
351351
352352-- | >>> printSQL $ unsafeFunction "f" true
353353-- f(TRUE)
354−unsafeFunction :: ByteString -> x :--> y
354+unsafeFunction :: ByteString -> x --> y
355355unsafeFunction fun x = UnsafeExpression $
356356 fun <> parenthesized (renderSQL x)
357357
358358function
359− :: (Has sch schemas schema, Has fun schema ('Function ('[x] '::--> 'Returns y)))
359+ :: (Has sch schemas schema, Has fun schema ('Function ('[x] :=> 'Returns y)))
360360 => QualifiedAlias sch fun
361361 -> FunctionDB schemas x y
362362function = unsafeFunction . renderSQL
363363
364364-- | >>> printSQL $ unsafeFunctionN "f" (currentTime :* localTimestamp :* false *: literal 'a')
365365-- f(CURRENT_TIME, LOCALTIMESTAMP, FALSE, E'a')
366−unsafeFunctionN :: SListI xs => ByteString -> FunctionN xs y
366+unsafeFunctionN :: SListI xs => ByteString -> xs ---> y
367367unsafeFunctionN fun xs = UnsafeExpression $
368368 fun <> parenthesized (renderCommaSeparated renderSQL xs)
369369
370370functionN
371371 :: ( Has sch schemas schema
372− , Has fun schema ('Function (xs '::--> 'Returns y))
372+ , Has fun schema ('Function (xs :=> 'Returns y))
373373 , SListI xs )
374374 => QualifiedAlias sch fun
375375 -> FunctionNDB schemas xs y
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Collection.hs+5−5
@@ -42,7 +42,7 @@ import qualified Generics.SOP as SOP
4242
4343import Squeal.PostgreSQL.Alias
4444import Squeal.PostgreSQL.Expression
45−import Squeal.PostgreSQL.Expression.SetOf
45+import Squeal.PostgreSQL.Expression.Set
4646import Squeal.PostgreSQL.List
4747import Squeal.PostgreSQL.Render
4848import Squeal.PostgreSQL.Schema
@@ -111,20 +111,20 @@ array2 xss = UnsafeExpression $ "ARRAY" <>
111111
112112-- | >>> printSQL $ cardinality (array [null_, false, true])
113113-- cardinality(ARRAY[NULL, FALSE, TRUE])
114−cardinality :: null ('PGvararray ty) :--> null 'PGint8
114+cardinality :: null ('PGvararray ty) --> null 'PGint8
115115cardinality = unsafeFunction "cardinality"
116116
117117-- | >>> printSQL $ array [null_, false, true] & index 2
118118-- (ARRAY[NULL, FALSE, TRUE])[2]
119119index
120120 :: Word64 -- ^ index
121− -> null ('PGvararray ty) :--> NullifyType ty
121+ -> null ('PGvararray ty) --> NullifyType ty
122122index n expr = UnsafeExpression $
123123 parenthesized (renderSQL expr) <> "[" <> fromString (show n) <> "]"
124124
125125-- | Expand an array to a set of rows
126−unnest :: SetOfFunction "unnest" (null ('PGvararray ty)) '["unnest" ::: ty]
127−unnest = unsafeSetOfFunction
126+unnest :: SetFunction "unnest" (null ('PGvararray ty)) '["unnest" ::: ty]
127+unnest = unsafeSetFunction
128128
129129-- | A row constructor is an expression that builds a row value
130130-- (also called a composite value) using values for its member fields.
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Comparison.hs+6−6
@@ -162,7 +162,7 @@ isNotDistinctFrom = unsafeBinaryOp "IS NOT DISTINCT FROM"
162162>>> printSQL $ true & isTrue
163163(TRUE IS TRUE)
164164-}
165−isTrue :: null0 'PGbool :--> null1 'PGbool
165+isTrue :: null0 'PGbool --> null1 'PGbool
166166isTrue = unsafeRightOp "IS TRUE"
167167
168168{- | is false or unknown
@@ -170,7 +170,7 @@ isTrue = unsafeRightOp "IS TRUE"
170170>>> printSQL $ true & isNotTrue
171171(TRUE IS NOT TRUE)
172172-}
173−isNotTrue :: null0 'PGbool :--> null1 'PGbool
173+isNotTrue :: null0 'PGbool --> null1 'PGbool
174174isNotTrue = unsafeRightOp "IS NOT TRUE"
175175
176176{- | is false
@@ -178,7 +178,7 @@ isNotTrue = unsafeRightOp "IS NOT TRUE"
178178>>> printSQL $ true & isFalse
179179(TRUE IS FALSE)
180180-}
181−isFalse :: null0 'PGbool :--> null1 'PGbool
181+isFalse :: null0 'PGbool --> null1 'PGbool
182182isFalse = unsafeRightOp "IS FALSE"
183183
184184{- | is true or unknown
@@ -186,7 +186,7 @@ isFalse = unsafeRightOp "IS FALSE"
186186>>> printSQL $ true & isNotFalse
187187(TRUE IS NOT FALSE)
188188-}
189−isNotFalse :: null0 'PGbool :--> null1 'PGbool
189+isNotFalse :: null0 'PGbool --> null1 'PGbool
190190isNotFalse = unsafeRightOp "IS NOT FALSE"
191191
192192{- | is unknown
@@ -194,7 +194,7 @@ isNotFalse = unsafeRightOp "IS NOT FALSE"
194194>>> printSQL $ true & isUnknown
195195(TRUE IS UNKNOWN)
196196-}
197−isUnknown :: null0 'PGbool :--> null1 'PGbool
197+isUnknown :: null0 'PGbool --> null1 'PGbool
198198isUnknown = unsafeRightOp "IS UNKNOWN"
199199
200200{- | is true or false
@@ -202,5 +202,5 @@ isUnknown = unsafeRightOp "IS UNKNOWN"
202202>>> printSQL $ true & isNotUnknown
203203(TRUE IS NOT UNKNOWN)
204204-}
205−isNotUnknown :: null0 'PGbool :--> null1 'PGbool
205+isNotUnknown :: null0 'PGbool --> null1 'PGbool
206206isNotUnknown = unsafeRightOp "IS NOT UNKNOWN"
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Json.hs+40−47
@@ -81,7 +81,7 @@ import GHC.TypeLits
8181
8282import Squeal.PostgreSQL.Alias
8383import Squeal.PostgreSQL.Expression
84−import Squeal.PostgreSQL.Expression.SetOf
84+import Squeal.PostgreSQL.Expression.Set
8585import Squeal.PostgreSQL.Expression.Type
8686import Squeal.PostgreSQL.List
8787import Squeal.PostgreSQL.Query
@@ -188,7 +188,7 @@ Table 9.45: JSON creation functions
188188-- otherwise, a scalar value is produced. For any scalar type other than a
189189-- number, a Boolean, or a null value, the text representation will be used, in
190190-- such a fashion that it is a valid json value.
191−toJson :: null ty :--> null 'PGjson
191+toJson :: null ty --> null 'PGjson
192192toJson = unsafeFunction "to_json"
193193
194194-- | Returns the value as jsonb. Arrays and composites are converted
@@ -197,26 +197,26 @@ toJson = unsafeFunction "to_json"
197197-- otherwise, a scalar value is produced. For any scalar type other than a
198198-- number, a Boolean, or a null value, the text representation will be used, in
199199-- such a fashion that it is a valid jsonb value.
200−toJsonb :: null ty :--> null 'PGjsonb
200+toJsonb :: null ty --> null 'PGjsonb
201201toJsonb = unsafeFunction "to_jsonb"
202202
203203-- | Returns the array as a JSON array. A PostgreSQL multidimensional array
204204-- becomes a JSON array of arrays.
205−arrayToJson :: null ('PGvararray ty) :--> null 'PGjson
205+arrayToJson :: null ('PGvararray ty) --> null 'PGjson
206206arrayToJson = unsafeFunction "array_to_json"
207207
208208-- | Returns the row as a JSON object.
209−rowToJson :: null ('PGcomposite ty) :--> null 'PGjson
209+rowToJson :: null ('PGcomposite ty) --> null 'PGjson
210210rowToJson = unsafeFunction "row_to_json"
211211
212212-- | Builds a possibly-heterogeneously-typed JSON array out of a variadic
213213-- argument list.
214−jsonBuildArray :: SOP.SListI tuple => FunctionN tuple (null 'PGjson)
214+jsonBuildArray :: SOP.SListI tuple => tuple ---> null 'PGjson
215215jsonBuildArray = unsafeFunctionN "json_build_array"
216216
217217-- | Builds a possibly-heterogeneously-typed (binary) JSON array out of a
218218-- variadic argument list.
219−jsonbBuildArray :: SOP.SListI tuple => FunctionN tuple (null 'PGjsonb)
219+jsonbBuildArray :: SOP.SListI tuple => tuple ---> null 'PGjsonb
220220jsonbBuildArray = unsafeFunctionN "jsonb_build_array"
221221
222222-- | Builds a possibly-heterogeneously-typed JSON object out of a variadic
@@ -224,10 +224,10 @@ jsonbBuildArray = unsafeFunctionN "jsonb_build_array"
224224-- and values.
225225class SOP.SListI tys => JsonBuildObject tys where
226226
227− jsonBuildObject :: FunctionN tys (null 'PGjson)
227+ jsonBuildObject :: tys ---> null 'PGjson
228228 jsonBuildObject = unsafeFunctionN "json_build_object"
229229
230− jsonbBuildObject :: FunctionN tys (null 'PGjsonb)
230+ jsonbBuildObject :: tys ---> null 'PGjsonb
231231 jsonbBuildObject = unsafeFunctionN "jsonb_build_object"
232232
233233instance JsonBuildObject '[]
@@ -240,7 +240,7 @@ instance (JsonBuildObject tys, key `In` PGJsonKey)
240240-- which are taken as a key/value pair.
241241jsonObject
242242 :: null ('PGfixarray '[n,2] ('NotNull 'PGtext))
243− :--> null 'PGjson
243+ --> null 'PGjson
244244jsonObject = unsafeFunction "json_object"
245245
246246-- | Builds a binary JSON object out of a text array.
@@ -249,24 +249,24 @@ jsonObject = unsafeFunction "json_object"
249249-- which are taken as a key/value pair.
250250jsonbObject
251251 :: null ('PGfixarray '[n,2] ('NotNull 'PGtext))
252− :--> null 'PGjsonb
252+ --> null 'PGjsonb
253253jsonbObject = unsafeFunction "jsonb_object"
254254
255255-- | This is an alternate form of 'jsonObject' that takes two arrays; one for
256256-- keys and one for values, that are zipped pairwise to create a JSON object.
257−jsonZipObject :: FunctionN
257+jsonZipObject ::
258258 '[ null ('PGvararray ('NotNull 'PGtext))
259259 , null ('PGvararray ('NotNull 'PGtext)) ]
260− ( null 'PGjson )
260+ ---> null 'PGjson
261261jsonZipObject = unsafeFunctionN "json_object"
262262
263263-- | This is an alternate form of 'jsonbObject' that takes two arrays; one for
264264-- keys and one for values, that are zipped pairwise to create a binary JSON
265265-- object.
266−jsonbZipObject :: FunctionN
266+jsonbZipObject ::
267267 '[ null ('PGvararray ('NotNull 'PGtext))
268268 , null ('PGvararray ('NotNull 'PGtext)) ]
269− ( null 'PGjsonb )
269+ ---> null 'PGjsonb
270270jsonbZipObject = unsafeFunctionN "jsonb_object"
271271
272272{-----------------------------------------
@@ -274,31 +274,31 @@ Table 9.46: JSON processing functions
274274-----------------------------------------}
275275
276276-- | Returns the number of elements in the outermost JSON array.
277−jsonArrayLength :: null 'PGjson :--> null 'PGint4
277+jsonArrayLength :: null 'PGjson --> null 'PGint4
278278jsonArrayLength = unsafeFunction "json_array_length"
279279
280280-- | Returns the number of elements in the outermost binary JSON array.
281−jsonbArrayLength :: null 'PGjsonb :--> null 'PGint4
281+jsonbArrayLength :: null 'PGjsonb --> null 'PGint4
282282jsonbArrayLength = unsafeFunction "jsonb_array_length"
283283
284284-- | Returns the type of the outermost JSON value as a text string. Possible
285285-- types are object, array, string, number, boolean, and null.
286−jsonTypeof :: null 'PGjson :--> null 'PGtext
286+jsonTypeof :: null 'PGjson --> null 'PGtext
287287jsonTypeof = unsafeFunction "json_typeof"
288288
289289-- | Returns the type of the outermost binary JSON value as a text string.
290290-- Possible types are object, array, string, number, boolean, and null.
291−jsonbTypeof :: null 'PGjsonb :--> null 'PGtext
291+jsonbTypeof :: null 'PGjsonb --> null 'PGtext
292292jsonbTypeof = unsafeFunction "jsonb_typeof"
293293
294294-- | Returns its argument with all object fields that have null values omitted.
295295-- Other null values are untouched.
296−jsonStripNulls :: null 'PGjson :--> null 'PGjson
296+jsonStripNulls :: null 'PGjson --> null 'PGjson
297297jsonStripNulls = unsafeFunction "json_strip_nulls"
298298
299299-- | Returns its argument with all object fields that have null values omitted.
300300-- Other null values are untouched.
301−jsonbStripNulls :: null 'PGjsonb :--> null 'PGjsonb
301+jsonbStripNulls :: null 'PGjsonb --> null 'PGjsonb
302302jsonbStripNulls = unsafeFunction "jsonb_strip_nulls"
303303
304304-- | @ jsonbSet target path new_value create_missing @
@@ -309,12 +309,9 @@ jsonbStripNulls = unsafeFunction "jsonb_strip_nulls"
309309-- item designated by path does not exist. As with the path orientated
310310-- operators, negative integers that appear in path count from the end of JSON
311311-- arrays.
312−jsonbSet
313− :: FunctionN
314− '[ null 'PGjsonb
315− , null ('PGvararray ('NotNull 'PGtext))
316− , null 'PGjsonb
317− , null 'PGbool ] (null 'PGjsonb)
312+jsonbSet ::
313+ '[ null 'PGjsonb, null ('PGvararray ('NotNull 'PGtext))
314+ , null 'PGjsonb, null 'PGbool ] ---> null 'PGjsonb
318315jsonbSet = unsafeFunctionN "jsonbSet"
319316
320317-- | @ jsonbInsert target path new_value insert_after @
@@ -326,27 +323,23 @@ jsonbSet = unsafeFunctionN "jsonbSet"
326323-- path is in JSONB object, @new_value@ will be inserted only if target does not
327324-- exist. As with the path orientated operators, negative integers that appear
328325-- in path count from the end of JSON arrays.
329−jsonbInsert
330− :: FunctionN
331− '[ null 'PGjsonb
332− , null ('PGvararray ('NotNull 'PGtext))
333− , null 'PGjsonb
334− , null 'PGbool ] (null 'PGjsonb)
326+jsonbInsert ::
327+ '[ null 'PGjsonb, null ('PGvararray ('NotNull 'PGtext))
328+ , null 'PGjsonb, null 'PGbool ] ---> null 'PGjsonb
335329jsonbInsert = unsafeFunctionN "jsonb_insert"
336330
337331-- | Returns its argument as indented JSON text.
338−jsonbPretty :: null 'PGjsonb :--> null 'PGtext
332+jsonbPretty :: null 'PGjsonb --> null 'PGtext
339333jsonbPretty = unsafeFunction "jsonb_pretty"
340334
341−
342335{- | Expands the outermost JSON object into a set of key/value pairs.
343336
344337>>> printSQL (select Star (from (jsonEach (literal (Json (object ["a" .= "foo", "b" .= "bar"]))))))
345338SELECT * FROM json_each(('{"a":"foo","b":"bar"}' :: json))
346339-}
347−jsonEach :: SetOfFunction "json_each" (null 'PGjson)
340+jsonEach :: SetFunction "json_each" (null 'PGjson)
348341 '["key" ::: 'NotNull 'PGtext, "value" ::: 'NotNull 'PGjson]
349−jsonEach = unsafeSetOfFunction
342+jsonEach = unsafeSetFunction
350343
351344{- | Expands the outermost binary JSON object into a set of key/value pairs.
352345
@@ -354,9 +347,9 @@ jsonEach = unsafeSetOfFunction
354347SELECT * FROM jsonb_each(('{"a":"foo","b":"bar"}' :: jsonb))
355348-}
356349jsonbEach
357− :: SetOfFunction "jsonb_each" (nullity 'PGjsonb)
350+ :: SetFunction "jsonb_each" (nullity 'PGjsonb)
358351 '["key" ::: 'NotNull 'PGtext, "value" ::: 'NotNull 'PGjson]
359−jsonbEach = unsafeSetOfFunction
352+jsonbEach = unsafeSetFunction
360353
361354{- | Expands the outermost JSON object into a set of key/value pairs.
362355
@@ -364,9 +357,9 @@ jsonbEach = unsafeSetOfFunction
364357SELECT * FROM json_each_text(('{"a":"foo","b":"bar"}' :: json))
365358-}
366359jsonEachText
367− :: SetOfFunction "json_each_text" (null 'PGjson)
360+ :: SetFunction "json_each_text" (null 'PGjson)
368361 '["key" ::: 'NotNull 'PGtext, "value" ::: 'NotNull 'PGtext]
369−jsonEachText = unsafeSetOfFunction
362+jsonEachText = unsafeSetFunction
370363
371364{- | Expands the outermost binary JSON object into a set of key/value pairs.
372365
@@ -374,9 +367,9 @@ jsonEachText = unsafeSetOfFunction
374367SELECT * FROM jsonb_each_text(('{"a":"foo","b":"bar"}' :: jsonb))
375368-}
376369jsonbEachText
377− :: SetOfFunction "jsonb_each_text" (null 'PGjsonb)
370+ :: SetFunction "jsonb_each_text" (null 'PGjsonb)
378371 '["key" ::: 'NotNull 'PGtext, "value" ::: 'NotNull 'PGtext]
379−jsonbEachText = unsafeSetOfFunction
372+jsonbEachText = unsafeSetFunction
380373
381374{- | Returns set of keys in the outermost JSON object.
382375
@@ -384,9 +377,9 @@ jsonbEachText = unsafeSetOfFunction
384377json_object_keys(('{"a":"foo","b":"bar"}' :: json))
385378-}
386379jsonObjectKeys
387− :: SetOfFunction "json_object_keys" (nullity 'PGjson)
380+ :: SetFunction "json_object_keys" (nullity 'PGjson)
388381 '["json_object_keys" ::: 'NotNull 'PGtext]
389−jsonObjectKeys = unsafeSetOfFunction
382+jsonObjectKeys = unsafeSetFunction
390383
391384{- | Returns set of keys in the outermost JSON object.
392385
@@ -394,9 +387,9 @@ jsonObjectKeys = unsafeSetOfFunction
394387jsonb_object_keys(('{"a":"foo","b":"bar"}' :: jsonb))
395388-}
396389jsonbObjectKeys
397− :: SetOfFunction "jsonb_object_keys" (null 'PGjsonb)
390+ :: SetFunction "jsonb_object_keys" (null 'PGjsonb)
398391 '["jsonb_object_keys" ::: 'NotNull 'PGtext]
399−jsonbObjectKeys = unsafeSetOfFunction
392+jsonbObjectKeys = unsafeSetFunction
400393
401394-- | Build rows from Json types.
402395type JsonPopulateFunction fun json
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Logic.hs+1−1
@@ -47,7 +47,7 @@ false = UnsafeExpression "FALSE"
4747
4848-- | >>> printSQL $ not_ true
4949-- (NOT TRUE)
50−not_ :: null 'PGbool :--> null 'PGbool
50+not_ :: null 'PGbool --> null 'PGbool
5151not_ = unsafeLeftOp "NOT"
5252
5353-- | >>> printSQL $ true .&& false
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Math.hs+4−6
@@ -37,9 +37,7 @@ import Squeal.PostgreSQL.Schema
3737-- in printSQL expression
3838-- :}
3939-- atan2(pi(), 2)
40−atan2_
41− :: float `In` PGFloating
42− => FunctionN '[ null float, null float] (null float)
40+atan2_ :: float `In` PGFloating => '[ null float, null float] ---> null float
4341atan2_ = unsafeFunctionN "atan2"
4442
4543
@@ -78,7 +76,7 @@ rem_ = unsafeBinaryOp "%"
7876-- in printSQL expression
7977-- :}
8078-- trunc(pi())
81−trunc :: frac `In` PGFloating => null frac :--> null frac
79+trunc :: frac `In` PGFloating => null frac --> null frac
8280trunc = unsafeFunction "trunc"
8381
8482-- | >>> :{
@@ -88,7 +86,7 @@ trunc = unsafeFunction "trunc"
8886-- in printSQL expression
8987-- :}
9088-- round(pi())
91−round_ :: frac `In` PGFloating => null frac :--> null frac
89+round_ :: frac `In` PGFloating => null frac --> null frac
9290round_ = unsafeFunction "round"
9391
9492-- | >>> :{
@@ -98,5 +96,5 @@ round_ = unsafeFunction "round"
9896-- in printSQL expression
9997-- :}
10098-- ceiling(pi())
101−ceiling_ :: frac `In` PGFloating => null frac :--> null frac
99+ceiling_ :: frac `In` PGFloating => null frac --> null frac
102100ceiling_ = unsafeFunction "ceiling"
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Null.hs+4−4
@@ -44,7 +44,7 @@ null_ = UnsafeExpression "NULL"
4444--
4545-- >>> printSQL $ notNull true
4646-- TRUE
47−notNull :: 'NotNull ty :--> 'Null ty
47+notNull :: 'NotNull ty --> 'Null ty
4848notNull = UnsafeExpression . renderSQL
4949
5050-- | return the leftmost value which is not NULL
@@ -69,12 +69,12 @@ fromNull notNullx nullx = coalesce [nullx] notNullx
6969
7070-- | >>> printSQL $ null_ & isNull
7171-- NULL IS NULL
72−isNull :: 'Null ty :--> null 'PGbool
72+isNull :: 'Null ty --> null 'PGbool
7373isNull x = UnsafeExpression $ renderSQL x <+> "IS NULL"
7474
7575-- | >>> printSQL $ null_ & isNotNull
7676-- NULL IS NOT NULL
77−isNotNull :: 'Null ty :--> null 'PGbool
77+isNotNull :: 'Null ty --> null 'PGbool
7878isNotNull x = UnsafeExpression $ renderSQL x <+> "IS NOT NULL"
7979
8080-- | analagous to `maybe` using @IS NULL@
@@ -100,5 +100,5 @@ matchNull y f x = ifThenElse (isNull x) y
100100>>> printSQL expr
101101NULLIF(FALSE, ($1 :: bool))
102102-}
103−nullIf :: FunctionN '[ 'NotNull ty, 'NotNull ty] ('Null ty)
103+nullIf :: '[ 'NotNull ty, 'NotNull ty] ---> 'Null ty
104104nullIf = unsafeFunctionN "NULLIF"
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Range.hs+8−9
@@ -205,27 +205,26 @@ whole = NonEmpty Infinite Infinite
205205(@-) :: Operator (null ('PGrange ty)) (null ('PGrange ty)) (null ('PGrange ty))
206206(@-) = unsafeBinaryOp "-"
207207
208−lowerBound :: null ('PGrange ty) :--> 'Null ty
208+lowerBound :: null ('PGrange ty) --> 'Null ty
209209lowerBound = unsafeFunction "lower"
210210
211−upperBound :: null ('PGrange ty) :--> 'Null ty
211+upperBound :: null ('PGrange ty) --> 'Null ty
212212upperBound = unsafeFunction "upper"
213213
214−isEmpty :: null ('PGrange ty) :--> 'Null 'PGbool
214+isEmpty :: null ('PGrange ty) --> 'Null 'PGbool
215215isEmpty = unsafeFunction "isempty"
216216
217−lowerInc :: null ('PGrange ty) :--> 'Null 'PGbool
217+lowerInc :: null ('PGrange ty) --> 'Null 'PGbool
218218lowerInc = unsafeFunction "lower_inc"
219219
220−lowerInf :: null ('PGrange ty) :--> 'Null 'PGbool
220+lowerInf :: null ('PGrange ty) --> 'Null 'PGbool
221221lowerInf = unsafeFunction "lower_inf"
222222
223−upperInc :: null ('PGrange ty) :--> 'Null 'PGbool
223+upperInc :: null ('PGrange ty) --> 'Null 'PGbool
224224upperInc = unsafeFunction "upper_inc"
225225
226−upperInf :: null ('PGrange ty) :--> 'Null 'PGbool
226+upperInf :: null ('PGrange ty) --> 'Null 'PGbool
227227upperInf = unsafeFunction "upper_inf"
228228
229−rangeMerge :: FunctionN
230− '[ null ('PGrange ty), null ('PGrange ty)] (null ('PGrange ty))
229+rangeMerge :: '[null ('PGrange ty), null ('PGrange ty)] ---> null ('PGrange ty)
231230rangeMerge = unsafeFunctionN "range_merge"
squeal-postgresql/src/Squeal/PostgreSQL/Expression/SetOf.hs → squeal-postgresql/src/Squeal/PostgreSQL/Expression/Set.hsrenamed+52−40
@@ -1,5 +1,5 @@
11{-|
2−Module: Squeal.PostgreSQL.Expression.SetOf
2+Module: Squeal.PostgreSQL.Expression.Set
33Description: Set returning functions
44Copyright: (c) Eitan Chatav, 2019
55Maintainer: eitan@morphism.tech
@@ -20,16 +20,18 @@ Set returning functions
2020 , TypeOperators
2121#-}
2222
23−module Squeal.PostgreSQL.Expression.SetOf
23+module Squeal.PostgreSQL.Expression.Set
2424 ( generateSeries
2525 , generateSeriesStep
2626 , generateSeriesTimestamp
27− , SetOfFunction
28− , unsafeSetOfFunction
29− , setOfFunction
30− , SetOfFunctionN
31− , unsafeSetOfFunctionN
32− , setOfFunctionN
27+ , SetFunction
28+ , unsafeSetFunction
29+ , SetFunctionDB
30+ , setFunction
31+ , SetFunctionN
32+ , unsafeSetFunctionN
33+ , SetFunctionNDB
34+ , setFunctionN
3335 ) where
3436
3537import GHC.TypeLits
@@ -46,55 +48,65 @@ import Squeal.PostgreSQL.Schema
4648{- |
4749A @RankNType@ for set returning functions with 1 argument.
4850-}
49−type SetOfFunction fun ty setof
51+type SetFunction fun ty row
5052 = forall outer commons schemas params
5153 . Expression outer commons 'Ungrouped schemas params '[] ty
5254 -- ^ input
53− -> FromClause outer commons schemas params '[fun ::: setof]
55+ -> FromClause outer commons schemas params '[fun ::: row]
56+ -- ^ output
57+
58+type SetFunctionDB fun schemas ty row
59+ = forall outer commons params
60+ . Expression outer commons 'Ungrouped schemas params '[] ty
61+ -- ^ input
62+ -> FromClause outer commons schemas params '[fun ::: row]
5463 -- ^ output
5564
5665-- | Escape hatch for a set returning function with 1 argument.
57−unsafeSetOfFunction
58− :: forall fun ty setof. KnownSymbol fun
59− => SetOfFunction fun ty setof -- ^ set returning function
60−unsafeSetOfFunction x = UnsafeFromClause $
66+unsafeSetFunction
67+ :: forall fun ty row. KnownSymbol fun
68+ => SetFunction fun ty row -- ^ set returning function
69+unsafeSetFunction x = UnsafeFromClause $
6170 renderSymbol @fun <> parenthesized (renderSQL x)
6271
63−setOfFunction
72+setFunction
6473 :: ( Has sch schemas schema
65− , Has fun schema ('Function ('[ty] '::--> 'ReturnsTable setof)) )
66− => Alias fun
67− -> ( forall outer commons params
68− . Expression outer commons 'Ungrouped schemas params '[] ty
69− -> FromClause outer commons schemas params '[fun ::: setof] )
70−setOfFunction _ = unsafeSetOfFunction
74+ , Has fun schema ('Function ('[ty] :=> 'ReturnsTable row)) )
75+ => QualifiedAlias sch fun
76+ -> SetFunctionDB fun schemas ty row
77+setFunction _ = unsafeSetFunction
7178
7279{- |
7380A @RankNType@ for set returning functions with multiple argument.
7481-}
75−type SetOfFunctionN fun tys setof
82+type SetFunctionN fun tys row
7683 = forall outer commons schemas params
7784 . NP (Expression outer commons 'Ungrouped schemas params '[]) tys
7885 -- ^ inputs
79− -> FromClause outer commons schemas params '[fun ::: setof]
86+ -> FromClause outer commons schemas params '[fun ::: row]
8087 -- ^ output
8188
8289-- | Escape hatch for a set returning function with multiple argument.
83−unsafeSetOfFunctionN
84− :: forall fun tys setof. (SOP.SListI tys, KnownSymbol fun)
85− => SetOfFunctionN fun tys setof -- ^ set returning function
86−unsafeSetOfFunctionN xs = UnsafeFromClause $
90+unsafeSetFunctionN
91+ :: forall fun tys row. (SOP.SListI tys, KnownSymbol fun)
92+ => SetFunctionN fun tys row -- ^ set returning function
93+unsafeSetFunctionN xs = UnsafeFromClause $
8794 renderSymbol @fun <> parenthesized (renderCommaSeparated renderSQL xs)
8895
89−setOfFunctionN
96+type SetFunctionNDB fun schemas tys row
97+ = forall outer commons params
98+ . NP (Expression outer commons 'Ungrouped schemas params '[]) tys
99+ -- ^ inputs
100+ -> FromClause outer commons schemas params '[fun ::: row]
101+ -- ^ output
102+
103+setFunctionN
90104 :: ( Has sch schemas schema
91− , Has fun schema ('Function (tys '::--> 'ReturnsTable setof))
105+ , Has fun schema ('Function (tys :=> 'ReturnsTable row))
92106 , SOP.SListI tys )
93− => Alias fun
94− -> ( forall outer commons params
95− . NP (Expression outer commons 'Ungrouped schemas params '[]) tys
96− -> FromClause outer commons schemas params '[fun ::: setof] )
97−setOfFunctionN _ = unsafeSetOfFunctionN
107+ => QualifiedAlias sch fun
108+ -> SetFunctionNDB fun schemas tys row
109+setFunctionN _ = unsafeSetFunctionN
98110
99111{- | @generateSeries (start *: stop)@
100112
@@ -102,9 +114,9 @@ Generate a series of values, from @start@ to @stop@ with a step size of one
102114-}
103115generateSeries
104116 :: ty `In` '[ 'PGint4, 'PGint8, 'PGnumeric]
105− => SetOfFunctionN "generate_series" '[ null ty, null ty]
117+ => SetFunctionN "generate_series" '[ null ty, null ty]
106118 '["generate_series" ::: null ty] -- ^ set returning function
107−generateSeries = unsafeSetOfFunctionN
119+generateSeries = unsafeSetFunctionN
108120
109121{- | @generateSeries (start :* stop *: step)@
110122
@@ -112,9 +124,9 @@ Generate a series of values, from @start@ to @stop@ with a step size of @step@
112124-}
113125generateSeriesStep
114126 :: ty `In` '[ 'PGint4, 'PGint8, 'PGnumeric]
115− => SetOfFunctionN "generate_series" '[null ty, null ty, null ty]
127+ => SetFunctionN "generate_series" '[null ty, null ty, null ty]
116128 '["generate_series" ::: null ty] -- ^ set returning function
117−generateSeriesStep = unsafeSetOfFunctionN
129+generateSeriesStep = unsafeSetFunctionN
118130
119131{- | @generateSeries (start :* stop *: step)@
120132
@@ -122,6 +134,6 @@ Generate a series of values, from @start@ to @stop@ with a step size of @step@
122134-}
123135generateSeriesTimestamp
124136 :: ty `In` '[ 'PGtimestamp, 'PGtimestamptz]
125− => SetOfFunctionN "generate_series" '[null ty, null ty, null 'PGinterval]
137+ => SetFunctionN "generate_series" '[null ty, null ty, null 'PGinterval]
126138 '["generate_series" ::: null ty] -- ^ set returning function
127−generateSeriesTimestamp = unsafeSetOfFunctionN
139+generateSeriesTimestamp = unsafeSetFunctionN
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Text.hs+3−3
@@ -30,17 +30,17 @@ import Squeal.PostgreSQL.Schema
3030
3131-- | >>> printSQL $ lower "ARRRGGG"
3232-- lower(E'ARRRGGG')
33−lower :: null 'PGtext :--> null 'PGtext
33+lower :: null 'PGtext --> null 'PGtext
3434lower = unsafeFunction "lower"
3535
3636-- | >>> printSQL $ upper "eeee"
3737-- upper(E'eeee')
38−upper :: null 'PGtext :--> null 'PGtext
38+upper :: null 'PGtext --> null 'PGtext
3939upper = unsafeFunction "upper"
4040
4141-- | >>> printSQL $ charLength "four"
4242-- char_length(E'four')
43−charLength :: null 'PGtext :--> null 'PGint4
43+charLength :: null 'PGtext --> null 'PGint4
4444charLength = unsafeFunction "char_length"
4545
4646-- | The `like` expression returns true if the @string@ matches
squeal-postgresql/src/Squeal/PostgreSQL/Expression/TextSearch.hs+19−20
@@ -55,7 +55,7 @@ import Squeal.PostgreSQL.Schema
5555(.|) = unsafeBinaryOp "||"
5656
5757-- | negate a `Squeal.PostgreSQL.Expression.Type.tsquery`
58−(.!) :: null 'PGtsquery :--> null 'PGtsquery
58+(.!) :: null 'PGtsquery --> null 'PGtsquery
5959(.!) = unsafeLeftOp "!!"
6060
6161-- | `Squeal.PostgreSQL.Expression.Type.tsquery` followed by
@@ -66,51 +66,50 @@ import Squeal.PostgreSQL.Schema
6666-- | convert array of lexemes to `Squeal.PostgreSQL.Expression.Type.tsvector`
6767arrayToTSvector
6868 :: null ('PGvararray ('NotNull 'PGtext))
69− :--> null 'PGtsvector
69+ --> null 'PGtsvector
7070arrayToTSvector = unsafeFunction "array_to_tsvector"
7171
7272-- | number of lexemes in `Squeal.PostgreSQL.Expression.Type.tsvector`
73−tsvectorLength :: null 'PGtsvector :--> null 'PGint4
73+tsvectorLength :: null 'PGtsvector --> null 'PGint4
7474tsvectorLength = unsafeFunction "length"
7575
7676-- | number of lexemes plus operators in `Squeal.PostgreSQL.Expression.Type.tsquery`
77−numnode :: null 'PGtsquery :--> null 'PGint4
77+numnode :: null 'PGtsquery --> null 'PGint4
7878numnode = unsafeFunction "numnode"
7979
8080-- | produce `Squeal.PostgreSQL.Expression.Type.tsquery` ignoring punctuation
81−plainToTSquery :: null 'PGtext :--> null 'PGtsquery
81+plainToTSquery :: null 'PGtext --> null 'PGtsquery
8282plainToTSquery = unsafeFunction "plainto_tsquery"
8383
8484-- | produce `Squeal.PostgreSQL.Expression.Type.tsquery` that searches for a phrase,
8585-- ignoring punctuation
86−phraseToTSquery :: null 'PGtext :--> null 'PGtsquery
86+phraseToTSquery :: null 'PGtext --> null 'PGtsquery
8787phraseToTSquery = unsafeFunction "phraseto_tsquery"
8888
8989-- | produce `Squeal.PostgreSQL.Expression.Type.tsquery` from a web search style query
90−websearchToTSquery :: null 'PGtext :--> null 'PGtsquery
90+websearchToTSquery :: null 'PGtext --> null 'PGtsquery
9191websearchToTSquery = unsafeFunction "websearch_to_tsquery"
9292
9393-- | get indexable part of a `Squeal.PostgreSQL.Expression.Type.tsquery`
94−queryTree :: null 'PGtsquery :--> null 'PGtext
94+queryTree :: null 'PGtsquery --> null 'PGtext
9595queryTree = unsafeFunction "query_tree"
9696
9797-- | normalize words and convert to `Squeal.PostgreSQL.Expression.Type.tsquery`
98−toTSquery :: null 'PGtext :--> null 'PGtsquery
98+toTSquery :: null 'PGtext --> null 'PGtsquery
9999toTSquery = unsafeFunction "to_tsquery"
100100
101101-- | reduce document text to `Squeal.PostgreSQL.Expression.Type.tsvector`
102102toTSvector
103103 :: ty `In` '[ 'PGtext, 'PGjson, 'PGjsonb]
104− => null ty :--> null 'PGtsvector
104+ => null ty --> null 'PGtsvector
105105toTSvector = unsafeFunction "to_tsvector"
106106
107107-- | assign weight to each element of `Squeal.PostgreSQL.Expression.Type.tsvector`
108−setWeight
109− :: FunctionN '[null 'PGtsvector, null ('PGchar 1)] (null 'PGtsvector)
108+setWeight :: '[null 'PGtsvector, null ('PGchar 1)] ---> null 'PGtsvector
110109setWeight = unsafeFunctionN "set_weight"
111110
112111-- | remove positions and weights from `Squeal.PostgreSQL.Expression.Type.tsvector`
113−strip :: null 'PGtsvector :--> null 'PGtsvector
112+strip :: null 'PGtsvector --> null 'PGtsvector
114113strip = unsafeFunction "strip"
115114
116115-- | @jsonToTSvector (document *: filter)@
@@ -123,7 +122,7 @@ strip = unsafeFunction "strip"
123122-- "boolean" (to include all Boolean values in the string format "true"/"false"),
124123-- "key" (to include all keys) or "all" (to include all above).
125124-- These values can be combined together to include, e.g. all string and numeric values.
126−jsonToTSvector :: FunctionN '[null 'PGjson, null 'PGjson] (null 'PGtsvector)
125+jsonToTSvector :: '[null 'PGjson, null 'PGjson] ---> null 'PGtsvector
127126jsonToTSvector = unsafeFunctionN "json_to_tsvector"
128127
129128-- | @jsonbToTSvector (document *: filter)@
@@ -136,23 +135,23 @@ jsonToTSvector = unsafeFunctionN "json_to_tsvector"
136135-- "boolean" (to include all Boolean values in the string format "true"/"false"),
137136-- "key" (to include all keys) or "all" (to include all above).
138137-- These values can be combined together to include, e.g. all string and numeric values.
139−jsonbToTSvector :: FunctionN '[null 'PGjsonb, null 'PGjsonb] (null 'PGtsvector)
138+jsonbToTSvector :: '[null 'PGjsonb, null 'PGjsonb] ---> null 'PGtsvector
140139jsonbToTSvector = unsafeFunctionN "jsonb_to_tsvector"
141140
142141-- | remove given lexeme from `Squeal.PostgreSQL.Expression.Type.tsvector`
143−tsDelete :: FunctionN
142+tsDelete ::
144143 '[null 'PGtsvector, null ('PGvararray ('NotNull 'PGtext))]
145− (null 'PGtsvector)
144+ ---> null 'PGtsvector
146145tsDelete = unsafeFunctionN "ts_delete"
147146
148147-- | select only elements with given weights from `Squeal.PostgreSQL.Expression.Type.tsvector`
149−tsFilter :: FunctionN
148+tsFilter ::
150149 '[null 'PGtsvector, null ('PGvararray ('NotNull ('PGchar 1)))]
151− (null 'PGtsvector)
150+ ---> null 'PGtsvector
152151tsFilter = unsafeFunctionN "ts_filter"
153152
154153-- | display a `Squeal.PostgreSQL.Expression.Type.tsquery` match
155154tsHeadline
156155 :: document `In` '[ 'PGtext, 'PGjson, 'PGjsonb]
157− => FunctionN '[null document, null 'PGtsquery] (null 'PGtext)
156+ => '[null document, null 'PGtsquery] ---> null 'PGtext
158157tsHeadline = unsafeFunctionN "ts_headline"
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Time.hs+7−12
@@ -17,6 +17,7 @@ Date/Time functions and operators
1717 , OverloadedStrings
1818 , PolyKinds
1919 , RankNTypes
20+ , TypeOperators
2021#-}
2122
2223module Squeal.PostgreSQL.Expression.Time
@@ -85,9 +86,7 @@ Create date from year, month and day fields
8586>>> printSQL (makeDate (1984 :* 7 *: 3))
8687make_date(1984, 7, 3)
8788-}
88−makeDate :: FunctionN
89− '[ null 'PGint4, null 'PGint4, null 'PGint4 ]
90− ( null 'PGdate )
89+makeDate :: '[ null 'PGint4, null 'PGint4, null 'PGint4 ] ---> null 'PGdate
9190makeDate = unsafeFunctionN "make_date"
9291
9392{-|
@@ -96,9 +95,7 @@ Create time from hour, minute and seconds fields
9695>>> printSQL (makeTime (8 :* 15 *: 23.5))
9796make_time(8, 15, 23.5)
9897-}
99−makeTime :: FunctionN
100− '[ null 'PGint4, null 'PGint4, null 'PGfloat8 ]
101− ( null 'PGtime )
98+makeTime :: '[ null 'PGint4, null 'PGint4, null 'PGfloat8 ] ---> null 'PGtime
10299makeTime = unsafeFunctionN "make_time"
103100
104101{-|
@@ -107,10 +104,9 @@ Create timestamp from year, month, day, hour, minute and seconds fields
107104>>> printSQL (makeTimestamp (2013 :* 7 :* 15 :* 8 :* 15 *: 23.5))
108105make_timestamp(2013, 7, 15, 8, 15, 23.5)
109106-}
110−makeTimestamp :: FunctionN
107+makeTimestamp ::
111108 '[ null 'PGint4, null 'PGint4, null 'PGint4
112− , null 'PGint4, null 'PGint4, null 'PGfloat8 ]
113− ( null 'PGtimestamp )
109+ , null 'PGint4, null 'PGint4, null 'PGfloat8 ] ---> null 'PGtimestamp
114110makeTimestamp = unsafeFunctionN "make_timestamp"
115111
116112{-|
@@ -121,10 +117,9 @@ the current time zone is used
121117>>> printSQL (makeTimestamptz (2013 :* 7 :* 15 :* 8 :* 15 *: 23.5))
122118make_timestamptz(2013, 7, 15, 8, 15, 23.5)
123119-}
124−makeTimestamptz :: FunctionN
120+makeTimestamptz ::
125121 '[ null 'PGint4, null 'PGint4, null 'PGint4
126− , null 'PGint4, null 'PGint4, null 'PGfloat8 ]
127− ( null 'PGtimestamptz )
122+ , null 'PGint4, null 'PGint4, null 'PGfloat8 ] ---> null 'PGtimestamptz
128123makeTimestamptz = unsafeFunctionN "make_timestamptz"
129124
130125{-|
squeal-postgresql/src/Squeal/PostgreSQL/Expression/Type.hs+108−3
@@ -23,14 +23,16 @@ Type expressions.
2323 , ScopedTypeVariables
2424 , TypeApplications
2525 , TypeOperators
26+ , UndecidableInstances
2627#-}
2728
2829module Squeal.PostgreSQL.Expression.Type
29− ( TypeExpression (..)
30− , cast
30+ ( -- * type casting
31+ cast
3132 , astype
3233 , inferredtype
33− , PGTyped (..)
34+ -- * type expressions
35+ , TypeExpression (..)
3436 , typedef
3537 , typetable
3638 , typeview
@@ -77,6 +79,22 @@ module Squeal.PostgreSQL.Expression.Type
7779 , tsrange
7880 , tstzrange
7981 , daterange
82+ -- * column type definitions
83+ , ColumnTypeExpression (..)
84+ , nullable
85+ , notNullable
86+ , default_
87+ , serial2
88+ , smallserial
89+ , serial4
90+ , serial
91+ , serial8
92+ , bigserial
93+ , hask
94+ -- * type inference
95+ , PGTyped (..)
96+ , PGNullityTyped (..)
97+ , FieldTyped (..)
8098 ) where
8199
82100import Control.DeepSeq
@@ -90,6 +108,7 @@ import qualified Generics.SOP as SOP
90108
91109import Squeal.PostgreSQL.Alias
92110import Squeal.PostgreSQL.Expression
111+import Squeal.PostgreSQL.PG
93112import Squeal.PostgreSQL.Render
94113import Squeal.PostgreSQL.Schema
95114
@@ -131,6 +150,7 @@ astype = cast
131150inferredtype
132151 :: PGTyped schemas ty
133152 => Expression outer common grp schemas params from ty
153+ -- ^ value
134154 -> Expression outer common grp schemas params from ty
135155inferredtype = astype pgtype
136156
@@ -150,6 +170,7 @@ instance RenderSQL (TypeExpression schemas ty) where
150170typedef
151171 :: (Has sch schemas schema, Has td schema ('Typedef ty))
152172 => QualifiedAlias sch td
173+ -- ^ type alias
153174 -> TypeExpression schemas (null ty)
154175typedef = UnsafeTypeExpression . renderSQL
155176
@@ -158,6 +179,7 @@ typedef = UnsafeTypeExpression . renderSQL
158179typetable
159180 :: (Has sch schemas schema, Has tab schema ('Table table))
160181 => QualifiedAlias sch tab
182+ -- ^ table alias
161183 -> TypeExpression schemas (null ('PGcomposite (TableToRow table)))
162184typetable = UnsafeTypeExpression . renderSQL
163185
@@ -166,6 +188,7 @@ typetable = UnsafeTypeExpression . renderSQL
166188typeview
167189 :: (Has sch schemas schema, Has vw schema ('View view))
168190 => QualifiedAlias sch vw
191+ -- ^ view alias
169192 -> TypeExpression schemas (null ('PGcomposite view))
170193typeview = UnsafeTypeExpression . renderSQL
171194
@@ -339,3 +362,85 @@ instance (SOP.All KnownNat dims, PGTyped schemas ty)
339362instance PGTyped schemas (null 'PGtsvector) where pgtype = tsvector
340363instance PGTyped schemas (null 'PGtsquery) where pgtype = tsquery
341364instance PGTyped schemas (null 'PGoid) where pgtype = oid
365+instance PGTyped schemas (null ('PGrange 'PGint4)) where pgtype = int4range
366+instance PGTyped schemas (null ('PGrange 'PGint8)) where pgtype = int8range
367+instance PGTyped schemas (null ('PGrange 'PGnumeric)) where pgtype = numrange
368+instance PGTyped schemas (null ('PGrange 'PGtimestamp)) where pgtype = tsrange
369+instance PGTyped schemas (null ('PGrange 'PGtimestamptz)) where pgtype = tstzrange
370+instance PGTyped schemas (null ('PGrange 'PGdate)) where pgtype = daterange
371+
372+-- | Lift `PGTyped` to a field
373+class FieldTyped schemas ty where
374+ fieldtype :: Aliased (TypeExpression schemas) ty
375+instance (KnownSymbol alias, PGTyped schemas ty)
376+ => FieldTyped schemas (alias ::: ty) where
377+ fieldtype = pgtype `As` Alias
378+
379+-- | `ColumnTypeExpression`s are used in `createTable` commands.
380+newtype ColumnTypeExpression (schemas :: SchemasType) (ty :: ColumnType)
381+ = UnsafeColumnTypeExpression { renderColumnTypeExpression :: ByteString }
382+ deriving (GHC.Generic,Show,Eq,Ord,NFData)
383+instance RenderSQL (ColumnTypeExpression schemas ty) where
384+ renderSQL = renderColumnTypeExpression
385+
386+-- | used in `createTable` commands as a column constraint to note that
387+-- @NULL@ may be present in a column
388+nullable
389+ :: TypeExpression schemas (nullity ty)
390+ -> ColumnTypeExpression schemas ('NoDef :=> 'Null ty)
391+nullable ty = UnsafeColumnTypeExpression $ renderSQL ty <+> "NULL"
392+
393+-- | used in `createTable` commands as a column constraint to ensure
394+-- @NULL@ is not present in a column
395+notNullable
396+ :: TypeExpression schemas (nullity ty)
397+ -> ColumnTypeExpression schemas ('NoDef :=> 'NotNull ty)
398+notNullable ty = UnsafeColumnTypeExpression $ renderSQL ty <+> "NOT NULL"
399+
400+-- | used in `createTable` commands as a column constraint to give a default
401+default_
402+ :: Expression '[] '[] 'Ungrouped schemas '[] '[] ty
403+ -> ColumnTypeExpression schemas ('NoDef :=> ty)
404+ -> ColumnTypeExpression schemas ('Def :=> ty)
405+default_ x ty = UnsafeColumnTypeExpression $
406+ renderSQL ty <+> "DEFAULT" <+> renderExpression x
407+
408+-- | not a true type, but merely a notational convenience for creating
409+-- unique identifier columns with type `PGint2`
410+serial2, smallserial
411+ :: ColumnTypeExpression schemas ('Def :=> 'NotNull 'PGint2)
412+serial2 = UnsafeColumnTypeExpression "serial2"
413+smallserial = UnsafeColumnTypeExpression "smallserial"
414+-- | not a true type, but merely a notational convenience for creating
415+-- unique identifier columns with type `PGint4`
416+serial4, serial
417+ :: ColumnTypeExpression schemas ('Def :=> 'NotNull 'PGint4)
418+serial4 = UnsafeColumnTypeExpression "serial4"
419+serial = UnsafeColumnTypeExpression "serial"
420+-- | not a true type, but merely a notational convenience for creating
421+-- unique identifier columns with type `PGint8`
422+serial8, bigserial
423+ :: ColumnTypeExpression schemas ('Def :=> 'NotNull 'PGint8)
424+serial8 = UnsafeColumnTypeExpression "serial8"
425+bigserial = UnsafeColumnTypeExpression "bigserial"
426+
427+-- | Like @PGTyped@ but also accounts for nullity.
428+class PGNullityTyped schemas (nullty :: NullityType) where
429+ pgNullityType :: ColumnTypeExpression schemas ('NoDef :=> nullty)
430+
431+instance PGTyped schemas ('Null ty) => PGNullityTyped schemas ('Null ty) where
432+ pgNullityType = nullable (pgtype @_ @('Null ty))
433+
434+instance PGTyped schemas ('NotNull ty) => PGNullityTyped schemas ('NotNull ty) where
435+ pgNullityType = notNullable (pgtype @_ @('NotNull ty))
436+
437+-- | Allow you to specify pg column types in relation to haskell types.
438+-- >>> printSQL $ hask @(Maybe String)
439+-- text NULL
440+--
441+-- >>> printSQL $ hask @Double
442+-- float8 NOT NULL
443+hask
444+ :: forall hask schemas. PGNullityTyped schemas (NullPG hask)
445+ => ColumnTypeExpression schemas ('NoDef :=> NullPG hask)
446+hask = pgNullityType
squeal-postgresql/src/Squeal/PostgreSQL/List.hs+11−2
@@ -84,7 +84,12 @@ infixl 9 *:
8484-- | @Elem@ is a promoted `Data.List.elem`.
8585type family Elem x xs where
8686 Elem x '[] = 'False
87− Elem x (x ': xs) = 'True
87+ Elem x (x ': _) = 'True
88+ Elem x (_ ': x ': _) = 'True
89+ Elem x (_ ': _ ': x ': _) = 'True
90+ Elem x (_ ': _ ': _ ': x ': _) = 'True
91+ Elem x (_ ': _ ': _ ': _ ': x ': _) = 'True
92+ Elem x (_ ': _ ': _ ': _ ': _ ': x ': _) = 'True
8893 Elem x (_ ': xs) = Elem x xs
8994
9095-- | @In x xs@ is a constraint that proves that @x@ is in @xs@.
@@ -99,5 +104,9 @@ Length '[Char,String,Bool,Double] :: Nat
99104= 4
100105-}
101106type family Length (xs :: [k]) :: Nat where
102− Length (x : xs) = 1 + Length xs
107+ Length (_ ': _ ': _ ': _ ': _ : xs) = 5 + Length xs
108+ Length (_ ': _ ': _ ': _ : xs) = 4 + Length xs
109+ Length (_ ': _ ': _ : xs) = 3 + Length xs
110+ Length (_ ': _ : xs) = 2 + Length xs
111+ Length (_ : xs) = 1 + Length xs
103112 Length '[] = 0
squeal-postgresql/src/Squeal/PostgreSQL/Migration.hs+2
@@ -167,6 +167,8 @@ import qualified GHC.Generics as GHC
167167import Squeal.PostgreSQL.Alias
168168import Squeal.PostgreSQL.Binary
169169import Squeal.PostgreSQL.Definition
170+import Squeal.PostgreSQL.Definition.Table
171+import Squeal.PostgreSQL.Definition.Table.Constraint
170172import Squeal.PostgreSQL.Expression.Comparison
171173import Squeal.PostgreSQL.Expression.Parameter
172174import Squeal.PostgreSQL.Expression.Time
squeal-postgresql/src/Squeal/PostgreSQL/PG.hs+22−12
@@ -47,7 +47,7 @@ module Squeal.PostgreSQL.PG
4747 , Enumerated (..)
4848 , VarArray (..)
4949 , FixArray (..)
50− , Oid (..)
50+ , LibPQ.Oid (..)
5151 -- * Type families
5252 , LabelsPG
5353 , DimPG
@@ -66,7 +66,6 @@ import Data.Int (Int16, Int32, Int64)
6666import Data.Scientific (Scientific)
6767import Data.Time (Day, DiffTime, LocalTime, TimeOfDay, TimeZone, UTCTime)
6868import Data.Vector (Vector)
69−import Data.Word (Word32)
7069import Data.UUID.Types (UUID)
7170import GHC.TypeLits
7271import Network.IP.Addr (NetAddr, IP)
@@ -75,6 +74,7 @@ import qualified Data.ByteString.Lazy as Lazy (ByteString)
7574import qualified Data.ByteString as Strict (ByteString)
7675import qualified Data.Text.Lazy as Lazy (Text)
7776import qualified Data.Text as Strict (Text)
77+import qualified Database.PostgreSQL.LibPQ as LibPQ
7878import qualified GHC.Generics as GHC
7979import qualified Generics.SOP as SOP
8080import qualified Generics.SOP.Record as SOP
@@ -108,7 +108,7 @@ type instance PG Int32 = 'PGint4
108108-- | `PGint8`
109109type instance PG Int64 = 'PGint8
110110-- | `PGint2`
111−type instance PG Oid = 'PGoid
111+type instance PG LibPQ.Oid = 'PGoid
112112-- | `PGnumeric`
113113type instance PG Scientific = 'PGnumeric
114114-- | `PGfloat4`
@@ -178,8 +178,16 @@ type family RowPG (hask :: Type) :: RowType where
178178
179179-- | `RowOf` applies `NullPG` to the fields of a list.
180180type family RowOf (record :: [(Symbol, Type)]) :: RowType where
181− RowOf '[] = '[]
181+ RowOf (col ::: ty ': col1 ::: ty1 ': col2 ::: ty2 ': col3 ::: ty3 ': col4 ::: ty4 ': record) =
182+ col ::: NullPG ty ': col1 ::: NullPG ty1 ': col2 ::: NullPG ty2 ': col3 ::: NullPG ty3 ': col4 ::: NullPG ty4 ': RowOf record
183+ RowOf (col ::: ty ': col1 ::: ty1 ': col2 ::: ty2 ': col3 ::: ty3 ': record) =
184+ col ::: NullPG ty ': col1 ::: NullPG ty1 ': col2 ::: NullPG ty2 ': col3 ::: NullPG ty3 ': RowOf record
185+ RowOf (col ::: ty ': col1 ::: ty1 ': col2 ::: ty2 ': record) =
186+ col ::: NullPG ty ': col1 ::: NullPG ty1 ': col2 ::: NullPG ty2 ': RowOf record
187+ RowOf (col ::: ty ': col1 ::: ty1 ': record) =
188+ col ::: NullPG ty ': col1::: NullPG ty1 ': RowOf record
182189 RowOf (col ::: ty ': record) = col ::: NullPG ty ': RowOf record
190+ RowOf '[] = '[]
183191
184192{- | `NullPG` turns a Haskell type into a `NullityType`.
185193
@@ -206,8 +214,17 @@ type family TuplePG (hask :: Type) :: [NullityType] where
206214
207215-- | `TupleOf` turns a list of Haskell `Type`s into a list of `NullityType`s.
208216type family TupleOf (tuple :: [Type]) :: [NullityType] where
209− TupleOf '[] = '[]
217+ TupleOf (hask ': hask1 ': hask2 ': hask3 ': hask4 ': hask5 ': tuple) =
218+ NullPG hask ': NullPG hask1 ': NullPG hask2 ': NullPG hask3 ': NullPG hask4 ': NullPG hask5 ': TupleOf tuple
219+ TupleOf (hask ': hask1 ': hask2 ': hask3 ': hask4 ': tuple) =
220+ NullPG hask ': NullPG hask1 ': NullPG hask2 ': NullPG hask3 ': NullPG hask4 ': TupleOf tuple
221+ TupleOf (hask ': hask1 ': hask2 ': hask3 ': tuple) =
222+ NullPG hask ': NullPG hask1 ': NullPG hask2 ': NullPG hask3 ': TupleOf tuple
223+ TupleOf (hask ': hask1 ': hask2 ': tuple) =
224+ NullPG hask ': NullPG hask1 ': NullPG hask2 ': TupleOf tuple
225+ TupleOf (hask ': hask1 ': tuple) = NullPG hask ': NullPG hask1 ': TupleOf tuple
210226 TupleOf (hask ': tuple) = NullPG hask ': TupleOf tuple
227+ TupleOf '[] = '[]
211228
212229-- | `TupleCodeOf` takes the `SOP.Code` of a haskell `Type`
213230-- and if it's a simple product returns it, otherwise giving a `TypeError`.
@@ -369,10 +386,3 @@ newtype FixArray arr = FixArray {getFixArray :: arr}
369386 deriving anyclass (SOP.HasDatatypeInfo, SOP.Generic)
370387-- | `PGfixarray` @(@`DimPG` @hask) (@`FixPG` @hask)@
371388type instance PG (FixArray hask) = 'PGfixarray (DimPG hask) (FixPG hask)
372−
373−{- | Object identifiers (`Oid`s) are used internally by PostgreSQL
374−as primary keys for various system tables.
375−-}
376−newtype Oid = Oid { getOid :: Word32 }
377− deriving stock (Eq, Ord, Show, Read, GHC.Generic)
378− deriving anyclass (SOP.HasDatatypeInfo, SOP.Generic)
squeal-postgresql/src/Squeal/PostgreSQL/PQ.hs+41−17
@@ -78,6 +78,7 @@ import UnliftIO (MonadUnliftIO (..), bracket, catch, handle, try)
7878import Data.ByteString (ByteString)
7979import Data.Foldable
8080import Data.Function ((&))
81+import Data.Functor ((<&>))
8182import Data.Kind
8283import Data.Text (pack, Text)
8384import Data.Traversable
@@ -86,6 +87,7 @@ import PostgreSQL.Binary.Encoding (encodingBytes)
8687
8788import qualified Control.Monad.Fail as Fail
8889import qualified Database.PostgreSQL.LibPQ as LibPQ
90+import qualified PostgreSQL.Binary.Encoding as Encoding
8991
9092import Squeal.PostgreSQL.Binary
9193import Squeal.PostgreSQL.Definition
@@ -315,13 +317,13 @@ a default instance.
315317-}
316318class Monad pq => MonadPQ schemas pq | pq -> schemas where
317319 manipulateParams
318− :: ToParams x params
320+ :: (ToParams x params, All OidOfParam params)
319321 => Manipulation '[] schemas params ys
320322 -- ^ `insertInto`, `update` or `deleteFrom`
321323 -> x -> pq (K LibPQ.Result ys)
322324 default manipulateParams
323325 :: (MonadTrans t, MonadPQ schemas pq1, pq ~ t pq1)
324− => ToParams x params
326+ => (ToParams x params, All OidOfParam params)
325327 => Manipulation '[] schemas params ys
326328 -- ^ `insertInto`, `update` or `deleteFrom`
327329 -> x -> pq (K LibPQ.Result ys)
@@ -329,7 +331,7 @@ class Monad pq => MonadPQ schemas pq | pq -> schemas where
329331 manipulateParams manipulation params
330332
331333 manipulateParams_
332− :: ToParams x params
334+ :: (ToParams x params, All OidOfParam params)
333335 => Manipulation '[] schemas params '[]
334336 -- ^ `insertInto`, `update` or `deleteFrom`
335337 -> x -> pq ()
@@ -342,7 +344,7 @@ class Monad pq => MonadPQ schemas pq | pq -> schemas where
342344 manipulate_ = void . manipulate
343345
344346 runQueryParams
345− :: ToParams x params
347+ :: (ToParams x params, All OidOfParam params)
346348 => Query '[] '[] schemas params ys
347349 -- ^ `select` and friends
348350 -> x -> pq (K LibPQ.Result ys)
@@ -355,19 +357,19 @@ class Monad pq => MonadPQ schemas pq | pq -> schemas where
355357 runQuery q = runQueryParams q ()
356358
357359 traversePrepared
358− :: (ToParams x params, Traversable list)
360+ :: (ToParams x params, Traversable list, All OidOfParam params)
359361 => Manipulation '[] schemas params ys
360362 -- ^ `insertInto`, `update`, or `deleteFrom`, and friends
361363 -> list x -> pq (list (K LibPQ.Result ys))
362364 default traversePrepared
363365 :: (MonadTrans t, MonadPQ schemas pq1, pq ~ t pq1)
364− => (ToParams x params, Traversable list)
366+ => (ToParams x params, Traversable list, All OidOfParam params)
365367 => Manipulation '[] schemas params ys -> list x -> pq (list (K LibPQ.Result ys))
366368 traversePrepared manipulation params = lift $
367369 traversePrepared manipulation params
368370
369371 forPrepared
370− :: (ToParams x params, Traversable list)
372+ :: (ToParams x params, Traversable list, All OidOfParam params)
371373 => list x
372374 -> Manipulation '[] schemas params ys
373375 -- ^ `insertInto`, `update` or `deleteFrom`
@@ -375,13 +377,13 @@ class Monad pq => MonadPQ schemas pq | pq -> schemas where
375377 forPrepared = flip traversePrepared
376378
377379 traversePrepared_
378− :: (ToParams x params, Foldable list)
380+ :: (ToParams x params, Foldable list, All OidOfParam params)
379381 => Manipulation '[] schemas params '[]
380382 -- ^ `insertInto`, `update` or `deleteFrom`
381383 -> list x -> pq ()
382384 default traversePrepared_
383385 :: (MonadTrans t, MonadPQ schemas pq1, pq ~ t pq1)
384− => (ToParams x params, Foldable list)
386+ => (ToParams x params, Foldable list, All OidOfParam params)
385387 => Manipulation '[] schemas params '[]
386388 -- ^ `insertInto`, `update` or `deleteFrom`
387389 -> list x -> pq ()
@@ -389,7 +391,7 @@ class Monad pq => MonadPQ schemas pq | pq -> schemas where
389391 traversePrepared_ manipulation params
390392
391393 forPrepared_
392− :: (ToParams x params, Foldable list)
394+ :: (ToParams x params, Foldable list, All OidOfParam params)
393395 => list x
394396 -> Manipulation '[] schemas params '[]
395397 -- ^ `insertInto`, `update` or `deleteFrom`
@@ -409,9 +411,17 @@ instance (MonadIO io, schemas0 ~ schemas, schemas1 ~ schemas)
409411 (UnsafeManipulation q :: Manipulation '[] schemas ps ys) (params :: x) =
410412 PQ $ \ (K conn) -> do
411413 let
412− toParam' encoding =
413− (LibPQ.invalidOid, encodingBytes encoding, LibPQ.Binary)
414− params' = fmap (fmap toParam') (hcollapse (toParams @x @ps params))
414+ paramSet
415+ :: forall param. OidOfParam param
416+ => K (Maybe Encoding.Encoding) param
417+ -> K (Maybe (LibPQ.Oid, ByteString, LibPQ.Format)) param
418+ paramSet (K maybeEncoding) = K $
419+ maybeEncoding <&> \encoding ->
420+ (oidOfParam @param, encodingBytes encoding, LibPQ.Binary)
421+ params'
422+ = hcollapse
423+ . hcmap (Proxy @OidOfParam) paramSet
424+ $ toParams @x @ps params
415425 q' = q <> ";"
416426 resultMaybe <- liftIO $ LibPQ.execParams conn q' params' LibPQ.Binary
417427 case resultMaybe of
@@ -424,8 +434,15 @@ instance (MonadIO io, schemas0 ~ schemas, schemas1 ~ schemas)
424434 traversePrepared
425435 (UnsafeManipulation q :: Manipulation '[] schemas xs ys) (list :: list x) =
426436 PQ $ \ (K conn) -> liftIO $ do
427− let temp = "temporary_statement"
428− prepResultMaybe <- LibPQ.prepare conn temp q Nothing
437+ let
438+ temp = "temporary_statement"
439+ paramOid :: forall p. OidOfParam p => K LibPQ.Oid p
440+ paramOid = K (oidOfParam @p)
441+ paramOids :: NP (K LibPQ.Oid) xs
442+ paramOids = hcpure (Proxy @OidOfParam) paramOid
443+ paramOids' :: [LibPQ.Oid]
444+ paramOids' = hcollapse paramOids
445+ prepResultMaybe <- LibPQ.prepare conn temp q (Just paramOids')
429446 case prepResultMaybe of
430447 Nothing -> throw $ ResultException
431448 "traversePrepared: LibPQ.prepare returned no results"
@@ -451,8 +468,15 @@ instance (MonadIO io, schemas0 ~ schemas, schemas1 ~ schemas)
451468 traversePrepared_
452469 (UnsafeManipulation q :: Manipulation '[] schemas xs '[]) (list :: list x) =
453470 PQ $ \ (K conn) -> liftIO $ do
454− let temp = "temporary_statement"
455− prepResultMaybe <- LibPQ.prepare conn temp q Nothing
471+ let
472+ temp = "temporary_statement"
473+ paramOid :: forall p. OidOfParam p => K LibPQ.Oid p
474+ paramOid = K (oidOfParam @p)
475+ paramOids :: NP (K LibPQ.Oid) xs
476+ paramOids = hcpure (Proxy @OidOfParam) paramOid
477+ paramOids' :: [LibPQ.Oid]
478+ paramOids' = hcollapse paramOids
479+ prepResultMaybe <- LibPQ.prepare conn temp q (Just paramOids')
456480 case prepResultMaybe of
457481 Nothing -> throw $ ResultException
458482 "traversePrepared_: LibPQ.prepare returned no results"
squeal-postgresql/src/Squeal/PostgreSQL/Query.hs+33−25
@@ -554,9 +554,9 @@ select_ = select . List
554554-- be subject to the elimination of duplicate rows using `selectDistinct`.
555555selectDistinct
556556 :: (SListI columns, columns ~ (col ': cols))
557− => Selection outer commons 'Ungrouped schemas params from columns
557+ => Selection outer commons grp schemas params from columns
558558 -- ^ selection
559− -> TableExpression outer commons 'Ungrouped schemas params from
559+ -> TableExpression outer commons grp schemas params from
560560 -- ^ intermediate virtual table
561561 -> Query outer commons schemas params columns
562562selectDistinct selection tabexpr = UnsafeQuery $
@@ -568,9 +568,9 @@ selectDistinct selection tabexpr = UnsafeQuery $
568568-- of a general `Selection`.
569569selectDistinct_
570570 :: (SListI columns, columns ~ (col ': cols))
571− => NP (Aliased (Expression outer commons 'Ungrouped schemas params from)) columns
571+ => NP (Aliased (Expression outer commons grp schemas params from)) columns
572572 -- ^ select list
573− -> TableExpression outer commons 'Ungrouped schemas params from
573+ -> TableExpression outer commons grp schemas params from
574574 -- ^ intermediate virtual table
575575 -> Query outer commons schemas params columns
576576selectDistinct_ = selectDistinct . List
@@ -580,14 +580,21 @@ selectDistinct_ = selectDistinct . List
580580the given expressions evaluate to equal. The DISTINCT ON expressions are
581581interpreted using the same rules as for ORDER BY. ORDER BY is used to
582582ensure that the desired row appears first.
583+
584+The DISTINCT ON expression(s) must match the leftmost ORDER BY expression(s).
585+The ORDER BY clause will normally contain additional expression(s) that
586+determine the desired precedence of rows within each DISTINCT ON group.
587+
588+In order to guarantee they match and reduce redundancy, this function
589+will prepend the The DISTINCT ON expressions to the ORDER BY clause.
583590-}
584591selectDistinctOn
585592 :: (SListI columns, columns ~ (col ': cols))
586− => [SortExpression outer commons 'Ungrouped schemas params from]
587− -- ^ distinct on and return the first row in ordering
588− -> Selection outer commons 'Ungrouped schemas params from columns
593+ => [SortExpression outer commons grp schemas params from]
594+ -- ^ DISTINCT ON expression(s) and prepended to ORDER BY clause
595+ -> Selection outer commons grp schemas params from columns
589596 -- ^ selection
590− -> TableExpression outer commons 'Ungrouped schemas params from
597+ -> TableExpression outer commons grp schemas params from
591598 -- ^ intermediate virtual table
592599 -> Query outer commons schemas params columns
593600selectDistinctOn distincts selection tab = UnsafeQuery $
@@ -608,11 +615,11 @@ selectDistinctOn distincts selection tab = UnsafeQuery $
608615-- of a general `Selection`.
609616selectDistinctOn_
610617 :: (SListI columns, columns ~ (col ': cols))
611− => [SortExpression outer commons 'Ungrouped schemas params from]
618+ => [SortExpression outer commons grp schemas params from]
612619 -- ^ distinct on and return the first row in ordering
613− -> NP (Aliased (Expression outer commons 'Ungrouped schemas params from)) columns
620+ -> NP (Aliased (Expression outer commons grp schemas params from)) columns
614621 -- ^ selection
615− -> TableExpression outer commons 'Ungrouped schemas params from
622+ -> TableExpression outer commons grp schemas params from
616623 -- ^ intermediate virtual table
617624 -> Query outer commons schemas params columns
618625selectDistinctOn_ distincts = selectDistinctOn distincts . List
@@ -750,7 +757,7 @@ instance RenderSQL (TableExpression outer commons grp schemas params from) where
750757from
751758 :: FromClause outer commons schemas params from -- ^ table reference
752759 -> TableExpression outer commons 'Ungrouped schemas params from
753−from tab = TableExpression tab [] NoGroups NoHaving [] [] []
760+from tab = TableExpression tab [] noGroups NoHaving [] [] []
754761
755762-- | A `where_` is an endomorphism of `TableExpression`s which adds a
756763-- search condition to the `whereClause`.
@@ -771,7 +778,7 @@ groupBy
771778groupBy bys rels = TableExpression
772779 { fromClause = fromClause rels
773780 , whereClause = whereClause rels
774− , groupByClause = Group bys
781+ , groupByClause = group bys
775782 , havingClause = Having []
776783 , orderByClause = []
777784 , limitClause = limitClause rels
@@ -985,19 +992,20 @@ instance (Has rel rels cols, Has col cols ty, bys ~ '[ '(rel, col)])
985992-- done on @NoGroups@ while all output `Expression`s must be aggregated
986993-- in @Group Nil@. In general, all output `Expression`s in the
987994-- complement of @bys@ must be aggregated in @Group bys@.
988−data GroupByClause grp from where
989− NoGroups :: GroupByClause 'Ungrouped from
990− Group
991− :: SListI bys
992− => NP (By from) bys
993− -> GroupByClause ('Grouped bys) from
994−
995−-- | Renders a `GroupByClause`.
995+newtype GroupByClause grp from = UnsafeGroupByClause
996+ { renderGroupByClause :: ByteString }
997+ deriving (GHC.Generic,Show,Eq,Ord,NFData)
996998instance RenderSQL (GroupByClause grp from) where
997− renderSQL = \case
998− NoGroups -> ""
999− Group Nil -> ""
1000− Group bys -> " GROUP BY" <+> renderCommaSeparated renderSQL bys
999+ renderSQL = renderGroupByClause
1000+noGroups :: GroupByClause 'Ungrouped from
1001+noGroups = UnsafeGroupByClause ""
1002+group
1003+ :: SListI bys
1004+ => NP (By from) bys
1005+ -> GroupByClause ('Grouped bys) from
1006+group bys = UnsafeGroupByClause $ case bys of
1007+ Nil -> ""
1008+ bys' -> " GROUP BY" <+> renderCommaSeparated renderSQL bys'
10011009
10021010-- | A `HavingClause` is used to eliminate groups that are not of interest.
10031011-- An `Ungrouped` `TableExpression` may only use `NoHaving` while a `Grouped`
squeal-postgresql/src/Squeal/PostgreSQL/Schema.hs+69−9
@@ -45,7 +45,7 @@ module Squeal.PostgreSQL.Schema
4545 , TableType
4646 , SchemumType (..)
4747 , IndexType (..)
48− , FunctionType (..)
48+ , FunctionType
4949 , OperatorType (..)
5050 , ReturnsType (..)
5151 , SchemaType
@@ -69,7 +69,9 @@ module Squeal.PostgreSQL.Schema
6969 , DropIfExists
7070 , DropSchemumIfExists
7171 , Alter
72+ , AlterIfExists
7273 , Rename
74+ , RenameIfExists
7375 , ConstraintInvolves
7476 , DropIfConstraintsInvolve
7577 , IsNotElem
@@ -258,9 +260,19 @@ type FromType = [(Symbol,RowType)]
258260
259261-- | `ColumnsToRow` removes column constraints.
260262type family ColumnsToRow (columns :: ColumnsType) :: RowType where
261− ColumnsToRow '[] = '[]
262− ColumnsToRow (column ::: constraint :=> ty ': columns) =
263+ ColumnsToRow (column ::: _ :=> ty ': column1 ::: _ :=> ty1 ': column2 ::: _ :=> ty2 ': column3 ::: _ :=> ty3 ': column4 ::: _ :=> ty4 ': column5 ::: _ :=> ty5 ': columns) =
264+ column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': column3 ::: ty3 ': column4 ::: ty4 ': column5 ::: ty5 ': ColumnsToRow columns
265+ ColumnsToRow (column ::: _ :=> ty ': column1 ::: _ :=> ty1 ': column2 ::: _ :=> ty2 ': column3 ::: _ :=> ty3 ': column4 ::: _ :=> ty4 ': columns) =
266+ column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': column3 ::: ty3 ': column4 ::: ty4 ': ColumnsToRow columns
267+ ColumnsToRow (column ::: _ :=> ty ': column1 ::: _ :=> ty1 ': column2 ::: _ :=> ty2 ': column3 ::: _ :=> ty3 ': columns) =
268+ column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': column3 ::: ty3 ': ColumnsToRow columns
269+ ColumnsToRow (column ::: _ :=> ty ': column1 ::: _ :=> ty1 ': column2 ::: _ :=> ty2 ': columns) =
270+ column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': ColumnsToRow columns
271+ ColumnsToRow (column ::: _ :=> ty ': column1 ::: _ :=> ty1 ': columns) =
272+ column ::: ty ': column1 ::: ty1 ': ColumnsToRow columns
273+ ColumnsToRow (column ::: _ :=> ty ': columns) =
263274 column ::: ty ': ColumnsToRow columns
275+ ColumnsToRow '[] = '[]
264276
265277-- | `TableToColumns` removes table constraints.
266278type family TableToColumns (table :: TableType) :: ColumnsType where
@@ -293,14 +305,24 @@ instance ty0 ~ ty1 => SamePGType
293305
294306-- | `AllNotNull` is a constraint that proves a `ColumnsType` has no @NULL@s.
295307type family AllNotNull (columns :: ColumnsType) :: Constraint where
308+ AllNotNull (_ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': columns) = AllNotNull columns
309+ AllNotNull (_ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': columns) = AllNotNull columns
310+ AllNotNull (_ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': columns) = AllNotNull columns
311+ AllNotNull (_ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': columns) = AllNotNull columns
312+ AllNotNull (_ ::: _ :=> 'NotNull _ ': _ ::: _ :=> 'NotNull _ ': columns) = AllNotNull columns
313+ AllNotNull (_ ::: _ :=> 'NotNull _ ': columns) = AllNotNull columns
296314 AllNotNull '[] = ()
297− AllNotNull (column ::: def :=> 'NotNull ty ': columns) = AllNotNull columns
298315
299316-- | `NotAllNull` is a constraint that proves a `ColumnsType` has some
300317-- @NOT NULL@.
301318type family NotAllNull (columns :: ColumnsType) :: Constraint where
302− NotAllNull (column ::: def :=> 'NotNull ty ': columns) = ()
303− NotAllNull (column ::: def :=> 'Null ty ': columns) = NotAllNull columns
319+ NotAllNull (_ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> 'NotNull _ ': _) = ()
320+ NotAllNull (_ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> 'NotNull _ ': _) = ()
321+ NotAllNull (_ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> 'NotNull _ ': _) = ()
322+ NotAllNull (_ ::: _ :=> _ _ ': _ ::: _ :=> _ _ ': _ ::: _ :=> 'NotNull _ ': _) = ()
323+ NotAllNull (_ ::: _ :=> _ _ ': _ ::: _ :=> 'NotNull _ ': _) = ()
324+ NotAllNull (_ ::: _ :=> 'NotNull _ ': _) = ()
325+ NotAllNull (_ ::: _ :=> 'Null _ ': columns) = NotAllNull columns
304326
305327-- | `NullifyType` is an idempotent that nullifies a `NullityType`.
306328type family NullifyType (ty :: NullityType) :: NullityType where
@@ -309,17 +331,37 @@ type family NullifyType (ty :: NullityType) :: NullityType where
309331
310332-- | `NullifyRow` is an idempotent that nullifies a `RowType`.
311333type family NullifyRow (columns :: RowType) :: RowType where
312− NullifyRow '[] = '[]
334+ NullifyRow (column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': column3 ::: ty3 ': column4 ::: ty4 ': column5 ::: ty5 ': columns) =
335+ column ::: NullifyType ty ': column1 ::: NullifyType ty1 ': column2 ::: NullifyType ty2 ': column3 ::: NullifyType ty3 ': column4 ::: NullifyType ty4 ': column5 ::: NullifyType ty5 ': NullifyRow columns
336+ NullifyRow (column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': column3 ::: ty3 ': column4 ::: ty4 ': columns) =
337+ column ::: NullifyType ty ': column1 ::: NullifyType ty1 ': column2 ::: NullifyType ty2 ': column3 ::: NullifyType ty3 ': column4 ::: NullifyType ty4 ': NullifyRow columns
338+ NullifyRow (column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': column3 ::: ty3 ': columns) =
339+ column ::: NullifyType ty ': column1 ::: NullifyType ty1 ': column2 ::: NullifyType ty2 ': column3 ::: NullifyType ty3 ': NullifyRow columns
340+ NullifyRow (column ::: ty ': column1 ::: ty1 ': column2 ::: ty2 ': columns) =
341+ column ::: NullifyType ty ': column1 ::: NullifyType ty1 ': column2 ::: NullifyType ty2 ': NullifyRow columns
342+ NullifyRow (column ::: ty ': column1 ::: ty1 ': columns) =
343+ column ::: NullifyType ty ': column1 ::: NullifyType ty1 ': NullifyRow columns
313344 NullifyRow (column ::: ty ': columns) =
314345 column ::: NullifyType ty ': NullifyRow columns
346+ NullifyRow '[] = '[]
315347
316348-- | `NullifyFrom` is an idempotent that nullifies a `FromType`
317349-- used to nullify the left or right hand side of an outer join
318350-- in a `Squeal.PostgreSQL.Query.FromClause`.
319351type family NullifyFrom (tables :: FromType) :: FromType where
320− NullifyFrom '[] = '[]
352+ NullifyFrom (table ::: columns ': table1 ::: columns1 ': table2 ::: columns2 ': table3 ::: columns3 ': table4 ::: columns4 ': table5 ::: columns5 ': tables) =
353+ table ::: NullifyRow columns ': table1 ::: NullifyRow columns1 ': table2 ::: NullifyRow columns2 ': table3 ::: NullifyRow columns3 ': table4 ::: NullifyRow columns5 ': table5 ::: NullifyRow columns4 ': NullifyFrom tables
354+ NullifyFrom (table ::: columns ': table1 ::: columns1 ': table2 ::: columns2 ': table3 ::: columns3 ': table4 ::: columns4 ': tables) =
355+ table ::: NullifyRow columns ': table1 ::: NullifyRow columns1 ': table2 ::: NullifyRow columns2 ': table3 ::: NullifyRow columns3 ': table4 ::: NullifyRow columns4 ': NullifyFrom tables
356+ NullifyFrom (table ::: columns ': table1 ::: columns1 ': table2 ::: columns2 ': table3 ::: columns3 ': tables) =
357+ table ::: NullifyRow columns ': table1 ::: NullifyRow columns1 ': table2 ::: NullifyRow columns2 ': table3 ::: NullifyRow columns3 ': NullifyFrom tables
358+ NullifyFrom (table ::: columns ': table1 ::: columns1 ': table2 ::: columns2 ': tables) =
359+ table ::: NullifyRow columns ': table1 ::: NullifyRow columns1 ': table2 ::: NullifyRow columns2 ': NullifyFrom tables
360+ NullifyFrom (table ::: columns ': table1 ::: columns1 ': tables) =
361+ table ::: NullifyRow columns ': table1 ::: NullifyRow columns1 ': NullifyFrom tables
321362 NullifyFrom (table ::: columns ': tables) =
322363 table ::: NullifyRow columns ': NullifyFrom tables
364+ NullifyFrom '[] = '[]
323365
324366-- | @Create alias x xs@ adds @alias ::: x@ to the end of @xs@ and is used in
325367-- `Squeal.PostgreSQL.Definition.createTable` statements and in @ALTER TABLE@
@@ -396,16 +438,34 @@ type family DropSchemumIfExists alias sch xs where
396438-- with the type @x@ and is used in `Squeal.PostgreSQL.Definition.alterTable`
397439-- and `Squeal.PostgreSQL.Definition.alterColumn`.
398440type family Alter alias x xs where
441+ Alter alias x '[] = TypeError
442+ ('Text "Alter: alias "
443+ ':<>: 'ShowType alias
444+ ':<>: 'Text " does not exist" )
399445 Alter alias x1 (alias ::: x0 ': xs) = alias ::: x1 ': xs
400446 Alter alias x1 (x0 ': xs) = x0 ': Alter alias x1 xs
401447
448+type family AlterIfExists alias x xs where
449+ AlterIfExists alias x '[] = '[]
450+ AlterIfExists alias x1 (alias ::: x0 ': xs) = alias ::: x1 ': xs
451+ AlterIfExists alias x1 (x0 ': xs) = x0 ': AlterIfExists alias x1 xs
452+
402453-- | @Rename alias0 alias1 xs@ replaces the alias @alias0@ by @alias1@ in @xs@
403454-- and is used in `Squeal.PostgreSQL.Definition.alterTableRename` and
404455-- `Squeal.PostgreSQL.Definition.renameColumn`.
405456type family Rename alias0 alias1 xs where
457+ Rename alias0 alias1 '[] = TypeError
458+ ('Text "Rename: alias "
459+ ':<>: 'ShowType alias0
460+ ':<>: 'Text " does not exist" )
406461 Rename alias0 alias1 ((alias0 ::: x0) ': xs) = (alias1 ::: x0) ': xs
407462 Rename alias0 alias1 (x ': xs) = x ': Rename alias0 alias1 xs
408463
464+type family RenameIfExists alias0 alias1 xs where
465+ RenameIfExists alias x '[] = '[]
466+ RenameIfExists alias0 alias1 ((alias0 ::: x0) ': xs) = (alias1 ::: x0) ': xs
467+ RenameIfExists alias0 alias1 (x ': xs) = x ': RenameIfExists alias0 alias1 xs
468+
409469-- | Check if a `TableConstraint` involves a column
410470type family ConstraintInvolves column constraint where
411471 ConstraintInvolves column ('Check columns) = column `Elem` columns
@@ -433,7 +493,7 @@ data SchemumType
433493 | Operator OperatorType
434494 | UnsafeSchemum Symbol
435495
436−data FunctionType = (::-->) [NullityType] ReturnsType
496+type FunctionType = ([NullityType], ReturnsType)
437497
438498data IndexType
439499 = Btree
squeal-postgresql/test/Property.hs+2−2
@@ -92,14 +92,14 @@ roundtrips = Group "roundtrips"
9292 -- genTimeWithZone = (,) <$> genTimeOfDay <*> genTimeZone
9393
9494roundtrip
95− :: (ToParam x ty, FromValue ty x, Show x, Eq x)
95+ :: (OidOf ty, ToParam x ty, FromValue ty x, Show x, Eq x)
9696 => TypeExpression schemas ('NotNull ty)
9797 -> Gen x
9898 -> (PropertyName, Property)
9999roundtrip = roundtripOn id
100100
101101roundtripOn
102− :: (ToParam x ty, FromValue ty x, Show x, Eq x)
102+ :: (OidOf ty, ToParam x ty, FromValue ty x, Show x, Eq x)
103103 => (x -> x)
104104 -> TypeExpression schemas ('NotNull ty)
105105 -> Gen x
squeal-postgresql/test/Spec.hs+11
@@ -123,3 +123,14 @@ spec = before_ setupDB . after_ dropDB $ do
123123 getRows =<< runQuery query
124124 (fromOnly <$> rangesOut :: [Range Int32]) `shouldBe`
125125 [ atLeast 3, 3 <=..< 5, Empty, whole ]
126+
127+ describe "Parameters" $ do
128+
129+ it "should run queries that don't reference all their parameters" $ do
130+
131+ out <- withConnection connectionString $ do
132+ let
133+ query :: Query_ (Public '[]) (Char,Int32) (Only Int32)
134+ query = values_ (param @2 `as` #fromOnly)
135+ firstRow =<< runQueryParams query ('a', 3 :: Int32)
136+ (fromOnly <$> out :: Maybe Int32) `shouldBe` Just 3