pokemon-showdown/chat-plugins/poll.js
2015-11-12 04:43:29 +01:00

273 lines
9.6 KiB
JavaScript

/*
* Poll chat plugin
* By bumbadadabum and Zarel.
*/
'use strict';
const permission = 'announce';
let Poll = (function () {
function Poll(room, question, options) {
if (room.pollNumber) {
room.pollNumber++;
} else {
room.pollNumber = 1;
}
this.room = room;
this.question = question;
this.voters = {};
this.voterIps = {};
this.totalVotes = 0;
this.timeout = null;
this.timeoutMins = 0;
this.options = new Map();
for (let i = 0; i < options.length; i++) {
this.options.set(i + 1, {name: options[i], votes: 0});
}
}
Poll.prototype.vote = function (user, option) {
let ip = user.latestIp;
let userid = user.userid;
if (userid in this.voters || ip in this.voterIps) {
return user.sendTo(this.room, "You have already voted for this poll.");
}
this.voters[userid] = option;
this.voterIps[ip] = option;
this.options.get(option).votes++;
this.totalVotes++;
this.update();
};
Poll.prototype.blankvote = function (user, option) {
let ip = user.latestIp;
let userid = user.userid;
if (userid in this.voters || ip in this.voterIps) {
return user.sendTo(this.room, "You're already looking at the results.");
} else {
this.voters[userid] = 0;
this.voterIps[ip] = 0;
}
this.update();
};
Poll.prototype.generateVotes = function () {
let output = '<div class="infobox"><p style="margin: 2px 0 5px 0"><span style="border:1px solid #6A6;color:#484;border-radius:4px;padding:0 3px"><i class="fa fa-bar-chart"></i> Poll</span> <strong style="font-size:11pt">' + Tools.escapeHTML(this.question) + '</strong></p>';
this.options.forEach(function (option, number) {
output += '<div style="margin-top: 3px"><button value="/poll vote ' + number + '" name="send" title="Vote for ' + number + '. ' + Tools.escapeHTML(option.name) + '">' + number + '. <strong>' + Tools.escapeHTML(option.name) + '</strong></button></div>';
});
output += '</div>';
return output;
};
Poll.prototype.generateResults = function (ended, option) {
let icon = '<span style="border:1px solid #' + (ended ? '777;color:#555' : '6A6;color:#484') + ';border-radius:4px;padding:0 3px"><i class="fa fa-bar-chart"></i> ' + (ended ? "Poll ended" : "Poll") + '</span>';
let output = '<div class="infobox"><p style="margin: 2px 0 5px 0">' + icon + ' <strong style="font-size:11pt">' + Tools.escapeHTML(this.question) + '</strong></p>';
let iter = this.options.entries();
let i = iter.next();
let c = 0;
let colors = ['#79A', '#8A8', '#88B'];
while (!i.done) {
let percentage = Math.round((i.value[1].votes * 100) / (this.totalVotes || 1));
output += '<div style="margin-top: 3px">' + i.value[0] + '. <strong>' + (i.value[0] === option ? '<em>' : '') + Tools.escapeHTML(i.value[1].name) + (i.value[0] === option ? '</em>' : '') + '</strong> <small>(' + i.value[1].votes + ' vote' + (i.value[1].votes === 1 ? '' : 's') + ')</small><br /><span style="font-size:7pt;background:' + colors[c % 3] + ';padding-right:' + (percentage * 3) + 'px"></span><small>&nbsp;' + percentage + '%</small></div>';
i = iter.next();
c++;
}
output += '</div>';
return output;
};
Poll.prototype.update = function () {
let results = [];
for (let i = 0; i <= this.options.size; i++) {
results.push(this.generateResults(false, i));
}
// Update the poll results for everyone that has voted
for (let i in this.room.users) {
let user = this.room.users[i];
if (user.userid in this.voters) {
user.sendTo(this.room, '|uhtmlchange|poll' + this.room.pollNumber + '|' + results[this.voters[user.userid]]);
} else if (user.latestIp in this.voterIps) {
user.sendTo(this.room, '|uhtmlchange|poll' + this.room.pollNumber + '|' + results[this.voterIps[user.latestIp]]);
}
}
};
Poll.prototype.display = function (user, broadcast) {
let votes = this.generateVotes();
let results = [];
for (let i = 0; i <= this.options.size; i++) {
results.push(this.generateResults(false, i));
}
let target = {};
if (broadcast) {
target = this.room.users;
} else {
target[0] = user;
}
for (let i in target) {
let thisUser = target[i];
if (thisUser.userid in this.voters) {
thisUser.sendTo(this.room, '|uhtml|poll' + this.room.pollNumber + '|' + results[this.voters[thisUser.userid]]);
} else if (thisUser.latestIp in this.voterIps) {
thisUser.sendTo(this.room, '|uhtml|poll' + this.room.pollNumber + '|' + results[this.voterIps[thisUser.latestIp]]);
} else {
thisUser.sendTo(this.room, '|uhtml|poll' + this.room.pollNumber + '|' + votes);
}
}
};
Poll.prototype.end = function () {
let results = this.generateResults(true);
this.room.send('|uhtmlchange|poll' + this.room.pollNumber + '|<div class="infobox">(The poll has ended &ndash; scroll down to see the results)</div>');
this.room.send('|html|' + results);
};
return Poll;
})();
exports.commands = {
poll: {
create: 'new',
new: function (target, room, user, connection, cmd, message) {
if (target.length > 1024) return this.errorReply("Poll too long.");
let params = target.split(target.includes('|') ? '|' : ',').map(function (param) { return param.trim(); });
if (!this.can(permission, null, room)) return false;
if (room.poll) return this.errorReply("There is already a poll in progress in this room.");
if (params.length < 3) return this.errorReply("Not enough arguments for /poll new.");
let options = [];
for (let i = 1; i < params.length; i++) {
options.push(params[i]);
}
if (options.length > 8) {
return this.errorReply("Too many options for poll (maximum is 8).");
}
room.poll = new Poll(room, params[0], options);
room.poll.display(user, true);
this.logEntry("" + user.name + " used " + message);
return this.privateModCommand("(A poll was started by " + user.name + ".)");
},
newhelp: ["/poll create [question], [option1], [option2], [...] - Creates a poll. Requires: % @ # & ~"],
vote: function (target, room, user) {
if (!room.poll) return this.errorReply("There is no poll running in this room.");
if (!target) return this.errorReply("Please specify an option.");
if (target === 'blank') {
room.poll.blankvote(user);
return;
}
let parsed = parseInt(target, 10);
if (isNaN(parsed)) return this.errorReply("To vote, specify the number of the option.");
if (!room.poll.options.has(parsed)) return this.sendReply("Option not in poll.");
room.poll.vote(user, parsed);
},
votehelp: ["/poll vote [number] - Votes for option [number]."],
timer: function (target, room, user) {
if (!room.poll) return this.errorReply("There is no poll running in this room.");
if (target) {
if (!this.can(permission, null, room)) return false;
if (target === 'clear') {
if (room.poll.timeout) {
clearTimeout(room.poll.timeout);
room.poll.timeout = null;
room.poll.timeoutMins = 0;
return room.add("The timeout for the poll was cleared.");
} else {
return this.errorReply("No timer to clear.");
}
}
let timeout = parseFloat(target);
if (isNaN(timeout) || timeout <= 0) return this.errorReply("Invalid time given.");
if (room.poll.timeout) clearTimeout(room.poll.timeout);
room.poll.timeoutMins = timeout;
room.poll.timeout = setTimeout((function () {
room.poll.end();
delete room.poll;
}), (timeout * 60000));
room.add("The timeout for the poll was set to " + timeout + " minutes.");
return this.privateModCommand("(The poll timeout was set to " + timeout + " minutes by " + user.name + ".)");
} else {
if (!this.canBroadcast()) return;
if (room.poll.timeout) {
return this.sendReply("The timeout for the poll is " + room.poll.timeoutMins + " minutes.");
} else {
return this.sendReply("There's no timer for this poll.");
}
}
},
timerhelp: ["/poll timer [minutes] - Sets the poll to automatically end after [minutes] minutes. Requires: % @ # & ~", "/poll timer clear - Clears the poll's timer. Requires: % @ # & ~"],
results: function (target, room, user) {
if (!room.poll) return this.errorReply("There is no poll running in this room.");
return room.poll.blankvote(user);
},
resultshelp: ["/poll results - Shows the results of the poll without voting. NOTE: you can't go back and vote after using this."],
close: 'end',
stop: 'end',
end: function (target, room, user) {
if (!this.can(permission, null, room)) return false;
if (!room.poll) return this.errorReply("There is no poll running in this room.");
if (room.poll.timeout) clearTimeout(room.poll.timeout);
room.poll.end();
delete room.poll;
return this.privateModCommand("(The poll was ended by " + user.name + ".)");
},
endhelp: ["/poll end - Ends a poll and displays the results. Requires: % @ # & ~"],
show: 'display',
display: function (target, room, user) {
if (!room.poll) return this.errorReply("There is no poll running in this room.");
if (!this.canBroadcast()) return;
room.update();
room.poll.display(user, this.broadcasting);
},
displayhelp: ["/poll display - Displays the poll"],
'': function (target, room, user) {
this.parse('/help poll');
}
},
pollhelp: ["/poll allows rooms to run their own polls. These polls are limited to one poll at a time per room.",
"Accepts the following commands:",
"/poll create [question], [option1], [option2], [...] - Creates a poll. Requires: % @ # & ~",
"/poll vote [number] - Votes for option [number].",
"/poll timer [minutes] - Sets the poll to automatically end after [minutes]. Requires: % @ # & ~",
"/poll results - Shows the results of the poll without voting. NOTE: you can't go back and vote after using this.",
"/poll display - Displays the poll",
"/poll end - Ends a poll and displays the results. Requires: % @ # & ~"]
};