123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318 |
- 'use strict'
- exports.__esModule = true
- exports.processRequest = void 0
- var _util = _interopRequireDefault(require('util'))
- var _busboy = _interopRequireDefault(require('busboy'))
- var _fsCapacitor = require('fs-capacitor')
- var _httpErrors = _interopRequireDefault(require('http-errors'))
- var _objectPath = _interopRequireDefault(require('object-path'))
- var _constants = require('./constants')
- var _ignoreStream = require('./ignoreStream')
- var _isEnumerableObject = require('./isEnumerableObject')
- // istanbul ignore next
- function _interopRequireDefault(obj) {
- return obj && obj.__esModule ? obj : { default: obj }
- }
- class Upload {
- constructor() {
- this.promise = new Promise((resolve, reject) => {
- this.resolve = file => {
- this.file = file
- resolve(file)
- }
- this.reject = reject
- })
- this.promise.catch(() => {})
- }
- }
- const processRequest = (
- request,
- response,
- { maxFieldSize = 1000000, maxFileSize = Infinity, maxFiles = Infinity } = {}
- ) =>
- new Promise((resolve, reject) => {
- let released
- let exitError
- let currentStream
- let operations
- let operationsPath
- let map
- const parser = new _busboy.default({
- headers: request.headers,
- limits: {
- fieldSize: maxFieldSize,
- fields: 2,
- fileSize: maxFileSize,
- files: maxFiles
- }
- })
- const exit = error => {
- if (exitError) return
- exitError = error
- reject(exitError)
- parser.destroy()
- if (currentStream) currentStream.destroy(exitError)
- if (map)
- for (const upload of map.values())
- if (!upload.file) upload.reject(exitError)
- request.unpipe(parser)
- setImmediate(() => {
- request.resume()
- })
- }
- const release = () => {
- // istanbul ignore next
- if (released) return
- released = true
- if (map)
- for (const upload of map.values())
- if (upload.file) upload.file.capacitor.destroy()
- }
- const abort = () => {
- exit(
- (0, _httpErrors.default)(
- 499,
- 'Request disconnected during file upload stream parsing.'
- )
- )
- }
- parser.on(
- 'field',
- (fieldName, value, fieldNameTruncated, valueTruncated) => {
- if (exitError) return
- if (valueTruncated)
- return exit(
- (0, _httpErrors.default)(
- 413,
- `The ‘${fieldName}’ multipart field value exceeds the ${maxFieldSize} byte size limit.`
- )
- )
- switch (fieldName) {
- case 'operations':
- try {
- operations = JSON.parse(value)
- } catch (error) {
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid JSON in the ‘operations’ multipart field (${_constants.SPEC_URL}).`
- )
- )
- }
- if (
- !(0, _isEnumerableObject.isEnumerableObject)(operations) &&
- !Array.isArray(operations)
- )
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid type for the ‘operations’ multipart field (${_constants.SPEC_URL}).`
- )
- )
- operationsPath = (0, _objectPath.default)(operations)
- break
- case 'map': {
- if (!operations)
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Misordered multipart fields; ‘map’ should follow ‘operations’ (${_constants.SPEC_URL}).`
- )
- )
- let parsedMap
- try {
- parsedMap = JSON.parse(value)
- } catch (error) {
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid JSON in the ‘map’ multipart field (${_constants.SPEC_URL}).`
- )
- )
- }
- if (!(0, _isEnumerableObject.isEnumerableObject)(parsedMap))
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid type for the ‘map’ multipart field (${_constants.SPEC_URL}).`
- )
- )
- const mapEntries = Object.entries(parsedMap)
- if (mapEntries.length > maxFiles)
- return exit(
- (0, _httpErrors.default)(
- 413,
- `${maxFiles} max file uploads exceeded.`
- )
- )
- map = new Map()
- for (const [fieldName, paths] of mapEntries) {
- if (!Array.isArray(paths))
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid type for the ‘map’ multipart field entry key ‘${fieldName}’ array (${_constants.SPEC_URL}).`
- )
- )
- map.set(fieldName, new Upload())
- for (const [index, path] of paths.entries()) {
- if (typeof path !== 'string')
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid type for the ‘map’ multipart field entry key ‘${fieldName}’ array index ‘${index}’ value (${_constants.SPEC_URL}).`
- )
- )
- try {
- operationsPath.set(path, map.get(fieldName).promise)
- } catch (error) {
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Invalid object path for the ‘map’ multipart field entry key ‘${fieldName}’ array index ‘${index}’ value ‘${path}’ (${_constants.SPEC_URL}).`
- )
- )
- }
- }
- }
- resolve(operations)
- }
- }
- }
- )
- parser.on('file', (fieldName, stream, filename, encoding, mimetype) => {
- if (exitError) {
- ;(0, _ignoreStream.ignoreStream)(stream)
- return
- }
- if (!map) {
- ;(0, _ignoreStream.ignoreStream)(stream)
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Misordered multipart fields; files should follow ‘map’ (${_constants.SPEC_URL}).`
- )
- )
- }
- currentStream = stream
- stream.on('end', () => {
- currentStream = null
- })
- const upload = map.get(fieldName)
- if (!upload) {
- ;(0, _ignoreStream.ignoreStream)(stream)
- return
- }
- const capacitor = new _fsCapacitor.WriteStream()
- capacitor.on('error', () => {
- stream.unpipe()
- stream.resume()
- })
- stream.on('limit', () => {
- stream.unpipe()
- capacitor.destroy(
- (0, _httpErrors.default)(
- 413,
- `File truncated as it exceeds the ${maxFileSize} byte size limit.`
- )
- )
- })
- stream.on('error', error => {
- stream.unpipe() // istanbul ignore next
- capacitor.destroy(exitError || error)
- })
- stream.pipe(capacitor)
- const file = {
- filename,
- mimetype,
- encoding,
- createReadStream() {
- const error = capacitor.error || (released ? exitError : null)
- if (error) throw error
- return capacitor.createReadStream()
- }
- }
- let capacitorStream
- Object.defineProperty(file, 'stream', {
- get: _util.default.deprecate(function() {
- if (!capacitorStream) capacitorStream = this.createReadStream()
- return capacitorStream
- }, 'File upload property ‘stream’ is deprecated. Use ‘createReadStream()’ instead.')
- })
- Object.defineProperty(file, 'capacitor', {
- value: capacitor
- })
- upload.resolve(file)
- })
- parser.once('filesLimit', () =>
- exit(
- (0, _httpErrors.default)(413, `${maxFiles} max file uploads exceeded.`)
- )
- )
- parser.once('finish', () => {
- request.unpipe(parser)
- request.resume()
- if (!operations)
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Missing multipart field ‘operations’ (${_constants.SPEC_URL}).`
- )
- )
- if (!map)
- return exit(
- (0, _httpErrors.default)(
- 400,
- `Missing multipart field ‘map’ (${_constants.SPEC_URL}).`
- )
- )
- for (const upload of map.values())
- if (!upload.file)
- upload.reject(
- (0, _httpErrors.default)(400, 'File missing in the request.')
- )
- })
- parser.once('error', exit)
- response.once('finish', release)
- response.once('close', release)
- request.once('close', abort)
- request.once('end', () => {
- request.removeListener('close', abort)
- })
- request.pipe(parser)
- })
- exports.processRequest = processRequest
|