From 7655bccbdfa2ad41aabcb63d6bc324b5b6a7e380 Mon Sep 17 00:00:00 2001
From: srqc <98553181+2srqc@users.noreply.github.com>
Date: Thu, 9 Feb 2023 19:35:51 -0500
Subject: [PATCH] Create ArchiveCmd.ts

---
 .../plugins/Utility/commands/ArchiveCmd.ts    | 72 +++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 backend/src/plugins/Utility/commands/ArchiveCmd.ts

diff --git a/backend/src/plugins/Utility/commands/ArchiveCmd.ts b/backend/src/plugins/Utility/commands/ArchiveCmd.ts
new file mode 100644
index 00000000..2d84c16a
--- /dev/null
+++ b/backend/src/plugins/Utility/commands/ArchiveCmd.ts
@@ -0,0 +1,72 @@
+import { GuildPluginData } from "knub";
+import { commandTypeHelpers as ct } from "../../../commandTypes";
+import { SavedMessage } from "../../../data/entities/SavedMessage";
+import { getBaseUrl, sendErrorMessage, sendSuccessMessage } from "../../../pluginUtils";
+import { utilityCmd, UtilityPluginType } from "../types";
+
+const DEFAULT_COUNT = 50;
+const MAX_COUNT = 1000;
+
+export async function archiveMessages(
+  pluginData: GuildPluginData<UtilityPluginType>,
+  messagesToArchive: SavedMessage[],
+) {
+  messagesToArchive = Array.from(messagesToArchive).sort((a, b) => (a.posted_at > b.posted_at ? 1 : -1));
+
+  const archiveId = await pluginData.state.archives.createFromSavedMessages(messagesToArchive, pluginData.guild);
+  const baseUrl = getBaseUrl(pluginData);
+
+  return pluginData.state.archives.getUrl(baseUrl, archiveId);
+}
+
+const opts = {
+  count: ct.number({ option: true, shortcut: "c" }),
+};
+
+export interface ArchiveArgs {
+  userId: string;
+  count?: number;
+}
+
+export async function archiveCmd(pluginData: GuildPluginData<UtilityPluginType>, args: ArchiveArgs | any, msg) {
+  args.count = args.count ?? DEFAULT_COUNT;
+
+  if (args.count > MAX_COUNT || args.count <= 0) {
+    sendErrorMessage(pluginData, msg.channel, `Archive count must be between 1 and ${MAX_COUNT}`);
+    return;
+  }
+
+  const archivingMessage = msg.channel.send("Archiving...");
+  const messagesToArchive = await pluginData.state.savedMessages.getUserMessages(args.userId, args.count);
+
+  if (messagesToArchive.length > 0) {
+    const archiveResult = await archiveMessages(pluginData, messagesToArchive);
+    let responseText = `Archived ${messagesToArchive.length} message${messagesToArchive.length === 1 ? "" : "s"}`;
+
+    responseText += `\n${archiveResult}`;
+    await sendSuccessMessage(pluginData, msg.channel, responseText);
+  } else {
+    const responseText = `Found no messages to archive!`;
+    await sendErrorMessage(pluginData, msg.channel, responseText);
+  }
+
+  await (await archivingMessage).delete();
+}
+
+export const ArchiveCmd = utilityCmd({
+  trigger: ["archive"],
+  description: "Archive a number of messages for a user",
+  usage: "!archive 106391128718245888",
+  permission: "can_archive",
+
+  signature: [
+    {
+      userId: ct.userId(),
+      ...opts,
+    },
+  ],
+
+  async run({ message: msg, args, pluginData }) {
+    archiveCmd(pluginData, args, msg);
+  },
+});