@nestjs/swagger
Version:
Nest - modern, fast, powerful node.js web framework (@swagger)
41 lines (40 loc) • 2.03 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.PickType = PickType;
const mapped_types_1 = require("@nestjs/mapped-types");
const lodash_1 = require("lodash");
const constants_1 = require("../constants");
const decorators_1 = require("../decorators");
const metadata_loader_1 = require("../plugin/metadata-loader");
const model_properties_accessor_1 = require("../services/model-properties-accessor");
const mapped_types_utils_1 = require("./mapped-types.utils");
const modelPropertiesAccessor = new model_properties_accessor_1.ModelPropertiesAccessor();
function PickType(classRef, keys) {
const fields = modelPropertiesAccessor
.getModelProperties(classRef.prototype)
.filter((item) => keys.includes(item));
const isInheritedPredicate = (propertyKey) => keys.includes(propertyKey);
class PickTypeClass {
constructor() {
(0, mapped_types_1.inheritPropertyInitializers)(this, classRef, isInheritedPredicate);
}
}
(0, mapped_types_1.inheritValidationMetadata)(classRef, PickTypeClass, isInheritedPredicate);
(0, mapped_types_1.inheritTransformationMetadata)(classRef, PickTypeClass, isInheritedPredicate);
function applyFields(fields) {
(0, mapped_types_utils_1.clonePluginMetadataFactory)(PickTypeClass, classRef.prototype, (metadata) => (0, lodash_1.pick)(metadata, keys));
fields.forEach((propertyKey) => {
const metadata = Reflect.getMetadata(constants_1.DECORATORS.API_MODEL_PROPERTIES, classRef.prototype, propertyKey);
const decoratorFactory = (0, decorators_1.ApiProperty)(metadata);
decoratorFactory(PickTypeClass.prototype, propertyKey);
});
}
applyFields(fields);
metadata_loader_1.MetadataLoader.addRefreshHook(() => {
const fields = modelPropertiesAccessor
.getModelProperties(classRef.prototype)
.filter((item) => keys.includes(item));
applyFields(fields);
});
return PickTypeClass;
}