3
0
mirror of https://github.com/reality/dbot.git synced 2024-12-24 03:33:07 +01:00

Added AV vote counting.

This commit is contained in:
Joe MacMahon 2012-06-19 14:44:50 +01:00
parent c1c951f5c7
commit d192a17d19
3 changed files with 72 additions and 12 deletions

View File

@ -100,20 +100,19 @@ var poll = function(dbot) {
if(polls.hasOwnProperty(name)) {
if(polls[name].av) {
var prefs = vote.split(',');
prefs = prefs.uniq();
var valid = true;
prefs.each(function(pref) {
console.log(pref);
console.log(polls[name].options);
valid = valid && polls[name].options.hasOwnProperty(pref);
valid = valid && polls[name].options.indexOf(pref) != -1;
});
if(valid){
if(polls[name].votes.hasOwnProperty(event.user)) {
polls[name].votes[event.user] = prefs;
event.reply(dbot.t('av_changed_vote', {'vote': vote, 'poll': name, 'user': event.user}));
event.reply(dbot.t('av_changed_vote', {'vote': prefs.join(','), 'poll': name, 'user': event.user}));
} else {
polls[name].votes[event.user] = prefs;
event.reply(dbot.t('av_voted', {'vote': vote, 'poll': name, 'user': event.user}));
event.reply(dbot.t('av_voted', {'vote': prefs.join(','), 'poll': name, 'user': event.user}));
}
} else {
event.reply(dbot.t('invalid_vote', {'vote': vote}));
@ -151,33 +150,70 @@ var poll = function(dbot) {
}
},
'~count': function(event) {
'~winner': function(event) {
var name = event.input[1];
if(polls.hasOwnProperty(name)) {
var winner;
if(polls[name].av) {
var finished = false;
var rounds = [];
var eliminated = [];
var voted;
for(var roundn = 0; !finished; roundn++) {
var roundLoser;
// Populate candidates for this round
rounds[roundn] = {};
polls[name].options.each(function (option) {
if(!eliminated.hasOwnProperty(option))
if(eliminated.indexOf(option) == -1)
rounds[roundn][option] = 0;
});
// Count votes
polls[name].votes.each(function (name, vote) {
console.log(name);
voted = false;
vote.each(function (pref) {
if(!voted && rounds[roundn].hasOwnProperty(pref)) {
rounds[roundn][pref]++;
voted = true;
}
});
});
break;
// Check for 50%
var max = 0;
var min = polls[name].votes.length()
rounds[roundn].each(function (option, count) {
if(count > max) {
winner = option;
max = count;
}
if(count < min) {
roundLoser = option;
min = count;
}
});
if((2*max) > polls[name].votes.length()) {
finished = true;
break;
}
// Eliminate loser
eliminated.push(roundLoser);
}
} else {
event.reply("Not yet implemented.");
var max = 0;
polls[name].votes.each(function (name, count) {
if(count > max) {
winner = name;
max = count;
}
});
}
event.reply(dbot.t('winner', {'poll': name, 'description': polls[name].description, 'winner': winner}));
} else {
event.reply(dbot.t('poll_unexistent', {'name': name}));
}
@ -188,7 +224,7 @@ var poll = function(dbot) {
commands['~rmoption'].regex = [/~rmoption ([^ ]+) ([^ ]+)/, 3];
commands['~vote'].regex = [/~vote ([^ ]+) ([^ ]+)/, 3];
commands['~pdesc'].regex = [/~pdesc ([^ ]+)/, 2];
commands['~count'].regex = [/~count ([^ ]+)/, 2];
commands['~winner'].regex = [/~winner ([^ ]+)/, 2];
return {
'name': 'poll',

View File

@ -48,6 +48,18 @@ Array.prototype.allGroupings = function() {
return groupings;
}
Array.prototype.uniq = function() {
var hash = {}
var result = [];
this.each(function(item) {
if(!hash.hasOwnProperty(item)){
hash[item] = true;
result.push(item);
}
});
return result;
}
/*** String ***/
String.prototype.valMatch = function(regex, expLength) {
@ -158,6 +170,14 @@ Object.prototype.each = function(fun) {
fun(key, this[key]);
};
Object.prototype.length = function() {
var l = 0;
for(key in this)
if(this.hasOwnProperty(key))
l++;
return l;
}
/*** Integer ***/
Number.prototype.chanceIn = function(x, y) {

View File

@ -253,5 +253,9 @@
"av_changed_vote": {
"english": "{user} changed their vote in {poll} to '{vote}'.",
"spanish" : "{user} cambió su voto en {poll} a '{vote}'."
},
"winner": {
"english": "The winner of poll '{poll}' ({description}) is: '{winner}'.",
"spanish": "La ganera de la votación '{poll}' ({description}) es: '{winner}'."
}
}