feat(network): embed colors

Discord-InterChat · Sep 15, 2023 · df6d4a8 · df6d4a8
1 parent 83d219d
commit df6d4a8
Show file tree

Hide file tree

Showing 3 changed files with 71 additions and 1 deletion.
diff --git a/prisma/schema.prisma b/prisma/schema.prisma
@@ -78,6 +78,7 @@ model connectedList {
   compact    Boolean
   invite     String?
   profFilter Boolean
+  embedColor String?
   webhookURL String
   date       DateTime @default(now())
   hub        hubs?    @relation(fields: [hubId], references: [id])

diff --git a/src/Events/messageCreate.ts b/src/Events/messageCreate.ts
@@ -1,7 +1,7 @@
 import checks from '../Scripts/message/checks';
 import messageContentModifiers from '../Scripts/message/messageContentModifiers';
 import cleanup from '../Scripts/message/cleanup';
-import { APIMessage, ActionRowBuilder, ButtonBuilder, ButtonStyle, EmbedBuilder, Message, User, WebhookClient, WebhookMessageCreateOptions } from 'discord.js';
+import { APIMessage, ActionRowBuilder, ButtonBuilder, ButtonStyle, EmbedBuilder, HexColorString, Message, User, WebhookClient, WebhookMessageCreateOptions } from 'discord.js';
 import { getDb, colors } from '../Utils/functions/utils';
 import { censor } from '../Utils/functions/wordFilter';
 import { messageData } from '@prisma/client';
@@ -135,6 +135,15 @@ export default {
             threadId: connection.parentId ? connection.channelId : undefined,
             allowedMentions: { parse: [] },
           };
+
+          if (connection.embedColor && webhookMessage.embeds
+          && /^#[0-9A-F]{6}$/i.test(connection.embedColor)
+          ) {
+            webhookMessage.embeds.push(EmbedBuilder.from(webhookMessage.embeds[0])
+              .setColor(connection.embedColor as HexColorString),
+            );
+            webhookMessage.embeds.splice(0, 1);
+          }
         }
 
         const webhook = new WebhookClient({ url: connection.webhookURL });

diff --git a/src/Scripts/network/displaySettings.ts b/src/Scripts/network/displaySettings.ts
@@ -72,6 +72,11 @@ export = {
             .setEmoji(emoji.icons.store)
             .setDescription('Set a different channel for the network.')
             .setValue('change_channel'),
+          new StringSelectMenuOptionBuilder()
+            .setLabel('Embed Color')
+            .setEmoji('🎨')
+            .setDescription('Set the color of the embeds sent in the network.')
+            .setValue('embed_color'),
         ),
     ]);
 
@@ -164,6 +169,61 @@ export = {
       }
 
       switch (settingsMenu.values[0]) {
+        /* Embed color selection */
+        case 'embed_color': {
+          const modal = new ModalBuilder()
+            .setTitle('Set Embed Color')
+            .setCustomId(settingsMenu.id)
+            .addComponents(
+              new ActionRowBuilder<TextInputBuilder>().addComponents(
+                new TextInputBuilder()
+                  .setCustomId('embed_color')
+                  .setStyle(TextInputStyle.Short)
+                  .setLabel('Embed Color')
+                  .setPlaceholder('Provide a hex color code.')
+                  .setValue(updConnection.embedColor || '#000000'),
+              ),
+            );
+
+          await settingsMenu.showModal(modal);
+
+          const modalSubmit = await settingsMenu.awaitModalSubmit({
+            time: 60_000,
+            filter: (i) => i.customId === modal.data.custom_id,
+          }).catch((e) => {
+            if (!e.message.includes('reason: time')) {
+              logger.error(e);
+              captureException(e);
+            }
+            return null;
+          });
+
+          if (!modalSubmit) return;
+
+
+          const embedColor = modalSubmit.fields.getTextInputValue('embed_color');
+
+          const hex_regex = /^#[0-9A-F]{6}$/i;
+          if (!hex_regex.test(embedColor)) {
+            modalSubmit.reply({
+              content: `${emoji.normal.no} Invalid hex color code. Please try again.`,
+              ephemeral: true,
+            });
+            return;
+          }
+
+          await db.connectedList.update({
+            where: { channelId: updConnection.channelId },
+            data: { embedColor },
+          });
+
+          modalSubmit.reply({
+            content: `${emoji.normal.yes} Embed color successfully set to \`${embedColor}\`!`,
+            ephemeral: true,
+          });
+          break;
+        }
+
         /* Compact / Normal mode toggle  */
         case 'compact': {
           await db.connectedList.update({