From 2f32e40aa83f15569c826d6abf77702056cc4882 Mon Sep 17 00:00:00 2001
From: Yuzu <yuzu@kirameki.cafe>
Date: Sat, 18 Sep 2021 08:37:16 -0700
Subject: [PATCH] Create Settings.ts

---
 src/discord/Settings.ts | 132 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 src/discord/Settings.ts

diff --git a/src/discord/Settings.ts b/src/discord/Settings.ts
new file mode 100644
index 0000000..589e027
--- /dev/null
+++ b/src/discord/Settings.ts
@@ -0,0 +1,132 @@
+import { Message, Interaction, CommandInteraction } from "discord.js";
+import { makeInfoEmbed, makeErrorEmbed, makeSuccessEmbed, makeProcessingEmbed, sendMessageOrInteractionResponse, sendReply } from "../utils/DiscordMessage";
+import DiscordProvider from "../providers/Discord";
+import {registerAllGuildsCommands, unregisterAllGuildsCommands} from "../utils/DiscordInteraction";
+import Prisma from "../providers/Prisma";
+import Users from "../services/Users";
+import Cache from "../providers/Cache";
+
+const EMBEDS = {
+    SETTINGS_INFO: (data: Message | Interaction) => {
+        return makeInfoEmbed({
+            title: 'Settings',
+            description: `Change how ${DiscordProvider.client.user?.username} behave on ${data.guild?.name}`,
+            fields: [
+                {
+                    name: 'Available arguments',
+                    value: '``setPrefix``'
+                }
+            ],
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+    },
+    PROCESSING: (data: Message | Interaction) => {
+        return makeProcessingEmbed({
+            icon: (data instanceof Message) ? undefined : '⌛',
+            title: `Performing actions`,
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+    },
+    NO_PERMISSION: (data: Message | Interaction, permissions: string) => {
+        return makeErrorEmbed({
+            title: 'Developer only',
+            description: 'You need ``' + permissions + '`` permission on this guild!',
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+    },
+    NO_PREFIX_PROVIDED: (data: Message | Interaction) => {
+        return makeErrorEmbed ({
+            title: 'No prefix provided',
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+    },
+    PREFIX_UPDATED: (data: Message | Interaction, newPrefix: string) => {
+        return makeSuccessEmbed ({
+            title: 'Prefix updated',
+            description: `From now onwards, I shall be called by using \`\`${newPrefix}\`\``,
+            user: (data instanceof Interaction) ? data.user : data.author
+        });
+    },
+}
+
+export default class Settings {
+    async onCommand(command: string, args: any, message: Message) {
+        if(command.toLowerCase() !== 'settings') return;
+        await this.process(message, args);
+    }
+
+    async interactionCreate(interaction: CommandInteraction) { 
+        if(interaction.isCommand()) {
+            if(typeof interaction.commandName === 'undefined') return;
+            if((interaction.commandName).toLowerCase() !== 'settings') return;
+            await this.process(interaction, interaction.options);
+        }
+    }
+
+    async process(data: Interaction | Message, args: any) {
+        
+        const isSlashCommand = data instanceof CommandInteraction && data.isCommand();
+        const isMessage = data instanceof Message;
+
+        if(!isSlashCommand && !isMessage) return;
+
+        if(data.guild === null || data.guildId === null) return;
+        
+        const Guild = await Prisma.client.guild.findFirst({ where: { id: data.guildId }})
+        if(!Guild) throw new Error('TODO: Handle if guild is not found');
+
+        const funct = {
+            setPrefix: async(data: Message | Interaction) => {
+
+                let newPrefix: string | undefined;
+                if(data instanceof Message) {
+                    let _name: string;
+                    if(typeof args[1] === 'undefined')
+                        return await sendMessageOrInteractionResponse(data, { embeds: [EMBEDS.NO_PREFIX_PROVIDED(data)] });
+
+                    let __name = args;
+                    __name.shift();
+                    _name = __name.join(" ");
+                    newPrefix = _name;
+                }
+                else if(data instanceof CommandInteraction)
+                    newPrefix = data.options.getString('prefix')!;
+
+                if(!newPrefix)
+                    return await sendMessageOrInteractionResponse(data, { embeds: [EMBEDS.NO_PREFIX_PROVIDED(data)] });
+
+                let placeholder = await sendMessageOrInteractionResponse(data, { embeds: [EMBEDS.PROCESSING(data)] });
+                await Prisma.client.guild.update({ where: { id: data.guildId! }, data: { prefix: newPrefix }});
+                Cache.updateGuildCache(data.guildId!);
+
+                if(isMessage)
+                    return (placeholder as Message).edit({ embeds: [EMBEDS.PREFIX_UPDATED(data, newPrefix)] });
+                else if(isSlashCommand)
+                    return await sendMessageOrInteractionResponse(data, { embeds: [EMBEDS.PREFIX_UPDATED(data, newPrefix)]}, true);
+            }
+        }
+
+        let query;
+
+        if(isMessage) {
+            if(data === null || !data.guildId || data.member === null || data.guild === null) return;
+            
+            if(args.length === 0) {
+                return await sendMessageOrInteractionResponse(data, { embeds: [EMBEDS.SETTINGS_INFO(data)] });
+            }
+            query = args[0].toLowerCase();
+
+        }
+        else if(isSlashCommand) {
+            query = args.getSubcommand();
+        }
+
+        switch(query) {
+            case "info":
+                return await sendMessageOrInteractionResponse(data, { embeds: [EMBEDS.SETTINGS_INFO(data)] });
+            case "setprefix":
+                return await funct.setPrefix(data);
+        }
+
+    }
+}
\ No newline at end of file