{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE NamedFieldPuns #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TupleSections #-}
{-# LANGUAGE NoImplicitPrelude #-}

module Data.Morpheus.Validation.Query.UnionSelection
  ( validateUnionSelection,
    validateInterfaceSelection,
  )
where

import qualified Data.HashMap.Lazy as HM
import Data.Mergeable (OrdMap)
import Data.Morpheus.Internal.Utils
  ( empty,
    fromElems,
    mergeConcat,
    selectOr,
    startHistory,
  )
import Data.Morpheus.Types.Internal.AST.Base (Position (..))
import Data.Morpheus.Types.Internal.AST.Name (TypeName)
import Data.Morpheus.Types.Internal.AST.Selection
  ( Fragment (..),
    Selection (..),
    SelectionContent (..),
    SelectionSet,
    UnionTag (..),
  )
import Data.Morpheus.Types.Internal.AST.Stage (RAW, VALID)
import Data.Morpheus.Types.Internal.AST.TypeCategory
  ( IMPLEMENTABLE,
    OUT,
    toCategory,
  )
import Data.Morpheus.Types.Internal.AST.TypeSystem
  ( TypeContent (..),
    TypeDefinition (..),
    UnionTypeDefinition,
    mkType,
  )
import Data.Morpheus.Types.Internal.Validation
  ( FragmentValidator,
    Scope (..),
    askInterfaceTypes,
    askTypeMember,
    asksScope,
  )
import Data.Morpheus.Validation.Query.Fragment
  ( ValidateFragmentSelection,
    castFragmentType,
    validateSpread,
  )
import Relude hiding (empty, join)

-- returns all Fragments used for Possible Types
splitFragment ::
  (ValidateFragmentSelection s) =>
  ( Fragment RAW ->
    FragmentValidator s (SelectionSet VALID)
  ) ->
  [TypeDefinition IMPLEMENTABLE VALID] ->
  Selection RAW ->
  FragmentValidator s (Either UnionTag (Selection RAW))
splitFragment :: (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> [TypeDefinition IMPLEMENTABLE VALID]
-> Selection RAW
-> FragmentValidator s (Either UnionTag (Selection RAW))
splitFragment Fragment RAW -> FragmentValidator s (SelectionSet VALID)
_ [TypeDefinition IMPLEMENTABLE VALID]
_ x :: Selection RAW
x@Selection {} = Either UnionTag (Selection RAW)
-> FragmentValidator s (Either UnionTag (Selection RAW))
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Selection RAW -> Either UnionTag (Selection RAW)
forall a b. b -> Either a b
Right Selection RAW
x)
splitFragment Fragment RAW -> FragmentValidator s (SelectionSet VALID)
f [TypeDefinition IMPLEMENTABLE VALID]
types (Spread Directives RAW
_ Ref FragmentName
ref) = UnionTag -> Either UnionTag (Selection RAW)
forall a b. a -> Either a b
Left (UnionTag -> Either UnionTag (Selection RAW))
-> Validator VALID (OperationContext VALID s) UnionTag
-> FragmentValidator s (Either UnionTag (Selection RAW))
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> [TypeName]
-> Ref FragmentName
-> Validator VALID (OperationContext VALID s) UnionTag
forall (s :: Stage).
ValidateFragmentSelection s =>
(Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> [TypeName] -> Ref FragmentName -> FragmentValidator s UnionTag
validateSpread Fragment RAW -> FragmentValidator s (SelectionSet VALID)
f (TypeDefinition IMPLEMENTABLE VALID -> TypeName
forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> TypeName
typeName (TypeDefinition IMPLEMENTABLE VALID -> TypeName)
-> [TypeDefinition IMPLEMENTABLE VALID] -> [TypeName]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [TypeDefinition IMPLEMENTABLE VALID]
types) Ref FragmentName
ref
splitFragment Fragment RAW -> FragmentValidator s (SelectionSet VALID)
f [TypeDefinition IMPLEMENTABLE VALID]
types (InlineFragment fragment :: Fragment RAW
fragment@Fragment {TypeName
fragmentType :: forall (stage :: Stage). Fragment stage -> TypeName
fragmentType :: TypeName
fragmentType}) =
  UnionTag -> Either UnionTag (Selection RAW)
forall a b. a -> Either a b
Left (UnionTag -> Either UnionTag (Selection RAW))
-> (MergeMap 'False FieldName (Selection VALID) -> UnionTag)
-> MergeMap 'False FieldName (Selection VALID)
-> Either UnionTag (Selection RAW)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. TypeName -> SelectionSet VALID -> UnionTag
UnionTag TypeName
fragmentType
    (MergeMap 'False FieldName (Selection VALID)
 -> Either UnionTag (Selection RAW))
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
-> FragmentValidator s (Either UnionTag (Selection RAW))
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> ( Maybe FragmentName
-> Position
-> [TypeName]
-> Fragment RAW
-> FragmentValidator s (Fragment RAW)
forall (s :: Stage) (s1 :: Stage).
Maybe FragmentName
-> Position
-> [TypeName]
-> Fragment s
-> FragmentValidator s1 (Fragment s)
castFragmentType Maybe FragmentName
forall a. Maybe a
Nothing (Fragment RAW -> Position
forall (stage :: Stage). Fragment stage -> Position
fragmentPosition Fragment RAW
fragment) (TypeDefinition IMPLEMENTABLE VALID -> TypeName
forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> TypeName
typeName (TypeDefinition IMPLEMENTABLE VALID -> TypeName)
-> [TypeDefinition IMPLEMENTABLE VALID] -> [TypeName]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [TypeDefinition IMPLEMENTABLE VALID]
types) Fragment RAW
fragment
            FragmentValidator s (Fragment RAW)
-> (Fragment RAW
    -> Validator
         VALID
         (OperationContext VALID s)
         (MergeMap 'False FieldName (Selection VALID)))
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= Fragment RAW
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
Fragment RAW -> FragmentValidator s (SelectionSet VALID)
f
        )

exploreFragments ::
  (ValidateFragmentSelection s) =>
  ( Fragment RAW ->
    FragmentValidator s (SelectionSet VALID)
  ) ->
  OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID) ->
  SelectionSet RAW ->
  FragmentValidator s ([UnionTag], SelectionSet RAW)
exploreFragments :: (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> SelectionSet RAW
-> FragmentValidator s ([UnionTag], SelectionSet RAW)
exploreFragments Fragment RAW -> FragmentValidator s (SelectionSet VALID)
validateFragment OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
types SelectionSet RAW
selectionSet = do
  ([UnionTag]
tags, [Selection RAW]
selections) <- [Either UnionTag (Selection RAW)] -> ([UnionTag], [Selection RAW])
forall a b. [Either a b] -> ([a], [b])
partitionEithers ([Either UnionTag (Selection RAW)]
 -> ([UnionTag], [Selection RAW]))
-> Validator
     VALID (OperationContext VALID s) [Either UnionTag (Selection RAW)]
-> Validator
     VALID (OperationContext VALID s) ([UnionTag], [Selection RAW])
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Selection RAW
 -> Validator
      VALID (OperationContext VALID s) (Either UnionTag (Selection RAW)))
-> [Selection RAW]
-> Validator
     VALID (OperationContext VALID s) [Either UnionTag (Selection RAW)]
forall (t :: * -> *) (f :: * -> *) a b.
(Traversable t, Applicative f) =>
(a -> f b) -> t a -> f (t b)
traverse ((Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> [TypeDefinition IMPLEMENTABLE VALID]
-> Selection RAW
-> Validator
     VALID (OperationContext VALID s) (Either UnionTag (Selection RAW))
forall (s :: Stage).
ValidateFragmentSelection s =>
(Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> [TypeDefinition IMPLEMENTABLE VALID]
-> Selection RAW
-> FragmentValidator s (Either UnionTag (Selection RAW))
splitFragment Fragment RAW -> FragmentValidator s (SelectionSet VALID)
validateFragment (OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> [TypeDefinition IMPLEMENTABLE VALID]
forall (t :: * -> *) a. Foldable t => t a -> [a]
toList OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
types)) (MergeMap 'True FieldName (Selection RAW) -> [Selection RAW]
forall (t :: * -> *) a. Foldable t => t a -> [a]
toList MergeMap 'True FieldName (Selection RAW)
SelectionSet RAW
selectionSet)
  Position
selectionPosition <- Position -> Maybe Position -> Position
forall a. a -> Maybe a -> a
fromMaybe (Int -> Int -> Position
Position Int
0 Int
0) (Maybe Position -> Position)
-> Validator VALID (OperationContext VALID s) (Maybe Position)
-> Validator VALID (OperationContext VALID s) Position
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Scope -> Maybe Position)
-> Validator VALID (OperationContext VALID s) (Maybe Position)
forall (s :: Stage) ctx (m :: * -> *) a.
MonadReader (ValidatorContext s ctx) m =>
(Scope -> a) -> m a
asksScope Scope -> Maybe Position
position
  ([UnionTag]
tags,)
    (MergeMap 'True FieldName (Selection RAW)
 -> ([UnionTag], MergeMap 'True FieldName (Selection RAW)))
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'True FieldName (Selection RAW))
-> Validator
     VALID
     (OperationContext VALID s)
     ([UnionTag], MergeMap 'True FieldName (Selection RAW))
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [Selection RAW]
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'True FieldName (Selection RAW))
forall (m :: * -> *) k a (map :: * -> * -> *).
(Monad m, KeyOf k a, FromList m map k a) =>
[a] -> m (map k a)
fromElems
      ( ( Selection :: forall (s :: Stage).
Position
-> Maybe FieldName
-> FieldName
-> Arguments s
-> Directives s
-> SelectionContent s
-> Selection s
Selection
            { selectionName :: FieldName
selectionName = FieldName
"__typename",
              selectionAlias :: Maybe FieldName
selectionAlias = Maybe FieldName
forall a. Maybe a
Nothing,
              Position
selectionPosition :: Position
selectionPosition :: Position
selectionPosition,
              selectionArguments :: Arguments RAW
selectionArguments = Arguments RAW
forall coll. Empty coll => coll
empty,
              selectionContent :: SelectionContent RAW
selectionContent = SelectionContent RAW
forall (s :: Stage). SelectionContent s
SelectionField,
              selectionDirectives :: Directives RAW
selectionDirectives = Directives RAW
forall coll. Empty coll => coll
empty
            }
        )
          Selection RAW -> [Selection RAW] -> [Selection RAW]
forall a. a -> [a] -> [a]
: [Selection RAW]
selections
      )

-- sorts Fragment by conditional Types
-- [
--   ( Type for Tag User , [ Fragment for User] )
--   ( Type for Tag Product , [ Fragment for Product] )
-- ]
tagUnionFragments ::
  [UnionTag] ->
  OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID) ->
  HashMap TypeName [SelectionSet VALID]
tagUnionFragments :: [UnionTag]
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> HashMap TypeName [SelectionSet VALID]
tagUnionFragments [UnionTag]
fragments OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
types = ([TypeName] -> [MergeMap 'False FieldName (Selection VALID)])
-> HashMap TypeName [TypeName]
-> HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [TypeName] -> [MergeMap 'False FieldName (Selection VALID)]
[TypeName] -> [SelectionSet VALID]
categorizeType HashMap TypeName [TypeName]
getSelectedTypes
  where
    getSelectedTypes :: HashMap TypeName [TypeName]
    getSelectedTypes :: HashMap TypeName [TypeName]
getSelectedTypes = [Item (HashMap TypeName [TypeName])] -> HashMap TypeName [TypeName]
forall l. IsList l => [Item l] -> l
fromList ((UnionTag -> (TypeName, [TypeName]))
-> [UnionTag] -> [(TypeName, [TypeName])]
forall a b. (a -> b) -> [a] -> [b]
map UnionTag -> (TypeName, [TypeName])
select [UnionTag]
fragments)
      where
        select :: UnionTag -> (TypeName, [TypeName])
select UnionTag {TypeName
unionTagName :: UnionTag -> TypeName
unionTagName :: TypeName
unionTagName} =
          ( TypeName
unionTagName,
            [TypeName]
-> (TypeDefinition IMPLEMENTABLE VALID -> [TypeName])
-> TypeName
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> [TypeName]
forall k (c :: * -> *) d a.
IsMap k c =>
d -> (a -> d) -> k -> c a -> d
selectOr
              [TypeName
unionTagName]
              TypeDefinition IMPLEMENTABLE VALID -> [TypeName]
forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> [TypeName]
getCompatibleTypes
              TypeName
unionTagName
              OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
types
          )
    categorizeType ::
      [TypeName] -> [SelectionSet VALID]
    categorizeType :: [TypeName] -> [SelectionSet VALID]
categorizeType [TypeName]
compatibleTypes =
      UnionTag -> MergeMap 'False FieldName (Selection VALID)
UnionTag -> SelectionSet VALID
unionTagSelection
        (UnionTag -> MergeMap 'False FieldName (Selection VALID))
-> [UnionTag] -> [MergeMap 'False FieldName (Selection VALID)]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (UnionTag -> Bool) -> [UnionTag] -> [UnionTag]
forall a. (a -> Bool) -> [a] -> [a]
filter
          ((TypeName -> [TypeName] -> Bool
forall (f :: * -> *) a.
(Foldable f, DisallowElem f, Eq a) =>
a -> f a -> Bool
`elem` [TypeName]
compatibleTypes) (TypeName -> Bool) -> (UnionTag -> TypeName) -> UnionTag -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. UnionTag -> TypeName
unionTagName)
          [UnionTag]
fragments

getCompatibleTypes :: TypeDefinition a s -> [TypeName]
getCompatibleTypes :: TypeDefinition a s -> [TypeName]
getCompatibleTypes TypeDefinition {TypeName
typeName :: TypeName
typeName :: forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> TypeName
typeName, typeContent :: forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> TypeContent 'True a s
typeContent = DataObject {[TypeName]
objectImplements :: forall (a :: TypeCategory) (s :: Stage).
CondTypeContent OBJECT a s -> [TypeName]
objectImplements :: [TypeName]
objectImplements}} = TypeName
typeName TypeName -> [TypeName] -> [TypeName]
forall a. a -> [a] -> [a]
: [TypeName]
objectImplements
getCompatibleTypes TypeDefinition {TypeName
typeName :: TypeName
typeName :: forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> TypeName
typeName} = [TypeName
typeName]

joinClusters ::
  SelectionSet VALID ->
  HashMap TypeName [SelectionSet VALID] ->
  FragmentValidator s (SelectionContent VALID)
joinClusters :: SelectionSet VALID
-> HashMap TypeName [SelectionSet VALID]
-> FragmentValidator s (SelectionContent VALID)
joinClusters SelectionSet VALID
selSet HashMap TypeName [SelectionSet VALID]
typedSelections
  | HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
-> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
HashMap TypeName [SelectionSet VALID]
typedSelections = SelectionContent VALID
-> FragmentValidator s (SelectionContent VALID)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (SelectionSet VALID -> SelectionContent VALID
forall (s :: Stage). SelectionSet s -> SelectionContent s
SelectionSet SelectionSet VALID
selSet)
  | Bool
otherwise =
    ((TypeName, [MergeMap 'False FieldName (Selection VALID)])
 -> Validator VALID (OperationContext VALID s) UnionTag)
-> [(TypeName, [MergeMap 'False FieldName (Selection VALID)])]
-> Validator VALID (OperationContext VALID s) [UnionTag]
forall (t :: * -> *) (f :: * -> *) a b.
(Traversable t, Applicative f) =>
(a -> f b) -> t a -> f (t b)
traverse (TypeName, [MergeMap 'False FieldName (Selection VALID)])
-> Validator VALID (OperationContext VALID s) UnionTag
forall (s :: Stage).
(TypeName, [SelectionSet VALID]) -> FragmentValidator s UnionTag
mkUnionTag (HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
-> [(TypeName, [MergeMap 'False FieldName (Selection VALID)])]
forall k v. HashMap k v -> [(k, v)]
HM.toList HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
HashMap TypeName [SelectionSet VALID]
typedSelections)
      Validator VALID (OperationContext VALID s) [UnionTag]
-> ([UnionTag] -> FragmentValidator s (SelectionContent VALID))
-> FragmentValidator s (SelectionContent VALID)
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= (MergeMap 'False TypeName UnionTag -> SelectionContent VALID)
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False TypeName UnionTag)
-> FragmentValidator s (SelectionContent VALID)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (SelectionSet VALID
-> UnionSelection VALID -> SelectionContent VALID
UnionSelection SelectionSet VALID
selSet) (Validator
   VALID
   (OperationContext VALID s)
   (MergeMap 'False TypeName UnionTag)
 -> FragmentValidator s (SelectionContent VALID))
-> ([UnionTag]
    -> Validator
         VALID
         (OperationContext VALID s)
         (MergeMap 'False TypeName UnionTag))
-> [UnionTag]
-> FragmentValidator s (SelectionContent VALID)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. HistoryT
  (Validator VALID (OperationContext VALID s))
  (MergeMap 'False TypeName UnionTag)
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False TypeName UnionTag)
forall (m :: * -> *) a. HistoryT m a -> m a
startHistory (HistoryT
   (Validator VALID (OperationContext VALID s))
   (MergeMap 'False TypeName UnionTag)
 -> Validator
      VALID
      (OperationContext VALID s)
      (MergeMap 'False TypeName UnionTag))
-> ([UnionTag]
    -> HistoryT
         (Validator VALID (OperationContext VALID s))
         (MergeMap 'False TypeName UnionTag))
-> [UnionTag]
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False TypeName UnionTag)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [UnionTag]
-> HistoryT
     (Validator VALID (OperationContext VALID s))
     (MergeMap 'False TypeName UnionTag)
forall (m :: * -> *) k a (map :: * -> * -> *).
(Monad m, KeyOf k a, FromList m map k a) =>
[a] -> m (map k a)
fromElems
  where
    mkUnionTag :: (TypeName, [SelectionSet VALID]) -> FragmentValidator s UnionTag
    mkUnionTag :: (TypeName, [SelectionSet VALID]) -> FragmentValidator s UnionTag
mkUnionTag (TypeName
typeName, [SelectionSet VALID]
fragments) = TypeName -> SelectionSet VALID -> UnionTag
UnionTag TypeName
typeName (MergeMap 'False FieldName (Selection VALID) -> UnionTag)
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
-> FragmentValidator s UnionTag
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> HistoryT
  (Validator VALID (OperationContext VALID s))
  (MergeMap 'False FieldName (Selection VALID))
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
forall (m :: * -> *) a. HistoryT m a -> m a
startHistory (NonEmpty (MergeMap 'False FieldName (Selection VALID))
-> HistoryT
     (Validator VALID (OperationContext VALID s))
     (MergeMap 'False FieldName (Selection VALID))
forall (m :: * -> *) a e.
(Monad m, Merge m a, MonadError e m) =>
NonEmpty a -> m a
mergeConcat (MergeMap 'False FieldName (Selection VALID)
SelectionSet VALID
selSet MergeMap 'False FieldName (Selection VALID)
-> [MergeMap 'False FieldName (Selection VALID)]
-> NonEmpty (MergeMap 'False FieldName (Selection VALID))
forall a. a -> [a] -> NonEmpty a
:| [MergeMap 'False FieldName (Selection VALID)]
[SelectionSet VALID]
fragments))

validateInterfaceSelection ::
  ValidateFragmentSelection s =>
  (Fragment RAW -> FragmentValidator s (SelectionSet VALID)) ->
  (TypeDefinition IMPLEMENTABLE VALID -> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID)) ->
  TypeDefinition IMPLEMENTABLE VALID ->
  SelectionSet RAW ->
  FragmentValidator s (SelectionContent VALID)
validateInterfaceSelection :: (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> (TypeDefinition IMPLEMENTABLE VALID
    -> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID))
-> TypeDefinition IMPLEMENTABLE VALID
-> SelectionSet RAW
-> FragmentValidator s (SelectionContent VALID)
validateInterfaceSelection
  Fragment RAW -> FragmentValidator s (SelectionSet VALID)
validateFragment
  TypeDefinition IMPLEMENTABLE VALID
-> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID)
validate
  typeDef :: TypeDefinition IMPLEMENTABLE VALID
typeDef@TypeDefinition {TypeName
typeName :: TypeName
typeName :: forall (a :: TypeCategory) (s :: Stage).
TypeDefinition a s -> TypeName
typeName}
  SelectionSet RAW
inputSelectionSet = do
    OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
possibleTypes <- TypeDefinition IMPLEMENTABLE VALID
-> Validator
     VALID
     (OperationContext VALID s)
     (OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID))
forall (m :: * -> *) (s :: Stage) ctx.
(MonadError GQLError m, MonadReader (ValidatorContext s ctx) m,
 FromCategory (TypeContent 'True) ANY IMPLEMENTABLE) =>
TypeDefinition IMPLEMENTABLE s
-> m (OrdMap TypeName (TypeDefinition IMPLEMENTABLE s))
askInterfaceTypes TypeDefinition IMPLEMENTABLE VALID
typeDef
    ([UnionTag]
spreads, MergeMap 'True FieldName (Selection RAW)
selectionSet) <- (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> SelectionSet RAW
-> FragmentValidator s ([UnionTag], SelectionSet RAW)
forall (s :: Stage).
ValidateFragmentSelection s =>
(Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> SelectionSet RAW
-> FragmentValidator s ([UnionTag], SelectionSet RAW)
exploreFragments Fragment RAW -> FragmentValidator s (SelectionSet VALID)
validateFragment OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
possibleTypes SelectionSet RAW
inputSelectionSet
    MergeMap 'False FieldName (Selection VALID)
validSelectionSet <- TypeDefinition IMPLEMENTABLE VALID
-> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID)
validate TypeDefinition IMPLEMENTABLE VALID
typeDef MergeMap 'True FieldName (Selection RAW)
SelectionSet RAW
selectionSet
    let tags :: HashMap TypeName [SelectionSet VALID]
tags = [UnionTag]
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> HashMap TypeName [SelectionSet VALID]
tagUnionFragments [UnionTag]
spreads OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
possibleTypes
    let interfaces :: [MergeMap 'False FieldName (Selection VALID)]
interfaces = [MergeMap 'False FieldName (Selection VALID)]
-> ([MergeMap 'False FieldName (Selection VALID)]
    -> [MergeMap 'False FieldName (Selection VALID)])
-> TypeName
-> HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
-> [MergeMap 'False FieldName (Selection VALID)]
forall k (c :: * -> *) d a.
IsMap k c =>
d -> (a -> d) -> k -> c a -> d
selectOr [] [MergeMap 'False FieldName (Selection VALID)]
-> [MergeMap 'False FieldName (Selection VALID)]
forall a. a -> a
id TypeName
typeName HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
HashMap TypeName [SelectionSet VALID]
tags
    MergeMap 'False FieldName (Selection VALID)
defaultSelection <- HistoryT
  (Validator VALID (OperationContext VALID s))
  (MergeMap 'False FieldName (Selection VALID))
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
forall (m :: * -> *) a. HistoryT m a -> m a
startHistory (HistoryT
   (Validator VALID (OperationContext VALID s))
   (MergeMap 'False FieldName (Selection VALID))
 -> Validator
      VALID
      (OperationContext VALID s)
      (MergeMap 'False FieldName (Selection VALID)))
-> HistoryT
     (Validator VALID (OperationContext VALID s))
     (MergeMap 'False FieldName (Selection VALID))
-> Validator
     VALID
     (OperationContext VALID s)
     (MergeMap 'False FieldName (Selection VALID))
forall a b. (a -> b) -> a -> b
$ NonEmpty (MergeMap 'False FieldName (Selection VALID))
-> HistoryT
     (Validator VALID (OperationContext VALID s))
     (MergeMap 'False FieldName (Selection VALID))
forall (m :: * -> *) a e.
(Monad m, Merge m a, MonadError e m) =>
NonEmpty a -> m a
mergeConcat (MergeMap 'False FieldName (Selection VALID)
validSelectionSet MergeMap 'False FieldName (Selection VALID)
-> [MergeMap 'False FieldName (Selection VALID)]
-> NonEmpty (MergeMap 'False FieldName (Selection VALID))
forall a. a -> [a] -> NonEmpty a
:| [MergeMap 'False FieldName (Selection VALID)]
interfaces)
    SelectionSet VALID
-> HashMap TypeName [SelectionSet VALID]
-> FragmentValidator s (SelectionContent VALID)
forall (s :: Stage).
SelectionSet VALID
-> HashMap TypeName [SelectionSet VALID]
-> FragmentValidator s (SelectionContent VALID)
joinClusters MergeMap 'False FieldName (Selection VALID)
SelectionSet VALID
defaultSelection (TypeName
-> HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
-> HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
forall k v. (Eq k, Hashable k) => k -> HashMap k v -> HashMap k v
HM.delete TypeName
typeName HashMap TypeName [MergeMap 'False FieldName (Selection VALID)]
HashMap TypeName [SelectionSet VALID]
tags)

mkUnionRootType :: FragmentValidator s (TypeDefinition IMPLEMENTABLE VALID)
mkUnionRootType :: FragmentValidator s (TypeDefinition IMPLEMENTABLE VALID)
mkUnionRootType = (TypeName
-> TypeContent 'True IMPLEMENTABLE VALID
-> TypeDefinition IMPLEMENTABLE VALID
forall (a :: TypeCategory) (s :: Stage).
TypeName -> TypeContent 'True a s -> TypeDefinition a s
`mkType` [TypeName]
-> FieldsDefinition OUT VALID
-> CondTypeContent OBJECT IMPLEMENTABLE VALID
forall (s :: Stage) (a :: TypeCategory).
[TypeName] -> FieldsDefinition OUT s -> CondTypeContent OBJECT a s
DataObject [] FieldsDefinition OUT VALID
forall coll. Empty coll => coll
empty) (TypeName -> TypeDefinition IMPLEMENTABLE VALID)
-> Validator VALID (OperationContext VALID s) TypeName
-> FragmentValidator s (TypeDefinition IMPLEMENTABLE VALID)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Scope -> TypeName)
-> Validator VALID (OperationContext VALID s) TypeName
forall (s :: Stage) ctx (m :: * -> *) a.
MonadReader (ValidatorContext s ctx) m =>
(Scope -> a) -> m a
asksScope Scope -> TypeName
currentTypeName

validateUnionSelection ::
  ValidateFragmentSelection s =>
  (Fragment RAW -> FragmentValidator s (SelectionSet VALID)) ->
  (TypeDefinition IMPLEMENTABLE VALID -> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID)) ->
  UnionTypeDefinition OUT VALID ->
  SelectionSet RAW ->
  FragmentValidator s (SelectionContent VALID)
validateUnionSelection :: (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> (TypeDefinition IMPLEMENTABLE VALID
    -> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID))
-> UnionTypeDefinition OUT VALID
-> SelectionSet RAW
-> FragmentValidator s (SelectionContent VALID)
validateUnionSelection Fragment RAW -> FragmentValidator s (SelectionSet VALID)
validateFragment TypeDefinition IMPLEMENTABLE VALID
-> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID)
validate UnionTypeDefinition OUT VALID
members SelectionSet RAW
selectionSet = do
  OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
unionTypes <- (UnionMember OUT VALID
 -> Validator
      VALID
      (OperationContext VALID s)
      (TypeDefinition IMPLEMENTABLE VALID))
-> UnionTypeDefinition OUT VALID
-> Validator
     VALID
     (OperationContext VALID s)
     (OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID))
forall (t :: * -> *) (f :: * -> *) a b.
(Traversable t, Applicative f) =>
(a -> f b) -> t a -> f (t b)
traverse ((TypeDefinition OBJECT VALID -> TypeDefinition IMPLEMENTABLE VALID)
-> Validator
     VALID (OperationContext VALID s) (TypeDefinition OBJECT VALID)
-> Validator
     VALID
     (OperationContext VALID s)
     (TypeDefinition IMPLEMENTABLE VALID)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap TypeDefinition OBJECT VALID -> TypeDefinition IMPLEMENTABLE VALID
forall (a :: TypeCategory -> Stage -> *) (k :: TypeCategory)
       (k' :: TypeCategory) (s :: Stage).
ToCategory a k k' =>
a k s -> a k' s
toCategory (Validator
   VALID (OperationContext VALID s) (TypeDefinition OBJECT VALID)
 -> Validator
      VALID
      (OperationContext VALID s)
      (TypeDefinition IMPLEMENTABLE VALID))
-> (UnionMember OUT VALID
    -> Validator
         VALID (OperationContext VALID s) (TypeDefinition OBJECT VALID))
-> UnionMember OUT VALID
-> Validator
     VALID
     (OperationContext VALID s)
     (TypeDefinition IMPLEMENTABLE VALID)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. UnionMember OUT VALID
-> Validator
     VALID (OperationContext VALID s) (TypeDefinition OBJECT VALID)
forall (m :: * -> *) c (cat :: TypeCategory) (s :: Stage) ctx.
Constraints m c cat s ctx =>
UnionMember cat s -> m (TypeDefinition (ToOBJECT cat) s)
askTypeMember) UnionTypeDefinition OUT VALID
members
  ([UnionTag]
spreads, MergeMap 'True FieldName (Selection RAW)
selSet) <- (Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> SelectionSet RAW
-> FragmentValidator s ([UnionTag], SelectionSet RAW)
forall (s :: Stage).
ValidateFragmentSelection s =>
(Fragment RAW -> FragmentValidator s (SelectionSet VALID))
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> SelectionSet RAW
-> FragmentValidator s ([UnionTag], SelectionSet RAW)
exploreFragments Fragment RAW -> FragmentValidator s (SelectionSet VALID)
validateFragment OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
unionTypes SelectionSet RAW
selectionSet
  TypeDefinition IMPLEMENTABLE VALID
typeDef <- Validator
  VALID
  (OperationContext VALID s)
  (TypeDefinition IMPLEMENTABLE VALID)
forall (s :: Stage).
FragmentValidator s (TypeDefinition IMPLEMENTABLE VALID)
mkUnionRootType
  MergeMap 'False FieldName (Selection VALID)
validSelection <- TypeDefinition IMPLEMENTABLE VALID
-> SelectionSet RAW -> FragmentValidator s (SelectionSet VALID)
validate TypeDefinition IMPLEMENTABLE VALID
typeDef MergeMap 'True FieldName (Selection RAW)
SelectionSet RAW
selSet
  SelectionSet VALID
-> HashMap TypeName [SelectionSet VALID]
-> FragmentValidator s (SelectionContent VALID)
forall (s :: Stage).
SelectionSet VALID
-> HashMap TypeName [SelectionSet VALID]
-> FragmentValidator s (SelectionContent VALID)
joinClusters MergeMap 'False FieldName (Selection VALID)
SelectionSet VALID
validSelection ([UnionTag]
-> OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
-> HashMap TypeName [SelectionSet VALID]
tagUnionFragments [UnionTag]
spreads OrdMap TypeName (TypeDefinition IMPLEMENTABLE VALID)
unionTypes)