@vegardit/prisma-generator-nestjs-dto
Version:
Generates DTO and Entity classes from Prisma Schema for NestJS
81 lines (80 loc) • 3.58 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 helpers_1 = require("../helpers");
const computeCreateDtoParams = ({ model, allModels, templateHelpers, }) => {
let hasEnum = false;
const imports = [];
const apiExtraModels = [];
const extraClasses = [];
const relationScalarFields = (0, helpers_1.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 = (0, helpers_1.generateRelationInput)({
field,
model,
allModels,
templateHelpers,
preAndSuffixClassName: templateHelpers.createDtoName,
canCreateAnnotation: annotations_1.DTO_RELATION_CAN_CRAEATE_ON_CREATE,
canConnectAnnotation: annotations_1.DTO_RELATION_CAN_CONNECT_ON_CREATE,
});
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;
(0, helpers_1.concatIntoArray)(relationInputType.imports, imports);
(0, helpers_1.concatIntoArray)(relationInputType.generatedClasses, extraClasses);
(0, helpers_1.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, (0, helpers_1.mapDMMFToParsedField)(field, overrides)];
}, []);
if (apiExtraModels.length || hasEnum) {
const destruct = [];
if (apiExtraModels.length)
destruct.push('ApiExtraModels');
if (hasEnum)
destruct.push('ApiProperty');
imports.unshift({ from: '@nestjs/swagger', destruct });
}
const importPrismaClient = (0, helpers_1.makeImportsFromPrismaClient)(fields);
if (importPrismaClient)
imports.unshift(importPrismaClient);
return {
model,
fields,
imports: (0, helpers_1.zipImportStatementParams)(imports),
extraClasses,
apiExtraModels,
};
};
exports.computeCreateDtoParams = computeCreateDtoParams;