forked from brl/citadel
57 lines
1.6 KiB
JavaScript
57 lines
1.6 KiB
JavaScript
var fs = require('fs');
|
|
var system = require('system');
|
|
var page = require('webpage').create();
|
|
|
|
// Examine console log for message from chart drawing
|
|
page.onConsoleMessage = function(msg) {
|
|
console.log(msg);
|
|
if (msg === "ALL CHARTS READY") {
|
|
window.charts_ready = true;
|
|
}
|
|
else if (msg.slice(0, 11) === "CHART READY") {
|
|
var chart_id = msg.split(" ")[2];
|
|
console.log('grabbing ' + chart_id);
|
|
var png_data = page.evaluate(function (chart_id) {
|
|
var chart_div = document.getElementById(chart_id + '_png');
|
|
return chart_div.outerHTML;
|
|
}, chart_id);
|
|
fs.write(args[2] + '/' + chart_id + '.png', png_data, 'w');
|
|
}
|
|
};
|
|
|
|
// Check command line arguments
|
|
var args = system.args;
|
|
if (args.length != 3) {
|
|
console.log("USAGE: " + args[0] + " REPORT_HTML OUT_DIR\n");
|
|
phantom.exit(1);
|
|
}
|
|
|
|
// Open the web page
|
|
page.open(args[1], function(status) {
|
|
if (status == 'fail') {
|
|
console.log("Failed to open file '" + args[1] + "'");
|
|
phantom.exit(1);
|
|
}
|
|
});
|
|
|
|
// Check status every 100 ms
|
|
interval = window.setInterval(function () {
|
|
//console.log('waiting');
|
|
if (window.charts_ready) {
|
|
clearTimeout(timer);
|
|
clearInterval(interval);
|
|
|
|
var fname = args[1].replace(/\/+$/, "").split("/").pop()
|
|
console.log("saving " + fname);
|
|
fs.write(args[2] + '/' + fname, page.content, 'w');
|
|
phantom.exit(0);
|
|
}
|
|
}, 100);
|
|
|
|
// Time-out after 10 seconds
|
|
timer = window.setTimeout(function () {
|
|
clearInterval(interval);
|
|
console.log("ERROR: timeout");
|
|
phantom.exit(1);
|
|
}, 10000);
|