import {
	AllowedMentions,
	BotCommand,
	castDurationContent,
	emojis,
	format,
	Moderation,
	muteResponse,
	type ArgType,
	type CommandMessage,
	type MuteResponse,
	type OptArgType,
	type SlashMessage
} from '#lib';
import assert from 'assert/strict';
import { ApplicationCommandOptionType, type GuildMember } from 'discord.js';

export default class MuteCommand extends BotCommand {
	public constructor() {
		super('mute', {
			aliases: ['mute'],
			category: 'moderation',
			description: 'Mute a user.',
			usage: ['mute <member> [reasonAndDuration]'],
			examples: ['mute ironm00n 1 day commands in #general'],
			args: [
				{
					id: 'user',
					description: 'The user to mute.',
					type: 'user',
					prompt: 'What user would you like to mute?',
					retry: '{error} Choose a valid user to mute.',
					slashType: ApplicationCommandOptionType.User
				},
				{
					id: 'reason_and_duration',
					description: 'The reason and duration of the mute.',
					type: 'contentWithDuration',
					match: 'rest',
					prompt: 'Why should this user be muted and for how long?',
					retry: '{error} Choose a valid mute reason and duration.',
					optional: true,
					slashType: ApplicationCommandOptionType.String
				},
				{
					id: 'force',
					description: 'Override permission checks.',
					flag: '--force',
					match: 'flag',
					optional: true,
					slashType: false,
					only: 'text',
					ownerOnly: true
				}
			],
			slash: true,
			channel: 'guild',
			clientPermissions: ['ManageRoles'],
			userPermissions: ['ManageMessages']
		});
	}

	public override async exec(
		message: CommandMessage | SlashMessage,
		args: {
			user: ArgType<'user'>;
			reason_and_duration: OptArgType<'contentWithDuration'> | string;
			force: ArgType<'flag'>;
		}
	) {
		assert(message.inGuild());
		assert(message.member);

		const { duration, content } = await castDurationContent(args.reason_and_duration, message);

		const member = await message.guild.members.fetch(args.user.id).catch(() => null);
		if (!member)
			return await message.util.reply(`${emojis.error} The user you selected is not in the server or is not a valid user.`);

		const useForce = args.force && message.author.isOwner();
		const canModerateResponse = await Moderation.permissionCheck(message.member, member, 'mute', true, useForce);

		if (canModerateResponse !== true) {
			return message.util.reply(canModerateResponse);
		}

		const responseCode = await member.customMute({
			reason: content,
			moderator: message.member,
			duration
		});

		return await message.util.reply({
			content: MuteCommand.formatCode(this.client.utils.prefix(message), member, responseCode),
			allowedMentions: AllowedMentions.none()
		});
	}

	public static formatCode(prefix: string, member: GuildMember, code: MuteResponse): string {
		const victim = format.input(member.user.tag);
		switch (code) {
			case muteResponse.MISSING_PERMISSIONS:
				return `${emojis.error} Could not mute ${victim} because I am missing the **Manage Roles** permission.`;
			case muteResponse.NO_MUTE_ROLE:
				return `${emojis.error} Could not mute ${victim}, you must set a mute role with \`${prefix}config muteRole\`.`;
			case muteResponse.MUTE_ROLE_INVALID:
				return `${emojis.error} Could not mute ${victim} because the current mute role no longer exists. Please set a new mute role with \`${prefix}config muteRole\`.`;
			case muteResponse.MUTE_ROLE_NOT_MANAGEABLE:
				return `${emojis.error} Could not mute ${victim} because I cannot assign the current mute role, either change the role's position or set a new mute role with \`${prefix}config muteRole\`.`;
			case muteResponse.ACTION_ERROR:
				return `${emojis.error} Could not mute ${victim}, there was an error assigning them the mute role.`;
			case muteResponse.MODLOG_ERROR:
				return `${emojis.error} There was an error creating a modlog entry, please report this to my developers.`;
			case muteResponse.PUNISHMENT_ENTRY_ADD_ERROR:
				return `${emojis.error} There was an error creating a punishment entry, please report this to my developers.`;
			case muteResponse.DM_ERROR:
				return `${emojis.warn} Muted ${victim} however I could not send them a dm.`;
			case muteResponse.SUCCESS:
				return `${emojis.success} Successfully muted ${victim}.`;
			default:
				return `${emojis.error} An error occurred: ${format.input(code)}}`;
		}
	}
}