puppeteer/utils/ChromiumDownloader.js

219 lines
6.7 KiB
JavaScript
Raw Normal View History

/**
* Copyright 2017 Google Inc. All rights reserved.
*
* 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.
*/
let os = require('os');
let https = require('https');
let fs = require('fs');
let path = require('path');
let extract = require('extract-zip');
let util = require('util');
let URL = require('url');
let removeRecursive = require('rimraf');
let DOWNLOADS_FOLDER = path.join(__dirname, '..', '.local-chromium');
let downloadURLs = {
2017-06-21 20:51:06 +00:00
linux: 'https://storage.googleapis.com/chromium-browser-snapshots/Linux_x64/%d/chrome-linux.zip',
mac: 'https://storage.googleapis.com/chromium-browser-snapshots/Mac/%d/chrome-mac.zip',
win32: 'https://storage.googleapis.com/chromium-browser-snapshots/Win/%d/chrome-win32.zip',
win64: 'https://storage.googleapis.com/chromium-browser-snapshots/Win_x64/%d/chrome-win32.zip',
};
module.exports = {
2017-06-21 20:51:06 +00:00
/**
* @return {!Array<string>}
*/
2017-06-21 20:51:06 +00:00
supportedPlatforms: function() {
return Object.keys(downloadURLs);
},
2017-06-21 20:51:06 +00:00
/**
* @return {string}
*/
2017-06-21 20:51:06 +00:00
currentPlatform: function() {
let platform = os.platform();
2017-06-21 20:51:06 +00:00
if (platform === 'darwin')
return 'mac';
if (platform === 'linux')
return 'linux';
if (platform === 'win32')
return os.arch() === 'x64' ? 'win64' : 'win32';
return '';
},
2017-06-21 20:51:06 +00:00
/**
* @param {string} platform
* @param {string} revision
* @return {!Promise<boolean>}
*/
2017-06-21 20:51:06 +00:00
canDownloadRevision: function(platform, revision) {
console.assert(downloadURLs[platform], 'Unknown platform: ' + platform);
let url = URL.parse(util.format(downloadURLs[platform], revision));
let options = {
2017-06-21 20:51:06 +00:00
method: 'HEAD',
host: url.host,
path: url.pathname,
};
let resolve;
let promise = new Promise(x => resolve = x);
let request = https.request(options, response => {
2017-06-21 20:51:06 +00:00
resolve(response.statusCode === 200);
});
request.on('error', error => {
console.error(error);
resolve(false);
});
request.end();
return promise;
},
2017-06-21 20:51:06 +00:00
/**
* @param {string} platform
* @param {string} revision
* @param {?function(number, number)} progressCallback
* @return {!Promise}
*/
2017-06-21 20:51:06 +00:00
downloadRevision: async function(platform, revision, progressCallback) {
let url = downloadURLs[platform];
2017-06-21 20:51:06 +00:00
console.assert(url, `Unsupported platform: ${platform}`);
url = util.format(url, revision);
let zipPath = path.join(DOWNLOADS_FOLDER, `download-${platform}-${revision}.zip`);
let folderPath = getFolderPath(platform, revision);
2017-06-21 20:51:06 +00:00
if (fs.existsSync(folderPath))
return;
try {
if (!fs.existsSync(DOWNLOADS_FOLDER))
fs.mkdirSync(DOWNLOADS_FOLDER);
2017-06-21 20:51:06 +00:00
await downloadFile(url, zipPath, progressCallback);
await extractZip(zipPath, folderPath);
} finally {
if (fs.existsSync(zipPath))
fs.unlinkSync(zipPath);
}
},
2017-06-21 20:51:06 +00:00
/**
* @return {!Array<!{platform:string, revision: string}>}
*/
downloadedRevisions: function() {
if (!fs.existsSync(DOWNLOADS_FOLDER))
return [];
let fileNames = fs.readdirSync(DOWNLOADS_FOLDER);
return fileNames.map(fileName => parseFolderPath(fileName)).filter(revision => !!revision);
},
/**
* @param {string} platform
* @param {string} revision
*/
removeRevision: async function(platform, revision) {
console.assert(downloadURLs[platform], `Unsupported platform: ${platform}`);
const folderPath = getFolderPath(platform, revision);
console.assert(fs.existsSync(folderPath));
await new Promise(fulfill => removeRecursive(folderPath, fulfill));
},
/**
* @param {string} platform
* @param {string} revision
* @return {?{executablePath: string}}
*/
2017-06-21 20:51:06 +00:00
revisionInfo: function(platform, revision) {
console.assert(downloadURLs[platform], `Unsupported platform: ${platform}`);
let folderPath = getFolderPath(platform, revision);
2017-06-21 20:51:06 +00:00
if (!fs.existsSync(folderPath))
return null;
let executablePath = '';
2017-06-21 20:51:06 +00:00
if (platform === 'mac')
executablePath = path.join(folderPath, 'chrome-mac', 'Chromium.app', 'Contents', 'MacOS', 'Chromium');
else if (platform === 'linux')
executablePath = path.join(folderPath, 'chrome-linux', 'chrome');
else if (platform === 'win32' || platform === 'win64')
executablePath = path.join(folderPath, 'chrome-win32', 'chrome.exe');
else
throw 'Unsupported platfrom: ' + platfrom;
return {
executablePath: executablePath
};
},
};
/**
* @param {string} platform
* @param {string} revision
* @return {string}
*/
function getFolderPath(platform, revision) {
return path.join(DOWNLOADS_FOLDER, platform + '-' + revision);
}
/**
* @param {string} folderPath
* @return {?{platform: string, revision: string}}
*/
function parseFolderPath(folderPath) {
let name = path.basename(folderPath);
let splits = name.split('-');
if (splits.length !== 2)
return null;
let [platform, revision] = splits;
if (!downloadURLs[platform])
return null;
return {platform, revision};
}
/**
* @param {string} url
* @param {string} destinationPath
* @param {?function(number, number)} progressCallback
* @return {!Promise}
*/
function downloadFile(url, destinationPath, progressCallback) {
let fulfill, reject;
let promise = new Promise((x, y) => { fulfill = x; reject = y; });
let request = https.get(url, response => {
2017-06-21 20:51:06 +00:00
if (response.statusCode !== 200) {
let error = new Error(`Download failed: server returned code ${response.statusCode}. URL: ${url}`);
2017-06-21 20:51:06 +00:00
// consume response data to free up memory
response.resume();
reject(error);
return;
}
let file = fs.createWriteStream(destinationPath);
2017-06-21 20:51:06 +00:00
file.on('finish', () => fulfill());
file.on('error', error => reject(error));
response.pipe(file);
let totalBytes = parseInt(response.headers['content-length'], 10);
2017-06-21 20:51:06 +00:00
if (progressCallback)
response.on('data', onData.bind(null, totalBytes));
});
request.on('error', error => reject(error));
return promise;
function onData(totalBytes, chunk) {
progressCallback(totalBytes, chunk.length);
}
}
/**
* @param {string} zipPath
* @param {string} folderPath
* @return {!Promise<?Error>}
*/
function extractZip(zipPath, folderPath) {
2017-06-21 20:51:06 +00:00
return new Promise(fulfill => extract(zipPath, {dir: folderPath}, fulfill));
}