Squashed commit of the following:

commit e8a57468de Author: Tarrgon <61888458+Tarrgon@users.noreply.github.com> Date: Sat May 31 11:10:56 2025 -0400 Add record command commit 5a8ca6bb54 Author: Tarrgon <61888458+Tarrgon@users.noreply.github.com> Date: Sat May 31 11:10:30 2025 -0400 Move context menus to their own folder commit 2d7c14ed7a Merge: 33a6ba8 5107759 Author: Tarrgon <61888458+Tarrgon@users.noreply.github.com> Date: Sat May 31 10:54:27 2025 -0400 Merge branch 'alt-data' into records commit 5107759923 Author: Tarrgon <61888458+Tarrgon@users.noreply.github.com> Date: Fri May 30 15:26:21 2025 -0400 Recursive alt lookup data
2025-05-31 11:11:37 -04:00
parent 33a6ba893b
commit f5268ace77
11 changed files with 306 additions and 50 deletions
@@ -0,0 +1,104 @@
+import { ActionRowBuilder, ButtonBuilder, ButtonStyle, Embed, EmbedBuilder, Guild } from 'discord.js';
+import { Database } from '../shared/Database';
+import { E621User, MessageContent, Record, RecordCategory } from '../types';
+import { getE621User, getUserRecords } from './e621-utils';
+import { config } from '../config';
+import { e621IdsFromAltData, comprehensiveAltLookupFromDiscord } from './alt-utils';
+
+type RecordWithUserData = Record & { user: E621User, creator: E621User, updater: E621User }
+type AllRecords = RecordWithUserData[];
+
+const RECORDS_PER_PAGE = 5;
+
+export async function getAllRecordsFromDiscordId(id: string, guild: Guild): Promise<AllRecords> {
+  const altData = await comprehensiveAltLookupFromDiscord(id, guild);
+  const userCache: Map<number, E621User> = new Map();
+
+  const allRecords: AllRecords = [];
+
+  const e621UserIds = e621IdsFromAltData(altData);
+
+  for (const id of e621UserIds) {
+    const user = userCache.get(id) ?? await getE621User(id);
+    if (!user) continue;
+    userCache.set(id, user);
+
+    const records = await getUserRecords(id);
+    for (const record of records) {
+      const creator = userCache.get(record.creator_id) ?? await getE621User(record.creator_id);
+      if (!creator) continue;
+      userCache.set(record.creator_id, creator);
+
+      const updater = userCache.get(record.updater_id) ?? await getE621User(record.updater_id);
+      if (!updater) continue;
+      userCache.set(record.updater_id, updater);
+
+      allRecords.push({
+        user,
+        creator,
+        updater,
+        ...record
+      });
+    }
+  }
+
+  return allRecords;
+}
+
+export async function getRecordMessageFromDiscordId(id: string, page: number, guild: Guild): Promise<MessageContent | null> {
+  const records = await getAllRecordsFromDiscordId(id, guild);
+
+  if (records.length == 0) return null;
+
+  page = page - 1;
+
+  const maxPage = Math.floor(records.length / RECORDS_PER_PAGE);
+
+  const embeds: EmbedBuilder[] = [];
+
+  for (let i = page * RECORDS_PER_PAGE; i < page * RECORDS_PER_PAGE + RECORDS_PER_PAGE; i++) {
+    if (i >= records.length) break;
+
+    embeds.push(getRecordEmbed(records[i]));
+  }
+
+  const prevPage = new ButtonBuilder()
+    .setLabel('Previous Page')
+    .setCustomId(`records-previous_${id}_${page + 1}`)
+    .setDisabled(page == 0)
+    .setStyle(ButtonStyle.Primary);
+
+  const nextPage = new ButtonBuilder()
+    .setLabel('Next Page')
+    .setCustomId(`records-next_${id}_${page + 1}`)
+    .setDisabled(page >= maxPage)
+    .setStyle(ButtonStyle.Primary);
+
+  const row = new ActionRowBuilder<ButtonBuilder>()
+    .addComponents(prevPage, nextPage);
+
+  return { content: `Records found for <@${id}>`, embeds, components: [row] };
+}
+
+function getRecordColor(category: RecordCategory) {
+  switch (category) {
+    case 'positive': return 0x00ff00;
+    case 'negative': return 0xff0000;
+    case 'neutral': return 0xaaaaaa;
+  }
+}
+
+function getRecordEmbed(record: RecordWithUserData): EmbedBuilder {
+  const isUpdated = record.updated_at != record.created_at;
+  const creator = isUpdated ? record.updater : record.creator;
+  return new EmbedBuilder()
+    .setColor(getRecordColor(record.category))
+    .setTitle(`Record from ${record.creator.name} for ${record.user.name}`)
+    .setDescription(record.body.trim())
+    .setURL(`${config.E621_BASE_URL}/user_feedbacks/${record.id}`)
+    .setAuthor({
+      name: `${isUpdated ? 'Last updated by' : 'Created by'}: ${creator.name}`,
+      url: `${config.E621_BASE_URL}/users/${creator.id}`
+    })
+    .setTimestamp(new Date(record.updated_at));
+}