@nestjs/swagger
Version:
Nest - modern, fast, powerful node.js web framework (@swagger)
23 lines (22 loc) • 1.07 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
const merge_options_1 = require("./merge-options");
const controller_class_visitor_1 = require("./visitors/controller-class.visitor");
const model_class_visitor_1 = require("./visitors/model-class.visitor");
const modelClassVisitor = new model_class_visitor_1.ModelClassVisitor();
const controllerClassVisitor = new controller_class_visitor_1.ControllerClassVisitor();
const isFilenameMatched = (patterns, filename) => patterns.some(path => filename.includes(path));
exports.before = (options, program) => {
options = merge_options_1.mergePluginOptions(options);
return (ctx) => {
return (sf) => {
if (isFilenameMatched(options.dtoFileNameSuffix, sf.fileName)) {
return modelClassVisitor.visit(sf, ctx, program, options);
}
if (isFilenameMatched(options.controllerFileNameSuffix, sf.fileName)) {
return controllerClassVisitor.visit(sf, ctx, program);
}
return sf;
};
};
};