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

setTimeout(function() {
  system.stderr.writeLine("Timing out...");
  phantom.exit(2);
}, 5000);

var userAgent = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_2) AppleWebKit/537.17 (KHTML, like Gecko) Chrome/24.0.1312.57 Safari/537.17';

var url    = system.args[1];
var format = system.args[2] || "PNG";
var width  = parseInt(system.args[3]) || 1280;
var height = parseInt(system.args[4]); // default: fit content
var crop   = true;

// NOTE: This does not work as "window" size, so height will
// not do as expected here.
//
// See https://github.com/ariya/phantomjs/issues/10619
page.viewportSize = {
  width: width,
  height: height
};

page.customHeaders = {
  url: 'https://grabshot.herokuapp.com'
};
page.onInitialized = function() {
  // Make sure assets are fetched with the write Referer (else
  // TypeKit and possibly others don't work...).
  page.customHeaders.Referer = url;
};

page.settings.userAgent = userAgent;

function render() {
  var result = page.evaluate(function () {
    return {
      title: document.title,
      width: document.body.clientWidth,
      height: document.body.clientHeight,
    };
  });

  if(crop && height && result.height > height) {
    // Cropping isn't exactly what we want, but PhantomJS does
    // not yet have a "window size" concept (See NOTE above).
    page.clipRect = {
      top: 0,
      left: 0,
      width: width,
      height: height
    };

    result.height = height;
    result.width  = width;
  }

  result.imageData = page.renderBase64(format);
  result.format = format;

  system.stdout.writeLine(JSON.stringify(result));
  phantom.exit();
}

function handleError(type, msg, trace) {
  var msgStack = [type + 'ERROR: ' + msg];
  if (trace) {
    msgStack.push('TRACE:');
    trace.forEach(function(t) {
      msgStack.push(' -> ' + (t.file || t.sourceURL) + ': ' + t.line + (t.function ? ' (in function ' + t.function + ')' : ''));
    });
  }
  system.stderr.write(msgStack.join('\n'));
}

page.onError = function(msg, trace) {
  handleError('PAGE', msg, trace);
};

phantom.onError = function(msg, trace) {
  handleError('PHANTOM', msg, trace);
  phantom.exit(1);
};

page.onLoadFinished = function (status) {
  var loaded = page.evaluate(function() { return !!document.body; });

  if(status == "success" || loaded == true) {
    setTimeout(render, 300);
  }
};

page.open(url);

@@ -156,17 +156,18 @@
 (2);%0A%7D, 
-5
+10
 000);%0A%0Av

Bump page timeout to 10s