l1l1th/src/index.js

const { REST, Routes, Client, Collection, GatewayIntentBits, Events, Partials, InteractionType, ActivityType } = require("discord.js");
const { default: Groq } = require("groq-sdk");
const { knex } = require("./db.js");
const path = require("node:path");
const fs = require("node:fs");
require("dotenv").config();

const client = new Client({
  intents: Object.keys(GatewayIntentBits).map(i => GatewayIntentBits[i]),
  partials: Object.keys(Partials).map(i => Partials[i])
});

client.commands = new Collection();
client.groq = new Groq({ apiKey: process.env.GROQ_API_KEY });

client.prompts = [];
var promptsDir = path.join(__dirname, "prompts");

const commands = [];
const foldersPath = path.join(__dirname, "commands");
const commandFolders = fs.readdirSync(foldersPath);

for (const folder of commandFolders) {
  const commandsPath = path.join(foldersPath, folder);
  const commandFiles = fs.readdirSync(commandsPath).filter(file => file.endsWith(".js"));
  for (const file of commandFiles) {
    const filePath = path.join(commandsPath, file);
    const command = require(filePath);
    if ("data" in command && "execute" in command) {
      client.commands.set(command.data.name, command);
      commands.push(command.data.toJSON());
    } else {
      console.log(`[WARNING] The command at ${filePath} is missing a required "data" or "execute" property.`);
    }
  }
}

client.on(Events.InteractionCreate, async interaction => {
  const command = interaction.client.commands.get(interaction.commandName);
  if (!command) {
    console.error(`No command matching ${interaction.commandName} was found.`);
    return;
  }

  if (interaction.isAutocomplete()) {
    try {
      await command.autocomplete(interaction);
    } catch (error) {
      console.error(error);
    }
  } else if (interaction.isCommand()) {
    console.log(`${interaction.user.username} ran ${(interaction.isChatInputCommand() ? "/" : '') + interaction.commandName}`);

    try {
      interaction.defaultModel = "llama-3.1-70b-versatile";
      await command.execute(interaction);
    } catch (err) {
      console.error(err);
      const data = {
        content: `${err.name}: ${err.message}`,
        ephemeral: true
      };
      try {
        if (interaction.replied || interaction.deferred) {
          await interaction.followUp(data);
        } else {
          await interaction.reply(data);
        }
      } catch { }
    }
  }
});

client.once(Events.ClientReady, async () => {
  console.log(`Ready! Logged in as ${client.user.displayName}`);
  client.user.setActivity({
    name: "kissing my sister",
    type: ActivityType.Custom
  });

  if (!(await knex.schema.hasTable("blacklists")))
    await knex.schema.createTable("blacklists", function (table) {
      table.string("user").primary();
      table.string("blacklist");
    });

  if (!(await knex.schema.hasTable("serverconfigs")))
    await knex.schema.createTable("serverconfigs", function (table) {
      table.string("id").primary();
      table.string("data");
    });

  if (!(await knex.schema.hasTable("userconfigs")))
    await knex.schema.createTable("userconfigs", function (table) {
      table.string("id").primary();
      table.string("data");
    });

  var user = client.user.toJSON();

  for (const prompt of fs.readdirSync(promptsDir)) {
    var data = fs.readFileSync(path.join(promptsDir, prompt), "utf-8");
    for (const info in user) {
      data = data.replaceAll(`{{${info}}}`, user[info]);
    }
    client.prompts[prompt.substring(0, prompt.lastIndexOf("."))] = data;
  }

  const rest = new REST().setToken(client.token);

  try {
    console.log(`Started refreshing ${commands.length} application (/) commands.`);

    const data = await rest.put(
      Routes.applicationCommands(client.user.id), { body: commands },
    );

    console.log(`Successfully reloaded ${data.length} application (/) commands.`);
  } catch (error) {
    console.error(error);
  }
});

client.login(process.env.TOKEN);