Adding support for PDF output

routes/index.js

@@ -15,16 +15,18 @@ module.exports = function(app, useCors) {
     }
 
     var url = utils.url(req.param('url'));
+    var format = req.param('format') || 'png';
+
     // required options
     var options = {
       uri: 'http://localhost:' + rasterizerService.getPort() + '/',
       headers: { url: url }
     };
-    ['width', 'height', 'clipRect', 'javascriptEnabled', 'loadImages', 'localToRemoteUrlAccessEnabled', 'userAgent', 'userName', 'password', 'delay'].forEach(function(name) {
+    ['width', 'height', 'clipRect', 'javascriptEnabled', 'loadImages', 'localToRemoteUrlAccessEnabled', 'userAgent', 'userName', 'password', 'delay', 'pdfPaperSize', 'pdfOrientation'].forEach(function(name) {
       if (req.param(name, false)) options.headers[name] = req.param(name);
     });
 
-    var filename = 'screenshot_' + utils.md5(url + JSON.stringify(options)) + '.png';
+    var filename = 'screenshot_' + utils.md5(url + JSON.stringify(options)) + '.'+format;
     options.headers.filename = filename;
 
     var filePath = join(rasterizerService.getPath(), filename);

scripts/rasterizer.js

@@ -17,6 +17,11 @@ defaultViewportSize = {
   height: ~~defaultViewportSize[1] || 600
 };
 
+var defaultPaperSize = {
+  format: 'Letter',
+  orientation: 'Portrait'
+};
+
 var pageSettings = ['javascriptEnabled', 'loadImages', 'localToRemoteUrlAccessEnabled', 'userAgent', 'userName', 'password'];
 
 var server, service;
@@ -72,6 +77,14 @@ service = server.listen(port, function(request, response) {
       width: request.headers.width || defaultViewportSize.width,
       height: request.headers.height || defaultViewportSize.height
     };
+
+    if(/\.pdf$/.test(request.headers.filename)) {
+      page.paperSize = {
+        format: request.headers.pdfPaperSize || defaultPaperSize.format,
+        orientation: request.headers.pdfOrientation || defaultPaperSize.orientation
+      };
+    }
+
     if (request.headers.clipRect) {
       page.clipRect = JSON.parse(request.headers.clipRect);
     }