tracman-server/server.js

/* IMPORTS */
var express = require('express'),
	crash = require('express-crash'),
	bodyParser = require('body-parser'),
	cookieParser = require('cookie-parser'),
	session = require('express-session'),
	mongoose = require('mongoose'),
	nunjucks = require('nunjucks'),
	passport = require('passport'),
	flash = require('connect-flash'),
	secret = require('./config/secrets.js'),
	User = require('./config/models/user.js'),
	app = express(),
	http = require('http').Server(app),
	io = require('socket.io')(http);

/* SETUP */

//  Templates
nunjucks.configure(__dirname+'/views', {
	autoescape: true,
	express: app
});

//  Session
app.use(session({
	secret: secret.session,
	saveUninitialized: true,
	resave: true
}));
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({
	extended: true
}));
app.use(cookieParser(secret.cookie));
app.use(flash());

//  Auth
app.use(passport.initialize());
app.use(passport.session());
require('./config/auth.js');
passport.serializeUser(function(user,done) {
	done(null, user.id);
});
passport.deserializeUser(function(id,done) {
	User.findById(id, function(err, user) {
		if(!err) done(null, user);
		else done(err, null);
	});
});

//  Database
mongoose.connect(secret.mongoSetup, {
	server:{socketOptions:{
		keepAlive:1, connectTimeoutMS:30000 }},
	replset:{socketOptions:{
		keepAlive:1, connectTimeoutMS:30000 }}
});

//  Routes
app.use(
	require('./config/routes/index.js'),
	require('./config/routes/auth.js'),
	require('./config/routes/feedback.js'),
	require('./config/routes/misc.js')
);
app.use('/trac', require('./config/routes/trac.js'));
app.use('/invited', require('./config/routes/invite.js'));
app.use('/dashboard', require('./config/routes/dashboard.js'));
app.use('/admin', require('./config/routes/admin.js'));
app.use('/static', express.static(__dirname+'/static'));


// Handle errors
// if (secret.url=='https://tracman.org') {
	var handle404 = function(err,req,res,next) {
		if (err) { console.log('404 handling error: '+err); }
		res.render('error.html', {code:404});
	};
	var handle500 = function(err,req,res,next) {
		if (err) { console.log('500 handling error: '+err); }
		res.render('error.html', {code:500});
	};
	app.use(crash.handle404(handle404));
	app.use(crash.handle500(handle500));
	crash.trapRoute(app);
	crash.handle(app, handle404, handle500);
// }

/* RUNTIME */

// Check for tracking users
function checkForUsers(room) {
	if (room) {
		io.to('app-'+room).emit('activate',
			(io.of("/").adapter.rooms[room])?'true':'false'
		);
	} else {
		User.find({}, function(err, users){
			if (err) { console.log('Sockets error finding all users in all rooms: '+err); }
			users.forEach( function(user){
				checkForUsers(user.id);
			});
		});
	}
}

// Websockets
io.on('connection', function(socket) {
	
	socket.on('room', function(room) {
		socket.join(room);
		if (room.slice(0,4)!='app-'){
			User.findById({_id:room}, function(err, user) {
				if (err) { console.log('Sockets error finding tracked user of room '+room+'\n'+err); }
				if (user) {
					io.to('app-'+room).emit('activate','true'); }
			});
		} else {
			checkForUsers(room.slice(4));
		}
	});
	
	socket.on('app', function(loc){
		loc.time = Date.now();
		io.to(loc.usr).emit('trac', loc);
		User.findByIdAndUpdate(loc.usr, {last:{
			lat: parseFloat(loc.lat),
			lon: parseFloat(loc.lon),
			dir: parseFloat(loc.dir||0),
			spd: parseFloat(loc.spd||0),
			time: Date.now()
		}}, function(err, user) {
			if (err) { console.log('Could not update last location of user '+loc.user+'\n'+err); }
			if (!user) { console.log("No user found: "+loc.user); }
		});
	});

	socket.onclose = function(reason){
		var closedroom = Object.keys(
			socket.adapter.sids[socket.id]).slice(1)[0];
		setTimeout(function() {
			checkForUsers(closedroom);
		}, 3000);
		Object.getPrototypeOf(this).onclose.call(this,reason);
	}

});

// Listen
http.listen(secret.port, function(){
	console.log('Listening at '+secret.url+':'+secret.port);
	checkForUsers();
});

module.exports = app;