123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300 |
- /*!
- * finalhandler
- * Copyright(c) 2014-2017 Douglas Christopher Wilson
- * MIT Licensed
- */
- 'use strict'
- /**
- * Module dependencies.
- * @private
- */
- var debug = require('debug')('finalhandler')
- var encodeUrl = require('encodeurl')
- var escapeHtml = require('escape-html')
- var onFinished = require('on-finished')
- var parseUrl = require('parseurl')
- var statuses = require('statuses')
- var unpipe = require('unpipe')
- /**
- * Module variables.
- * @private
- */
- var DOUBLE_SPACE_REGEXP = /\x20{2}/g
- var NEWLINE_REGEXP = /\n/g
- /* istanbul ignore next */
- var defer = typeof setImmediate === 'function'
- ? setImmediate
- : function (fn) { process.nextTick(fn.bind.apply(fn, arguments)) }
- var isFinished = onFinished.isFinished
- /**
- * Create a minimal HTML document.
- *
- * @param {string} message
- * @private
- */
- function createHtmlDocument (message) {
- var body = escapeHtml(message)
- .replace(NEWLINE_REGEXP, '<br>')
- .replace(DOUBLE_SPACE_REGEXP, ' ')
- return '<!DOCTYPE html>\n' +
- '<html lang="en">\n' +
- '<head>\n' +
- '<meta charset="utf-8">\n' +
- '<title>Error</title>\n' +
- '</head>\n' +
- '<body>\n' +
- '<pre>' + body + '</pre>\n' +
- '</body>\n' +
- '</html>\n'
- }
- /**
- * Module exports.
- * @public
- */
- module.exports = finalhandler
- /**
- * Create a function to handle the final response.
- *
- * @param {Request} req
- * @param {Response} res
- * @param {Object} [options]
- * @return {Function}
- * @public
- */
- function finalhandler (req, res, options) {
- var opts = options || {}
- // get environment
- var env = opts.env || process.env.NODE_ENV || 'development'
- // get error callback
- var onerror = opts.onerror
- return function (err) {
- var headers
- var msg
- var status
- // ignore 404 on in-flight response
- if (!err && res._header) {
- debug('cannot 404 after headers sent')
- return
- }
- // unhandled error
- if (err) {
- // respect status code from error
- status = getErrorStatusCode(err)
- // respect headers from error
- if (status !== undefined) {
- headers = getErrorHeaders(err)
- }
- // fallback to status code on response
- if (status === undefined) {
- status = getResponseStatusCode(res)
- }
- // get error message
- msg = getErrorMessage(err, status, env)
- } else {
- // not found
- status = 404
- msg = 'Cannot ' + req.method + ' ' + encodeUrl(parseUrl.original(req).pathname)
- }
- debug('default %s', status)
- // schedule onerror callback
- if (err && onerror) {
- defer(onerror, err, req, res)
- }
- // cannot actually respond
- if (res._header) {
- debug('cannot %d after headers sent', status)
- req.socket.destroy()
- return
- }
- // send response
- send(req, res, status, headers, msg)
- }
- }
- /**
- * Get headers from Error object.
- *
- * @param {Error} err
- * @return {object}
- * @private
- */
- function getErrorHeaders (err) {
- if (!err.headers || typeof err.headers !== 'object') {
- return undefined
- }
- var headers = Object.create(null)
- var keys = Object.keys(err.headers)
- for (var i = 0; i < keys.length; i++) {
- var key = keys[i]
- headers[key] = err.headers[key]
- }
- return headers
- }
- /**
- * Get message from Error object, fallback to status message.
- *
- * @param {Error} err
- * @param {number} status
- * @param {string} env
- * @return {string}
- * @private
- */
- function getErrorMessage (err, status, env) {
- var msg
- if (env !== 'production') {
- // use err.stack, which typically includes err.message
- msg = err.stack
- // fallback to err.toString() when possible
- if (!msg && typeof err.toString === 'function') {
- msg = err.toString()
- }
- }
- return msg || statuses[status]
- }
- /**
- * Get status code from Error object.
- *
- * @param {Error} err
- * @return {number}
- * @private
- */
- function getErrorStatusCode (err) {
- // check err.status
- if (typeof err.status === 'number' && err.status >= 400 && err.status < 600) {
- return err.status
- }
- // check err.statusCode
- if (typeof err.statusCode === 'number' && err.statusCode >= 400 && err.statusCode < 600) {
- return err.statusCode
- }
- return undefined
- }
- /**
- * Get status code from response.
- *
- * @param {OutgoingMessage} res
- * @return {number}
- * @private
- */
- function getResponseStatusCode (res) {
- var status = res.statusCode
- // default status code to 500 if outside valid range
- if (typeof status !== 'number' || status < 400 || status > 599) {
- status = 500
- }
- return status
- }
- /**
- * Send response.
- *
- * @param {IncomingMessage} req
- * @param {OutgoingMessage} res
- * @param {number} status
- * @param {object} headers
- * @param {string} message
- * @private
- */
- function send (req, res, status, headers, message) {
- function write () {
- // response body
- var body = createHtmlDocument(message)
- // response status
- res.statusCode = status
- res.statusMessage = statuses[status]
- // response headers
- setHeaders(res, headers)
- // security headers
- res.setHeader('Content-Security-Policy', "default-src 'self'")
- res.setHeader('X-Content-Type-Options', 'nosniff')
- // standard headers
- res.setHeader('Content-Type', 'text/html; charset=utf-8')
- res.setHeader('Content-Length', Buffer.byteLength(body, 'utf8'))
- if (req.method === 'HEAD') {
- res.end()
- return
- }
- res.end(body, 'utf8')
- }
- if (isFinished(req)) {
- write()
- return
- }
- // unpipe everything from the request
- unpipe(req)
- // flush the request
- onFinished(req, write)
- req.resume()
- }
- /**
- * Set response headers from an object.
- *
- * @param {OutgoingMessage} res
- * @param {object} headers
- * @private
- */
- function setHeaders (res, headers) {
- if (!headers) {
- return
- }
- var keys = Object.keys(headers)
- for (var i = 0; i < keys.length; i++) {
- var key = keys[i]
- res.setHeader(key, headers[key])
- }
- }
|