SupportBot/Structures/Client.js

const fs = require("fs");
const path = require("path");
const Discord = require("discord.js");
const { GatewayIntentBits, Partials } = require("discord.js");
const { Command, Event } = require("./Addon.js");
const configStore = require("./ConfigStore.js");
const { getPrimaryGuild } = require("./GuildManager.js");

const supportbot = configStore.supportbot;
const cmdconfig = configStore.commands;

const c = {
  reset: "\x1b[0m",
  gray: "\x1b[90m",
  green: "\x1b[32m",
  yellow: "\x1b[33m",
  cyan: "\x1b[36m",
  red: "\x1b[31m",
  white: "\x1b[37m",
  bold: "\x1b[1m",
};

function line() {
  console.log(`${c.gray}────────────────────────────────────────────────────────${c.reset}`);
}

function section(title) {
  console.log(`\n${c.green}${title}${c.reset}`);
  console.log(`${c.gray}${"-".repeat(title.length)}${c.reset}`);
}

function walk(dir, ext = ".js", fileList = []) {
  const files = fs.readdirSync(dir);
  for (const file of files) {
    const full = `${dir}/${file}`;
    const stat = fs.statSync(full);
    if (stat.isDirectory()) {
      walk(full, ext, fileList);
    } else if (file.endsWith(ext)) {
      fileList.push(full);
    }
  }
  return fileList;
}

class Client extends Discord.Client {
  constructor() {
    super({
      intents: [
        GatewayIntentBits.Guilds,
        GatewayIntentBits.GuildMembers,
        GatewayIntentBits.GuildMessages,
        GatewayIntentBits.GuildMessageReactions,
        GatewayIntentBits.GuildEmojisAndStickers,
        GatewayIntentBits.MessageContent,
      ],
      partials: [Partials.Message, Partials.Channel, Partials.Reaction],
    });

    this.commands = new Discord.Collection();
  }

  async getChannel(channel, guild) {
    if (!channel) return null;
    const channelStr = String(channel).toLowerCase();
    return guild.channels.cache.find(
      (c) =>
        (c.type === Discord.ChannelType.GuildText || c.type === Discord.ChannelType.GuildNews) &&
        (c.id === channel || (c.name && c.name.toLowerCase() === channelStr)),
    );
  }

  async getRole(role, guild) {
    if (!role) return null;
    const roleStr = String(role).toLowerCase();
    return guild.roles.cache.find(
      (r) => r.id === role || (r.name && r.name.toLowerCase() === roleStr),
    );
  }

  async getCategory(category, guild) {
    if (!category) return null;
    const catStr = String(category).toLowerCase();
    return guild.channels.cache.find(
      (c) =>
        c.type === Discord.ChannelType.GuildCategory &&
        (c.id === category || (c.name && c.name.toLowerCase() === catStr)),
    );
  }

  async start(token) {
    const commandFiles = walk("./Commands");

    const allCommands = commandFiles.map((file) => ({
      file,
      command: require(`../${file}`),
    }));

    const enabledCommands = allCommands.filter(({ command }) => {
      const matchingConfig = Object.values(cmdconfig).find(
        (entry) =>
          entry &&
          typeof entry === "object" &&
          entry.Command &&
          String(entry.Command).toLowerCase() === String(command.name).toLowerCase(),
      );

      if (!matchingConfig) {
        return true;
      }

      return matchingConfig.Enabled !== false;
    });

    section("Commands");

    enabledCommands.forEach(({ command, file }) => {
      console.log(
        `${c.green}✓${c.reset} ${c.white}${command.name}${c.reset} ${c.gray}(${path.basename(file)})${c.reset}`,
      );
      this.commands.set(command.name, command);
    });

    if (supportbot.General.Addons.Enabled) {
      const addonFiles = fs.readdirSync("./Addons").filter((file) => file.endsWith(".js"));

      section("Addons");

      addonFiles.forEach((file) => {
        let addon = require(`../Addons/${file}`);
        
        // Handle single or multiple addons per file
        const addonArray = Array.isArray(addon) ? addon : [addon];
        
        addonArray.forEach((item, index) => {
          // If name is missing, use filename (handling index for arrays)
          if (!item.name) {
             item.name = index === 0 ? file.split(".")[0] : `${file.split(".")[0]}_${index}`;
          }

          console.log(`${c.green}✓${c.reset} ${c.white}${item.name}${c.reset}`);

          if (item instanceof Command) {
            this.commands.set(item.name, item);
          }

          if (item.events && Array.isArray(item.events)) {
            item.events.forEach((event) => {
              if (event instanceof Event) {
                this.on(event.event, (...args) => event.run(this, ...args));
              }
            });
          }
          
          // If it's a direct Event instance
          if (item instanceof Event) {
            this.on(item.event, (...args) => item.run(this, ...args));
          }
        });
      });
    }

    this.on("clientReady", async () => {
      const guild = getPrimaryGuild(this);
      if (!guild) return;
      await guild.commands.set(this.commands);
      const { logSlashCommandsRegistered } = require("./BotStartup.js");
      logSlashCommandsRegistered(this);
    });

    section("Events");

    fs.readdirSync("./Events")
      .filter((file) => file.endsWith(".js"))
      .forEach((file) => {
        const event = require(`../Events/${file}`);
        console.log(
          `${c.green}✓${c.reset} ${c.white}${event.event}${c.reset} ${c.gray}(${file})${c.reset}`,
        );
        this.on(event.event, (...args) => event.run(this, ...args));
      });

    console.log("");

    if (process.argv[2] !== "test") {
      await this.login(token);
    } else {
      console.log(`${c.red}RUNNING IN TEST MODE, IF NOT INTENDED, PLEASE USE npm start${c.reset}`);
    }
  }
}

module.exports = Client;