seyfert/packages/cache/mod.ts

import type { Emoji, Session, SymCache } from "./deps.ts";
import type { CachedGuild } from "./guilds.ts";
import type { CachedUser } from "./users.ts";
import type { CachedDMChannel } from "./channels.ts";
import { Collection } from "./Collection.ts";
import { memberBootstrapper } from "./members.ts";
import { userBootstrapper } from "./users.ts";
import { channelBootstrapper } from "./channels.ts";
import { guildBootstrapper } from "./guilds.ts";
import {
    messageBootstrapper,
    reactionBootstrapper,
    reactionBootstrapperDeletions
} from "./messages.ts";

export const cache_sym = Symbol("@cache");

export interface SessionCache extends SymCache {
    guilds: Collection<CachedGuild>;
    users: Collection<CachedUser>;
    dms: Collection<CachedDMChannel>;
    emojis: Collection<Emoji>;
    session: Session;
}

export function enableCache(session: Session): SessionCache {
    const cache = {
        guilds: new Collection<CachedGuild>(session),
        users: new Collection<CachedUser>(session),
        dms: new Collection<CachedDMChannel>(session),
        emojis: new Collection<Emoji>(session),
        cache: cache_sym,
        session,
    };

    session.on("raw", (data) => {
        // deno-lint-ignore no-explicit-any
        const raw = data.d as any;

        if (!raw) return;

        switch (data.t) {
            // TODO: add more events
            // for now users have to use the bootstrappers that are not implemented yet
            case "MESSAGE_CREATE":
                messageBootstrapper(cache, raw, false);
                break;
            case "MESSAGE_UPDATE":
                messageBootstrapper(cache, raw, !raw.edited_timestamp);
                break;
            case "CHANNEL_CREATE":
                channelBootstrapper(cache, raw);
                break;
            case "GUILD_MEMBER_ADD":
                memberBootstrapper(cache, raw, raw.guild_id);
                break;
            case "GUILD_CREATE":
                guildBootstrapper(cache, raw);
                break;
            case "GUILD_DELETE":
                cache.guilds.delete(raw.id);
                break;
            case "MESSAGE_REACTION_ADD":
                reactionBootstrapper(cache, raw, false);
                break;
            case "MESSAGE_REACTION_REMOVE":
                reactionBootstrapper(cache, raw, false);
                break;
            case "MESSAGE_REACTION_REMOVE_ALL":
                reactionBootstrapperDeletions(cache, raw);
                break;
            case "READY":
                userBootstrapper(cache, raw.user);
                break;
            default:
                session.emit("debug", `NOT CACHED: ${JSON.stringify(raw)}`);
        }
    });

    return cache;
}

export default enableCache;