@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
89 lines (88 loc) • 4.2 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.CreateDtoParamsComputer = void 0;
const annotations_1 = require("../annotations");
const field_classifiers_1 = require("../field-classifiers");
const base_model_params_computer_1 = require("./base-model-params-computer");
class CreateDtoParamsComputer extends base_model_params_computer_1.BaseModelParamsComputer {
computeParams(model, allModels, addExposePropertyDecorator) {
let hasEnum = false;
const imports = [];
const apiExtraModels = [];
const extraClasses = [];
const relationScalarFields = 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.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.concatIntoArray(relationInputType.imports, imports);
Helpers.concatIntoArray(relationInputType.generatedClasses, extraClasses);
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.mapDMMFToParsedField(field, overrides)];
}, []);
const hasApiPropertyDoc = 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.makeImportsFromPrismaClient(fields);
if (importPrismaClient)
imports.unshift(importPrismaClient);
return {
model,
fields,
imports: Helpers.zipImportStatementParams(imports),
extraClasses,
apiExtraModels,
};
}
;
}
exports.CreateDtoParamsComputer = CreateDtoParamsComputer;