forked from GitHub/dbot
update poll [#131]
This commit is contained in:
reality
parent
529194e4df
commit
2e875e7434
187
modules/poll/commands.js
Normal file
187
modules/poll/commands.js
Normal file
@ -0,0 +1,187 @@
|
|||||||
|
var _ = require('underscore')._;
|
||||||
|
|
||||||
|
var commands = function(dbot) {
|
||||||
|
var polls = dbot.db.polls;
|
||||||
|
var commands = {
|
||||||
|
'~newpoll': function(event) {
|
||||||
|
var name = event.input[1],
|
||||||
|
options = event.input[2].split(','),
|
||||||
|
description = event.input[3];
|
||||||
|
|
||||||
|
if(_.has(polls, name)) {
|
||||||
|
event.reply(dbot.t('poll_exists', { 'name': name }));
|
||||||
|
} else {
|
||||||
|
polls[name] = {
|
||||||
|
'name': name,
|
||||||
|
'description': description,
|
||||||
|
'owner': event.user,
|
||||||
|
'votes': {},
|
||||||
|
'votees': {}
|
||||||
|
};
|
||||||
|
for(var i=0;i<options.length;i++) {
|
||||||
|
polls[name]['votes'][options[i]] = 0;
|
||||||
|
}
|
||||||
|
event.reply(dbot.t('poll_created', {
|
||||||
|
'name': name,
|
||||||
|
'description': description,
|
||||||
|
'url': dbot.t('url', {
|
||||||
|
'host': dbot.config.web.webHost,
|
||||||
|
'port': dbot.config.web.webPort,
|
||||||
|
'path': 'polls/' + name
|
||||||
|
})
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
'~addoption': function(event) {
|
||||||
|
var name = event.input[1],
|
||||||
|
option = event.input[2];
|
||||||
|
|
||||||
|
if(_.has(polls, name)) {
|
||||||
|
if(polls[name].owner === event.user) {
|
||||||
|
if(!_.has(polls[name].votes, option)) {
|
||||||
|
polls[name]['votes'][option] = 0;
|
||||||
|
event.reply(dbot.t('option_added', {
|
||||||
|
'user': event.user,
|
||||||
|
'name': name,
|
||||||
|
'option': option
|
||||||
|
}));
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('option_exists', {
|
||||||
|
'option': option,
|
||||||
|
'name': name,
|
||||||
|
'user': event.user
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('not_poll_owner', {
|
||||||
|
'user': event.user,
|
||||||
|
'name': name
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('poll_unexistent', {'name': name}));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
'~rmoption': function(event) {
|
||||||
|
var name = event.input[1],
|
||||||
|
option = event.input[2];
|
||||||
|
|
||||||
|
if(_.has(polls, name)) {
|
||||||
|
if(polls[name].owner === event.user) {
|
||||||
|
if(_.has(polls[name].votes, option)) {
|
||||||
|
delete polls[name]['votes'][option];
|
||||||
|
event.reply(dbot.t('option_removed', {
|
||||||
|
'user': event.user,
|
||||||
|
'name': name,
|
||||||
|
'option': option
|
||||||
|
}));
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('invalid_vote', { 'vote': option }));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('not_poll_owner', { 'name': name }));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('poll_unexistent', { 'name': name }));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
'~vote': function(event) {
|
||||||
|
var name = event.input[1],
|
||||||
|
vote = event.input[2];
|
||||||
|
|
||||||
|
if(_.has(polls, name)) {
|
||||||
|
if(_.has(polls[name].votes, vote)) {
|
||||||
|
if(_.has(polls[name].votees, event.user)) {
|
||||||
|
var oldVote = polls[name].votees[event.user];
|
||||||
|
polls[name].votes[oldVote]--;
|
||||||
|
polls[name].votes[vote]++;
|
||||||
|
polls[name].votees[event.user] = vote;
|
||||||
|
|
||||||
|
event.reply(dbot.t('changed_vote', {
|
||||||
|
'vote': vote,
|
||||||
|
'poll': name,
|
||||||
|
'count': polls[name].votes[vote],
|
||||||
|
'user': event.user
|
||||||
|
}));
|
||||||
|
} else {
|
||||||
|
polls[name].votes[vote]++;
|
||||||
|
polls[name].votees[event.user] = vote;
|
||||||
|
event.reply(dbot.t('voted', {
|
||||||
|
'vote': vote,
|
||||||
|
'poll': name,
|
||||||
|
'count': polls[name].votes[vote],
|
||||||
|
'user': event.user
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('invalid_vote', { 'vote': vote }));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('poll_unexistent', { 'name': name }));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
'~pdesc': function(event) {
|
||||||
|
var name = event.input[1];
|
||||||
|
|
||||||
|
if(_.has(polls, name)) {
|
||||||
|
event.reply(dbot.t('poll_describe', {
|
||||||
|
'name': name,
|
||||||
|
'description': polls[name].description,
|
||||||
|
'url': dbot.t('url', {
|
||||||
|
'host': dbot.config.web.webHost,
|
||||||
|
'port': dbot.config.web.webPort,
|
||||||
|
'path': 'polls/' + name
|
||||||
|
})
|
||||||
|
}));
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('poll_unexistent', { 'name': name }));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
'~count': function(event) {
|
||||||
|
var name = event.input[1];
|
||||||
|
|
||||||
|
if(_.has(polls, name)) {
|
||||||
|
var order;
|
||||||
|
var votesArr = [];
|
||||||
|
|
||||||
|
var order = _.chain(polls[name].votes)
|
||||||
|
.pairs()
|
||||||
|
.sortBy(function(option) { return option[1] })
|
||||||
|
.reverse()
|
||||||
|
.value();
|
||||||
|
|
||||||
|
var orderString = "";
|
||||||
|
for(var i=0;i<order.length;i++) {
|
||||||
|
orderString += order[i][0] +
|
||||||
|
" (" + order[i][1] + "), ";
|
||||||
|
}
|
||||||
|
orderString = orderString.slice(0, -2);
|
||||||
|
|
||||||
|
event.reply(dbot.t('count', {
|
||||||
|
'poll': name,
|
||||||
|
'description': polls[name].description,
|
||||||
|
'places': orderString
|
||||||
|
}));
|
||||||
|
} else {
|
||||||
|
event.reply(dbot.t('poll_unexistent', {'name': name}));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
commands['~newpoll'].regex = [/~newpoll ([^ ]+) options=([^ ]+) (.+)/, 4];
|
||||||
|
commands['~addoption'].regex = [/~addoption ([^ ]+) ([^ ]+)/, 3];
|
||||||
|
commands['~rmoption'].regex = [/~rmoption ([^ ]+) ([^ ]+)/, 3];
|
||||||
|
commands['~vote'].regex = [/~vote ([^ ]+) ([^ ]+)/, 3];
|
||||||
|
commands['~pdesc'].regex = [/~pdesc ([^ ]+)/, 2];
|
||||||
|
commands['~count'].regex = [/~count ([^ ]+)/, 2];
|
||||||
|
|
||||||
|
return commands;
|
||||||
|
};
|
||||||
|
|
||||||
|
exports.fetch = function(dbot) {
|
||||||
|
return commands(dbot);
|
||||||
|
}
|
||||||
45
modules/poll/pages.js
Normal file
45
modules/poll/pages.js
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
var _ = require('underscore')._;
|
||||||
|
|
||||||
|
var pages = function(dbot) {
|
||||||
|
var polls = dbot.db.polls;
|
||||||
|
var pages = {
|
||||||
|
// Shows the results of a poll
|
||||||
|
'/polls/:key': function(req, res) {
|
||||||
|
var key = req.params.key.toLowerCase();
|
||||||
|
if(_.has(dbot.db.polls, key)) {
|
||||||
|
var totalVotes = _.reduce(dbot.db.polls[key].votes,
|
||||||
|
function(memo, option) {
|
||||||
|
return memo += option;
|
||||||
|
}, 0);
|
||||||
|
res.render('polls', {
|
||||||
|
'name': dbot.config.name,
|
||||||
|
'description': dbot.db.polls[key].description,
|
||||||
|
'votees': Object.keys(dbot.db.polls[key].votees),
|
||||||
|
'options': dbot.db.polls[key].votes,
|
||||||
|
locals: {
|
||||||
|
'totalVotes': totalVotes,
|
||||||
|
'url_regex': RegExp.prototype.url_regex()
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
res.render('error', {
|
||||||
|
'name': dbot.config.name,
|
||||||
|
'message': 'No polls under that key.'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
// Lists all of the polls
|
||||||
|
'/polls': function(req, res) {
|
||||||
|
res.render('polllist', {
|
||||||
|
'name': dbot.config.name,
|
||||||
|
'polllist': Object.keys(dbot.db.polls)
|
||||||
|
});
|
||||||
|
},
|
||||||
|
};
|
||||||
|
return pages;
|
||||||
|
};
|
||||||
|
|
||||||
|
exports.fetch = function(dbot) {
|
||||||
|
return pages(dbot);
|
||||||
|
};
|
||||||
@ -1,228 +1,8 @@
|
|||||||
var _ = require('underscore')._;
|
|
||||||
|
|
||||||
var poll = function(dbot) {
|
var poll = function(dbot) {
|
||||||
var polls = dbot.db.polls;
|
this.name = poll;
|
||||||
var commands = {
|
this.ignorable = true;
|
||||||
'~newpoll': function(event) {
|
|
||||||
var name = event.input[1],
|
|
||||||
options = event.input[2].split(','),
|
|
||||||
description = event.input[3];
|
|
||||||
|
|
||||||
if(_.has(polls, name)) {
|
|
||||||
event.reply(dbot.t('poll_exists', { 'name': name }));
|
|
||||||
} else {
|
|
||||||
polls[name] = {
|
|
||||||
'name': name,
|
|
||||||
'description': description,
|
|
||||||
'owner': event.user,
|
|
||||||
'votes': {},
|
|
||||||
'votees': {}
|
|
||||||
};
|
|
||||||
for(var i=0;i<options.length;i++) {
|
|
||||||
polls[name]['votes'][options[i]] = 0;
|
|
||||||
}
|
|
||||||
event.reply(dbot.t('poll_created', {
|
|
||||||
'name': name,
|
|
||||||
'description': description,
|
|
||||||
'url': dbot.t('url', {
|
|
||||||
'host': dbot.config.web.webHost,
|
|
||||||
'port': dbot.config.web.webPort,
|
|
||||||
'path': 'polls/' + name
|
|
||||||
})
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
'~addoption': function(event) {
|
|
||||||
var name = event.input[1],
|
|
||||||
option = event.input[2];
|
|
||||||
|
|
||||||
if(_.has(polls, name)) {
|
|
||||||
if(polls[name].owner === event.user) {
|
|
||||||
if(!_.has(polls[name].votes, name)) {
|
|
||||||
polls[name]['votes'][option] = 0;
|
|
||||||
event.reply(dbot.t('option_added', {
|
|
||||||
'user': event.user,
|
|
||||||
'name': name,
|
|
||||||
'option': option
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('option_exists', {
|
|
||||||
'option': option,
|
|
||||||
'name': name,
|
|
||||||
'user': event.user
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('not_poll_owner', {
|
|
||||||
'user': event.user,
|
|
||||||
'name': name
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('poll_unexistent', {'name': name}));
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
'~rmoption': function(event) {
|
|
||||||
var name = event.input[1],
|
|
||||||
option = event.input[2];
|
|
||||||
|
|
||||||
if(_.has(polls, name)) {
|
|
||||||
if(polls[name].owner === event.user) {
|
|
||||||
if(_.has(polls[name].votes, option)) {
|
|
||||||
delete polls[name]['votes'][option];
|
|
||||||
event.reply(dbot.t('option_removed', {
|
|
||||||
'user': event.user,
|
|
||||||
'name': name,
|
|
||||||
'option': option
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('invalid_vote', { 'vote': option }));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('not_poll_owner', { 'name': name }));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('poll_unexistent', { 'name': name }));
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
'~vote': function(event) {
|
|
||||||
var name = event.input[1],
|
|
||||||
vote = event.input[2];
|
|
||||||
|
|
||||||
if(_.has(polls, name)) {
|
|
||||||
if(_.has(polls[name].votes, vote)) {
|
|
||||||
if(_.has(polls[name].votees, event.user)) {
|
|
||||||
var oldVote = polls[name].votees[event.user];
|
|
||||||
polls[name].votes[oldVote]--;
|
|
||||||
polls[name].votes[vote]++;
|
|
||||||
polls[name].votees[event.user] = vote;
|
|
||||||
|
|
||||||
event.reply(dbot.t('changed_vote', {
|
|
||||||
'vote': vote,
|
|
||||||
'poll': name,
|
|
||||||
'count': polls[name].votes[vote],
|
|
||||||
'user': event.user
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
polls[name].votes[vote]++;
|
|
||||||
polls[name].votees[event.user] = vote;
|
|
||||||
event.reply(dbot.t('voted', {
|
|
||||||
'vote': vote,
|
|
||||||
'poll': name,
|
|
||||||
'count': polls[name].votes[vote],
|
|
||||||
'user': event.user
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('invalid_vote', { 'vote': vote }));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('poll_unexistent', { 'name': name }));
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
'~pdesc': function(event) {
|
|
||||||
var name = event.input[1];
|
|
||||||
|
|
||||||
if(_.has(polls, name)) {
|
|
||||||
event.reply(dbot.t('poll_describe', {
|
|
||||||
'name': name,
|
|
||||||
'description': polls[name].description,
|
|
||||||
'url': dbot.t('url', {
|
|
||||||
'host': dbot.config.web.webHost,
|
|
||||||
'port': dbot.config.web.webPort,
|
|
||||||
'path': 'polls/' + name
|
|
||||||
})
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('poll_unexistent', { 'name': name }));
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
'~count': function(event) {
|
|
||||||
var name = event.input[1];
|
|
||||||
|
|
||||||
if(_.has(polls, name)) {
|
|
||||||
var order;
|
|
||||||
var votesArr = [];
|
|
||||||
|
|
||||||
var order = _.chain(polls[name].votes)
|
|
||||||
.pairs()
|
|
||||||
.sortBy(function(option) { return option[1] })
|
|
||||||
.reverse()
|
|
||||||
.value();
|
|
||||||
|
|
||||||
var orderString = "";
|
|
||||||
for(var i=0;i<order.length;i++) {
|
|
||||||
orderString += order[i][0] +
|
|
||||||
" (" + order[i][1] + "), ";
|
|
||||||
}
|
|
||||||
orderString = orderString.slice(0, -2);
|
|
||||||
|
|
||||||
event.reply(dbot.t('count', {
|
|
||||||
'poll': name,
|
|
||||||
'description': polls[name].description,
|
|
||||||
'places': orderString
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
event.reply(dbot.t('poll_unexistent', {'name': name}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
commands['~newpoll'].regex = [/~newpoll ([^ ]+) options=([^ ]+) (.+)/, 4];
|
|
||||||
commands['~addoption'].regex = [/~addoption ([^ ]+) ([^ ]+)/, 3];
|
|
||||||
commands['~rmoption'].regex = [/~rmoption ([^ ]+) ([^ ]+)/, 3];
|
|
||||||
commands['~vote'].regex = [/~vote ([^ ]+) ([^ ]+)/, 3];
|
|
||||||
commands['~pdesc'].regex = [/~pdesc ([^ ]+)/, 2];
|
|
||||||
commands['~count'].regex = [/~count ([^ ]+)/, 2];
|
|
||||||
|
|
||||||
var pages = {
|
|
||||||
// Shows the results of a poll
|
|
||||||
'/polls/:key': function(req, res) {
|
|
||||||
var key = req.params.key.toLowerCase();
|
|
||||||
if(_.has(dbot.db.polls, key)) {
|
|
||||||
var totalVotes = _.reduce(dbot.db.polls[key].votes,
|
|
||||||
function(memo, option) {
|
|
||||||
return memo += option;
|
|
||||||
}, 0);
|
|
||||||
res.render('polls', {
|
|
||||||
'name': dbot.config.name,
|
|
||||||
'description': dbot.db.polls[key].description,
|
|
||||||
'votees': Object.keys(dbot.db.polls[key].votees),
|
|
||||||
'options': dbot.db.polls[key].votes,
|
|
||||||
locals: {
|
|
||||||
'totalVotes': totalVotes,
|
|
||||||
'url_regex': RegExp.prototype.url_regex()
|
|
||||||
}
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
res.render('error', {
|
|
||||||
'name': dbot.config.name,
|
|
||||||
'message': 'No polls under that key.'
|
|
||||||
});
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
// Lists all of the polls
|
|
||||||
'/polls': function(req, res) {
|
|
||||||
res.render('polllist', {
|
|
||||||
'name': dbot.config.name,
|
|
||||||
'polllist': Object.keys(dbot.db.polls)
|
|
||||||
});
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
return {
|
|
||||||
'name': 'poll',
|
|
||||||
'ignorable': true,
|
|
||||||
'commands': commands,
|
|
||||||
'pages': pages
|
|
||||||
};
|
|
||||||
};
|
};
|
||||||
|
|
||||||
exports.fetch = function(dbot) {
|
exports.fetch = function(dbot) {
|
||||||
return poll(dbot);
|
return new poll(dbot);
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user