commands/moderation/NotesCommand.ts

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* SudoBot is free software; you can redistribute it and/or modify it
* under the terms of the GNU Affero General Public License as published by 
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* 
* SudoBot is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of 
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License 
* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
*/

import { CommandInteraction, Interaction, InteractionCollector, Message, MessageActionRow, MessageButton, Permissions, 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 Note from '../../models/Note';

export default class NotesCommand extends BaseCommand {
    supportsInteractions: boolean = true;
    permissions = [Permissions.FLAGS.MANAGE_MESSAGES];

    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.find({
            guild_id: msg.guild!.id,
            user_id: user.id,
        }).skip(offset).limit(limit).sort("createdAt");

        let str = '';
        
        const maxPage = Math.ceil((await Note.count({
            guild_id: msg.guild!.id,
            user_id: user.id,
        })) / limit);

        for await (const note of notes) {
            str += `**Note ID**: ${note.id}\n`;
            str += `Note taken by: ${note.mod_tag}\n`;
            str += `Date: ${note.createdAt.toLocaleString()}\n`;
            str += `Content:\n\`\`\`\n${note.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 || isNaN(parseInt(splitted[2]))) 
                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	rakinar2	577	/**
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2022 OSN Inc.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import { CommandInteraction, Interaction, InteractionCollector, Message, MessageActionRow, MessageButton, Permissions, User } from 'discord.js';
21			import BaseCommand from '../../utils/structures/BaseCommand';
22			import DiscordClient from '../../client/Client';
23			import CommandOptions from '../../types/CommandOptions';
24			import InteractionOptions from '../../types/InteractionOptions';
25			import MessageEmbed from '../../client/MessageEmbed';
26			import getUser from '../../utils/getUser';
27			import Note from '../../models/Note';
28
29			export default class NotesCommand extends BaseCommand {
30			supportsInteractions: boolean = true;
31			permissions = [Permissions.FLAGS.MANAGE_MESSAGES];
32
33			constructor() {
34			super('notes', 'moderation', []);
35			}
36
37			async genEmbed(client: DiscordClient, msg: Message \| Interaction, user: User, page: number = 1) {
38			const limit = 5;
39			const offset = ((page < 1 ? 1 : page) - 1) * limit;
40
41			const notes = await Note.find({
42			guild_id: msg.guild!.id,
43			user_id: user.id,
44			}).skip(offset).limit(limit).sort("createdAt");
45
46			let str = '';
47
48			const maxPage = Math.ceil((await Note.count({
49			guild_id: msg.guild!.id,
50			user_id: user.id,
51			})) / limit);
52
53			for await (const note of notes) {
54			str += `Note ID: ${note.id}\n`;
55			str += `Note taken by: ${note.mod_tag}\n`;
56			str += `Date: ${note.createdAt.toLocaleString()}\n`;
57			str += `Content:\n\`\`\`\n${note.content}\n\`\`\`\n`;
58			str += '\n';
59			}
60
61			return {
62			embeds: [
63			new MessageEmbed({
64			author: {
65			name: user.tag,
66			iconURL: user.displayAvatarURL()
67			},
68			title: 'Notes',
69			description: str === '' ? 'No notes.' : str,
70			timestamp: new Date(),
71			})
72			],
73			components: [
74			this.createActionRow(page, maxPage)
75			]
76			};
77			}
78
79			createActionRow(page: number, max: number) {
80			console.log(max);
81
82			const back = new MessageButton({
83			customId: 'notes-back-',
84			label: '<<',
85			style: 'PRIMARY'
86			});
87
88			const next = new MessageButton({
89			customId: 'notes-next-',
90			label: '>>',
91			style: 'PRIMARY'
92			});
93
94			let nextPage = page + 1;
95			console.log(nextPage);
96
97			if (nextPage > max) {
98			nextPage = max;
99			next.setDisabled(true);
100			}
101			else {
102			next.setDisabled(false);
103			}
104
105			let prevPage = page - 1;
106			console.log(prevPage);
107
108			if (prevPage <= 0) {
109			prevPage = 1;
110			back.setDisabled(true);
111			}
112			else {
113			back.setDisabled(false);
114			}
115
116			next.setCustomId('notes-next-' + nextPage);
117			back.setCustomId('notes-back-' + prevPage);
118
119			return new MessageActionRow()
120			.addComponents(
121			back,
122			next
123			);
124			}
125
126			async update(client: DiscordClient, interaction: Interaction, message: Message) {
127			console.log('here');
128
129			if (interaction.isButton() && interaction.customId.startsWith('notes-')) {
130			const splitted = interaction.customId.split('-');
131
132			if (splitted[2] === undefined \|\| isNaN(parseInt(splitted[2])))
133			return;
134
135			if (splitted[1] === 'next' \|\| splitted[1] === 'back') {
136			const options = await this.genEmbed(client, interaction, (global as any).user, parseInt(splitted[2]));
137
138			try {
139			await interaction.update(options);
140			}
141			catch (e) {
142			console.log(e);
143			}
144			}
145			}
146			}
147
148			async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
149			if (!options.isInteraction && typeof options.args[0] === 'undefined') {
150			await msg.reply({
151			embeds: [
152			new MessageEmbed()
153			.setColor('#f14a60')
154			.setDescription(`This command requires at least one argument.`)
155			]
156			});
157
158			return;
159			}
160
161			let user: User \| null \| undefined;
162
163			if (options.isInteraction) {
164			user = await <User> options.options.getUser('user');
165			}
166			else {
167			try {
168			user = await getUser(client, msg as Message, options);
169
170			if (!user)
171			throw new Error();
172			}
173			catch (e) {
174			console.log(e);
175
176			await msg.reply({
177			embeds: [
178			new MessageEmbed()
179			.setColor('#f14a60')
180			.setDescription(`Invalid user given.`)
181			]
182			});
183
184			return;
185			}
186			}
187
188			(global as any).user = user;
189
190			let message = <Message> await msg.reply(await this.genEmbed(client, msg, user));
191
192			if (msg instanceof CommandInteraction)
193			message = <Message> await msg.fetchReply();
194
195			const collector = new InteractionCollector(client, {
196			guild: msg.guild!,
197			channel: msg.channel!,
198			max: 20,
199			componentType: 'BUTTON',
200			interactionType: 'MESSAGE_COMPONENT',
201			message,
202			time: 30000,
203			filter(i) {
204			return i.isButton() && i.member?.user.id === msg.member!.user.id;
205			}
206			});
207
208			collector.on('collect', async i => {
209			await this.update(client, i, message);
210			});
211
212			collector.on('end', async () => {
213			try {
214			await message.edit({
215			components: []
216			});
217			}
218			catch (e) {
219			console.log(e);
220			}
221			});
222			}
223			}