@nestjs/swagger
Version:
Nest - modern, fast, powerful node.js web framework (@swagger)
142 lines (141 loc) • 5.22 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.DocumentBuilder = void 0;
const common_1 = require("@nestjs/common");
const lodash_1 = require("lodash");
const document_base_1 = require("./fixtures/document.base");
const global_parameters_storage_1 = require("./storages/global-parameters.storage");
const global_responses_storage_1 = require("./storages/global-responses.storage");
class DocumentBuilder {
constructor() {
this.logger = new common_1.Logger(DocumentBuilder.name);
this.document = (0, document_base_1.buildDocumentBase)();
}
setTitle(title) {
this.document.info.title = title;
return this;
}
setDescription(description) {
this.document.info.description = description;
return this;
}
setVersion(version) {
this.document.info.version = version;
return this;
}
setTermsOfService(termsOfService) {
this.document.info.termsOfService = termsOfService;
return this;
}
setContact(name, url, email) {
this.document.info.contact = { name, url, email };
return this;
}
setLicense(name, url) {
this.document.info.license = { name, url };
return this;
}
setOpenAPIVersion(version) {
if (version.match(/^\d\.\d\.\d$/)) {
this.document.openapi = version;
}
else {
this.logger.warn('The OpenApi version is invalid. Expecting format "x.x.x"');
}
return this;
}
addServer(url, description, variables) {
this.document.servers.push({ url, description, variables });
return this;
}
setExternalDoc(description, url) {
this.document.externalDocs = { description, url };
return this;
}
setBasePath(path) {
this.logger.warn('The "setBasePath" method has been deprecated. Now, a global prefix is populated automatically. If you want to ignore it, take a look here: https://docs.nestjs.com/recipes/swagger#global-prefix. Alternatively, you can use "addServer" method to set up multiple different paths.');
return this;
}
addTag(name, description = '', externalDocs) {
this.document.tags = this.document.tags.concat((0, lodash_1.pickBy)({
name,
description,
externalDocs
}, (0, lodash_1.negate)(lodash_1.isUndefined)));
return this;
}
addExtension(extensionKey, extensionProperties, location = 'root') {
if (!extensionKey.startsWith('x-')) {
throw new Error('Extension key is not prefixed. Please ensure you prefix it with `x-`.');
}
if (location === 'root') {
this.document[extensionKey] = (0, lodash_1.clone)(extensionProperties);
}
else {
this.document[location][extensionKey] = (0, lodash_1.clone)(extensionProperties);
}
return this;
}
addSecurity(name, options) {
this.document.components.securitySchemes = Object.assign(Object.assign({}, (this.document.components.securitySchemes || {})), { [name]: options });
return this;
}
addGlobalResponse(...respones) {
const groupedByStatus = respones.reduce((acc, response) => {
const { status = 'default' } = response;
acc[status] = (0, lodash_1.omit)(response, 'status');
return acc;
}, {});
global_responses_storage_1.GlobalResponsesStorage.add(groupedByStatus);
return this;
}
addGlobalParameters(...parameters) {
global_parameters_storage_1.GlobalParametersStorage.add(...parameters);
return this;
}
addSecurityRequirements(name, requirements = []) {
let securityRequirement;
if ((0, lodash_1.isString)(name)) {
securityRequirement = { [name]: requirements };
}
else {
securityRequirement = name;
}
this.document.security = (this.document.security || []).concat(Object.assign({}, securityRequirement));
return this;
}
addBearerAuth(options = {
type: 'http'
}, name = 'bearer') {
this.addSecurity(name, Object.assign({ scheme: 'bearer', bearerFormat: 'JWT' }, options));
return this;
}
addOAuth2(options = {
type: 'oauth2'
}, name = 'oauth2') {
this.addSecurity(name, Object.assign({ type: 'oauth2', flows: {} }, options));
return this;
}
addApiKey(options = {
type: 'apiKey'
}, name = 'api_key') {
this.addSecurity(name, Object.assign({ type: 'apiKey', in: 'header', name }, options));
return this;
}
addBasicAuth(options = {
type: 'http'
}, name = 'basic') {
this.addSecurity(name, Object.assign({ type: 'http', scheme: 'basic' }, options));
return this;
}
addCookieAuth(cookieName = 'connect.sid', options = {
type: 'apiKey'
}, securityName = 'cookie') {
this.addSecurity(securityName, Object.assign({ type: 'apiKey', in: 'cookie', name: cookieName }, options));
return this;
}
build() {
return this.document;
}
}
exports.DocumentBuilder = DocumentBuilder;