use std::{
collections::{BTreeMap, HashMap, HashSet},
convert::TryFrom,
};
#[cfg(feature = "messages")]
use std::ops::DerefMut;
#[cfg(feature = "encryption")]
use std::time::Duration;
#[cfg(feature = "encryption")]
use matrix_sdk_crypto::EncryptionSettings;
#[cfg(feature = "messages")]
use matrix_sdk_common::events::{
room::redaction::SyncRedactionEvent, AnyPossiblyRedactedSyncMessageEvent, AnySyncMessageEvent,
};
use matrix_sdk_common::{
api::r0::sync::sync_events::{RoomSummary, UnreadNotificationsCount},
events::{
presence::{PresenceEvent, PresenceEventContent},
room::{
aliases::AliasesEventContent,
canonical_alias::CanonicalAliasEventContent,
encryption::EncryptionEventContent,
member::{MemberEventContent, MembershipChange, MembershipState},
name::NameEventContent,
power_levels::{NotificationPowerLevels, PowerLevelsEventContent},
tombstone::TombstoneEventContent,
},
AnyStrippedStateEvent, AnySyncRoomEvent, AnySyncStateEvent, EventType, StrippedStateEvent,
SyncStateEvent,
},
identifiers::{EventEncryptionAlgorithm, RoomAliasId, RoomId, UserId},
int, uint, Int, UInt,
};
use serde::{Deserialize, Serialize};
use tracing::{debug, error, trace};
#[cfg(feature = "messages")]
use super::message::MessageQueue;
use super::RoomMember;
#[derive(Clone, Debug, Default, PartialEq, Serialize, Deserialize)]
pub struct RoomName {
name: Option<String>,
canonical_alias: Option<RoomAliasId>,
aliases: Vec<RoomAliasId>,
pub heroes: Vec<String>,
pub joined_member_count: Option<UInt>,
pub invited_member_count: Option<UInt>,
}
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
pub struct PowerLevels {
pub ban: Int,
pub events: BTreeMap<EventType, Int>,
pub events_default: Int,
pub invite: Int,
pub kick: Int,
pub redact: Int,
pub state_default: Int,
pub users_default: Int,
pub notifications: Int,
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub struct EncryptionInfo {
algorithm: EventEncryptionAlgorithm,
rotation_period_ms: u64,
rotation_period_messages: u64,
}
impl Default for EncryptionInfo {
fn default() -> Self {
Self {
algorithm: EventEncryptionAlgorithm::MegolmV1AesSha2,
rotation_period_ms: 604_800_000,
rotation_period_messages: 100,
}
}
}
impl EncryptionInfo {
pub fn algorithm(&self) -> &EventEncryptionAlgorithm {
&self.algorithm
}
pub fn rotation_period(&self) -> u64 {
self.rotation_period_ms
}
pub fn rotation_period_messages(&self) -> u64 {
self.rotation_period_messages
}
}
impl From<&SyncStateEvent<EncryptionEventContent>> for EncryptionInfo {
fn from(event: &SyncStateEvent<EncryptionEventContent>) -> Self {
EncryptionInfo {
algorithm: event.content.algorithm.clone(),
rotation_period_ms: event
.content
.rotation_period_ms
.map_or(604_800_000, Into::into),
rotation_period_messages: event.content.rotation_period_msgs.map_or(100, Into::into),
}
}
}
#[cfg(feature = "encryption")]
impl Into<EncryptionSettings> for EncryptionInfo {
fn into(self) -> EncryptionSettings {
EncryptionSettings {
algorithm: self.algorithm,
rotation_period: Duration::from_millis(self.rotation_period_ms),
rotation_period_msgs: self.rotation_period_messages,
}
}
}
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
pub struct Tombstone {
body: String,
replacement: RoomId,
}
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub struct Room {
pub room_id: RoomId,
pub room_name: RoomName,
pub own_user_id: UserId,
pub creator: Option<UserId>,
pub direct_target: Option<UserId>,
pub invited_members: HashMap<UserId, RoomMember>,
pub joined_members: HashMap<UserId, RoomMember>,
#[cfg(feature = "messages")]
#[cfg_attr(feature = "docs", doc(cfg(messages)))]
#[serde(with = "super::message::ser_deser")]
pub messages: MessageQueue,
pub typing_users: Vec<UserId>,
pub power_levels: Option<PowerLevels>,
pub encrypted: Option<EncryptionInfo>,
pub unread_highlight: Option<UInt>,
pub unread_notifications: Option<UInt>,
pub tombstone: Option<Tombstone>,
}
impl RoomName {
pub fn push_alias(&mut self, alias: RoomAliasId) -> bool {
self.aliases.push(alias);
true
}
pub fn set_canonical(&mut self, alias: RoomAliasId) -> bool {
self.canonical_alias = Some(alias);
true
}
pub fn set_name(&mut self, name: &str) -> bool {
self.name = Some(name.to_string());
true
}
pub fn calculate_name(
&self,
own_user_id: &UserId,
invited_members: &HashMap<UserId, RoomMember>,
joined_members: &HashMap<UserId, RoomMember>,
) -> String {
if let Some(name) = &self.name {
let name = name.trim();
name.to_string()
} else if let Some(alias) = &self.canonical_alias {
let alias = alias.alias().trim();
alias.to_string()
} else if !self.aliases.is_empty() && !self.aliases[0].alias().is_empty() {
self.aliases[0].alias().trim().to_string()
} else {
let joined = self.joined_member_count.unwrap_or_else(|| uint!(0));
let invited = self.invited_member_count.unwrap_or_else(|| uint!(0));
let heroes = UInt::new(self.heroes.len() as u64).unwrap();
let invited_joined = (invited + joined).saturating_sub(uint!(1));
let members = joined_members.values().chain(invited_members.values());
if heroes >= invited_joined {
let mut names = members
.filter(|m| m.user_id != *own_user_id)
.take(3)
.map(|mem| {
mem.display_name
.clone()
.unwrap_or_else(|| mem.user_id.localpart().to_string())
})
.collect::<Vec<String>>();
names.sort();
names.join(", ")
} else if heroes < invited_joined && invited + joined > uint!(1) {
let mut names = members
.filter(|m| m.user_id != *own_user_id)
.take(3)
.map(|mem| {
mem.display_name
.clone()
.unwrap_or_else(|| mem.user_id.localpart().to_string())
})
.collect::<Vec<String>>();
names.sort();
format!("{}, and {} others", names.join(", "), (joined + invited))
} else {
"Empty room".to_string()
}
}
}
}
impl Room {
pub fn new(room_id: &RoomId, own_user_id: &UserId) -> Self {
Room {
room_id: room_id.clone(),
room_name: RoomName::default(),
own_user_id: own_user_id.clone(),
creator: None,
direct_target: None,
invited_members: HashMap::new(),
joined_members: HashMap::new(),
#[cfg(feature = "messages")]
messages: MessageQueue::new(),
typing_users: Vec::new(),
power_levels: None,
encrypted: None,
unread_highlight: None,
unread_notifications: None,
tombstone: None,
}
}
pub fn display_name(&self) -> String {
self.room_name.calculate_name(
&self.own_user_id,
&self.invited_members,
&self.joined_members,
)
}
pub fn is_encrypted(&self) -> bool {
self.encrypted.is_some()
}
pub fn encryption_info(&self) -> Option<&EncryptionInfo> {
self.encrypted.as_ref()
}
fn add_member(
&mut self,
target_member: &UserId,
event: &SyncStateEvent<MemberEventContent>,
) -> (bool, HashMap<UserId, bool>) {
let new_member = RoomMember::new(event, &self.room_id);
if self.joined_members.contains_key(&new_member.user_id) {
error!("add_member called on event of an already joined user");
return (false, HashMap::new());
}
let disambiguations =
self.disambiguation_updates(target_member, None, new_member.display_name.clone());
debug!("add_member: disambiguations: {:#?}", disambiguations);
match event.content.membership {
MembershipState::Join => {
self.invited_members.remove(target_member);
self.joined_members
.insert(target_member.clone(), new_member)
}
MembershipState::Invite => self
.invited_members
.insert(target_member.clone(), new_member),
_ => panic!("Room::add_member called on event that is neither `join` nor `invite`."),
};
for (id, is_ambiguous) in disambiguations.iter() {
self.get_member_mut(id).unwrap().display_name_ambiguous = *is_ambiguous;
}
(true, disambiguations)
}
fn remove_member(
&mut self,
target_member: &UserId,
event: &SyncStateEvent<MemberEventContent>,
) -> (bool, HashMap<UserId, bool>) {
let leaving_member = RoomMember::new(event, &self.room_id);
if self.get_member(target_member).is_none() {
return (false, HashMap::new());
}
let disambiguations =
self.disambiguation_updates(target_member, leaving_member.display_name, None);
debug!("remove_member: disambiguations: {:#?}", disambiguations);
for (id, is_ambiguous) in disambiguations.iter() {
self.get_member_mut(id).unwrap().display_name_ambiguous = *is_ambiguous;
}
self.joined_members
.remove(target_member)
.or_else(|| self.invited_members.remove(target_member));
(true, disambiguations)
}
pub fn member_is_tracked(&self, user_id: &UserId) -> bool {
self.invited_members.contains_key(&user_id) || self.joined_members.contains_key(&user_id)
}
pub fn get_member(&self, user_id: &UserId) -> Option<&RoomMember> {
self.joined_members
.get(user_id)
.or_else(|| self.invited_members.get(user_id))
}
pub fn get_member_mut(&mut self, user_id: &UserId) -> Option<&mut RoomMember> {
match self.joined_members.get_mut(user_id) {
None => self.invited_members.get_mut(user_id),
Some(m) => Some(m),
}
}
fn display_name_equivalence_set(&self, name: &str) -> HashSet<UserId> {
let members = self
.invited_members
.iter()
.chain(self.joined_members.iter());
members
.filter(|(_, member)| {
member
.display_name
.as_ref()
.map(|other_name| other_name == name)
.unwrap_or(false)
})
.map(|(_, member)| member.user_id.clone())
.collect()
}
fn disambiguation_updates(
&self,
member: &UserId,
old_name: Option<String>,
new_name: Option<String>,
) -> HashMap<UserId, bool> {
let old_name_eq_set = match old_name {
None => HashSet::new(),
Some(name) => self.display_name_equivalence_set(&name),
};
let disambiguate_old = match old_name_eq_set.len().saturating_sub(1) {
n if n > 1 => vec![(member.clone(), false)].into_iter().collect(),
1 => old_name_eq_set.into_iter().map(|m| (m, false)).collect(),
0 => HashMap::new(),
_ => panic!("impossible"),
};
let mut new_name_eq_set = match new_name {
None => HashSet::new(),
Some(name) => self.display_name_equivalence_set(&name),
};
new_name_eq_set.insert(member.clone());
let disambiguate_new = match new_name_eq_set.len() {
1 => HashMap::new(),
2 => new_name_eq_set.into_iter().map(|m| (m, true)).collect(),
_ => vec![(member.clone(), true)].into_iter().collect(),
};
disambiguate_old
.into_iter()
.chain(disambiguate_new.into_iter())
.collect()
}
fn push_room_alias(&mut self, alias: &RoomAliasId) -> bool {
self.room_name.push_alias(alias.clone());
true
}
fn canonical_alias(&mut self, alias: &RoomAliasId) -> bool {
self.room_name.set_canonical(alias.clone());
true
}
fn set_room_name(&mut self, name: &str) -> bool {
self.room_name.set_name(name);
true
}
fn set_room_power_level(&mut self, event: &SyncStateEvent<PowerLevelsEventContent>) -> bool {
let PowerLevelsEventContent {
ban,
events,
events_default,
invite,
kick,
redact,
state_default,
users_default,
notifications: NotificationPowerLevels { room },
..
} = &event.content;
let power = PowerLevels {
ban: *ban,
events: events.clone(),
events_default: *events_default,
invite: *invite,
kick: *kick,
redact: *redact,
state_default: *state_default,
users_default: *users_default,
notifications: *room,
};
self.power_levels = Some(power);
true
}
pub(crate) fn set_room_summary(&mut self, summary: &RoomSummary) {
let RoomSummary {
heroes,
joined_member_count,
invited_member_count,
..
} = summary;
self.room_name.heroes = heroes.clone();
self.room_name.invited_member_count = *invited_member_count;
self.room_name.joined_member_count = *joined_member_count;
}
pub(crate) fn set_unread_notice_count(&mut self, notifications: &UnreadNotificationsCount) {
self.unread_highlight = notifications.highlight_count;
self.unread_notifications = notifications.notification_count;
}
pub fn handle_membership(
&mut self,
event: &SyncStateEvent<MemberEventContent>,
state_event: bool,
) -> (bool, HashMap<UserId, bool>) {
use MembershipChange::*;
use MembershipState::*;
trace!(
"Received {} event: {}",
if state_event { "state" } else { "timeline" },
event.event_id
);
let target_user = match UserId::try_from(event.state_key.clone()) {
Ok(id) => id,
Err(e) => {
error!("Received a member event with invalid state_key: {}", e);
return (false, HashMap::new());
}
};
if state_event && !self.member_is_tracked(&target_user) {
debug!(
"handle_membership: User {user_id} {state} the room {room_id} ({room_name})",
user_id = target_user,
state = event.content.membership.describe(),
room_id = self.room_id,
room_name = self.display_name(),
);
match event.content.membership {
Join | Invite => self.add_member(&target_user, event),
_ => (false, HashMap::new()),
}
} else {
let change = event.membership_change();
debug!(
"handle_membership: User {user_id} {action} the room {room_id} ({room_name})",
user_id = target_user,
action = change.describe(),
room_id = self.room_id,
room_name = self.display_name(),
);
match change {
Invited | Joined => self.add_member(&target_user, event),
Kicked | Banned | KickedAndBanned | InvitationRejected | Left => {
self.remove_member(&target_user, event)
}
ProfileChanged { .. } => self.update_member_profile(&target_user, event, change),
_ => (false, HashMap::new()),
}
}
}
#[cfg(feature = "messages")]
#[cfg_attr(feature = "docs", doc(cfg(messages)))]
pub fn handle_message(&mut self, event: &AnySyncMessageEvent) -> bool {
self.messages
.push(AnyPossiblyRedactedSyncMessageEvent::Regular(event.clone()))
}
#[cfg(feature = "messages")]
#[cfg_attr(feature = "docs", doc(cfg(messages)))]
pub fn handle_redaction(&mut self, redacted_event: &SyncRedactionEvent) -> bool {
use crate::{identifiers::RoomVersionId, models::message::PossiblyRedactedExt};
if let Some(mut msg) = self
.messages
.iter_mut()
.find(|msg| &redacted_event.redacts == msg.event_id())
{
match msg.deref_mut() {
AnyPossiblyRedactedSyncMessageEvent::Regular(event) => {
*msg = AnyPossiblyRedactedSyncMessageEvent::Redacted(
event
.clone()
.redact(redacted_event.clone(), RoomVersionId::Version6),
);
}
AnyPossiblyRedactedSyncMessageEvent::Redacted(_) => return false,
}
true
} else {
false
}
}
pub fn handle_room_aliases(&mut self, event: &SyncStateEvent<AliasesEventContent>) -> bool {
match event.content.aliases.as_slice() {
[alias] => self.push_room_alias(alias),
[alias, ..] => self.push_room_alias(alias),
_ => false,
}
}
pub fn handle_canonical(&mut self, event: &SyncStateEvent<CanonicalAliasEventContent>) -> bool {
match &event.content.alias {
Some(name) => self.canonical_alias(&name),
_ => false,
}
}
pub fn handle_room_name(&mut self, event: &SyncStateEvent<NameEventContent>) -> bool {
match event.content.name() {
Some(name) => self.set_room_name(name),
_ => false,
}
}
pub fn handle_stripped_room_name(
&mut self,
event: &StrippedStateEvent<NameEventContent>,
) -> bool {
match event.content.name() {
Some(name) => self.set_room_name(name),
_ => false,
}
}
pub fn handle_power_level(&mut self, event: &SyncStateEvent<PowerLevelsEventContent>) -> bool {
let mut updated = self.set_room_power_level(event);
let mut max_power = event.content.users_default;
for power in event.content.users.values() {
max_power = *power.max(&max_power);
}
for user in event.content.users.keys() {
if let Some(member) = self.joined_members.get_mut(user) {
if Room::update_member_power(member, event, max_power) {
updated = true;
}
}
}
updated
}
fn handle_tombstone(&mut self, event: &SyncStateEvent<TombstoneEventContent>) -> bool {
self.tombstone = Some(Tombstone {
body: event.content.body.clone(),
replacement: event.content.replacement_room.clone(),
});
true
}
pub fn handle_direct(&mut self, user_id: &UserId) -> bool {
if let Some(old_target) = &self.direct_target {
if old_target == user_id {
return false;
}
}
self.direct_target = Some(user_id.clone());
true
}
fn handle_encryption_event(&mut self, event: &SyncStateEvent<EncryptionEventContent>) -> bool {
self.encrypted = Some(event.into());
true
}
pub fn receive_timeline_event(&mut self, event: &AnySyncRoomEvent) -> bool {
match event {
AnySyncRoomEvent::State(event) => match event {
AnySyncStateEvent::RoomMember(event) => self.handle_membership(event, false).0,
AnySyncStateEvent::RoomName(event) => self.handle_room_name(event),
AnySyncStateEvent::RoomCanonicalAlias(event) => self.handle_canonical(event),
AnySyncStateEvent::RoomAliases(event) => self.handle_room_aliases(event),
AnySyncStateEvent::RoomPowerLevels(event) => self.handle_power_level(event),
AnySyncStateEvent::RoomTombstone(event) => self.handle_tombstone(event),
AnySyncStateEvent::RoomEncryption(event) => self.handle_encryption_event(event),
_ => false,
},
AnySyncRoomEvent::Message(event) => match event {
#[cfg(feature = "messages")]
AnySyncMessageEvent::RoomMessage(_) => self.handle_message(event),
#[cfg(feature = "messages")]
AnySyncMessageEvent::RoomRedaction(event) => self.handle_redaction(event),
_ => false,
},
AnySyncRoomEvent::RedactedMessage(_) | AnySyncRoomEvent::RedactedState(_) => false,
}
}
pub fn receive_state_event(&mut self, event: &AnySyncStateEvent) -> bool {
match event {
AnySyncStateEvent::RoomMember(member) => self.handle_membership(member, true).0,
AnySyncStateEvent::RoomName(name) => self.handle_room_name(name),
AnySyncStateEvent::RoomCanonicalAlias(c_alias) => self.handle_canonical(c_alias),
AnySyncStateEvent::RoomAliases(alias) => self.handle_room_aliases(alias),
AnySyncStateEvent::RoomPowerLevels(power) => self.handle_power_level(power),
AnySyncStateEvent::RoomTombstone(tomb) => self.handle_tombstone(tomb),
AnySyncStateEvent::RoomEncryption(encrypt) => self.handle_encryption_event(encrypt),
_ => false,
}
}
pub fn receive_stripped_state_event(&mut self, event: &AnyStrippedStateEvent) -> bool {
match event {
AnyStrippedStateEvent::RoomName(event) => self.handle_stripped_room_name(event),
_ => false,
}
}
pub fn receive_presence_event(&mut self, event: &PresenceEvent) -> bool {
let PresenceEvent {
content:
PresenceEventContent {
avatar_url,
currently_active,
displayname,
last_active_ago,
presence,
status_msg,
},
..
} = event;
if let Some(member) = self.joined_members.get_mut(&event.sender) {
if member.display_name == *displayname
&& member.avatar_url == *avatar_url
&& member.presence.as_ref() == Some(presence)
&& member.status_msg == *status_msg
&& member.last_active_ago == *last_active_ago
&& member.currently_active == *currently_active
{
false
} else {
member.presence_events.push(event.clone());
member.avatar_url = avatar_url.clone();
member.currently_active = *currently_active;
member.display_name = displayname.clone();
member.last_active_ago = *last_active_ago;
member.presence = Some(presence.clone());
member.status_msg = status_msg.clone();
true
}
} else {
false
}
}
pub fn update_member_profile(
&mut self,
target_member: &UserId,
event: &SyncStateEvent<MemberEventContent>,
change: MembershipChange,
) -> (bool, HashMap<UserId, bool>) {
let member = self.get_member(target_member);
let member = match member {
Some(member) => member,
None => {
debug!("update_member_profile [{}]: Got a profile update for user {} but he's not a room member",
self.room_id, target_member);
return (false, HashMap::new());
}
};
let old_name = member.display_name.clone();
let new_name = event.content.displayname.clone();
match change {
MembershipChange::ProfileChanged {
displayname_changed,
avatar_url_changed,
} => {
if displayname_changed {
debug!(
"update_member_profile [{}]: {} changed display name from {:#?} to {:#?}",
self.room_id, target_member, old_name, &new_name
);
}
if avatar_url_changed {
debug!(
"update_member_profile [{}]: {} changed avatar URL from {:#?} to {:#?}",
self.room_id, target_member, &member.avatar_url, &new_name
);
}
}
_ => {
error!(
"update_member_profile [{}]: got a ProfileChanged but nothing changed",
self.room_id
);
return (false, HashMap::new());
}
}
let disambiguations =
self.disambiguation_updates(target_member, old_name, new_name.clone());
for (id, is_ambiguous) in disambiguations.iter() {
if self.get_member_mut(id).is_none() {
debug!("update_member_profile [{}]: Tried disambiguating display name for {} but he's not there",
self.room_id,
id);
} else {
self.get_member_mut(id).unwrap().display_name_ambiguous = *is_ambiguous;
}
}
debug!(
"update_member_profile [{}]: disambiguations: {:#?}",
self.room_id, &disambiguations
);
let changed = match self.get_member_mut(target_member) {
Some(member) => {
member.display_name = new_name;
member.avatar_url = event.content.avatar_url.clone();
true
}
None => {
error!(
"update_member_profile [{}]: user {} does not exist",
self.room_id, target_member
);
false
}
};
(changed, disambiguations)
}
pub fn update_member_power(
member: &mut RoomMember,
event: &SyncStateEvent<PowerLevelsEventContent>,
max_power: Int,
) -> bool {
let changed;
if let Some(user_power) = event.content.users.get(&member.user_id) {
changed = member.power_level != Some(*user_power);
member.power_level = Some(*user_power);
} else {
changed = member.power_level != Some(event.content.users_default);
member.power_level = Some(event.content.users_default);
}
if max_power > int!(0) {
let normalized = {
let pl: i64 = member.power_level.unwrap_or_default().into();
let max: i64 = max_power.into();
Int::new((pl * 100_i64) / max)
};
member.power_level_norm = normalized;
}
changed
}
}
trait Describe {
fn describe(&self) -> String;
}
impl Describe for MembershipState {
fn describe(&self) -> String {
match self {
Self::Ban => "is banned in",
Self::Invite => "is invited to",
Self::Join => "is a member of",
Self::Knock => "is requesting access to",
Self::Leave => "has left",
_ => "unhandled case of MembershipState",
}
.to_string()
}
}
impl Describe for MembershipChange {
fn describe(&self) -> String {
match self {
Self::Invited => "got invited to",
Self::Joined => "joined",
Self::Kicked => "got kicked from",
Self::Banned => "got banned from",
Self::Unbanned => "got unbanned from",
Self::KickedAndBanned => "got kicked and banned from",
Self::InvitationRejected => "rejected the invitation to",
Self::InvitationRevoked => "got their invitation revoked from",
Self::Left => "left",
Self::ProfileChanged {
displayname_changed,
avatar_url_changed,
} => match (*displayname_changed, *avatar_url_changed) {
(true, true) => "changed their displayname and avatar in",
(true, false) => "changed their displayname in",
(false, true) => "changed their avatar in",
_ => {
error!("Got ProfileChanged but nothing changed");
"impossible: changed nothing in their profile in"
}
},
Self::None => "did nothing in",
Self::NotImplemented => "NOT IMPLEMENTED",
Self::Error => "ERROR",
_ => "unhandled case of MembershipChange",
}
.to_string()
}
}
#[cfg(test)]
mod test {
use super::*;
#[cfg(not(target_arch = "wasm32"))]
use crate::{events::room::encryption::EncryptionEventContent, Raw};
use crate::{
events::Unsigned,
identifiers::{event_id, room_id, user_id, UserId},
BaseClient, Session,
};
use matrix_sdk_common::{int, Int};
use matrix_sdk_test::{async_test, sync_response, EventBuilder, EventsJson, SyncResponseFile};
use std::{ops::Deref, time::SystemTime};
#[cfg(target_arch = "wasm32")]
use wasm_bindgen_test::*;
async fn get_client() -> BaseClient {
let session = Session {
access_token: "1234".to_owned(),
user_id: user_id!("@example:localhost"),
device_id: "DEVICEID".into(),
};
let client = BaseClient::new().unwrap();
client.restore_login(session).await.unwrap();
client
}
fn get_room_id() -> RoomId {
room_id!("!SVkFJHzfwvuaIEawgC:localhost")
}
#[async_test]
async fn user_presence() {
let client = get_client().await;
let mut response = sync_response(SyncResponseFile::Default);
client.receive_sync_response(&mut response).await.unwrap();
let rooms_lock = &client.joined_rooms();
let rooms = rooms_lock.read().await;
let room = &rooms
.get(&room_id!("!SVkFJHzfwvuaIEawgC:localhost"))
.unwrap()
.read()
.await;
assert_eq!(1, room.joined_members.len());
assert!(room.deref().power_levels.is_some())
}
#[cfg(feature = "messages")]
#[test]
fn message_edit_deser() {
let json = matrix_sdk_test::test_json::MESSAGE_EDIT.deref();
let event = serde_json::from_value::<Raw<AnySyncMessageEvent>>(json.clone()).unwrap();
if let Ok(AnySyncMessageEvent::RoomMessage(ev)) = event.deserialize() {
if let matrix_sdk_common::events::room::message::MessageEventContent::Text(content) =
ev.content
{
assert_eq!(content.body, " * edited message");
assert!(content.relates_to.is_some());
}
} else {
panic!("{:?}", event);
}
}
#[async_test]
async fn member_is_not_both_invited_and_joined() {
let client = get_client().await;
let room_id = get_room_id();
let user_id1 = user_id!("@example:localhost");
let user_id2 = user_id!("@example2:localhost");
let member2_invite_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example2",
"membership": "invite"
},
"event_id": "$16345217l517tabbz:localhost",
"membership": "join",
"origin_server_ts": 1455123234,
"sender": format!("{}", user_id1),
"state_key": format!("{}", user_id2),
"type": "m.room.member",
"unsigned": {
"age": 1989321234,
"replaces_state": "$1622a2311315tkjoA:localhost"
}
});
let member2_join_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example2",
"membership": "join"
},
"event_id": "$163409224327jkbba:localhost",
"membership": "join",
"origin_server_ts": 1455123238,
"sender": format!("{}", user_id2),
"state_key": format!("{}", user_id2),
"type": "m.room.member",
"prev_content": {
"avatar_url": null,
"displayname": "example2",
"membership": "invite"
},
"unsigned": {
"age": 1989321214,
"replaces_state": "$16345217l517tabbz:localhost"
}
});
let mut event_builder = EventBuilder::new();
let mut member1_join_sync_response = event_builder
.add_room_event(EventsJson::Member)
.build_sync_response();
let mut member2_invite_sync_response = event_builder
.add_custom_joined_event(&room_id, member2_invite_event)
.build_sync_response();
let mut member2_join_sync_response = event_builder
.add_custom_joined_event(&room_id, member2_join_event)
.build_sync_response();
async fn invited_or_joined_but_not_both(client: &BaseClient, room: &RoomId, user: &UserId) {
let room = client.get_joined_room(&room).await.unwrap();
let room = room.read().await;
assert!(
room.invited_members.get(&user).is_none()
|| room.joined_members.get(&user).is_none()
);
assert!(
room.invited_members.get(&user).is_some()
|| room.joined_members.get(&user).is_some()
);
};
client
.receive_sync_response(&mut member1_join_sync_response)
.await
.unwrap();
invited_or_joined_but_not_both(&client, &room_id, &user_id1).await;
client
.receive_sync_response(&mut member2_invite_sync_response)
.await
.unwrap();
invited_or_joined_but_not_both(&client, &room_id, &user_id1).await;
invited_or_joined_but_not_both(&client, &room_id, &user_id2).await;
client
.receive_sync_response(&mut member2_join_sync_response)
.await
.unwrap();
invited_or_joined_but_not_both(&client, &room_id, &user_id1).await;
invited_or_joined_but_not_both(&client, &room_id, &user_id2).await;
}
#[async_test]
async fn test_member_display_name() {
let client = get_client().await;
let room_id = get_room_id();
let user_id1 = user_id!("@example:localhost");
let user_id2 = user_id!("@example2:localhost");
let user_id3 = user_id!("@example3:localhost");
let member2_join_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example",
"membership": "join"
},
"event_id": "$16345217l517tabbz:localhost",
"membership": "join",
"origin_server_ts": 1455123234,
"sender": format!("{}", user_id2),
"state_key": format!("{}", user_id2),
"type": "m.room.member",
"prev_content": {
"avatar_url": null,
"displayname": "example",
"membership": "invite"
},
"unsigned": {
"age": 1989321234,
"replaces_state": "$1622a2311315tkjoA:localhost"
}
});
let member1_invites_member2_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example",
"membership": "invite"
},
"event_id": "$16345217l517tabbz:localhost",
"membership": "invite",
"origin_server_ts": 1455123238,
"sender": format!("{}", user_id1),
"state_key": format!("{}", user_id2),
"type": "m.room.member",
"unsigned": {
"age": 1989321238,
"replaces_state": "$1622a2311315tkjoA:localhost"
}
});
let member2_name_change_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "changed",
"membership": "join"
},
"event_id": "$16345217l517tabbz:localhost",
"membership": "join",
"origin_server_ts": 1455123238,
"sender": format!("{}", user_id2),
"state_key": format!("{}", user_id2),
"type": "m.room.member",
"prev_content": {
"avatar_url": null,
"displayname": "example",
"membership": "join"
},
"unsigned": {
"age": 1989321238,
"replaces_state": "$1622a2311315tkjoA:localhost"
}
});
let member2_leave_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example",
"membership": "leave"
},
"event_id": "$263452333l22bggbz:localhost",
"membership": "leave",
"origin_server_ts": 1455123228,
"sender": format!("{}", user_id2),
"state_key": format!("{}", user_id2),
"type": "m.room.member",
"prev_content": {
"avatar_url": null,
"displayname": "example",
"membership": "join"
},
"unsigned": {
"age": 1989321221,
"replaces_state": "$16345217l517tabbz:localhost"
}
});
let member3_join_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example",
"membership": "join"
},
"event_id": "$16845287981ktggba:localhost",
"membership": "join",
"origin_server_ts": 1455123244,
"sender": format!("{}", user_id3),
"state_key": format!("{}", user_id3),
"type": "m.room.member",
"prev_content": {
"avatar_url": null,
"displayname": "example",
"membership": "invite"
},
"unsigned": {
"age": 1989321254,
"replaces_state": "$1622l2323445kabrA:localhost"
}
});
let member3_leave_event = serde_json::json!({
"content": {
"avatar_url": null,
"displayname": "example",
"membership": "leave"
},
"event_id": "$11121987981abfgr:localhost",
"membership": "leave",
"origin_server_ts": 1455123230,
"sender": format!("{}", user_id3),
"state_key": format!("{}", user_id3),
"type": "m.room.member",
"prev_content": {
"avatar_url": null,
"displayname": "example",
"membership": "join"
},
"unsigned": {
"age": 1989321244,
"replaces_state": "$16845287981ktggba:localhost"
}
});
let mut event_builder = EventBuilder::new();
let mut member1_join_sync_response = event_builder
.add_room_event(EventsJson::Member)
.build_sync_response();
let mut member2_join_sync_response = event_builder
.add_custom_joined_event(&room_id, member2_join_event.clone())
.build_sync_response();
let mut member3_join_sync_response = event_builder
.add_custom_joined_event(&room_id, member3_join_event)
.build_sync_response();
let mut member2_and_member3_leave_sync_response = event_builder
.add_custom_joined_event(&room_id, member2_leave_event)
.add_custom_joined_event(&room_id, member3_leave_event)
.build_sync_response();
let mut member2_rejoins_when_invited_sync_response = event_builder
.add_custom_joined_event(&room_id, member1_invites_member2_event)
.add_custom_joined_event(&room_id, member2_join_event)
.build_sync_response();
let mut member1_name_change_sync_response = event_builder
.add_room_event(EventsJson::MemberNameChange)
.build_sync_response();
let mut member2_name_change_sync_response = event_builder
.add_custom_joined_event(&room_id, member2_name_change_event)
.build_sync_response();
client
.receive_sync_response(&mut member1_join_sync_response)
.await
.unwrap();
{
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
let display_name1 = room.get_member(&user_id1).unwrap().disambiguated_name();
assert_eq!("example", display_name1);
}
client
.receive_sync_response(&mut member2_join_sync_response)
.await
.unwrap();
client
.receive_sync_response(&mut member3_join_sync_response)
.await
.unwrap();
{
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
let display_name1 = room.get_member(&user_id1).unwrap().disambiguated_name();
let display_name2 = room.get_member(&user_id2).unwrap().disambiguated_name();
let display_name3 = room.get_member(&user_id3).unwrap().disambiguated_name();
assert_eq!(format!("example ({})", user_id1), display_name1);
assert_eq!(format!("example ({})", user_id2), display_name2);
assert_eq!(format!("example ({})", user_id3), display_name3);
}
client
.receive_sync_response(&mut member2_and_member3_leave_sync_response)
.await
.unwrap();
{
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
let display_name1 = room.get_member(&user_id1).unwrap().disambiguated_name();
assert_eq!("example", display_name1);
}
client
.receive_sync_response(&mut member2_rejoins_when_invited_sync_response)
.await
.unwrap();
{
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
let display_name1 = room.get_member(&user_id1).unwrap().disambiguated_name();
let display_name2 = room.get_member(&user_id2).unwrap().disambiguated_name();
assert_eq!(format!("example ({})", user_id1), display_name1);
assert_eq!(format!("example ({})", user_id2), display_name2);
}
client
.receive_sync_response(&mut member1_name_change_sync_response)
.await
.unwrap();
{
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
let display_name1 = room.get_member(&user_id1).unwrap().disambiguated_name();
let display_name2 = room.get_member(&user_id2).unwrap().disambiguated_name();
assert_eq!("changed", display_name1);
assert_eq!("example", display_name2);
}
client
.receive_sync_response(&mut member2_name_change_sync_response)
.await
.unwrap();
{
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
let display_name1 = room.get_member(&user_id1).unwrap().disambiguated_name();
let display_name2 = room.get_member(&user_id2).unwrap().disambiguated_name();
assert_eq!(format!("changed ({})", user_id1), display_name1);
assert_eq!(format!("changed ({})", user_id2), display_name2);
}
}
#[async_test]
async fn room_events() {
let client = get_client().await;
let room_id = get_room_id();
let user_id = user_id!("@example:localhost");
let mut response = EventBuilder::default()
.add_state_event(EventsJson::Member)
.add_state_event(EventsJson::PowerLevels)
.build_sync_response();
client.receive_sync_response(&mut response).await.unwrap();
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
assert_eq!(room.joined_members.len(), 1);
assert!(room.power_levels.is_some());
assert_eq!(room.power_levels.as_ref().unwrap().kick, int!(50));
let admin = room.joined_members.get(&user_id).unwrap();
assert_eq!(admin.power_level.unwrap(), int!(100));
}
#[async_test]
async fn calculate_aliases() {
let client = get_client().await;
let room_id = get_room_id();
let mut response = EventBuilder::default()
.add_state_event(EventsJson::Aliases)
.build_sync_response();
client.receive_sync_response(&mut response).await.unwrap();
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
assert_eq!("tutorial", room.display_name());
}
#[async_test]
async fn calculate_alias() {
let client = get_client().await;
let room_id = get_room_id();
let mut response = EventBuilder::default()
.add_state_event(EventsJson::Alias)
.build_sync_response();
client.receive_sync_response(&mut response).await.unwrap();
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
assert_eq!("tutorial", room.display_name());
}
#[async_test]
async fn calculate_name() {
let client = get_client().await;
let room_id = get_room_id();
let mut response = EventBuilder::default()
.add_state_event(EventsJson::Name)
.build_sync_response();
client.receive_sync_response(&mut response).await.unwrap();
let room = client.get_joined_room(&room_id).await.unwrap();
let room = room.read().await;
assert_eq!("room name", room.display_name());
}
#[async_test]
async fn calculate_room_names_from_summary() {
let mut response = sync_response(SyncResponseFile::DefaultWithSummary);
let session = Session {
access_token: "1234".to_owned(),
user_id: user_id!("@example:localhost"),
device_id: "DEVICEID".into(),
};
let client = BaseClient::new().unwrap();
client.restore_login(session).await.unwrap();
client.receive_sync_response(&mut response).await.unwrap();
let mut room_names = vec![];
for room in client.joined_rooms().read().await.values() {
room_names.push(room.read().await.display_name())
}
assert_eq!(vec!["example2"], room_names);
}
#[cfg(feature = "messages")]
#[async_test]
async fn message_queue_redaction_event() {
let room_id = get_room_id();
let mut response = sync_response(SyncResponseFile::DefaultWithSummary);
let session = Session {
access_token: "1234".to_owned(),
user_id: user_id!("@example:localhost"),
device_id: "DEVICEID".into(),
};
let client = BaseClient::new().unwrap();
client.restore_login(session).await.unwrap();
client.receive_sync_response(&mut response).await.unwrap();
let json = serde_json::json!({
"content": {
"reason": "😀"
},
"event_id": "$151957878228ssqrJ:localhost",
"origin_server_ts": 151957878,
"sender": "@example:localhost",
"type": "m.room.redaction",
"redacts": "$152037280074GZeOm:localhost"
});
let mut event: Raw<AnySyncRoomEvent> = serde_json::from_value(json).unwrap();
client
.receive_joined_timeline_event(&room_id, &mut event)
.await
.unwrap();
for room in client.joined_rooms().read().await.values() {
let queue = &room.read().await.messages;
if let crate::events::AnyPossiblyRedactedSyncMessageEvent::Redacted(
crate::events::AnyRedactedSyncMessageEvent::RoomMessage(event),
) = &queue.msgs[0]
{
assert_eq!(event.event_id, event_id!("$152037280074GZeOm:localhost"))
} else {
panic!("message event in message queue should be redacted")
}
}
}
#[async_test]
#[cfg(not(target_arch = "wasm32"))]
async fn encryption_info_test() {
let room_id = get_room_id();
let mut response = sync_response(SyncResponseFile::DefaultWithSummary);
let user_id = user_id!("@example:localhost");
let session = Session {
access_token: "1234".to_owned(),
user_id: user_id.clone(),
device_id: "DEVICEID".into(),
};
let client = BaseClient::new().unwrap();
client.restore_login(session).await.unwrap();
client.receive_sync_response(&mut response).await.unwrap();
let mut content = EncryptionEventContent::new(EventEncryptionAlgorithm::MegolmV1AesSha2);
content.rotation_period_ms = Some(100_000u32.into());
content.rotation_period_msgs = Some(100u32.into());
let event = SyncStateEvent {
event_id: event_id!("$h29iv0s8:example.com"),
origin_server_ts: SystemTime::now(),
sender: user_id,
state_key: "".into(),
unsigned: Unsigned::default(),
content,
prev_content: None,
};
let room = client.get_joined_room(&room_id).await.unwrap();
assert!(!room.read().await.is_encrypted());
room.write().await.handle_encryption_event(&event);
assert!(room.read().await.is_encrypted());
let room_lock = room.read().await;
let encryption_info = room_lock.encryption_info().unwrap();
assert_eq!(
encryption_info.algorithm(),
&EventEncryptionAlgorithm::MegolmV1AesSha2
);
assert_eq!(encryption_info.rotation_period(), 100_000);
assert_eq!(encryption_info.rotation_period_messages(), 100);
}
#[test]
fn power_level_overflow() {
let room_id = get_room_id();
let user_id = user_id!("@example:localhost");
let content = MemberEventContent {
avatar_url: None,
displayname: Some("user1".into()),
is_direct: Some(false),
membership: MembershipState::Join,
third_party_invite: None,
};
let member = SyncStateEvent {
event_id: event_id!("$h29iv0s8:example.com"),
origin_server_ts: SystemTime::now(),
sender: user_id.clone(),
state_key: "@example:localhost".into(),
unsigned: Unsigned::default(),
content,
prev_content: None,
};
let mut room_member = RoomMember::new(&member, &room_id);
let mut content = PowerLevelsEventContent::default();
*content
.users
.entry(user_id.clone())
.or_insert_with(|| Int::new(4503599627370495).unwrap()) =
Int::new(4503599627370495).unwrap();
let power = SyncStateEvent {
event_id: event_id!("$h29iv0s8:example.com"),
origin_server_ts: SystemTime::now(),
sender: user_id,
state_key: "".into(),
unsigned: Unsigned::default(),
content,
prev_content: None,
};
Room::update_member_power(
&mut room_member,
&power,
Int::new(4503599627370495).unwrap(),
);
assert_eq!(room_member.power_level_norm, Some(int!(100)))
}
}