graphql-engine/server/src-lib/Hasura/Backends/Postgres/DDL.hs
Rakesh Emmadi 9ef603360c server: generalize schema cache building (#496)
Co-authored-by: Vamshi Surabhi <vamshi@hasura.io>
Co-authored-by: Vladimir Ciobanu <admin@cvlad.info>
Co-authored-by: Antoine Leblanc <antoine@hasura.io>
Co-authored-by: Stylish Haskell Bot <stylish-haskell@users.noreply.github.com>
GitOrigin-RevId: 9d631878037637f3ed2994b5d0525efd978f7b8f
2021-02-14 06:08:46 +00:00

56 lines
2.0 KiB
Haskell

module Hasura.Backends.Postgres.DDL
( parseCollectableType
, module M
)
where
import qualified Data.Text as T
import Data.Aeson
import Hasura.Backends.Postgres.SQL.DML
import Hasura.Backends.Postgres.Translate.Column
import Hasura.Prelude
import Hasura.RQL.IR.BoolExp
import Hasura.RQL.Types.Column
import Hasura.RQL.Types.Error
import Hasura.SQL.Backend
import Hasura.SQL.Types
import Hasura.Server.Utils
import Hasura.Session
import Hasura.Backends.Postgres.DDL.BoolExp as M
import Hasura.Backends.Postgres.DDL.Field as M
import Hasura.Backends.Postgres.DDL.Function as M
import Hasura.Backends.Postgres.DDL.Source as M
import Hasura.Backends.Postgres.DDL.Table as M
parseCollectableType
:: (MonadError QErr m)
=> CollectableType (ColumnType 'Postgres)
-> Value -> m (PartialSQLExp 'Postgres)
parseCollectableType pgType = \case
-- When it is a special variable
String t
| isSessionVariable t -> return $ mkTypedSessionVar pgType $ mkSessionVariable t
| isReqUserId t -> return $ mkTypedSessionVar pgType userIdHeader
-- Typical value as Aeson's value
val -> case pgType of
CollectableTypeScalar cvType ->
PSESQLExp . toTxtValue . ColumnValue cvType <$> parseScalarValueColumnType cvType val
CollectableTypeArray ofType -> do
vals <- runAesonParser parseJSON val
scalarValues <- parseScalarValuesColumnType ofType vals
return . PSESQLExp $ SETyAnn
(SEArray $ map (toTxtValue . ColumnValue ofType) scalarValues)
(mkTypeAnn $ CollectableTypeArray (unsafePGColumnToBackend ofType))
mkTypedSessionVar
:: CollectableType (ColumnType 'Postgres)
-> SessionVariable -> PartialSQLExp 'Postgres
mkTypedSessionVar columnType =
PSESessVar (unsafePGColumnToBackend <$> columnType)
isReqUserId :: Text -> Bool
isReqUserId = (== "req_user_id") . T.toLower