@graphql-codegen/typed-document-node
Version:
GraphQL Code Generator plugin for generating ready-to-use TypedDocumentNode based on GraphQL operations
33 lines (32 loc) • 1.7 kB
JavaScript
import { extname } from 'path';
import { oldVisit } from '@graphql-codegen/plugin-helpers';
import { DocumentMode, optimizeOperations, } from '@graphql-codegen/visitor-plugin-common';
import { concatAST, Kind } from 'graphql';
import { TypeScriptDocumentNodesVisitor } from './visitor.js';
export const plugin = (schema, rawDocuments, config) => {
const documents = config.flattenGeneratedTypes ? optimizeOperations(schema, rawDocuments) : rawDocuments;
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: allAst.definitions.length === 0 ? [] : visitor.getImports(),
content: [visitor.fragments, ...visitorResult.definitions.filter(t => typeof t === 'string')].join('\n'),
};
};
export const validate = async (schema, documents, config, outputFile) => {
if (config && config.documentMode === DocumentMode.string) {
throw new Error(`Plugin "typed-document-node" does not allow using 'documentMode: string' configuration!`);
}
if (extname(outputFile) !== '.ts' && extname(outputFile) !== '.tsx') {
throw new Error(`Plugin "typed-document-node" requires extension to be ".ts" or ".tsx"!`);
}
};