mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 12:31:52 +03:00
146 lines
5.3 KiB
Haskell
146 lines
5.3 KiB
Haskell
{-# LANGUAGE StrictData #-}
|
|
|
|
module Hasura.GraphQL.Context
|
|
( RoleContext(..)
|
|
, GQLContext(..)
|
|
, ParserFn
|
|
, SourceConfigWith(..)
|
|
, RootField(..)
|
|
, QueryDB(..)
|
|
, MutationDB(..)
|
|
, ActionQuery(..)
|
|
, ActionMutation(..)
|
|
, RemoteFieldG (..)
|
|
, RemoteField
|
|
, QueryRootField
|
|
, MutationRootField
|
|
, SubscriptionRootField
|
|
, QueryDBRoot(..)
|
|
, MutationDBRoot(..)
|
|
, traverseQueryDB
|
|
, traverseActionQuery
|
|
) where
|
|
|
|
import Hasura.Prelude
|
|
|
|
import qualified Data.Aeson as J
|
|
import qualified Data.Kind as T
|
|
import qualified Language.GraphQL.Draft.Syntax as G
|
|
|
|
import Data.Aeson.TH
|
|
|
|
import qualified Hasura.RQL.IR.Delete as IR
|
|
import qualified Hasura.RQL.IR.Insert as IR
|
|
import qualified Hasura.RQL.IR.Select as IR
|
|
import qualified Hasura.RQL.IR.Update as IR
|
|
import qualified Hasura.RQL.Types.Action as RQL
|
|
import qualified Hasura.RQL.Types.Backend as RQL
|
|
import qualified Hasura.RQL.Types.Common as RQL
|
|
import qualified Hasura.RQL.Types.RemoteSchema as RQL
|
|
import qualified Hasura.SQL.AnyBackend as AB
|
|
|
|
import Hasura.GraphQL.Parser
|
|
import Hasura.SQL.Backend
|
|
|
|
|
|
-- | For storing both a normal GQLContext and one for the backend variant.
|
|
-- Currently, this is to enable the backend variant to have certain insert
|
|
-- permissions which the frontend variant does not.
|
|
|
|
data RoleContext a
|
|
= RoleContext
|
|
{ _rctxDefault :: !a -- ^ The default context for normal sessions
|
|
, _rctxBackend :: !(Maybe a) -- ^ The context for sessions with backend privilege.
|
|
} deriving (Show, Eq, Functor, Foldable, Traversable)
|
|
$(deriveToJSON hasuraJSON ''RoleContext)
|
|
|
|
data GQLContext = GQLContext
|
|
{ gqlQueryParser :: ParserFn (InsOrdHashMap G.Name (QueryRootField UnpreparedValue))
|
|
, gqlMutationParser :: Maybe (ParserFn (InsOrdHashMap G.Name (MutationRootField UnpreparedValue)))
|
|
}
|
|
|
|
instance J.ToJSON GQLContext where
|
|
toJSON GQLContext{} = J.String "The GraphQL schema parsers"
|
|
|
|
type ParserFn a
|
|
= G.SelectionSet G.NoFragments Variable
|
|
-> Either (NESeq ParseError) (a, QueryReusability)
|
|
|
|
data SourceConfigWith (db :: BackendType -> T.Type) (b :: BackendType) =
|
|
SourceConfigWith (RQL.SourceConfig b) (db b)
|
|
|
|
data RootField (db :: BackendType -> T.Type) remote action raw where
|
|
RFDB
|
|
:: RQL.SourceName
|
|
-> AB.AnyBackend (SourceConfigWith db)
|
|
-> RootField db remote action raw
|
|
RFRemote :: remote -> RootField db remote action raw
|
|
RFAction :: action -> RootField db remote action raw
|
|
RFRaw :: raw -> RootField db remote action raw
|
|
|
|
data QueryDB (b :: BackendType) v
|
|
= QDBMultipleRows (IR.AnnSimpleSelG b v)
|
|
| QDBSingleRow (IR.AnnSimpleSelG b v)
|
|
| QDBAggregation (IR.AnnAggregateSelectG b v)
|
|
| QDBConnection (IR.ConnectionSelect b v)
|
|
|
|
data MutationDB (b :: BackendType) v
|
|
= MDBInsert (IR.AnnInsert b v)
|
|
| MDBUpdate (IR.AnnUpdG b v)
|
|
| MDBDelete (IR.AnnDelG b v)
|
|
| MDBFunction RQL.JsonAggSelect (IR.AnnSimpleSelG b v)
|
|
-- ^ This represents a VOLATILE function, and is AnnSimpleSelG for easy
|
|
-- re-use of non-VOLATILE function tracking code.
|
|
|
|
data ActionQuery (b :: BackendType) v
|
|
= AQQuery !(RQL.AnnActionExecution b v)
|
|
| AQAsync !(RQL.AnnActionAsyncQuery b v)
|
|
|
|
data ActionMutation (b :: BackendType) v
|
|
= AMSync !(RQL.AnnActionExecution b v)
|
|
| AMAsync !RQL.AnnActionMutationAsync
|
|
|
|
data RemoteFieldG var
|
|
= RemoteFieldG
|
|
{ _rfRemoteSchemaInfo :: !RQL.RemoteSchemaInfo
|
|
, _rfField :: !(G.Field G.NoFragments var)
|
|
} deriving (Functor, Foldable, Traversable)
|
|
|
|
type RemoteField = RemoteFieldG RQL.RemoteSchemaVariable
|
|
|
|
-- The `db` type argument of @RootField@ expects only one type argument, the backend `b`, as not all
|
|
-- types stored in a RootField will have a second parameter like @QueryDB@ does: they all only have
|
|
-- in common the fact that they're parametric over the backend. To define @QueryRootField@ in terms
|
|
-- of @QueryDB@ (and likewise for mutations), we need a type-level function `b -> QueryDB b (v
|
|
-- b)`. Sadly, neither type synonyms nor type families may be partially applied. Hence the need for
|
|
-- @QueryDBRoot@ and @MutationDBRoot@.
|
|
newtype QueryDBRoot v b = QDBR (QueryDB b (v b))
|
|
newtype MutationDBRoot v b = MDBR (MutationDB b (v b))
|
|
|
|
|
|
type QueryRootField v = RootField (QueryDBRoot v) RemoteField (ActionQuery 'Postgres (v 'Postgres)) J.Value
|
|
type MutationRootField v = RootField (MutationDBRoot v) RemoteField (ActionMutation 'Postgres (v 'Postgres)) J.Value
|
|
type SubscriptionRootField v = RootField (QueryDBRoot v) Void Void Void
|
|
|
|
|
|
traverseQueryDB
|
|
:: forall f a b backend
|
|
. (Applicative f, RQL.Backend backend)
|
|
=> (a -> f b)
|
|
-> QueryDB backend a
|
|
-> f (QueryDB backend b)
|
|
traverseQueryDB f = \case
|
|
QDBMultipleRows s -> QDBMultipleRows <$> IR.traverseAnnSimpleSelect f s
|
|
QDBSingleRow s -> QDBSingleRow <$> IR.traverseAnnSimpleSelect f s
|
|
QDBAggregation s -> QDBAggregation <$> IR.traverseAnnAggregateSelect f s
|
|
QDBConnection s -> QDBConnection <$> IR.traverseConnectionSelect f s
|
|
|
|
traverseActionQuery
|
|
:: (Applicative f, RQL.Backend backend)
|
|
=> (a -> f b)
|
|
-> ActionQuery backend a
|
|
-> f (ActionQuery backend b)
|
|
traverseActionQuery f = \case
|
|
AQQuery actionExecution -> AQQuery <$> RQL.traverseAnnActionExecution f actionExecution
|
|
AQAsync actionAsyncQ -> AQAsync <$> RQL.traverseAnnActionAsyncQuery f actionAsyncQ
|