join roles, sticky roles, join messages, support threads etc

5 files changed, 208 insertions, 3 deletions

diff --git a/src/listeners/guild/guildCreate.ts b/src/listeners/guild/guildCreate.ts
new file mode 100644
index 0000000..21a7ab0
--- /dev/null
+++ b/src/listeners/guild/guildCreate.ts
@@ -0,0 +1,18 @@
+import { BushListener, Guild } from '../../lib';
+import { BushClientEvents } from '../../lib/extensions/discord.js/BushClientEvents';
+
+export default class GuildCreateListener extends BushListener {
+	public constructor() {
+		super('guildCreate', {
+			emitter: 'client',
+			event: 'guildCreate', // when the bot joins a guild
+			category: 'client'
+		});
+	}
+
+	public override async exec(...[guild]: BushClientEvents['guildCreate']): Promise<void> {
+		void client.console.info('JoinGuild', `Joined <<${guild.name}>> with <<${guild.memberCount?.toLocaleString()}>> members.`);
+		const g = await Guild.findByPk(guild.id);
+		if (!g) void Guild.create({ id: guild.id });
+	}
+}
diff --git a/src/listeners/guild/guildDelete.ts b/src/listeners/guild/guildDelete.ts
new file mode 100644
index 0000000..a59f45e
--- /dev/null
+++ b/src/listeners/guild/guildDelete.ts
@@ -0,0 +1,16 @@
+import { BushListener } from '../../lib';
+import { BushClientEvents } from '../../lib/extensions/discord.js/BushClientEvents';
+
+export default class GuildDeleteListener extends BushListener {
+	public constructor() {
+		super('guildDelete', {
+			emitter: 'client',
+			event: 'guildDelete', //when the bot leaves a guild
+			category: 'client'
+		});
+	}
+
+	public override exec(...[guild]: BushClientEvents['guildDelete']): void {
+		void client.console.info('LeaveGuild', `Left <<${guild.name}>> with <<${guild.memberCount?.toLocaleString()}>> members.`);
+	}
+}
diff --git a/src/listeners/guild/guildMemberAdd.ts b/src/listeners/guild/guildMemberAdd.ts
new file mode 100644
index 0000000..bf6e0b6
--- /dev/null
+++ b/src/listeners/guild/guildMemberAdd.ts
@@ -0,0 +1,103 @@
+import { MessageEmbed, Snowflake, Util } from 'discord.js';
+import { BushGuildMember, BushListener, BushTextChannel, StickyRole } from '../../lib';
+import { BushClientEvents } from '../../lib/extensions/discord.js/BushClientEvents';
+
+export default class GuildMemberAddListener extends BushListener {
+	public constructor() {
+		super('guildMemberAdd', {
+			emitter: 'client',
+			event: 'guildMemberAdd',
+			category: 'client'
+		});
+	}
+
+	public override async exec(...[member]: BushClientEvents['guildMemberAdd']): Promise<void> {
+		void this.sendWelcomeMessage(member);
+		void this.joinAndStickyRoles(member);
+	}
+
+	public async sendWelcomeMessage(member: BushGuildMember): Promise<void> {
+		if (client.config.isDevelopment) return;
+		const welcomeChannel = await member.guild.getSetting('welcomeChannel');
+		if (!welcomeChannel) return;
+		const welcome = this.client.channels.cache.get(welcomeChannel) as BushTextChannel | undefined;
+		if (!welcome) return;
+		if (member.guild.id !== welcome?.guild.id) throw new Error('Welcome channel must be in the guild.');
+		const embed = new MessageEmbed()
+			.setDescription(
+				`${this.client.util.emojis.join} **${Util.escapeMarkdown(
+					member.user.tag
+				)}** joined the server. There are now ${member.guild.memberCount.toLocaleString()} members.`
+			)
+			.setColor(this.client.util.colors.green);
+		await welcome
+			.send({ embeds: [embed] })
+			.then(() => this.client.console.info('OnJoin', `Sent a message for <<${member.user.tag}>> in <<${member.guild.name}>>.`))
+			.catch(() =>
+				this.client.console.warn('OnJoin', `Failed to send message for <<${member.user.tag}>> in <<${member.guild.name}>>.`)
+			);
+	}
+
+	public async joinAndStickyRoles(member: BushGuildMember): Promise<void> {
+		if (client.config.isDevelopment) return;
+		if (await member.guild.hasFeature('stickyRoles')) {
+			const hadRoles = await StickyRole.findOne({ where: { guild: member.guild.id, user: member.id } });
+			if (hadRoles?.roles?.length) {
+				const rolesArray = hadRoles.roles
+					.map((roleID: Snowflake) => {
+						const role = member.guild.roles.cache.get(roleID);
+						if (role && !member.roles.cache.has(roleID)) {
+							if (role.name !== '@everyone' || !role.managed) return role.id;
+						}
+					})
+					.filter((role) => role) as Snowflake[];
+				if (hadRoles.nickname && member.manageable) {
+					void member.setNickname(hadRoles.nickname).catch(() => {});
+				}
+				if (rolesArray?.length) {
+					const addedRoles = await member.roles
+						.add(rolesArray, "Returning member's previous roles.")
+						.catch(
+							() => void this.client.console.warn('ReturnRoles', `There was an error returning <<${member.user.tag}>>'s roles.`)
+						);
+					if (addedRoles) {
+						void this.client.console.info(
+							'RoleData',
+							`Assigned sticky roles to <<${member.user.tag}>> in <<${member.guild.name}>>.`
+						);
+					} else if (!addedRoles) {
+						const failedRoles: string[] = [];
+						for (let i = 0; i < rolesArray.length; i++) {
+							await member.roles
+								.add(rolesArray[i], "[Fallback] Returning member's previous roles.")
+								.catch(() => failedRoles.push(rolesArray[i]));
+						}
+						if (failedRoles.length) {
+							void this.client.console.warn('RoleData', 'Failed assigning the following roles on Fallback:' + failedRoles);
+						} else {
+							void this.client.console.info(
+								'RoleData',
+								`[Fallback] Assigned sticky roles to <<${member.user.tag}>> in <<${member.guild.name}>>.`
+							);
+						}
+					}
+				}
+			}
+		} else {
+			const joinRoles = await member.guild.getSetting('joinRoles');
+			if (!joinRoles) return;
+			await member.roles
+				.add(joinRoles, 'Join roles.')
+				.then(() =>
+					this.client.console.info('RoleData', `Assigned join roles to <<${member.user.tag}>> in <<${member.guild.name}>>.`)
+				)
+				.catch(
+					() =>
+						void this.client.console.warn(
+							'OnJoin',
+							`Failed to assign join roles to <<${member.user.tag}>>, in <<${member.guild.name}>>.`
+						)
+				);
+		}
+	}
+}
diff --git a/src/listeners/guild/guildMemberRemove.ts b/src/listeners/guild/guildMemberRemove.ts
new file mode 100644
index 0000000..f25108f
--- /dev/null
+++ b/src/listeners/guild/guildMemberRemove.ts
@@ -0,0 +1,67 @@
+import { MessageEmbed, Util } from 'discord.js';
+import { BushGuildMember, BushListener, BushTextChannel, PartialBushGuildMember, StickyRole } from '../../lib';
+import { BushClientEvents } from '../../lib/extensions/discord.js/BushClientEvents';
+
+export default class GuildMemberRemoveListener extends BushListener {
+	public constructor() {
+		super('guildMemberRemove', {
+			emitter: 'client',
+			event: 'guildMemberRemove',
+			category: 'guild'
+		});
+	}
+
+	public override async exec(...[member]: BushClientEvents['guildMemberRemove']): Promise<void> {
+		void this.sendWelcomeMessage(member);
+		void this.stickyRoles(member);
+	}
+
+	public async sendWelcomeMessage(member: BushGuildMember | PartialBushGuildMember): Promise<void> {
+		if (client.config.isDevelopment) return;
+		const user = member.partial ? await client.users.fetch(member.id) : member.user;
+		await util.sleep(0.05); // ban usually triggers after member leave
+		const isBan = member.guild.bans.cache.has(member.id);
+		const welcomeChannel = await member.guild.getSetting('welcomeChannel');
+		if (!welcomeChannel) return;
+		const welcome = this.client.channels.cache.get(welcomeChannel) as BushTextChannel | undefined;
+		if (member.guild.id !== welcome?.guild.id) throw new Error('Welcome channel must be in the guild.');
+		const embed: MessageEmbed = new MessageEmbed()
+			.setDescription(
+				`${this.client.util.emojis.leave} **${Util.escapeBold(user.tag)}** ${
+					isBan ? 'banned from' : 'left'
+				} the server. There are now ${welcome.guild.memberCount.toLocaleString()} members.`
+			)
+			.setColor(isBan ? util.colors.orange : util.colors.red);
+		welcome
+			.send({ embeds: [embed] })
+			.then(() => client.console.info('OnLeave', `Sent a message for <<${user.tag}>> in <<${member.guild.name}>>.`))
+			.catch(() =>
+				this.client.console.warn('OnLeave', `Failed to send message for <<${user.tag}>> in <<${member.guild.name}>>.`)
+			);
+	}
+
+	public async stickyRoles(member: BushGuildMember | PartialBushGuildMember): Promise<void> {
+		if (!(await member.guild.hasFeature('stickyRoles'))) return;
+		if (member.partial) throw new Error('Partial member, cannot save sticky roles.');
+		const rolesArray = member.roles.cache.filter((role) => role.name !== '@everyone').map((role) => role.id);
+		const nickname = member.nickname;
+		if (rolesArray) {
+			const [row, isNew] = await StickyRole.findOrBuild({
+				where: {
+					user: member.user.id,
+					guild: member.guild.id
+				},
+				defaults: {
+					user: member.user.id,
+					guild: member.guild.id,
+					roles: rolesArray
+				}
+			});
+			row.roles = rolesArray;
+			if (nickname) row.nickname = nickname;
+			await row
+				.save()
+				.then(() => this.client.console.info('RoleData', `${isNew ? 'Created' : 'Updated'} info for <<${member.user.tag}>>.`));
+		}
+	}
+}
diff --git a/src/listeners/guild/syncUnban.ts b/src/listeners/guild/syncUnban.ts
index b1e4fd9..9a6a607 100644
--- a/src/listeners/guild/syncUnban.ts
+++ b/src/listeners/guild/syncUnban.ts
@@ -1,15 +1,16 @@
 import { ActivePunishment, ActivePunishmentType, BushListener } from '@lib';
-import { ClientEvents } from 'discord.js';
+import { BushClientEvents } from '../../lib/extensions/discord.js/BushClientEvents';
 
 export default class SyncUnbanListener extends BushListener {
 	public constructor() {
 		super('guildBanRemove', {
 			emitter: 'client',
-			event: 'guildBanRemove'
+			event: 'guildBanRemove',
+			category: 'guild'
 		});
 	}
 
-	public override async exec(...[ban]: ClientEvents['guildBanRemove']): Promise<void> {
+	public override async exec(...[ban]: BushClientEvents['guildBanRemove']): Promise<void> {
 		const bans = await ActivePunishment.findAll({
 			where: {
 				user: ban.user.id,