{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE NamedFieldPuns #-}
{-# LANGUAGE PolyKinds #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE UndecidableInstances #-}
{-# LANGUAGE NoImplicitPrelude #-}
module Data.Morpheus.Server.Deriving.Encode
( deriveModel,
EncodeConstraints,
ContextValue (..),
)
where
import Control.Monad.Except (MonadError)
import qualified Data.Map as M
import Data.Morpheus.App.Internal.Resolving
( LiftOperation,
ObjectTypeResolver,
Resolver,
ResolverState,
ResolverValue,
ResolverValue (..),
RootResolverValue (..),
getArguments,
liftResolverState,
mkEnum,
mkObject,
mkUnion,
requireObject,
)
import Data.Morpheus.Internal.Ext (GQLResult)
import Data.Morpheus.Kind
( CUSTOM,
DerivingKind,
SCALAR,
TYPE,
WRAPPER,
)
import Data.Morpheus.Server.Deriving.Channels
( ChannelsConstraint,
channelResolver,
)
import Data.Morpheus.Server.Deriving.Decode
( DecodeConstraint,
decodeArguments,
)
import Data.Morpheus.Server.Deriving.Utils
( ConsRep (..),
DataType (..),
FieldRep (..),
TypeConstraint (..),
TypeRep (..),
isUnionRef,
toFieldRes,
toValue,
)
import Data.Morpheus.Server.Types.GQLType
( GQLType,
KIND,
__isEmptyType,
)
import Data.Morpheus.Server.Types.Types
( TypeGuard (..),
)
import Data.Morpheus.Types
( RootResolver (..),
)
import Data.Morpheus.Types.GQLScalar
( EncodeScalar (..),
)
import Data.Morpheus.Types.GQLWrapper (EncodeWrapper (..))
import Data.Morpheus.Types.Internal.AST
( GQLError,
IN,
MUTATION,
OperationType,
QUERY,
SUBSCRIPTION,
TypeRef (..),
)
import GHC.Generics
( Generic (..),
)
import Relude
newtype ContextValue (kind :: DerivingKind) a = ContextValue
{ ContextValue kind a -> a
unContextValue :: a
}
class Encode (m :: Type -> Type) resolver where
encode :: resolver -> m (ResolverValue m)
instance (EncodeKind (KIND a) m a) => Encode m a where
encode :: a -> m (ResolverValue m)
encode a
resolver = ContextValue (KIND a) a -> m (ResolverValue m)
forall (kind :: DerivingKind) (m :: * -> *) a.
EncodeKind kind m a =>
ContextValue kind a -> m (ResolverValue m)
encodeKind (a -> ContextValue (KIND a) a
forall (kind :: DerivingKind) a. a -> ContextValue kind a
ContextValue a
resolver :: ContextValue (KIND a) a)
class EncodeKind (kind :: DerivingKind) (m :: Type -> Type) (a :: Type) where
encodeKind :: ContextValue kind a -> m (ResolverValue m)
instance
( EncodeWrapper f,
Encode m a,
Monad m
) =>
EncodeKind WRAPPER m (f a)
where
encodeKind :: ContextValue WRAPPER (f a) -> m (ResolverValue m)
encodeKind = (a -> m (ResolverValue m)) -> f a -> m (ResolverValue m)
forall (wrapper :: * -> *) (m :: * -> *) a.
(EncodeWrapper wrapper, Monad m) =>
(a -> m (ResolverValue m)) -> wrapper a -> m (ResolverValue m)
encodeWrapper a -> m (ResolverValue m)
forall (m :: * -> *) resolver.
Encode m resolver =>
resolver -> m (ResolverValue m)
encode (f a -> m (ResolverValue m))
-> (ContextValue WRAPPER (f a) -> f a)
-> ContextValue WRAPPER (f a)
-> m (ResolverValue m)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ContextValue WRAPPER (f a) -> f a
forall (kind :: DerivingKind) a. ContextValue kind a -> a
unContextValue
instance
( EncodeScalar a,
Monad m
) =>
EncodeKind SCALAR m a
where
encodeKind :: ContextValue SCALAR a -> m (ResolverValue m)
encodeKind = ResolverValue m -> m (ResolverValue m)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (ResolverValue m -> m (ResolverValue m))
-> (ContextValue SCALAR a -> ResolverValue m)
-> ContextValue SCALAR a
-> m (ResolverValue m)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ScalarValue -> ResolverValue m
forall (m :: * -> *). ScalarValue -> ResolverValue m
ResScalar (ScalarValue -> ResolverValue m)
-> (ContextValue SCALAR a -> ScalarValue)
-> ContextValue SCALAR a
-> ResolverValue m
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> ScalarValue
forall a. EncodeScalar a => a -> ScalarValue
encodeScalar (a -> ScalarValue)
-> (ContextValue SCALAR a -> a)
-> ContextValue SCALAR a
-> ScalarValue
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ContextValue SCALAR a -> a
forall (kind :: DerivingKind) a. ContextValue kind a -> a
unContextValue
instance
( EncodeConstraint m a,
MonadError GQLError m
) =>
EncodeKind TYPE m a
where
encodeKind :: ContextValue TYPE a -> m (ResolverValue m)
encodeKind = ResolverValue m -> m (ResolverValue m)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (ResolverValue m -> m (ResolverValue m))
-> (ContextValue TYPE a -> ResolverValue m)
-> ContextValue TYPE a
-> m (ResolverValue m)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> ResolverValue m
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverValue m
exploreResolvers (a -> ResolverValue m)
-> (ContextValue TYPE a -> a)
-> ContextValue TYPE a
-> ResolverValue m
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ContextValue TYPE a -> a
forall (kind :: DerivingKind) a. ContextValue kind a -> a
unContextValue
instance (Monad m, Encode m [(k, v)]) => EncodeKind CUSTOM m (Map k v) where
encodeKind :: ContextValue CUSTOM (Map k v) -> m (ResolverValue m)
encodeKind = [(k, v)] -> m (ResolverValue m)
forall (m :: * -> *) resolver.
Encode m resolver =>
resolver -> m (ResolverValue m)
encode ([(k, v)] -> m (ResolverValue m))
-> (ContextValue CUSTOM (Map k v) -> [(k, v)])
-> ContextValue CUSTOM (Map k v)
-> m (ResolverValue m)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Map k v -> [(k, v)]
forall k a. Map k a -> [(k, a)]
M.toList (Map k v -> [(k, v)])
-> (ContextValue CUSTOM (Map k v) -> Map k v)
-> ContextValue CUSTOM (Map k v)
-> [(k, v)]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ContextValue CUSTOM (Map k v) -> Map k v
forall (kind :: DerivingKind) a. ContextValue kind a -> a
unContextValue
instance (MonadError GQLError m, EncodeConstraint m guard, EncodeConstraint m union) => EncodeKind CUSTOM m (TypeGuard guard union) where
encodeKind :: ContextValue CUSTOM (TypeGuard guard union) -> m (ResolverValue m)
encodeKind (ContextValue (ResolveType union
value)) = ResolverValue m -> m (ResolverValue m)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (union -> ResolverValue m
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverValue m
exploreResolvers union
value)
encodeKind (ContextValue (ResolveInterface guard
value)) = ResolverValue m -> m (ResolverValue m)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (guard -> ResolverValue m
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverValue m
exploreResolvers guard
value)
instance
( DecodeConstraint a,
Generic a,
Monad m,
Encode (Resolver o e m) b,
LiftOperation o
) =>
EncodeKind CUSTOM (Resolver o e m) (a -> b)
where
encodeKind :: ContextValue CUSTOM (a -> b)
-> Resolver o e m (ResolverValue (Resolver o e m))
encodeKind (ContextValue a -> b
f) =
Resolver o e m (Arguments VALID)
forall (o :: OperationType) (m :: * -> *) e.
(LiftOperation o, Monad m) =>
Resolver o e m (Arguments VALID)
getArguments
Resolver o e m (Arguments VALID)
-> (Arguments VALID -> Resolver o e m a) -> Resolver o e m a
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= ResolverState a -> Resolver o e m a
forall (o :: OperationType) (m :: * -> *) a e.
(LiftOperation o, Monad m) =>
ResolverState a -> Resolver o e m a
liftResolverState (ResolverState a -> Resolver o e m a)
-> (Arguments VALID -> ResolverState a)
-> Arguments VALID
-> Resolver o e m a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Arguments VALID -> ResolverState a
forall a. DecodeConstraint a => Arguments VALID -> ResolverState a
decodeArguments
Resolver o e m a
-> (a -> Resolver o e m (ResolverValue (Resolver o e m)))
-> Resolver o e m (ResolverValue (Resolver o e m))
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= b -> Resolver o e m (ResolverValue (Resolver o e m))
forall (m :: * -> *) resolver.
Encode m resolver =>
resolver -> m (ResolverValue m)
encode (b -> Resolver o e m (ResolverValue (Resolver o e m)))
-> (a -> b) -> a -> Resolver o e m (ResolverValue (Resolver o e m))
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> b
f
instance
(Monad m, Encode (Resolver o e m) b, LiftOperation o) =>
EncodeKind CUSTOM (Resolver o e m) (Resolver o e m b)
where
encodeKind :: ContextValue CUSTOM (Resolver o e m b)
-> Resolver o e m (ResolverValue (Resolver o e m))
encodeKind (ContextValue Resolver o e m b
value) = Resolver o e m b
value Resolver o e m b
-> (b -> Resolver o e m (ResolverValue (Resolver o e m)))
-> Resolver o e m (ResolverValue (Resolver o e m))
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= b -> Resolver o e m (ResolverValue (Resolver o e m))
forall (m :: * -> *) resolver.
Encode m resolver =>
resolver -> m (ResolverValue m)
encode
convertNode ::
forall m.
MonadError GQLError m =>
DataType (m (ResolverValue m)) ->
ResolverValue m
convertNode :: DataType (m (ResolverValue m)) -> ResolverValue m
convertNode
DataType
{ TypeName
dataTypeName :: forall v. DataType v -> TypeName
dataTypeName :: TypeName
dataTypeName,
Bool
tyIsUnion :: forall v. DataType v -> Bool
tyIsUnion :: Bool
tyIsUnion,
tyCons :: forall v. DataType v -> ConsRep v
tyCons = cons :: ConsRep (m (ResolverValue m))
cons@ConsRep {[FieldRep (m (ResolverValue m))]
consFields :: forall v. ConsRep v -> [FieldRep v]
consFields :: [FieldRep (m (ResolverValue m))]
consFields, TypeName
consName :: forall v. ConsRep v -> TypeName
consName :: TypeName
consName}
}
| Bool
tyIsUnion = [FieldRep (m (ResolverValue m))] -> ResolverValue m
encodeUnion [FieldRep (m (ResolverValue m))]
consFields
| Bool
otherwise = TypeName -> [ResolverEntry m] -> ResolverValue m
forall (m :: * -> *).
TypeName -> [ResolverEntry m] -> ResolverValue m
mkObject TypeName
dataTypeName (FieldRep (m (ResolverValue m)) -> ResolverEntry m
forall k (m :: k -> *) (a :: k). FieldRep (m a) -> (FieldName, m a)
toFieldRes (FieldRep (m (ResolverValue m)) -> ResolverEntry m)
-> [FieldRep (m (ResolverValue m))] -> [ResolverEntry m]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [FieldRep (m (ResolverValue m))]
consFields)
where
encodeUnion ::
[FieldRep (m (ResolverValue m))] ->
ResolverValue m
encodeUnion :: [FieldRep (m (ResolverValue m))] -> ResolverValue m
encodeUnion [] = TypeName -> ResolverValue m
forall (m :: * -> *). TypeName -> ResolverValue m
mkEnum TypeName
consName
encodeUnion [FieldRep {fieldTypeRef :: forall a. FieldRep a -> TypeRef
fieldTypeRef = TypeRef {TypeName
typeConName :: TypeRef -> TypeName
typeConName :: TypeName
typeConName}, m (ResolverValue m)
fieldValue :: forall a. FieldRep a -> a
fieldValue :: m (ResolverValue m)
fieldValue}]
| TypeName -> ConsRep (m (ResolverValue m)) -> Bool
forall k. TypeName -> ConsRep k -> Bool
isUnionRef TypeName
dataTypeName ConsRep (m (ResolverValue m))
cons = m (ResolverValue m) -> ResolverValue m
forall (m :: * -> *). m (ResolverValue m) -> ResolverValue m
ResLazy (Maybe TypeName -> ObjectTypeResolver m -> ResolverValue m
forall (m :: * -> *).
Maybe TypeName -> ObjectTypeResolver m -> ResolverValue m
ResObject (TypeName -> Maybe TypeName
forall a. a -> Maybe a
Just TypeName
typeConName) (ObjectTypeResolver m -> ResolverValue m)
-> m (ObjectTypeResolver m) -> m (ResolverValue m)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (m (ResolverValue m)
fieldValue m (ResolverValue m)
-> (ResolverValue m -> m (ObjectTypeResolver m))
-> m (ObjectTypeResolver m)
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= ResolverValue m -> m (ObjectTypeResolver m)
forall (f :: * -> *) (m :: * -> *).
MonadError GQLError f =>
ResolverValue m -> f (ObjectTypeResolver m)
requireObject))
encodeUnion [FieldRep (m (ResolverValue m))]
fields = TypeName -> [ResolverEntry m] -> ResolverValue m
forall (m :: * -> *).
Monad m =>
TypeName -> [ResolverEntry m] -> ResolverValue m
mkUnion TypeName
consName (FieldRep (m (ResolverValue m)) -> ResolverEntry m
forall k (m :: k -> *) (a :: k). FieldRep (m a) -> (FieldName, m a)
toFieldRes (FieldRep (m (ResolverValue m)) -> ResolverEntry m)
-> [FieldRep (m (ResolverValue m))] -> [ResolverEntry m]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [FieldRep (m (ResolverValue m))]
fields)
exploreResolvers ::
forall m a.
( EncodeConstraint m a,
MonadError GQLError m
) =>
a ->
ResolverValue m
exploreResolvers :: a -> ResolverValue m
exploreResolvers =
DataType (m (ResolverValue m)) -> ResolverValue m
forall (m :: * -> *).
MonadError GQLError m =>
DataType (m (ResolverValue m)) -> ResolverValue m
convertNode
(DataType (m (ResolverValue m)) -> ResolverValue m)
-> (a -> DataType (m (ResolverValue m))) -> a -> ResolverValue m
forall b c a. (b -> c) -> (a -> b) -> a -> c
. TypeConstraint (Encode m) (m (ResolverValue m)) Identity
-> Proxy IN -> a -> DataType (m (ResolverValue m))
forall (proxy :: TypeCategory -> *) (kind :: TypeCategory)
(constraint :: * -> Constraint) value a.
(GQLType a, CategoryValue kind, Generic a,
TypeRep constraint value (Rep a)) =>
TypeConstraint constraint value Identity
-> proxy kind -> a -> DataType value
toValue
( (forall a. Encode m a => Identity a -> m (ResolverValue m))
-> TypeConstraint (Encode m) (m (ResolverValue m)) Identity
forall (c :: * -> Constraint) v (f :: * -> *).
(forall a. c a => f a -> v) -> TypeConstraint c v f
TypeConstraint (a -> m (ResolverValue m)
forall (m :: * -> *) resolver.
Encode m resolver =>
resolver -> m (ResolverValue m)
encode (a -> m (ResolverValue m))
-> (Identity a -> a) -> Identity a -> m (ResolverValue m)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Identity a -> a
forall a. Identity a -> a
runIdentity) ::
TypeConstraint (Encode m) (m (ResolverValue m)) Identity
)
(Proxy IN
forall k (t :: k). Proxy t
Proxy @IN)
objectResolvers ::
( EncodeConstraint m a,
MonadError GQLError m
) =>
a ->
ResolverState (ObjectTypeResolver m)
objectResolvers :: a -> ResolverState (ObjectTypeResolver m)
objectResolvers a
value = ResolverValue m -> ResolverState (ObjectTypeResolver m)
forall (f :: * -> *) (m :: * -> *).
MonadError GQLError f =>
ResolverValue m -> f (ObjectTypeResolver m)
requireObject (a -> ResolverValue m
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverValue m
exploreResolvers a
value)
type EncodeConstraint (m :: Type -> Type) a =
( GQLType a,
Generic a,
TypeRep (Encode m) (m (ResolverValue m)) (Rep a)
)
type EncodeObjectConstraint (o :: OperationType) e (m :: Type -> Type) a =
EncodeConstraint (Resolver o e m) (a (Resolver o e m))
type EncodeConstraints e m query mut sub =
( ChannelsConstraint e m sub,
EncodeObjectConstraint QUERY e m query,
EncodeObjectConstraint MUTATION e m mut,
EncodeObjectConstraint SUBSCRIPTION e m sub
)
deriveModel ::
forall e m query mut sub.
(Monad m, EncodeConstraints e m query mut sub) =>
RootResolver m e query mut sub ->
GQLResult (RootResolverValue e m)
deriveModel :: RootResolver m e query mut sub -> GQLResult (RootResolverValue e m)
deriveModel RootResolver {query (Resolver QUERY e m)
mut (Resolver MUTATION e m)
sub (Resolver SUBSCRIPTION e m)
subscriptionResolver :: forall (m :: * -> *) event (query :: (* -> *) -> *)
(mutation :: (* -> *) -> *) (subscription :: (* -> *) -> *).
RootResolver m event query mutation subscription
-> subscription (Resolver SUBSCRIPTION event m)
mutationResolver :: forall (m :: * -> *) event (query :: (* -> *) -> *)
(mutation :: (* -> *) -> *) (subscription :: (* -> *) -> *).
RootResolver m event query mutation subscription
-> mutation (Resolver MUTATION event m)
queryResolver :: forall (m :: * -> *) event (query :: (* -> *) -> *)
(mutation :: (* -> *) -> *) (subscription :: (* -> *) -> *).
RootResolver m event query mutation subscription
-> query (Resolver QUERY event m)
subscriptionResolver :: sub (Resolver SUBSCRIPTION e m)
mutationResolver :: mut (Resolver MUTATION e m)
queryResolver :: query (Resolver QUERY e m)
..} =
RootResolverValue e m -> GQLResult (RootResolverValue e m)
forall (f :: * -> *) a. Applicative f => a -> f a
pure
RootResolverValue :: forall e (m :: * -> *).
ResolverState (ObjectTypeResolver (Resolver QUERY e m))
-> ResolverState (ObjectTypeResolver (Resolver MUTATION e m))
-> ResolverState (ObjectTypeResolver (Resolver SUBSCRIPTION e m))
-> Maybe (Selection VALID -> ResolverState (Channel e))
-> RootResolverValue e m
RootResolverValue
{ queryResolver :: ResolverState (ObjectTypeResolver (Resolver QUERY e m))
queryResolver = query (Resolver QUERY e m)
-> ResolverState (ObjectTypeResolver (Resolver QUERY e m))
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverState (ObjectTypeResolver m)
objectResolvers query (Resolver QUERY e m)
queryResolver,
mutationResolver :: ResolverState (ObjectTypeResolver (Resolver MUTATION e m))
mutationResolver = mut (Resolver MUTATION e m)
-> ResolverState (ObjectTypeResolver (Resolver MUTATION e m))
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverState (ObjectTypeResolver m)
objectResolvers mut (Resolver MUTATION e m)
mutationResolver,
subscriptionResolver :: ResolverState (ObjectTypeResolver (Resolver SUBSCRIPTION e m))
subscriptionResolver = sub (Resolver SUBSCRIPTION e m)
-> ResolverState (ObjectTypeResolver (Resolver SUBSCRIPTION e m))
forall (m :: * -> *) a.
(EncodeConstraint m a, MonadError GQLError m) =>
a -> ResolverState (ObjectTypeResolver m)
objectResolvers sub (Resolver SUBSCRIPTION e m)
subscriptionResolver,
Maybe (Selection VALID -> ResolverState (Channel e))
channelMap :: Maybe (Selection VALID -> ResolverState (Channel e))
channelMap :: Maybe (Selection VALID -> ResolverState (Channel e))
channelMap
}
where
channelMap :: Maybe (Selection VALID -> ResolverState (Channel e))
channelMap
| Proxy (sub (Resolver SUBSCRIPTION e m)) -> Bool
forall a (f :: * -> *). GQLType a => f a -> Bool
__isEmptyType (Proxy (sub (Resolver SUBSCRIPTION e m))
forall k (t :: k). Proxy t
Proxy :: Proxy (sub (Resolver SUBSCRIPTION e m))) = Maybe (Selection VALID -> ResolverState (Channel e))
forall a. Maybe a
Nothing
| Bool
otherwise = (Selection VALID -> ResolverState (Channel e))
-> Maybe (Selection VALID -> ResolverState (Channel e))
forall a. a -> Maybe a
Just (sub (Resolver SUBSCRIPTION e m)
-> Selection VALID -> ResolverState (Channel e)
forall e (m :: * -> *) (subs :: (* -> *) -> *).
ChannelsConstraint e m subs =>
subs (Resolver SUBSCRIPTION e m)
-> Selection VALID -> ResolverState (Channel e)
channelResolver sub (Resolver SUBSCRIPTION e m)
subscriptionResolver)