{-|
Module: Squeal.PostgreSQL.Expression.Composite
Description: composite functions
Copyright: (c) Eitan Chatav, 2019
Maintainer: eitan@morphism.tech
Stability: experimental

composite functions
-}

{-# LANGUAGE
    AllowAmbiguousTypes
  , DataKinds
  , FlexibleContexts
  , FlexibleInstances
  , MultiParamTypeClasses
  , OverloadedLabels
  , OverloadedStrings
  , RankNTypes
  , ScopedTypeVariables
  , TypeApplications
  , TypeFamilies
  , TypeOperators
  , UndecidableInstances
#-}

module Squeal.PostgreSQL.Expression.Composite
  ( -- * Composite Functions
    row
  , rowStar
  , field
  ) where

import qualified Generics.SOP as SOP

import Squeal.PostgreSQL.Type.Alias
import Squeal.PostgreSQL.Expression
import Squeal.PostgreSQL.Type.List
import Squeal.PostgreSQL.Render
import Squeal.PostgreSQL.Type.Schema

-- $setup
-- >>> import Squeal.PostgreSQL

-- | A row constructor is an expression that builds a row value
-- (also called a composite value) using values for its member fields.
--
-- >>> :{
-- type Complex = 'PGcomposite
--   '[ "real"      ::: 'NotNull 'PGfloat8
--    , "imaginary" ::: 'NotNull 'PGfloat8 ]
-- :}
--
-- >>> let i = row (0 `as` #real :* 1 `as` #imaginary) :: Expression grp lat with db params from ('NotNull Complex)
-- >>> printSQL i
-- ROW((0.0 :: float8), (1.0 :: float8))
row
  :: SOP.SListI row
  => NP (Aliased (Expression grp lat with db params from)) row
  -- ^ zero or more expressions for the row field values
  -> Expression grp lat with db params from (null ('PGcomposite row))
row :: forall (row :: [(Symbol, NullType)]) (grp :: Grouping)
       (lat :: FromType) (with :: FromType) (db :: SchemasType)
       (params :: [NullType]) (from :: FromType)
       (null :: PGType -> NullType).
SListI row =>
NP (Aliased (Expression grp lat with db params from)) row
-> Expression grp lat with db params from (null ('PGcomposite row))
row NP (Aliased (Expression grp lat with db params from)) row
exprs = forall (grp :: Grouping) (lat :: FromType) (with :: FromType)
       (db :: SchemasType) (params :: [NullType]) (from :: FromType)
       (ty :: NullType).
ByteString -> Expression grp lat with db params from ty
UnsafeExpression forall a b. (a -> b) -> a -> b
$ ByteString
"ROW" forall a. Semigroup a => a -> a -> a
<> ByteString -> ByteString
parenthesized
  (forall {k} (xs :: [k]) (expression :: k -> *).
SListI xs =>
(forall (x :: k). expression x -> ByteString)
-> NP expression xs -> ByteString
renderCommaSeparated (\ (Expression grp lat with db params from ty
expr `As` Alias alias
_) -> forall sql. RenderSQL sql => sql -> ByteString
renderSQL Expression grp lat with db params from ty
expr) NP (Aliased (Expression grp lat with db params from)) row
exprs)

-- | A row constructor on all columns in a table expression.
rowStar
  :: Has tab from row
  => Alias tab -- ^ intermediate table
  -> Expression grp lat with db params from (null ('PGcomposite row))
rowStar :: forall (tab :: Symbol) (from :: FromType)
       (row :: [(Symbol, NullType)]) (grp :: Grouping) (lat :: FromType)
       (with :: FromType) (db :: SchemasType) (params :: [NullType])
       (null :: PGType -> NullType).
Has tab from row =>
Alias tab
-> Expression grp lat with db params from (null ('PGcomposite row))
rowStar Alias tab
tab = forall (grp :: Grouping) (lat :: FromType) (with :: FromType)
       (db :: SchemasType) (params :: [NullType]) (from :: FromType)
       (ty :: NullType).
ByteString -> Expression grp lat with db params from ty
UnsafeExpression forall a b. (a -> b) -> a -> b
$ ByteString
"ROW" forall a. Semigroup a => a -> a -> a
<>
  ByteString -> ByteString
parenthesized (forall sql. RenderSQL sql => sql -> ByteString
renderSQL Alias tab
tab forall a. Semigroup a => a -> a -> a
<> ByteString
".*")

-- | >>> :{
-- type Complex = 'PGcomposite
--   '[ "real"      ::: 'NotNull 'PGfloat8
--    , "imaginary" ::: 'NotNull 'PGfloat8 ]
-- type Schema = '["complex" ::: 'Typedef Complex]
-- :}
--
-- >>> let i = row (0 `as` #real :* 1 `as` #imaginary) :: Expression lat '[] grp (Public Schema) from params ('NotNull Complex)
-- >>> printSQL $ i & field #complex #imaginary
-- (ROW((0.0 :: float8), (1.0 :: float8))::"complex")."imaginary"
field
  :: ( relss ~ DbRelations db
     , Has sch relss rels
     , Has rel rels row
     , Has field row ty
     )
  => QualifiedAlias sch rel -- ^ row type
  -> Alias field -- ^ field name
  -> Expression grp lat with db params from ('NotNull ('PGcomposite row))
  -> Expression grp lat with db params from ty
field :: forall (relss :: [(Symbol, FromType)]) (db :: SchemasType)
       (sch :: Symbol) (rels :: FromType) (rel :: Symbol)
       (row :: [(Symbol, NullType)]) (field :: Symbol) (ty :: NullType)
       (grp :: Grouping) (lat :: FromType) (with :: FromType)
       (params :: [NullType]) (from :: FromType).
(relss ~ DbRelations db, Has sch relss rels, Has rel rels row,
 Has field row ty) =>
QualifiedAlias sch rel
-> Alias field
-> Expression
     grp lat with db params from ('NotNull ('PGcomposite row))
-> Expression grp lat with db params from ty
field QualifiedAlias sch rel
rel Alias field
fld Expression
  grp lat with db params from ('NotNull ('PGcomposite row))
expr = forall (grp :: Grouping) (lat :: FromType) (with :: FromType)
       (db :: SchemasType) (params :: [NullType]) (from :: FromType)
       (ty :: NullType).
ByteString -> Expression grp lat with db params from ty
UnsafeExpression forall a b. (a -> b) -> a -> b
$
  ByteString -> ByteString
parenthesized (forall sql. RenderSQL sql => sql -> ByteString
renderSQL Expression
  grp lat with db params from ('NotNull ('PGcomposite row))
expr forall a. Semigroup a => a -> a -> a
<> ByteString
"::" forall a. Semigroup a => a -> a -> a
<> forall sql. RenderSQL sql => sql -> ByteString
renderSQL QualifiedAlias sch rel
rel)
    forall a. Semigroup a => a -> a -> a
<> ByteString
"." forall a. Semigroup a => a -> a -> a
<> forall sql. RenderSQL sql => sql -> ByteString
renderSQL Alias field
fld