@nestjs/swagger
Version:
Nest - modern, fast, powerful node.js web framework (@swagger)
51 lines (50 loc) • 2.57 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 plugin_constants_1 = require("../plugin/plugin-constants");
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);
});
if (PickTypeClass[plugin_constants_1.METADATA_FACTORY_NAME]) {
const pluginMetadata = PickTypeClass[plugin_constants_1.METADATA_FACTORY_NAME]();
Object.keys(pluginMetadata).forEach((key) => {
if (!fields.includes(key)) {
const decoratorFactory = (0, decorators_1.ApiProperty)(pluginMetadata[key]);
decoratorFactory(PickTypeClass.prototype, key);
}
});
}
}
applyFields(fields);
metadata_loader_1.MetadataLoader.addRefreshHook(() => {
const fields = modelPropertiesAccessor
.getModelProperties(classRef.prototype)
.filter((item) => keys.includes(item));
applyFields(fields);
});
return PickTypeClass;
}