From 2e875e7434ea3c2135bb45107fc7e938dd3ccbc2 Mon Sep 17 00:00:00 2001
From: reality <tinmachin3@gmail.com>
Date: Tue, 15 Jan 2013 16:21:43 +0000
Subject: [PATCH] update poll [#131]

---
 modules/poll/commands.js | 187 ++++++++++++++++++++++++++++++++
 modules/poll/pages.js    |  45 ++++++++
 modules/poll/poll.js     | 226 +--------------------------------------
 3 files changed, 235 insertions(+), 223 deletions(-)
 create mode 100644 modules/poll/commands.js
 create mode 100644 modules/poll/pages.js

diff --git a/modules/poll/commands.js b/modules/poll/commands.js
new file mode 100644
index 0000000..0758f66
--- /dev/null
+++ b/modules/poll/commands.js
@@ -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);
+}
diff --git a/modules/poll/pages.js b/modules/poll/pages.js
new file mode 100644
index 0000000..0c8ea41
--- /dev/null
+++ b/modules/poll/pages.js
@@ -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);
+};
diff --git a/modules/poll/poll.js b/modules/poll/poll.js
index 0ab8cda..f7f2d4b 100644
--- a/modules/poll/poll.js
+++ b/modules/poll/poll.js
@@ -1,228 +1,8 @@
-var _ = require('underscore')._;
-
 var poll = 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, 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
-    };
+    this.name = poll;
+    this.ignorable = true;
 };
 
 exports.fetch = function(dbot) {
-    return poll(dbot);
+    return new poll(dbot);
 }