From aa6a5fce42294fa1a8846fe2b7748d3c803b99fd Mon Sep 17 00:00:00 2001
From: reality <tinmachin3@gmail.com>
Date: Sun, 24 Mar 2013 12:26:33 +0000
Subject: [PATCH] databankise users listener, add getChannel to users api
 [#272]

---
 modules/users/api.js   | 14 ++++++++
 modules/users/users.js | 75 ++++++++++++++++++++++++------------------
 2 files changed, 57 insertions(+), 32 deletions(-)

diff --git a/modules/users/api.js b/modules/users/api.js
index 7e8434f..7d6c30d 100644
--- a/modules/users/api.js
+++ b/modules/users/api.js
@@ -25,6 +25,20 @@ var api = function(dbot) {
             });
         },
 
+        'getChannel': function(server, channel, callback) {
+            var channel = false;
+            this.db.search('channel_users', {
+                'server': server,
+                'name': channel
+            }, function(result) {
+                channel = result;
+            }, function(err) {
+                if(!err) {
+                    callback(channel);
+                }
+            });
+        },
+
         'getRandomChannelUser': function(server, channel, callback) {
             var channel;
             this.db.search('channel_users', { 
diff --git a/modules/users/users.js b/modules/users/users.js
index 5e713de..6e38ad6 100644
--- a/modules/users/users.js
+++ b/modules/users/users.js
@@ -6,16 +6,6 @@ var _ = require('underscore')._;
 
 var users = function(dbot) {
     this.knownUsers = dbot.db.knownUsers;
-    this.getServerUsers = function(server) {
-        var knownUsers = this.knownUsers;
-        if(!_.has(knownUsers, server)) {
-            knownUsers[server] = { 'users': [], 'aliases': {}, 'channelUsers': {} };
-        }
-        if(!_.has(knownUsers[server], 'channelUsers')) {
-            knownUsers[server].channelUsers = {};
-        }
-        return knownUsers[server];    
-    };
 
     this.updateAliases = function(event, oldUser, newUser) {
         var knownUsers = this.getServerUsers(event.server);
@@ -35,32 +25,53 @@ var users = function(dbot) {
     };
         
     this.listener = function(event) {
-        /*var knownUsers = this.getServerUsers(event.server); 
-        var nick = event.user;
-
         if(event.action == 'JOIN' && nick != dbot.config.name) {
-            if(!_.has(knownUsers.channelUsers, event.channel.name)) {
-                knownUsers.channelUsers[event.channel.name] = [];
-            }
-            var channelUsers = knownUsers.channelUsers[event.channel.name];
-
-            if(this.api.isKnownUser(event.server, nick)) {
-                nick = this.api.resolveUser(event.server, nick);
-            } else {
-                knownUsers.users.push(nick);
-                dbot.api.event.emit('new_user', [ event.server, nick ]);
-            }
-
-            if(!_.include(channelUsers, nick)) {
-                channelUsers.push(nick);
+            this.api.resolveUser(event.server, event.user, function(user) {
+                if(!user) { // User does not yet exist 
+                    var id = uuid.v4();
+                    this.db.create('users', id, {
+                        'uuid': id,
+                        'primaryNick': event.user,
+                        'currentNick': event.user,
+                        'server': event.server,
+                        'channels': [ event.channel ],
+                        'aliases': []
+                    }, function(err, result) {
+                        if(!err) {
+                            user = result;
+                            dbot.api.event.emit('new_user', [ user ]);
+                        }
+                    });
+                } 
+                
+                if(!_.include(user.channels, event.channel)) { // User not yet channel user
+                    user.channels.push(event.channel);
+                    this.db.save('users', user.id, user, function(err) {
+                        if(!err) {
+                            this.api.getChannel(event.server, event.channel, function(channel) {
+                                channel.users.push(user.primaryNick);
+                                this.db.save('channel_users', channel.id, channel, function(err)) {
+                                    if(!err) {
+                                        dbot.api.event.emit('new_channel_user', [ user ]);
+                                    }
+                                });
+                            }
+                        }
+                    });
+                }
             }
         } else if(event.action == 'NICK') {
-            var newNick = event.newNick;
-            if(!this.api.isKnownUser(newNick)) {
-                knownUsers.aliases[newNick] = this.api.resolveUser(event.server, event.user);
-                dbot.api.event.emit('nick_change', [ event.server, newNick ]);
+            if(!this.api.isKnownUser(event.newNick)) {
+                this.api.resolveUser(event.server, event.user, function(user) {
+                    user.aliases.push(event.newNick);
+                    this.db.save('users', user.id, function(err) {
+                        if(!err) {
+                            dbot.api.event.emit('new_user_alias', [ user, event.newNick ]);
+                        }
+                    });
+                });
             }
-        }*/
+        }
     }.bind(this);
     this.on =  ['JOIN', 'NICK'];