import { REST } from '@discordjs/rest';
import { Routes } from 'discord-api-types/v9';
import { Client } from 'discord.js';
import { readdir } from 'fs/promises';
import { removeExtension } from '../utils/misc';

/** Load all the commands. */
export default async (client: Client) => {
	const rest = new REST({ version: '10' }).setToken(client.token ?? '');

	const command_categories = (await readdir(__dirname))
		.filter(element => !element.endsWith('.js') && !element.endsWith('.ts'));

	const commands = (
		await Promise.all(
			// For each categorie
			command_categories.map(async command_category => {
				// Retrieve all the commands
				const command_files = await readdir(`${__dirname}/${command_category}`);

				// Add the category to the collection for the help command
				client.commands.categories.set(
					command_category,
					command_files.map(removeExtension),
				);

				// Add the command
				return Promise.all(
					command_files.map(async command_file => {
						const command = (
							await import(`../commands/${command_category}/${command_file}`)
						).default;

						// Add it to the collection so the interaction will work
						command.data = command.data(client);
						client.commands.list.set(command.data.name, command);

						return command;
					}),
				);
			}),
		)
	).flat(2);

	// Send guilds commands to Discord
	const scopedCommands = new Map<string, unknown[]>();

	// Add commands to guild where the bot is
	const allowedGuilds = client.guilds.cache;

	// Assign each commands to the guilds
	commands.filter((c) => c.scope().length > 0).forEach((c) => {
		c.scope().forEach((guild: string) => {
			if (allowedGuilds.get(guild) !== undefined) {
				const guildCommands = scopedCommands.get(guild);
				if (guildCommands === undefined) {
					scopedCommands.set(guild, [c.data.toJSON()]);
				} else {
					guildCommands.push(c.data.toJSON());
					scopedCommands.set(guild, guildCommands);
				}
			}
		});
	});

	scopedCommands
		.forEach(async (command, guild) => await rest.put(
			Routes.applicationGuildCommands(client.user?.id as string, guild), {
				body: command,
			}));

	// Send global commands to Discord
	const globalCommands = commands.filter((c) => c.scope().length == 0);
	return await rest.put(Routes.applicationCommands(client.user?.id as string), {
		body: globalCommands.map((c) => c.data.toJSON()),
	});
};