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 path = require('path');
const camelCase = require('camelcase');
const config = require('../config');
const logger = require('../logger');
@ -55,44 +56,25 @@ class Controller {
return uploadedFileName;
}
// static collectFiles(request) {
// logger.info('Checking if files are expected in schema');
// const requestFiles = {};
// if (request.openapi.schema.requestBody !== undefined) {
// const [contentType] = request.headers['content-type'].split(';');
// if (contentType === 'multipart/form-data') {
// const contentSchema = request.openapi.schema.requestBody.content[contentType].schema;
// Object.entries(contentSchema.properties).forEach(([name, property]) => {
// if (property.type === 'string' && ['binary', 'base64'].indexOf(property.format) > -1) {
// const fileObject = request.files.find(file => file.fieldname === name);
// 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 getRequestBodyName(request) {
const codeGenDefinedBodyName = request.openapi.schema['x-codegen-request-body-name'];
if (codeGenDefinedBodyName !== undefined) {
return codeGenDefinedBodyName;
}
const refObjectPath = request.openapi.schema.requestBody.content['application/json'].schema.$ref;
if (refObjectPath !== undefined && refObjectPath.length > 0) {
return (refObjectPath.substr(refObjectPath.lastIndexOf('/') + 1));
}
return 'body';
}
static collectRequestParams(request) {
const requestParams = {};
if (request.openapi.schema.requestBody !== undefined) {
const { content } = request.openapi.schema.requestBody;
if (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;
}
const requestBodyName = camelCase(this.getRequestBodyName(request));
requestParams[requestBodyName] = request.body;
} else if (content['multipart/form-data'] !== undefined) {
Object.keys(content['multipart/form-data'].schema.properties).forEach(
(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) => {
if (param.in === 'path') {
requestParams[param.name] = request.openapi.pathParams[param.name];

View File

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