File Explorer

1'use strict'2 3const { InvalidArgumentError, RequestAbortedError } = require('../core/errors')4const DecoratorHandler = require('../handler/decorator-handler')5 6class DumpHandler extends DecoratorHandler {7  #maxSize = 1024 * 10248  #dumped = false9  #size = 010  #controller = null11  aborted = false12  reason = false13 14  constructor ({ maxSize, signal }, handler) {15    if (maxSize != null && (!Number.isFinite(maxSize) || maxSize < 1)) {16      throw new InvalidArgumentError('maxSize must be a number greater than 0')17    }18 19    super(handler)20 21    this.#maxSize = maxSize ?? this.#maxSize22    // this.#handler = handler23  }24 25  #abort (reason) {26    this.aborted = true27    this.reason = reason28  }29 30  onRequestStart (controller, context) {31    controller.abort = this.#abort.bind(this)32    this.#controller = controller33 34    return super.onRequestStart(controller, context)35  }36 37  onResponseStart (controller, statusCode, headers, statusMessage) {38    const contentLength = headers['content-length']39 40    if (contentLength != null && contentLength > this.#maxSize) {41      throw new RequestAbortedError(42        `Response size (${contentLength}) larger than maxSize (${43          this.#maxSize44        })`45      )46    }47 48    if (this.aborted === true) {49      return true50    }51 52    return super.onResponseStart(controller, statusCode, headers, statusMessage)53  }54 55  onResponseError (controller, err) {56    if (this.#dumped) {57      return58    }59 60    // On network errors before connect, controller will be null61    err = this.#controller?.reason ?? err62 63    super.onResponseError(controller, err)64  }65 66  onResponseData (controller, chunk) {67    this.#size = this.#size + chunk.length68 69    if (this.#size >= this.#maxSize) {70      this.#dumped = true71 72      if (this.aborted === true) {73        super.onResponseError(controller, this.reason)74      } else {75        super.onResponseEnd(controller, {})76      }77    }78 79    return true80  }81 82  onResponseEnd (controller, trailers) {83    if (this.#dumped) {84      return85    }86 87    if (this.#controller.aborted === true) {88      super.onResponseError(controller, this.reason)89      return90    }91 92    super.onResponseEnd(controller, trailers)93  }94}95 96function createDumpInterceptor (97  { maxSize: defaultMaxSize } = {98    maxSize: 1024 * 102499  }100) {101  return dispatch => {102    return function Intercept (opts, handler) {103      const { dumpMaxSize = defaultMaxSize } = opts104 105      const dumpHandler = new DumpHandler({ maxSize: dumpMaxSize, signal: opts.signal }, handler)106 107      return dispatch(opts, dumpHandler)108    }109  }110}111 112module.exports = createDumpInterceptor113