@weverson_na/prisma-generator-nestjs-dto
Version:
Advanced Prisma Generator with Smart Merge v2: Creates DTO and Entity classes with AST-based preservation, intelligent import management, and modular architecture for NestJS
87 lines (86 loc) • 3.95 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.computeCreateDtoParams = void 0;
const annotations_1 = require("../annotations");
const field_classifiers_1 = require("../field-classifiers");
const template_helpers_1 = require("../template-helpers");
const helpers_1 = require("../helpers");
const computeCreateDtoParams = ({ model, allModels, templateHelpers, addExposePropertyDecorator, }) => {
let hasEnum = false;
const imports = [];
const apiExtraModels = [];
const extraClasses = [];
const relationScalarFields = helpers_1.Helpers.getRelationScalars(model.fields);
const relationScalarFieldNames = Object.keys(relationScalarFields);
const fields = model.fields.reduce((result, field) => {
const { name } = field;
const overrides = {};
if ((0, field_classifiers_1.isReadOnly)(field))
return result;
if ((0, field_classifiers_1.isRelation)(field)) {
if (!(0, field_classifiers_1.isAnnotatedWithOneOf)(field, annotations_1.DTO_RELATION_MODIFIERS_ON_CREATE)) {
return result;
}
const relationInputType = helpers_1.Helpers.generateRelationInput({
field,
model,
allModels,
templateHelpers,
preAndSuffixClassName: (name) => templateHelpers.createDtoName(name),
canCreateAnnotation: annotations_1.DTO_RELATION_CAN_CRAEATE_ON_CREATE,
canConnectAnnotation: annotations_1.DTO_RELATION_CAN_CONNECT_ON_CREATE,
addExposePropertyDecorator,
});
const isDtoRelationRequired = (0, field_classifiers_1.isAnnotatedWith)(field, annotations_1.DTO_RELATION_REQUIRED);
if (isDtoRelationRequired)
overrides.isRequired = true;
if (field.isList)
overrides.isRequired = false;
overrides.type = relationInputType.type;
overrides.isList = false;
helpers_1.Helpers.concatIntoArray(relationInputType.imports, imports);
helpers_1.Helpers.concatIntoArray(relationInputType.generatedClasses, extraClasses);
helpers_1.Helpers.concatIntoArray(relationInputType.apiExtraModels, apiExtraModels);
}
if (relationScalarFieldNames.includes(name))
return result;
const isDtoOptional = (0, field_classifiers_1.isAnnotatedWith)(field, annotations_1.DTO_CREATE_OPTIONAL);
if (!isDtoOptional) {
if ((0, field_classifiers_1.isIdWithDefaultValue)(field))
return result;
if ((0, field_classifiers_1.isUpdatedAt)(field))
return result;
if ((0, field_classifiers_1.isRequiredWithDefaultValue)(field))
return result;
}
if (isDtoOptional) {
overrides.isRequired = false;
}
if (field.kind === 'enum')
hasEnum = true;
return [...result, helpers_1.Helpers.mapDMMFToParsedField(field, overrides)];
}, []);
const hasApiPropertyDoc = template_helpers_1.TemplateHelpers.hasSomeApiPropertyDoc(fields);
if (apiExtraModels.length || hasEnum || hasApiPropertyDoc) {
const destruct = [];
if (apiExtraModels.length)
destruct.push('ApiExtraModels');
if (hasEnum)
destruct.push('ApiProperty');
if (hasApiPropertyDoc) {
destruct.push('ApiProperty');
}
imports.unshift({ from: '@nestjs/swagger', destruct });
}
const importPrismaClient = helpers_1.Helpers.makeImportsFromPrismaClient(fields);
if (importPrismaClient)
imports.unshift(importPrismaClient);
return {
model,
fields,
imports: helpers_1.Helpers.zipImportStatementParams(imports),
extraClasses,
apiExtraModels,
};
};
exports.computeCreateDtoParams = computeCreateDtoParams;