import { BushCommand, BushMessage, BushSlashMessage, Global } from '@lib';
import { ArgumentOptions, Flag } from 'discord-akairo';
import { User } from 'discord.js';

export default class SuperUserCommand extends BushCommand {
	public constructor() {
		super('superuser', {
			aliases: ['superuser', 'su'],
			category: 'dev',
			description: {
				content: 'A command to manage superusers.',
				usage: 'superuser <add/remove> <user>',
				examples: ['superuser add IRONM00N']
			},
			clientPermissions: ['SEND_MESSAGES'],
			ownerOnly: true
		});
	}

	override *args(): IterableIterator<ArgumentOptions | Flag> {
		const action = yield {
			id: 'action',
			type: ['add', 'remove'],
			prompt: {
				start: 'Would you like to `add` or `remove` a user from the superuser list?',
				retry: '{error} Choose if you would like to `add` or `remove` a user.',
				optional: false
			}
		};
		const user = yield {
			id: 'user',
			type: 'user',
			match: 'restContent',
			prompt: {
				start: `Who would you like to ${action ?? 'add/remove'} from the superuser list?`,
				retry: `Choose a valid user to ${action ?? 'add/remove'} from the superuser list.`,
				optional: false
			}
		};
		return { action, user };
	}

	public override async exec(
		message: BushMessage | BushSlashMessage,
		{ action, user }: { action: 'add' | 'remove'; user: User }
	): Promise<unknown> {
		if (!message.author.isOwner())
			return await message.util.reply(`${util.emojis.error} Only my developers can run this command.`);

		const superUsers: string[] = (await Global.findByPk(client.config.environment))?.superUsers ?? [];

		if (action === 'add' ? superUsers.includes(user.id) : !superUsers.includes(user.id))
			return message.util.reply(`${util.emojis.warn} \`${user.tag}\` is ${action === 'add' ? 'already' : 'not'} a superuser.`);

		const success = await util.insertOrRemoveFromGlobal(action, 'superUsers', user.id).catch(() => false);

		if (success) {
			return await message.util.reply(
				`${util.emojis.success} ${action == 'remove' ? '' : 'made'} \`${user.tag}\` ${
					action == 'remove' ? 'is no longer ' : ''
				}a superuser.`
			);
		} else {
			return await message.util.reply(
				`${util.emojis.error} There was an error ${action == 'remove' ? `removing` : 'making'} \`${user.tag}\` ${
					action == 'remove' ? `from` : 'to'
				} the superuser list.`
			);
		}
	}
}