Files
phantomjs/src/bootstrap.js
T
Ivan De Marino 75403737c4 Added "includeJs()" to "page".
* It includes a script in the page
* It uses a callback to ensure any code dependent on the include runs afterwards
* It uses the signal "javaScriptAlertSent" to do the trick (is there another way to be notified of the "onLoad" event from outside the page context?)
* It uses a "private" slot "_appendScriptElement" to pass the script url in the page context (is there a better way?)
2011-06-10 17:08:46 +01:00

105 lines
3.7 KiB
JavaScript

// This allows creating a new web page using the construct "new WebPage",
// which feels more natural than "phantom.createWebPage()".
window.WebPage = function() {
var page = phantom.createWebPage();
// deep copy
page.settings = JSON.parse(JSON.stringify(phantom.defaultPageSettings));
// private, don't touch this
page._handlers = {};
page.__defineSetter__("onLoadStarted", function(f) {
if (this._handlers && typeof this._handlers.loadStarted === 'function') {
try {
this.loadStarted.disconnect(this._handlers.loadStarted);
} catch (e) {}
}
this._handlers.loadStarted = f;
this.loadStarted.connect(this._handlers.loadStarted);
});
page.__defineSetter__("onLoadFinished", function(f) {
if (this._handlers && typeof this._handlers.loadFinished === 'function') {
try {
this.loadFinished.disconnect(this._handlers.loadFinished);
} catch (e) {}
}
this._handlers.loadFinished = f;
this.loadFinished.connect(this._handlers.loadFinished);
});
page.__defineSetter__("onResourceRequested", function(f) {
if (this._handlers && typeof this._handlers.resourceRequested === 'function') {
try {
this.resourceRequested.disconnect(this._handlers.resourceRequested);
} catch (e) {}
}
this._handlers.resourceRequested = f;
this.resourceRequested.connect(this._handlers.resourceRequested);
});
page.__defineSetter__("onResourceReceived", function(f) {
if (this.handlers && typeof this.handlers.resourceReceived === 'function') {
try {
this.resourceReceived.disconnect(this.handlers.resourceReceived);
} catch (e) {}
}
this.handlers.resourceReceived = f;
this.resourceReceived.connect(this.handlers.resourceReceived);
});
page.onAlert = function (msg) {};
page.onConsoleMessage = function (msg) {};
page.open = function () {
if (typeof this.onAlert === 'function') {
this.javaScriptAlertSent.connect(this.onAlert);
}
if (typeof this.onConsoleMessage === 'function') {
this.javaScriptConsoleMessageSent.connect(this.onConsoleMessage);
}
if (arguments.length === 1) {
this.openUrl(arguments[0], 'get', this.settings);
return;
}
if (arguments.length === 2) {
this.onLoadFinished = arguments[1];
this.openUrl(arguments[0], 'get', this.settings);
return;
} else if (arguments.length === 3) {
this.onLoadFinished = arguments[2];
this.openUrl(arguments[0], arguments[1], this.settings);
return;
} else if (arguments.length === 4) {
this.onLoadFinished = arguments[3];
this.openUrl(arguments[0], {
operation: arguments[1],
data: arguments[2]
}, this.settings);
return;
}
throw "Wrong use of WebPage#open";
};
page.includeJs = function(scriptUrl, onScriptLoaded) {
// Register temporary signal handler for 'alert()'
this.javaScriptAlertSent.connect(function(msgFromAlert) {
if ( msgFromAlert === scriptUrl ) {
// Resource loaded, time to fire the callback
onScriptLoaded(scriptUrl);
// And disconnect the signal handler
try {
this.javaScriptAlertSent.disconnect(this);
} catch (e) {}
}
});
// Append the script tag to the body
this._appendScriptElement(scriptUrl);
};
return page;
}