commands/moderation/NotesCommand.ts

import { BanOptions, CommandInteraction, EmojiIdentifierResolvable, GuildMember, Interaction, InteractionCollector, Message, MessageActionRow, MessageButton, MessageOptions, ReplyOptions, TextChannel, User } from 'discord.js';
import BaseCommand from '../../utils/structures/BaseCommand';
import DiscordClient from '../../client/Client';
import CommandOptions from '../../types/CommandOptions';
import InteractionOptions from '../../types/InteractionOptions';
import MessageEmbed from '../../client/MessageEmbed';
import getUser from '../../utils/getUser';
import getMember from '../../utils/getMember';
import History from '../../automod/History';
import { fetchEmoji } from '../../utils/Emoji';
import PunishmentType from '../../types/PunishmentType';
import Note from '../../models/Note';

export default class NotesCommand extends BaseCommand {
    supportsInteractions: boolean = true;

    constructor() {
        super('notes', 'moderation', []);
    }

    async genEmbed(client: DiscordClient, msg: Message | Interaction, user: User, page: number = 1) {
        const limit = 5;
        const offset = ((page < 1 ? 1 : page) - 1) * limit;

        const notes = await Note.findAndCountAll({
            where: {
                guild_id: msg.guild!.id,
                user_id: user.id
            },
            order: [
                ['createdAt', 'DESC']
            ],
            limit,
            offset
        });

        let str = '';
        const maxPage = Math.ceil(notes.count / limit);

        for await (const note of notes.rows) {
            str += `**Note ID**: ${note.get().id}\n`;
            str += `Note taken by: ${note.get().mod_tag}\n`;
            str += `Date: ${note.get().createdAt.toLocaleString()}\n`;
            str += `Content:\n\`\`\`\n${note.get().content}\n\`\`\`\n`;
            str += '\n';
        }

        return {
            embeds: [
                new MessageEmbed({
                    author: {
                        name: user.tag,
                        iconURL: user.displayAvatarURL()
                    },
                    title: 'Notes',
                    description: str === '' ? 'No notes.' : str,
                    timestamp: new Date(),
                })
            ],
            components: [
                this.createActionRow(page, maxPage)
            ]
        };
    }

    createActionRow(page: number, max: number) {
        console.log(max);
        
        const back = new MessageButton({
            customId: 'notes-back-',
            label: '<<',
            style: 'PRIMARY'
        });

        const next = new MessageButton({
            customId: 'notes-next-',
            label: '>>',
            style: 'PRIMARY'
        });

        let nextPage = page + 1;
        console.log(nextPage);

        if (nextPage > max) {
            nextPage = max;
            next.setDisabled(true);
        }
        else {
            next.setDisabled(false);
        }

        let prevPage = page - 1;
        console.log(prevPage);

        if (prevPage <= 0) {
            prevPage = 1;
            back.setDisabled(true);
        }
        else {
            back.setDisabled(false);
        }

        next.setCustomId('notes-next-' + nextPage);
        back.setCustomId('notes-back-' + prevPage);

        return new MessageActionRow()
            .addComponents(
                back,
                next
            );
    }

    async update(client: DiscordClient, interaction: Interaction, message: Message) {
        console.log('here');
        
        if (interaction.isButton() && interaction.customId.startsWith('notes-')) {
            const splitted = interaction.customId.split('-');

            if (splitted[2] === undefined || parseInt(splitted[2]) === NaN) 
                return;

            if (splitted[1] === 'next' || splitted[1] === 'back') {
                const options = await this.genEmbed(client, interaction, (global as any).user, parseInt(splitted[2]));
                
                try {
                    await interaction.update(options);
                }
                catch (e) {
                    console.log(e);                    
                }
            }
        }
    }

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[0] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least one argument.`)
                ]
            });

            return;
        }

        let user: User | null | undefined;

        if (options.isInteraction) {
            user = await <User> options.options.getUser('user');
        }
        else {
            try {
                user = await getUser(client, msg as Message, options);

                if (!user) 
                    throw new Error();
            }
            catch (e) {
                console.log(e);
                
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(`Invalid user given.`)
                    ]
                });

                return;
            }
        }

        (global as any).user = user;

        let message = <Message> await msg.reply(await this.genEmbed(client, msg, user));

        if (msg instanceof CommandInteraction)
            message = <Message> await msg.fetchReply();

        const collector = new InteractionCollector(client, {
            guild: msg.guild!,
            channel: msg.channel!,
            max: 20,
            componentType: 'BUTTON',
            interactionType: 'MESSAGE_COMPONENT',
            message,
            time: 30000,
            filter(i) {
                return i.isButton() && i.member?.user.id === msg.member!.user.id;
            }
        });

        collector.on('collect', async i => {
            await this.update(client, i, message);
        });

        collector.on('end', async () => {
            try {
                await message.edit({
                    components: []
                });
            }
            catch (e) {
                console.log(e);                
            }
        });
    }
}
1	import { BanOptions, CommandInteraction, EmojiIdentifierResolvable, GuildMember, Interaction, InteractionCollector, Message, MessageActionRow, MessageButton, MessageOptions, ReplyOptions, TextChannel, User } from 'discord.js';
2	import BaseCommand from '../../utils/structures/BaseCommand';
3	import DiscordClient from '../../client/Client';
4	import CommandOptions from '../../types/CommandOptions';
5	import InteractionOptions from '../../types/InteractionOptions';
6	import MessageEmbed from '../../client/MessageEmbed';
7	import getUser from '../../utils/getUser';
8	import getMember from '../../utils/getMember';
9	import History from '../../automod/History';
10	import { fetchEmoji } from '../../utils/Emoji';
11	import PunishmentType from '../../types/PunishmentType';
12	import Note from '../../models/Note';
13
14	export default class NotesCommand extends BaseCommand {
15	supportsInteractions: boolean = true;
16
17	constructor() {
18	super('notes', 'moderation', []);
19	}
20
21	async genEmbed(client: DiscordClient, msg: Message \| Interaction, user: User, page: number = 1) {
22	const limit = 5;
23	const offset = ((page < 1 ? 1 : page) - 1) * limit;
24
25	const notes = await Note.findAndCountAll({
26	where: {
27	guild_id: msg.guild!.id,
28	user_id: user.id
29	},
30	order: [
31	['createdAt', 'DESC']
32	],
33	limit,
34	offset
35	});
36
37	let str = '';
38	const maxPage = Math.ceil(notes.count / limit);
39
40	for await (const note of notes.rows) {
41	str += `Note ID: ${note.get().id}\n`;
42	str += `Note taken by: ${note.get().mod_tag}\n`;
43	str += `Date: ${note.get().createdAt.toLocaleString()}\n`;
44	str += `Content:\n\`\`\`\n${note.get().content}\n\`\`\`\n`;
45	str += '\n';
46	}
47
48	return {
49	embeds: [
50	new MessageEmbed({
51	author: {
52	name: user.tag,
53	iconURL: user.displayAvatarURL()
54	},
55	title: 'Notes',
56	description: str === '' ? 'No notes.' : str,
57	timestamp: new Date(),
58	})
59	],
60	components: [
61	this.createActionRow(page, maxPage)
62	]
63	};
64	}
65
66	createActionRow(page: number, max: number) {
67	console.log(max);
68
69	const back = new MessageButton({
70	customId: 'notes-back-',
71	label: '<<',
72	style: 'PRIMARY'
73	});
74
75	const next = new MessageButton({
76	customId: 'notes-next-',
77	label: '>>',
78	style: 'PRIMARY'
79	});
80
81	let nextPage = page + 1;
82	console.log(nextPage);
83
84	if (nextPage > max) {
85	nextPage = max;
86	next.setDisabled(true);
87	}
88	else {
89	next.setDisabled(false);
90	}
91
92	let prevPage = page - 1;
93	console.log(prevPage);
94
95	if (prevPage <= 0) {
96	prevPage = 1;
97	back.setDisabled(true);
98	}
99	else {
100	back.setDisabled(false);
101	}
102
103	next.setCustomId('notes-next-' + nextPage);
104	back.setCustomId('notes-back-' + prevPage);
105
106	return new MessageActionRow()
107	.addComponents(
108	back,
109	next
110	);
111	}
112
113	async update(client: DiscordClient, interaction: Interaction, message: Message) {
114	console.log('here');
115
116	if (interaction.isButton() && interaction.customId.startsWith('notes-')) {
117	const splitted = interaction.customId.split('-');
118
119	if (splitted[2] === undefined \|\| parseInt(splitted[2]) === NaN)
120	return;
121
122	if (splitted[1] === 'next' \|\| splitted[1] === 'back') {
123	const options = await this.genEmbed(client, interaction, (global as any).user, parseInt(splitted[2]));
124
125	try {
126	await interaction.update(options);
127	}
128	catch (e) {
129	console.log(e);
130	}
131	}
132	}
133	}
134
135	async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
136	if (!options.isInteraction && typeof options.args[0] === 'undefined') {
137	await msg.reply({
138	embeds: [
139	new MessageEmbed()
140	.setColor('#f14a60')
141	.setDescription(`This command requires at least one argument.`)
142	]
143	});
144
145	return;
146	}
147
148	let user: User \| null \| undefined;
149
150	if (options.isInteraction) {
151	user = await <User> options.options.getUser('user');
152	}
153	else {
154	try {
155	user = await getUser(client, msg as Message, options);
156
157	if (!user)
158	throw new Error();
159	}
160	catch (e) {
161	console.log(e);
162
163	await msg.reply({
164	embeds: [
165	new MessageEmbed()
166	.setColor('#f14a60')
167	.setDescription(`Invalid user given.`)
168	]
169	});
170
171	return;
172	}
173	}
174
175	(global as any).user = user;
176
177	let message = <Message> await msg.reply(await this.genEmbed(client, msg, user));
178
179	if (msg instanceof CommandInteraction)
180	message = <Message> await msg.fetchReply();
181
182	const collector = new InteractionCollector(client, {
183	guild: msg.guild!,
184	channel: msg.channel!,
185	max: 20,
186	componentType: 'BUTTON',
187	interactionType: 'MESSAGE_COMPONENT',
188	message,
189	time: 30000,
190	filter(i) {
191	return i.isButton() && i.member?.user.id === msg.member!.user.id;
192	}
193	});
194
195	collector.on('collect', async i => {
196	await this.update(client, i, message);
197	});
198
199	collector.on('end', async () => {
200	try {
201	await message.edit({
202	components: []
203	});
204	}
205	catch (e) {
206	console.log(e);
207	}
208	});
209	}
210	}