commands/moderation/ShotCommand.ts

import { BanOptions, CommandInteraction, ContextMenuInteraction, GuildMember, Interaction, Message, 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 getMember from '../../utils/getMember';
import Punishment from '../../models/Punishment';
import PunishmentType from '../../types/PunishmentType';

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

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

    async run(client: DiscordClient, msg: Message | CommandInteraction | ContextMenuInteraction, 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: GuildMember;
        let dm = true;
        let reason: string | undefined;

        if (options.isInteraction) {
            user = await <GuildMember> (msg instanceof ContextMenuInteraction ? options.options.getMember('user') : options.options.getMember('member'));

            if (!user) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription("Invalid member given.")
                    ]
                });
    
                return;
            }

            if (options.options.getString('reason')) {
                reason = await <string> options.options.getString('reason');
            }
        }
        else {
            try {
                const user2 = await getMember((msg as Message), options);

                if (!user2) {
                    throw new Error('Invalid user');
                }

                user = user2;
            }
            catch (e) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(`Invalid user given.`)
                    ]
                });
    
                return;
            }

            console.log(user);

            if (options.args[1]) {
                const args = [...options.args];
                args.shift();
                reason = await args.join(' ');
            }
        }

        try {            
            await Punishment.create({
                type: PunishmentType.SHOT,
                user_id: user.id,
                guild_id: msg.guild!.id,
                mod_id: msg.member!.user.id,
                mod_tag: (msg.member!.user as User).tag,
                reason,
                createdAt: new Date()
            });

            // await History.create(user.id, msg.guild!, 'bean', msg.member!.user.id, typeof reason === 'undefined' ? null : reason);

            try {
                await user.send({
                    embeds: [
                        new MessageEmbed()
                            .setAuthor({
                                iconURL: <string> msg.guild!.iconURL(),
                                name: `\tYou got a shot in ${msg.guild!.name}`
                            })
                            .addFields([
                                {
                                    name: "Reason",
                                    value: typeof reason === 'undefined' ? '*No reason provided*' : reason
                                }
                            ])
                    ]
                });
            }
            catch (e) {
                console.log(e);
                dm = false;
            }

            // await client.logger.logBeaned(user, typeof reason === 'undefined' ? '*No reason provided*' : reason, msg.member!.user as User);
        }
        catch (e) {
            console.log(e);            
        }

        await msg.reply({
            embeds: [
                new MessageEmbed()
                .setAuthor({
                    name: user.user.tag,
                    iconURL: user.user.displayAvatarURL(),
                })
                .setDescription(user.user.tag + " got a shot." + (!dm ? "\nThey have DMs disabled. They will not know that they got a shot." : ''))
                .addFields([
                    {
                        name: "Given by",
                        value: (msg.member!.user as User).tag
                    },
                    {
                        name: "Reason",
                        value: reason === undefined ? "*No reason provided*" : reason
                    }
                ])
            ]
        });
    }
}
1	rakin	125	import { BanOptions, CommandInteraction, ContextMenuInteraction, GuildMember, Interaction, Message, User } from 'discord.js';
2	rakin	51	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 getMember from '../../utils/getMember';
8	rakin	86	import Punishment from '../../models/Punishment';
9			import PunishmentType from '../../types/PunishmentType';
10	rakin	51
11	rakin	197	export default class ShotCommand extends BaseCommand {
12	rakin	51	supportsInteractions: boolean = true;
13	rakin	125	supportsContextMenu: boolean = true;
14	rakin	51
15			constructor() {
16	rakin	197	super('shot', 'moderation', ['Shot']);
17	rakin	51	}
18
19	rakin	125	async run(client: DiscordClient, msg: Message \| CommandInteraction \| ContextMenuInteraction, options: CommandOptions \| InteractionOptions) {
20	rakin	51	if (!options.isInteraction && typeof options.args[0] === 'undefined') {
21			await msg.reply({
22			embeds: [
23			new MessageEmbed()
24			.setColor('#f14a60')
25			.setDescription(`This command requires at least one argument.`)
26			]
27			});
28
29			return;
30			}
31
32			let user: GuildMember;
33	rakin	160	let dm = true;
34	rakin	51	let reason: string \| undefined;
35
36			if (options.isInteraction) {
37	rakin	125	user = await <GuildMember> (msg instanceof ContextMenuInteraction ? options.options.getMember('user') : options.options.getMember('member'));
38	rakin	51
39			if (!user) {
40			await msg.reply({
41			embeds: [
42			new MessageEmbed()
43			.setColor('#f14a60')
44	rakin	125	.setDescription("Invalid member given.")
45	rakin	51	]
46			});
47
48			return;
49			}
50
51			if (options.options.getString('reason')) {
52			reason = await <string> options.options.getString('reason');
53			}
54			}
55			else {
56			try {
57			const user2 = await getMember((msg as Message), options);
58
59			if (!user2) {
60			throw new Error('Invalid user');
61			}
62
63			user = user2;
64			}
65			catch (e) {
66			await msg.reply({
67			embeds: [
68			new MessageEmbed()
69			.setColor('#f14a60')
70			.setDescription(`Invalid user given.`)
71			]
72			});
73
74			return;
75			}
76
77			console.log(user);
78
79			if (options.args[1]) {
80			const args = [...options.args];
81			args.shift();
82			reason = await args.join(' ');
83			}
84			}
85
86			try {
87	rakin	86	await Punishment.create({
88	rakin	197	type: PunishmentType.SHOT,
89	rakin	86	user_id: user.id,
90			guild_id: msg.guild!.id,
91			mod_id: msg.member!.user.id,
92			mod_tag: (msg.member!.user as User).tag,
93	rakin	336	reason,
94			createdAt: new Date()
95	rakin	86	});
96
97	rakin	197	// await History.create(user.id, msg.guild!, 'bean', msg.member!.user.id, typeof reason === 'undefined' ? null : reason);
98	rakin	51
99	rakin	189	try {
100			await user.send({
101			embeds: [
102			new MessageEmbed()
103			.setAuthor({
104			iconURL: <string> msg.guild!.iconURL(),
105	rakin	197	name: `\tYou got a shot in ${msg.guild!.name}`
106	rakin	189	})
107			.addFields([
108			{
109			name: "Reason",
110			value: typeof reason === 'undefined' ? 'No reason provided' : reason
111			}
112			])
113			]
114			});
115	rakin	160	}
116			catch (e) {
117			console.log(e);
118			dm = false;
119			}
120	rakin	51
121	rakin	197	// await client.logger.logBeaned(user, typeof reason === 'undefined' ? 'No reason provided' : reason, msg.member!.user as User);
122	rakin	51	}
123			catch (e) {
124			console.log(e);
125			}
126
127	rakin	197	await msg.reply({
128	rakin	51	embeds: [
129			new MessageEmbed()
130			.setAuthor({
131			name: user.user.tag,
132			iconURL: user.user.displayAvatarURL(),
133			})
134	rakin	197	.setDescription(user.user.tag + " got a shot." + (!dm ? "\nThey have DMs disabled. They will not know that they got a shot." : ''))
135	rakin	51	.addFields([
136			{
137	rakin	197	name: "Given by",
138	rakin	51	value: (msg.member!.user as User).tag
139			},
140			{
141			name: "Reason",
142			value: reason === undefined ? "No reason provided" : reason
143			}
144			])
145			]
146			});
147			}
148	rakin	160	}