|
@@ -2,17 +2,13 @@
|
|
|
|
|
|
const url = require('url')
|
|
|
const { EventEmitter } = require('events')
|
|
|
-const { Readable } = require('stream')
|
|
|
+const { Readable, Writable } = require('stream')
|
|
|
const { app } = require('electron')
|
|
|
const { Session } = process.electronBinding('session')
|
|
|
-const { net, Net } = process.electronBinding('net')
|
|
|
-const { URLRequest } = net
|
|
|
+const { net, Net, _isValidHeaderName, _isValidHeaderValue } = process.electronBinding('net')
|
|
|
+const { URLLoader } = net
|
|
|
|
|
|
-// Net is an EventEmitter.
|
|
|
-Object.setPrototypeOf(Net.prototype, EventEmitter.prototype)
|
|
|
-EventEmitter.call(net)
|
|
|
-
|
|
|
-Object.setPrototypeOf(URLRequest.prototype, EventEmitter.prototype)
|
|
|
+Object.setPrototypeOf(URLLoader.prototype, EventEmitter.prototype)
|
|
|
|
|
|
const kSupportedProtocols = new Set(['http:', 'https:'])
|
|
|
|
|
@@ -40,32 +36,24 @@ const discardableDuplicateHeaders = new Set([
|
|
|
])
|
|
|
|
|
|
class IncomingMessage extends Readable {
|
|
|
- constructor (urlRequest) {
|
|
|
+ constructor (responseHead) {
|
|
|
super()
|
|
|
- this.urlRequest = urlRequest
|
|
|
- this.shouldPush = false
|
|
|
- this.data = []
|
|
|
- this.urlRequest.on('data', (event, chunk) => {
|
|
|
- this._storeInternalData(chunk)
|
|
|
- this._pushInternalData()
|
|
|
- })
|
|
|
- this.urlRequest.on('end', () => {
|
|
|
- this._storeInternalData(null)
|
|
|
- this._pushInternalData()
|
|
|
- })
|
|
|
+ this._shouldPush = false
|
|
|
+ this._data = []
|
|
|
+ this._responseHead = responseHead
|
|
|
}
|
|
|
|
|
|
get statusCode () {
|
|
|
- return this.urlRequest.statusCode
|
|
|
+ return this._responseHead.statusCode
|
|
|
}
|
|
|
|
|
|
get statusMessage () {
|
|
|
- return this.urlRequest.statusMessage
|
|
|
+ return this._responseHead.statusMessage
|
|
|
}
|
|
|
|
|
|
get headers () {
|
|
|
const filteredHeaders = {}
|
|
|
- const rawHeaders = this.urlRequest.rawResponseHeaders
|
|
|
+ const rawHeaders = this._responseHead.headers
|
|
|
Object.keys(rawHeaders).forEach(header => {
|
|
|
if (header in filteredHeaders && discardableDuplicateHeaders.has(header)) {
|
|
|
// do nothing with discardable duplicate headers
|
|
@@ -88,11 +76,11 @@ class IncomingMessage extends Readable {
|
|
|
}
|
|
|
|
|
|
get httpVersionMajor () {
|
|
|
- return this.urlRequest.httpVersionMajor
|
|
|
+ return this._responseHead.httpVersion.major
|
|
|
}
|
|
|
|
|
|
get httpVersionMinor () {
|
|
|
- return this.urlRequest.httpVersionMinor
|
|
|
+ return this._responseHead.httpVersion.minor
|
|
|
}
|
|
|
|
|
|
get rawTrailers () {
|
|
@@ -104,170 +92,197 @@ class IncomingMessage extends Readable {
|
|
|
}
|
|
|
|
|
|
_storeInternalData (chunk) {
|
|
|
- this.data.push(chunk)
|
|
|
+ this._data.push(chunk)
|
|
|
+ this._pushInternalData()
|
|
|
}
|
|
|
|
|
|
_pushInternalData () {
|
|
|
- while (this.shouldPush && this.data.length > 0) {
|
|
|
- const chunk = this.data.shift()
|
|
|
- this.shouldPush = this.push(chunk)
|
|
|
+ while (this._shouldPush && this._data.length > 0) {
|
|
|
+ const chunk = this._data.shift()
|
|
|
+ this._shouldPush = this.push(chunk)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
_read () {
|
|
|
- this.shouldPush = true
|
|
|
+ this._shouldPush = true
|
|
|
this._pushInternalData()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-URLRequest.prototype._emitRequestEvent = function (isAsync, ...rest) {
|
|
|
- if (isAsync) {
|
|
|
- process.nextTick(() => {
|
|
|
- this.clientRequest.emit(...rest)
|
|
|
- })
|
|
|
- } else {
|
|
|
- this.clientRequest.emit(...rest)
|
|
|
+/** Writable stream that buffers up everything written to it. */
|
|
|
+class SlurpStream extends Writable {
|
|
|
+ constructor () {
|
|
|
+ super()
|
|
|
+ this._data = Buffer.alloc(0)
|
|
|
}
|
|
|
-}
|
|
|
-
|
|
|
-URLRequest.prototype._emitResponseEvent = function (isAsync, ...rest) {
|
|
|
- if (isAsync) {
|
|
|
- process.nextTick(() => {
|
|
|
- this._response.emit(...rest)
|
|
|
- })
|
|
|
- } else {
|
|
|
- this._response.emit(...rest)
|
|
|
+ _write (chunk, encoding, callback) {
|
|
|
+ this._data = Buffer.concat([this._data, chunk])
|
|
|
+ callback()
|
|
|
}
|
|
|
+ data () { return this._data }
|
|
|
}
|
|
|
|
|
|
-class ClientRequest extends EventEmitter {
|
|
|
- constructor (options, callback) {
|
|
|
+class ChunkedBodyStream extends Writable {
|
|
|
+ constructor (clientRequest) {
|
|
|
super()
|
|
|
+ this._clientRequest = clientRequest
|
|
|
+ }
|
|
|
|
|
|
- if (!app.isReady()) {
|
|
|
- throw new Error('net module can only be used after app is ready')
|
|
|
- }
|
|
|
-
|
|
|
- if (typeof options === 'string') {
|
|
|
- options = url.parse(options)
|
|
|
+ _write (chunk, encoding, callback) {
|
|
|
+ if (this._downstream) {
|
|
|
+ this._downstream.write(chunk).then(callback, callback)
|
|
|
} else {
|
|
|
- options = Object.assign({}, options)
|
|
|
+ // the contract of _write is that we won't be called again until we call
|
|
|
+ // the callback, so we're good to just save a single chunk.
|
|
|
+ this._pendingChunk = chunk
|
|
|
+ this._pendingCallback = callback
|
|
|
+
|
|
|
+ // The first write to a chunked body stream begins the request.
|
|
|
+ this._clientRequest._startRequest()
|
|
|
}
|
|
|
+ }
|
|
|
|
|
|
- const method = (options.method || 'GET').toUpperCase()
|
|
|
- let urlStr = options.url
|
|
|
+ _final (callback) {
|
|
|
+ this._downstream.done()
|
|
|
+ callback()
|
|
|
+ }
|
|
|
|
|
|
- if (!urlStr) {
|
|
|
- const urlObj = {}
|
|
|
- const protocol = options.protocol || 'http:'
|
|
|
- if (!kSupportedProtocols.has(protocol)) {
|
|
|
- throw new Error('Protocol "' + protocol + '" not supported')
|
|
|
+ startReading (pipe) {
|
|
|
+ if (this._downstream) {
|
|
|
+ throw new Error('two startReading calls???')
|
|
|
+ }
|
|
|
+ this._downstream = pipe
|
|
|
+ if (this._pendingChunk) {
|
|
|
+ const doneWriting = (maybeError) => {
|
|
|
+ const cb = this._pendingCallback
|
|
|
+ delete this._pendingCallback
|
|
|
+ delete this._pendingChunk
|
|
|
+ cb(maybeError)
|
|
|
}
|
|
|
- urlObj.protocol = protocol
|
|
|
-
|
|
|
- if (options.host) {
|
|
|
- urlObj.host = options.host
|
|
|
- } else {
|
|
|
- if (options.hostname) {
|
|
|
- urlObj.hostname = options.hostname
|
|
|
- } else {
|
|
|
- urlObj.hostname = 'localhost'
|
|
|
- }
|
|
|
+ this._downstream.write(this._pendingChunk).then(doneWriting, doneWriting)
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
|
|
|
- if (options.port) {
|
|
|
- urlObj.port = options.port
|
|
|
- }
|
|
|
- }
|
|
|
+function parseOptions (options) {
|
|
|
+ if (typeof options === 'string') {
|
|
|
+ options = url.parse(options)
|
|
|
+ } else {
|
|
|
+ options = { ...options }
|
|
|
+ }
|
|
|
|
|
|
- if (options.path && / /.test(options.path)) {
|
|
|
- // The actual regex is more like /[^A-Za-z0-9\-._~!$&'()*+,;=/:@]/
|
|
|
- // with an additional rule for ignoring percentage-escaped characters
|
|
|
- // but that's a) hard to capture in a regular expression that performs
|
|
|
- // well, and b) possibly too restrictive for real-world usage. That's
|
|
|
- // why it only scans for spaces because those are guaranteed to create
|
|
|
- // an invalid request.
|
|
|
- throw new TypeError('Request path contains unescaped characters')
|
|
|
- }
|
|
|
- const pathObj = url.parse(options.path || '/')
|
|
|
- urlObj.pathname = pathObj.pathname
|
|
|
- urlObj.search = pathObj.search
|
|
|
- urlObj.hash = pathObj.hash
|
|
|
- urlStr = url.format(urlObj)
|
|
|
- }
|
|
|
+ const method = (options.method || 'GET').toUpperCase()
|
|
|
+ let urlStr = options.url
|
|
|
|
|
|
- const redirectPolicy = options.redirect || 'follow'
|
|
|
- if (!['follow', 'error', 'manual'].includes(redirectPolicy)) {
|
|
|
- throw new Error('redirect mode should be one of follow, error or manual')
|
|
|
+ if (!urlStr) {
|
|
|
+ const urlObj = {}
|
|
|
+ const protocol = options.protocol || 'http:'
|
|
|
+ if (!kSupportedProtocols.has(protocol)) {
|
|
|
+ throw new Error('Protocol "' + protocol + '" not supported')
|
|
|
}
|
|
|
+ urlObj.protocol = protocol
|
|
|
|
|
|
- const urlRequestOptions = {
|
|
|
- method: method,
|
|
|
- url: urlStr,
|
|
|
- redirect: redirectPolicy
|
|
|
- }
|
|
|
- if (options.session) {
|
|
|
- if (options.session instanceof Session) {
|
|
|
- urlRequestOptions.session = options.session
|
|
|
+ if (options.host) {
|
|
|
+ urlObj.host = options.host
|
|
|
+ } else {
|
|
|
+ if (options.hostname) {
|
|
|
+ urlObj.hostname = options.hostname
|
|
|
} else {
|
|
|
- throw new TypeError('`session` should be an instance of the Session class')
|
|
|
+ urlObj.hostname = 'localhost'
|
|
|
}
|
|
|
- } else if (options.partition) {
|
|
|
- if (typeof options.partition === 'string') {
|
|
|
- urlRequestOptions.partition = options.partition
|
|
|
- } else {
|
|
|
- throw new TypeError('`partition` should be a string')
|
|
|
+
|
|
|
+ if (options.port) {
|
|
|
+ urlObj.port = options.port
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- const urlRequest = new URLRequest(urlRequestOptions)
|
|
|
+ if (options.path && / /.test(options.path)) {
|
|
|
+ // The actual regex is more like /[^A-Za-z0-9\-._~!$&'()*+,;=/:@]/
|
|
|
+ // with an additional rule for ignoring percentage-escaped characters
|
|
|
+ // but that's a) hard to capture in a regular expression that performs
|
|
|
+ // well, and b) possibly too restrictive for real-world usage. That's
|
|
|
+ // why it only scans for spaces because those are guaranteed to create
|
|
|
+ // an invalid request.
|
|
|
+ throw new TypeError('Request path contains unescaped characters')
|
|
|
+ }
|
|
|
+ const pathObj = url.parse(options.path || '/')
|
|
|
+ urlObj.pathname = pathObj.pathname
|
|
|
+ urlObj.search = pathObj.search
|
|
|
+ urlObj.hash = pathObj.hash
|
|
|
+ urlStr = url.format(urlObj)
|
|
|
+ }
|
|
|
|
|
|
- // Set back and forward links.
|
|
|
- this.urlRequest = urlRequest
|
|
|
- urlRequest.clientRequest = this
|
|
|
+ const redirectPolicy = options.redirect || 'follow'
|
|
|
+ if (!['follow', 'error', 'manual'].includes(redirectPolicy)) {
|
|
|
+ throw new Error('redirect mode should be one of follow, error or manual')
|
|
|
+ }
|
|
|
|
|
|
- // This is a copy of the extra headers structure held by the native
|
|
|
- // net::URLRequest. The main reason is to keep the getHeader API synchronous
|
|
|
- // after the request starts.
|
|
|
- this.extraHeaders = {}
|
|
|
+ if (options.headers != null && typeof options.headers !== 'object') {
|
|
|
+ throw new TypeError('headers must be an object')
|
|
|
+ }
|
|
|
|
|
|
- if (options.headers) {
|
|
|
- for (const key in options.headers) {
|
|
|
- this.setHeader(key, options.headers[key])
|
|
|
- }
|
|
|
+ const urlLoaderOptions = {
|
|
|
+ method: method,
|
|
|
+ url: urlStr,
|
|
|
+ redirectPolicy,
|
|
|
+ extraHeaders: options.headers || {}
|
|
|
+ }
|
|
|
+ for (const [name, value] of Object.entries(urlLoaderOptions.extraHeaders)) {
|
|
|
+ if (!_isValidHeaderName(name)) {
|
|
|
+ throw new Error(`Invalid header name: '${name}'`)
|
|
|
}
|
|
|
+ if (!_isValidHeaderValue(value.toString())) {
|
|
|
+ throw new Error(`Invalid value for header '${name}': '${value}'`)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (options.session) {
|
|
|
+ if (options.session instanceof Session) {
|
|
|
+ urlLoaderOptions.session = options.session
|
|
|
+ } else {
|
|
|
+ throw new TypeError('`session` should be an instance of the Session class')
|
|
|
+ }
|
|
|
+ } else if (options.partition) {
|
|
|
+ if (typeof options.partition === 'string') {
|
|
|
+ urlLoaderOptions.partition = options.partition
|
|
|
+ } else {
|
|
|
+ throw new TypeError('`partition` should be a string')
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return urlLoaderOptions
|
|
|
+}
|
|
|
|
|
|
- // Set when the request uses chunked encoding. Can be switched
|
|
|
- // to true only once and never set back to false.
|
|
|
- this.chunkedEncodingEnabled = false
|
|
|
-
|
|
|
- urlRequest.on('response', () => {
|
|
|
- const response = new IncomingMessage(urlRequest)
|
|
|
- urlRequest._response = response
|
|
|
- this.emit('response', response)
|
|
|
- })
|
|
|
+class ClientRequest extends Writable {
|
|
|
+ constructor (options, callback) {
|
|
|
+ super({ autoDestroy: true })
|
|
|
|
|
|
- urlRequest.on('login', (event, authInfo, callback) => {
|
|
|
- const handled = this.emit('login', authInfo, callback)
|
|
|
- if (!handled) {
|
|
|
- // If there were no listeners, cancel the authentication request.
|
|
|
- callback()
|
|
|
- }
|
|
|
- })
|
|
|
+ if (!app.isReady()) {
|
|
|
+ throw new Error('net module can only be used after app is ready')
|
|
|
+ }
|
|
|
|
|
|
if (callback) {
|
|
|
this.once('response', callback)
|
|
|
}
|
|
|
- }
|
|
|
|
|
|
- get chunkedEncoding () {
|
|
|
- return this.chunkedEncodingEnabled
|
|
|
+ const { redirectPolicy, ...urlLoaderOptions } = parseOptions(options)
|
|
|
+ this._urlLoaderOptions = urlLoaderOptions
|
|
|
+ this._redirectPolicy = redirectPolicy
|
|
|
+ this._started = false
|
|
|
}
|
|
|
|
|
|
set chunkedEncoding (value) {
|
|
|
- if (!this.urlRequest.notStarted) {
|
|
|
- throw new Error('Can\'t set the transfer encoding, headers have been sent')
|
|
|
+ if (this._started) {
|
|
|
+ throw new Error('chunkedEncoding can only be set before the request is started')
|
|
|
+ }
|
|
|
+ if (typeof this._chunkedEncoding !== 'undefined') {
|
|
|
+ throw new Error('chunkedEncoding can only be set once')
|
|
|
+ }
|
|
|
+ this._chunkedEncoding = !!value
|
|
|
+ if (this._chunkedEncoding) {
|
|
|
+ this._body = new ChunkedBodyStream(this)
|
|
|
+ this._urlLoaderOptions.body = (pipe) => {
|
|
|
+ this._body.startReading(pipe)
|
|
|
+ }
|
|
|
}
|
|
|
- this.chunkedEncodingEnabled = value
|
|
|
}
|
|
|
|
|
|
setHeader (name, value) {
|
|
@@ -277,13 +292,18 @@ class ClientRequest extends EventEmitter {
|
|
|
if (value == null) {
|
|
|
throw new Error('`value` required in setHeader("' + name + '", value)')
|
|
|
}
|
|
|
- if (!this.urlRequest.notStarted) {
|
|
|
+ if (this._started || this._firstWrite) {
|
|
|
throw new Error('Can\'t set headers after they are sent')
|
|
|
}
|
|
|
+ if (!_isValidHeaderName(name)) {
|
|
|
+ throw new Error(`Invalid header name: '${name}'`)
|
|
|
+ }
|
|
|
+ if (!_isValidHeaderValue(value.toString())) {
|
|
|
+ throw new Error(`Invalid value for header '${name}': '${value}'`)
|
|
|
+ }
|
|
|
|
|
|
const key = name.toLowerCase()
|
|
|
- this.extraHeaders[key] = value
|
|
|
- this.urlRequest.setExtraHeader(name, value.toString())
|
|
|
+ this._urlLoaderOptions.extraHeaders[key] = value
|
|
|
}
|
|
|
|
|
|
getHeader (name) {
|
|
@@ -291,12 +311,8 @@ class ClientRequest extends EventEmitter {
|
|
|
throw new Error('`name` is required for getHeader(name)')
|
|
|
}
|
|
|
|
|
|
- if (!this.extraHeaders) {
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
const key = name.toLowerCase()
|
|
|
- return this.extraHeaders[key]
|
|
|
+ return this._urlLoaderOptions.extraHeaders[key]
|
|
|
}
|
|
|
|
|
|
removeHeader (name) {
|
|
@@ -304,91 +320,142 @@ class ClientRequest extends EventEmitter {
|
|
|
throw new Error('`name` is required for removeHeader(name)')
|
|
|
}
|
|
|
|
|
|
- if (!this.urlRequest.notStarted) {
|
|
|
+ if (this._started || this._firstWrite) {
|
|
|
throw new Error('Can\'t remove headers after they are sent')
|
|
|
}
|
|
|
|
|
|
const key = name.toLowerCase()
|
|
|
- delete this.extraHeaders[key]
|
|
|
- this.urlRequest.removeExtraHeader(name)
|
|
|
+ delete this._urlLoaderOptions.extraHeaders[key]
|
|
|
}
|
|
|
|
|
|
- _write (chunk, encoding, callback, isLast) {
|
|
|
- const chunkIsString = typeof chunk === 'string'
|
|
|
- const chunkIsBuffer = chunk instanceof Buffer
|
|
|
- if (!chunkIsString && !chunkIsBuffer) {
|
|
|
- throw new TypeError('First argument must be a string or Buffer')
|
|
|
- }
|
|
|
-
|
|
|
- if (chunkIsString) {
|
|
|
- // We convert all strings into binary buffers.
|
|
|
- chunk = Buffer.from(chunk, encoding)
|
|
|
- }
|
|
|
-
|
|
|
- // Since writing to the network is asynchronous, we conservatively
|
|
|
- // assume that request headers are written after delivering the first
|
|
|
- // buffer to the network IO thread.
|
|
|
- if (this.urlRequest.notStarted) {
|
|
|
- this.urlRequest.setChunkedUpload(this.chunkedEncoding)
|
|
|
- }
|
|
|
-
|
|
|
- // Headers are assumed to be sent on first call to _writeBuffer,
|
|
|
- // i.e. after the first call to write or end.
|
|
|
- const result = this.urlRequest.write(chunk, isLast)
|
|
|
-
|
|
|
- // The write callback is fired asynchronously to mimic Node.js.
|
|
|
- if (callback) {
|
|
|
- process.nextTick(callback)
|
|
|
+ _write (chunk, encoding, callback) {
|
|
|
+ this._firstWrite = true
|
|
|
+ if (!this._body) {
|
|
|
+ this._body = new SlurpStream()
|
|
|
+ this._body.on('finish', () => {
|
|
|
+ this._urlLoaderOptions.body = this._body.data()
|
|
|
+ this._startRequest()
|
|
|
+ })
|
|
|
}
|
|
|
-
|
|
|
- return result
|
|
|
+ // TODO: is this the right way to forward to another stream?
|
|
|
+ this._body.write(chunk, encoding, callback)
|
|
|
}
|
|
|
|
|
|
- write (data, encoding, callback) {
|
|
|
- if (this.urlRequest.finished) {
|
|
|
- const error = new Error('Write after end')
|
|
|
- process.nextTick(writeAfterEndNT, this, error, callback)
|
|
|
- return true
|
|
|
+ _final (callback) {
|
|
|
+ if (this._body) {
|
|
|
+ // TODO: is this the right way to forward to another stream?
|
|
|
+ this._body.end(callback)
|
|
|
+ } else {
|
|
|
+ // end() called without a body, go ahead and start the request
|
|
|
+ this._startRequest()
|
|
|
+ callback()
|
|
|
}
|
|
|
-
|
|
|
- return this._write(data, encoding, callback, false)
|
|
|
}
|
|
|
|
|
|
- end (data, encoding, callback) {
|
|
|
- if (this.urlRequest.finished) {
|
|
|
- return false
|
|
|
+ _startRequest () {
|
|
|
+ this._started = true
|
|
|
+ const stringifyValues = (obj) => {
|
|
|
+ const ret = {}
|
|
|
+ for (const k in obj) {
|
|
|
+ ret[k] = obj[k].toString()
|
|
|
+ }
|
|
|
+ return ret
|
|
|
}
|
|
|
+ const opts = { ...this._urlLoaderOptions, extraHeaders: stringifyValues(this._urlLoaderOptions.extraHeaders) }
|
|
|
+ this._urlLoader = new URLLoader(opts)
|
|
|
+ this._urlLoader.on('response-started', (event, finalUrl, responseHead) => {
|
|
|
+ const response = this._response = new IncomingMessage(responseHead)
|
|
|
+ this.emit('response', response)
|
|
|
+ })
|
|
|
+ this._urlLoader.on('data', (event, data) => {
|
|
|
+ this._response._storeInternalData(Buffer.from(data))
|
|
|
+ })
|
|
|
+ this._urlLoader.on('complete', () => {
|
|
|
+ if (this._response) { this._response._storeInternalData(null) }
|
|
|
+ })
|
|
|
+ this._urlLoader.on('error', (event, netErrorString) => {
|
|
|
+ const error = new Error(netErrorString)
|
|
|
+ if (this._response) this._response.destroy(error)
|
|
|
+ this._die(error)
|
|
|
+ })
|
|
|
|
|
|
- if (typeof data === 'function') {
|
|
|
- callback = data
|
|
|
- encoding = null
|
|
|
- data = null
|
|
|
- } else if (typeof encoding === 'function') {
|
|
|
- callback = encoding
|
|
|
- encoding = null
|
|
|
- }
|
|
|
+ this._urlLoader.on('login', (event, authInfo, callback) => {
|
|
|
+ const handled = this.emit('login', authInfo, callback)
|
|
|
+ if (!handled) {
|
|
|
+ // If there were no listeners, cancel the authentication request.
|
|
|
+ callback()
|
|
|
+ }
|
|
|
+ })
|
|
|
+
|
|
|
+ this._urlLoader.on('redirect', (event, redirectInfo, headers) => {
|
|
|
+ const { statusCode, newMethod, newUrl } = redirectInfo
|
|
|
+ if (this._redirectPolicy === 'error') {
|
|
|
+ this._die(new Error(`Attempted to redirect, but redirect policy was 'error'`))
|
|
|
+ } else if (this._redirectPolicy === 'manual') {
|
|
|
+ let _followRedirect = false
|
|
|
+ this._followRedirectCb = () => { _followRedirect = true }
|
|
|
+ try {
|
|
|
+ this.emit('redirect', statusCode, newMethod, newUrl, headers)
|
|
|
+ } finally {
|
|
|
+ this._followRedirectCb = null
|
|
|
+ if (!_followRedirect) {
|
|
|
+ this._die(new Error('Redirect was cancelled'))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else if (this._redirectPolicy === 'follow') {
|
|
|
+ // Calling followRedirect() when the redirect policy is 'follow' is
|
|
|
+ // allowed but does nothing. (Perhaps it should throw an error
|
|
|
+ // though...? Since the redirect will happen regardless.)
|
|
|
+ try {
|
|
|
+ this._followRedirectCb = () => {}
|
|
|
+ this.emit('redirect', statusCode, newMethod, newUrl, headers)
|
|
|
+ } finally {
|
|
|
+ this._followRedirectCb = null
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ this._die(new Error(`Unexpected redirect policy '${this._redirectPolicy}'`))
|
|
|
+ }
|
|
|
+ })
|
|
|
|
|
|
- data = data || ''
|
|
|
+ this._urlLoader.on('upload-progress', (event, position, total) => {
|
|
|
+ this._uploadProgress = { active: true, started: true, current: position, total }
|
|
|
+ this.emit('upload-progress', position, total) // Undocumented, for now
|
|
|
+ })
|
|
|
|
|
|
- return this._write(data, encoding, callback, true)
|
|
|
+ this._urlLoader.on('download-progress', (event, current) => {
|
|
|
+ if (this._response) {
|
|
|
+ this._response.emit('download-progress', current) // Undocumented, for now
|
|
|
+ }
|
|
|
+ })
|
|
|
}
|
|
|
|
|
|
followRedirect () {
|
|
|
- this.urlRequest.followRedirect()
|
|
|
+ if (this._followRedirectCb) {
|
|
|
+ this._followRedirectCb()
|
|
|
+ } else {
|
|
|
+ throw new Error('followRedirect() called, but was not waiting for a redirect')
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
abort () {
|
|
|
- this.urlRequest.cancel()
|
|
|
+ if (!this._aborted) {
|
|
|
+ process.nextTick(() => { this.emit('abort') })
|
|
|
+ }
|
|
|
+ this._aborted = true
|
|
|
+ this._die()
|
|
|
}
|
|
|
|
|
|
- getUploadProgress () {
|
|
|
- return this.urlRequest.getUploadProgress()
|
|
|
+ _die (err) {
|
|
|
+ this.destroy(err)
|
|
|
+ if (this._urlLoader) {
|
|
|
+ this._urlLoader.cancel()
|
|
|
+ if (this._response) this._response.destroy(err)
|
|
|
+ }
|
|
|
}
|
|
|
-}
|
|
|
|
|
|
-function writeAfterEndNT (self, error, callback) {
|
|
|
- self.emit('error', error)
|
|
|
- if (callback) callback(error)
|
|
|
+ getUploadProgress () {
|
|
|
+ return { ...this._uploadProgress } || { active: false }
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
Net.prototype.request = function (options, callback) {
|