Pass body to service (#6041)

* Fixed a bug where request body names were not being transferred appropriately based on openapi definitions

* Added changes made to output samples

* Fixed formatting on Controller.mustache

* Added sample code generated after last change
This commit is contained in:
YishTish 2020-06-19 15:44:08 +03:00 committed by GitHub
parent 93bd8571d3
commit 5cdc9e9e35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 59 additions and 109 deletions

View File

@ -1,5 +1,6 @@
const fs = require('fs'); const fs = require('fs');
const path = require('path'); const path = require('path');
const camelCase = require('camelcase');
const config = require('../config'); const config = require('../config');
const logger = require('../logger'); const logger = require('../logger');
@ -55,44 +56,25 @@ class Controller {
return uploadedFileName; return uploadedFileName;
} }
// static collectFiles(request) { static getRequestBodyName(request) {
// logger.info('Checking if files are expected in schema'); const codeGenDefinedBodyName = request.openapi.schema['x-codegen-request-body-name'];
// const requestFiles = {}; if (codeGenDefinedBodyName !== undefined) {
// if (request.openapi.schema.requestBody !== undefined) { return codeGenDefinedBodyName;
// const [contentType] = request.headers['content-type'].split(';'); }
// if (contentType === 'multipart/form-data') { const refObjectPath = request.openapi.schema.requestBody.content['application/json'].schema.$ref;
// const contentSchema = request.openapi.schema.requestBody.content[contentType].schema; if (refObjectPath !== undefined && refObjectPath.length > 0) {
// Object.entries(contentSchema.properties).forEach(([name, property]) => { return (refObjectPath.substr(refObjectPath.lastIndexOf('/') + 1));
// if (property.type === 'string' && ['binary', 'base64'].indexOf(property.format) > -1) { }
// const fileObject = request.files.find(file => file.fieldname === name); return 'body';
// const fileArray = fileObject.originalname.split('.'); }
// const extension = fileArray.pop();
// fileArray.push(`_${Date.now()}`);
// const uploadedFileName = `${fileArray.join('')}.${extension}`;
// fs.renameSync(path.join(config.FILE_UPLOAD_PATH, fileObject.filename),
// path.join(config.FILE_UPLOAD_PATH, uploadedFileName));
// requestFiles[name] = uploadedFileName;
// }
// });
// } else if (request.openapi.schema.requestBody.content[contentType] !== undefined
// && request.files !== undefined) {
// [request.body] = request.files;
// }
// }
// return requestFiles;
// }
static collectRequestParams(request) { static collectRequestParams(request) {
const requestParams = {}; const requestParams = {};
if (request.openapi.schema.requestBody !== undefined) { if (request.openapi.schema.requestBody !== undefined) {
const { content } = request.openapi.schema.requestBody; const { content } = request.openapi.schema.requestBody;
if (content['application/json'] !== undefined) { if (content['application/json'] !== undefined) {
const schema = request.openapi.schema.requestBody.content['application/json']; const requestBodyName = camelCase(this.getRequestBodyName(request));
if (schema.$ref) { requestParams[requestBodyName] = request.body;
requestParams[schema.$ref.substr(schema.$ref.lastIndexOf('.'))] = request.body;
} else {
requestParams.body = request.body;
}
} else if (content['multipart/form-data'] !== undefined) { } else if (content['multipart/form-data'] !== undefined) {
Object.keys(content['multipart/form-data'].schema.properties).forEach( Object.keys(content['multipart/form-data'].schema.properties).forEach(
(property) => { (property) => {
@ -106,14 +88,7 @@ class Controller {
); );
} }
} }
// if (request.openapi.schema.requestBody.content['application/json'] !== undefined) {
// const schema = request.openapi.schema.requestBody.content['application/json'];
// if (schema.$ref) {
// requestParams[schema.$ref.substr(schema.$ref.lastIndexOf('.'))] = request.body;
// } else {
// requestParams.body = request.body;
// }
// }
request.openapi.schema.parameters.forEach((param) => { request.openapi.schema.parameters.forEach((param) => {
if (param.in === 'path') { if (param.in === 'path') {
requestParams[param.name] = request.openapi.pathParams[param.name]; requestParams[param.name] = request.openapi.pathParams[param.name];

View File

@ -1,5 +1,6 @@
const fs = require('fs'); const fs = require('fs');
const path = require('path'); const path = require('path');
const camelCase = require('camelcase');
const config = require('../config'); const config = require('../config');
const logger = require('../logger'); const logger = require('../logger');
@ -55,44 +56,25 @@ class Controller {
return uploadedFileName; return uploadedFileName;
} }
// static collectFiles(request) { static getRequestBodyName(request) {
// logger.info('Checking if files are expected in schema'); const codeGenDefinedBodyName = request.openapi.schema['x-codegen-request-body-name'];
// const requestFiles = {}; if (codeGenDefinedBodyName !== undefined) {
// if (request.openapi.schema.requestBody !== undefined) { return codeGenDefinedBodyName;
// const [contentType] = request.headers['content-type'].split(';'); }
// if (contentType === 'multipart/form-data') { const refObjectPath = request.openapi.schema.requestBody.content['application/json'].schema.$ref;
// const contentSchema = request.openapi.schema.requestBody.content[contentType].schema; if (refObjectPath !== undefined && refObjectPath.length > 0) {
// Object.entries(contentSchema.properties).forEach(([name, property]) => { return (refObjectPath.substr(refObjectPath.lastIndexOf('/') + 1));
// if (property.type === 'string' && ['binary', 'base64'].indexOf(property.format) > -1) { }
// const fileObject = request.files.find(file => file.fieldname === name); return 'body';
// const fileArray = fileObject.originalname.split('.'); }
// const extension = fileArray.pop();
// fileArray.push(`_${Date.now()}`);
// const uploadedFileName = `${fileArray.join('')}.${extension}`;
// fs.renameSync(path.join(config.FILE_UPLOAD_PATH, fileObject.filename),
// path.join(config.FILE_UPLOAD_PATH, uploadedFileName));
// requestFiles[name] = uploadedFileName;
// }
// });
// } else if (request.openapi.schema.requestBody.content[contentType] !== undefined
// && request.files !== undefined) {
// [request.body] = request.files;
// }
// }
// return requestFiles;
// }
static collectRequestParams(request) { static collectRequestParams(request) {
const requestParams = {}; const requestParams = {};
if (request.openapi.schema.requestBody !== undefined) { if (request.openapi.schema.requestBody !== undefined) {
const { content } = request.openapi.schema.requestBody; const { content } = request.openapi.schema.requestBody;
if (content['application/json'] !== undefined) { if (content['application/json'] !== undefined) {
const schema = request.openapi.schema.requestBody.content['application/json']; const requestBodyName = camelCase(this.getRequestBodyName(request));
if (schema.$ref) { requestParams[requestBodyName] = request.body;
requestParams[schema.$ref.substr(schema.$ref.lastIndexOf('.'))] = request.body;
} else {
requestParams.body = request.body;
}
} else if (content['multipart/form-data'] !== undefined) { } else if (content['multipart/form-data'] !== undefined) {
Object.keys(content['multipart/form-data'].schema.properties).forEach( Object.keys(content['multipart/form-data'].schema.properties).forEach(
(property) => { (property) => {
@ -106,14 +88,7 @@ class Controller {
); );
} }
} }
// if (request.openapi.schema.requestBody.content['application/json'] !== undefined) {
// const schema = request.openapi.schema.requestBody.content['application/json'];
// if (schema.$ref) {
// requestParams[schema.$ref.substr(schema.$ref.lastIndexOf('.'))] = request.body;
// } else {
// requestParams.body = request.body;
// }
// }
request.openapi.schema.parameters.forEach((param) => { request.openapi.schema.parameters.forEach((param) => {
if (param.in === 'path') { if (param.in === 'path') {
requestParams[param.name] = request.openapi.pathParams[param.name]; requestParams[param.name] = request.openapi.pathParams[param.name];