legit just copy utilibot v2 code

4 files changed, 406 insertions, 0 deletions

diff --git a/src/commands/moderation/BanCommand.ts b/src/commands/moderation/BanCommand.ts
new file mode 100644
index 0000000..300101b
--- /dev/null
+++ b/src/commands/moderation/BanCommand.ts
@@ -0,0 +1,137 @@
+import { User } from 'discord.js';
+import { BotCommand } from '../../lib/extensions/BotCommand';
+import { BotMessage } from '../../lib/extensions/BotMessage';
+import { Ban, Modlog, ModlogType } from '../../lib/types/Models';
+import moment from 'moment';
+
+const durationAliases: Record<string, string[]> = {
+	weeks: ['w', 'weeks', 'week', 'wk', 'wks'],
+	days: ['d', 'days', 'day'],
+	hours: ['h', 'hours', 'hour', 'hr', 'hrs'],
+	minutes: ['m', 'min', 'mins', 'minutes', 'minute'],
+	months: ['mo', 'month', 'months']
+};
+const durationRegex = /(?:(\d+)(d(?:ays?)?|h(?:ours?|rs?)?|m(?:inutes?|ins?)?|mo(?:nths?)?|w(?:eeks?|ks?)?)(?: |$))/g;
+
+export default class PrefixCommand extends BotCommand {
+	constructor() {
+		super('ban', {
+			aliases: ['ban'],
+			args: [
+				{
+					id: 'user',
+					type: 'user',
+					prompt: {
+						start: 'What user would you like to ban?',
+						retry: 'Invalid response. What user would you like to ban?'
+					}
+				},
+				{
+					id: 'reason'
+				},
+				{
+					id: 'time',
+					match: 'option',
+					flag: '--time'
+				}
+			],
+			clientPermissions: ['BAN_MEMBERS'],
+			userPermissions: ['BAN_MEMBERS']
+		});
+	}
+	async exec(
+		message: BotMessage,
+		{ user, reason, time }: { user: User; reason?: string; time?: string }
+	): Promise<void> {
+		const duration = moment.duration();
+		let modlogEnry: Modlog;
+		let banEntry: Ban;
+		const translatedTime: string[] = [];
+		try {
+			try {
+				if (time) {
+					const parsed = [...time.matchAll(durationRegex)];
+					if (parsed.length < 1) {
+						await message.util.send('Invalid time.');
+						return;
+					}
+					for (const part of parsed) {
+						const translated = Object.keys(durationAliases).find((k) =>
+							durationAliases[k].includes(part[2])
+						);
+						translatedTime.push(part[1] + ' ' + translated);
+						duration.add(
+							Number(part[1]),
+							translated as 'weeks' | 'days' | 'hours' | 'months' | 'minutes'
+						);
+					}
+					modlogEnry = Modlog.build({
+						user: user.id,
+						guild: message.guild.id,
+						reason,
+						type: ModlogType.TEMPBAN,
+						duration: duration.asMilliseconds(),
+						moderator: message.author.id
+					});
+					banEntry = Ban.build({
+						user: user.id,
+						guild: message.guild.id,
+						reason,
+						expires: new Date(new Date().getTime() + duration.asMilliseconds()),
+						modlog: modlogEnry.id
+					});
+				} else {
+					modlogEnry = Modlog.build({
+						user: user.id,
+						guild: message.guild.id,
+						reason,
+						type: ModlogType.BAN,
+						moderator: message.author.id
+					});
+					banEntry = Ban.build({
+						user: user.id,
+						guild: message.guild.id,
+						reason,
+						modlog: modlogEnry.id
+					});
+				}
+				await modlogEnry.save();
+				await banEntry.save();
+			} catch (e) {
+				console.error(e);
+				await message.util.send(
+					'Error saving to database. Please report this to a developer.'
+				);
+				return;
+			}
+			try {
+				await user.send(
+					`You were banned in ${message.guild.name} ${
+						translatedTime.length >= 1
+							? `for ${translatedTime.join(', ')}`
+							: 'permanently'
+					} with reason \`${reason || 'No reason given'}\``
+				);
+			} catch (e) {
+				await message.channel.send('Error sending message to user');
+			}
+			await message.guild.members.ban(user, {
+				reason: `Banned by ${message.author.tag} with ${
+					reason ? `reason ${reason}` : 'no reason'
+				}`
+			});
+			await message.util.send(
+				`Banned <@!${user.id}> ${
+					translatedTime.length >= 1
+						? `for ${translatedTime.join(', ')}`
+						: 'permanently'
+				} with reason \`${reason || 'No reason given'}\``
+			);
+		} catch {
+			await message.util.send('Error banning :/');
+			await modlogEnry.destroy();
+			await banEntry.destroy();
+			return;
+		}
+	}
+}
diff --git a/src/commands/moderation/KickCommand.ts b/src/commands/moderation/KickCommand.ts
new file mode 100644
index 0000000..0dc4276
--- /dev/null
+++ b/src/commands/moderation/KickCommand.ts
@@ -0,0 +1,72 @@
+import { BotCommand } from '../../lib/extensions/BotCommand';
+import { BotMessage } from '../../lib/extensions/BotMessage';
+import { Modlog, ModlogType } from '../../lib/types/Models';
+import { GuildMember } from 'discord.js';
+
+export default class PrefixCommand extends BotCommand {
+	constructor() {
+		super('kick', {
+			aliases: ['kick'],
+			args: [
+				{
+					id: 'user',
+					type: 'member',
+					prompt: {
+						start: 'What user would you like to kick?',
+						retry: 'Invalid response. What user would you like to kick?'
+					}
+				},
+				{
+					id: 'reason'
+				}
+			],
+			clientPermissions: ['KICK_MEMBERS'],
+			userPermissions: ['KICK_MEMBERS']
+		});
+	}
+	async exec(
+		message: BotMessage,
+		{ user, reason }: { user: GuildMember; reason?: string }
+	): Promise<void> {
+		let modlogEnry: Modlog;
+		try {
+			modlogEnry = Modlog.build({
+				user: user.id,
+				guild: message.guild.id,
+				moderator: message.author.id,
+				type: ModlogType.KICK,
+				reason
+			});
+			await modlogEnry.save();
+		} catch (e) {
+			console.error(e);
+			await message.util.send(
+				'Error saving to database. Please report this to a developer.'
+			);
+			return;
+		}
+		try {
+			await user.send(
+				`You were kicked in ${message.guild.name} with reason \`${
+					reason || 'No reason given'
+				}\``
+			);
+		} catch (e) {
+			await message.channel.send('Error sending message to user');
+		}
+		try {
+			await user.kick(
+				`Kicked by ${message.author.tag} with ${
+					reason ? `reason ${reason}` : 'no reason'
+				}`
+			);
+		} catch {
+			await message.util.send('Error kicking :/');
+			await modlogEnry.destroy();
+			return;
+		}
+		await message.util.send(
+			`Kicked <@!${user.id}> with reason \`${reason || 'No reason given'}\``
+		);
+	}
+}
diff --git a/src/commands/moderation/ModlogCommand.ts b/src/commands/moderation/ModlogCommand.ts
new file mode 100644
index 0000000..ea35198
--- /dev/null
+++ b/src/commands/moderation/ModlogCommand.ts
@@ -0,0 +1,143 @@
+import { BotCommand } from '../../lib/extensions/BotCommand';
+import { Message } from 'discord.js';
+import { Modlog } from '../../lib/types/Models';
+import { MessageEmbed } from 'discord.js';
+import moment from 'moment';
+import { stripIndent } from 'common-tags';
+import { Argument } from 'discord-akairo';
+
+export default class ModlogCommand extends BotCommand {
+	constructor() {
+		super('modlog', {
+			aliases: ['modlog', 'modlogs'],
+			args: [
+				{
+					id: 'search',
+					prompt: {
+						start: 'What modlog id or user would you like to see?'
+					}
+				},
+				{
+					id: 'page',
+					type: 'number'
+				}
+			],
+			userPermissions: ['MANAGE_MESSAGES']
+		});
+	}
+	*args(): unknown {
+		const search = yield {
+			id: 'search',
+			type: Argument.union('user', 'string'),
+			prompt: {
+				start: 'What modlog id or user would you like to see?'
+			}
+		};
+		if (typeof search === 'string') return { search, page: null };
+		else {
+			const page = yield {
+				id: 'page',
+				type: 'number',
+				prompt: {
+					start: 'What page?',
+					retry: 'Not a number. What page?',
+					optional: true
+				}
+			};
+			return { search, page };
+		}
+	}
+	async exec(
+		message: Message,
+		{ search, page }: { search: string; page: number }
+	): Promise<void> {
+		const foundUser = await this.client.util.resolveUserAsync(search);
+		if (foundUser) {
+			const logs = await Modlog.findAll({
+				where: {
+					guild: message.guild.id,
+					user: foundUser.id
+				},
+				order: [['createdAt', 'ASC']]
+			});
+			const niceLogs: string[] = [];
+			for (const log of logs) {
+				niceLogs.push(stripIndent`
+					ID: ${log.id}
+					Type: ${log.type.toLowerCase()}
+					User: <@!${log.user}> (${log.user})
+					Moderator: <@!${log.moderator}> (${log.moderator})
+					Duration: ${
+						log.duration
+							? moment.duration(log.duration, 'milliseconds').humanize()
+							: 'N/A'
+					}
+					Reason: ${log.reason || 'None given'}
+					${this.client.util.ordinal(logs.indexOf(log) + 1)} action
+				`);
+			}
+			const chunked: string[][] = this.client.util.chunk(niceLogs, 3);
+			const embedPages = chunked.map(
+				(e, i) =>
+					new MessageEmbed({
+						title: `Modlogs page ${i + 1}`,
+						description: e.join(
+							'\n-------------------------------------------------------\n'
+						),
+						footer: {
+							text: `Page ${i + 1}/${chunked.length}`
+						}
+					})
+			);
+			if (page) {
+				await message.util.send(embedPages[page - 1]);
+				return;
+			} else {
+				await message.util.send(embedPages[0]);
+				return;
+			}
+		} else if (search) {
+			const entry = await Modlog.findByPk(search);
+			if (!entry) {
+				await message.util.send('That modlog does not exist.');
+				return;
+			}
+			await message.util.send(
+				new MessageEmbed({
+					title: `Modlog ${entry.id}`,
+					fields: [
+						{
+							name: 'Type',
+							value: entry.type.toLowerCase(),
+							inline: true
+						},
+						{
+							name: 'Duration',
+							value: `${
+								entry.duration
+									? moment.duration(entry.duration, 'milliseconds').humanize()
+									: 'N/A'
+							}`,
+							inline: true
+						},
+						{
+							name: 'Reason',
+							value: `${entry.reason || 'None given'}`,
+							inline: true
+						},
+						{
+							name: 'Moderator',
+							value: `<@!${entry.moderator}> (${entry.moderator})`,
+							inline: true
+						},
+						{
+							name: 'User',
+							value: `<@!${entry.user}> (${entry.user})`,
+							inline: true
+						}
+					]
+				})
+			);
+		}
+	}
+}
diff --git a/src/commands/moderation/WarnCommand.ts b/src/commands/moderation/WarnCommand.ts
new file mode 100644
index 0000000..676615d
--- /dev/null
+++ b/src/commands/moderation/WarnCommand.ts
@@ -0,0 +1,54 @@
+import { GuildMember } from 'discord.js';
+import { BotCommand } from '../../lib/extensions/BotCommand';
+import { BotMessage } from '../../lib/extensions/BotMessage';
+import { Modlog, ModlogType } from '../../lib/types/Models';
+
+export default class WarnCommand extends BotCommand {
+	public constructor() {
+		super('warn', {
+			aliases: ['warn'],
+			userPermissions: ['MANAGE_MESSAGES'],
+			args: [
+				{
+					id: 'member',
+					type: 'member'
+				},
+				{
+					id: 'reason',
+					match: 'rest'
+				}
+			]
+		});
+	}
+	public async exec(
+		message: BotMessage,
+		{ member, reason }: { member: GuildMember; reason: string }
+	): Promise<void> {
+		try {
+			const entry = Modlog.build({
+				user: member.id,
+				guild: message.guild.id,
+				moderator: message.author.id,
+				type: ModlogType.WARN,
+				reason
+			});
+			await entry.save();
+		} catch (e) {
+			await message.util.send(
+				'Error saving to database, please contact the developers'
+			);
+			return;
+		}
+		try {
+			await member.send(
+				`You were warned in ${message.guild.name} for reason "${reason}".`
+			);
+		} catch (e) {
+			await message.util.send('Error messaging user, warning still saved.');
+			return;
+		}
+		await message.util.send(
+			`${member.user.tag} was warned for reason "${reason}".`
+		);
+	}
+}