Splitted the node server in modules

2014-08-07 19:26:58 +02:00
parent 9b6b0f39f0
commit 7fcdf2aaa2
17 changed files with 407 additions and 250 deletions
@@ -0,0 +1,23 @@
+/**
+ * Yellow Lab Tools home page controller
+ */
+
+var async   = require('async');
+var fs      = require ('fs');
+
+var indexController = function(req, res) {
+    'use strict';
+
+    async.parallel({
+        
+        htmlTemplate: function(callback) {
+            fs.readFile('./app/node_views/index.html', {encoding: 'utf8'}, callback);
+        }
+
+    }, function(err, results) {
+        res.setHeader('Content-Type', 'text/html');
+        res.send(results.htmlTemplate);
+    });
+ };
+
+ module.exports = indexController;
@@ -0,0 +1,82 @@
+/**
+ * Controller for the test launching page (the waiting page, after the user submited a test on the index page)
+ */
+
+var async   = require('async');
+var fs      = require ('fs');
+
+var launchTestController = function(req, res, testQueue) {
+    'use strict';
+
+    // Generate test id
+    var testId = (Date.now()*1000 + Math.round(Math.random()*1000)).toString(36);
+
+    var resultsPath = 'results/' + testId;
+    var phantomasResultsPath = resultsPath + '/results.json';
+    
+    var url = req.body.url;
+    if (url.indexOf('http://') !== 0 && url.indexOf('https://') !== 0) {
+        url = 'http://' + url;
+    }
+
+    var options = {};
+    if (req.body.timeout) {
+        options.timeout = req.body.timeout;
+    }
+
+    async.waterfall([
+        
+        function htmlTemplate(callback) {
+            fs.readFile('./app/node_views/launchTest.html', {encoding: 'utf8'}, callback);
+        },
+
+        function sendResponse(html, callback) {
+
+            html = html.replace('%%TEST_URL%%', url);
+            html = html.replace('%%TEST_ID%%', testId);
+
+            res.setHeader('Content-Type', 'text/html');
+            res.send(html);
+
+            callback();
+        },
+
+        function createFolder(callback) {
+            // Create results folder
+            fs.mkdir(resultsPath, callback);
+        },
+
+        function executePhantomas(callback) {
+            console.log('Adding test ' + testId + ' on ' + url + ' to the queue');
+            
+            var task = {
+                testId: testId,
+                url: url,
+                options: options
+            };
+
+            testQueue.push(task, callback);
+        },
+
+        function writeResults(json, resultsObject, callback) {
+            console.log('Saving Phantomas results file to ' + phantomasResultsPath);
+            fs.writeFile(phantomasResultsPath, JSON.stringify(json, null, 4), callback);
+        }
+
+    ], function(err) {
+        if (err) {
+            console.log('An error occured while launching the phantomas test : ', err);
+
+            fs.writeFile(phantomasResultsPath, JSON.stringify({url: url, error: err}, null, 4), function(err) {
+                if (err) {
+                    console.log('Could not even write an error message on file ' + phantomasResultsPath);
+                    console.log(err);
+                }
+            });
+        } else {
+            testQueue.testComplete(testId);
+        }
+    });
+ };
+
+ module.exports = launchTestController;
@@ -0,0 +1,41 @@
+/**
+ * The page that dispays the results
+ */
+
+var async   = require('async');
+var fs      = require ('fs');
+
+var resultsController = function(req, res) {
+    'use strict';
+
+    var testId = req.params.testId;
+    var resultsPath = 'results/' + testId;
+    var phantomasResultsPath = resultsPath + '/results.json';
+
+    console.log('Opening test ' + testId + ' results as HTML');
+
+    async.parallel({
+        
+        htmlTemplate: function(callback) {
+            fs.readFile('./app/node_views/results.html', {encoding: 'utf8'}, callback);
+        },
+
+        phantomasResults: function(callback) {
+            fs.readFile(phantomasResultsPath, {encoding: 'utf8'}, callback);
+        }
+
+    }, function(err, results) {
+        if (err) {
+            console.log(err);
+            return res.status(404).send('Sorry, test not found...');
+        }
+
+        var html = results.htmlTemplate;
+        html = html.replace('%%RESULTS%%', results.phantomasResults);
+
+        res.setHeader('Content-Type', 'text/html');
+        res.send(html);
+    });
+};
+
+module.exports = resultsController;
@@ -0,0 +1,53 @@
+/**
+ * Socket.io handler
+ */
+
+var fs = require('fs');
+
+var waitingQueueSocket = function(socket, testQueue) {
+    
+    socket.on('waiting', function(testId) {
+        console.log('User waiting for test id ' + testId);
+
+        sendTestStatus(testId);
+
+        testQueue.on('testComplete', function(id) {
+            if (testId === id) {
+                socket.emit('complete');
+                console.log('Sending complete event to test id ' + testId);
+            }
+        });
+
+        testQueue.on('queueMoving', function() {
+            var positionInQueue = testQueue.indexOf(testId);
+            if (positionInQueue >= 0) {
+                socket.emit('position', positionInQueue);
+                console.log('Sending position to test id ' + testId);
+            }
+        });
+    });
+
+    // Finds the status of a test and send it to the client
+    function sendTestStatus(testId) {
+        // Check task position in queue
+        var positionInQueue = testQueue.indexOf(testId);
+
+        if (positionInQueue >= 0) {
+            socket.emit('position', positionInQueue);
+        } else {
+            // Find in results files
+            var exists = fs.exists('results/' + testId + '/results.json', function(exists) {
+                if (exists) {
+                    // TODO : find a way to make sure the file is completely written
+                    setTimeout(function() {
+                        socket.emit('complete');
+                    }, 4000);
+                } else {
+                    socket.emit('404');
+                }
+            });
+        }
+    }
+};
+
+module.exports = waitingQueueSocket;