initial commit.

helpers/phantom.js

@@ -0,0 +1,64 @@
+'use strict';
+
+require('../models/schema');
+var config = require('config');
+var phantom = require('node-phantom-simple');
+
+module.exports = {
+  // type = "pdf" or "png"
+  takeScreenshot: function(space,type,on_success,on_error) {
+    var spaceId = space._id;
+    var space_url = config.get("endpoint")+"/api/spaces/"+spaceId+"/html";
+
+    var export_path = "/tmp/"+spaceId+"."+type;
+
+    var timeout = 5000;
+    if (type=="pdf") timeout = 30000;
+
+    space_url += "?api_token="+config.get("phantom_api_secret");
+
+    console.log("[space-screenshot] url: "+space_url);
+    console.log("[space-screenshot] export_path: "+export_path);
+
+    var on_success_called = false;
+
+    var on_exit = function(exit_code) {
+      if (exit_code>0) {
+        console.log("phantom abnormal exit for url "+space_url);
+        if (!on_success_called && on_error) {
+          on_error();
+        }
+      }
+    };
+
+    phantom.create({ path: require('phantomjs-prebuilt').path }, function (err, browser) {
+      return browser.createPage(function (err, page) {
+        console.log("page created, opening ",space_url);
+
+        if (type=="pdf") {
+          var psz = {
+            width: space.advanced.width+"px",
+            height: space.advanced.height+"px"
+          };
+          page.set('paperSize', psz);
+        }
+
+        page.set('settings.resourceTimeout',timeout);
+        page.set('settings.javascriptEnabled',false);
+
+        return page.open(space_url, function (err,status) {
+          page.render(export_path, function() {
+            on_success_called = true;
+            if (on_success) {
+              on_success(export_path);
+            }
+            page.close();
+            browser.exit();
+          });
+        });
+      });
+    }, {
+      onExit: on_exit
+    });
+  }
+};