format: prettify entire project
This commit is contained in:
103
node_modules/undici/lib/handler/wrap-handler.js
generated
vendored
103
node_modules/undici/lib/handler/wrap-handler.js
generated
vendored
@ -1,95 +1,116 @@
|
||||
'use strict'
|
||||
'use strict';
|
||||
|
||||
const { InvalidArgumentError } = require('../core/errors')
|
||||
const { InvalidArgumentError } = require('../core/errors');
|
||||
|
||||
module.exports = class WrapHandler {
|
||||
#handler
|
||||
#handler;
|
||||
|
||||
constructor (handler) {
|
||||
this.#handler = handler
|
||||
constructor(handler) {
|
||||
this.#handler = handler;
|
||||
}
|
||||
|
||||
static wrap (handler) {
|
||||
static wrap(handler) {
|
||||
// TODO (fix): More checks...
|
||||
return handler.onRequestStart ? handler : new WrapHandler(handler)
|
||||
return handler.onRequestStart ? handler : new WrapHandler(handler);
|
||||
}
|
||||
|
||||
// Unwrap Interface
|
||||
|
||||
onConnect (abort, context) {
|
||||
return this.#handler.onConnect?.(abort, context)
|
||||
onConnect(abort, context) {
|
||||
return this.#handler.onConnect?.(abort, context);
|
||||
}
|
||||
|
||||
onHeaders (statusCode, rawHeaders, resume, statusMessage) {
|
||||
return this.#handler.onHeaders?.(statusCode, rawHeaders, resume, statusMessage)
|
||||
onHeaders(statusCode, rawHeaders, resume, statusMessage) {
|
||||
return this.#handler.onHeaders?.(
|
||||
statusCode,
|
||||
rawHeaders,
|
||||
resume,
|
||||
statusMessage
|
||||
);
|
||||
}
|
||||
|
||||
onUpgrade (statusCode, rawHeaders, socket) {
|
||||
return this.#handler.onUpgrade?.(statusCode, rawHeaders, socket)
|
||||
onUpgrade(statusCode, rawHeaders, socket) {
|
||||
return this.#handler.onUpgrade?.(statusCode, rawHeaders, socket);
|
||||
}
|
||||
|
||||
onData (data) {
|
||||
return this.#handler.onData?.(data)
|
||||
onData(data) {
|
||||
return this.#handler.onData?.(data);
|
||||
}
|
||||
|
||||
onComplete (trailers) {
|
||||
return this.#handler.onComplete?.(trailers)
|
||||
onComplete(trailers) {
|
||||
return this.#handler.onComplete?.(trailers);
|
||||
}
|
||||
|
||||
onError (err) {
|
||||
onError(err) {
|
||||
if (!this.#handler.onError) {
|
||||
throw err
|
||||
throw err;
|
||||
}
|
||||
|
||||
return this.#handler.onError?.(err)
|
||||
return this.#handler.onError?.(err);
|
||||
}
|
||||
|
||||
// Wrap Interface
|
||||
|
||||
onRequestStart (controller, context) {
|
||||
this.#handler.onConnect?.((reason) => controller.abort(reason), context)
|
||||
onRequestStart(controller, context) {
|
||||
this.#handler.onConnect?.((reason) => controller.abort(reason), context);
|
||||
}
|
||||
|
||||
onRequestUpgrade (controller, statusCode, headers, socket) {
|
||||
const rawHeaders = []
|
||||
onRequestUpgrade(controller, statusCode, headers, socket) {
|
||||
const rawHeaders = [];
|
||||
for (const [key, val] of Object.entries(headers)) {
|
||||
rawHeaders.push(Buffer.from(key), Array.isArray(val) ? val.map(v => Buffer.from(v)) : Buffer.from(val))
|
||||
rawHeaders.push(
|
||||
Buffer.from(key),
|
||||
Array.isArray(val) ? val.map((v) => Buffer.from(v)) : Buffer.from(val)
|
||||
);
|
||||
}
|
||||
|
||||
this.#handler.onUpgrade?.(statusCode, rawHeaders, socket)
|
||||
this.#handler.onUpgrade?.(statusCode, rawHeaders, socket);
|
||||
}
|
||||
|
||||
onResponseStart (controller, statusCode, headers, statusMessage) {
|
||||
const rawHeaders = []
|
||||
onResponseStart(controller, statusCode, headers, statusMessage) {
|
||||
const rawHeaders = [];
|
||||
for (const [key, val] of Object.entries(headers)) {
|
||||
rawHeaders.push(Buffer.from(key), Array.isArray(val) ? val.map(v => Buffer.from(v)) : Buffer.from(val))
|
||||
rawHeaders.push(
|
||||
Buffer.from(key),
|
||||
Array.isArray(val) ? val.map((v) => Buffer.from(v)) : Buffer.from(val)
|
||||
);
|
||||
}
|
||||
|
||||
if (this.#handler.onHeaders?.(statusCode, rawHeaders, () => controller.resume(), statusMessage) === false) {
|
||||
controller.pause()
|
||||
if (
|
||||
this.#handler.onHeaders?.(
|
||||
statusCode,
|
||||
rawHeaders,
|
||||
() => controller.resume(),
|
||||
statusMessage
|
||||
) === false
|
||||
) {
|
||||
controller.pause();
|
||||
}
|
||||
}
|
||||
|
||||
onResponseData (controller, data) {
|
||||
onResponseData(controller, data) {
|
||||
if (this.#handler.onData?.(data) === false) {
|
||||
controller.pause()
|
||||
controller.pause();
|
||||
}
|
||||
}
|
||||
|
||||
onResponseEnd (controller, trailers) {
|
||||
const rawTrailers = []
|
||||
onResponseEnd(controller, trailers) {
|
||||
const rawTrailers = [];
|
||||
for (const [key, val] of Object.entries(trailers)) {
|
||||
rawTrailers.push(Buffer.from(key), Array.isArray(val) ? val.map(v => Buffer.from(v)) : Buffer.from(val))
|
||||
rawTrailers.push(
|
||||
Buffer.from(key),
|
||||
Array.isArray(val) ? val.map((v) => Buffer.from(v)) : Buffer.from(val)
|
||||
);
|
||||
}
|
||||
|
||||
this.#handler.onComplete?.(rawTrailers)
|
||||
this.#handler.onComplete?.(rawTrailers);
|
||||
}
|
||||
|
||||
onResponseError (controller, err) {
|
||||
onResponseError(controller, err) {
|
||||
if (!this.#handler.onError) {
|
||||
throw new InvalidArgumentError('invalid onError method')
|
||||
throw new InvalidArgumentError('invalid onError method');
|
||||
}
|
||||
|
||||
this.#handler.onError?.(err)
|
||||
this.#handler.onError?.(err);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
Reference in New Issue
Block a user