setupCacheRemovals.ts

import { Bot, DiscordChannel, DiscordGuildBanAddRemove, DiscordGuildMemberRemove, DiscordGuildRoleDelete, DiscordUnavailableGuild, type DesiredPropertiesBehavior, type TransformersDesiredProperties } from '@discordeno/bot';
import { BotWithProxyCache, ProxyCacheTypes } from './index.js';

export const setupCacheRemovals = <T extends ProxyCacheTypes<Props, Behavior>, Props extends TransformersDesiredProperties, Behavior extends DesiredPropertiesBehavior, B extends Bot<Props, Behavior>>(bot: BotWithProxyCache<T, Props, Behavior, B>) => {
    const { CHANNEL_DELETE, GUILD_BAN_ADD, GUILD_DELETE, GUILD_MEMBER_REMOVE, GUILD_ROLE_DELETE, THREAD_DELETE } = bot.handlers;

    bot.handlers.CHANNEL_DELETE = (_, data, shardId) => {
        const payload = data.d as DiscordChannel;
        // HANDLER BEFORE DELETING, BECAUSE HANDLER RUNS TRANSFORMER WHICH RE CACHES
        CHANNEL_DELETE(bot, data, shardId);

        const id = bot.transformers.snowflake(payload.id);

        setTimeout(() => bot.cache.channels.delete(id), 5000);
    };

    bot.handlers.GUILD_DELETE = (_, data, shardId) => {
        const payload = data.d as DiscordUnavailableGuild;
        const id = bot.transformers.snowflake(payload.id);

        // Remove the guild from cache
        bot.cache.options.bulk?.removeGuild?.(id);

        GUILD_DELETE(bot, data, shardId);
    };

    bot.handlers.GUILD_MEMBER_REMOVE = (_, data, shardId) => {
        const payload = data.d as DiscordGuildMemberRemove;

        GUILD_MEMBER_REMOVE(bot, data, shardId);

        bot.cache.members.delete(bot.transformers.snowflake(payload.user.id), bot.transformers.snowflake(payload.guild_id));
    };

    bot.handlers.GUILD_BAN_ADD = (_, data, shardId) => {
        const payload = data.d as DiscordGuildBanAddRemove;
        GUILD_BAN_ADD(bot, data, shardId);

        bot.cache.members.delete(bot.transformers.snowflake(payload.user.id), bot.transformers.snowflake(payload.guild_id));
    };

    bot.handlers.GUILD_ROLE_DELETE = (_, data, shardId) => {
        const payload = data.d as DiscordGuildRoleDelete;
        const id = bot.transformers.snowflake(payload.role_id);

        bot.cache.options.bulk?.removeRole?.(id);

        GUILD_ROLE_DELETE(bot, data, shardId);
    };

    bot.handlers.THREAD_DELETE = (_, data, shardId) => {
        const payload = data.d as DiscordChannel;
        // HANDLER BEFORE DELETING, BECAUSE HANDLER RUNS TRANSFORMER WHICH RE CACHES
        THREAD_DELETE(bot, data, shardId);

        const id = bot.transformers.snowflake(payload.id);

        setTimeout(() => bot.cache.channels.delete(id), 5000);
    };
};