xiao/framework/Registry.js

const { Collection } = require('@discordjs/collection');
const fs = require('fs');
const path = require('path');
const Group = require('./Group');

module.exports = class Registry {
	constructor(client) {
		Object.defineProperty(this, 'client', { value: client });

		this.commands = new Collection();
		this.groups = new Collection();
		this.types = new Collection();
	}

	findCommands(query) {
		query = query.toLowerCase();
		return this.commands.filter(command => command.name === query || command.aliases.includes(query));
	}

	registerCommand(command) {
		const nameReplaced = command.name.replace(/-/g, '');
		if (nameReplaced !== command.name) command.aliases.push(nameReplaced);
		for (const alias of command.aliases) {
			const replaced = alias.replace(/-/g, '');
			if (replaced === alias) continue;
			command.aliases.push(replaced);
		}
		this.commands.set(command.name, command);
		return this;
	}

	registerCommandsIn(dir) {
		const groups = fs.readdirSync(dir);
		for (const group of groups) {
			const commands = fs.readdirSync(path.join(dir, group));
			for (const command of commands) {
				if (!command.endsWith('.js')) continue;
				const Required = require(path.join(dir, group, command));
				this.registerCommand(new Required(this.client));
			}
		}
		return this;
	}

	findGroups(query) {
		query = query.toLowerCase();
		return this.groups.filter(group => group.id === query || group.name.toLowerCase() === query);
	}

	registerGroup(group) {
		this.groups.set(group.id, group);
		return this;
	}

	registerGroups(groups) {
		for (const [id, name] of groups) {
			const group = new Group(this.client, id, name);
			this.registerGroup(group);
		}
		return this;
	}

	registerType(type) {
		this.types.set(type.id, type);
		return this;
	}

	registerTypesIn(dir) {
		const types = fs.readdirSync(dir);
		for (const type of types) {
			if (!type.endsWith('.js')) continue;
			const Required = require(path.join(dir, type));
			this.registerType(new Required(this.client));
		}
		return this;
	}

	registerDefaultTypes() {
		return this.registerTypesIn(path.join(__dirname, 'types'));
	}

	get totalUses() {
		let uses = 0;
		for (const command of this.commands.values()) {
			if (command.ownerOnly || command.unknown || command.uses === undefined) continue;
			uses += command.uses;
		}
		return uses;
	}
};