UNPKG

stackpress

Version:

Incept is a content management framework.

132 lines (131 loc) 5.14 kB
import * as typemap from '../../schema/config/typemaps.js'; export default function generate(directory, registry) { for (const model of registry.model.values()) { const file = `${model.name}/types.ts`; const source = directory.createSourceFile(file, '', { overwrite: true }); generateModel(source, model); } for (const fieldset of registry.fieldset.values()) { const file = `${fieldset.name}/types.ts`; const source = directory.createSourceFile(file, '', { overwrite: true }); generateFieldset(source, fieldset); } const source = directory.createSourceFile('types.ts', '', { overwrite: true }); for (const model of registry.model.values()) { source.addExportDeclaration({ isTypeOnly: true, moduleSpecifier: `./${model.name}/types.js`, namedExports: [model.title, `${model.title}Input`, `${model.title}Extended`] }); } for (const fieldset of registry.fieldset.values()) { source.addExportDeclaration({ isTypeOnly: true, moduleSpecifier: `./${fieldset.name}/types.js`, namedExports: [fieldset.title, `${fieldset.title}Input`] }); } } ; export function generateModel(source, model) { const columns = Array.from(model.columns.values()); const enums = Array.from(model.enums.values()).map(column => column.type); if (enums.length > 0) { source.addImportDeclaration({ moduleSpecifier: '../enums.js', namedImports: enums }); } for (const column of model.relations) { const relation = column.relation; if (!relation) continue; const model = relation?.parent.model; source.addImportDeclaration({ moduleSpecifier: `../${model.name}/types.js`, namedImports: [model.title] }); } for (const column of model.fieldsets.values()) { source.addImportDeclaration({ moduleSpecifier: `../${column.type}/types.js`, namedImports: [column.type] }); } source.addTypeAlias({ isExported: true, name: model.title, type: (`{ ${columns.filter(column => !!column.typemap.model || !!column.enum || !!column.fieldset).map(column => (`${column.name}${!column.required ? '?' : ''}: ${column.typemap.model || column.type}${column.multiple ? '[]' : ''}`)).join(',\n')} }`) }); if (model.relations.length) { source.addTypeAlias({ isExported: true, name: `${model.title}Extended`, type: (`${model.title} & { ${model.relations.map(column => (`${column.name}${!column.required ? '?' : ''}: ${column.type}${column.multiple ? '[]' : ''}`)).join(',\n')} }`) }); } else { source.addTypeAlias({ isExported: true, name: `${model.title}Extended`, type: model.title }); } const inputs = columns .filter(column => !column.generated) .filter(column => [ ...Object.keys(typemap.model), ...model.enums.map(column => column.type), ...model.fieldsets.map(column => column.fieldset?.title) ].includes(column.type)); source.addTypeAlias({ isExported: true, name: `${model.title}Input`, type: (`{ ${inputs.map(column => (`${column.name}${!column.required || typeof column.default !== 'undefined' ? '?' : ''}: ${column.typemap.model || column.type}${column.multiple ? '[]' : ''}`)).join(',\n')} }`) }); } ; export function generateFieldset(source, fieldset) { const columns = Array.from(fieldset.columns.values()); const enums = Array.from(fieldset.enums.values()).map(column => column.type); if (enums.length > 0) { source.addImportDeclaration({ moduleSpecifier: '../enums.js', namedImports: enums }); } for (const column of fieldset.fieldsets.values()) { source.addImportDeclaration({ moduleSpecifier: `../${column.type}/types.js`, namedImports: [column.type] }); } source.addTypeAlias({ isExported: true, name: fieldset.title, type: (`{ ${columns.filter(column => !!column.typemap.model || !!column.enum || !!column.fieldset).map(column => (`${column.name}${!column.required ? '?' : ''}: ${column.typemap.model || column.type}${column.multiple ? '[]' : ''}`)).join(',\n')} }`) }); const inputs = columns .filter(column => !column.generated) .filter(column => [ ...Object.keys(typemap.model), ...fieldset.enums.map(column => column.type), ...fieldset.fieldsets.map(column => column.fieldset?.title) ].includes(column.type)); source.addTypeAlias({ isExported: true, name: `${fieldset.title}Input`, type: (`{ ${inputs.map(column => (`${column.name}${!column.required || typeof column.default !== 'undefined' ? '?' : ''}: ${column.typemap.model || column.type}${column.multiple ? '[]' : ''}`)).join(',\n')} }`) }); } ;