puppeteer/phantomjs/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.
 */

var fs = require('fs');
var path = require('path');
var vm = require('vm');
var url = require('url');

/**
 * @param {!Object} context
 * @param {string} scriptPath
 */
module.exports.create = function(context, scriptPath) {
    var 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() {
            var 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;
            var 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) {
            var 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;
}