docs: update pdf example

This commit is contained in:
Alex Rudenko 2024-04-29 08:01:25 +02:00
parent e1606acfc8
commit 8bf9f81283
6 changed files with 26724 additions and 2 deletions

View File

@ -14,8 +14,6 @@ const puppeteer = require('puppeteer');
await page.goto('https://news.ycombinator.com', { await page.goto('https://news.ycombinator.com', {
waitUntil: 'networkidle2', waitUntil: 'networkidle2',
}); });
// page.pdf() is currently supported only in headless mode.
// @see https://bugs.chromium.org/p/chromium/issues/detail?id=753118
await page.pdf({ await page.pdf({
path: 'hn.pdf', path: 'hn.pdf',
format: 'letter', format: 'letter',

View File

@ -0,0 +1,40 @@
/**
* @internal
*/
class BrowserWebSocketTransport {
static create(url) {
return new Promise((resolve, reject) => {
const ws = new WebSocket(url);
ws.addEventListener('open', () => {
return resolve(new BrowserWebSocketTransport(ws));
});
ws.addEventListener('error', reject);
});
}
#ws;
onmessage;
onclose;
constructor(ws) {
this.#ws = ws;
this.#ws.addEventListener('message', event => {
if (this.onmessage) {
this.onmessage.call(null, event.data);
}
});
this.#ws.addEventListener('close', () => {
if (this.onclose) {
this.onclose.call(null);
}
});
// Silently ignore all errors - we don't know what to do with them.
this.#ws.addEventListener('error', () => { });
}
send(message) {
this.#ws.send(message);
}
close() {
this.#ws.close();
}
}
export { BrowserWebSocketTransport };

View File

@ -0,0 +1,64 @@
import NodeWebSocket from 'ws';
/**
* @internal
*/
const packageVersion = '22.7.1';
/**
* @license
* Copyright 2018 Google Inc.
* SPDX-License-Identifier: Apache-2.0
*/
/**
* @internal
*/
class NodeWebSocketTransport {
static create(url, headers) {
return new Promise((resolve, reject) => {
const ws = new NodeWebSocket(url, [], {
followRedirects: true,
perMessageDeflate: false,
maxPayload: 256 * 1024 * 1024, // 256Mb
headers: {
'User-Agent': `Puppeteer ${packageVersion}`,
...headers,
},
});
ws.addEventListener('open', () => {
return resolve(new NodeWebSocketTransport(ws));
});
ws.addEventListener('error', reject);
});
}
#ws;
onmessage;
onclose;
constructor(ws) {
this.#ws = ws;
this.#ws.addEventListener('message', event => {
setImmediate(() => {
if (this.onmessage) {
this.onmessage.call(null, event.data);
}
});
});
this.#ws.addEventListener('close', () => {
setImmediate(() => {
if (this.onclose) {
this.onclose.call(null);
}
});
});
// Silently ignore all errors - we don't know what to do with them.
this.#ws.addEventListener('error', () => { });
}
send(message) {
this.#ws.send(message);
}
close() {
this.#ws.close();
}
}
export { NodeWebSocketTransport };

View File

@ -0,0 +1,214 @@
import { spawnSync, spawn } from 'child_process';
import { PassThrough } from 'stream';
import debug from 'debug';
import { C as CDPSessionEvent, d as debugError, l as lastValueFrom, f as fromEmitterEvent, t as tap, a as filter, m as map, b as bufferCount, c as concatMap, e as takeUntil, g as fromEvent, h as guarded, i as asyncDisposeSymbol, j as from } from './main.js';
/**
* @license
* Copyright 2023 Google Inc.
* SPDX-License-Identifier: Apache-2.0
*/
var __runInitializers = (undefined && undefined.__runInitializers) || function (thisArg, initializers, value) {
var useValue = arguments.length > 2;
for (var i = 0; i < initializers.length; i++) {
value = useValue ? initializers[i].call(thisArg, value) : initializers[i].call(thisArg);
}
return useValue ? value : void 0;
};
var __esDecorate = (undefined && undefined.__esDecorate) || function (ctor, descriptorIn, decorators, contextIn, initializers, extraInitializers) {
function accept(f) { if (f !== void 0 && typeof f !== "function") throw new TypeError("Function expected"); return f; }
var kind = contextIn.kind, key = kind === "getter" ? "get" : kind === "setter" ? "set" : "value";
var target = !descriptorIn && ctor ? contextIn["static"] ? ctor : ctor.prototype : null;
var descriptor = descriptorIn || (target ? Object.getOwnPropertyDescriptor(target, contextIn.name) : {});
var _, done = false;
for (var i = decorators.length - 1; i >= 0; i--) {
var context = {};
for (var p in contextIn) context[p] = p === "access" ? {} : contextIn[p];
for (var p in contextIn.access) context.access[p] = contextIn.access[p];
context.addInitializer = function (f) { if (done) throw new TypeError("Cannot add initializers after decoration has completed"); extraInitializers.push(accept(f || null)); };
var result = (0, decorators[i])(kind === "accessor" ? { get: descriptor.get, set: descriptor.set } : descriptor[key], context);
if (kind === "accessor") {
if (result === void 0) continue;
if (result === null || typeof result !== "object") throw new TypeError("Object expected");
if (_ = accept(result.get)) descriptor.get = _;
if (_ = accept(result.set)) descriptor.set = _;
if (_ = accept(result.init)) initializers.unshift(_);
}
else if (_ = accept(result)) {
if (kind === "field") initializers.unshift(_);
else descriptor[key] = _;
}
}
if (target) Object.defineProperty(target, contextIn.name, descriptor);
done = true;
};
var __setFunctionName = (undefined && undefined.__setFunctionName) || function (f, name, prefix) {
if (typeof name === "symbol") name = name.description ? "[".concat(name.description, "]") : "";
return Object.defineProperty(f, "name", { configurable: true, value: prefix ? "".concat(prefix, " ", name) : name });
};
const CRF_VALUE = 30;
const DEFAULT_FPS = 30;
const debugFfmpeg = debug('puppeteer:ffmpeg');
/**
* @public
*/
let ScreenRecorder = (() => {
let _classSuper = PassThrough;
let _instanceExtraInitializers = [];
let _private_writeFrame_decorators;
let _private_writeFrame_descriptor;
let _stop_decorators;
return class ScreenRecorder extends _classSuper {
static {
const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(_classSuper[Symbol.metadata] ?? null) : void 0;
__esDecorate(this, _private_writeFrame_descriptor = { value: __setFunctionName(async function (buffer) {
const error = await new Promise(resolve => {
this.#process.stdin.write(buffer, resolve);
});
if (error) {
console.log(`ffmpeg failed to write: ${error.message}.`);
}
}, "#writeFrame") }, _private_writeFrame_decorators, { kind: "method", name: "#writeFrame", static: false, private: true, access: { has: obj => #writeFrame in obj, get: obj => obj.#writeFrame }, metadata: _metadata }, null, _instanceExtraInitializers);
__esDecorate(this, null, _stop_decorators, { kind: "method", name: "stop", static: false, private: false, access: { has: obj => "stop" in obj, get: obj => obj.stop }, metadata: _metadata }, null, _instanceExtraInitializers);
if (_metadata) Object.defineProperty(this, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
}
#page = (__runInitializers(this, _instanceExtraInitializers), void 0);
#process;
#controller = new AbortController();
#lastFrame;
/**
* @internal
*/
constructor(page, width, height, { speed, scale, crop, format, path } = {}) {
super({ allowHalfOpen: false });
path ??= 'ffmpeg';
// Tests if `ffmpeg` exists.
const { error } = spawnSync(path);
if (error) {
throw error;
}
this.#process = spawn(path,
// See https://trac.ffmpeg.org/wiki/Encode/VP9 for more information on flags.
[
['-loglevel', 'error'],
// Reduces general buffering.
['-avioflags', 'direct'],
// Reduces initial buffering while analyzing input fps and other stats.
[
'-fpsprobesize',
'0',
'-probesize',
'32',
'-analyzeduration',
'0',
'-fflags',
'nobuffer',
],
// Forces input to be read from standard input, and forces png input
// image format.
['-f', 'image2pipe', '-c:v', 'png', '-i', 'pipe:0'],
// Overwrite output and no audio.
['-y', '-an'],
// This drastically reduces stalling when cpu is overbooked. By default
// VP9 tries to use all available threads?
['-threads', '1'],
// Specifies the frame rate we are giving ffmpeg.
['-framerate', `${DEFAULT_FPS}`],
// Specifies the encoding and format we are using.
this.#getFormatArgs(format ?? 'webm'),
// Disable bitrate.
['-b:v', '0'],
// Filters to ensure the images are piped correctly.
[
'-vf',
`${speed ? `setpts=${1 / speed}*PTS,` : ''}crop='min(${width},iw):min(${height},ih):0:0',pad=${width}:${height}:0:0${crop ? `,crop=${crop.width}:${crop.height}:${crop.x}:${crop.y}` : ''}${scale ? `,scale=iw*${scale}:-1` : ''}`,
],
'pipe:1',
].flat(), { stdio: ['pipe', 'pipe', 'pipe'] });
this.#process.stdout.pipe(this);
this.#process.stderr.on('data', (data) => {
debugFfmpeg(data.toString('utf8'));
});
this.#page = page;
const { client } = this.#page.mainFrame();
client.once(CDPSessionEvent.Disconnected, () => {
void this.stop().catch(debugError);
});
this.#lastFrame = lastValueFrom(fromEmitterEvent(client, 'Page.screencastFrame').pipe(tap(event => {
void client.send('Page.screencastFrameAck', {
sessionId: event.sessionId,
});
}), filter(event => {
return event.metadata.timestamp !== undefined;
}), map(event => {
return {
buffer: Buffer.from(event.data, 'base64'),
timestamp: event.metadata.timestamp,
};
}), bufferCount(2, 1), concatMap(([{ timestamp: previousTimestamp, buffer }, { timestamp }]) => {
return from(Array(Math.round(DEFAULT_FPS * Math.max(timestamp - previousTimestamp, 0))).fill(buffer));
}), map(buffer => {
void this.#writeFrame(buffer);
return [buffer, performance.now()];
}), takeUntil(fromEvent(this.#controller.signal, 'abort'))), { defaultValue: [Buffer.from([]), performance.now()] });
}
#getFormatArgs(format) {
switch (format) {
case 'webm':
return [
// Sets the codec to use.
['-c:v', 'vp9'],
// Sets the format
['-f', 'webm'],
// Sets the quality. Lower the better.
['-crf', `${CRF_VALUE}`],
// Sets the quality and how efficient the compression will be.
['-deadline', 'realtime', '-cpu-used', '8'],
].flat();
case 'gif':
return [
// Sets the frame rate and uses a custom palette generated from the
// input.
[
'-vf',
'fps=5,split[s0][s1];[s0]palettegen=stats_mode=diff[p];[s1][p]paletteuse',
],
// Sets the format
['-f', 'gif'],
].flat();
}
}
get #writeFrame() { return _private_writeFrame_descriptor.value; }
/**
* Stops the recorder.
*
* @public
*/
async stop() {
if (this.#controller.signal.aborted) {
return;
}
// Stopping the screencast will flush the frames.
await this.#page._stopScreencast().catch(debugError);
this.#controller.abort();
// Repeat the last frame for the remaining frames.
const [buffer, timestamp] = await this.#lastFrame;
await Promise.all(Array(Math.max(1, Math.round((DEFAULT_FPS * (performance.now() - timestamp)) / 1000)))
.fill(buffer)
.map(this.#writeFrame.bind(this)));
// Close stdin to notify FFmpeg we are done.
this.#process.stdin.end();
await new Promise(resolve => {
this.#process.once('close', resolve);
});
}
/**
* @internal
*/
async [(_private_writeFrame_decorators = [guarded()], _stop_decorators = [guarded()], asyncDisposeSymbol)]() {
await this.stop();
}
};
})();
export { ScreenRecorder };

File diff suppressed because it is too large Load Diff

File diff suppressed because one or more lines are too long