stackpress
Version:
Incept is a content management framework.
132 lines (131 loc) • 5.14 kB
JavaScript
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')}
}`)
});
}
;