kirbe/model/KirbeResponse.js

module.exports = class KirbeResponse {
  constructor(res) {
    this.coreRes = res;
    this.headers = {};
    
    this.statusCode    = 200;
    this.statusMessage = null;

    this.data = Buffer.alloc(0);
  }

  body(body) {
    if (typeof body === 'object' && !Buffer.isBuffer(body)) {
	  if (!this.headers['content-type']) this.headers['content-type'] = 'application/json'; this.data = JSON.stringify(body);
    } else this.data = body;
    return this;
  }

  header(a, b) {
    if (typeof a === 'object') Object.keys(a).forEach((v) => this.headers[v.toLowerCase()] = a[v]);
    else this.headers[a.toLowerCase()] = b;
    return this;
  }

  status(code, message) {
    this.statusCode    = code;
    this.statusMessage = message;
    return this;
  }

  writeHead(status, headers) {
    if (status)  this.status(status);
    if (headers) this.header(headers);
  }

  end(data) {
    if (data) this.body(data);
    if (this.statusMessage) this.coreRes.writeHead(this.statusCode, this.statusMessage, this.headers);
    else this.coreRes.writeHead(this.statusCode, this.headers);
    this.coreRes.end(this.data);
    return this;
  }
};