140 lines
3.8 KiB
JavaScript
140 lines
3.8 KiB
JavaScript
/**
|
|
* 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;
|
|
}
|