majjo/stoatchat
1
0
Fork 0
mirror of https://github.com/stoatchat/stoatchat.git synced 2026-08-30 22:04:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit from fork

Browse Source 
* fix: check membership before subscribing to servers

Signed-off-by: IAmTomahawkx <iamtomahawkx@gmail.com>

* remove active_server when leaving a server.

Signed-off-by: IAmTomahawkx <iamtomahawkx@gmail.com>

---------

Signed-off-by: IAmTomahawkx <iamtomahawkx@gmail.com>
This commit is contained in:
Tom
2026-08-02 10:52:25 -07:00
committed by GitHub
parent 4283c0b33b
commit fa74826eea
3 changed files with 28 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -567,6 +567,7 @@ impl State {
EventV1::ServerMemberLeave { id, user, .. } => {
if user == &self.cache.user_id {
self.remove_subscription(id).await;
self.remove_active_server(id).await;
if let Some(server) = self.cache.servers.remove(id) {
for channel in &server.channels {

15
crates/bonfire/src/events/state.rs
View File

@@ -216,4 +216,19 @@ impl State {
subscribed.remove(subscription);
}
// Remove a server from the active server state.
pub async fn remove_active_server(&mut self, server_id: &str) -> Option<()> {
let removed = {
let mut lock = self.active_servers.lock().await;
lock.remove(server_id).is_some()
};
if removed {
self.remove_subscription(&format!("{server_id}u")).await;
Some(())
} else {
None
}
}
}

18
crates/bonfire/src/websocket.rs
View File

@@ -189,6 +189,7 @@ pub async fn client(db: &'static Database, stream: TcpStream, addr: SocketAddr)
read,
&write,
kill_signal_1_s,
db,
);
join!(listener, worker);
@@ -420,6 +421,7 @@ async fn worker_with_kill_signal(
read: WsReader,
write: &Mutex<WsWriter>,
kill_signal_s: async_channel::Sender<()>,
db: &Database,
) {
worker(
addr,
@@ -431,6 +433,7 @@ async fn worker_with_kill_signal(
kill_signal_r,
read,
write,
db,
)
.await;
kill_signal_s.send(()).await.ok();
@@ -447,6 +450,7 @@ async fn worker(
kill_signal_r: async_channel::Receiver<()>,
mut read: WsReader,
write: &Mutex<WsWriter>,
db: &Database,
) {
loop {
let t1 = read.try_next().fuse();
@@ -508,13 +512,15 @@ async fn worker(
.await;
}
ClientMessage::Subscribe { server_id } => {
let mut servers = active_servers.lock().await;
let has_item = servers.contains_key(&server_id);
servers.insert(server_id, ());
if db.fetch_member(&server_id, &user_id).await.is_ok() {
let mut servers = active_servers.lock().await;
let has_item = servers.contains_key(&server_id);
servers.insert(server_id, ());
if !has_item {
// Poke the listener to adjust subscriptions
topic_signal_s.send(()).await.ok();
if !has_item {
// Poke the listener to adjust subscriptions
topic_signal_s.send(()).await.ok();
}
}
}
ClientMessage::Ping { data, responded } => {