import {
	BotCommand,
	clientSendAndPermCheck,
	colors,
	emojis,
	type ArgType,
	type CommandMessage,
	type OptArgType,
	type SlashMessage
} from '#lib';
import assert from 'assert/strict';
import { ApplicationCommandOptionType, EmbedBuilder } from 'discord.js';
import { VM } from 'vm2';
assert(VM);

export default class JavascriptCommand extends BotCommand {
	public constructor() {
		super('javascript', {
			aliases: ['javascript', 'js'],
			category: 'dev',
			description: 'Evaluate code in a sand boxed environment.',
			usage: ['javascript <code> [--depth #]'],
			examples: ['javascript 9+10'],
			args: [
				{
					id: 'code',
					description: 'The code you would like to run in a sand boxed environment.',
					match: 'rest',
					prompt: 'What code would you like to run in a sand boxed environment?',
					retry: '{error} Invalid code to run in a sand boxed environment.',
					slashType: ApplicationCommandOptionType.String
				},
				{
					id: 'sel_depth',
					description: 'How deep to inspect the output.',
					match: 'option',
					type: 'integer',
					flag: '--depth',
					default: 0,
					prompt: 'How deep would you like to inspect the output?',
					slashType: ApplicationCommandOptionType.Integer,
					optional: true
				}
			],
			slash: true,
			superUserOnly: true,
			clientPermissions: (m) => clientSendAndPermCheck(m),
			userPermissions: []
		});
	}

	public override async exec(
		message: CommandMessage | SlashMessage,
		args: { code: ArgType<'string'>; sel_depth: OptArgType<'integer'> }
	) {
		if (!message.author.isSuperUser()) return await message.util.reply(`${emojis.error} Only super users can run this command.`);
		if (message.util.isSlashMessage(message)) {
			await message.interaction.deferReply({ ephemeral: false });
		}
		const code = args.code.replace(/[“”]/g, '"').replace(/```*(?:js)?/g, '');
		const embed = new EmbedBuilder();
		const input = await this.client.utils.inspectCleanRedactCodeblock(code, 'js');

		try {
			const rawOutput = /^(9\s*?\+\s*?10)|(10\s*?\+\s*?9)$/.test(code)
				? '21'
				: new VM({ eval: true, wasm: true, timeout: 1_000, fixAsync: true }).run(`${code}`);
			const output = await this.client.utils.inspectCleanRedactCodeblock(rawOutput, 'js', {
				depth: args.sel_depth ?? 0,
				getters: true,
				inspectStrings: true,
				colors: false
			});

			embed.setTitle(`${emojis.successFull} Successfully Evaluated Expression`).setColor(colors.success);
			embed.addFields({ name: '📥 Input', value: input }, { name: '📤 Output', value: output });
		} catch (e) {
			embed.setTitle(`${emojis.errorFull} Unable to Evaluate Expression`).setColor(colors.error);
			embed.addFields(
				{ name: '📥 Input', value: input },
				{ name: '📤 Error', value: await this.client.utils.inspectCleanRedactCodeblock(e, 'js', { colors: false }) }
			);
		}

		embed.setTimestamp().setFooter({ text: message.author.tag, iconURL: message.author.displayAvatarURL() ?? undefined });

		await message.util.reply({ embeds: [embed] });
	}
}