3
0
mirror of https://github.com/reality/dbot.git synced 2024-11-24 04:49:25 +01:00

tranform users, fix for bind in module loader [#131]

This commit is contained in:
reality 2013-01-14 21:03:47 +00:00
parent 8542ceb9e9
commit 8b7953e906
4 changed files with 188 additions and 173 deletions

42
modules/users/api.js Normal file
View File

@ -0,0 +1,42 @@
var _ = require('underscore')._;
var api = function(dbot) {
var api = {
'resolveUser': function(server, nick, useLowercase) {
var knownUsers = this.getServerUsers(server);
var user = nick;
if(!_.include(knownUsers.users, nick) && _.has(knownUsers.aliases, nick)) {
user = knownUsers.aliases[nick];
}
if(useLowercase) user = user.toLowerCase();
return user;
},
'isKnownUser': function(server, nick) {
var knownUsers = this.getServerUsers(server);
return (_.include(knownUsers.users, nick) || _.has(knownUsers.aliases, nick));
},
'isPrimaryUser': function(server, nick) {
var knownUsers = this.getServerUsers(server);
return _.include(knownUsers.users, nick);
},
'getAliases': function(server, nick) {
var knownUsers = this.getServerUsers(server);
return _.chain(knownUsers.aliases)
.keys()
.filter(function(user) {
return knownUsers.aliases[user] == nick;
}, this)
.value();
}
};
return api;
};
exports.fetch = function(dbot) {
return api(dbot);
};

91
modules/users/commands.js Normal file
View File

@ -0,0 +1,91 @@
var _ = require('underscore')._;
var commands = function(dbot) {
var commands = {
'~alias': function(event) {
var knownUsers = this.getServerUsers(event.server),
alias = event.params[1].trim();
if(_.include(knownUsers.users, alias)) {
var aliasCount = _.reduce(knownUsers.aliases, function(memo, user) {
if(user == alias) return memo += 1;
}, 0, this);
event.reply(dbot.t('primary', {
'user': alias,
'count': aliasCount
}));
} else if(_.has(knownUsers.aliases, alias)) {
event.reply(dbot.t('alias', {
'alias': alias,
'user': knownUsers.aliases[alias]
}));
} else {
event.reply(dbot.t('unknown_alias', { 'alias': alias }));
}
},
'~setaliasparent': function(event) {
var knownUsers = this.getServerUsers(event.server);
var newParent = event.params[1];
if(_.has(knownUsers.aliases, newParent)) {
var newAlias = knownUsers.aliases[newParent];
// Replace user entry
knownUsers.users = _.without(knownUsers.users, newAlias);
knownUsers.users.push(newParent);
// Replace channels entries with new primary user
this.updateChannels(event, newAlias, newParent);
// Remove alias for new parent & add alias for new alias
delete knownUsers.aliases[newParent];
knownUsers.aliases[newAlias] = newParent;
// Update aliases to point to new primary user
this.updateAliases(event, newAlias, newParent);
event.reply(dbot.t('aliasparentset', {
'newParent': newParent,
'newAlias': newAlias
}));
dbot.api.stats.fixStats(event.server, newAlias);
} else {
event.reply(dbot.t('unknown_alias', { 'alias': newParent }));
}
},
'~mergeusers': function(event) {
var knownUsers = this.getServerUsers(event.server);
var primaryUser = event.params[1];
var secondaryUser = event.params[2];
if(_.include(knownUsers.users, primaryUser) && _.include(knownUsers.users, secondaryUser)) {
knownUsers.users = _.without(knownUsers.users, secondaryUser);
knownUsers.aliases[secondaryUser] = primaryUser;
this.updateAliases(event, secondaryUser, primaryUser);
this.updateChannels(event, secondaryUser, primaryUser);
event.reply(dbot.t('merged_users', {
'old_user': secondaryUser,
'new_user': primaryUser
}));
dbot.api.stats.fixStats(event.server, secondaryUser);
} else {
event.reply(dbot.t('unprimary_error'));
}
}
};
commands['~setaliasparent'].access = 'moderator';
commands['~mergeusers'].access = 'moderator';
return commands;
};
exports.fetch = function(dbot) {
return commands(dbot);
};

View File

@ -2,12 +2,12 @@
* Name: Users * Name: Users
* Description: Track known users * Description: Track known users
*/ */
var web = require('./web'), var _ = require('underscore')._;
_ = require('underscore')._;
var users = function(dbot) { var users = function(dbot) {
var knownUsers = dbot.db.knownUsers; this.knownUsers = dbot.db.knownUsers;
var getServerUsers = function(server) { this.getServerUsers = function(server) {
var knownUsers = this.knownUsers;
if(!_.has(knownUsers, server)) { if(!_.has(knownUsers, server)) {
knownUsers[server] = { 'users': [], 'aliases': {}, 'channelUsers': {} }; knownUsers[server] = { 'users': [], 'aliases': {}, 'channelUsers': {} };
} }
@ -17,8 +17,8 @@ var users = function(dbot) {
return knownUsers[server]; return knownUsers[server];
}; };
var updateAliases = function(event, oldUser, newUser) { this.updateAliases = function(event, oldUser, newUser) {
var knownUsers = getServerUsers(event.server); var knownUsers = this.getServerUsers(event.server);
_.each(knownUsers.aliases, function(user, alias) { _.each(knownUsers.aliases, function(user, alias) {
if(user == oldUser) { if(user == oldUser) {
knownUsers.aliases[alias] = newUser; knownUsers.aliases[alias] = newUser;
@ -26,197 +26,78 @@ var users = function(dbot) {
}, this); }, this);
}; };
var updateChannels = function(event, oldUser, newUser) { this.updateChannels = function(event, oldUser, newUser) {
var channelUsers = getServerUsers(event.server).channelUsers; var channelUsers = this.getServerUsers(event.server).channelUsers;
channelUsers = _.each(channelUsers, function(channel, channelName) { channelUsers = _.each(channelUsers, function(channel, channelName) {
channelUsers[channelName] = _.without(channel, oldUser); channelUsers[channelName] = _.without(channel, oldUser);
channelUsers[channelName].push(newUser); channelUsers[channelName].push(newUser);
}, this); }, this);
}; };
dbot.instance.addListener('366', 'users', function(event) {
var knownUsers = getServerUsers(event.server); this.name = 'users';
if(!_.has(knownUsers.channelUsers, event.channel.name)) { this.ignorable = false;
knownUsers.channelUsers[event.channel.name] = [];
} this.listener = function(event) {
var channelUsers = knownUsers.channelUsers[event.channel.name]; var knownUsers = this.getServerUsers(event.server);
var nick = event.user;
_.each(event.channel.nicks, function(nick) { if(event.action == 'JOIN') {
nick = nick.name; if(!_.has(knownUsers.channelUsers, event.channel.name)) {
if(api.isKnownUser(event.server, nick)) { knownUsers.channelUsers[event.channel.name] = [];
nick = api.resolveUser(event.server, nick); }
var channelUsers = knownUsers.channelUsers[event.channel.name];
if(this.api.isKnownUser(event.server, nick)) {
nick = this.api.resolveUser(event.server, nick);
} else { } else {
knownUsers.users.push(nick); knownUsers.users.push(nick);
} }
if(!_.include(channelUsers, nick)) { if(!_.include(channelUsers, nick)) {
channelUsers.push(nick); channelUsers.push(nick);
} }
}, this); } else if(event.action == 'NICK') {
}); var newNick = event.params.substr(1);
if(_.has(knownUsers.aliases, event.user)) {
knownUsers.aliases[newNick] = knownUsers.aliases[event.user];
var api = {
'resolveUser': function(server, nick, useLowercase) {
var knownUsers = getServerUsers(server);
var user = nick;
if(!_.include(knownUsers.users, nick) && _.has(knownUsers.aliases, nick)) {
user = knownUsers.aliases[nick];
}
if(useLowercase) user = user.toLowerCase();
return user;
},
'isKnownUser': function(server, nick) {
var knownUsers = getServerUsers(server);
return (_.include(knownUsers.users, nick) || _.has(knownUsers.aliases, nick));
},
'isPrimaryUser': function(server, nick) {
return _.include(knownUsers.users, nick);
},
'getAliases': function(server, nick) {
var knownUsers = getServerUsers(server);
return _.chain(knownUsers.aliases)
.keys()
.filter(function(user) {
return knownUsers.aliases[user] == nick;
}, this)
.value();
}
};
var commands = {
'~alias': function(event) {
var knownUsers = getServerUsers(event.server),
alias = event.params[1].trim();
if(_.include(knownUsers.users, alias)) {
var aliasCount = _.reduce(knownUsers.aliases, function(memo, user) {
if(user == alias) return memo += 1;
}, 0, this);
event.reply(dbot.t('primary', {
'user': alias,
'count': aliasCount
}));
} else if(_.has(knownUsers.aliases, alias)) {
event.reply(dbot.t('alias', {
'alias': alias,
'user': knownUsers.aliases[alias]
}));
} else { } else {
event.reply(dbot.t('unknown_alias', { 'alias': alias })); if(!_.include(knownUsers.users, newNick)) {
} knownUsers.aliases[newNick] = event.user;
},
'~setaliasparent': function(event) {
var knownUsers = getServerUsers(event.server);
var newParent = event.params[1];
if(_.has(knownUsers.aliases, newParent)) {
var newAlias = knownUsers.aliases[newParent];
// Replace user entry
knownUsers.users = _.without(knownUsers.users, newAlias);
knownUsers.users.push(newParent);
// Replace channels entries with new primary user
updateChannels(event, newAlias, newParent);
// Remove alias for new parent & add alias for new alias
delete knownUsers.aliases[newParent];
knownUsers.aliases[newAlias] = newParent;
// Update aliases to point to new primary user
updateAliases(event, newAlias, newParent);
event.reply(dbot.t('aliasparentset', {
'newParent': newParent,
'newAlias': newAlias
}));
dbot.api.stats.fixStats(event.server, newAlias);
} else {
event.reply(dbot.t('unknown_alias', { 'alias': newParent }));
}
},
'~mergeusers': function(event) {
var knownUsers = getServerUsers(event.server);
var primaryUser = event.params[1];
var secondaryUser = event.params[2];
if(_.include(knownUsers.users, primaryUser) && _.include(knownUsers.users, secondaryUser)) {
knownUsers.users = _.without(knownUsers.users, secondaryUser);
knownUsers.aliases[secondaryUser] = primaryUser;
updateAliases(event, secondaryUser, primaryUser);
updateChannels(event, secondaryUser, primaryUser);
event.reply(dbot.t('merged_users', {
'old_user': secondaryUser,
'new_user': primaryUser
}));
dbot.api.stats.fixStats(event.server, secondaryUser);
} else {
event.reply(dbot.t('unprimary_error'));
}
}
};
commands['~setaliasparent'].access = 'moderator';
commands['~mergeusers'].access = 'moderator';
return {
'name': 'users',
'ignorable': false,
'commands': commands,
'api': api,
'pages': web.getPages(dbot),
'listener': function(event) {
var knownUsers = getServerUsers(event.server);
var nick = event.user;
if(event.action == 'JOIN') {
if(!_.has(knownUsers.channelUsers, event.channel.name)) {
knownUsers.channelUsers[event.channel.name] = [];
} }
var channelUsers = knownUsers.channelUsers[event.channel.name]; }
}
}.bind(this);
this.on = ['JOIN', 'NICK'];
this.onLoad = function() {
// Trigger updateNickLists to stat current users in channel
dbot.instance.addListener('366', 'users', function(event) {
var knownUsers = this.getServerUsers(event.server);
if(!_.has(knownUsers.channelUsers, event.channel.name)) {
knownUsers.channelUsers[event.channel.name] = [];
}
var channelUsers = knownUsers.channelUsers[event.channel.name];
if(api.isKnownUser(event.server, nick)) { _.each(event.channel.nicks, function(nick) {
nick = api.resolveUser(event.server, nick); nick = nick.name;
if(this.api.isKnownUser(event.server, nick)) {
nick = this.api.resolveUser(event.server, nick);
} else { } else {
knownUsers.users.push(nick); knownUsers.users.push(nick);
} }
if(!_.include(channelUsers, nick)) { if(!_.include(channelUsers, nick)) {
channelUsers.push(nick); channelUsers.push(nick);
} }
} else if(event.action == 'NICK') { }, this);
var newNick = event.params.substr(1); }.bind(this));
if(_.has(knownUsers.aliases, event.user)) {
knownUsers.aliases[newNick] = knownUsers.aliases[event.user]; var connections = dbot.instance.connections;
} else { _.each(connections, function(connection) {
if(!_.include(knownUsers.users, newNick)) { connection.updateNickLists();
knownUsers.aliases[newNick] = event.user; });
}
}
}
},
'on': ['JOIN', 'NICK'],
'onLoad': function() {
// Trigger updateNickLists to stat current users in channel
var connections = dbot.instance.connections;
_.each(connections, function(connection) {
connection.updateNickLists();
});
}
}; };
}; };
exports.fetch = function(dbot) { exports.fetch = function(dbot) {
return users(dbot); return new users(dbot);
}; };

1
run.js
View File

@ -194,6 +194,7 @@ DBot.prototype.reloadModules = function() {
if(_.has(config, property) && _.has(config[property], itemName)) { if(_.has(config, property) && _.has(config[property], itemName)) {
_.extend(item, config[property][itemName]); _.extend(item, config[property][itemName]);
} }
module[property][itemName] = item.bind(module);
}, this); }, this);
_.extend(this[property], module[property]); _.extend(this[property], module[property]);
}, this); }, this);