From fb32a04ad38be8d0836dc99259b6ef05a0825830 Mon Sep 17 00:00:00 2001
From: PalmDevs <git@palmdevs.me>
Date: Mon, 24 Jun 2024 01:06:40 +0700
Subject: [PATCH] feat(bots/discord/utils): add functions for role presets

---
 bots/discord/src/utils/discord/rolePresets.ts | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 bots/discord/src/utils/discord/rolePresets.ts

diff --git a/bots/discord/src/utils/discord/rolePresets.ts b/bots/discord/src/utils/discord/rolePresets.ts
new file mode 100644
index 0000000..a0623d4
--- /dev/null
+++ b/bots/discord/src/utils/discord/rolePresets.ts
@@ -0,0 +1,58 @@
+import { config, database } from '$/context'
+import { appliedPresets } from '$/database/schemas'
+import type { GuildMember } from 'discord.js'
+import { and, eq } from 'drizzle-orm'
+
+// TODO: Fix this type
+type PresetKey = string
+
+export const applyRolePreset = async (member: GuildMember, presetName: PresetKey, untilMs: number | null) => {
+    const afterInsert = await commonOperations(presetName, member, true)
+    const until = untilMs ? Math.ceil(untilMs / 1000) : null
+
+    await database
+        .insert(appliedPresets)
+        .values({
+            memberId: member.id,
+            guildId: member.guild.id,
+            preset: presetName,
+            until,
+        })
+        .onConflictDoUpdate({
+            target: [appliedPresets.memberId, appliedPresets.preset, appliedPresets.guildId],
+            set: { until },
+        })
+        .then(afterInsert)
+}
+
+export const removeRolePreset = async (member: GuildMember, presetName: PresetKey) => {
+    const afterDelete = await commonOperations(presetName, member, false)
+
+    await database
+        .delete(appliedPresets)
+        .where(
+            and(
+                eq(appliedPresets.memberId, member.id),
+                eq(appliedPresets.preset, presetName),
+                eq(appliedPresets.guildId, member.guild.id),
+            ),
+        )
+        .execute()
+        .then(afterDelete)
+}
+
+/**
+ * Inserts (if not already present) an entry in the database, sets the member's roles
+ * @returns The currently applied presets AND a callback function to run after correcting the presets in the database
+ */
+const commonOperations = async (presetName: string, member: GuildMember, applying: boolean) => {
+    const preset = config.rolePresets?.guilds[presetName]?.[member.guild.id]
+    if (!preset) throw new Error(`The preset "${presetName}" does not exist for this server`)
+
+    const roles = new Set(member.roles.cache.keys())
+
+    for (const role of preset.give) roles[applying ? 'add' : 'delete'](role)
+    for (const role of preset.take) roles[applying ? 'delete' : 'add'](role)
+
+    return () => member.roles.set(Array.from(roles))
+}