//from https://github.com/makinacorpus/django-screamshot

var page = require('webpage').create(),
    system = require('system');

/** 
 * arguments:
 * [1] => URL
 * [2] => output. Use /dev/stdout if you want to capture.
 * [3] => size
 */

var address = system.args[1],
    output  = system.args[2];

config = {}
system.args.forEach(function(arg, i) {
    if (i > 2) {
        namev = arg.split('=');
        config[namev[0].replace('--', '')] = namev[1];
    }
});

var method = config.method || 'get',
     width = config.width || 1400,
    height = config.height || 1,
      wait = config.wait || 200;

/**
 * please note: the default height is intentionaly left as 1.
 * the thing is, if you skip the height, phantomjs falls back
 * to some default viewport. and that's not what we want. we
 * want to set the width, and let the height auto-calculate.
 */
page.viewportSize = {width: width, height: height}

format = config.format || 'png';

if (format == 'pdf') {
    page.paperSize = {
        format: 'A4',
        orientation: config.orientation || 'portrait',
        margin: config.margin || '0cm'
    }
}

page.settings.resourceTimeout = 10000;

page.open(address, function(status){
    if (status == 'success') {
        setTimeout(function(){
            page.render(output, {format: format});
            phantom.exit();
        }, wait);
    }
});