puppeteer/phantom_shim/Phantom.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 fs = require('fs');
let path = require('path');
let vm = require('vm');
let url = require('url');

/**
 * @param {!Object} context
 * @param {string} scriptPath
 */
module.exports.create = function(context, scriptPath) {
  const phantom = {
    page: {
      onConsoleMessage: null,
    },

    /**
         * @param {string} relative
         * @param {string} base
         * @return {string}
         */
    resolveRelativeUrl: function(relative, base) {
      return url.resolve(base, relative);
    },

    /**
         * @param {string} url
         * @return {string}
         */
    fullyDecodeUrl: function(url) {
      return decodeURI(url);
    },

    libraryPath: path.dirname(scriptPath),

    onError: null,

    /**
         * @return {string}
         */
    get outputEncoding() {
      return 'UTF-8';
    },

    /**
         * @param {string} value
         */
    set outputEncoding(value) {
      throw new Error('Phantom.outputEncoding setter is not implemented');
    },

    /**
         * @return {boolean}
         */
    get cookiesEnabled() {
      return true;
    },

    /**
         * @param {boolean} value
         */
    set cookiesEnabled(value) {
      throw new Error('Phantom.cookiesEnabled setter is not implemented');
    },

    /**
         * @return {!{major: number, minor: number, patch: number}}
         */
    get version() {
      let versionParts = require('../package.json').version.split('.');
      return {
        major: parseInt(versionParts[0], 10),
        minor: parseInt(versionParts[1], 10),
        patch: parseInt(versionParts[2], 10),
      };
    },

    /**
         * @param {number=} code
         */
    exit: function(code) {
      process.exit(code);
    },

    /**
         * @param {string} filePath
         * @return {boolean}
         */
    injectJs: function(filePath) {
      filePath = path.resolve(phantom.libraryPath, filePath);
      if (!fs.existsSync(filePath))
        return false;
      let code = fs.readFileSync(filePath, 'utf8');
      if (code.startsWith('#!'))
        code = code.substring(code.indexOf('\n'));
      vm.runInContext(code, context, {
        filename: filePath,
        displayErrors: true
      });
      return true;
    },

    /**
         * @param {string} moduleSource
         * @param {string} filename
         */
    loadModule: function(moduleSource, filename) {
      let code = [
        '(function(require, exports, module) {\n',
        moduleSource,
        '\n}.call({},',
        'require.cache[\'' + filename + '\']._getRequire(),',
        'require.cache[\'' + filename + '\'].exports,',
        'require.cache[\'' + filename + '\']',
        '));'
      ].join('');
      vm.runInContext(code, context, {
        filename: filename,
        displayErrors: true
      });
    },
  };
  return phantom;
};