1mod updates;
16
17use std::{cmp::Ordering, collections::BTreeSet, fmt, sync::Arc};
18
19use eyeball_im::VectorDiff;
20use futures_util::{StreamExt as _, stream};
21use matrix_sdk_base::{
22 apply_redaction,
23 event_cache::{Event, Gap},
24 linked_chunk::{LinkedChunkId, OwnedLinkedChunkId, Position, Update},
25 serde_helpers::{extract_redaction_target, extract_relation},
26 sync::Timeline,
27 task_monitor::BackgroundTaskHandle,
28};
29use matrix_sdk_common::executor::spawn;
30use ruma::{
31 EventId, MilliSecondsSinceUnixEpoch, OwnedEventId, OwnedRoomId, OwnedUserId, RoomId,
32 events::{relation::RelationType, room::redaction::SyncRoomRedactionEvent},
33 room_version_rules::RoomVersionRules,
34};
35use tokio::sync::broadcast::{Receiver, Sender};
36use tracing::{debug, instrument, trace, warn};
37
38pub(super) use self::updates::PinnedEventsCacheUpdateSender;
39#[cfg(feature = "e2e-encryption")]
40use super::super::redecryptor::MaybeResolvedEvent;
41use super::{
42 super::{
43 EventCacheError, EventsOrigin, Result,
44 deduplicator::{DeduplicationOutcome, filter_duplicate_events},
45 persistence::{find_event, send_updates_to_store},
46 states::{
47 CacheStateLock, ReloadPreprocessing, StateLock, StateLockWriteGuard,
48 selectors::PinnedEventsStateSelector,
49 },
50 },
51 EventLocation, TimelineVectorDiffs,
52 event_linked_chunk::{EventLinkedChunk, sort_positions_descending},
53 room::RoomEventCacheLinkedChunkUpdate,
54};
55use crate::{Room, client::WeakClient, config::RequestConfig, room::WeakRoom};
56
57pub struct PinnedEventsCacheState {
58 room_id: OwnedRoomId,
60
61 own_user_id: OwnedUserId,
63
64 room_version_rules: RoomVersionRules,
66
67 chunk: EventLinkedChunk,
74
75 pub update_sender: PinnedEventsCacheUpdateSender,
77
78 linked_chunk_update_sender: Sender<RoomEventCacheLinkedChunkUpdate>,
83}
84
85#[cfg(not(tarpaulin_include))]
86impl fmt::Debug for PinnedEventsCacheState {
87 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
88 f.debug_struct("PinnedEventsCacheState")
89 .field("room_id", &self.room_id)
90 .field("chunk", &self.chunk)
91 .finish_non_exhaustive()
92 }
93}
94
95impl<'a> StateLockWriteGuard<'a, PinnedEventsCacheState> {
96 #[must_use = "Propagate `VectorDiff` updates via `TimelineVectorDiffs`"]
102 pub async fn reload(
103 &mut self,
104 preprocessing: ReloadPreprocessing,
105 ) -> Result<Vec<VectorDiff<Event>>> {
106 match preprocessing {
107 ReloadPreprocessing::ForgetAll => {
108 self.state.chunk.reset();
111 self.propagate_changes().await?;
112 }
113
114 ReloadPreprocessing::None => {}
115 }
116
117 self.reload_from_storage().await?;
120
121 Ok(self.state.chunk.updates_as_vector_diffs())
122 }
123
124 async fn handle_sync(&mut self, timeline: Timeline) -> Result<()> {
125 let DeduplicationOutcome {
126 all_events: events,
127 in_memory_duplicated_event_ids,
128 in_store_duplicated_event_ids,
129 non_empty_all_duplicates: all_duplicates,
130 } = filter_duplicate_events(
131 &self.state.own_user_id,
132 &self.store,
133 LinkedChunkId::PinnedEvents(&self.state.room_id),
134 &self.state.chunk,
135 timeline.events,
136 )
137 .await?;
138
139 if all_duplicates {
140 return Ok(());
143 }
144
145 self.remove_events(in_memory_duplicated_event_ids, in_store_duplicated_event_ids).await?;
151
152 self.state.chunk.push_live_events(None, &events);
154
155 self.propagate_changes().await?;
156 self.notify_subscribers(EventsOrigin::Sync);
157
158 for event in &events {
160 self.maybe_apply_new_redaction(event).await?;
162 }
163
164 Ok(())
165 }
166
167 #[instrument(skip_all)]
172 pub async fn remove_events(
173 &mut self,
174 in_memory_events: Vec<(OwnedEventId, Position)>,
175 in_store_events: Vec<(OwnedEventId, Position)>,
176 ) -> Result<()> {
177 if !in_store_events.is_empty() {
179 let mut positions = in_store_events
180 .into_iter()
181 .map(|(_event_id, position)| position)
182 .collect::<Vec<_>>();
183
184 sort_positions_descending(&mut positions);
185
186 let updates =
187 positions.into_iter().map(|pos| Update::RemoveItem { at: pos }).collect::<Vec<_>>();
188
189 self.apply_store_only_updates(updates).await?;
190 }
191
192 if in_memory_events.is_empty() {
194 return Ok(());
196 }
197
198 self.state
200 .chunk
201 .remove_events_by_position(
202 in_memory_events.into_iter().map(|(_event_id, position)| position).collect(),
203 )
204 .expect("failed to remove an event");
205
206 self.propagate_changes().await
207 }
208
209 async fn apply_store_only_updates(&mut self, updates: Vec<Update<Event, Gap>>) -> Result<()> {
215 self.send_updates_to_store(updates).await
216 }
217
218 #[instrument(skip_all)]
221 async fn maybe_apply_new_redaction(&mut self, event: &Event) -> Result<()> {
222 let Some(event_id) =
223 extract_redaction_target(event.raw(), &self.room_version_rules.redaction)
224 else {
225 return Ok(());
226 };
227
228 let Some((location, mut target_event)) = self.find_event(&event_id).await? else {
230 trace!("redacted event is missing from the linked chunk");
231 return Ok(());
232 };
233
234 let target_event_raw = target_event.raw();
235
236 if let Ok(deserialized) = target_event_raw.deserialize()
238 && deserialized.is_redacted()
239 {
240 return Ok(());
241 }
242
243 if let Some(redacted_event) = apply_redaction(
244 target_event_raw,
245 event.raw().cast_ref_unchecked::<SyncRoomRedactionEvent>(),
246 &self.room_version_rules.redaction,
247 ) {
248 target_event.replace_raw(redacted_event.cast_unchecked());
256
257 self.replace_event_at(location, target_event.clone()).await?;
258 }
259
260 Ok(())
261 }
262
263 pub(super) async fn find_event(
265 &self,
266 event_id: &EventId,
267 ) -> Result<Option<(EventLocation, Event)>> {
268 find_event(event_id, &self.room_id, &self.chunk, &self.store).await
269 }
270
271 pub async fn replace_event_at(
278 &mut self,
279 location: EventLocation,
280 new_event: Event,
281 ) -> Result<()> {
282 match location {
283 EventLocation::Memory(position) => {
284 self.state
285 .chunk
286 .replace_event_at(position, new_event)
287 .expect("should have been a valid position of an item");
288 self.propagate_changes().await?;
291 }
292 EventLocation::Store => {
293 self.save_events([new_event]).await?;
294 }
295 }
296
297 Ok(())
298 }
299
300 pub async fn save_events(&mut self, events: impl IntoIterator<Item = Event>) -> Result<()> {
302 let store = self.store.clone();
303 let room_id = self.state.room_id.clone();
304 let events = events.into_iter().collect::<Vec<_>>();
305
306 spawn(async move {
308 for event in events {
309 store.save_event(&room_id, event).await?;
310 }
311
312 Result::Ok(())
313 })
314 .await
315 .expect("joining failed")?;
316
317 Ok(())
318 }
319
320 async fn reload_from_storage(&mut self) -> Result<()> {
323 let room_id = self.state.room_id.clone();
324 let linked_chunk_id = LinkedChunkId::PinnedEvents(&room_id);
325
326 let (last_chunk, chunk_id_gen) = self.store.load_last_chunk(linked_chunk_id).await?;
327
328 let Some(last_chunk) = last_chunk else {
329 if self.state.chunk.events().next().is_some() {
332 self.state.chunk.reset();
333 self.notify_subscribers(EventsOrigin::Sync);
334 }
335
336 return Ok(());
337 };
338
339 {
340 let mut current_chunk_identifier = last_chunk.identifier;
341 self.state.chunk.shrink_to_last_reloaded_chunk(
342 Some(last_chunk),
343 chunk_id_gen,
344 None,
346 )?;
347
348 while let Some(previous_chunk) =
350 self.store.load_previous_chunk(linked_chunk_id, current_chunk_identifier).await?
351 {
352 current_chunk_identifier = previous_chunk.identifier;
353 self.state.chunk.insert_new_chunk_as_first(previous_chunk)?;
354 }
355 }
356
357 self.state.chunk.store_updates().take();
359
360 self.notify_subscribers(EventsOrigin::Cache);
362
363 Ok(())
364 }
365
366 async fn replace_all_events(&mut self, new_events: Vec<Event>) -> Result<()> {
367 trace!("resetting all pinned events in linked chunk");
368
369 let previous_pinned_event_ids = self.state.current_event_ids();
370
371 if new_events
372 .iter()
373 .filter_map(|e| e.event_id())
374 .map(ToOwned::to_owned)
375 .collect::<BTreeSet<_>>()
376 == previous_pinned_event_ids.into_iter().collect()
377 {
378 return Ok(());
380 }
381
382 if self.state.chunk.events().next().is_some() {
383 self.state.chunk.reset();
384 }
385
386 self.state.chunk.push_live_events(None, &new_events);
387 self.propagate_changes().await?;
388 self.notify_subscribers(EventsOrigin::Sync);
389
390 Ok(())
391 }
392
393 pub async fn propagate_changes(&mut self) -> Result<()> {
396 let updates = self.state.chunk.store_updates().take();
397
398 self.send_updates_to_store(updates).await
399 }
400
401 async fn send_updates_to_store(&mut self, updates: Vec<Update<Event, Gap>>) -> Result<()> {
402 let linked_chunk_id = OwnedLinkedChunkId::PinnedEvents(self.room_id.clone());
403
404 send_updates_to_store(
405 &self.store,
406 linked_chunk_id,
407 &self.state.linked_chunk_update_sender,
408 updates,
409 )
410 .await
411 }
412
413 fn notify_subscribers(&mut self, origin: EventsOrigin) {
415 let diffs = self.state.chunk.updates_as_vector_diffs();
416
417 if !diffs.is_empty() {
418 self.update_sender.send(TimelineVectorDiffs { diffs, origin });
419 }
420 }
421}
422
423impl PinnedEventsCacheState {
424 pub(super) fn current_event_ids(&self) -> Vec<OwnedEventId> {
426 self.chunk
427 .events()
428 .filter_map(|(_position, event)| event.event_id().map(ToOwned::to_owned))
429 .collect()
430 }
431
432 fn has_exactly_pinned_events(&self, pinned_event_ids: &[OwnedEventId]) -> bool {
441 let pinned_event_ids: BTreeSet<&EventId> =
442 pinned_event_ids.iter().map(|event_id| &**event_id).collect();
443 let event_ids: BTreeSet<&EventId> =
444 self.chunk.events().filter_map(|(_position, event)| event.event_id()).collect();
445
446 if !pinned_event_ids.is_subset(&event_ids) {
447 return false;
448 }
449
450 self.chunk.events().all(|(_position, event)| {
454 event.event_id().is_some_and(|event_id| pinned_event_ids.contains(event_id))
455 || extract_relation(event.raw()).is_some_and(|(relation_type, related_event_id)| {
456 relation_type != RelationType::Thread && event_ids.contains(&*related_event_id)
457 })
458 || extract_redaction_target(event.raw(), &self.room_version_rules.redaction)
459 .is_some_and(|redacted_event_id| event_ids.contains(&*redacted_event_id))
460 })
461 }
462}
463
464#[derive(Clone)]
468pub struct PinnedEventsCache {
469 inner: Arc<PinnedEventsCacheInner>,
470
471 _task: Arc<BackgroundTaskHandle>,
474}
475
476struct PinnedEventsCacheInner {
478 room_id: OwnedRoomId,
480
481 state: CacheStateLock<PinnedEventsStateSelector>,
485}
486
487impl PinnedEventsCache {
488 pub(in super::super) async fn new(
490 weak_room: &WeakRoom,
491 own_user_id: OwnedUserId,
492 room_version_rules: RoomVersionRules,
493 linked_chunk_update_sender: Sender<RoomEventCacheLinkedChunkUpdate>,
494 state: &StateLock,
495 ) -> Result<Self> {
496 let room = weak_room.get().ok_or(EventCacheError::ClientDropped)?;
497 let room_id = room.room_id().to_owned();
498
499 let cache_state = state
500 .try_insert_once_with(
501 PinnedEventsStateSelector::new(room_id.clone()),
502 |_store_guard| async {
503 Ok(PinnedEventsCacheState {
504 room_id: room_id.clone(),
505 own_user_id,
506 room_version_rules,
507 chunk: EventLinkedChunk::new(),
508 update_sender: PinnedEventsCacheUpdateSender::new(),
509 linked_chunk_update_sender,
510 })
511 },
512 )
513 .await?;
514
515 let inner = Arc::new(PinnedEventsCacheInner { room_id, state: cache_state });
516
517 let task = room
518 .client()
519 .task_monitor()
520 .spawn_infinite_task(
521 "pinned_event_listener_task",
522 Self::pinned_event_listener_task(room, inner.clone()),
523 )
524 .abort_on_drop();
525
526 Ok(Self { inner, _task: Arc::new(task) })
527 }
528
529 pub fn room_id(&self) -> &RoomId {
531 &self.inner.room_id
532 }
533
534 pub(super) fn state(&self) -> &CacheStateLock<PinnedEventsStateSelector> {
536 &self.inner.state
537 }
538
539 pub async fn subscribe(&self) -> Result<(Vec<Event>, Receiver<TimelineVectorDiffs>)> {
541 let guard = self.inner.state.read().await?;
542 let events = guard.state.chunk.events().map(|(_position, item)| item.clone()).collect();
543
544 let recv = guard.state.update_sender.new_pinned_events_receiver();
545
546 Ok((events, recv))
547 }
548
549 #[cfg(feature = "e2e-encryption")]
553 pub(in super::super) async fn replace_in_memory_utds(
554 &self,
555 resolved_events: &[MaybeResolvedEvent],
556 ) -> Result<()> {
557 let mut state = self.inner.state.write().await?;
558
559 let _ = state.state.chunk.store_updates().take();
562
563 if state.state.chunk.replace_utds(resolved_events) {
564 state.propagate_changes().await?;
565 state.notify_subscribers(EventsOrigin::Cache);
566 }
567
568 Ok(())
569 }
570
571 #[instrument(skip_all, fields(room_id = %self.inner.room_id))]
573 pub(super) async fn handle_joined_room_update(&self, timeline: Timeline) -> Result<()> {
574 self.handle_timeline(timeline).await
575 }
576
577 #[instrument(skip_all, fields(room_id = %self.inner.room_id))]
579 pub(super) async fn handle_left_room_update(&self, timeline: Timeline) -> Result<()> {
580 self.handle_timeline(timeline).await
581 }
582
583 async fn handle_timeline(&self, timeline: Timeline) -> Result<()> {
586 if timeline.events.is_empty() {
587 return Ok(());
588 }
589
590 trace!("adding new {} events", timeline.events.len());
591
592 self.inner.state.write().await?.handle_sync(timeline).await
593 }
594
595 #[instrument(fields(%room_id = room.room_id()), skip(room, inner))]
596 async fn pinned_event_listener_task(room: Room, inner: Arc<PinnedEventsCacheInner>) {
597 debug!("pinned events listener task started");
598
599 let reload_from_network = async |room: Room| {
600 let events = match Self::reload_pinned_events(room).await {
601 Ok(Some(events)) => events,
602 Ok(None) => Vec::new(),
603 Err(err) => {
604 warn!("error when loading pinned events: {err}");
605 return;
606 }
607 };
608
609 match inner.state.write().await {
612 Ok(mut guard) => {
613 guard.replace_all_events(events).await.unwrap_or_else(|err| {
614 warn!("error when replacing pinned events: {err}");
615 });
616 }
617
618 Err(err) => {
619 warn!("error when acquiring write lock to replace pinned events: {err}");
620 }
621 }
622 };
623
624 match inner.state.write().await {
626 Ok(mut guard) => {
627 guard.reload_from_storage().await.unwrap_or_else(|err| {
629 warn!("error when reloading pinned events from storage, at start: {err}");
630 });
631
632 let actual_pinned_events =
635 pinned_event_ids_to_load(&room, room.pinned_event_ids().unwrap_or_default());
636
637 if !guard.state.has_exactly_pinned_events(&actual_pinned_events) {
638 drop(guard);
640 reload_from_network(room.clone()).await;
641 }
642 }
643
644 Err(err) => {
645 warn!("error when acquiring write lock to initialize pinned events: {err}");
646 }
647 }
648
649 let weak_room =
650 WeakRoom::new(WeakClient::from_client(&room.client()), room.room_id().to_owned());
651
652 let mut stream = room.pinned_event_ids_stream();
653
654 drop(room);
655
656 while let Some(new_list) = stream.next().await {
658 trace!("handling update");
659
660 let Some(room) = weak_room.get() else {
661 debug!("room has been dropped, ending pinned events listener task");
662 break;
663 };
664
665 let new_list = pinned_event_ids_to_load(&room, new_list);
666
667 let guard = match inner.state.read().await {
668 Ok(guard) => guard,
669 Err(err) => {
670 warn!("error when acquiring read lock to handle pinned events update: {err}");
671 break;
672 }
673 };
674
675 if guard.state.has_exactly_pinned_events(&new_list) {
677 continue;
679 }
680
681 drop(guard);
682
683 reload_from_network(room).await;
685 }
686
687 debug!("pinned events listener task ended");
688 }
689
690 async fn reload_pinned_events(room: Room) -> Result<Option<Vec<Event>>> {
699 let max_concurrent_requests =
700 room.client().event_cache().config().max_pinned_events_concurrent_requests;
701
702 let pinned_event_ids =
703 pinned_event_ids_to_load(&room, room.pinned_event_ids().unwrap_or_default());
704
705 if pinned_event_ids.is_empty() {
706 return Ok(Some(Vec::new()));
707 }
708
709 let mut num_successful_loads = 0;
710
711 let mut loaded_events: Vec<Event> =
712 stream::iter(pinned_event_ids.clone().into_iter().map(|event_id| {
713 let room = room.clone();
714 let filter = vec![RelationType::Annotation, RelationType::Replacement];
715 let request_config = RequestConfig::default().retry_limit(3);
716
717 async move {
718 let (target, mut relations) = room
719 .load_or_fetch_event_with_relations(
720 &event_id,
721 Some(filter),
722 Some(request_config),
723 )
724 .await?;
725
726 relations.insert(0, target);
727 Ok::<_, crate::Error>(relations)
728 }
729 }))
730 .buffer_unordered(max_concurrent_requests)
731 .inspect(|result| {
733 if result.is_ok() {
734 num_successful_loads += 1;
735 }
736 })
737 .flat_map(stream::iter)
739 .flat_map(stream::iter)
741 .collect()
742 .await;
743
744 if num_successful_loads != pinned_event_ids.len() {
745 warn!(
746 "only successfully loaded {} out of {} pinned events",
747 num_successful_loads,
748 pinned_event_ids.len()
749 );
750 }
751
752 if loaded_events.is_empty() {
753 return Err(EventCacheError::UnableToLoadPinnedEvents);
757 }
758
759 loaded_events.sort_by(compare_pinned_items);
764
765 Ok(Some(loaded_events))
766 }
767}
768
769impl fmt::Debug for PinnedEventsCache {
770 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
771 f.debug_struct("PinnedEventsCache").finish_non_exhaustive()
772 }
773}
774
775fn pinned_event_ids_to_load(room: &Room, pinned_event_ids: Vec<OwnedEventId>) -> Vec<OwnedEventId> {
781 let max_events_to_load = room.client().event_cache().config().max_pinned_events_to_load;
782 pinned_event_ids.into_iter().rev().take(max_events_to_load).rev().collect()
783}
784
785fn compare_pinned_items(a: &Event, b: &Event) -> Ordering {
786 let a_time: Option<MilliSecondsSinceUnixEpoch> = a.timestamp_raw();
787 let b_time: Option<MilliSecondsSinceUnixEpoch> = b.timestamp_raw();
788
789 compare_by_optional_timestamp(a_time, b_time)
790}
791
792fn compare_by_optional_timestamp(
793 a: Option<MilliSecondsSinceUnixEpoch>,
794 b: Option<MilliSecondsSinceUnixEpoch>,
795) -> Ordering {
796 match (a, b) {
797 (None, None) => Ordering::Equal,
798 (None, Some(_)) => Ordering::Greater,
799 (Some(_), None) => Ordering::Less,
800 (Some(a), Some(b)) => a.cmp(&b),
801 }
802}
803
804#[cfg(not(target_family = "wasm"))]
805#[cfg(test)]
806mod tests {
807 use proptest::prelude::*;
808 use ruma::UInt;
809
810 use super::*;
811
812 fn any_timestamp() -> impl Strategy<Value = Option<MilliSecondsSinceUnixEpoch>> {
813 prop::option::of(
814 any::<u32>().prop_map(|value| MilliSecondsSinceUnixEpoch(UInt::from(value))),
815 )
816 }
817
818 #[test]
819 fn sort_pinned_events_never_panics_only_nones() {
820 let mut vec = vec![None; 100_000];
821 vec.sort_by(|a, b| compare_by_optional_timestamp(*a, *b))
822 }
823
824 proptest! {
825 #[test]
826 fn sort_pinned_events_never_panics(mut v in prop::collection::vec(any_timestamp(), 0..1000)) {
827 v.sort_by(
828 |a, b| compare_by_optional_timestamp(*a, *b))
829 }
830
831 #[test]
832 fn compare_pinned_events_reflexive(a in any_timestamp()) {
833 prop_assert_eq!(compare_by_optional_timestamp(a, a), Ordering::Equal);
834 }
835
836 #[test]
837 fn compare_pinned_events_antisymmetric(a in any_timestamp(), b in any_timestamp()) {
838 let ab = compare_by_optional_timestamp(a, b);
839 let ba = compare_by_optional_timestamp(b, a);
840
841 prop_assert_eq!(ab, ba.reverse());
842 }
843
844 #[test]
845 fn compare_pinned_events_transitive(
846 a in any_timestamp(),
847 b in any_timestamp(),
848 c in any_timestamp()
849 ) {
850 let ab = compare_by_optional_timestamp(a, b);
851 let bc = compare_by_optional_timestamp(b, c);
852 let ac = compare_by_optional_timestamp(a, c);
853
854 if ab == Ordering::Less && bc == Ordering::Less {
855 prop_assert_eq!(ac, Ordering::Less);
856 }
857
858 if ab == Ordering::Equal && bc == Ordering::Equal {
859 prop_assert_eq!(ac, Ordering::Equal);
860 }
861
862 if ab == Ordering::Greater && bc == Ordering::Greater {
863 prop_assert_eq!(ac, Ordering::Greater);
864 }
865 }
866 }
867}