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:
parent
8542ceb9e9
commit
8b7953e906
42
modules/users/api.js
Normal file
42
modules/users/api.js
Normal 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
91
modules/users/commands.js
Normal 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);
|
||||||
|
};
|
@ -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
1
run.js
@ -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);
|
||||||
|
Loading…
Reference in New Issue
Block a user