StyAndCo/zeppelin
3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-05-10 20:35:02 +00:00
Code Releases Activity

Utility: add --counts flag to !roles

Browse source
This commit is contained in:
Dragory 2019-02-17 15:42:52 +02:00
parent 625120e07e
commit 2e13495d77
1 changed files with 53 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

59
src/plugins/Utility.ts
View file

@ -1,5 +1,5 @@
import { decorators as d } from "knub";
import { Channel, EmbedOptions, Member, Message, TextChannel, User, VoiceChannel } from "eris";
import { Channel, EmbedOptions, Member, Message, Role, TextChannel, User, VoiceChannel } from "eris";
import {
chunkArray,
embedPadding,
@ -87,32 +87,73 @@ export class UtilityPlugin extends ZeppelinPlugin {
}
}
@d.command("roles", "[search:string$]")
@d.command("roles", "[search:string$]", {
options: [
{
name: "counts",
type: "bool",
},
],
})
@d.permission("roles")
async rolesCmd(msg: Message, args: { search?: string }) {
let roles = (msg.channel as TextChannel).guild.roles.map(role => `${role.name} ${role.id}`);
async rolesCmd(msg: Message, args: { search?: string; counts?: boolean }) {
let roles: Array<{ _memberCount?: number } & Role> = Array.from((msg.channel as TextChannel).guild.roles.values());
if (args.search) {
const searchStr = args.search.toLowerCase();
roles = roles.filter(r => r.toLowerCase().includes(searchStr));
roles = roles.filter(r => r.name.toLowerCase().includes(searchStr) || r.id === searchStr);
}
if (args.counts) {
// If the user requested role member counts as well, calculate them and sort the roles by their member count
const roleCounts: Map<string, number> = Array.from(this.guild.members.values()).reduce((map, member) => {
for (const roleId of member.roles) {
if (!map.has(roleId)) map.set(roleId, 0);
map.set(roleId, map.get(roleId) + 1);
}
return map;
}, new Map());
// The "everyone" role always has all members in it
roleCounts.set(this.guildId, this.guild.memberCount);
for (const role of roles) {
role._memberCount = roleCounts.has(role.id) ? roleCounts.get(role.id) : 0;
}
roles.sort((a, b) => {
if (a.toLowerCase() > b.toLowerCase()) return 1;
if (a.toLowerCase() < b.toLowerCase()) return -1;
if (a._memberCount > b._memberCount) return -1;
if (a._memberCount < b._memberCount) return 1;
return 0;
});
} else {
// Otherwise sort by name
roles.sort((a, b) => {
if (a.name.toLowerCase() > b.name.toLowerCase()) return 1;
if (a.name.toLowerCase() < b.name.toLowerCase()) return -1;
return 0;
});
}
const chunks = chunkArray(roles, 20);
for (const [i, chunk] of chunks.entries()) {
const roleLines = chunk.map(role => {
let line = `${role.id} ${role.name}`;
if (role._memberCount != null) {
line += role._memberCount === 1 ? ` (${role._memberCount} member)` : ` (${role._memberCount} members)`;
}
return line;
});
if (i === 0) {
msg.channel.createMessage(
trimLines(`
${args.search ? "Total roles found" : "Total roles"}: ${roles.length}
\`\`\`py\n${chunk.join("\n")}\`\`\`
\`\`\`py\n${roleLines.join("\n")}\`\`\`
`),
);
} else {
msg.channel.createMessage("```py\n" + chunk.join("\n") + "```");
msg.channel.createMessage("```py\n" + roleLines.join("\n") + "```");
}
}
}