discordeno/packages/plugins/cache/src/addCacheCollections.ts

import {
  BigString,
  Bot,
  Channel,
  Collection,
  Guild,
  Member,
  Message,
  ModifyWebhook,
  PresenceUpdate,
  User,
  Webhook
} from '@discordeno/bot'

export type BotWithCache<B extends Bot = Bot> = Omit<B, 'helpers'> & CacheProps & {
  helpers: BotHelpersWithCache<B['helpers']>
}

export type BotHelpersWithCache<T> = Omit<T, 'editWebhook'> & {
  /** The added channelId argument at the end is used to validate permission checks */
  editWebhook: (webhookId: BigString, options: ModifyWebhook, fromChannelId?: bigint) => Promise<Webhook>
}

export interface CacheProps {
  guilds: Collection<bigint, Guild>
  users: Collection<bigint, User>
  members: Collection<bigint, Member>
  channels: Collection<bigint, Channel>
  messages: Collection<bigint, Message>
  presences: Collection<bigint, PresenceUpdate>
  dispatchedGuildIds: Set<bigint>
  dispatchedChannelIds: Set<bigint>
}

export function addCacheCollections<B extends Bot> (bot: B): BotWithCache<B> {
  const cacheBot = bot as unknown as BotWithCache<B>
  cacheBot.guilds = new Collection()
  cacheBot.users = new Collection()
  cacheBot.members = new Collection()
  cacheBot.channels = new Collection()
  cacheBot.messages = new Collection()
  cacheBot.presences = new Collection()
  cacheBot.dispatchedGuildIds = new Set()
  cacheBot.dispatchedChannelIds = new Set()

  return cacheBot
}