feat: add User-Agent with Puppeteer version to WebSocket request (#5614)

This will add the User-Agent header to WebSocket requests.
This commit is contained in:
headlesstesting 2021-09-15 09:40:43 +02:00 committed by GitHub
parent 300be5d167
commit 6a2bf0aaba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -18,11 +18,16 @@ import NodeWebSocket from 'ws';
export class NodeWebSocketTransport implements ConnectionTransport { export class NodeWebSocketTransport implements ConnectionTransport {
static create(url: string): Promise<NodeWebSocketTransport> { static create(url: string): Promise<NodeWebSocketTransport> {
// eslint-disable-next-line @typescript-eslint/no-var-requires
const pkg = require('../../../../package.json');
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
const ws = new NodeWebSocket(url, [], { const ws = new NodeWebSocket(url, [], {
followRedirects: true, followRedirects: true,
perMessageDeflate: false, perMessageDeflate: false,
maxPayload: 256 * 1024 * 1024, // 256Mb maxPayload: 256 * 1024 * 1024, // 256Mb
headers: {
'User-Agent': `Puppeteer ${pkg.version}`,
},
}); });
ws.addEventListener('open', () => ws.addEventListener('open', () =>