index.js

//
// Computing With Large Data Sets
// Visualization project
//
// December 2011
// @author Josh Leitzel <josh.leitzel@nyu.edu>
//

var http = require('http'),
    _ = require('underscore')._,
    childProcess = require('child_process'),
    fs = require('fs'),
    express = require('express');

// Constants
var GRAPHS_DIR = 'graphs/',
    DATA_ROOT = 'data/',
    DATA_USER = 'uploads', // relative to DATA_ROOT
    SCRIPT_MAP = {
      cluster : 'r/cluster.r',
      regression : 'r/regression.r'
    };

// Clear tmp directory
console.log('Clearing public/tmp/ directory...');
childProcess.exec('rm public/tmp/*.tar');

var server = express.createServer();

server.use(express.bodyParser())
  .use(express.static(__dirname + '/public'))

server.get('/cleargraphs', function (req, res) {
  console.log('Clearing graphs...');
  childProcess.exec('rm public/graphs/*.png', function () {
    childProcess.exec('rm public/graphs/*.pdf', function () {
      res.end('done');
    });
  })
});

server.get('/genes', function (req, res) {
  console.log('Getting genes...');
  fs.readdir('clusters/dist', function (err, files) {
    res.end(_.map(files, function (file) {
      return file.substring(0, file.length - 7); // chop off the `.sqlite`
    }).join(','));
  });
});

server.get('/graphs', function (req, res) {
  console.log('Getting graphs...');
  fs.readdir('public/graphs', function (err, files) {
    res.end(_.map(_.filter(files, function (file) { return file.substring(file.length - 4, file.length) === '.png'; }), function (file) { return 'graphs/' + file; }).join(','));
  });
});

server.get('/tar', function (req, res) {
  console.log('Building tarball of pngs...');
  var publicTarURL = 'tmp/graphs_png_' + new Date().getTime() + '.tar';
  childProcess.exec('tar czf public/' + publicTarURL + ' public/graphs/*.png', function (error, stdout, stderr) {
    if (error) {
      throw error;
    }
    res.end(publicTarURL);
  });
});

server.get('/pdf', function (req, res) {
  console.log('Building tarball of pdfs...');
  var publicTarURL = 'tmp/graphs_pdf_' + new Date().getTime() + '.tar';
  childProcess.exec('tar czf public/' + publicTarURL + ' public/graphs/*.pdf', function (error, stdout, stderr) {
    if (error) {
      throw error;
    }
    res.end(publicTarURL);
  });
});

server.post('/process', function (req, res) {
  console.log('Processing graph...');
  var clusters = req.body.clusters ? 'clusters=' + req.body.clusters : '',
      genes = req.body.genes ? 'genes=' + req.body.genes : '',
      graph = 'graph=' + req.body.graph,
      process = req.body.process,
      visual = req.body.visual,
      options = req.body.options;

  if (process != 'cluster' && process != 'regression') {
    res.end('Error: data sent is not valid');
  }

  var args = _.union(['r/process.r'], graph, clusters, genes, visual, options);
  console.log('Invoking Rscript ' + args.join(' '));
  childProcess.exec('Rscript ' + args.join(' '), function (error, stdout, stderr) {
    if (error) {
      throw error;
    }
    res.end(stdout || stderr);
  });
});
server.listen(1337);