app.js

// Expose modules in ./support for demo purposes
require.paths.unshift(__dirname + '/../../support');

/**
 * Module dependencies.
 */
var express = require('express')
	, crypto = require('crypto')
	, ejs = require('ejs')
	, url = require('url')
	, formidable = require('formidable')
	, sys = require('sys')
	, util = require('util')
	, fs = require('fs')
	, path = require('path')
	, data = require(__dirname + '/data')
	, docs = require(__dirname + '/docs');


var app = module.exports = express.createServer();

// Optional since express defaults to CWD/views
app.set('views', __dirname + '/views');
// Set our default template engine to "ejs"
app.set('view engine', 'html');
app.register('.html', require('ejs'));
//Public directory
var pub = __dirname + '/public';
app.use(express.static(pub));


//Uploads directory
var uploads = __dirname + '/fileUploads';
app.use(express.static(uploads));

app.use(express.bodyParser());
app.use(express.cookieParser());
app.use(express.session({secret: 'secret salt yay!'}));
app.use('/', express.errorHandler({dump:true, stack:true}));

app.dynamicHelpers({
	messages: require('express-messages')
});

app.dynamicHelpers({
	message: function(req){
		var err = req.session.error
			,	msg = req.session.success;
			delete req.session.error;
			delete req.session.success;
			if (err) return err;
			if (msg) return msg;
	}
});

app.get('/', function(req, res){
  res.render('home');
});

app.get('/login', function(req, res){
	if (req.session.user) {
		req.session.success = 'Welcome, ' + 
													'<a href=/user/' + req.session.user.username + 
													'>' + req.session.user.username + '</a>';
	} else {
		req.session.success = 'Please login:';
	}
	res.render('login');
});

app.post('/login', function(req,res){
	console.log('username:' + req.body.username + ' password:' + req.body.password);
	authenticate(req.body.username, req.body.password, function(err, user){
		if (user){
			req.session.regenerate(function(){
				req.session.user = user;
				//res.render('user', {user: user});
				res.redirect('/user/' + req.body.username);
			});
		} else {
			req.session.error = err.message;
			res.redirect('back');
		}
	});
});

app.get('/user/:username?', function(req, res){
	if (req.session.user){
		var username = "";
		if (req.params.username) username = req.params.username;
		else username = req.session.user.username;
		
		if (req.session.user.username == username){
			data.getPageList(req.session.user, function(err, userPages){
				if (err) {
					console.log("Grave error when getting page list!")
					return;
				}
				res.render('user', {user: req.session.user, pages: userPages});
			});
		} else {
			res.redirect('/logout');
		}
	} else {
		res.redirect('/logout');
	}
});

app.get('/user/:username/:pageNum.html', function(req,res){
	data.getPage(req.params.username, req.params.pageNum, function(err, reply){
		if (err){
			console.log("page could not be found: " + req.params.username + "/" + req.params.pageNum);
			res.render('404');
			//do a 404
		} else{
			res.render('page', {data: reply, layout: false});
		}
	});
});

app.get('/logout', function(req,res){
	//destroy the session
	req.session.destroy(function(){
		res.redirect('/login');
	});
});

app.get('/client/?', function(req,res){
	if (req.session.user){
		if (req.query.user){
			var username = req.query.user;
			var page = req.query.page;
			data.getDirtyBit(username, page, function(reply){
				if (reply){
					res.render('editor', {layout: false});
				} else{
					res.render('client', {layout: false});
				}
			});
			
		} else{
			res.render('client', {layout: false});
		}
	} else {
		res.redirect('/logout');
	}
});

app.get('/processed/:username/:pageNum', function(req,res){
	data.getProcessedPage(req.params.username, req.params.pageNum, function(err, reply){
		if (reply){
			res.send(reply);
		} else{
			console.log('There was an error getting the processed page');
		}
	});
});

app.get('/dirty/:username/:pageNum', function(req,res){
	data.getDirtyBit(req.params.username, req.params.pageNum, function(reply){
		if (reply){
			res.send({dirty: true});
		} else{
			res.send({dirty: false});
		}
	});
});

app.post('/savepage/:pageNum?', function(req,res){
	if (req.session.user){
		var username = req.session.user.username;
		if (req.params.pageNum){
			var pageNum = req.params.pageNum;
			data.savePage(username, req.body.data, pageNum);
			data.saveProcessedPage(username, req.body.canvas, req.body.style, pageNum);
		} else{
			data.savePage(username, req.body.data);
			data.saveProcessedPage(username, req.body.canvas, req.body.style)
		}
	} else {
		res.redirect('/logout');
	}
});

app.get('/register', function(req,res){
	res.render('register');
});

app.post('/register', function(req,res){
	register(req.body.username, req.body.password, function(err, user){
		if (user){
			req.session.regenerate(function(){
				req.session.user = user;
				//res.render('user', {user: user});
				res.redirect('/user/' + user.username);
			});
		} else{
			req.flash('error', err.message);
			res.redirect('back');
		}
	});
});

app.post('/fetch', function(req,res){
	if (req.session.user){
		console.log("fetching " + req.body.pageUrl);
		
		docs.loadDocument(req.body.pageUrl, function(err, allHtml, html, style){
			if (err){
				res.redirect('404')
			} else{
				data.savePage(req.session.user.username, allHtml, false, true);
				data.saveProcessedPage(req.session.user.username, html, style);
				res.redirect('/user');
			}
		});
	} else{
		res.redirect('/logout');
	}
});

app.get('/img/:imagename', function(req,res){
     
     returnImage(req,res,req.session.user.username, req.params.imagename, true);

});

app.get('/imglist', function(req,res){
	listImages(req, res, req.session.user.username);
});

app.post('/img', function(req,res){

	//data.saveImage(urlObj.query['qqfile'], req.body.data);
	upload_file(req,res,req.session.user.username );


});

app.get('/img/:imagename', function(req,res){
     
     returnImage(req,res,req.session.user.username, req.params.imagename);

});

app.listen(3000);
console.log('Express app started on port 3000');

//Helper functions bellow
function authenticate(username, pass, fn){
	data.getUser(username, function(user){
		//user exists?
		if (!user) return fn(new Error('Invalid username'));
		//Check the password
		if (user.password == pass) return fn(null, user);
		//otherwise throw error, password is invalid
		fn(new Error('Invalid password'));
	});
};

function register(username, pass, fn){
	//Try to authenticate with credential to see if user exists
	authenticate(username, pass, function(err, user){
		if (user || (err.message === 'Invalid password')){
			return fn(new Error('Username is already taken'));
		}
		var newUser = data.addUser(username, pass);
		fn(null, newUser);
	});
};




/*
 * Handle file upload
 */
function upload_file(req, res, user) {

    // parse a file upload
   var form = new formidable.IncomingForm(),
       outstream,
       stats;



	
    form
      .on('filebegin', function(name, file) {
			file.path = "fileUploads/images/";
      })
      .on('file', function(field, file) {
      
	      path.exists('fileUploads/images/'+ user, function (exists) {

		      	if (exists) { 
		      		stats = fs.lstatSync('fileUploads/images/'+ user);		      	
		      		if (stats.isDirectory()) {
		      			
		        		
		        		fs.rename(file.path,'fileUploads/images/'+ user + "/"+file.name, function() { 
		        			console.log("Image saved");
		        		} );

		        	}
		        	else {
			        	
						//Make a folder for any file uploads
						fs.mkdir('fileUploads/images/' + user, 0777, function (err) {
						    if (err) {
						        console.log(err);
						    } else {
						        console.log('Directory created');
						    }
						});
						fs.rename(file.path,'fileUploads/images/'+ user + "/"+file.name, function() { 
							console.log("Image saved");
						} );

		        	}
		        
		        }
		        else {
		        	
					//Make a folder for any file uploads
					fs.mkdir('fileUploads/images/' + user, 0777, function (err) {
					    if (err) {
					        console.log(err);
					    } else {
					        console.log('Directory created');
					    }
					});
		        	fs.rename(file.path,'fileUploads/images/'+ user + "/"+file.name, function() { 
		        		console.log("Image saved");
		        	
		        	} );
		        }
	        });
        
		 
        
      })
      .on('end', function() {
        res.writeHead(200, {'content-type': 'text/plain'});
        res.end();
      });
    form.parse(req);


};

function listImages(req, res, user) {
	
		var list= "";
		
		path.exists('fileUploads/images/'+ user, function (exists) {

		      	if (exists) { 
		      		stats = fs.lstatSync('fileUploads/images/'+ user);		      	
		      		if (stats.isDirectory()) {
						fs.readdir('fileUploads/images/'+ user, function(err,files){
							
							if (files) {
								for (file in files) {
									if ((files[file] != '') && (files[file] != '.DS_Store')){
										list+= files[file] + ",";
									}
								}
							}
	        				res.writeHead(200, {'content-type': 'text/plain'});
    						res.end(list);
							
						});
		        	}
		        	else {
			        	
						//Make a folder for any file uploads
						fs.mkdir('fileUploads/images/' + user, 0777, function (err) {
						    if (err) {
						        console.log(err);
						    } else {
						        console.log('Directory created');
						    }
						});
						res.writeHead(200, {'content-type': 'text/plain'});
    					res.end(list);
		        	}
		        
		        }
		        else {

					//Make a folder for any file uploads
					fs.mkdir('fileUploads/images/' + user, 0777, function (err) {
					    if (err) {
					        console.log(err);
					    } else {
					        console.log('Directory created');
					    }
					});

	        		res.writeHead(200, {'content-type': 'text/plain'});
    				res.end(list);

		        }
		        

	  });

};

function returnImage(req, res, user, image) {
	
		var stats;
		

		
		path.exists('fileUploads/images/'+ user, function (exists) {

		      	if (exists) { 
		      		stats = fs.lstatSync('fileUploads/images/'+ user + "/" + image);		      	
		      		if (stats.isFile()) {

     						res.writeHead(200, {'content-type': 'text/plain'});
     						res.end('/images/'+user+'/'+ image);	
							
						
		        	}
		        	else {
						res.render('404');
		        	}
		        
		        }
		        else {

					res.render('404');

		        }
		        

	  });

};