UNPKG

@graphql-codegen/flow-operations

Version:

GraphQL Code Generator plugin for generating Flow types for GraphQL operations (query/mutation/subscription/fragment)

36 lines (35 loc) 1.6 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.plugin = void 0; const graphql_1 = require("graphql"); const plugin_helpers_1 = require("@graphql-codegen/plugin-helpers"); const visitor_plugin_common_1 = require("@graphql-codegen/visitor-plugin-common"); const visitor_js_1 = require("./visitor.js"); const plugin = (schema, rawDocuments, config) => { const documents = config.flattenGeneratedTypes ? (0, visitor_plugin_common_1.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 = (0, graphql_1.concatAST)(documents.map(v => v.document)); const includedFragments = allAst.definitions.filter(d => d.kind === graphql_1.Kind.FRAGMENT_DEFINITION); const allFragments = [ ...includedFragments.map(fragmentDef => ({ node: fragmentDef, name: fragmentDef.name.value, onType: fragmentDef.typeCondition.name.value, isExternal: false, })), ...(config.externalFragments || []), ]; const visitor = new visitor_js_1.FlowDocumentsVisitor(schema, config, allFragments); const visitorResult = (0, plugin_helpers_1.oldVisit)(allAst, { leave: visitor, }); return { prepend: ['// @flow\n', ...visitor.getImports()], content: [prefix, ...visitorResult.definitions].join('\n'), }; }; exports.plugin = plugin;