UNPKG

@graphql-codegen/typescript-document-nodes

Version:

GraphQL Code Generator plugin for generating TypeScript modules with embedded GraphQL document nodes

27 lines (26 loc) 1.19 kB
import { oldVisit } from '@graphql-codegen/plugin-helpers'; import { concatAST, Kind } from 'graphql'; import { TypeScriptDocumentNodesVisitor } from './visitor.js'; export const plugin = (schema, documents, config) => { const allAst = concatAST(documents.map(v => v.document)); const allFragments = [ ...allAst.definitions.filter(d => d.kind === Kind.FRAGMENT_DEFINITION).map(fragmentDef => ({ node: fragmentDef, name: fragmentDef.name.value, onType: fragmentDef.typeCondition.name.value, isExternal: false, })), ...(config.externalFragments || []), ]; const visitor = new TypeScriptDocumentNodesVisitor(schema, allFragments, config, documents); const visitorResult = oldVisit(allAst, { leave: visitor }); return { prepend: visitor.getImports(), content: [visitor.fragments, ...visitorResult.definitions.filter(t => typeof t === 'string')].join('\n'), }; }; export const validate = async (schema, documents, config, outputFile) => { if (!outputFile.endsWith('.ts')) { throw new Error(`Plugin "typescript-document-nodes" requires extension to be ".ts"!`); } };