Migrate SelfGrantableRoles to new Plugin structure

backend/src/plugins/SelfGrantableRoles/commands/RoleAddCmd.ts

@@ -0,0 +1,124 @@
+import { selfGrantableRolesCmd } from "../types";
+import { commandTypeHelpers as ct } from "../../../commandTypes";
+import { getApplyingEntries } from "../util/getApplyingEntries";
+import { sendErrorMessage, sendSuccessMessage } from "src/pluginUtils";
+import { splitRoleNames } from "../util/splitRoleNames";
+import { normalizeRoleNames } from "../util/normalizeRoleNames";
+import { findMatchingRoles } from "../util/findMatchingRoles";
+import { Role } from "eris";
+
+export const RoleAddCmd = selfGrantableRolesCmd({
+  trigger: ["role", "role add"],
+  permission: null,
+
+  signature: {
+    roleNames: ct.string({ rest: true }),
+  },
+
+  async run({ message: msg, args, pluginData }) {
+    const lock = await pluginData.locks.acquire(`grantableRoles:${msg.author.id}`);
+
+    const applyingEntries = getApplyingEntries(pluginData, msg);
+    if (applyingEntries.length === 0) {
+      lock.unlock();
+      return;
+    }
+
+    const roleNames = normalizeRoleNames(splitRoleNames(args.roleNames));
+    const matchedRoleIds = findMatchingRoles(roleNames, applyingEntries);
+
+    const hasUnknownRoles = matchedRoleIds.length !== roleNames.length;
+
+    const rolesToAdd: Map<string, Role> = Array.from(matchedRoleIds.values())
+      .map(id => pluginData.guild.roles.get(id))
+      .filter(Boolean)
+      .reduce((map, role) => {
+        map.set(role.id, role);
+        return map;
+      }, new Map());
+
+    if (!rolesToAdd.size) {
+      sendErrorMessage(
+        pluginData,
+        msg.channel,
+        `<@!${msg.author.id}> Unknown ${args.roleNames.length === 1 ? "role" : "roles"}`,
+      );
+      lock.unlock();
+      return;
+    }
+
+    // Grant the roles
+    const newRoleIds = new Set([...rolesToAdd.keys(), ...msg.member.roles]);
+
+    // Remove extra roles (max_roles) for each entry
+    const skipped: Set<Role> = new Set();
+    const removed: Set<Role> = new Set();
+
+    for (const entry of applyingEntries) {
+      if (entry.max_roles === 0) continue;
+
+      let foundRoles = 0;
+
+      for (const roleId of newRoleIds) {
+        if (entry.roles[roleId]) {
+          if (foundRoles < entry.max_roles) {
+            foundRoles++;
+          } else {
+            newRoleIds.delete(roleId);
+            rolesToAdd.delete(roleId);
+
+            if (msg.member.roles.includes(roleId)) {
+              removed.add(pluginData.guild.roles.get(roleId));
+            } else {
+              skipped.add(pluginData.guild.roles.get(roleId));
+            }
+          }
+        }
+      }
+    }
+
+    try {
+      await msg.member.edit({
+        roles: Array.from(newRoleIds),
+      });
+    } catch (e) {
+      sendErrorMessage(
+        pluginData,
+        msg.channel,
+        `<@!${msg.author.id}> Got an error while trying to grant you the roles`,
+      );
+      return;
+    }
+
+    const mentionRoles = pluginData.config.get().mention_roles;
+    const addedRolesStr = Array.from(rolesToAdd.values()).map(r => (mentionRoles ? `<@&${r.id}>` : `**${r.name}**`));
+    const addedRolesWord = rolesToAdd.size === 1 ? "role" : "roles";
+
+    const messageParts = [];
+    messageParts.push(`Granted you the ${addedRolesStr.join(", ")} ${addedRolesWord}`);
+
+    if (skipped.size || removed.size) {
+      const skippedRolesStr = skipped.size
+        ? "skipped " +
+          Array.from(skipped.values())
+            .map(r => (mentionRoles ? `<@&${r.id}>` : `**${r.name}**`))
+            .join(",")
+        : null;
+      const removedRolesStr = removed.size
+        ? "removed " + Array.from(removed.values()).map(r => (mentionRoles ? `<@&${r.id}>` : `**${r.name}**`))
+        : null;
+
+      const skippedRemovedStr = [skippedRolesStr, removedRolesStr].filter(Boolean).join(" and ");
+
+      messageParts.push(`${skippedRemovedStr} due to role limits`);
+    }
+
+    if (hasUnknownRoles) {
+      messageParts.push("couldn't recognize some of the roles");
+    }
+
+    sendSuccessMessage(pluginData, msg.channel, `<@!${msg.author.id}> ${messageParts.join("; ")}`);
+
+    lock.unlock();
+  },
+});

backend/src/plugins/SelfGrantableRoles/commands/RoleHelpCmd.ts

@@ -0,0 +1,52 @@
+import { selfGrantableRolesCmd } from "../types";
+import { asSingleLine, trimLines } from "src/utils";
+import { getApplyingEntries } from "../util/getApplyingEntries";
+
+export const RoleHelpCmd = selfGrantableRolesCmd({
+  trigger: ["role help", "role"],
+  permission: null,
+
+  async run({ message: msg, pluginData }) {
+    const applyingEntries = getApplyingEntries(pluginData, msg);
+    if (applyingEntries.length === 0) return;
+
+    const allPrimaryAliases = [];
+    for (const entry of applyingEntries) {
+      for (const aliases of Object.values(entry.roles)) {
+        if (aliases[0]) {
+          allPrimaryAliases.push(aliases[0]);
+        }
+      }
+    }
+
+    const prefix = pluginData.guildConfig.prefix;
+    const [firstRole, secondRole] = allPrimaryAliases;
+
+    const help1 = asSingleLine(`
+      To give yourself a role, type e.g. \`${prefix}role ${firstRole}\` where **${firstRole}** is the role you want.
+      ${secondRole ? `You can also add multiple roles at once, e.g. \`${prefix}role ${firstRole} ${secondRole}\`` : ""}
+    `);
+
+    const help2 = asSingleLine(`
+      To remove a role, type \`${prefix}role remove ${firstRole}\`,
+      again replacing **${firstRole}** with the role you want to remove.
+    `);
+
+    const helpMessage = trimLines(`
+      ${help1}
+
+      ${help2}
+
+      **Roles available to you:**
+      ${allPrimaryAliases.join(", ")}
+    `);
+
+    const helpEmbed = {
+      title: "How to get roles",
+      description: helpMessage,
+      color: parseInt("42bff4", 16),
+    };
+
+    msg.channel.createMessage({ embed: helpEmbed });
+  },
+});

backend/src/plugins/SelfGrantableRoles/commands/RoleRemoveCmd.ts

@@ -0,0 +1,75 @@
+import { selfGrantableRolesCmd } from "../types";
+import { commandTypeHelpers as ct } from "../../../commandTypes";
+import { getApplyingEntries } from "../util/getApplyingEntries";
+import { sendErrorMessage, sendSuccessMessage } from "src/pluginUtils";
+import { splitRoleNames } from "../util/splitRoleNames";
+import { normalizeRoleNames } from "../util/normalizeRoleNames";
+import { findMatchingRoles } from "../util/findMatchingRoles";
+
+export const RoleRemoveCmd = selfGrantableRolesCmd({
+  trigger: "role remove",
+  permission: null,
+
+  signature: {
+    roleNames: ct.string({ rest: true }),
+  },
+
+  async run({ message: msg, args, pluginData }) {
+    const lock = await pluginData.locks.acquire(`grantableRoles:${msg.author.id}`);
+
+    const applyingEntries = getApplyingEntries(pluginData, msg);
+    if (applyingEntries.length === 0) {
+      lock.unlock();
+      return;
+    }
+
+    const roleNames = normalizeRoleNames(splitRoleNames(args.roleNames));
+    const matchedRoleIds = findMatchingRoles(roleNames, applyingEntries);
+
+    const rolesToRemove = Array.from(matchedRoleIds.values()).map(id => pluginData.guild.roles.get(id));
+    const roleIdsToRemove = rolesToRemove.map(r => r.id);
+
+    // Remove the roles
+    if (rolesToRemove.length) {
+      const newRoleIds = msg.member.roles.filter(roleId => !roleIdsToRemove.includes(roleId));
+
+      try {
+        await msg.member.edit({
+          roles: newRoleIds,
+        });
+
+        const removedRolesStr = rolesToRemove.map(r => `**${r.name}**`);
+        const removedRolesWord = rolesToRemove.length === 1 ? "role" : "roles";
+
+        if (rolesToRemove.length !== roleNames.length) {
+          sendSuccessMessage(
+            pluginData,
+            msg.channel,
+            `<@!${msg.author.id}> Removed ${removedRolesStr.join(", ")} ${removedRolesWord};` +
+              ` couldn't recognize the other roles you mentioned`,
+          );
+        } else {
+          sendSuccessMessage(
+            pluginData,
+            msg.channel,
+            `<@!${msg.author.id}> Removed ${removedRolesStr.join(", ")} ${removedRolesWord}`,
+          );
+        }
+      } catch (e) {
+        sendSuccessMessage(
+          pluginData,
+          msg.channel,
+          `<@!${msg.author.id}> Got an error while trying to remove the roles`,
+        );
+      }
+    } else {
+      sendErrorMessage(
+        pluginData,
+        msg.channel,
+        `<@!${msg.author.id}> Unknown ${args.roleNames.length === 1 ? "role" : "roles"}`,
+      );
+    }
+
+    lock.unlock();
+  },
+});