UNPKG

@graphql-codegen/flutter-freezed

Version:

GraphQL Code Generator plugin to generate Freezed models from your GraphQL schema

23 lines (22 loc) 1.36 kB
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()); };