think-payload/index.js

Payload for ThinkJS 3.x

const parse = require('./lib/index.js');

module.exports = function(opts = {}) {
  const extendTypes = Object.assign({
    json: [],
    form: [],
    text: [],
    multipart: [],
    xml: []
  }, opts.extendTypes);

  // default json types
  const jsonTypes = [
    'application/json',
    ...extendTypes.json
  ];

  // default form types
  const formTypes = [
    'application/x-www-form-urlencoded',
    ...extendTypes.form
  ];

  // default text types
  const textTypes = [
    'text/plain',
    ...extendTypes.text
  ];

  // default multipart-form types
  const multipartTypes = [
    'multipart/form-data',
    ...extendTypes.multipart
  ];

  // default xml types
  const xmlTypes = [
    'text/xml',
    ...extendTypes.xml
  ];

  return function(ctx, next) {
    if (ctx.request.body !== undefined) return next();

    return parseBody(ctx, opts).then(body => {
      if (body.raw || typeof body === 'string') {
        if (ctx.request.rawBody === undefined) {
          ctx.request.rawBody = body.raw || body;
        }
        delete body.raw;
      }

      ctx.request.body = body;
      return next();
    }, _ => {
      ctx.throw(400, 'Incorrect parameter format');
    });
  };

  function parseBody(ctx, opts) {
    const methods = ['POST', 'PUT', 'DELETE', 'PATCH', 'LINK', 'UNLINK'];

    if (methods.every(method => ctx.method !== method)) {
      return Promise.resolve({});
    }

    if (ctx.request.is(jsonTypes)) {
      return parse.json(ctx, opts);
    }
    if (ctx.request.is(formTypes)) {
      return parse.form(ctx, opts);
    }
    if (ctx.request.is(textTypes)) {
      return parse.text(ctx, opts);
    }
    if (ctx.request.is(multipartTypes)) {
      return parse.multipart(ctx, opts);
    }
    if (ctx.request.is(xmlTypes)) {
      return parse.xml(ctx, opts);
    }

    return Promise.resolve({});
  }
};