HMC/handmade-revolt-backend
6
0
Fork 0
forked from jmug/stoatchat
Code Pull Requests Actions Packages Releases 2 Activity

feat: ready payload field customisation

Browse Source
This commit is contained in:
Zomatree
2025-05-05 22:14:10 +01:00
committed by Angelo Kontaxis
parent a186732ca5
commit db57706794
5 changed files with 65 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
crates/bonfire/src/events/impl.rs
View File

@@ -95,7 +95,7 @@ impl State {
pub async fn generate_ready_payload(
&mut self,
db: &Database,
fields: Vec<ReadyPayloadFields>,
fields: &ReadyPayloadFields,
) -> Result<EventV1> {
let user = self.clone_user();
self.cache.is_bot = user.bot.is_some();
@@ -168,7 +168,7 @@ impl State {
.await?;
// Fetch customisations.
let emojis = if fields.contains(&ReadyPayloadFields::Emoji) {
let emojis = if fields.emojis {
Some(
db.fetch_emoji_by_parent_ids(
&servers
@@ -183,17 +183,14 @@ impl State {
};
// Fetch user settings
let user_settings = if let Some(ReadyPayloadFields::UserSettings(keys)) = fields
.iter()
.find(|e| matches!(e, ReadyPayloadFields::UserSettings(_)))
{
Some(db.fetch_user_settings(&user.id, keys).await?)
let user_settings = if !fields.user_settings.is_empty() {
Some(db.fetch_user_settings(&user.id, &fields.user_settings).await?)
} else {
None
};
// Fetch channel unreads
let channel_unreads = if fields.contains(&ReadyPayloadFields::ChannelUnreads) {
let channel_unreads = if fields.channel_unreads {
Some(db.fetch_unreads(&user.id).await?)
} else {
None
@@ -241,22 +238,22 @@ impl State {
}
Ok(EventV1::Ready {
users: if fields.contains(&ReadyPayloadFields::Users) {
users: if fields.users {
Some(users)
} else {
None
},
servers: if fields.contains(&ReadyPayloadFields::Servers) {
servers: if fields.servers {
Some(servers.into_iter().map(Into::into).collect())
} else {
None
},
channels: if fields.contains(&ReadyPayloadFields::Channels) {
channels: if fields.channels {
Some(channels.into_iter().map(Into::into).collect())
} else {
None
},
members: if fields.contains(&ReadyPayloadFields::Members) {
members: if fields.members {
Some(members.into_iter().map(Into::into).collect())
} else {
None