openapi-generator/samples/server/petstore/nodejs-express-server/controllers/Controller.js

const fs = require('fs');
const path = require('path');
const config = require('../config');
const logger = require('../logger');

class Controller {
  static sendResponse(response, payload) {
    /**
     * The default response-code is 200. We want to allow to change that. in That case,
     * payload will be an object consisting of a code and a payload. If not customized
     * send 200 and the payload as received in this method.
     */
    response.status(payload.code || 200);
    const responsePayload = payload.payload !== undefined ? payload.payload : payload;
    if (responsePayload instanceof Object) {
      response.json(responsePayload);
    } else {
      response.end(responsePayload);
    }
  }

  static sendError(response, error) {
    response.status(error.code || 500);
    if (error.error instanceof Object) {
      response.json(error.error);
    } else {
      response.end(error.error || error.message);
    }
  }

  /**
   * Files have been uploaded to the directory defined by config.js as upload directory
   * Files have a temporary name, that was saved as 'filename' of the file object that is
   * referenced in reuquest.files array.
   * This method finds the file and changes it to the file name that was originally called
   * when it was uploaded. To prevent files from being overwritten, a timestamp is added between
   * the filename and its extension
   * @param request
   * @param fieldName
   * @returns {string}
   */
  static collectFile(request, fieldName) {
    let uploadedFileName = '';
    if (request.files && request.files.length > 0) {
      const fileObject = request.files.find(file => file.fieldname === fieldName);
      if (fileObject) {
        const fileArray = fileObject.originalname.split('.');
        const extension = fileArray.pop();
        fileArray.push(`_${Date.now()}`);
        uploadedFileName = `${fileArray.join('')}.${extension}`;
        fs.renameSync(path.join(config.FILE_UPLOAD_PATH, fileObject.filename),
          path.join(config.FILE_UPLOAD_PATH, uploadedFileName));
      }
    }
    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 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;
        }
      } else if (content['multipart/form-data'] !== undefined) {
        Object.keys(content['multipart/form-data'].schema.properties).forEach(
          (property) => {
            const propertyObject = content['multipart/form-data'].schema.properties[property];
            if (propertyObject.format !== undefined && propertyObject.format === 'binary') {
              requestParams[property] = this.collectFile(request, property);
            } else {
              requestParams[property] = request.body[property];
            }
          },
        );
      }
    }
    // 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];
      } else if (param.in === 'query') {
        requestParams[param.name] = request.query[param.name];
      } else if (param.in === 'header') {
        requestParams[param.name] = request.headers[param.name];
      }
    });
    return requestParams;
  }

  static async handleRequest(request, response, serviceOperation) {
    try {
      const serviceResponse = await serviceOperation(this.collectRequestParams(request));
      Controller.sendResponse(response, serviceResponse);
    } catch (error) {
      Controller.sendError(response, error);
    }
  }
}

module.exports = Controller;