move sync watcher from globals service to sync service

Signed-off-by: Jason Volk <jason@zemos.net>

src/service/sync/watch.rs

@@ -0,0 +1,117 @@
+use conduit::{implement, trace, Result};
+use futures::{pin_mut, stream::FuturesUnordered, FutureExt, StreamExt};
+use ruma::{DeviceId, UserId};
+
+#[implement(super::Service)]
+#[tracing::instrument(skip(self), level = "debug")]
+pub async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result {
+	let userid_bytes = user_id.as_bytes().to_vec();
+	let mut userid_prefix = userid_bytes.clone();
+	userid_prefix.push(0xFF);
+
+	let mut userdeviceid_prefix = userid_prefix.clone();
+	userdeviceid_prefix.extend_from_slice(device_id.as_bytes());
+	userdeviceid_prefix.push(0xFF);
+
+	let mut futures = FuturesUnordered::new();
+
+	// Return when *any* user changed their key
+	// TODO: only send for user they share a room with
+	futures.push(self.db.todeviceid_events.watch_prefix(&userdeviceid_prefix));
+
+	futures.push(self.db.userroomid_joined.watch_prefix(&userid_prefix));
+	futures.push(self.db.userroomid_invitestate.watch_prefix(&userid_prefix));
+	futures.push(self.db.userroomid_leftstate.watch_prefix(&userid_prefix));
+	futures.push(
+		self.db
+			.userroomid_notificationcount
+			.watch_prefix(&userid_prefix),
+	);
+	futures.push(
+		self.db
+			.userroomid_highlightcount
+			.watch_prefix(&userid_prefix),
+	);
+
+	// Events for rooms we are in
+	let rooms_joined = self.services.state_cache.rooms_joined(user_id);
+
+	pin_mut!(rooms_joined);
+	while let Some(room_id) = rooms_joined.next().await {
+		let Ok(short_roomid) = self.services.short.get_shortroomid(room_id).await else {
+			continue;
+		};
+
+		let roomid_bytes = room_id.as_bytes().to_vec();
+		let mut roomid_prefix = roomid_bytes.clone();
+		roomid_prefix.push(0xFF);
+
+		// Key changes
+		futures.push(self.db.keychangeid_userid.watch_prefix(&roomid_prefix));
+
+		// Room account data
+		let mut roomuser_prefix = roomid_prefix.clone();
+		roomuser_prefix.extend_from_slice(&userid_prefix);
+
+		futures.push(
+			self.db
+				.roomusertype_roomuserdataid
+				.watch_prefix(&roomuser_prefix),
+		);
+
+		// PDUs
+		let short_roomid = short_roomid.to_be_bytes().to_vec();
+		futures.push(self.db.pduid_pdu.watch_prefix(&short_roomid));
+
+		// EDUs
+		let typing_room_id = room_id.to_owned();
+		let typing_wait_for_update = async move {
+			self.services.typing.wait_for_update(&typing_room_id).await;
+		};
+
+		futures.push(typing_wait_for_update.boxed());
+		futures.push(
+			self.db
+				.readreceiptid_readreceipt
+				.watch_prefix(&roomid_prefix),
+		);
+	}
+
+	let mut globaluserdata_prefix = vec![0xFF];
+	globaluserdata_prefix.extend_from_slice(&userid_prefix);
+
+	futures.push(
+		self.db
+			.roomusertype_roomuserdataid
+			.watch_prefix(&globaluserdata_prefix),
+	);
+
+	// More key changes (used when user is not joined to any rooms)
+	futures.push(self.db.keychangeid_userid.watch_prefix(&userid_prefix));
+
+	// One time keys
+	futures.push(
+		self.db
+			.userid_lastonetimekeyupdate
+			.watch_prefix(&userid_bytes),
+	);
+
+	// Server shutdown
+	let server_shutdown = async move {
+		while self.services.server.running() {
+			self.services.server.signal.subscribe().recv().await.ok();
+		}
+	};
+
+	futures.push(server_shutdown.boxed());
+	if !self.services.server.running() {
+		return Ok(());
+	}
+
+	// Wait until one of them finds something
+	trace!(futures = futures.len(), "watch started");
+	futures.next().await;
+	trace!(futures = futures.len(), "watch finished");
+
+	Ok(())
+}