@graphql-codegen/flutter-freezed
Version:
GraphQL Code Generator plugin to generate Freezed models from your GraphQL schema
23 lines (22 loc) • 1.36 kB
JavaScript
import { oldVisit } from '@graphql-codegen/plugin-helpers';
import { transformSchemaAST } from '@graphql-codegen/schema-ast';
import { FreezedDeclarationBlock } from './freezed-declaration-blocks/index.js';
import { schemaVisitor } from './schema-visitor.js';
import { addFreezedImportStatements, DefaultFreezedPluginConfig } from './utils.js';
export const plugin = (schema, _documents, config) => {
// sets the defaults for the config
config = { ...new DefaultFreezedPluginConfig(config) };
const { schema: _schema, ast } = transformSchemaAST(schema, config);
const { freezedFactoryBlockRepository, ...visitor } = schemaVisitor(_schema, config);
const visitorResult = oldVisit(ast, { leave: visitor });
const generated = visitorResult.definitions.filter((def) => def instanceof FreezedDeclarationBlock);
return (addFreezedImportStatements(config.fileName) +
generated
.map(freezedDeclarationBlock => freezedDeclarationBlock.toString().replace(/==>factory==>.+\n/gm, s => {
const pattern = s.replace('==>factory==>', '').trim();
const [key, appliesOn, name, typeName] = pattern.split('==>');
return freezedFactoryBlockRepository.retrieve(key, appliesOn, name, typeName !== null && typeName !== void 0 ? typeName : null);
}))
.join('')
.trim());
};