123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.downloadBrowserWithProgressBar = downloadBrowserWithProgressBar;
- exports.logPolitely = logPolitely;
- var _fs = _interopRequireDefault(require("fs"));
- var _os = _interopRequireDefault(require("os"));
- var _path = _interopRequireDefault(require("path"));
- var _child_process = _interopRequireDefault(require("child_process"));
- var _fileUtils = require("../../utils/fileUtils");
- var _debugLogger = require("../../common/debugLogger");
- var _manualPromise = require("../../utils/manualPromise");
- var _utilsBundle = require("../../utilsBundle");
- var _ = require(".");
- var _userAgent = require("../../utils/userAgent");
- function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
- /**
- * Copyright 2017 Google Inc. All rights reserved.
- * Modifications copyright (c) Microsoft Corporation.
- *
- * 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
- *
- * http://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.
- */
- async function downloadBrowserWithProgressBar(title, browserDirectory, executablePath, downloadURLs, downloadFileName, downloadConnectionTimeout) {
- if (await (0, _fileUtils.existsAsync)((0, _.browserDirectoryToMarkerFilePath)(browserDirectory))) {
- // Already downloaded.
- _debugLogger.debugLogger.log('install', `${title} is already downloaded.`);
- return false;
- }
- const zipPath = _path.default.join(_os.default.tmpdir(), downloadFileName);
- try {
- const retryCount = 3;
- for (let attempt = 1; attempt <= retryCount; ++attempt) {
- _debugLogger.debugLogger.log('install', `downloading ${title} - attempt #${attempt}`);
- const url = downloadURLs[(attempt - 1) % downloadURLs.length];
- logPolitely(`Downloading ${title}` + _utilsBundle.colors.dim(` from ${url}`));
- const {
- error
- } = await downloadBrowserWithProgressBarOutOfProcess(title, browserDirectory, url, zipPath, executablePath, downloadConnectionTimeout);
- if (!error) {
- _debugLogger.debugLogger.log('install', `SUCCESS installing ${title}`);
- break;
- }
- if (await (0, _fileUtils.existsAsync)(zipPath)) await _fs.default.promises.unlink(zipPath);
- if (await (0, _fileUtils.existsAsync)(browserDirectory)) await _fs.default.promises.rmdir(browserDirectory, {
- recursive: true
- });
- const errorMessage = (error === null || error === void 0 ? void 0 : error.message) || '';
- _debugLogger.debugLogger.log('install', `attempt #${attempt} - ERROR: ${errorMessage}`);
- if (attempt >= retryCount) throw error;
- }
- } catch (e) {
- _debugLogger.debugLogger.log('install', `FAILED installation ${title} with error: ${e}`);
- process.exitCode = 1;
- throw e;
- } finally {
- if (await (0, _fileUtils.existsAsync)(zipPath)) await _fs.default.promises.unlink(zipPath);
- }
- logPolitely(`${title} downloaded to ${browserDirectory}`);
- return true;
- }
- /**
- * Node.js has a bug where the process can exit with 0 code even though there was an uncaught exception.
- * Thats why we execute it in a separate process and check manually if the destination file exists.
- * https://github.com/microsoft/playwright/issues/17394
- */
- function downloadBrowserWithProgressBarOutOfProcess(title, browserDirectory, url, zipPath, executablePath, connectionTimeout) {
- const cp = _child_process.default.fork(_path.default.join(__dirname, 'oopDownloadBrowserMain.js'));
- const promise = new _manualPromise.ManualPromise();
- const progress = getDownloadProgress();
- cp.on('message', message => {
- if ((message === null || message === void 0 ? void 0 : message.method) === 'log') _debugLogger.debugLogger.log('install', message.params.message);
- if ((message === null || message === void 0 ? void 0 : message.method) === 'progress') progress(message.params.done, message.params.total);
- });
- cp.on('exit', code => {
- if (code !== 0) {
- promise.resolve({
- error: new Error(`Download failure, code=${code}`)
- });
- return;
- }
- if (!_fs.default.existsSync((0, _.browserDirectoryToMarkerFilePath)(browserDirectory))) promise.resolve({
- error: new Error(`Download failure, ${(0, _.browserDirectoryToMarkerFilePath)(browserDirectory)} does not exist`)
- });else promise.resolve({
- error: null
- });
- });
- cp.on('error', error => {
- promise.resolve({
- error
- });
- });
- _debugLogger.debugLogger.log('install', `running download:`);
- _debugLogger.debugLogger.log('install', `-- from url: ${url}`);
- _debugLogger.debugLogger.log('install', `-- to location: ${zipPath}`);
- const downloadParams = {
- title,
- browserDirectory,
- url,
- zipPath,
- executablePath,
- connectionTimeout,
- userAgent: (0, _userAgent.getUserAgent)()
- };
- cp.send({
- method: 'download',
- params: downloadParams
- });
- return promise;
- }
- function logPolitely(toBeLogged) {
- const logLevel = process.env.npm_config_loglevel;
- const logLevelDisplay = ['silent', 'error', 'warn'].indexOf(logLevel || '') > -1;
- if (!logLevelDisplay) console.log(toBeLogged); // eslint-disable-line no-console
- }
- function getDownloadProgress() {
- if (process.stdout.isTTY) return getAnimatedDownloadProgress();
- return getBasicDownloadProgress();
- }
- function getAnimatedDownloadProgress() {
- let progressBar;
- let lastDownloadedBytes = 0;
- return (downloadedBytes, totalBytes) => {
- if (!progressBar) {
- progressBar = new _utilsBundle.progress(`${toMegabytes(totalBytes)} [:bar] :percent :etas`, {
- complete: '=',
- incomplete: ' ',
- width: 20,
- total: totalBytes
- });
- }
- const delta = downloadedBytes - lastDownloadedBytes;
- lastDownloadedBytes = downloadedBytes;
- progressBar.tick(delta);
- };
- }
- function getBasicDownloadProgress() {
- const totalRows = 10;
- const stepWidth = 8;
- let lastRow = -1;
- return (downloadedBytes, totalBytes) => {
- const percentage = downloadedBytes / totalBytes;
- const row = Math.floor(totalRows * percentage);
- if (row > lastRow) {
- lastRow = row;
- const percentageString = String(percentage * 100 | 0).padStart(3);
- // eslint-disable-next-line no-console
- console.log(`|${'■'.repeat(row * stepWidth)}${' '.repeat((totalRows - row) * stepWidth)}| ${percentageString}% of ${toMegabytes(totalBytes)}`);
- }
- };
- }
- function toMegabytes(bytes) {
- const mb = bytes / 1024 / 1024;
- return `${Math.round(mb * 10) / 10} MiB`;
- }
|