/
makeRemoteExecutableSchema.ts
117 lines (102 loc) · 3.35 KB
/
makeRemoteExecutableSchema.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import { ApolloLink } from 'apollo-link';
import {
GraphQLFieldResolver,
GraphQLSchema,
Kind,
GraphQLResolveInfo,
BuildSchemaOptions,
DocumentNode,
} from 'graphql';
import { Fetcher } from '../Interfaces';
import { buildSchema } from '../polyfills/index';
import { addTypenameToAbstract } from '../delegate/addTypenameToAbstract';
import { checkResultAndHandleErrors } from '../delegate/checkResultAndHandleErrors';
import linkToFetcher, { execute } from '../stitch/linkToFetcher';
import { observableToAsyncIterable } from '../stitch/observableToAsyncIterable';
import mapAsyncIterator from '../stitch/mapAsyncIterator';
import { wrapSchema } from './wrapSchema';
export type ResolverFn = (
rootValue?: any,
args?: any,
context?: any,
info?: GraphQLResolveInfo,
) => AsyncIterator<any>;
export default function makeRemoteExecutableSchema({
schema: schemaOrTypeDefs,
link,
fetcher,
createResolver = defaultCreateRemoteResolver,
createSubscriptionResolver = defaultCreateRemoteSubscriptionResolver,
buildSchemaOptions,
}: {
schema: GraphQLSchema | string;
link?: ApolloLink;
fetcher?: Fetcher;
createResolver?: (fetcher: Fetcher) => GraphQLFieldResolver<any, any>;
createSubscriptionResolver?: (
link: ApolloLink,
) => GraphQLFieldResolver<any, any>;
buildSchemaOptions?: BuildSchemaOptions;
}): GraphQLSchema {
let finalFetcher: Fetcher = fetcher;
if (finalFetcher == null && link != null) {
finalFetcher = linkToFetcher(link);
}
const targetSchema =
typeof schemaOrTypeDefs === 'string'
? buildSchema(schemaOrTypeDefs, buildSchemaOptions)
: schemaOrTypeDefs;
return wrapSchema({
schema: targetSchema,
createProxyingResolver: (_schema, _transforms, operation) => {
if (operation === 'query' || operation === 'mutation') {
return createResolver(finalFetcher);
}
return createSubscriptionResolver(link);
},
});
}
export function defaultCreateRemoteResolver(
fetcher: Fetcher,
): GraphQLFieldResolver<any, any> {
return async (_root, _args, context, info) => {
const fragments = Object.keys(info.fragments).map(
(fragment) => info.fragments[fragment],
);
let query: DocumentNode = {
kind: Kind.DOCUMENT,
definitions: [info.operation, ...fragments],
};
query = addTypenameToAbstract(info.schema, query);
const result = await fetcher({
query,
variables: info.variableValues,
context: { graphqlContext: context },
});
return checkResultAndHandleErrors(result, context, info);
};
}
export function defaultCreateRemoteSubscriptionResolver(
link: ApolloLink,
): GraphQLFieldResolver<any, any> {
return (_root, _args, context, info) => {
const fragments = Object.keys(info.fragments).map(
(fragment) => info.fragments[fragment],
);
let query: DocumentNode = {
kind: Kind.DOCUMENT,
definitions: [info.operation, ...fragments],
};
query = addTypenameToAbstract(info.schema, query);
const operation = {
query,
variables: info.variableValues,
context: { graphqlContext: context },
};
const observable = execute(link, operation);
const originalAsyncIterator = observableToAsyncIterable(observable);
return mapAsyncIterator(originalAsyncIterator, (result) => ({
[info.fieldName]: checkResultAndHandleErrors(result, context, info),
}));
};
}