@nestjs/swagger
Version:
Nest - modern, fast, powerful node.js web framework (@swagger)
62 lines (61 loc) • 2.38 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.isEnumMetadata = exports.isEnumDefined = exports.isEnumArray = exports.addEnumSchema = exports.addEnumArraySchema = exports.getEnumType = exports.getEnumValues = void 0;
const lodash_1 = require("lodash");
function getEnumValues(enumType) {
if (Array.isArray(enumType)) {
return enumType;
}
if (typeof enumType !== 'object') {
return [];
}
const values = [];
const uniqueValues = {};
for (const key in enumType) {
const value = enumType[key];
if (!uniqueValues.hasOwnProperty(value) &&
!uniqueValues.hasOwnProperty(key)) {
values.push(value);
uniqueValues[value] = value;
}
}
return values;
}
exports.getEnumValues = getEnumValues;
function getEnumType(values) {
const hasString = values.filter(lodash_1.isString).length > 0;
return hasString ? 'string' : 'number';
}
exports.getEnumType = getEnumType;
function addEnumArraySchema(paramDefinition, decoratorOptions) {
const paramSchema = paramDefinition.schema || {};
paramDefinition.schema = paramSchema;
paramSchema.type = 'array';
delete paramDefinition.isArray;
const enumValues = getEnumValues(decoratorOptions.enum);
paramSchema.items = {
type: getEnumType(enumValues),
enum: enumValues
};
if (decoratorOptions.enumName) {
paramDefinition.enumName = decoratorOptions.enumName;
}
}
exports.addEnumArraySchema = addEnumArraySchema;
function addEnumSchema(paramDefinition, decoratorOptions) {
const paramSchema = paramDefinition.schema || {};
const enumValues = getEnumValues(decoratorOptions.enum);
paramDefinition.schema = paramSchema;
paramSchema.enum = enumValues;
paramSchema.type = getEnumType(enumValues);
if (decoratorOptions.enumName) {
paramDefinition.enumName = decoratorOptions.enumName;
}
}
exports.addEnumSchema = addEnumSchema;
const isEnumArray = (obj) => obj.isArray && obj.enum;
exports.isEnumArray = isEnumArray;
const isEnumDefined = (obj) => obj.enum;
exports.isEnumDefined = isEnumDefined;
const isEnumMetadata = (metadata) => { var _a; return metadata.enum || (metadata.isArray && ((_a = metadata.items) === null || _a === void 0 ? void 0 : _a['enum'])); };
exports.isEnumMetadata = isEnumMetadata;