-
Notifications
You must be signed in to change notification settings - Fork 1.3k
/
index.ts
43 lines (36 loc) · 1.56 KB
/
index.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
import { oldVisit, PluginFunction, Types } from '@graphql-codegen/plugin-helpers';
import { concatAST, GraphQLSchema, Kind, FragmentDefinitionNode } from 'graphql';
import { FlowDocumentsVisitor } from './visitor';
import { LoadedFragment, optimizeOperations } from '@graphql-codegen/visitor-plugin-common';
import { FlowDocumentsPluginConfig } from './config';
export const plugin: PluginFunction<FlowDocumentsPluginConfig> = (
schema: GraphQLSchema,
rawDocuments: Types.DocumentFile[],
config: FlowDocumentsPluginConfig
) => {
const documents = config.flattenGeneratedTypes
? optimizeOperations(schema, rawDocuments, { includeFragments: true })
: rawDocuments;
const prefix = config.preResolveTypes
? ''
: `type $Pick<Origin: Object, Keys: Object> = $ObjMapi<Keys, <Key>(k: Key) => $ElementType<Origin, Key>>;\n`;
const allAst = concatAST(documents.map(v => v.document));
const includedFragments = allAst.definitions.filter(d => d.kind === Kind.FRAGMENT_DEFINITION);
const allFragments: LoadedFragment[] = [
...(includedFragments as FragmentDefinitionNode[]).map(fragmentDef => ({
node: fragmentDef,
name: fragmentDef.name.value,
onType: fragmentDef.typeCondition.name.value,
isExternal: false,
})),
...(config.externalFragments || []),
];
const visitor = new FlowDocumentsVisitor(schema, config, allFragments);
const visitorResult = oldVisit(allAst, {
leave: visitor,
});
return {
prepend: ['// @flow\n', ...visitor.getImports()],
content: [prefix, ...visitorResult.definitions].join('\n'),
};
};