routes.js

// This file is required by app.js. It sets up event listeners
// for the two main URL endpoints of the application - /create and /chat/:id
// and listens for socket.io messages.

// Use the gravatar module, to turn email addresses into avatar images:

var gravatar = require('gravatar');
var challenges = require('./flags.json');
console.log(challenges.hidden_path.route);
// Export a function, so that we can pass 
// the app and io instances from the app.js file:
module.exports = function(app,io,db){

	app.get('/', function(req, res){

		// Render views/home.html
		res.render('home');
	});

	app.get('/cz5Fc6sz7rppPf8B', function(req, res){
		var admin_token = req.get('x-admin-token');
		if (admin_token === "9PZxgVeZiMShe1KV"){
			res.render('admin_navbar', {flag: challenges.admin.flag});
		}else{
			res.status(500).send('Nope!');
		}
		
	});

	app.put("/" + challenges.hidden_path.route, function(req, res){
		res.status(200).send('Nice find! Here is your flag: ' + challenges.hidden_path.flag);
	});

	app.get('/create', function(req,res){

		// Generate unique id for the room
		var id = Math.round((Math.random() * 10000));

		// Redirect to the random room
		res.redirect('/chat/'+id);
	});

	app.get('/chat/:id', function(req,res){

		// Render the chant.html view
		res.render('chat');
	});

	// Initialize a new socket.io application, named 'chat'
	var chat = io.on('connection', function (socket) {
		console.log('New socket connection!');
		// When the client emits the 'load' event, reply with the 
		// number of people in this chat room

		socket.on('load',function(data){
			db.query("SELECT * from messages where room = ?;",
				[data],
				function(err, result){
					if (result.length != 0){
						chat.emit('tooMany', result);
						console.log(result);
					}
			});
			var room = findClientsSocket(io,data);
			if(room.length === 0 ) {

				socket.emit('peopleinchat', {number: 0});
			}
			else if(room.length === 1) {

				socket.emit('peopleinchat', {
					number: 1,
					user: room[0].username,
					avatar: room[0].avatar,
					id: data
				});
			}
			else if(room.length >= 2) {

			}
		});

		// When the client emits 'login', save his name and avatar,
		// and add them to the room
		socket.on('login', function(data) {

			var room = findClientsSocket(io, data.id);
			// Only two people per room are allowed
			if (room.length < 2) {

				// Use the socket object to store data. Each client gets
				// their own unique socket object

				socket.username = data.user;
				socket.room = data.id;
				socket.avatar = gravatar.url(data.avatar, {s: '140', r: 'x', d: 'mm'});

				// Tell the person what he should use for an avatar
				socket.emit('img', socket.avatar);


				// Add the client to the room
				socket.join(data.id);

				if (room.length == 1) {

					var usernames = [],
						avatars = [];

					usernames.push(room[0].username);
					usernames.push(socket.username);

					avatars.push(room[0].avatar);
					avatars.push(socket.avatar);

					// Send the startChat event to all the people in the
					// room, along with a list of people that are in it.

					chat.in(data.id).emit('startChat', {
						boolean: true,
						id: data.id,
						users: usernames,
						avatars: avatars
					});
				}
			}
			else {
				socket.emit('tooMany', {boolean: true});
			}
		});

		// Somebody left the chat
		socket.on('disconnect', function() {

			// Notify the other person in the chat room
			// that his partner has left

			socket.broadcast.to(this.room).emit('leave', {
				boolean: true,
				room: this.room,
				user: this.username,
				avatar: this.avatar
			});

			// leave the room
			socket.leave(socket.room);
		});


		// Handle the sending of messages
		socket.on('msg', function(data){

			// When the server receives a message, it sends it to the other person in the room.
			socket.broadcast.to(socket.room).emit('receive', {msg: data.msg, user: data.user, img: data.img});
			db.query("INSERT INTO messages (msg, user, room, avatar) VALUES (?, ?, ?, ?)",
			[data.msg, data.user, socket.room, socket.avatar],
			function(err, result){
					console.log('User ' + data.user + ' sent: ' 
						+ data.msg + ' in room: ' 
						+ socket.room + " with avatar: " 
						+ socket.avatar);
				});
			
		});
		socket.on('block', function() {
			// Notify the other person in the chat room
			// that his partner has left
			console.log('User Blocked');
			socket.broadcast.to(this.room).emit('leave', {
				boolean: true,
				room: this.room,
				user: this.username,
				avatar: this.avatar
			});

			// leave the room
			socket.leave(socket.room);

		});

	});
};

function findClientsSocket(io,roomId, namespace) {
	var res = [],
		ns = io.of(namespace ||"/");    // the default namespace is "/"

	if (ns) {
		for (var id in ns.connected) {
			if(roomId) {
				var index = ns.connected[id].rooms.indexOf(roomId) ;
				if(index !== -1) {
					res.push(ns.connected[id]);
				}
			}
			else {
				res.push(ns.connected[id]);
			}
		}
	}
	return res;
}