File Explorer

1'use strict'2 3const { InvalidArgumentError, SocketError } = require('../core/errors')4const { AsyncResource } = require('node:async_hooks')5const assert = require('node:assert')6const util = require('../core/util')7const { kHTTP2Stream } = require('../core/symbols')8const { addSignal, removeSignal } = require('./abort-signal')9 10class UpgradeHandler extends AsyncResource {11  constructor (opts, callback) {12    if (!opts || typeof opts !== 'object') {13      throw new InvalidArgumentError('invalid opts')14    }15 16    if (typeof callback !== 'function') {17      throw new InvalidArgumentError('invalid callback')18    }19 20    const { signal, opaque, responseHeaders } = opts21 22    if (signal && typeof signal.on !== 'function' && typeof signal.addEventListener !== 'function') {23      throw new InvalidArgumentError('signal must be an EventEmitter or EventTarget')24    }25 26    super('UNDICI_UPGRADE')27 28    this.responseHeaders = responseHeaders || null29    this.opaque = opaque || null30    this.callback = callback31    this.abort = null32    this.context = null33 34    addSignal(this, signal)35  }36 37  onConnect (abort, context) {38    if (this.reason) {39      abort(this.reason)40      return41    }42 43    assert(this.callback)44 45    this.abort = abort46    this.context = null47  }48 49  onHeaders () {50    throw new SocketError('bad upgrade', null)51  }52 53  onUpgrade (statusCode, rawHeaders, socket) {54    assert(socket[kHTTP2Stream] === true ? statusCode === 200 : statusCode === 101)55 56    const { callback, opaque, context } = this57 58    removeSignal(this)59 60    this.callback = null61    const headers = this.responseHeaders === 'raw' ? util.parseRawHeaders(rawHeaders) : util.parseHeaders(rawHeaders)62    this.runInAsyncScope(callback, null, null, {63      headers,64      socket,65      opaque,66      context67    })68  }69 70  onError (err) {71    const { callback, opaque } = this72 73    removeSignal(this)74 75    if (callback) {76      this.callback = null77      queueMicrotask(() => {78        this.runInAsyncScope(callback, null, err, { opaque })79      })80    }81  }82}83 84function upgrade (opts, callback) {85  if (callback === undefined) {86    return new Promise((resolve, reject) => {87      upgrade.call(this, opts, (err, data) => {88        return err ? reject(err) : resolve(data)89      })90    })91  }92 93  try {94    const upgradeHandler = new UpgradeHandler(opts, callback)95    const upgradeOpts = {96      ...opts,97      method: opts.method || 'GET',98      upgrade: opts.protocol || 'Websocket'99    }100 101    this.dispatch(upgradeOpts, upgradeHandler)102  } catch (err) {103    if (typeof callback !== 'function') {104      throw err105    }106    const opaque = opts?.opaque107    queueMicrotask(() => callback(err, { opaque }))108  }109}110 111module.exports = upgrade112