replies: Change reply embed format

Add reference author to "author#iconURL" and move username to "author#name" Signed-off-by: Woomymy <woomy@woomy.be>
federation-interservices-d-informatique · Feb 2, 2023 · 74b41bb · 74b41bb
1 parent 9e0cab7
commit 74b41bb
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 12 deletions.
diff --git a/src/events/messageCreate.ts b/src/events/messageCreate.ts
@@ -1,12 +1,13 @@
 import { clientEvent } from "@federation-interservices-d-informatique/fiibot-common";
 import { Prisma } from "@prisma/client";
-import { Colors, Embed, Message, MessageType, TextChannel } from "discord.js";
+import { Message, MessageType, TextChannel } from "discord.js";
 import { InterServerClient } from "../classes/InterServerClient";
 import {
     INTERSERVER_WH_NAME,
     SERVERS_HEADERS,
     ServersHeadersKey
 } from "../utils/constants.js";
+import { makeReplyEmbed } from "../utils/embeds.js";
 
 export default clientEvent({
     name: "messageCreate",
@@ -51,11 +52,7 @@ export default clientEvent({
                 const reference = await msg.fetchReference();
                 if (!reference) return;
 
-                msg.embeds.push({
-                    title: `En réponse à ${reference.author.username}`,
-                    description: `>>> ${reference.cleanContent}`,
-                    color: Colors.Blue
-                } as Embed);
+                msg.embeds.push(makeReplyEmbed(reference));
             }
 
             const whMessage = await webhook.send({

diff --git a/src/events/messageUpdate.ts b/src/events/messageUpdate.ts
@@ -1,12 +1,13 @@
 import { clientEvent } from "@federation-interservices-d-informatique/fiibot-common";
-import { Message, PartialMessage, TextChannel } from "discord.js";
+import { Message, MessageType, PartialMessage, TextChannel } from "discord.js";
 import { InterServerClient } from "../classes/InterServerClient";
 import { MessageCloneData } from "../typings";
 import {
     INTERSERVER_WH_NAME,
     ServersHeadersKey,
     SERVERS_HEADERS
 } from "../utils/constants.js";
+import { makeReplyEmbed } from "../utils/embeds.js";
 
 export default clientEvent({
     name: "messageUpdate",
@@ -34,11 +35,12 @@ export default clientEvent({
                 (hook) => hook.name === INTERSERVER_WH_NAME
             );
             if (webHook) {
-                const replyEmbed = oldmessage.embeds.find((e) =>
-                    e.title?.startsWith("En réponse à")
-                );
-                if (replyEmbed) {
-                    newmessage.embeds.push(replyEmbed);
+                // Handle message replies
+                if (newmessage.type === MessageType.Reply) {
+                    const reference = await newmessage.fetchReference();
+                    if (!reference) return;
+
+                    newmessage.embeds.push(makeReplyEmbed(reference));
                 }
                 try {
                     const hookMessage = await webHook.fetchMessage(clone.id);

diff --git a/src/utils/embeds.ts b/src/utils/embeds.ts
@@ -0,0 +1,16 @@
+import { Colors, Embed, Message } from "discord.js";
+
+/**
+ * Creates a "reply" embed
+ * @param {Message} [reference] The referenced message
+ */
+export function makeReplyEmbed(reference: Message): Embed {
+    return {
+        author: {
+            iconURL: reference.author.displayAvatarURL(),
+            name: `En réponse à ${reference.author.username}`
+        },
+        description: `>>> ${reference.cleanContent}`,
+        color: Colors.Blue
+    } as Embed;
+}