From ef1fdbed71d0dd28aac980a0c95dcec783d02615 Mon Sep 17 00:00:00 2001
From: psychobunny <psycho.bunny@hotmail.com>
Date: Thu, 6 Feb 2014 12:45:13 -0500
Subject: [PATCH] server-side voting logic

---
 src/favourites.js | 147 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 144 insertions(+), 3 deletions(-)

diff --git a/src/favourites.js b/src/favourites.js
index 935e4542da..5c3e9d5b74 100644
--- a/src/favourites.js
+++ b/src/favourites.js
@@ -8,11 +8,152 @@ var async = require('async'),
 (function (Favourites) {
 	"use strict";
 
-	Favourites.favourite = function (pid, room_id, uid, socket) {
+	function vote(type, postData, pid, room_id, uid, socket) {
 		var	websockets = require('./socket.io');
 
 		if (uid === 0) {
+			return socket.emit('event:alert', {
+				alert_id: 'post_vote',
+				title: '[[vote.not_logged_in.title]]',
+				message: '[[vote.not_logged_in.message]]',
+				type: 'danger',
+				timeout: 5000
+			});
+		} else if (uid === postData.uid) {
+			return socket.emit('event:alert', {
+				alert_id: 'post_vote',
+				title: '[[vote.cant_vote_self.title]]',
+				message: '[[vote.cant_vote_self.message]]',
+				type: 'danger',
+				timeout: 5000
+			});
+		}
+
+		//Favourites.hasVoted(type, pid, uid, function (err, hasVoted) {
+		//	if (!hasVoted) {
+				var notType = (type === 'upvote' ? 'downvote' : 'upvote');
+
+				db[type === 'upvote' ? 'sortedSetAdd' : 'sortedSetRemove']('uid:' + uid + ':upvote', postData.timestamp, pid);
+				db[type === 'upvote' ? 'sortedSetRemove' : 'sortedSetAdd']('uid:' + uid + ':downvote', postData.timestamp, pid);
+
+
+				user[type === 'upvote' ? 'incrementUserFieldBy' : 'decrementUserFieldBy'](postData.uid, 'reputation', 1, function (err, newreputation) {
+					db.sortedSetAdd('users:reputation', newreputation, postData.uid);
+				});
+
+				db.setAdd('pid:' + pid + ':' + type, uid, function(err) {
+					db.setCount('pid:' + pid + ':' + type, function(err, count) {
+						console.log(count);
+						posts.setPostField(pid, type, count);
+					});
+				});
+
+				db.setRemove('pid:' + pid + ':' + notType, uid, function(err) {
+					db.setCount('pid:' + pid + ':' + notType, function(err, count) {
+						posts.setPostField(pid, notType, count);
+					});
+				});
+
+				if (room_id) {
+					websockets.in(room_id).emit('event:' + (type === 'upvote' ? 'rep_up' : 'rep_down'), {
+						uid: postData.uid,
+						pid: pid
+					});
+				}
+
+				socket.emit('posts.' + type, {
+					pid: pid
+				});
+		//	}
+		//});
+	}
+
+	Favourites.upvote = function(pid, room_id, uid, socket) {
+		Favourites.unvote(pid, room_id, uid, socket, function(postData) {
+			vote('upvote', postData, pid, room_id, uid, socket);
+		});
+	};
+
+	Favourites.downvote = function(pid, room_id, uid, socket) {
+		Favourites.unvote(pid, room_id, uid, socket, function(postData) {
+			vote('downvote', postData, pid, room_id, uid, socket);
+		});
+	};
 
+	Favourites.unvote = function(pid, room_id, uid, socket, callback) {
+		var	websockets = require('./socket.io');
+
+		async.parallel({
+			upvoted: function(each) {
+				db.isSetMember('pid:' + pid + ':upvote', uid, each);
+			},
+			downvoted: function(each) {
+				db.isSetMember('pid:' + pid + ':downvote', uid, each);
+			}
+		}, function(err, results) {
+			posts.getPostFields(pid, ['uid', 'timestamp'], function (err, postData) {
+				if (results.upvoted) {
+					db.sortedSetRemove('uid:' + uid + ':upvote');
+
+					db.setRemove('pid:' + pid + ':upvote', uid, function(err) {
+						db.setCount('pid:' + pid + ':upvote', function(err, count) {
+							posts.setPostField(pid, 'upvote', count);
+						});
+					});
+
+					user.decrementUserFieldBy(postData.uid, 'reputation', 1, function (err, newreputation) {
+						db.sortedSetAdd('users:reputation', newreputation, postData.uid);
+					});
+
+					if (room_id) {
+						websockets.in(room_id).emit('event:rep_down', {
+							uid: postData.uid,
+							pid: pid
+						});
+					}
+				} else if (results.downvoted) {
+					db.sortedSetRemove('uid:' + uid + ':downvote');
+
+					db.setRemove('pid:' + pid + ':downvote', uid, function(err) {
+						db.setCount('pid:' + pid + ':downvote', function(err, count) {
+							posts.setPostField(pid, 'downvote', count);
+						});
+					});
+
+					user.incrementUserFieldBy(postData.uid, 'reputation', 1, function (err, newreputation) {
+						db.sortedSetAdd('users:reputation', newreputation, postData.uid);
+					});
+
+					if (room_id) {
+						websockets.in(room_id).emit('event:rep_up', {
+							uid: postData.uid,
+							pid: pid
+						});
+					}
+				}
+
+				if (results.upvoted || results.downvoted) {
+					socket.emit('posts.unvote', {
+						pid: pid
+					});
+				}
+
+				if (callback) {
+					callback(postData);
+				}
+			});
+		});
+
+	};
+
+	Favourites.hasVoted = function(type, pid, uid, callback) {
+		db.isSetMember('pid:' + pid + ':' + type, uid, callback);
+	};
+
+	Favourites.favourite = function (pid, room_id, uid, socket) {
+		var	websockets = require('./socket.io');
+
+		if (uid === 0) {
 			translator.mget(['topic:favourites.not_logged_in.message', 'topic:favourites.not_logged_in.title'], function(err, results) {
 				socket.emit('event:alert', {
 					alert_id: 'post_favourite',
@@ -46,7 +187,7 @@ var async = require('async'),
 					}
 
 					if (room_id) {
-						websockets.in(room_id).emit('event:rep_up', {
+						websockets.in(room_id).emit('event:favourited', {
 							uid: uid !== postData.uid ? postData.uid : 0,
 							pid: pid
 						});
@@ -86,7 +227,7 @@ var async = require('async'),
 					}
 
 					if (room_id) {
-						websockets.in(room_id).emit('event:rep_down', {
+						websockets.in(room_id).emit('event:unfavourited', {
 							uid: uid !== uid_of_poster ? uid_of_poster : 0,
 							pid: pid
 						});