use std::{
collections::{BTreeMap, HashSet},
sync::{Arc, atomic::AtomicBool},
};
use bitflags::bitflags;
use eyeball::Subscriber;
use matrix_sdk_common::{
ROOM_VERSION_FALLBACK, ROOM_VERSION_RULES_FALLBACK, deserialized_responses::TimelineEventKind,
};
use ruma::{
EventId, MilliSecondsSinceUnixEpoch, MxcUri, OwnedEventId, OwnedMxcUri, OwnedRoomAliasId,
OwnedRoomId, OwnedUserId, RoomAliasId, RoomId, RoomVersionId,
api::client::sync::sync_events::v3::RoomSummary as RumaSummary,
assign,
events::{
AnyStrippedStateEvent, AnySyncStateEvent, AnySyncTimelineEvent, StateEventType,
SyncStateEvent,
beacon_info::BeaconInfoEventContent,
call::member::{CallMemberEventContent, CallMemberStateKey, MembershipData},
direct::OwnedDirectUserIdentifier,
room::{
avatar::{self, RoomAvatarEventContent},
canonical_alias::RoomCanonicalAliasEventContent,
encryption::RoomEncryptionEventContent,
guest_access::{GuestAccess, RoomGuestAccessEventContent},
history_visibility::{HistoryVisibility, RoomHistoryVisibilityEventContent},
join_rules::{JoinRule, RoomJoinRulesEventContent},
name::RoomNameEventContent,
pinned_events::RoomPinnedEventsEventContent,
redaction::SyncRoomRedactionEvent,
tombstone::RoomTombstoneEventContent,
topic::RoomTopicEventContent,
},
tag::{TagEventContent, TagName, Tags},
},
room::RoomType,
room_version_rules::{AuthorizationRules, RedactionRules, RoomVersionRules},
serde::Raw,
};
use serde::{Deserialize, Serialize};
use tracing::{debug, error, field::debug, info, instrument, warn};
use super::{
AccountDataSource, EncryptionState, Room, RoomCreateWithCreatorEventContent, RoomDisplayName,
RoomHero, RoomNotableTags, RoomState, RoomSummary,
};
use crate::{
MinimalStateEvent, OriginalMinimalStateEvent,
deserialized_responses::RawSyncOrStrippedState,
latest_event::{LatestEvent, LatestEventValue},
notification_settings::RoomNotificationMode,
read_receipts::RoomReadReceipts,
store::{DynStateStore, StateStoreExt},
sync::UnreadNotificationsCount,
};
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct InviteAcceptanceDetails {
pub invite_accepted_at: MilliSecondsSinceUnixEpoch,
pub inviter: OwnedUserId,
}
impl Room {
pub fn subscribe_info(&self) -> Subscriber<RoomInfo> {
self.info.subscribe()
}
pub fn clone_info(&self) -> RoomInfo {
self.info.get()
}
pub fn set_room_info(
&self,
room_info: RoomInfo,
room_info_notable_update_reasons: RoomInfoNotableUpdateReasons,
) {
self.info.set(room_info);
if !room_info_notable_update_reasons.is_empty() {
let _ = self.room_info_notable_update_sender.send(RoomInfoNotableUpdate {
room_id: self.room_id.clone(),
reasons: room_info_notable_update_reasons,
});
} else {
let _ = self.room_info_notable_update_sender.send(RoomInfoNotableUpdate {
room_id: self.room_id.clone(),
reasons: RoomInfoNotableUpdateReasons::NONE,
});
}
}
}
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct BaseRoomInfo {
pub(crate) avatar: Option<MinimalStateEvent<RoomAvatarEventContent>>,
#[serde(skip_serializing_if = "BTreeMap::is_empty", default)]
pub(crate) beacons: BTreeMap<OwnedUserId, MinimalStateEvent<BeaconInfoEventContent>>,
pub(crate) canonical_alias: Option<MinimalStateEvent<RoomCanonicalAliasEventContent>>,
pub(crate) create: Option<MinimalStateEvent<RoomCreateWithCreatorEventContent>>,
pub(crate) dm_targets: HashSet<OwnedDirectUserIdentifier>,
pub(crate) encryption: Option<RoomEncryptionEventContent>,
pub(crate) guest_access: Option<MinimalStateEvent<RoomGuestAccessEventContent>>,
pub(crate) history_visibility: Option<MinimalStateEvent<RoomHistoryVisibilityEventContent>>,
pub(crate) join_rules: Option<MinimalStateEvent<RoomJoinRulesEventContent>>,
pub(crate) max_power_level: i64,
pub(crate) name: Option<MinimalStateEvent<RoomNameEventContent>>,
pub(crate) tombstone: Option<MinimalStateEvent<RoomTombstoneEventContent>>,
pub(crate) topic: Option<MinimalStateEvent<RoomTopicEventContent>>,
#[serde(skip_serializing_if = "BTreeMap::is_empty", default)]
pub(crate) rtc_member_events:
BTreeMap<CallMemberStateKey, MinimalStateEvent<CallMemberEventContent>>,
#[serde(default)]
pub(crate) is_marked_unread: bool,
#[serde(default)]
pub(crate) is_marked_unread_source: AccountDataSource,
#[serde(skip_serializing_if = "RoomNotableTags::is_empty", default)]
pub(crate) notable_tags: RoomNotableTags,
pub(crate) pinned_events: Option<RoomPinnedEventsEventContent>,
}
impl BaseRoomInfo {
pub fn new() -> Self {
Self::default()
}
pub fn room_version(&self) -> Option<&RoomVersionId> {
match self.create.as_ref()? {
MinimalStateEvent::Original(ev) => Some(&ev.content.room_version),
MinimalStateEvent::Redacted(ev) => Some(&ev.content.room_version),
}
}
pub fn handle_state_event(&mut self, ev: &AnySyncStateEvent) -> bool {
match ev {
AnySyncStateEvent::BeaconInfo(b) => {
self.beacons.insert(b.state_key().clone(), b.into());
}
AnySyncStateEvent::RoomEncryption(SyncStateEvent::Original(encryption)) => {
self.encryption = Some(encryption.content.clone());
}
AnySyncStateEvent::RoomAvatar(a) => {
self.avatar = Some(a.into());
}
AnySyncStateEvent::RoomName(n) => {
self.name = Some(n.into());
}
AnySyncStateEvent::RoomCreate(c) if self.create.is_none() => {
self.create = Some(c.into());
}
AnySyncStateEvent::RoomHistoryVisibility(h) => {
self.history_visibility = Some(h.into());
}
AnySyncStateEvent::RoomGuestAccess(g) => {
self.guest_access = Some(g.into());
}
AnySyncStateEvent::RoomJoinRules(c) => match c.join_rule() {
JoinRule::Invite
| JoinRule::Knock
| JoinRule::Private
| JoinRule::Restricted(_)
| JoinRule::KnockRestricted(_)
| JoinRule::Public => self.join_rules = Some(c.into()),
r => warn!("Encountered a custom join rule {}, skipping", r.as_str()),
},
AnySyncStateEvent::RoomCanonicalAlias(a) => {
self.canonical_alias = Some(a.into());
}
AnySyncStateEvent::RoomTopic(t) => {
self.topic = Some(t.into());
}
AnySyncStateEvent::RoomTombstone(t) => {
self.tombstone = Some(t.into());
}
AnySyncStateEvent::RoomPowerLevels(p) => {
self.max_power_level = p.power_levels(&AuthorizationRules::V1, vec![]).max().into();
}
AnySyncStateEvent::CallMember(m) => {
let Some(o_ev) = m.as_original() else {
return false;
};
let mut o_ev = o_ev.clone();
o_ev.content.set_created_ts_if_none(o_ev.origin_server_ts);
self.rtc_member_events
.insert(m.state_key().clone(), SyncStateEvent::Original(o_ev).into());
self.rtc_member_events.retain(|_, ev| {
ev.as_original().is_some_and(|o| !o.content.active_memberships(None).is_empty())
});
}
AnySyncStateEvent::RoomPinnedEvents(p) => {
self.pinned_events = p.as_original().map(|p| p.content.clone());
}
_ => return false,
}
true
}
pub fn handle_stripped_state_event(&mut self, ev: &AnyStrippedStateEvent) -> bool {
match ev {
AnyStrippedStateEvent::RoomEncryption(encryption) => {
if let Some(algorithm) = &encryption.content.algorithm {
let content = assign!(RoomEncryptionEventContent::new(algorithm.clone()), {
rotation_period_ms: encryption.content.rotation_period_ms,
rotation_period_msgs: encryption.content.rotation_period_msgs,
});
self.encryption = Some(content);
}
}
AnyStrippedStateEvent::RoomAvatar(a) => {
self.avatar = Some(a.into());
}
AnyStrippedStateEvent::RoomName(n) => {
self.name = Some(n.into());
}
AnyStrippedStateEvent::RoomCreate(c) if self.create.is_none() => {
self.create = Some(c.into());
}
AnyStrippedStateEvent::RoomHistoryVisibility(h) => {
self.history_visibility = Some(h.into());
}
AnyStrippedStateEvent::RoomGuestAccess(g) => {
self.guest_access = Some(g.into());
}
AnyStrippedStateEvent::RoomJoinRules(c) => match &c.content.join_rule {
JoinRule::Invite
| JoinRule::Knock
| JoinRule::Private
| JoinRule::Restricted(_)
| JoinRule::KnockRestricted(_)
| JoinRule::Public => self.join_rules = Some(c.into()),
r => warn!("Encountered a custom join rule {}, skipping", r.as_str()),
},
AnyStrippedStateEvent::RoomCanonicalAlias(a) => {
self.canonical_alias = Some(a.into());
}
AnyStrippedStateEvent::RoomTopic(t) => {
self.topic = Some(t.into());
}
AnyStrippedStateEvent::RoomTombstone(t) => {
self.tombstone = Some(t.into());
}
AnyStrippedStateEvent::RoomPowerLevels(p) => {
self.max_power_level = p.power_levels(&AuthorizationRules::V1, vec![]).max().into();
}
AnyStrippedStateEvent::CallMember(_) => {
return false;
}
AnyStrippedStateEvent::RoomPinnedEvents(p) => {
if let Some(pinned) = p.content.pinned.clone() {
self.pinned_events = Some(RoomPinnedEventsEventContent::new(pinned));
}
}
_ => return false,
}
true
}
pub(super) fn handle_redaction(&mut self, redacts: &EventId) {
let redaction_rules = self
.room_version()
.and_then(|room_version| room_version.rules())
.unwrap_or(ROOM_VERSION_RULES_FALLBACK)
.redaction;
if let Some(ev) = &mut self.avatar
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.canonical_alias
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.create
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.guest_access
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.history_visibility
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.join_rules
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.name
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.tombstone
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else if let Some(ev) = &mut self.topic
&& ev.event_id() == Some(redacts)
{
ev.redact(&redaction_rules);
} else {
self.rtc_member_events
.retain(|_, member_event| member_event.event_id() != Some(redacts));
}
}
pub fn handle_notable_tags(&mut self, tags: &Tags) {
let mut notable_tags = RoomNotableTags::empty();
if tags.contains_key(&TagName::Favorite) {
notable_tags.insert(RoomNotableTags::FAVOURITE);
}
if tags.contains_key(&TagName::LowPriority) {
notable_tags.insert(RoomNotableTags::LOW_PRIORITY);
}
self.notable_tags = notable_tags;
}
}
impl Default for BaseRoomInfo {
fn default() -> Self {
Self {
avatar: None,
beacons: BTreeMap::new(),
canonical_alias: None,
create: None,
dm_targets: Default::default(),
encryption: None,
guest_access: None,
history_visibility: None,
join_rules: None,
max_power_level: 100,
name: None,
tombstone: None,
topic: None,
rtc_member_events: BTreeMap::new(),
is_marked_unread: false,
is_marked_unread_source: AccountDataSource::Unstable,
notable_tags: RoomNotableTags::empty(),
pinned_events: None,
}
}
}
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct RoomInfo {
#[serde(default, alias = "version")]
pub(crate) data_format_version: u8,
pub(crate) room_id: OwnedRoomId,
pub(crate) room_state: RoomState,
pub(crate) notification_counts: UnreadNotificationsCount,
pub(crate) summary: RoomSummary,
pub(crate) members_synced: bool,
pub(crate) last_prev_batch: Option<String>,
pub(crate) sync_info: SyncInfo,
pub(crate) encryption_state_synced: bool,
pub(crate) latest_event: Option<Box<LatestEvent>>,
#[serde(default)]
pub(crate) new_latest_event: LatestEventValue,
#[serde(default)]
pub(crate) read_receipts: RoomReadReceipts,
pub(crate) base_info: Box<BaseRoomInfo>,
#[serde(skip)]
pub(crate) warned_about_unknown_room_version_rules: Arc<AtomicBool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub(crate) cached_display_name: Option<RoomDisplayName>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub(crate) cached_user_defined_notification_mode: Option<RoomNotificationMode>,
#[serde(default)]
pub(crate) recency_stamp: Option<RoomRecencyStamp>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub(crate) invite_acceptance_details: Option<InviteAcceptanceDetails>,
}
impl RoomInfo {
#[doc(hidden)] pub fn new(room_id: &RoomId, room_state: RoomState) -> Self {
Self {
data_format_version: 1,
room_id: room_id.into(),
room_state,
notification_counts: Default::default(),
summary: Default::default(),
members_synced: false,
last_prev_batch: None,
sync_info: SyncInfo::NoState,
encryption_state_synced: false,
latest_event: None,
new_latest_event: LatestEventValue::default(),
read_receipts: Default::default(),
base_info: Box::new(BaseRoomInfo::new()),
warned_about_unknown_room_version_rules: Arc::new(false.into()),
cached_display_name: None,
cached_user_defined_notification_mode: None,
recency_stamp: None,
invite_acceptance_details: None,
}
}
pub fn mark_as_joined(&mut self) {
self.set_state(RoomState::Joined);
}
pub fn mark_as_left(&mut self) {
self.set_state(RoomState::Left);
}
pub fn mark_as_invited(&mut self) {
self.set_state(RoomState::Invited);
}
pub fn mark_as_knocked(&mut self) {
self.set_state(RoomState::Knocked);
}
pub fn mark_as_banned(&mut self) {
self.set_state(RoomState::Banned);
}
pub fn set_state(&mut self, room_state: RoomState) {
if self.state() != RoomState::Joined && self.invite_acceptance_details.is_some() {
error!(room_id = %self.room_id, "The RoomInfo contains invite acceptance details but the room is not in the joined state");
}
self.invite_acceptance_details = None;
self.room_state = room_state;
}
pub fn mark_members_synced(&mut self) {
self.members_synced = true;
}
pub fn mark_members_missing(&mut self) {
self.members_synced = false;
}
pub fn are_members_synced(&self) -> bool {
self.members_synced
}
pub fn mark_state_partially_synced(&mut self) {
self.sync_info = SyncInfo::PartiallySynced;
}
pub fn mark_state_fully_synced(&mut self) {
self.sync_info = SyncInfo::FullySynced;
}
pub fn mark_state_not_synced(&mut self) {
self.sync_info = SyncInfo::NoState;
}
pub fn mark_encryption_state_synced(&mut self) {
self.encryption_state_synced = true;
}
pub fn mark_encryption_state_missing(&mut self) {
self.encryption_state_synced = false;
}
pub fn set_prev_batch(&mut self, prev_batch: Option<&str>) -> bool {
if self.last_prev_batch.as_deref() != prev_batch {
self.last_prev_batch = prev_batch.map(|p| p.to_owned());
true
} else {
false
}
}
pub fn state(&self) -> RoomState {
self.room_state
}
#[cfg(not(feature = "experimental-encrypted-state-events"))]
pub fn encryption_state(&self) -> EncryptionState {
if !self.encryption_state_synced {
EncryptionState::Unknown
} else if self.base_info.encryption.is_some() {
EncryptionState::Encrypted
} else {
EncryptionState::NotEncrypted
}
}
#[cfg(feature = "experimental-encrypted-state-events")]
pub fn encryption_state(&self) -> EncryptionState {
if !self.encryption_state_synced {
EncryptionState::Unknown
} else {
self.base_info
.encryption
.as_ref()
.map(|state| {
if state.encrypt_state_events {
EncryptionState::StateEncrypted
} else {
EncryptionState::Encrypted
}
})
.unwrap_or(EncryptionState::NotEncrypted)
}
}
pub fn set_encryption_event(&mut self, event: Option<RoomEncryptionEventContent>) {
self.base_info.encryption = event;
}
pub fn handle_encryption_state(
&mut self,
requested_required_states: &[(StateEventType, String)],
) {
if requested_required_states
.iter()
.any(|(state_event, _)| state_event == &StateEventType::RoomEncryption)
{
self.mark_encryption_state_synced();
}
}
pub fn handle_state_event(&mut self, event: &AnySyncStateEvent) -> bool {
let base_info_has_been_modified = self.base_info.handle_state_event(event);
if let AnySyncStateEvent::RoomEncryption(_) = event {
self.mark_encryption_state_synced();
}
base_info_has_been_modified
}
pub fn handle_stripped_state_event(&mut self, event: &AnyStrippedStateEvent) -> bool {
self.base_info.handle_stripped_state_event(event)
}
#[instrument(skip_all, fields(redacts))]
pub fn handle_redaction(
&mut self,
event: &SyncRoomRedactionEvent,
_raw: &Raw<SyncRoomRedactionEvent>,
) {
let redaction_rules = self.room_version_rules_or_default().redaction;
let Some(redacts) = event.redacts(&redaction_rules) else {
info!("Can't apply redaction, redacts field is missing");
return;
};
tracing::Span::current().record("redacts", debug(redacts));
if let Some(latest_event) = &mut self.latest_event {
tracing::trace!("Checking if redaction applies to latest event");
if latest_event.event_id().as_deref() == Some(redacts) {
match apply_redaction(latest_event.event().raw(), _raw, &redaction_rules) {
Some(redacted) => {
latest_event.event_mut().kind =
TimelineEventKind::PlainText { event: redacted };
debug!("Redacted latest event");
}
None => {
self.latest_event = None;
debug!("Removed latest event");
}
}
}
}
self.base_info.handle_redaction(redacts);
}
pub fn avatar_url(&self) -> Option<&MxcUri> {
self.base_info
.avatar
.as_ref()
.and_then(|e| e.as_original().and_then(|e| e.content.url.as_deref()))
}
pub fn update_avatar(&mut self, url: Option<OwnedMxcUri>) {
self.base_info.avatar = url.map(|url| {
let mut content = RoomAvatarEventContent::new();
content.url = Some(url);
MinimalStateEvent::Original(OriginalMinimalStateEvent { content, event_id: None })
});
}
pub fn avatar_info(&self) -> Option<&avatar::ImageInfo> {
self.base_info
.avatar
.as_ref()
.and_then(|e| e.as_original().and_then(|e| e.content.info.as_deref()))
}
pub fn update_notification_count(&mut self, notification_counts: UnreadNotificationsCount) {
self.notification_counts = notification_counts;
}
pub fn update_from_ruma_summary(&mut self, summary: &RumaSummary) -> bool {
let mut changed = false;
if !summary.is_empty() {
if !summary.heroes.is_empty() {
self.summary.room_heroes = summary
.heroes
.iter()
.map(|hero_id| RoomHero {
user_id: hero_id.to_owned(),
display_name: None,
avatar_url: None,
})
.collect();
changed = true;
}
if let Some(joined) = summary.joined_member_count {
self.summary.joined_member_count = joined.into();
changed = true;
}
if let Some(invited) = summary.invited_member_count {
self.summary.invited_member_count = invited.into();
changed = true;
}
}
changed
}
pub(crate) fn update_joined_member_count(&mut self, count: u64) {
self.summary.joined_member_count = count;
}
pub(crate) fn update_invited_member_count(&mut self, count: u64) {
self.summary.invited_member_count = count;
}
pub(crate) fn set_invite_acceptance_details(&mut self, details: InviteAcceptanceDetails) {
self.invite_acceptance_details = Some(details);
}
pub fn invite_acceptance_details(&self) -> Option<InviteAcceptanceDetails> {
self.invite_acceptance_details.clone()
}
pub(crate) fn update_heroes(&mut self, heroes: Vec<RoomHero>) {
self.summary.room_heroes = heroes;
}
pub fn heroes(&self) -> &[RoomHero] {
&self.summary.room_heroes
}
pub fn active_members_count(&self) -> u64 {
self.summary.joined_member_count.saturating_add(self.summary.invited_member_count)
}
pub fn invited_members_count(&self) -> u64 {
self.summary.invited_member_count
}
pub fn joined_members_count(&self) -> u64 {
self.summary.joined_member_count
}
pub fn canonical_alias(&self) -> Option<&RoomAliasId> {
self.base_info.canonical_alias.as_ref()?.as_original()?.content.alias.as_deref()
}
pub fn alt_aliases(&self) -> &[OwnedRoomAliasId] {
self.base_info
.canonical_alias
.as_ref()
.and_then(|ev| ev.as_original())
.map(|ev| ev.content.alt_aliases.as_ref())
.unwrap_or_default()
}
pub fn room_id(&self) -> &RoomId {
&self.room_id
}
pub fn room_version(&self) -> Option<&RoomVersionId> {
self.base_info.room_version()
}
pub fn room_version_rules_or_default(&self) -> RoomVersionRules {
use std::sync::atomic::Ordering;
self.base_info.room_version().and_then(|room_version| room_version.rules()).unwrap_or_else(
|| {
if self
.warned_about_unknown_room_version_rules
.compare_exchange(false, true, Ordering::Relaxed, Ordering::Relaxed)
.is_ok()
{
warn!("Unable to get the room version rules, defaulting to rules for room version {ROOM_VERSION_FALLBACK}");
}
ROOM_VERSION_RULES_FALLBACK
},
)
}
pub fn room_type(&self) -> Option<&RoomType> {
match self.base_info.create.as_ref()? {
MinimalStateEvent::Original(ev) => ev.content.room_type.as_ref(),
MinimalStateEvent::Redacted(ev) => ev.content.room_type.as_ref(),
}
}
pub fn creators(&self) -> Option<Vec<OwnedUserId>> {
match self.base_info.create.as_ref()? {
MinimalStateEvent::Original(ev) => Some(ev.content.creators()),
MinimalStateEvent::Redacted(ev) => Some(ev.content.creators()),
}
}
pub(super) fn guest_access(&self) -> &GuestAccess {
match &self.base_info.guest_access {
Some(MinimalStateEvent::Original(ev)) => &ev.content.guest_access,
_ => &GuestAccess::Forbidden,
}
}
pub fn history_visibility(&self) -> Option<&HistoryVisibility> {
match &self.base_info.history_visibility {
Some(MinimalStateEvent::Original(ev)) => Some(&ev.content.history_visibility),
_ => None,
}
}
pub fn history_visibility_or_default(&self) -> &HistoryVisibility {
match &self.base_info.history_visibility {
Some(MinimalStateEvent::Original(ev)) => &ev.content.history_visibility,
_ => &HistoryVisibility::Shared,
}
}
pub fn join_rule(&self) -> Option<&JoinRule> {
match &self.base_info.join_rules {
Some(MinimalStateEvent::Original(ev)) => Some(&ev.content.join_rule),
_ => None,
}
}
pub fn name(&self) -> Option<&str> {
let name = &self.base_info.name.as_ref()?.as_original()?.content.name;
(!name.is_empty()).then_some(name)
}
pub fn create(&self) -> Option<&RoomCreateWithCreatorEventContent> {
Some(&self.base_info.create.as_ref()?.as_original()?.content)
}
pub fn tombstone(&self) -> Option<&RoomTombstoneEventContent> {
Some(&self.base_info.tombstone.as_ref()?.as_original()?.content)
}
pub fn topic(&self) -> Option<&str> {
Some(&self.base_info.topic.as_ref()?.as_original()?.content.topic)
}
fn active_matrix_rtc_memberships(&self) -> Vec<(CallMemberStateKey, MembershipData<'_>)> {
let mut v = self
.base_info
.rtc_member_events
.iter()
.filter_map(|(user_id, ev)| {
ev.as_original().map(|ev| {
ev.content
.active_memberships(None)
.into_iter()
.map(move |m| (user_id.clone(), m))
})
})
.flatten()
.collect::<Vec<_>>();
v.sort_by_key(|(_, m)| m.created_ts());
v
}
fn active_room_call_memberships(&self) -> Vec<(CallMemberStateKey, MembershipData<'_>)> {
self.active_matrix_rtc_memberships()
.into_iter()
.filter(|(_user_id, m)| m.is_room_call())
.collect()
}
pub fn has_active_room_call(&self) -> bool {
!self.active_room_call_memberships().is_empty()
}
pub fn active_room_call_participants(&self) -> Vec<OwnedUserId> {
self.active_room_call_memberships()
.iter()
.map(|(call_member_state_key, _)| call_member_state_key.user_id().to_owned())
.collect()
}
pub fn latest_event(&self) -> Option<&LatestEvent> {
self.latest_event.as_deref()
}
pub fn set_new_latest_event(&mut self, new_value: LatestEventValue) {
self.new_latest_event = new_value;
}
pub fn update_recency_stamp(&mut self, stamp: RoomRecencyStamp) {
self.recency_stamp = Some(stamp);
}
pub fn pinned_event_ids(&self) -> Option<Vec<OwnedEventId>> {
self.base_info.pinned_events.clone().map(|c| c.pinned)
}
pub fn is_pinned_event(&self, event_id: &EventId) -> bool {
self.base_info
.pinned_events
.as_ref()
.map(|p| p.pinned.contains(&event_id.to_owned()))
.unwrap_or_default()
}
#[instrument(skip_all, fields(room_id = ?self.room_id))]
pub(crate) async fn apply_migrations(&mut self, store: Arc<DynStateStore>) -> bool {
let mut migrated = false;
if self.data_format_version < 1 {
info!("Migrating room info to version 1");
match store.get_room_account_data_event_static::<TagEventContent>(&self.room_id).await {
Ok(Some(raw_event)) => match raw_event.deserialize() {
Ok(event) => {
self.base_info.handle_notable_tags(&event.content.tags);
}
Err(error) => {
warn!("Failed to deserialize room tags: {error}");
}
},
Ok(_) => {
}
Err(error) => {
warn!("Failed to load room tags: {error}");
}
}
match store.get_state_event_static::<RoomPinnedEventsEventContent>(&self.room_id).await
{
Ok(Some(RawSyncOrStrippedState::Sync(raw_event))) => {
match raw_event.deserialize() {
Ok(event) => {
self.handle_state_event(&event.into());
}
Err(error) => {
warn!("Failed to deserialize room pinned events: {error}");
}
}
}
Ok(_) => {
}
Err(error) => {
warn!("Failed to load room pinned events: {error}");
}
}
self.data_format_version = 1;
migrated = true;
}
migrated
}
}
#[repr(transparent)]
#[derive(Clone, Copy, Debug, Serialize, Deserialize, PartialEq)]
#[serde(transparent)]
pub struct RoomRecencyStamp(u64);
impl From<u64> for RoomRecencyStamp {
fn from(value: u64) -> Self {
Self(value)
}
}
impl From<RoomRecencyStamp> for u64 {
fn from(value: RoomRecencyStamp) -> Self {
value.0
}
}
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
pub(crate) enum SyncInfo {
NoState,
PartiallySynced,
FullySynced,
}
pub fn apply_redaction(
event: &Raw<AnySyncTimelineEvent>,
raw_redaction: &Raw<SyncRoomRedactionEvent>,
rules: &RedactionRules,
) -> Option<Raw<AnySyncTimelineEvent>> {
use ruma::canonical_json::{RedactedBecause, redact_in_place};
let mut event_json = match event.deserialize_as() {
Ok(json) => json,
Err(e) => {
warn!("Failed to deserialize latest event: {e}");
return None;
}
};
let redacted_because = match RedactedBecause::from_raw_event(raw_redaction) {
Ok(rb) => rb,
Err(e) => {
warn!("Redaction event is not valid canonical JSON: {e}");
return None;
}
};
let redact_result = redact_in_place(&mut event_json, rules, Some(redacted_because));
if let Err(e) = redact_result {
warn!("Failed to redact event: {e}");
return None;
}
let raw = Raw::new(&event_json).expect("CanonicalJsonObject must be serializable");
Some(raw.cast_unchecked())
}
#[derive(Debug, Clone)]
pub struct RoomInfoNotableUpdate {
pub room_id: OwnedRoomId,
pub reasons: RoomInfoNotableUpdateReasons,
}
bitflags! {
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct RoomInfoNotableUpdateReasons: u8 {
const RECENCY_STAMP = 0b0000_0001;
const LATEST_EVENT = 0b0000_0010;
const READ_RECEIPT = 0b0000_0100;
const UNREAD_MARKER = 0b0000_1000;
const MEMBERSHIP = 0b0001_0000;
const DISPLAY_NAME = 0b0010_0000;
const NONE = 0b1000_0000;
}
}
impl Default for RoomInfoNotableUpdateReasons {
fn default() -> Self {
Self::empty()
}
}
#[cfg(test)]
mod tests {
use std::sync::Arc;
use assert_matches::assert_matches;
use matrix_sdk_common::deserialized_responses::TimelineEvent;
use matrix_sdk_test::{
async_test,
test_json::{TAG, sync_events::PINNED_EVENTS},
};
use ruma::{
assign, events::room::pinned_events::RoomPinnedEventsEventContent, owned_event_id,
owned_mxc_uri, owned_user_id, room_id, serde::Raw,
};
use serde_json::json;
use similar_asserts::assert_eq;
use super::{BaseRoomInfo, LatestEventValue, RoomInfo, SyncInfo};
use crate::{
RoomDisplayName, RoomHero, RoomState, StateChanges,
latest_event::LatestEvent,
notification_settings::RoomNotificationMode,
room::{RoomNotableTags, RoomSummary},
store::{IntoStateStore, MemoryStore},
sync::UnreadNotificationsCount,
};
#[test]
fn test_room_info_serialization() {
let info = RoomInfo {
data_format_version: 1,
room_id: room_id!("!gda78o:server.tld").into(),
room_state: RoomState::Invited,
notification_counts: UnreadNotificationsCount {
highlight_count: 1,
notification_count: 2,
},
summary: RoomSummary {
room_heroes: vec![RoomHero {
user_id: owned_user_id!("@somebody:example.org"),
display_name: None,
avatar_url: None,
}],
joined_member_count: 5,
invited_member_count: 0,
},
members_synced: true,
last_prev_batch: Some("pb".to_owned()),
sync_info: SyncInfo::FullySynced,
encryption_state_synced: true,
latest_event: Some(Box::new(LatestEvent::new(TimelineEvent::from_plaintext(
Raw::from_json_string(json!({"sender": "@u:i.uk"}).to_string()).unwrap(),
)))),
new_latest_event: LatestEventValue::None,
base_info: Box::new(
assign!(BaseRoomInfo::new(), { pinned_events: Some(RoomPinnedEventsEventContent::new(vec![owned_event_id!("$a")])) }),
),
read_receipts: Default::default(),
warned_about_unknown_room_version_rules: Arc::new(false.into()),
cached_display_name: None,
cached_user_defined_notification_mode: None,
recency_stamp: Some(42.into()),
invite_acceptance_details: None,
};
let info_json = json!({
"data_format_version": 1,
"room_id": "!gda78o:server.tld",
"room_state": "Invited",
"notification_counts": {
"highlight_count": 1,
"notification_count": 2,
},
"summary": {
"room_heroes": [{
"user_id": "@somebody:example.org",
"display_name": null,
"avatar_url": null
}],
"joined_member_count": 5,
"invited_member_count": 0,
},
"members_synced": true,
"last_prev_batch": "pb",
"sync_info": "FullySynced",
"encryption_state_synced": true,
"latest_event": {
"event": {
"kind": {"PlainText": {"event": {"sender": "@u:i.uk"}}},
"thread_summary": "None",
"timestamp": null,
},
},
"new_latest_event": "None",
"base_info": {
"avatar": null,
"canonical_alias": null,
"create": null,
"dm_targets": [],
"encryption": null,
"guest_access": null,
"history_visibility": null,
"is_marked_unread": false,
"is_marked_unread_source": "Unstable",
"join_rules": null,
"max_power_level": 100,
"name": null,
"tombstone": null,
"topic": null,
"pinned_events": {
"pinned": ["$a"]
},
},
"read_receipts": {
"num_unread": 0,
"num_mentions": 0,
"num_notifications": 0,
"latest_active": null,
"pending": [],
},
"recency_stamp": 42,
});
assert_eq!(serde_json::to_value(info).unwrap(), info_json);
}
#[async_test]
async fn test_room_info_migration_v1() {
let store = MemoryStore::new().into_state_store();
let room_info_json = json!({
"room_id": "!gda78o:server.tld",
"room_state": "Joined",
"notification_counts": {
"highlight_count": 1,
"notification_count": 2,
},
"summary": {
"room_heroes": [{
"user_id": "@somebody:example.org",
"display_name": null,
"avatar_url": null
}],
"joined_member_count": 5,
"invited_member_count": 0,
},
"members_synced": true,
"last_prev_batch": "pb",
"sync_info": "FullySynced",
"encryption_state_synced": true,
"latest_event": {
"event": {
"encryption_info": null,
"event": {
"sender": "@u:i.uk",
},
},
},
"base_info": {
"avatar": null,
"canonical_alias": null,
"create": null,
"dm_targets": [],
"encryption": null,
"guest_access": null,
"history_visibility": null,
"join_rules": null,
"max_power_level": 100,
"name": null,
"tombstone": null,
"topic": null,
},
"read_receipts": {
"num_unread": 0,
"num_mentions": 0,
"num_notifications": 0,
"latest_active": null,
"pending": []
},
"recency_stamp": 42,
});
let mut room_info: RoomInfo = serde_json::from_value(room_info_json).unwrap();
assert_eq!(room_info.data_format_version, 0);
assert!(room_info.base_info.notable_tags.is_empty());
assert!(room_info.base_info.pinned_events.is_none());
assert!(room_info.apply_migrations(store.clone()).await);
assert_eq!(room_info.data_format_version, 1);
assert!(room_info.base_info.notable_tags.is_empty());
assert!(room_info.base_info.pinned_events.is_none());
assert!(!room_info.apply_migrations(store.clone()).await);
assert_eq!(room_info.data_format_version, 1);
assert!(room_info.base_info.notable_tags.is_empty());
assert!(room_info.base_info.pinned_events.is_none());
let mut changes = StateChanges::default();
let raw_tag_event = Raw::new(&*TAG).unwrap().cast_unchecked();
let tag_event = raw_tag_event.deserialize().unwrap();
changes.add_room_account_data(&room_info.room_id, tag_event, raw_tag_event);
let raw_pinned_events_event = Raw::new(&*PINNED_EVENTS).unwrap().cast_unchecked();
let pinned_events_event = raw_pinned_events_event.deserialize().unwrap();
changes.add_state_event(&room_info.room_id, pinned_events_event, raw_pinned_events_event);
store.save_changes(&changes).await.unwrap();
room_info.data_format_version = 0;
assert!(room_info.apply_migrations(store.clone()).await);
assert_eq!(room_info.data_format_version, 1);
assert!(room_info.base_info.notable_tags.contains(RoomNotableTags::FAVOURITE));
assert!(room_info.base_info.pinned_events.is_some());
let new_room_info = RoomInfo::new(room_id!("!new_room:localhost"), RoomState::Joined);
assert_eq!(new_room_info.data_format_version, 1);
}
#[test]
fn test_room_info_deserialization() {
let info_json = json!({
"room_id": "!gda78o:server.tld",
"room_state": "Joined",
"notification_counts": {
"highlight_count": 1,
"notification_count": 2,
},
"summary": {
"room_heroes": [{
"user_id": "@somebody:example.org",
"display_name": "Somebody",
"avatar_url": "mxc://example.org/abc"
}],
"joined_member_count": 5,
"invited_member_count": 0,
},
"members_synced": true,
"last_prev_batch": "pb",
"sync_info": "FullySynced",
"encryption_state_synced": true,
"base_info": {
"avatar": null,
"canonical_alias": null,
"create": null,
"dm_targets": [],
"encryption": null,
"guest_access": null,
"history_visibility": null,
"join_rules": null,
"max_power_level": 100,
"name": null,
"tombstone": null,
"topic": null,
},
"cached_display_name": { "Calculated": "lol" },
"cached_user_defined_notification_mode": "Mute",
"recency_stamp": 42,
});
let info: RoomInfo = serde_json::from_value(info_json).unwrap();
assert_eq!(info.room_id, room_id!("!gda78o:server.tld"));
assert_eq!(info.room_state, RoomState::Joined);
assert_eq!(info.notification_counts.highlight_count, 1);
assert_eq!(info.notification_counts.notification_count, 2);
assert_eq!(
info.summary.room_heroes,
vec![RoomHero {
user_id: owned_user_id!("@somebody:example.org"),
display_name: Some("Somebody".to_owned()),
avatar_url: Some(owned_mxc_uri!("mxc://example.org/abc")),
}]
);
assert_eq!(info.summary.joined_member_count, 5);
assert_eq!(info.summary.invited_member_count, 0);
assert!(info.members_synced);
assert_eq!(info.last_prev_batch, Some("pb".to_owned()));
assert_eq!(info.sync_info, SyncInfo::FullySynced);
assert!(info.encryption_state_synced);
assert!(info.latest_event.is_none());
assert_matches!(info.new_latest_event, LatestEventValue::None);
assert!(info.base_info.avatar.is_none());
assert!(info.base_info.canonical_alias.is_none());
assert!(info.base_info.create.is_none());
assert_eq!(info.base_info.dm_targets.len(), 0);
assert!(info.base_info.encryption.is_none());
assert!(info.base_info.guest_access.is_none());
assert!(info.base_info.history_visibility.is_none());
assert!(info.base_info.join_rules.is_none());
assert_eq!(info.base_info.max_power_level, 100);
assert!(info.base_info.name.is_none());
assert!(info.base_info.tombstone.is_none());
assert!(info.base_info.topic.is_none());
assert_eq!(
info.cached_display_name.as_ref(),
Some(&RoomDisplayName::Calculated("lol".to_owned())),
);
assert_eq!(
info.cached_user_defined_notification_mode.as_ref(),
Some(&RoomNotificationMode::Mute)
);
assert_eq!(info.recency_stamp.as_ref(), Some(&42.into()));
}
#[test]
fn test_room_info_deserialization_without_optional_items() {
let info_json = json!({
"room_id": "!gda78o:server.tld",
"room_state": "Invited",
"notification_counts": {
"highlight_count": 1,
"notification_count": 2,
},
"summary": {
"room_heroes": [{
"user_id": "@somebody:example.org",
"display_name": "Somebody",
"avatar_url": "mxc://example.org/abc"
}],
"joined_member_count": 5,
"invited_member_count": 0,
},
"members_synced": true,
"last_prev_batch": "pb",
"sync_info": "FullySynced",
"encryption_state_synced": true,
"base_info": {
"avatar": null,
"canonical_alias": null,
"create": null,
"dm_targets": [],
"encryption": null,
"guest_access": null,
"history_visibility": null,
"join_rules": null,
"max_power_level": 100,
"name": null,
"tombstone": null,
"topic": null,
},
});
let info: RoomInfo = serde_json::from_value(info_json).unwrap();
assert_eq!(info.room_id, room_id!("!gda78o:server.tld"));
assert_eq!(info.room_state, RoomState::Invited);
assert_eq!(info.notification_counts.highlight_count, 1);
assert_eq!(info.notification_counts.notification_count, 2);
assert_eq!(
info.summary.room_heroes,
vec![RoomHero {
user_id: owned_user_id!("@somebody:example.org"),
display_name: Some("Somebody".to_owned()),
avatar_url: Some(owned_mxc_uri!("mxc://example.org/abc")),
}]
);
assert_eq!(info.summary.joined_member_count, 5);
assert_eq!(info.summary.invited_member_count, 0);
assert!(info.members_synced);
assert_eq!(info.last_prev_batch, Some("pb".to_owned()));
assert_eq!(info.sync_info, SyncInfo::FullySynced);
assert!(info.encryption_state_synced);
assert!(info.base_info.avatar.is_none());
assert!(info.base_info.canonical_alias.is_none());
assert!(info.base_info.create.is_none());
assert_eq!(info.base_info.dm_targets.len(), 0);
assert!(info.base_info.encryption.is_none());
assert!(info.base_info.guest_access.is_none());
assert!(info.base_info.history_visibility.is_none());
assert!(info.base_info.join_rules.is_none());
assert_eq!(info.base_info.max_power_level, 100);
assert!(info.base_info.name.is_none());
assert!(info.base_info.tombstone.is_none());
assert!(info.base_info.topic.is_none());
}
}