Add command to add servers from invites with eligibility check

backend/src/plugins/BotControl/commands/AddServerFromInviteCmd.ts

@@ -0,0 +1,68 @@
+import { ApiPermissions } from "@shared/apiPermissions";
+import { TextChannel } from "discord.js";
+import { commandTypeHelpers as ct } from "../../../commandTypes";
+import { isOwnerPreFilter, sendErrorMessage, sendSuccessMessage } from "../../../pluginUtils";
+import { DBDateFormat, isGuildInvite, isSnowflake, resolveInvite } from "../../../utils";
+import { botControlCmd } from "../types";
+import moment from "moment-timezone";
+import { isEligible } from "../functions/isEligible";
+
+export const AddServerFromInviteCmd = botControlCmd({
+  trigger: ["add_server_from_invite", "allow_server_from_invite"],
+  permission: "can_add_server_from_invite",
+
+  signature: {
+    user: ct.resolvedUser(),
+    inviteCode: ct.string(),
+  },
+
+  async run({ pluginData, message: msg, args }) {
+    const invite = await resolveInvite(pluginData.client, args.inviteCode, true);
+    if (!invite || !isGuildInvite(invite)) {
+      sendErrorMessage(pluginData, msg.channel as TextChannel, "Could not resolve invite"); // :D
+      return;
+    }
+
+    const existing = await pluginData.state.allowedGuilds.find(invite.guild.id);
+    if (existing) {
+      sendErrorMessage(pluginData, msg.channel as TextChannel, "Server is already allowed!");
+      return;
+    }
+
+    const { result, explanation } = await isEligible(pluginData, args.user, invite);
+    if (!result) {
+      sendErrorMessage(
+        pluginData,
+        msg.channel as TextChannel,
+        `Could not add server because it's not eligible: ${explanation}`,
+      );
+      return;
+    }
+
+    await pluginData.state.allowedGuilds.add(invite.guild.id, { name: invite.guild.name });
+    await pluginData.state.configs.saveNewRevision(`guild-${invite.guild.id}`, "plugins: {}", msg.author.id);
+
+    await pluginData.state.apiPermissionAssignments.addUser(invite.guild.id, args.user.id, [
+      ApiPermissions.ManageAccess,
+    ]);
+
+    if (args.user.id !== msg.author.id) {
+      // Add temporary access to user who added server
+      await pluginData.state.apiPermissionAssignments.addUser(
+        invite.guild.id,
+        msg.author.id,
+        [ApiPermissions.ManageAccess],
+        moment
+          .utc()
+          .add(1, "hour")
+          .format(DBDateFormat),
+      );
+    }
+
+    sendSuccessMessage(
+      pluginData,
+      msg.channel as TextChannel,
+      "Server was eligible and is now allowed to use Zeppelin!",
+    );
+  },
+});