Revert "Remove database-related message content usage"

This reverts commit 0b211db023.

src/utils/event-log-utils.ts

@@ -1,6 +1,9 @@
-import { APIEmbedField, EmbedBuilder, Guild, GuildBasedChannel, GuildTextBasedChannel } from 'discord.js';
+import { APIEmbedField, AttachmentBuilder, EmbedBuilder, Guild, GuildBasedChannel, GuildTextBasedChannel, MessageCreateOptions } from 'discord.js';
+import { Message } from '../events';
 import { Database } from '../shared/Database';
+import { LoggedMessage } from '../types';
 import { channelIsInStaffCategory } from './channel-utils';
+import { deserializeMessagePart, getModifiedAttachments, getModifiedStickers } from './message-utils';
 
 type CustomEventLogData = {
   title: string
@@ -10,6 +13,63 @@ type CustomEventLogData = {
   fields: APIEmbedField[] | null
 }
 
+export async function logEdit(loggedMessage: LoggedMessage, newMessage: Message<true>) {
+  const channel = await getEventLogChannel(newMessage.guild, newMessage.channel);
+
+  if (!channel) return;
+
+  const includeContentInEmbed = loggedMessage.content.length <= 1024 && newMessage.content.length <= 1024;
+
+  const fields: APIEmbedField[] = [];
+  fields.push(...getMainEmbeds(loggedMessage, newMessage));
+  fields.push(...getEditEmbeds(loggedMessage, newMessage, includeContentInEmbed));
+
+  const embed = new EmbedBuilder()
+    .setTitle('Edited Message')
+    .setColor(0xFFFF00)
+    .setTimestamp(newMessage.createdTimestamp)
+    .addFields(...fields);
+
+  const messagePayload: MessageCreateOptions = { embeds: [embed] };
+
+  if (!includeContentInEmbed) {
+    const before = new AttachmentBuilder(Buffer.from(loggedMessage.content), { name: 'before.txt' });
+    const after = new AttachmentBuilder(Buffer.from(newMessage.content), { name: 'after.txt' });
+
+    messagePayload.files = [before, after];
+  }
+
+  channel.send(messagePayload);
+}
+
+export async function logDeletion(loggedMessage: LoggedMessage, deletedMessage: Message<true>) {
+  const channel = await getEventLogChannel(deletedMessage.guild, deletedMessage.channel);
+
+  if (!channel) return;
+
+  const includeContentInEmbed = loggedMessage.content.length <= 1024;
+
+  const fields: APIEmbedField[] = [];
+  fields.push(...getMainEmbeds(loggedMessage, deletedMessage));
+  fields.push(...getDeletedEmbeds(loggedMessage, includeContentInEmbed));
+
+  const embed = new EmbedBuilder()
+    .setTitle('Deleted Message')
+    .setColor(0xFF0000)
+    .setTimestamp(deletedMessage.createdTimestamp)
+    .addFields(...fields);
+
+  const messagePayload: MessageCreateOptions = { embeds: [embed] };
+
+  if (!includeContentInEmbed) {
+    const before = new AttachmentBuilder(Buffer.from(loggedMessage.content), { name: 'content.txt' });
+
+    messagePayload.files = [before];
+  }
+
+  channel.send(messagePayload);
+}
+
 export async function logCustomEvent(guild: Guild, data: CustomEventLogData) {
   const channel = await getEventLogChannel(guild);
 
@@ -47,4 +107,113 @@ async function getEventLogChannel(guild: Guild, channel: GuildBasedChannel | nul
 
     return channel;
   }
+}
+
+function getMainEmbeds(loggedMessage: LoggedMessage, newMessage: Message<true>): APIEmbedField[] {
+  const channelString = `${newMessage.channel.toString()}\n${newMessage.channel.name}`;
+  const userString = `<@${loggedMessage.author_id}>\n${loggedMessage.author_name}`;
+
+  return [
+    {
+      name: 'Channel',
+      value: channelString,
+      inline: true
+    },
+    {
+      name: 'User',
+      value: userString,
+      inline: true
+    },
+    {
+      name: 'Message',
+      value: `[${newMessage.id}](${newMessage.url})`,
+      inline: true
+    },
+  ];
+}
+
+function getDeletedEmbeds(loggedMessage: LoggedMessage, includeContentInEmbed = true): APIEmbedField[] {
+  const fields: APIEmbedField[] = [];
+
+  if (includeContentInEmbed && loggedMessage.content != '') {
+    fields.push({
+      name: 'Content',
+      value: loggedMessage.content,
+      inline: false
+    });
+  }
+
+  for (const attachment of deserializeMessagePart(loggedMessage.attachments)) {
+    fields.push({
+      name: 'Attachment',
+      value: attachment,
+      inline: true
+    });
+  }
+
+  for (const sticker of deserializeMessagePart(loggedMessage.stickers)) {
+    fields.push({
+      name: 'Stickers',
+      value: sticker,
+      inline: true
+    });
+  }
+
+  return fields;
+}
+
+function getEditEmbeds(loggedMessage: LoggedMessage, newMessage: Message<true>, includeContentInEmbed = true): APIEmbedField[] {
+  const fields: APIEmbedField[] = [];
+  if (includeContentInEmbed && loggedMessage.content != newMessage.content) {
+    fields.push(
+      {
+        name: 'Before',
+        value: loggedMessage.content,
+        inline: false
+      },
+      {
+        name: 'After',
+        value: newMessage.content,
+        inline: false
+      }
+    );
+  }
+
+  const { addedAttachments, removedAttachments } = getModifiedAttachments(loggedMessage, newMessage);
+
+  for (const removedAttachment of removedAttachments) {
+    fields.push({
+      name: 'Removed Attachment',
+      value: removedAttachment,
+      inline: true
+    });
+  }
+
+  for (const addedAttachment of addedAttachments) {
+    fields.push({
+      name: 'Added Attachment',
+      value: addedAttachment,
+      inline: true
+    });
+  }
+
+  const { addedStickers, removedStickers } = getModifiedStickers(loggedMessage, newMessage);
+
+  for (const removedSticker of addedStickers) {
+    fields.push({
+      name: 'Removed Sticker',
+      value: removedSticker,
+      inline: true
+    });
+  }
+
+  for (const addedSticker of removedStickers) {
+    fields.push({
+      name: 'Added Sticker',
+      value: addedSticker,
+      inline: true
+    });
+  }
+
+  return fields;
 }

src/utils/message-utils.ts

@@ -1,4 +1,5 @@
 import { Message } from '../events';
+import { LoggedMessage } from '../types';
 
 export const ARRAY_SEPARATOR = '$';
 
@@ -15,6 +16,34 @@ export function deserializeMessagePart(part: string): string[] {
   return part.split(ARRAY_SEPARATOR).filter(e => e);
 }
 
+export function getModifiedAttachments(loggedMessage: LoggedMessage, newMessage: Message): { addedAttachments: string[], removedAttachments: string[] } {
+  const loggedAttachments = loggedMessage.attachments.split(ARRAY_SEPARATOR).filter(e => e);
+  const addedAttachments = newMessage.attachments.filter(a => !loggedAttachments.includes(`${a.name}:${a.id}`)).map(a => `${a.name}:${a.id}`);
+  const removedAttachments = loggedAttachments.filter(a => !newMessage.attachments.has(a.split(':').at(-1)!));
+
+  return { addedAttachments, removedAttachments };
+}
+
+export function getModifiedStickers(loggedMessage: LoggedMessage, newMessage: Message): { addedStickers: string[], removedStickers: string[] } {
+  const loggedStickers = loggedMessage.stickers.split(ARRAY_SEPARATOR).filter(e => e);
+  const addedStickers = newMessage.stickers.filter(s => !loggedStickers.includes(`${s.name}:${s.id}`)).map(s => `${s.name}:${s.id}`);
+  const removedStickers = loggedStickers.filter(s => !newMessage.stickers.has(s.split(':').at(-1)!));
+
+  return { addedStickers, removedStickers };
+}
+
+export function isEdited(loggedMessage: LoggedMessage, newMessage: Message) {
+  if (newMessage.content != loggedMessage.content) return true;
+
+  const { addedAttachments, removedAttachments } = getModifiedAttachments(loggedMessage, newMessage);
+
+  if (addedAttachments.length > 0 || removedAttachments.length > 0) return true;
+
+  const { addedStickers, removedStickers } = getModifiedStickers(loggedMessage, newMessage);
+
+  return addedStickers.length > 0 || removedStickers.length > 0;
+}
+
 export function isInSpoilerTags(content: string, index: number): boolean {
   if (!content.includes('||')) return false;