@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
JavaScript
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"!`);
}
};