File Explorer

1'use strict'2 3const util = require('../core/util')4const { InvalidArgumentError, RequestAbortedError } = require('../core/errors')5const DecoratorHandler = require('../handler/decorator-handler')6 7class DumpHandler extends DecoratorHandler {8  #maxSize = 1024 * 10249  #abort = null10  #dumped = false11  #aborted = false12  #size = 013  #reason = null14  #handler = null15 16  constructor ({ maxSize }, handler) {17    super(handler)18 19    if (maxSize != null && (!Number.isFinite(maxSize) || maxSize < 1)) {20      throw new InvalidArgumentError('maxSize must be a number greater than 0')21    }22 23    this.#maxSize = maxSize ?? this.#maxSize24    this.#handler = handler25  }26 27  onConnect (abort) {28    this.#abort = abort29 30    this.#handler.onConnect(this.#customAbort.bind(this))31  }32 33  #customAbort (reason) {34    this.#aborted = true35    this.#reason = reason36  }37 38  // TODO: will require adjustment after new hooks are out39  onHeaders (statusCode, rawHeaders, resume, statusMessage) {40    const headers = util.parseHeaders(rawHeaders)41    const contentLength = headers['content-length']42 43    if (contentLength != null && contentLength > this.#maxSize) {44      throw new RequestAbortedError(45        `Response size (${contentLength}) larger than maxSize (${46          this.#maxSize47        })`48      )49    }50 51    if (this.#aborted) {52      return true53    }54 55    return this.#handler.onHeaders(56      statusCode,57      rawHeaders,58      resume,59      statusMessage60    )61  }62 63  onError (err) {64    if (this.#dumped) {65      return66    }67 68    err = this.#reason ?? err69 70    this.#handler.onError(err)71  }72 73  onData (chunk) {74    this.#size = this.#size + chunk.length75 76    if (this.#size >= this.#maxSize) {77      this.#dumped = true78 79      if (this.#aborted) {80        this.#handler.onError(this.#reason)81      } else {82        this.#handler.onComplete([])83      }84    }85 86    return true87  }88 89  onComplete (trailers) {90    if (this.#dumped) {91      return92    }93 94    if (this.#aborted) {95      this.#handler.onError(this.reason)96      return97    }98 99    this.#handler.onComplete(trailers)100  }101}102 103function createDumpInterceptor (104  { maxSize: defaultMaxSize } = {105    maxSize: 1024 * 1024106  }107) {108  return dispatch => {109    return function Intercept (opts, handler) {110      const { dumpMaxSize = defaultMaxSize } =111        opts112 113      const dumpHandler = new DumpHandler(114        { maxSize: dumpMaxSize },115        handler116      )117 118      return dispatch(opts, dumpHandler)119    }120  }121}122 123module.exports = createDumpInterceptor124