File Explorer

1'use strict'2 3// const { parseHeaders } = require('../core/util')4const DecoratorHandler = require('../handler/decorator-handler')5const { ResponseError } = require('../core/errors')6 7class ResponseErrorHandler extends DecoratorHandler {8  #statusCode9  #contentType10  #decoder11  #headers12  #body13 14  constructor (_opts, { handler }) {15    super(handler)16  }17 18  #checkContentType (contentType) {19    return (this.#contentType ?? '').indexOf(contentType) === 020  }21 22  onRequestStart (controller, context) {23    this.#statusCode = 024    this.#contentType = null25    this.#decoder = null26    this.#headers = null27    this.#body = ''28 29    return super.onRequestStart(controller, context)30  }31 32  onResponseStart (controller, statusCode, headers, statusMessage) {33    this.#statusCode = statusCode34    this.#headers = headers35    this.#contentType = headers['content-type']36 37    if (this.#statusCode < 400) {38      return super.onResponseStart(controller, statusCode, headers, statusMessage)39    }40 41    if (this.#checkContentType('application/json') || this.#checkContentType('text/plain')) {42      this.#decoder = new TextDecoder('utf-8')43    }44  }45 46  onResponseData (controller, chunk) {47    if (this.#statusCode < 400) {48      return super.onResponseData(controller, chunk)49    }50 51    this.#body += this.#decoder?.decode(chunk, { stream: true }) ?? ''52  }53 54  onResponseEnd (controller, trailers) {55    if (this.#statusCode >= 400) {56      this.#body += this.#decoder?.decode(undefined, { stream: false }) ?? ''57 58      if (this.#checkContentType('application/json')) {59        try {60          this.#body = JSON.parse(this.#body)61        } catch {62          // Do nothing...63        }64      }65 66      let err67      const stackTraceLimit = Error.stackTraceLimit68      Error.stackTraceLimit = 069      try {70        err = new ResponseError('Response Error', this.#statusCode, {71          body: this.#body,72          headers: this.#headers73        })74      } finally {75        Error.stackTraceLimit = stackTraceLimit76      }77 78      super.onResponseError(controller, err)79    } else {80      super.onResponseEnd(controller, trailers)81    }82  }83 84  onResponseError (controller, err) {85    super.onResponseError(controller, err)86  }87}88 89module.exports = () => {90  return (dispatch) => {91    return function Intercept (opts, handler) {92      return dispatch(opts, new ResponseErrorHandler(opts, { handler }))93    }94  }95}96