diff --git a/lib/screencap/raster.js b/lib/screencap/raster.js index 884abe3..ab7c319 100644 --- a/lib/screencap/raster.js +++ b/lib/screencap/raster.js @@ -34,12 +34,23 @@ var page = new WebPage(), // Functions // function pickupNamedArguments() { - var i, pair; - for(i = 0; i < phantom.args.length; i++) { - pair = phantom.args[i].split(/=(.*)/); - args[pair[0]] = pair[1]; + var pair, scriptArgs; + if (typeof phantom.args != 'undefined') { + // phantomjs < 2.0 + scriptArgs = phantom.args; + } else { + // phantomjs 2.0 + var system = require('system'); + scriptArgs = system.args; + // remove first arg as always script name + scriptArgs.shift(); } + scriptArgs.forEach(function(arg, i) { + pair = arg.split(/=(.*)/); + args[pair[0]] = pair[1]; + }); + if(!args.width) { args.width = 1024; } if(!args.dpi) { args.dpi = 1; } if(args.url) { args.url = decodeURIComponent(args.url); }