puppeteer/phantom_shim/WebPage.js

/**
 * 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 await = require('./utilities').await;
let EventEmitter = require('events');
let fs = require('fs');
let path = require('path');
let PageEvents = require('../lib/Page').Events;

let noop = function() { };

class WebPage {
  /**
     * @param {!Browser} browser
     * @param {string} scriptPath
     * @param {!Object=} options
     */
  constructor(browser, scriptPath, options) {
    this._page = await(browser.newPage());
    this.settings = new WebPageSettings(this._page);

    options = options || {};
    options.settings = options.settings || {};
    if (options.settings.userAgent)
      this.settings.userAgent = options.settings.userAgent;
    if (options.viewportSize)
      await(this._page.setViewportSize(options.viewportSize));

    this.loading = false;
    this.loadingProgress = 0;
    this.clipRect = options.clipRect || {left: 0, top: 0, width: 0, height: 0};
    this.onConsoleMessage = null;
    this.onLoadFinished = null;
    this.onResourceError = null;
    this.onResourceReceived = null;
    this._onInitialized = undefined;
    this._deferEvaluate = false;

    this._currentFrame = this._page.mainFrame();

    this.libraryPath = path.dirname(scriptPath);

    this._onResourceRequestedCallback = undefined;
    this._onConfirmCallback = undefined;
    this._onAlertCallback = undefined;
    this._onError = noop;

    this._pageEvents = new AsyncEmitter(this._page);
    this._pageEvents.on(PageEvents.Response, response => this._onResponseReceived(response));
    this._pageEvents.on(PageEvents.RequestFinished, request => this._onRequestFinished(request));
    this._pageEvents.on(PageEvents.RequestFailed, event => (this.onResourceError || noop).call(null, event));
    this._pageEvents.on(PageEvents.ConsoleMessage, msg => (this.onConsoleMessage || noop).call(null, msg));
    this._pageEvents.on(PageEvents.Confirm, message => this._onConfirm(message));
    this._pageEvents.on(PageEvents.Alert, message => this._onAlert(message));
    this._pageEvents.on(PageEvents.Dialog, dialog => this._onDialog(dialog));
    this._pageEvents.on(PageEvents.PageError, error => (this._onError || noop).call(null, error.message, error.stack));
  }

  /**
   * @return {string}
   */
  get frameName() {
    return this._currentFrame.name();
  }

  /**
   * @return {number}
   */
  get framesCount() {
    return this._currentFrame.childFrames().length;
  }

  /**
   * @return {!Array<string>}
   */
  get framesName() {
    return this._currentFrame.childFrames().map(frame => frame.name());
  }

  /**
   * @return {string}
   */
  get focusedFrameName() {
    let focusedFrame = this._focusedFrame();
    return focusedFrame ? focusedFrame.name() : '';
  }

  /**
   * @return {?Frame}
   */
  _focusedFrame() {
    let frames = this._currentFrame.childFrames().slice();
    frames.push(this._currentFrame);
    let promises = frames.map(frame => frame.evaluate(() => document.hasFocus()));
    let result = await(Promise.all(promises));
    for (let i = 0; i < result.length; ++i) {
      if (result[i])
        return frames[i];
    }
    return null;
  }

  switchToFocusedFrame() {
    let frame = this._focusedFrame();
    this._currentFrame = frame;
  }

  /**
   * @param {(string|number)} frameName
   * @return {boolean}
   */
  switchToFrame(frameName) {
    let frame = null;
    if (typeof frameName === 'string')
      frame = this._currentFrame.childFrames().find(frame => frame.name() === frameName);
    else if (typeof frameName === 'number')
      frame = this._currentFrame.childFrames()[frameName];
    if (!frame)
      return false;
    this._currentFrame = frame;
    return true;
  }

  /**
   * @return {boolean}
   */
  switchToParentFrame() {
    let frame = this._currentFrame.parentFrame();
    if (!frame)
      return false;
    this._currentFrame = frame;
    return true;
  }

  switchToMainFrame() {
    this._currentFrame = this._page.mainFrame();
  }

  get onInitialized() {
    return this._onInitialized;
  }

  set onInitialized(value) {
    if (typeof value !== 'function')
      this._onInitialized = undefined;
    else
      this._onInitialized = value;
  }

  /**
     * @return {?function(!Object, !Request)}
     */
  get onResourceRequested() {
    return this._onResourceRequestedCallback;
  }

  /**
     * @return {?function(!Object, !Request)} callback
     */
  set onResourceRequested(callback) {
    this._onResourceRequestedCallback = callback;
    this._page.setRequestInterceptor(callback ? resourceInterceptor : null);

    /**
         * @param {!InterceptedRequest} request
         */
    function resourceInterceptor(request) {
      let requestData = new RequestData(request);
      let phantomRequest = new PhantomRequest(request);
      callback(requestData, phantomRequest);
      if (phantomRequest._aborted)
        request.abort();
      else
        request.continue();
    }
  }

  _onResponseReceived(response) {
    if (!this.onResourceReceived)
      return;
    let phantomResponse = new PhantomResponse(response, false /* isResponseFinished */);
    this.onResourceReceived.call(null, phantomResponse);
  }

  _onRequestFinished(request) {
    if (!this.onResourceReceived)
      return;
    let phantomResponse = new PhantomResponse(request.response(), true /* isResponseFinished */);
    this.onResourceReceived.call(null, phantomResponse);
  }

  /**
     * @param {string} url
     * @param {function()} callback
     */
  includeJs(url, callback) {
    this._page.addScriptTag(url).then(callback);
  }

  /**
     * @return {!{width: number, height: number}}
     */
  get viewportSize() {
    return this._page.viewportSize();
  }

  /**
     * @return {!Object}
     */
  get customHeaders() {
    return this._page.httpHeaders();
  }

  /**
     * @param {!Object} value
     */
  set customHeaders(value) {
    await(this._page.setHTTPHeaders(value));
  }

  /**
     * @param {string} filePath
     */
  injectJs(filePath) {
    if (!fs.existsSync(filePath))
      filePath = path.resolve(this.libraryPath, filePath);
    if (!fs.existsSync(filePath))
      return false;
    await(this._page.injectFile(filePath));
    return true;
  }

  /**
     * @return {string}
     */
  get plainText() {
    return await(this._page.plainText());
  }

  /**
     * @return {string}
     */
  get title() {
    return await(this._page.title());
  }

  /**
     * @return {(function()|undefined)}
     */
  get onError() {
    return this._onError;
  }

  /**
     * @param {(function()|undefined)} handler
     */
  set onError(handler) {
    if (typeof handler !== 'function')
      handler = undefined;
    this._onError = handler;
  }

  /**
     * @return {(function()|undefined)}
     */
  get onConfirm() {
    return this._onConfirmCallback;
  }

  /**
     * @param {function()} handler
     */
  set onConfirm(handler) {
    if (typeof handler !== 'function')
      handler = undefined;
    this._onConfirmCallback = handler;
  }

  /**
     * @return {(function()|undefined)}
     */
  get onAlert() {
    return this._onAlertCallback;
  }

  /**
     * @param {function()} handler
     */
  set onAlert(handler) {
    if (typeof handler !== 'function')
      handler = undefined;
    this._onAlertCallback = handler;
  }

  /**
     * @param {!Dialog} dialog
     */
  _onDialog(dialog) {
    if (dialog.type === 'alert' && this._onAlertCallback) {
      this._onAlertCallback.call(null, dialog.message());
      await(dialog.accept());
    } else if (dialog.type === 'confirm' && this._onConfirmCallback) {
      let result = this._onConfirmCallback.call(null, dialog.message());
      await(result ? dialog.accept() : dialog.dismiss());
    }
  }

  /**
     * @return {string}
     */
  get url() {
    return await(this._page.url());
  }

  /**
     * @param {string} html
     */
  set content(html) {
    await(this._page.setContent(html));
  }

  /**
     * @param {string} html
     * @param {function()=} callback
     */
  open(url, callback) {
    console.assert(arguments.length <= 2, 'WebPage.open does not support METHOD and DATA arguments');
    this._deferEvaluate = true;
    if (typeof this._onInitialized === 'function')
      this._onInitialized();
    this._deferEvaluate = false;
    this.loading = true;
    this.loadingProgress = 50;
    this._page.navigate(url).then(result => {
      this.loadingProgress = 100;
      this.loading = false;
      let status = result ? 'success' : 'fail';
      if (!result) {
        this.onResourceError.call(null, {
          url,
          errorString: 'SSL handshake failed'
        });
      }
      if (this.onLoadFinished)
        this.onLoadFinished.call(null, status);
      if (callback)
        callback.call(null, status);
      this.loadingProgress = 0;
    });
  }

  /**
     * @param {!{width: number, height: number}} options
     */
  set viewportSize(options) {
    await(this._page.setViewportSize(options));
  }

  /**
     * @param {function()} fun
     * @param {!Array<!Object>} args
     */
  evaluate(fun, ...args) {
    if (this._deferEvaluate)
      return await(this._page.evaluateOnInitialized(fun, ...args));
    return await(this._currentFrame.evaluate(fun, ...args));
  }

  /**
     * {string} fileName
     */
  render(fileName) {
    if (fileName.endsWith('pdf')) {
      let options = {};
      let paperSize = this.paperSize || {};
      options.margin = paperSize.margin;
      options.format = paperSize.format;
      options.landscape = paperSize.orientation === 'landscape';
      options.width = paperSize.width;
      options.height = paperSize.height;
      await(this._page.printToPDF(fileName, options));
    } else {
      let options = {};
      if (this.clipRect && (this.clipRect.left || this.clipRect.top || this.clipRect.width || this.clipRect.height)) {
        options.clip = {
          x: this.clipRect.left,
          y: this.clipRect.top,
          width: this.clipRect.width,
          height: this.clipRect.height
        };
      }
      options.path = fileName;
      await(this._page.screenshot(options));
    }
  }

  release() {
    this._page.close();
  }

  close() {
    this._page.close();
  }
}

class WebPageSettings {
  /**
     * @param {!Page} page
     */
  constructor(page) {
    this._page = page;
  }

  /**
     * @param {string} value
     */
  set userAgent(value) {
    await(this._page.setUserAgent(value));
  }

  /**
     * @return {string}
     */
  get userAgent() {
    return this._page.userAgent();
  }
}

class PhantomRequest {
  /**
   * @param {!InterceptedRequest} request
   */
  constructor(request) {
    this._request = request;
  }

  /**
   * @param {string} key
   * @param {string} value
   */
  setHeader(key, value) {
    this._request.headers.set(key, value);
  }

  abort() {
    this._aborted = true;
  }

  /**
   * @param {string} url
   */
  changeUrl(newUrl) {
    this._request.url = newUrl;
  }
}

class PhantomResponse {
  /**
   * @param {!Response} response
   * @param {boolean} isResponseFinished
   */
  constructor(response, isResponseFinished) {
    this.url = response.url;
    this.status = response.status;
    this.statusText = response.statusText;
    this.stage = isResponseFinished ? 'end' : 'start';
    this.headers = [];
    for (let entry of response.headers.entries()) {
      this.headers.push({
        name: entry[0],
        value: entry[1]
      });
    }
  }
}

class RequestData {
  /**
   * @param {!InterceptedRequest} request
   */
  constructor(request) {
    this.url = request.url,
    this.headers = {};
    for (let entry of request.headers.entries())
      this.headers[entry[0]] = entry[1];
  }
}

// To prevent reenterability, eventemitters should emit events
// only being in a consistent state.
// This is not the case for 'ws' npm module: https://goo.gl/sy3dJY
//
// Since out phantomjs environment uses nested event loops, we
// exploit this condition in 'ws', which probably never happens
// in case of regular I/O.
//
// This class is a wrapper around EventEmitter which re-emits events asynchronously,
// helping to overcome the issue.
class AsyncEmitter extends EventEmitter {
  /**
     * @param {!Page} page
     */
  constructor(page) {
    super();
    this._page = page;
    this._symbol = Symbol('AsyncEmitter');
    this.on('newListener', this._onListenerAdded);
    this.on('removeListener', this._onListenerRemoved);
  }

  _onListenerAdded(event, listener) {
    // Async listener calls original listener on next tick.
    let asyncListener = (...args) => {
      process.nextTick(() => listener.apply(null, args));
    };
    listener[this._symbol] = asyncListener;
    this._page.on(event, asyncListener);
  }

  _onListenerRemoved(event, listener) {
    this._page.removeListener(event, listener[this._symbol]);
  }
}

module.exports = WebPage;