import { ActivePunishment, ActivePunishmentType, BushTask, type BushGuild, type BushUser } from '#lib';
import assert from 'assert';
const { Op } = (await import('sequelize')).default;

export default class RemoveExpiredPunishmentsTask extends BushTask {
	public constructor() {
		super('removeExpiredPunishments', {
			delay: 15_000, // 15 seconds
			runOnStart: true
		});
	}

	public override async exec() {
		const expiredEntries = await ActivePunishment.findAll({
			where: {
				expires: {
					[Op.lt]: new Date() // Find all rows with an expiry date before now
				}
			}
		});

		void client.logger.verbose(
			`removeExpiredPunishments`,
			`Queried punishments, found <<${expiredEntries.length}>> expired punishments.`
		);

		for (const entry of expiredEntries) {
			const guild = client.guilds.cache.get(entry.guild) as BushGuild;
			const member = guild.members.cache.get(entry.user);
			const user = (await util.resolveNonCachedUser(entry.user)) as BushUser;

			if (!guild) continue;

			switch (entry.type) {
				case ActivePunishmentType.BAN: {
					assert(user);
					const result = await guild.bushUnban({ user: user, reason: 'Punishment expired.' });
					if (['success', 'user not banned'].includes(result)) await entry.destroy();
					else throw new Error(result);
					void client.logger.verbose(`removeExpiredPunishments`, `Unbanned ${entry.user}.`);
					break;
				}
				case ActivePunishmentType.BLOCK: {
					if (!member) {
						await entry.destroy(); // channel overrides are removed when the member leaves the guild
						continue;
					}
					const result = await member.bushUnblock({ reason: 'Punishment expired.', channel: entry.extraInfo });
					if (['success', 'user not blocked'].includes(result)) await entry.destroy();
					else throw new Error(result);
					void client.logger.verbose(`removeExpiredPunishments`, `Unblocked ${entry.user}.`);
					break;
				}
				case ActivePunishmentType.MUTE: {
					if (!member) continue;
					const result = await member.bushUnmute({ reason: 'Punishment expired.' });
					if (['success', 'failed to dm'].includes(result)) await entry.destroy();
					else throw new Error(result);
					void client.logger.verbose(`removeExpiredPunishments`, `Unmuted ${entry.user}.`);
					break;
				}
				case ActivePunishmentType.ROLE: {
					if (!member) continue;
					const role = guild?.roles?.cache?.get(entry.extraInfo);
					if (!role) throw new Error(`Cannot unmute ${member.user.tag} because I cannot find the mute role.`);
					const result = await member.bushRemoveRole({
						reason: 'Punishment expired.',
						role: role,
						addToModlog: true
					});

					if (['success', 'failed to dm'].includes(result)) await entry.destroy();
					else throw new Error(result);
					void client.logger.verbose(`removeExpiredPunishments`, `Removed a punishment role from ${entry.user}.`);
					break;
				}
			}
		}
	}
}