add screenshot functionality

felixcheung · Jan 23, 2015 · 5c04a49 · 5c04a49
1 parent 7879832
commit 5c04a49
Show file tree

Hide file tree

Showing 7 changed files with 2,914 additions and 12 deletions.
diff --git a/app/controllers/visualization.js b/app/controllers/visualization.js
@@ -1,6 +1,7 @@
 var models = require('../models');
 var Q = require('q');
 var _ = require('lodash');
+var webshot = require('webshot');
 
 
 exports.getData = function (req, res, next) {
@@ -298,4 +299,33 @@ exports.pym = function (req, res, next) {
                 }))
             });
     }).fail(next);
-};
+};
+
+
+
+exports.screenshot = function(req, res, next) {
+
+
+    var vizId = req.params.vid;
+    var host = req.headers.host;
+    var url = 'http://' + host + '/visualizations/' + vizId + '/iframe';
+
+    var width = req.query.width || 1024;
+    var height = req.query.height || 768;
+
+    var opts = {
+        screenSize: {
+            width: width,
+            height: height
+        }
+    };
+
+    webshot(url, opts, function(err, renderStream) {
+        res.setHeader('Content-Type', 'image/png');
+        renderStream.pipe(res);
+    });
+
+}
+
+
+
diff --git a/app/views/session/viz-data.jade b/app/views/session/viz-data.jade
@@ -31,6 +31,9 @@
                         li
                             a(href="/visualizations/#{viz.id}/data")
                                 | Raw Data
+                        li
+                            a(href="/visualizations/#{viz.id}/screenshot?width=1024&height=768")
+                                | Take Screenshot
                         li
                             a(href="/visualizations/#{viz.id}", data-attribute="permalink")
                                 | Internal Permalink

diff --git a/config/routes.js b/config/routes.js
@@ -41,6 +41,7 @@ module.exports = function (app) {
     app.post('/js/dynamic', staticController.bundleJSForExecution);
     app.post('/css/dynamic', staticController.buildSCSS);
     app.get('/visualizations/:vid/public', visualization.publicRead);
+    app.get('/visualizations/:vid/screenshot', visualization.screenshot);
     app.get('/sessions/:sid/public', session.publicRead);
     app.get('/visualizations/:vid/iframe', visualization.iframe);
     app.get('/visualizations/:vid/pym', visualization.pym);

diff --git a/package.json b/package.json
@@ -120,6 +120,7 @@
     "topojson": "^1.6.18",
     "validator": "~3.17.0",
     "view-helpers": "^0.1.5",
+    "webshot": "^0.15.3",
     "winston": "^0.7.3",
     "yargs": "^1.3.3"
   }