Tarrgon
103 lines
3.4 KiB
TypeScript
103 lines
3.4 KiB
TypeScript
import { ActionRowBuilder, ButtonBuilder, ButtonStyle, EmbedBuilder, Guild } from 'discord.js';
|
|
import { config } from '../config';
|
|
import { E621User, MessageContent, Record, RecordCategory } from '../types';
|
|
import { comprehensiveAltLookupFromDiscord, e621IdsFromAltData } from './alt-utils';
|
|
import { getE621User, getUserRecords } from './e621-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));
|
|
} |