123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
import { Bot, Collection, DiscordGuildEmojisUpdate } from "./deps.ts";import { addCacheCollections, BotWithCache } from "./src/addCacheCollections.ts";import { setupCacheEdits } from "./src/setupCacheEdits.ts";import { setupCacheRemovals } from "./src/setupCacheRemovals.ts";
// PLUGINS MUST TAKE A BOT ARGUMENT WHICH WILL BE MODIFIEDexport function enableCachePlugin<B extends Bot = Bot>(rawBot: B): BotWithCache<B> {  // MARK THIS PLUGIN BEING USED  rawBot.enabledPlugins.add("CACHE");
  // CUSTOMIZATION GOES HERE  const bot = addCacheCollections(rawBot);
  // Get the unmodified transformer.  const { guild, user, member, channel, message, presence, role } = bot.transformers;  // Override the transformer  bot.transformers.guild = function (_, payload) {    // Run the unmodified transformer    const result = guild(bot, payload);    // Cache the result    if (result) {      bot.guilds.set(result.id, result);
      const channels = payload.guild.channels || [];
      channels.forEach((channel) => {        bot.transformers.channel(bot, { channel, guildId: result.id });      });
      payload.guild.members?.forEach((member) => {        bot.transformers.member(bot, member, result.id, bot.transformers.snowflake(member.user!.id));      });    }
    // Return the result    return result;  };
  // Override the transformer  bot.transformers.user = function (...args) {    // Run the unmodified transformer    const result = user(...args);    // Cache the result    if (result) {      bot.users.set(result.id, result);    }    // Return the result    return result;  };
  // Override the transformer  bot.transformers.member = function (...args) {    // Run the unmodified transformer    const result = member(...args);    // Cache the result    if (result) {      bot.members.set(        bot.transformers.snowflake(`${result.id}${result.guildId}`),        result,      );    }    // Return the result    return result;  };
  // Override the transformer  bot.transformers.channel = function (...args) {    // Run the unmodified transformer    const result = channel(...args);    // Cache the result    if (result) {      bot.channels.set(result.id, result);    }    // Return the result    return result;  };
  // Override the transformer  bot.transformers.message = function (_, payload) {    // Run the unmodified transformer    const result = message(bot, payload);    // Cache the result    if (result) {      bot.messages.set(result.id, result);      // CACHE THE USER      const user = bot.transformers.user(bot, payload.author);      bot.users.set(user.id, user);
      if (payload.guild_id && payload.member) {        const guildId = bot.transformers.snowflake(payload.guild_id);        // CACHE THE MEMBER        bot.members.set(          bot.transformers.snowflake(`${payload.author.id}${payload.guild_id}`),          bot.transformers.member(bot, payload.member, guildId, user.id),        );      }    }
    // Return the result    return result;  };
  // Override the transformer  bot.transformers.presence = function (...args) {    // Run the unmodified transformer    const result = presence(...args);    // Cache the result    if (result) {      bot.presences.set(result.user.id, result);    }    // Return the result    return result;  };
  // Override the transformer  bot.transformers.role = function (...args) {    // Run the unmodified transformer    const result = role(...args);    // Cache the result    if (result) {      bot.guilds.get(result.guildId)?.roles.set(result.id, result);    }    // Return the result    return result;  };
  const { GUILD_EMOJIS_UPDATE } = bot.handlers;  bot.handlers.GUILD_EMOJIS_UPDATE = function (_, data, shardId) {    const payload = data.d as DiscordGuildEmojisUpdate;
    const guild = bot.guilds.get(bot.transformers.snowflake(payload.guild_id));    if (guild) {      guild.emojis = new Collection(payload.emojis.map((e) => {        const emoji = bot.transformers.emoji(bot, e);        return [emoji.id!, emoji];      }));    }
    GUILD_EMOJIS_UPDATE(bot, data, shardId);  };
  setupCacheRemovals(bot);  setupCacheEdits(bot);
  // PLUGINS MUST RETURN THE BOT  return bot;}
export default enableCachePlugin;export * from "./src/addCacheCollections.ts";export * from "./src/dispatchRequirements.ts";export * from "./src/setupCacheEdits.ts";export * from "./src/setupCacheRemovals.ts";export * from "./src/sweepers.ts";