egg/app/extend/response.js

'use strict';

const contentType = require('mime-types').contentType;
const isJSON = require('koa-is-json');
const LRU = require('ylru');

const REAL_STATUS = Symbol('Context#realStatus');

const typeLRUCache = new LRU(100);
function getType(type) {
  let mimeType = typeLRUCache.get(type);
  if (!mimeType) {
    mimeType = contentType(type);
    typeLRUCache.set(type, mimeType);
  }
  return mimeType;
}

module.exports = {
  set length(n) {
    // copy from koa
    // change header name to lower case
    this.set('content-length', n);
  },

  get length() {
    // copy from koa
    const len = this.header['content-length'];
    const body = this.body;

    if (len == null) {
      if (!body) return;
      if (typeof body === 'string') return Buffer.byteLength(body);
      if (Buffer.isBuffer(body)) return body.length;
      if (isJSON(body)) return Buffer.byteLength(JSON.stringify(body));
      return;
    }

    return parseInt(len, 10);
  },

  set type(type) {
    // copy from koa
    // Different:
    //  - change header name to lower case
    //  - improve mimeType detect by local lru cache
    type = getType(type);
    if (type) {
      this.set('content-type', type);
    } else {
      this.remove('content-type');
    }
  },

  get type() {
    // copy from koa
    const type = this.get('content-type');
    if (!type) return '';
    return type.split(';')[0];
  },

  /**
   * read response real status code.
   *
   * e.g.: Using 302 status redirect to the global error page
   * instead of show current 500 status page.
   * And access log should save 500 not 302,
   * then the `realStatus` can help us find out the real status code.
   * @member {Number} Context#realStatus
   */
  get realStatus() {
    if (this[REAL_STATUS]) {
      return this[REAL_STATUS];
    }
    return this.status;
  },

  /**
   * set response real status code.
   *
   * @member {Void} Response#realStatus
   * @param {Number} status the real status code
   */
  set realStatus(status) {
    this[REAL_STATUS] = status;
  },
};