Files
react-native-devtools-frontend/node_modules/@puppeteer/replay/lib/cli.js
T
Randolf 51ffdadc72 Update @puppeteer/replay to 2.9.0
Bug: none
Change-Id: I0372a04e8714f11338d7df7c2b2ce21b24f3048a
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/4226464
Commit-Queue: Simon Zünd <szuend@chromium.org>
Reviewed-by: Simon Zünd <szuend@chromium.org>
Auto-Submit: Randolf Jung <jrandolf@chromium.org>
2023-02-07 04:50:44 +00:00

205 lines
6.7 KiB
JavaScript
Executable File

#!/usr/bin/env node
import yargs from 'yargs';
import { hideBin } from 'yargs/helpers';
import { parse, createRunner, PuppeteerRunnerOwningBrowserExtension } from '../lib/main.js';
import { lstatSync, readFileSync, readdirSync } from 'fs';
import { extname, join, relative, isAbsolute } from 'path';
import { pathToFileURL } from 'url';
import { cwd } from 'process';
import Table from 'cli-table3';
import { bgGreen, bgRed, white } from 'colorette';
/**
Copyright 2022 Google LLC
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
https://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
function getJSONFilesFromFolder(path) {
return readdirSync(path)
.filter((file) => extname(file) === '.json')
.map((file) => join(path, file));
}
function getRecordingPaths(paths, log = true) {
const recordingPaths = [];
for (const path of paths) {
let isDirectory;
try {
isDirectory = lstatSync(path).isDirectory();
}
catch (err) {
log && console.error(`Couldn't find file/folder: ${path}`, err);
continue;
}
if (isDirectory) {
const filesInFolder = getJSONFilesFromFolder(path);
if (!filesInFolder.length)
log && console.error(`There is no recordings in: ${path}`);
recordingPaths.push(...filesInFolder);
}
else
recordingPaths.push(path);
}
return recordingPaths;
}
function getHeadlessEnvVar(headless) {
if (!headless) {
return true;
}
switch (headless.toLowerCase()) {
case '1':
case 'true':
return true;
case 'new':
return 'new';
case '0':
case 'false':
return false;
default:
throw new Error('PUPPETEER_HEADLESS: unrecognized value');
}
}
function createStatusReport(results) {
var _a;
const table = new Table({
head: ['Title', 'Status', 'File', 'Duration'],
chars: {
top: '═',
'top-mid': '╤',
'top-left': '╔',
'top-right': '╗',
bottom: '═',
'bottom-mid': '╧',
'bottom-left': '╚',
'bottom-right': '╝',
left: '║',
'left-mid': '╟',
mid: '─',
'mid-mid': '┼',
right: '║',
'right-mid': '╢',
middle: '│',
},
style: {
head: ['bold'],
},
});
const resultTextColor = white;
for (const result of results) {
const row = [];
const duration = ((_a = result.finishedAt) === null || _a === void 0 ? void 0 : _a.getTime()) - result.startedAt.getTime() || 0;
const status = result.success
? resultTextColor(bgGreen(' Success '))
: resultTextColor(bgRed(' Failure '));
row.push(result.title);
row.push(status);
row.push(relative(process.cwd(), result.file));
row.push(`${duration}ms`);
table.push(row);
}
return table;
}
async function importExtensionFromPath(path) {
const module = await import(pathToFileURL(isAbsolute(path) ? path : join(cwd(), path)).toString());
return module.default;
}
async function runFiles(files, opts = {
log: true,
headless: true,
}) {
let Extension = PuppeteerRunnerOwningBrowserExtension;
let browser;
if (opts.extension) {
Extension = await importExtensionFromPath(opts.extension);
}
const results = [];
for (const file of files) {
const result = {
title: '',
startedAt: new Date(),
finishedAt: new Date(),
file,
success: true,
};
opts.log && console.log(`Running ${file}...`);
try {
const content = readFileSync(file, 'utf-8');
const object = JSON.parse(content);
const recording = parse(object);
result.title = recording.title;
const { default: puppeteer } = await import('puppeteer');
browser = await puppeteer.launch({
headless: opts.headless,
});
const page = await browser.newPage();
const extension = new Extension(browser, page);
const runner = await createRunner(recording, extension);
await runner.run();
opts.log && console.log(`Finished running ${file}`);
}
catch (err) {
opts.log && console.error(`Error running ${file}`, err);
result.success = false;
}
finally {
result.finishedAt = new Date();
results.push(result);
await (browser === null || browser === void 0 ? void 0 : browser.close());
}
}
if (opts.log) {
const statusReport = createStatusReport(results);
console.log(statusReport.toString());
}
if (results.every((result) => result.success))
return;
throw new Error('Some recordings have failed to run.');
}
/**
Copyright 2022 Google LLC
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
https://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
yargs(hideBin(process.argv))
.command('$0 <files..>', 'run files', () => { }, async (argv) => {
const args = argv;
const recordingPaths = getRecordingPaths(args.files);
await runFiles(recordingPaths, {
log: true,
headless: getHeadlessEnvVar(args.headless || process.env['PUPPETEER_HEADLESS']),
extension: args.extension,
});
})
.option('headless', {
type: 'string',
description: "Run using the browser's headless mode.",
choices: ['new', 'true', '1', '0', 'false'],
})
.option('extension', {
alias: 'ext',
type: 'string',
description: 'Run using an extension identified by the path.',
})
.parse();
//# sourceMappingURL=cli.js.map