//! A cache of events received over a `Shard`, where storing at least some //! data from the event is possible. //! //! This acts as a cache, to avoid making requests over the REST API through //! the [`rest`] module where possible. All fields are public, and do not have //! getters, to allow you more flexibility with the stored data. However, this //! allows data to be "corrupted", and _may or may not_ cause misfunctions //! within the library. Mutate data at your own discretion. //! //! A "globally available" instance of the Cache is available at //! [`client::CACHE`]. This is the instance that is updated by the library, //! meaning you should _not_ need to maintain updating it yourself in any case. //! //! # Use by the Context //! //! The [`Context`] will automatically attempt to pull from the cache for you. //! For example, the [`Context::get_channel`] method will attempt to find the //! channel in the cache. If it can not find it, it will perform a request //! through the REST API, and then insert a clone of the channel - if found - //! into the Cache, giving you the original. //! //! This allows you to save a step, by only needing to perform the //! [`Context::get_channel`] call and not need to first search through the cache //! - and if not found - _then_ perform an HTTP request through the Context or //! [`rest`] module. //! //! Additionally, note that some information received through events can _not_ //! be retrieved through the REST API. This is information such as [`Role`]s in //! [`LiveGuild`]s. //! //! # Use by Models //! //! Most models of Discord objects, such as the [`Message`], [`PublicChannel`], //! or [`Emoji`], have methods for interacting with that single instance. This //! feature is only compiled if the `methods` feature is enabled. An example of //! this is [`LiveGuild::edit`], which performs a check to ensure that the //! current user is the owner of the guild, prior to actually performing the //! HTTP request. The cache is involved due to the function's use of unlocking //! the cache and retrieving the Id of the current user, and comparing it to //! the Id of the user that owns the guild. This is an inexpensive method of //! being able to access data required by these sugary methods. //! //! # Do I need the Cache? //! //! If you're asking this, the answer is likely "definitely yes" or //! "definitely no"; any in-between tends to be "yes". If you are low on RAM, //! and need to run on only a couple MB, then the answer is "definitely no". If //! you do not care about RAM and want your bot to be able to access data //! while needing to hit the REST API as little as possible, then the answer //! is "yes". //! //! [`Context`]: ../../client/struct.Context.html //! [`Context::get_channel`]: ../../client/struct.Context.html#method.get_channel //! [`Emoji`]: ../../model/struct.Emoji.html //! [`Group`]: ../../model/struct.Group.html //! [`LiveGuild`]: ../../model/struct.LiveGuild.html //! [`LiveGuild::edit`]: ../../model/struct.LiveGuild.html#method.edit //! [`Message`]: ../../model/struct.Message.html //! [`PublicChannel`]: ../../model/struct.PublicChannel.html //! [`Role`]: ../../model/struct.Role.html //! [`client::CACHE`]: ../../client/struct.CACHE.html //! [`rest`]: ../../client/rest/index.html use std::collections::hash_map::Entry; use std::collections::{HashMap, HashSet}; use std::default::Default; use std::sync::{Arc, RwLock}; use std::mem; use ::model::*; use ::model::event::*; /// A cache of all events received over a [`Connection`], where storing at least /// some data from the event is possible. /// /// This acts as a cache, to avoid making requests over the REST API through the /// [`rest`] module where possible. All fields are public, and do not have /// getters, to allow you more flexibility with the stored data. However, this /// allows data to be "corrupted", and _may or may not_ cause misfunctions /// within the library. Mutate data at your own discretion. /// /// # Use by the Context /// /// The [`Context`] will automatically attempt to pull from the cache for you. /// For example, the [`Context::get_channel`] method will attempt to find the /// channel in the cache. If it can not find it, it will perform a request /// through the REST API, and then insert a clone of the channel - if found - /// into the Cache. /// /// This allows you to only need to perform the `Context::get_channel` call, /// and not need to first search through the cache - and if not found - _then_ /// perform an HTTP request through the Context or `rest` module. /// /// Additionally, note that some information received through events can _not_ /// be retrieved through the REST API. This is information such as [`Role`]s in /// [`Guild`]s. /// /// [`Connection`]: ../../client/struct.Connection.html /// [`Context`]: ../../client/struct.Context.html /// [`Context::get_channel`]: ../../client/struct.Context.html#method.get_channel /// [`Guild`]: ../../model/struct.Guild.html /// [`Role`]: ../../model/struct.Role.html /// [`rest`]: ../../client/rest/index.html #[derive(Clone, Debug)] pub struct Cache { /// A map of channels in [`Guild`]s that the current user has received data /// for. /// /// When a [`Event::GuildDelete`] or [`Event::GuildUnavailable`] is /// received and processed by the cache, the relevant channels are also /// removed from this map. /// /// [`Event::GuildDelete`]: ../../model/event/struct.GuildDeleteEvent.html /// [`Event::GuildUnavailable`]: ../../model/event/struct.GuildUnavailableEvent.html /// [`Guild`]: ../../model/struct.Guild.html pub channels: HashMap>>, /// A map of the groups that the current user is in. /// /// For bot users this will always be empty, except for in [special cases]. /// /// [special cases]: index.html#special-cases-in-the-cache pub groups: HashMap>>, /// A map of guilds with full data available. This includes data like /// [`Role`]s and [`Emoji`]s that are not available through the REST API. /// /// [`Emoji`]: ../../model/struct.Emoji.html /// [`Role`]: ../../model/struct.Role.html pub guilds: HashMap>>, /// A map of notes that a user has made for individual users. /// /// An empty note is equivalent to having no note, and creating an empty /// note is equivalent to deleting a note. /// /// This will always be empty for bot users. pub notes: HashMap, /// A map of users' presences. This is updated in real-time. Note that /// status updates are often "eaten" by the gateway, and this should not /// be treated as being entirely 100% accurate. pub presences: HashMap, /// A map of direct message channels that the current user has open with /// other users. pub private_channels: HashMap>>, /// The total number of shards being used by the bot. pub shard_count: u64, /// A list of guilds which are "unavailable". Refer to the documentation for /// [`Event::GuildUnavailable`] for more information on when this can occur. /// /// Additionally, guilds are always unavailable for bot users when a Ready /// is received. Guilds are "sent in" over time through the receiving of /// [`Event::GuildCreate`]s. /// /// [`Event::GuildCreate`]: ../../model/enum.Event.html#variant.GuildCreate /// [`Event::GuildUnavailable`]: ../../model/enum.Event.html#variant.GuildUnavailable pub unavailable_guilds: HashSet, /// The current user "logged in" and for which events are being received /// for. /// /// The current user contains information that a regular [`User`] does not, /// such as whether it is a bot, whether the user is verified, etc. /// /// Refer to the documentation for [`CurrentUser`] for more information. /// /// [`CurrentUser`]: ../../model/struct.CurrentUser.html /// [`User`]: ../../model/struct.User.html pub user: CurrentUser, /// A map of users that the current user sees. /// /// Users are added to - and updated from - this map via the following /// received events: /// /// - [`ChannelRecipientAdd`][`ChannelRecipientAddEvent`] /// - [`GuildMemberAdd`][`GuildMemberAddEvent`] /// - [`GuildMemberRemove`][`GuildMemberRemoveEvent`] /// - [`GuildMembersChunk`][`GuildMembersChunkEvent`] /// - [`GuildSync`][`GuildSyncEvent`] /// - [`PresenceUpdate`][`PresenceUpdateEvent`] /// - [`Ready`][`ReadyEvent`] /// - [`RelationshipAdd`][`RelationshipAddEvent`] /// /// Note, however, that users are _not_ removed from the map on removal /// events such as [`GuildMemberRemove`][`GuildMemberRemoveEvent`], as other /// structs such as members or recipients may still exist. /// /// [`ChannelRecipientAddEvent`]: ../../model/event/struct.ChannelRecipientAddEvent.html /// [`GuildMemberAddEvent`]: ../../model/event/struct.GuildMemberAddEvent.html /// [`GuildMemberRemoveEvent`]: ../../model/event/struct.GuildMemberRemoveEvent.html /// [`GuildMemberUpdateEvent`]: ../../model/event/struct.GuildMemberUpdateEvent.html /// [`GuildMembersChunkEvent`]: ../../model/event/struct.GuildMembersChunkEvent.html /// [`GuildSyncEvent`]: ../../model/event/struct.GuildSyncEvent.html /// [`PresenceUpdateEvent`]: ../../model/event/struct.PresenceUpdateEvent.html /// [`Ready`]: ../../model/event/struct.ReadyEvent.html /// [`RelationshipAdd`][`RelationshipAddEvent`] pub users: HashMap>>, } impl Cache { /// Fetches the number of [`Member`]s that have not had data received. /// /// The important detail to note here is that this is the number of /// _member_s that have not had data downloaded. A single [`User`] may have /// multiple associated member objects that have not been received. /// /// This can be used in combination with [`Shard::chunk_guilds`], and can be /// used to determine how many members have not yet been downloaded. /// /// [`Member`]: ../../model/struct.Member.html /// [`User`]: ../../model/struct.User.html pub fn unknown_members(&self) -> u64 { let mut total = 0; for guild in self.guilds.values() { let guild = guild.read().unwrap(); let members = guild.members.len() as u64; if guild.member_count > members { total += guild.member_count - members; } } total } /// Fetches a vector of all [`PrivateChannel`] and [`Group`] Ids that are /// stored in the cache. /// /// # Examples /// /// If there are 6 private channels and 2 groups in the cache, then `8` Ids /// will be returned. /// /// [`Group`]: ../../model/struct.Group.html /// [`PrivateChannel`]: ../../model/struct.PrivateChannel.html pub fn all_private_channels(&self) -> Vec { self.groups .keys() .cloned() .chain(self.private_channels.keys().cloned()) .collect() } /// Fetches a vector of all [`Guild`]s' Ids that are stored in the cache. /// /// Note that if you are utilizing multiple [`Shard`]s, then the guilds /// retrieved over all shards are included in this count -- not just the /// current [`Context`]'s shard, if accessing from one. /// /// [`Context`]: ../../client/struct.Context.html /// [`Guild`]: ../../model/struct.Guild.html /// [`Shard`]: ../../client/gateway/struct.Shard.html pub fn all_guilds(&self) -> Vec { self.guilds .values() .map(|g| g.read().unwrap().id) .chain(self.unavailable_guilds.iter().cloned()) .collect() } /// Retrieves a [`Channel`] from the cache based on the given Id. /// /// This will search the [`channels`] map, the [`private_channels`] map, and /// then the map of [`groups`] to find the channel. /// /// If you know what type of channel you're looking for, you should instead /// manually retrieve from one of the respective maps or methods: /// /// - [`GuildChannel`]: [`get_guild_channel`] or [`channels`] /// - [`PrivateChannel`]: [`get_private_channel`] or [`private_channels`] /// - [`Group`]: [`get_group`] or [`groups`] /// /// [`Channel`]: ../../model/enum.Channel.html /// [`Group`]: ../../model/struct.Group.html /// [`Guild`]: ../../model/struct.Guild.html /// [`channels`]: #structfield.channels /// [`get_group`]: #method.get_group /// [`get_guild_channel`]: #method.get_guild_channel /// [`get_private_channel`]: #method.get_private_channel /// [`groups`]: #structfield.groups /// [`private_channels`]: #structfield.private_channels pub fn channel>(&self, id: C) -> Option { let id = id.into(); if let Some(channel) = self.channels.get(&id) { return Some(Channel::Guild(channel.clone())); } if let Some(private_channel) = self.private_channels.get(&id) { return Some(Channel::Private(private_channel.clone())); } if let Some(group) = self.groups.get(&id) { return Some(Channel::Group(group.clone())); } None } /// Retrieves a guild from the cache based on the given Id. /// /// The only advantage of this method is that you can pass in anything that /// is indirectly a [`GuildId`]. /// /// [`GuildId`]: ../../model/struct.GuildId.html #[inline] pub fn guild>(&self, id: G) -> Option>> { self.guilds.get(&id.into()).cloned() } /// Retrieves a reference to a [`Guild`]'s channel. Unlike [`get_channel`], /// this will only search guilds for the given channel. /// /// The only advantage of this method is that you can pass in anything that /// is indirectly a [`ChannelId`]. /// /// # Examples /// /// Getting a guild's channel via the Id of the message received through a /// [`Client::on_message`] event dispatch: /// /// ```rust,ignore /// use serenity::client::CACHE; /// /// let cache = CACHE.read().unwrap(); /// /// let channel = match cache.get_guild_channel(message.channel_id) { /// Some(channel) => channel, /// None => { /// if let Err(why) = message.channel_id.say("Could not find guild's channel data") { /// println!("Error sending message: {:?}", why); /// } /// /// return; /// }, /// }; /// ``` /// /// [`ChannelId`]: ../../model/struct.ChannelId.html /// [`Client::on_message`]: ../../client/struct.Client.html#method.on_message /// [`Guild`]: ../../model/struct.Guild.html /// [`get_channel`]: #method.get_channel #[inline] pub fn guild_channel>(&self, id: C) -> Option>> { self.channels.get(&id.into()).cloned() } /// Retrieves a reference to a [`Group`] from the cache based on the given /// associated channel Id. /// /// The only advantage of this method is that you can pass in anything that /// is indirectly a [`ChannelId`]. /// /// [`ChannelId`]: ../../model/struct.ChannelId.html /// [`Group`]: ../../model/struct.Group.html #[inline] pub fn group>(&self, id: C) -> Option>> { self.groups.get(&id.into()).cloned() } /// Retrieves a [`Guild`]'s member from the cache based on the guild's and /// user's given Ids. /// /// **Note**: This will clone the entire member. Instead, retrieve the guild /// and retrieve from the guild's [`members`] map to avoid this. /// /// # Examples /// /// Retrieving the member object of the user that posted a message, in a /// [`Client::on_message`] context: /// /// ```rust,ignore /// use serenity::client::CACHE; /// /// let cache = CACHE.read().unwrap(); /// let member = { /// let channel = match cache.get_guild_channel(message.channel_id) { /// Some(channel) => channel, /// None => { /// if let Err(why) = message.channel_id.say("Error finding channel data") { /// println!("Error sending message: {:?}", why); /// } /// }, /// }; /// /// match cache.get_member(channel.guild_id, message.author.id) { /// Some(member) => member, /// None => { /// if let Err(why) = message.channel_id.say("Error finding member data") { /// println!("Error sending message: {:?}", why); /// } /// }, /// } /// }; /// /// let msg = format!("You have {} roles", member.roles.len()); /// /// if let Err(why) = message.channel_id.say(&msg) { /// println!("Error sending message: {:?}", why); /// } /// ``` /// /// [`Client::on_message`]: ../../client/struct.Client.html#method.on_message /// [`Guild`]: ../../model/struct.Guild.html /// [`members`]: ../../model/struct.Guild.html#structfield.members pub fn member(&self, guild_id: G, user_id: U) -> Option where G: Into, U: Into { self.guilds .get(&guild_id.into()) .and_then(|guild| guild.write().unwrap().members.get(&user_id.into()).cloned()) } /// Retrieves a [`PrivateChannel`] from the cache's [`private_channels`] /// map, if it exists. /// /// The only advantage of this method is that you can pass in anything that /// is indirectly a [`ChannelId`]. #[inline] pub fn private_channel>(&self, channel_id: C) -> Option>> { self.private_channels.get(&channel_id.into()).cloned() } /// Retrieves a [`Guild`]'s role by their Ids. /// /// **Note**: This will clone the entire role. Instead, retrieve the guild /// and retrieve from the guild's [`roles`] map to avoid this. /// /// [`Guild`]: ../../model/struct.Guild.html /// [`roles`]: ../../model/struct.Guild.html#structfield.roles pub fn role(&self, guild_id: G, role_id: R) -> Option where G: Into, R: Into { self.guilds .get(&guild_id.into()) .and_then(|g| g.read().unwrap().roles.get(&role_id.into()).cloned()) } /// Retrieves a `User` from the cache's [`users`] map, if it exists. /// /// The only advantage of this method is that you can pass in anything that /// is indirectly a [`UserId`]. /// /// [`UserId`]: ../../model/struct.UserId.html /// [`users`]: #structfield.users #[inline] pub fn user>(&self, user_id: U) -> Option>> { self.users.get(&user_id.into()).cloned() } /// Alias of [`channel`]. /// /// [`channel`]: #method.channel #[deprecated(since="0.1.5", note="Use `channel` instead.")] #[inline] pub fn get_channel>(&self, id: C) -> Option { self.channel(id) } /// Alias of [`guild`]. /// /// [`guild`]: #method.guild #[deprecated(since="0.1.5", note="Use `guild` instead.")] #[inline] pub fn get_guild>(&self, id: G) -> Option>> { self.guild(id) } /// Alias of [`guild_channel`]. /// /// [`guild_channel`]: #method.guild_channel #[deprecated(since="0.1.5", note="Use `guild_channel` instead.")] #[inline] pub fn get_guild_channel>(&self, id: C) -> Option>> { self.guild_channel(id) } /// Alias of [`member`]. /// /// [`member`]: #method.member #[deprecated(since="0.1.5", note="Use `member` instead.")] #[inline] pub fn get_member(&self, guild_id: G, user_id: U) -> Option where G: Into, U: Into { self.member(guild_id, user_id) } /// Alias of [`private_channel`]. /// /// [`private_channel`]: #method.private_channel #[deprecated(since="0.1.5", note="Use `private_channel` instead.")] #[inline] pub fn get_private_channel>(&self, id: C) -> Option>> { self.private_channel(id) } /// Alias of [`role`]. /// /// [`role`]: #method.role #[deprecated(since="0.1.5", note="Use `role` instead.")] #[inline] pub fn get_role(&self, guild_id: G, role_id: R) -> Option where G: Into, R: Into { self.role(guild_id, role_id) } /// Alias of [`user`]. /// /// [`user`]: #method.user #[deprecated(since="0.1.5", note="Use `user` instead.")] #[inline] pub fn get_user>(&self, id: U) -> Option>> { self.user(id) } #[doc(hidden)] pub fn update_with_channel_create(&mut self, event: &ChannelCreateEvent) -> Option { match event.channel { Channel::Group(ref group) => { let group = group.clone(); let channel_id = { let writer = group.write().unwrap(); for (recipient_id, recipient) in &mut group.write().unwrap().recipients { self.update_user_entry(&recipient.read().unwrap()); *recipient = self.users[recipient_id].clone(); } writer.channel_id }; let ch = self.groups.insert(channel_id, group); ch.map(Channel::Group) }, Channel::Guild(ref channel) => { let (guild_id, channel_id) = { let channel = channel.read().unwrap(); (channel.guild_id, channel.id) }; self.channels.insert(channel_id, channel.clone()); self.guilds .get_mut(&guild_id) .and_then(|guild| { guild.write().unwrap().channels.insert(channel_id, channel.clone()) }).map(Channel::Guild) }, Channel::Private(ref channel) => { let channel = channel.clone(); let mut channel_writer = channel.write().unwrap(); let user_id = { let user_reader = channel_writer.recipient.read().unwrap(); self.update_user_entry(&user_reader); user_reader.id }; channel_writer.recipient = self.users[&user_id].clone(); let ch = self.private_channels.insert(channel_writer.id, channel.clone()); ch.map(Channel::Private) }, } } #[doc(hidden)] pub fn update_with_channel_delete(&mut self, event: &ChannelDeleteEvent) -> Option { match event.channel { Channel::Group(ref group) => { self.groups.remove(&group.read().unwrap().channel_id).map(Channel::Group) }, Channel::Private(ref channel) => { self.private_channels.remove(&channel.read().unwrap().id).map(Channel::Private) }, Channel::Guild(ref channel) => { let (channel_id, guild_id) = { let channel = channel.read().unwrap(); (channel.id, channel.guild_id) }; self.channels.remove(&channel_id); self.guilds .get_mut(&guild_id) .and_then(|guild| guild.write().unwrap().channels.remove(&channel_id)) .map(Channel::Guild) }, } } #[doc(hidden)] pub fn update_with_channel_pins_update(&mut self, event: &ChannelPinsUpdateEvent) { if let Some(channel) = self.channels.get(&event.channel_id) { channel.write().unwrap().last_pin_timestamp = event.last_pin_timestamp.clone(); return; } if let Some(channel) = self.private_channels.get_mut(&event.channel_id) { channel.write().unwrap().last_pin_timestamp = event.last_pin_timestamp.clone(); return; } if let Some(group) = self.groups.get_mut(&event.channel_id) { group.write().unwrap().last_pin_timestamp = event.last_pin_timestamp.clone(); return; } } #[doc(hidden)] pub fn update_with_channel_recipient_add(&mut self, event: &mut ChannelRecipientAddEvent) { self.update_user_entry(&event.user); let user = self.users[&event.user.id].clone(); self.groups .get_mut(&event.channel_id) .map(|group| { group.write() .unwrap() .recipients .insert(event.user.id, user); }); } #[doc(hidden)] pub fn update_with_channel_recipient_remove(&mut self, event: &ChannelRecipientRemoveEvent) { self.groups .get_mut(&event.channel_id) .map(|group| group.write().unwrap().recipients.remove(&event.user.id)); } #[doc(hidden)] pub fn update_with_channel_update(&mut self, event: &ChannelUpdateEvent) { match event.channel { Channel::Group(ref group) => { let (ch_id, no_recipients) = { let group = group.read().unwrap(); (group.channel_id, group.recipients.is_empty()) }; match self.groups.entry(ch_id) { Entry::Vacant(e) => { e.insert(group.clone()); }, Entry::Occupied(mut e) => { let mut dest = e.get_mut().write().unwrap(); if no_recipients { let recipients = mem::replace(&mut dest.recipients, HashMap::new()); dest.clone_from(&group.read().unwrap()); dest.recipients = recipients; } else { dest.clone_from(&group.read().unwrap()); } }, } }, Channel::Guild(ref channel) => { let (channel_id, guild_id) = { let channel = channel.read().unwrap(); (channel.id, channel.guild_id) }; self.channels.insert(channel_id, channel.clone()); self.guilds .get_mut(&guild_id) .map(|guild| { guild.write() .unwrap() .channels .insert(channel_id, channel.clone()) }); }, Channel::Private(ref channel) => { self.private_channels .get_mut(&channel.read().unwrap().id) .map(|private| private.clone_from(channel)); }, } } #[doc(hidden)] pub fn update_with_guild_create(&mut self, event: &GuildCreateEvent) { self.unavailable_guilds.remove(&event.guild.id); let mut guild = event.guild.clone(); for (user_id, member) in &mut guild.members { self.update_user_entry(&member.user.read().unwrap()); let user = self.users[user_id].clone(); member.user = user.clone(); } self.channels.extend(guild.channels.clone()); self.guilds.insert(event.guild.id, Arc::new(RwLock::new(guild))); } #[doc(hidden)] pub fn update_with_guild_delete(&mut self, event: &GuildDeleteEvent) -> Option>> { // Remove channel entries for the guild if the guild is found. self.guilds.remove(&event.guild.id).map(|guild| { for channel_id in guild.read().unwrap().channels.keys() { self.channels.remove(channel_id); } guild }) } #[doc(hidden)] pub fn update_with_guild_emojis_update(&mut self, event: &GuildEmojisUpdateEvent) { self.guilds .get_mut(&event.guild_id) .map(|guild| guild.write().unwrap().emojis.extend(event.emojis.clone())); } #[doc(hidden)] pub fn update_with_guild_member_add(&mut self, event: &mut GuildMemberAddEvent) { let user_id = event.member.user.read().unwrap().id; self.update_user_entry(&event.member.user.read().unwrap()); // Always safe due to being inserted above. event.member.user = self.users[&user_id].clone(); self.guilds .get_mut(&event.guild_id) .map(|guild| { let mut guild = guild.write().unwrap(); guild.member_count += 1; guild.members.insert(user_id, event.member.clone()); }); } #[doc(hidden)] pub fn update_with_guild_member_remove(&mut self, event: &GuildMemberRemoveEvent) -> Option { self.guilds .get_mut(&event.guild_id) .and_then(|guild| { let mut guild = guild.write().unwrap(); guild.member_count -= 1; guild.members.remove(&event.user.id) }) } #[doc(hidden)] pub fn update_with_guild_member_update(&mut self, event: &GuildMemberUpdateEvent) -> Option { self.update_user_entry(&event.user); if let Some(guild) = self.guilds.get_mut(&event.guild_id) { let mut guild = guild.write().unwrap(); let mut found = false; let item = if let Some(member) = guild.members.get_mut(&event.user.id) { let item = Some(member.clone()); member.nick.clone_from(&event.nick); member.roles.clone_from(&event.roles); member.user.write().unwrap().clone_from(&event.user); found = true; item } else { None }; if !found { guild.members.insert(event.user.id, Member { deaf: false, guild_id: Some(event.guild_id), joined_at: String::default(), mute: false, nick: event.nick.clone(), roles: event.roles.clone(), user: Arc::new(RwLock::new(event.user.clone())), }); } item } else { None } } #[doc(hidden)] pub fn update_with_guild_members_chunk(&mut self, event: &GuildMembersChunkEvent) { for member in event.members.values() { self.update_user_entry(&member.user.read().unwrap()); } self.guilds .get_mut(&event.guild_id) .map(|guild| guild.write().unwrap().members.extend(event.members.clone())); } #[doc(hidden)] pub fn update_with_guild_role_create(&mut self, event: &GuildRoleCreateEvent) { self.guilds .get_mut(&event.guild_id) .map(|guild| guild.write().unwrap().roles.insert(event.role.id, event.role.clone())); } #[doc(hidden)] pub fn update_with_guild_role_delete(&mut self, event: &GuildRoleDeleteEvent) -> Option { self.guilds .get_mut(&event.guild_id) .and_then(|guild| guild.write().unwrap().roles.remove(&event.role_id)) } #[doc(hidden)] pub fn update_with_guild_role_update(&mut self, event: &GuildRoleUpdateEvent) -> Option { self.guilds .get_mut(&event.guild_id) .and_then(|guild| { guild.write() .unwrap() .roles .get_mut(&event.role.id) .map(|role| mem::replace(role, event.role.clone())) }) } #[doc(hidden)] pub fn update_with_guild_unavailable(&mut self, event: &GuildUnavailableEvent) { self.unavailable_guilds.insert(event.guild_id); self.guilds.remove(&event.guild_id); } #[doc(hidden)] pub fn update_with_guild_update(&mut self, event: &GuildUpdateEvent) { self.guilds .get_mut(&event.guild.id) .map(|guild| { let mut guild = guild.write().unwrap(); guild.afk_timeout = event.guild.afk_timeout; guild.afk_channel_id.clone_from(&event.guild.afk_channel_id); guild.icon.clone_from(&event.guild.icon); guild.name.clone_from(&event.guild.name); guild.owner_id.clone_from(&event.guild.owner_id); guild.region.clone_from(&event.guild.region); guild.roles.clone_from(&event.guild.roles); guild.verification_level = event.guild.verification_level; }); } #[doc(hidden)] pub fn update_with_presences_replace(&mut self, event: &PresencesReplaceEvent) { self.presences.extend({ let mut p: HashMap = HashMap::default(); for presence in &event.presences { p.insert(presence.user_id, presence.clone()); } p }); } #[doc(hidden)] pub fn update_with_presence_update(&mut self, event: &mut PresenceUpdateEvent) { let user_id = event.presence.user_id; if let Some(user) = event.presence.user.as_mut() { self.update_user_entry(&user.read().unwrap()); *user = self.users[&user_id].clone(); } if let Some(guild_id) = event.guild_id { if let Some(guild) = self.guilds.get_mut(&guild_id) { let mut guild = guild.write().unwrap(); // If the member went offline, remove them from the presence list. if event.presence.status == OnlineStatus::Offline { guild.presences.remove(&event.presence.user_id); } else { guild.presences.insert(event.presence.user_id, event.presence.clone()); } } } else if event.presence.status == OnlineStatus::Offline { self.presences.remove(&event.presence.user_id); } else { self.presences.insert(event.presence.user_id, event.presence.clone()); } } #[doc(hidden)] pub fn update_with_ready(&mut self, event: &ReadyEvent) { let mut ready = event.ready.clone(); for guild in ready.guilds { match guild { GuildStatus::Offline(unavailable) => { self.guilds.remove(&unavailable.id); self.unavailable_guilds.insert(unavailable.id); }, GuildStatus::OnlineGuild(guild) => { self.unavailable_guilds.remove(&guild.id); self.guilds.insert(guild.id, Arc::new(RwLock::new(guild))); }, GuildStatus::OnlinePartialGuild(_) => {}, } } // The private channels sent in the READY contains both the actual // private channels and the groups. for (channel_id, channel) in ready.private_channels { match channel { Channel::Group(group) => { self.groups.insert(channel_id, group); }, Channel::Private(channel) => { self.private_channels.insert(channel_id, channel); }, Channel::Guild(guild) => warn!("Got a guild in DMs: {:?}", guild), } } for (user_id, presence) in &mut ready.presences { if let Some(ref user) = presence.user { self.update_user_entry(&user.read().unwrap()); } presence.user = self.users.get(user_id).cloned(); } self.presences.extend(ready.presences); self.shard_count = ready.shard.map_or(1, |s| s[1]); self.user = ready.user; } #[doc(hidden)] pub fn update_with_user_update(&mut self, event: &UserUpdateEvent) -> CurrentUser { mem::replace(&mut self.user, event.current_user.clone()) } #[doc(hidden)] pub fn update_with_voice_state_update(&mut self, event: &VoiceStateUpdateEvent) { if let Some(guild_id) = event.guild_id { if let Some(guild) = self.guilds.get_mut(&guild_id) { let mut guild = guild.write().unwrap(); if event.voice_state.channel_id.is_some() { // Update or add to the voice state list { let finding = guild.voice_states.get_mut(&event.voice_state.user_id); if let Some(srv_state) = finding { srv_state.clone_from(&event.voice_state); return; } } guild.voice_states.insert(event.voice_state.user_id, event.voice_state.clone()); } else { // Remove the user from the voice state list guild.voice_states.remove(&event.voice_state.user_id); } } return; } } // Adds or updates a user entry in the [`users`] map with a received user. // // [`users`]: #structfield.users fn update_user_entry(&mut self, user: &User) { match self.users.entry(user.id) { Entry::Vacant(e) => { e.insert(Arc::new(RwLock::new(user.clone()))); }, Entry::Occupied(mut e) => { e.get_mut().write().unwrap().clone_from(user); } } } } impl Default for Cache { fn default() -> Cache { Cache { channels: HashMap::default(), groups: HashMap::default(), guilds: HashMap::default(), notes: HashMap::default(), presences: HashMap::default(), private_channels: HashMap::default(), shard_count: 1, unavailable_guilds: HashSet::default(), user: CurrentUser { avatar: None, bot: false, discriminator: 0, email: None, id: UserId(0), mfa_enabled: false, name: String::default(), verified: false, }, users: HashMap::default(), } } }