File Explorer

1'use strict'2 3const Dispatcher = require('./dispatcher')4const UnwrapHandler = require('../handler/unwrap-handler')5const {6  ClientDestroyedError,7  ClientClosedError,8  InvalidArgumentError9} = require('../core/errors')10const { kDestroy, kClose, kClosed, kDestroyed, kDispatch } = require('../core/symbols')11 12const kOnDestroyed = Symbol('onDestroyed')13const kOnClosed = Symbol('onClosed')14const kWebSocketOptions = Symbol('webSocketOptions')15 16class DispatcherBase extends Dispatcher {17  /** @type {boolean} */18  [kDestroyed] = false;19 20  /** @type {Array<Function|null} */21  [kOnDestroyed] = null;22 23  /** @type {boolean} */24  [kClosed] = false;25 26  /** @type {Array<Function>|null} */27  [kOnClosed] = null28 29  /**30   * @param {import('../../types/dispatcher').DispatcherOptions} [opts]31   */32  constructor (opts) {33    super()34    this[kWebSocketOptions] = opts?.webSocket ?? {}35  }36 37  /**38   * @returns {import('../../types/dispatcher').WebSocketOptions}39   */40  get webSocketOptions () {41    return {42      maxFragments: this[kWebSocketOptions].maxFragments ?? 131072,43      maxPayloadSize: this[kWebSocketOptions].maxPayloadSize ?? 128 * 1024 * 1024 // 128 MB default44    }45  }46 47  /** @returns {boolean} */48  get destroyed () {49    return this[kDestroyed]50  }51 52  /** @returns {boolean} */53  get closed () {54    return this[kClosed]55  }56 57  close (callback) {58    if (callback === undefined) {59      return new Promise((resolve, reject) => {60        this.close((err, data) => {61          return err ? reject(err) : resolve(data)62        })63      })64    }65 66    if (typeof callback !== 'function') {67      throw new InvalidArgumentError('invalid callback')68    }69 70    if (this[kDestroyed]) {71      const err = new ClientDestroyedError()72      queueMicrotask(() => callback(err, null))73      return74    }75 76    if (this[kClosed]) {77      if (this[kOnClosed]) {78        this[kOnClosed].push(callback)79      } else {80        queueMicrotask(() => callback(null, null))81      }82      return83    }84 85    this[kClosed] = true86    this[kOnClosed] ??= []87    this[kOnClosed].push(callback)88 89    const onClosed = () => {90      const callbacks = this[kOnClosed]91      this[kOnClosed] = null92      for (let i = 0; i < callbacks.length; i++) {93        callbacks[i](null, null)94      }95    }96 97    // Should not error.98    this[kClose]()99      .then(() => this.destroy())100      .then(() => queueMicrotask(onClosed))101  }102 103  destroy (err, callback) {104    if (typeof err === 'function') {105      callback = err106      err = null107    }108 109    if (callback === undefined) {110      return new Promise((resolve, reject) => {111        this.destroy(err, (err, data) => {112          return err ? reject(err) : resolve(data)113        })114      })115    }116 117    if (typeof callback !== 'function') {118      throw new InvalidArgumentError('invalid callback')119    }120 121    if (this[kDestroyed]) {122      if (this[kOnDestroyed]) {123        this[kOnDestroyed].push(callback)124      } else {125        queueMicrotask(() => callback(null, null))126      }127      return128    }129 130    if (!err) {131      err = new ClientDestroyedError()132    }133 134    this[kDestroyed] = true135    this[kOnDestroyed] ??= []136    this[kOnDestroyed].push(callback)137 138    const onDestroyed = () => {139      const callbacks = this[kOnDestroyed]140      this[kOnDestroyed] = null141      for (let i = 0; i < callbacks.length; i++) {142        callbacks[i](null, null)143      }144    }145 146    // Should not error.147    this[kDestroy](err)148      .then(() => queueMicrotask(onDestroyed))149  }150 151  dispatch (opts, handler) {152    if (!handler || typeof handler !== 'object') {153      throw new InvalidArgumentError('handler must be an object')154    }155 156    handler = UnwrapHandler.unwrap(handler)157 158    try {159      if (!opts || typeof opts !== 'object') {160        throw new InvalidArgumentError('opts must be an object.')161      }162 163      if (this[kDestroyed] || this[kOnDestroyed]) {164        throw new ClientDestroyedError()165      }166 167      if (this[kClosed]) {168        throw new ClientClosedError()169      }170 171      return this[kDispatch](opts, handler)172    } catch (err) {173      if (typeof handler.onError !== 'function') {174        throw err175      }176 177      handler.onError(err)178 179      return false180    }181  }182}183 184module.exports = DispatcherBase185