/** * 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. */ var http = require('http'); var path = require('path'); var removeRecursive = require('rimraf').sync; var Page = require('./Page'); var childProcess = require('child_process'); var Downloader = require('../utils/ChromiumDownloader'); var Connection = require('./Connection'); var CHROME_PROFILE_PATH = path.resolve(__dirname, '..', '.dev_profile'); var browserId = 0; var DEFAULT_ARGS = [ '--disable-background-timer-throttling', '--no-first-run', ]; class Browser { /** * @param {(!Object|undefined)} options */ constructor(options) { options = options || {}; ++browserId; this._userDataDir = CHROME_PROFILE_PATH + browserId; this._remoteDebuggingPort = 9229; if (typeof options.remoteDebuggingPort === 'number') this._remoteDebuggingPort = options.remoteDebuggingPort; this._chromeArguments = DEFAULT_ARGS.concat([ `--user-data-dir=${this._userDataDir}`, `--remote-debugging-port=${this._remoteDebuggingPort}`, ]); if (typeof options.headless !== 'boolean' || options.headless) { this._chromeArguments.push(...[ `--headless`, `--disable-gpu`, ]); } if (typeof options.executablePath === 'string') { this._chromeExecutable = options.executablePath; } else { var chromiumRevision = require('../package.json').puppeteer.chromium_revision; var revisionInfo = Downloader.revisionInfo(Downloader.currentPlatform(), chromiumRevision); console.assert(revisionInfo, 'Chromium revision is not downloaded. Run npm install'); this._chromeExecutable = revisionInfo.executablePath; } if (Array.isArray(options.args)) this._chromeArguments.push(...options.args); this._terminated = false; this._chromeProcess = null; } /** * @return {!Promise} */ async newPage() { await this._ensureChromeIsRunning(); if (!this._chromeProcess || this._terminated) throw new Error('ERROR: this chrome instance is not alive any more!'); var client = await Connection.create(this._remoteDebuggingPort); var page = await Page.create(this, client); return page; } /** * @param {!Page} page */ async closePage(page) { if (!this._chromeProcess || this._terminated) throw new Error('ERROR: this chrome instance is not running'); await page.close(); } /** * @return {string} */ async version() { await this._ensureChromeIsRunning(); var version = await Connection.version(this._remoteDebuggingPort); return version.Browser; } async _ensureChromeIsRunning() { if (this._chromeProcess) return; this._chromeProcess = childProcess.spawn(this._chromeExecutable, this._chromeArguments, {}); // Cleanup as processes exit. process.on('exit', () => this._chromeProcess.kill()); this._chromeProcess.on('exit', () => { this._terminated = true; removeRecursive(this._userDataDir); }); await waitForChromeResponsive(this._remoteDebuggingPort); } close() { if (!this._chromeProcess) return; this._chromeProcess.kill(); } } module.exports = Browser; function waitForChromeResponsive(remoteDebuggingPort) { var fulfill; var promise = new Promise(x => fulfill = x); var options = { method: 'GET', host: 'localhost', port: remoteDebuggingPort, path: '/json/list' }; var probeTimeout = 100; sendRequest(); return promise; function sendRequest() { var req = http.request(options, res => { fulfill(); }); req.on('error', e => setTimeout(sendRequest, probeTimeout)); req.end(); } }