/** * Name: Users * Description: Track known users */ var _ = require('underscore')._, uuid = require('node-uuid'); var users = function(dbot) { /*** Internal API ***/ this.internalAPI = { 'createUser': function(server, nick, channel, callback) { var id = uuid.v4(); this.db.create('users', id, { 'id': id, 'primaryNick': nick, 'currentNick': nick, 'server': server, 'channels': [ channel ], 'aliases': [] }, function(err, result) { if(!err) { dbot.api.event.emit('new_user', [ result ]); callback(result); } }); }.bind(this), 'createChannel': function(server, name, callback) { var id = uuid.v4(); this.db.create('channel_users', id, { 'id': id, 'server': server, 'name': name, 'users': [] }, function(err, result) { if(!err) { dbot.api.event.emit('new_channel', [ result ]); callback(result); } }); }.bind(this), 'addChannelUser': function(user, channelName, callback) { this.api.getChannel(user.server, channelName, function(channel) { channel.users.push(user.id); this.db.save('channel_users', channel.id, channel, function(err) { if(!err) { dbot.api.event.emit('new_channel_user', [ user, channel ]); callback(); } }); }.bind(this)); }.bind(this), 'updateChannelPrimaryUser': function(server, oldUser, newUser) { this.db.search('channel_users', { 'server': server }, function(channel) { channel.users = _.without(channel.users, oldUser); if(!_.include(channel.users, newUser)) channel.users.push(newUser); this.db.save('channel_users', channel.id, channel, function(err) { if(err) { // QQ } }); }.bind(this), function(err) { if(err) { // QQ } }); }.bind(this) }; this.listener = function(event) { if(event.action == 'JOIN' && event.user != dbot.config.name) { this.api.resolveUser(event.server, event.user, function(user) { if(!user) { // User does not yet exist this.internalAPI.createUser(event.server, event.user, event.channel.name, function(result) { this.internalAPI.addChannelUser(result, event.channel.name, function(err) { }); }.bind(this)); } else { if(!_.include(user.channels, event.channel.name)) { // User not yet channel user users.channels.push(event.channel.name); this.internalAPI.addChannelUser(user, event.channel.name, function(err) { }); } user.currentNick = event.user; this.db.save('users', user.id, user, function(err) { if(err) { // QQ } }); } }.bind(this)); } else if(event.action == 'NICK') { this.api.resolveUser(event.server, event.user, function(user) { this.api.isKnownUser(event.server, event.newNick, function(isKnown) { user.currentNick = event.newNick; if(!isKnown) { user.aliases.push(event.newNick); } this.db.save('users', user.id, user, function(err) { if(!err) { dbot.api.event.emit('new_user_alias', [ user, event.newNick ]); } }); }.bind(this)); }.bind(this)); } }.bind(this); this.on = ['JOIN', 'NICK']; this.onLoad = function() { dbot.instance.addListener('366', 'users', function(event) { this.api.getChannel(event.server, event.channel.name, function(channel) { var checkChannelUsers = function(channel) { _.each(event.channel.nicks, function(nick) { var nick = nick.name; this.api.resolveUser(event.server, nick, function(user) { if(!user) { this.internalAPI.createUser(event.server, nick, event.channel.name, function(result) { channel.users.push(result.id); }.bind(this)); } else { if(!_.include(user.channels, event.channel)) { channel.users.push(user.id); } } }.bind(this)); }, this); process.nextTick(function() { this.db.save('channel_users', channel.id, channel, function(err) { }); }.bind(this)); }.bind(this); if(!channel) { // Channel does not yet exist this.internalAPI.createChannel(event.server, event.channel.name, checkChannelUsers); } else { checkChannelUsers(channel); } }.bind(this)); }.bind(this)); var connections = dbot.instance.connections; _.each(connections, function(connection) { connection.updateNickLists(); }); }; }; exports.fetch = function(dbot) { return new users(dbot); };