puppeteer-core
Version:
A high-level API to control headless Chrome over the DevTools Protocol
263 lines • 8.64 kB
JavaScript
"use strict";
var _a;
Object.defineProperty(exports, "__esModule", { value: true });
exports.BidiHTTPRequest = exports.requests = void 0;
const HTTPRequest_js_1 = require("../api/HTTPRequest.js");
const Errors_js_1 = require("../common/Errors.js");
const encoding_js_1 = require("../util/encoding.js");
const HTTPResponse_js_1 = require("./HTTPResponse.js");
exports.requests = new WeakMap();
/**
* @internal
*/
class BidiHTTPRequest extends HTTPRequest_js_1.HTTPRequest {
static from(bidiRequest, frame, redirect) {
const request = new _a(bidiRequest, frame, redirect);
request.#initialize();
return request;
}
#redirectChain;
#response = null;
id;
#frame;
#request;
constructor(request, frame, redirect) {
super();
exports.requests.set(request, this);
this.interception.enabled = request.isBlocked;
this.#request = request;
this.#frame = frame;
this.#redirectChain = redirect ? redirect.#redirectChain : [];
this.id = request.id;
}
get client() {
return this.#frame.client;
}
#initialize() {
this.#request.on('redirect', request => {
const httpRequest = _a.from(request, this.#frame, this);
this.#redirectChain.push(this);
request.once('success', () => {
this.#frame
.page()
.trustedEmitter.emit("requestfinished" /* PageEvent.RequestFinished */, httpRequest);
});
request.once('error', () => {
this.#frame
.page()
.trustedEmitter.emit("requestfailed" /* PageEvent.RequestFailed */, httpRequest);
});
void httpRequest.finalizeInterceptions();
});
this.#request.once('success', data => {
this.#response = HTTPResponse_js_1.BidiHTTPResponse.from(data, this, this.#frame.page().browser().cdpSupported);
});
this.#request.on('authenticate', this.#handleAuthentication);
this.#frame.page().trustedEmitter.emit("request" /* PageEvent.Request */, this);
if (this.#hasInternalHeaderOverwrite) {
this.interception.handlers.push(async () => {
await this.continue({
headers: this.headers(),
}, 0);
});
}
}
url() {
return this.#request.url;
}
resourceType() {
if (!this.#frame.page().browser().cdpSupported) {
throw new Errors_js_1.UnsupportedOperation();
}
return (this.#request.resourceType || 'other').toLowerCase();
}
method() {
return this.#request.method;
}
postData() {
if (!this.#frame.page().browser().cdpSupported) {
throw new Errors_js_1.UnsupportedOperation();
}
return this.#request.postData;
}
hasPostData() {
if (!this.#frame.page().browser().cdpSupported) {
throw new Errors_js_1.UnsupportedOperation();
}
return this.#request.hasPostData;
}
async fetchPostData() {
throw new Errors_js_1.UnsupportedOperation();
}
get #hasInternalHeaderOverwrite() {
return Boolean(Object.keys(this.#extraHTTPHeaders).length ||
Object.keys(this.#userAgentHeaders).length);
}
get #extraHTTPHeaders() {
return this.#frame?.page()._extraHTTPHeaders ?? {};
}
get #userAgentHeaders() {
return this.#frame?.page()._userAgentHeaders ?? {};
}
headers() {
const headers = {};
for (const header of this.#request.headers) {
headers[header.name.toLowerCase()] = header.value.value;
}
return {
...headers,
...this.#extraHTTPHeaders,
...this.#userAgentHeaders,
};
}
response() {
return this.#response;
}
failure() {
if (this.#request.error === undefined) {
return null;
}
return { errorText: this.#request.error };
}
isNavigationRequest() {
return this.#request.navigation !== undefined;
}
initiator() {
return {
...this.#request.initiator,
type: this.#request.initiator?.type ?? 'other',
};
}
redirectChain() {
return this.#redirectChain.slice();
}
frame() {
return this.#frame;
}
async continue(overrides, priority) {
return await super.continue({
headers: this.#hasInternalHeaderOverwrite ? this.headers() : undefined,
...overrides,
}, priority);
}
async _continue(overrides = {}) {
const headers = getBidiHeaders(overrides.headers);
this.interception.handled = true;
return await this.#request
.continueRequest({
url: overrides.url,
method: overrides.method,
body: overrides.postData
? {
type: 'base64',
value: (0, encoding_js_1.stringToBase64)(overrides.postData),
}
: undefined,
headers: headers.length > 0 ? headers : undefined,
})
.catch(error => {
this.interception.handled = false;
return (0, HTTPRequest_js_1.handleError)(error);
});
}
async _abort() {
this.interception.handled = true;
return await this.#request.failRequest().catch(error => {
this.interception.handled = false;
throw error;
});
}
async _respond(response, _priority) {
this.interception.handled = true;
let parsedBody;
if (response.body) {
parsedBody = HTTPRequest_js_1.HTTPRequest.getResponse(response.body);
}
const headers = getBidiHeaders(response.headers);
const hasContentLength = headers.some(header => {
return header.name === 'content-length';
});
if (response.contentType) {
headers.push({
name: 'content-type',
value: {
type: 'string',
value: response.contentType,
},
});
}
if (parsedBody?.contentLength && !hasContentLength) {
headers.push({
name: 'content-length',
value: {
type: 'string',
value: String(parsedBody.contentLength),
},
});
}
const status = response.status || 200;
return await this.#request
.provideResponse({
statusCode: status,
headers: headers.length > 0 ? headers : undefined,
reasonPhrase: HTTPRequest_js_1.STATUS_TEXTS[status],
body: parsedBody?.base64
? {
type: 'base64',
value: parsedBody?.base64,
}
: undefined,
})
.catch(error => {
this.interception.handled = false;
throw error;
});
}
#authenticationHandled = false;
#handleAuthentication = async () => {
if (!this.#frame) {
return;
}
const credentials = this.#frame.page()._credentials;
if (credentials && !this.#authenticationHandled) {
this.#authenticationHandled = true;
void this.#request.continueWithAuth({
action: 'provideCredentials',
credentials: {
type: 'password',
username: credentials.username,
password: credentials.password,
},
});
}
else {
void this.#request.continueWithAuth({
action: 'cancel',
});
}
};
timing() {
return this.#request.timing();
}
}
exports.BidiHTTPRequest = BidiHTTPRequest;
_a = BidiHTTPRequest;
function getBidiHeaders(rawHeaders) {
const headers = [];
for (const [name, value] of Object.entries(rawHeaders ?? [])) {
if (!Object.is(value, undefined)) {
const values = Array.isArray(value) ? value : [value];
for (const value of values) {
headers.push({
name: name.toLowerCase(),
value: {
type: 'string',
value: String(value),
},
});
}
}
}
return headers;
}
//# sourceMappingURL=HTTPRequest.js.map