index.js

'use strict'
const fp = require('fastify-plugin')

const { Errors } = require('./lib/index')

module.exports = fp(
  function fastifyRacePlugin (fastify, globalOpts, next) {
    const controllers = new WeakMap()
    let error

    if (globalOpts != null && typeof globalOpts !== 'object') {
      return next(new Errors.BAD_PARAMS('object', typeof globalOpts))
    }

    globalOpts = Object.assign(
      {},
      { handleOnError: true, onRequestClosed: null },
      globalOpts
    )

    if (typeof globalOpts.handleOnError !== 'boolean') {
      error = new Errors.BAD_PARAMS('boolean', typeof globalOpts.handleOnError)
    } else if (
      globalOpts.onRequestClosed != null &&
      typeof globalOpts.onRequestClosed !== 'function'
    ) {
      error = new Errors.BAD_PARAMS(
        'function',
        typeof globalOpts.onRequestClosed
      )
    }

    fastify.decorateRequest('race', race)
    fastify.addHook('onResponse', fastifyRacingCleaner)

    return next(error)

    function fastifyRacingCleaner (request, _reply, done) {
      if (controllers.has(request)) {
        const { controller, cbs } = controllers.get(request)

        if (controller.signal.aborted === false) {
          for (const cb of cbs) {
            controller.signal.removeEventListener('abort', cb, {
              once: true
            })
          }
        }

        controllers.delete(request)
      }

      done()
    }

    function race (opts = globalOpts) {
      const { raw, id: reqId } = this
      const handleError = typeof opts === 'function' ? true : opts.handleOnError
      const cb = typeof opts === 'function' ? opts : opts.onRequestClosed

      if (controllers.has(this)) {
        const { controller: ctrl, cbs } = controllers.get(this)

        if (ctrl.signal.aborted === true) {
          throw new Errors.ALREADY_ABORTED(reqId)
        }

        if (raw.socket.destroyed === true) {
          throw new Errors.SOCKET_CLOSED(reqId)
        }

        if (cb != null) {
          ctrl.signal.addEventListener('abort', cb, {
            once: true
          })

          controllers.set(this, { controller: ctrl, cbs: cbs.concat(cb) })
        }

        return ctrl.signal
      } else {
        // eslint-disable-next-line no-undef
        const controller = new AbortController()

        if (cb != null) {
          controller.signal.addEventListener('abort', cb, {
            once: true
          })
        }

        if (cb == null) controller.signal.then = theneable.bind(this)

        if (raw.socket.destroyed) {
          throw new Errors.SOCKET_CLOSED(reqId)
        } else {
          raw.once(
            'close',
            function () {
              if (controllers.has(this)) {
                const { controller: ctrl } = controllers.get(this)
                if (ctrl.signal.aborted === false) controller.abort()
              }
            }.bind(this)
          )

          if (handleError === true || cb != null) {
            raw.once(
              'error',
              function (err) {
                if (controllers.has(this)) {
                  const { controller: ctrl } = controllers.get(this)
                  if (ctrl.signal.aborted === false) controller.abort(err)
                }
              }.bind(this)
            )
          }
        }

        controllers.set(this, { controller, cbs: cb != null ? [cb] : [] })

        return controller.signal
      }

      function theneable (resolve, reject) {
        const { controller, cbs } = controllers.get(this)

        if (raw.socket.destroyed === true) {
          return reject(Errors.SOCKET_CLOSED(this.id))
        }

        if (controller.signal.aborted === true) {
          return reject(Errors.ALREADY_ABORTED(this.id))
        }

        try {
          controller.signal.addEventListener('abort', theneableHandler, {
            once: true
          })

          controllers.set(this, {
            controller,
            cbs: cbs.concat(theneableHandler)
          })
        } catch (err) {
          reject(err)
        }

        function theneableHandler (evt) {
          const event = {
            type: evt.type,
            reason: controller.signal?.reason
          }

          resolve(event)
        }
      }
    }
  },
  {
    fastify: '>=3.24.1',
    name: 'fastify-racing'
  }
)