52 lines
9.3 KiB
JavaScript
52 lines
9.3 KiB
JavaScript
|
"use strict";
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
const testcafe_hammerhead_1 = require("testcafe-hammerhead");
|
||
|
|
const string_1 = require("../utils/string");
|
||
|
|
const cdp_1 = require("../utils/cdp");
|
||
|
|
class ProxylessRequestHookEventProvider extends testcafe_hammerhead_1.RequestHookEventProvider {
|
||
|
|
static _hasResponseWithBody(context) {
|
||
|
|
return context.onResponseEventData.some((eventData) => eventData.opts.includeBody);
|
||
|
|
}
|
||
|
|
static async _setResponseBody({ pipelineContext, resourceBody, eventFactory, event, client }) {
|
||
|
|
if ((resourceBody === null || resourceBody === void 0 ? void 0 : resourceBody.length) || (0, cdp_1.isPreflightRequest)(event)) {
|
||
|
|
eventFactory.setResponseBody(resourceBody);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const hasOnResponseWithBody = ProxylessRequestHookEventProvider._hasResponseWithBody(pipelineContext);
|
||
|
|
if (!hasOnResponseWithBody)
|
||
|
|
return;
|
||
|
|
const responseObj = await client.Fetch.getResponseBody({ requestId: event.requestId });
|
||
|
|
const responseBody = (0, string_1.getResponseAsBuffer)(responseObj);
|
||
|
|
eventFactory.setResponseBody(responseBody);
|
||
|
|
}
|
||
|
|
async onRequest(event, contextInfo) {
|
||
|
|
if (!this.hasRequestEventListeners())
|
||
|
|
return;
|
||
|
|
const { pipelineContext, eventFactory } = contextInfo.getContextData(event);
|
||
|
|
await pipelineContext.onRequestHookRequest(this, eventFactory);
|
||
|
|
}
|
||
|
|
async onResponse(event, resourceBody, contextInfo, client) {
|
||
|
|
let modified = false;
|
||
|
|
if (!this.hasRequestEventListeners())
|
||
|
|
return modified;
|
||
|
|
const { pipelineContext, eventFactory } = contextInfo.getContextData(event);
|
||
|
|
eventFactory.update(event);
|
||
|
|
await pipelineContext.onRequestHookConfigureResponse(this, eventFactory);
|
||
|
|
if (eventFactory.headersModified)
|
||
|
|
modified = true;
|
||
|
|
await ProxylessRequestHookEventProvider._setResponseBody({
|
||
|
|
pipelineContext,
|
||
|
|
resourceBody,
|
||
|
|
eventFactory,
|
||
|
|
event,
|
||
|
|
client,
|
||
|
|
});
|
||
|
|
await Promise.all(pipelineContext.onResponseEventData.map(async (eventData) => {
|
||
|
|
await pipelineContext.onRequestHookResponse(this, eventFactory, eventData.rule, eventData.opts);
|
||
|
|
}));
|
||
|
|
return modified;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.default = ProxylessRequestHookEventProvider;
|
||
|
|
module.exports = exports.default;
|
||
|
|
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZXZlbnQtcHJvdmlkZXIuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi8uLi9zcmMvcHJveHlsZXNzL3JlcXVlc3QtaG9va3MvZXZlbnQtcHJvdmlkZXIudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7QUFBQSw2REFJNkI7QUFJN0IsNENBQXNEO0FBQ3RELHNDQUFrRDtBQUtsRCxNQUFxQixpQ0FBa0MsU0FBUSw4Q0FBd0I7SUFDM0UsTUFBTSxDQUFDLG9CQUFvQixDQUFFLE9BQWlDO1FBQ2xFLE9BQU8sT0FBTyxDQUFDLG1CQUFtQixDQUFDLElBQUksQ0FBQyxDQUFDLFNBQThCLEVBQUUsRUFBRSxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsV0FBVyxDQUFDLENBQUM7SUFDNUcsQ0FBQztJQUVPLE1BQU0sQ0FBQyxLQUFLLENBQUMsZ0JBQWdCLENBQUUsRUFBRSxlQUFlLEVBQUUsWUFBWSxFQUFFLFlBQVksRUFBRSxLQUFLLEVBQUUsTUFBTSxFQUF5SztRQUN4USxJQUFJLENBQUEsWUFBWSxhQUFaLFlBQVksdUJBQVosWUFBWSxDQUFFLE1BQU0sS0FBSSxJQUFBLHdCQUFrQixFQUFDLEtBQUssQ0FBQyxFQUFFO1lBQ2xELFlBQW9ELENBQUMsZUFBZSxDQUFDLFlBQXNCLENBQUMsQ0FBQztZQUU5RixPQUFPO1NBQ1Y7UUFHRCxNQUFNLHFCQUFxQixHQUFHLGlDQUFpQyxDQUFDLG9CQUFvQixDQUFDLGVBQWUsQ0FBQyxDQUFDO1FBRXRHLElBQUksQ0FBQyxxQkFBcUI7WUFDdEIsT0FBTztRQUVYLE1BQU0sV0FBVyxHQUFJLE1BQU0sTUFBTSxDQUFDLEtBQUssQ0FBQyxlQUFlLENBQUMsRUFBRSxTQUFTLEVBQUUsS0FBSyxDQUFDLFNBQVMsRUFBRSxDQUFDLENBQUM7UUFDeEYsTUFBTSxZQUFZLEdBQUcsSUFBQSw0QkFBbUIsRUFBQyxXQUFXLENBQUMsQ0FBQztRQUVyRCxZQUFvRCxDQUFDLGVBQWUsQ0FBQyxZQUFZLENBQUMsQ0FBQztJQUN4RixDQUFDO0lBRU0sS0FBSyxDQUFDLFNBQVMsQ0FBRSxLQUF5QixFQUFFLFdBQXdDO1FBQ3ZGLElBQUksQ0FBQyxJQUFJLENBQUMsd0JBQXdCLEVBQUU7WUFDaEMsT0FBTztRQUVYLE1BQU0sRUFBRSxlQUFlLEVBQUUsWUFBWSxFQUFFLEdBQUcsV0FBVyxDQUFDLGNBQWMsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUU1RSxNQUFNLGVBQWUsQ0FBQyxvQkFBb0IsQ0FBQyxJQUFJLEVBQUUsWUFBWSxDQUFDLENBQUM7SUFDbkUsQ0FBQztJQUVNLEtBQUssQ0FBQyxVQUFVLENBQUUsS0FBeUIsRUFBRSxZQUEyQixFQUFFLFdBQXdDLEVBQUUsTUFBbUI7UUFDMUksSUFBSSxRQUFRLEdBQUcsS0FBSyxDQUFDO1FBRXJCLElBQUksQ0FBQyxJQUFJLENBQUMsd0JBQXdCLEVBQUU7WUFDaEMsT0FBTyxRQUFRLENBQUM7UUFFcEIsTUFBTSxFQUFFLGVBQWUsRUFBRSxZQUFZLEVBQUUsR0FBRyxXQUFXLENBQUMsY0FBYyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBRTNFLFlBQW9ELENBQUMsTUFBTSxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBRXBFLE1BQU0sZUFBZSxDQUFDLDhCQUE4QixDQUFDLElBQUksRUFBRSxZQUFZLENBQUMsQ0FBQztRQUV6RSxJQUFLLFlBQW9ELENBQUMsZUFBZTtZQUNyRSxRQUFRLEdBQUcsSUFBSSxDQUFDO1FBRXBCLE1BQU0saUNBQWlDLENBQUMsZ0JBQWdCLENBQUM7WUFDckQsZUFBZTtZQUNmLFlBQVk7WUFDWixZQUFZO1lBQ1osS0FBSztZQUNMLE1BQU07U0FDVCxDQUFDLENBQUM7UUFFSCxNQUFNLE9BQU8sQ0FBQyxHQUFHLENBQUMsZUFBZSxDQUFDLG1CQUFtQixDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUMsU0FBUyxFQUFDLEVBQUU7WUFDeEUsTUFBTSxlQUFlLENBQUMscUJBQXFCLENBQUMsSUFBSSxFQUFFLFlBQVksRUFBRSxTQUFTLENBQUMsSUFBSSxFQUFFLFNBQVMsQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUNwRyxDQUFDLENBQUMsQ0FBQyxDQUFDO1FBRUosT0FBTyxRQUFRLENBQUM7SUFDcEIsQ0FBQztDQUNKO0FBOURELG9EQThEQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCB7XG4gICAgQmFzZVJlcXVlc3RIb29rRXZlbnRGYWN0b3J5LFxuICAgIE9uUmVzcG9uc2VFdmVudERhdGEsXG4gICAgUmVxdWVzdEhvb2tFdmVudFByb3ZpZGVyLFxufSBmcm9tICd0ZXN0Y2FmZS1oYW1tZXJoZWFkJztcbmltcG9ydCBQcm94eWxlc3NQaXBlbGluZUNvbnRleHQgZnJvbSAnLi9waXBlbGluZS1jb250ZXh0JztcbmltcG9ydCBSZXF1ZXN0UGF1c2VkRXZlbnRCYXNlZEV2ZW50RmFjdG9yeSBmcm9tICcuL2V2ZW50LWZhY3RvcnkvcmVxdWVzdC1wYXVzZWQtZXZlbnQtYmFzZWQnO1xuaW1wb3J0IHsgUHJvdG9jb2xBcGkgfSBmcm9tICdjaHJvbWUtcmVtb3RlLWludGVyZmFjZSc7XG5pbXBvcnQgeyBnZXRSZXNwb25zZUFzQnVmZmVyIH0gZnJvbSAnLi4vdXRpbHMvc3RyaW5nJztcbmltcG9ydCB7IGlzUHJlZmxpZ2h0UmVxdWVzdCB9IGZyb20gJy4uL3V0aWxzL2NkcCc7XG5pbXBvcnQgUHJvdG9jb2wgZnJvbSAnZGV2dG9vbHMtcHJvdG9jb2wnO1xuaW1wb3J0IFJlcXVlc3RQYXVzZWRFdmVudCA9IFByb3RvY29sLkZldGNoLlJlcXVlc3RQYXVzZWRFdmVudDtcbmltcG9ydCBQcm94eWxlc3NSZXF1ZXN0Q29udGV4dEluZm8gZnJvbSAnLi4vcmVxdWVzdC1waXBlbGluZS9jb250ZXh0LWluZm8nO1xuXG5leHBvcnQgZGVmYXVsdCBjbGFzcyBQcm94eWxlc3NSZXF1ZXN0SG9va0V2ZW50UHJvdmlkZXIgZXh0ZW5kcyBSZXF1ZXN0SG9va0V2ZW50UHJvdmlkZXIge1xuICAgIHByaXZhdGUgc3RhdGljIF9oYXNSZXNwb25zZVdpdGhCb2R5IChjb250ZXh0OiBQcm94eWxlc3NQaXBlbGluZUNvbnRleHQpOiBib29sZWFuIHtcbiAgICAgICAgcmV0dXJuIGNvbnRleHQub25SZXNwb25zZUV2ZW50RGF0YS5zb21lKChldmVudERhdGE6IE9uUmVzcG9uc2VFdmVudERhdGEpID0+IGV2ZW50RGF0YS5vcHRzLmluY2x1ZGVCb2R5KTtcbiAgICB9XG5cbiAgICBwcml2YXRlIHN0YXRpYyBhc3luYyBfc2V0UmVzcG9uc2VCb2R5ICh7IHBpcGVsaW5lQ29udGV4dCwgcmVzb3VyY2VCb2R5LCBldmVudEZhY3RvcnksIGV2ZW50LCBjbGllbnQgfTogeyBwaXBlbGluZUNvbnRleHQ6IFByb3h5bGVzc1BpcGVsaW5lQ29udGV4dCwgcmVzb3VyY2VCb2R5OiBCdWZmZXIgfCBudWxsLCBldmVud
|