2020-03-28 10:58:30 +00:00
|
|
|
// Copyright 2020 Damir Jelić
|
|
|
|
// Copyright 2020 The Matrix.org Foundation C.I.C.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
2020-05-14 15:09:18 +00:00
|
|
|
use std::borrow::Cow;
|
2020-04-23 08:52:47 +00:00
|
|
|
use std::collections::{BTreeMap, HashMap};
|
2020-04-03 13:07:40 +00:00
|
|
|
use std::convert::TryFrom;
|
2020-03-28 10:58:30 +00:00
|
|
|
|
2020-05-07 01:17:21 +00:00
|
|
|
#[cfg(feature = "messages")]
|
2020-04-29 11:58:31 +00:00
|
|
|
use super::message::MessageQueue;
|
2020-04-03 13:07:40 +00:00
|
|
|
use super::RoomMember;
|
2020-03-29 19:54:26 +00:00
|
|
|
|
2020-05-07 19:21:06 +00:00
|
|
|
use crate::api::r0::sync::sync_events::{RoomSummary, UnreadNotificationsCount};
|
2020-03-28 10:58:30 +00:00
|
|
|
use crate::events::collections::all::{RoomEvent, StateEvent};
|
2020-07-01 13:44:44 +00:00
|
|
|
use crate::events::presence::{PresenceEvent, PresenceEventContent};
|
2020-03-28 10:58:30 +00:00
|
|
|
use crate::events::room::{
|
|
|
|
aliases::AliasesEvent,
|
|
|
|
canonical_alias::CanonicalAliasEvent,
|
2020-04-01 14:06:28 +00:00
|
|
|
encryption::EncryptionEvent,
|
2020-03-29 12:05:40 +00:00
|
|
|
member::{MemberEvent, MembershipChange},
|
2020-03-28 10:58:30 +00:00
|
|
|
name::NameEvent,
|
2020-04-05 16:41:00 +00:00
|
|
|
power_levels::{NotificationPowerLevels, PowerLevelsEvent, PowerLevelsEventContent},
|
2020-04-23 10:45:00 +00:00
|
|
|
tombstone::TombstoneEvent,
|
2020-03-28 10:58:30 +00:00
|
|
|
};
|
2020-05-07 00:46:06 +00:00
|
|
|
use crate::events::stripped::{AnyStrippedStateEvent, StrippedRoomName};
|
2020-05-14 11:58:09 +00:00
|
|
|
use crate::events::{Algorithm, EventType};
|
2020-05-07 01:17:21 +00:00
|
|
|
|
|
|
|
#[cfg(feature = "messages")]
|
|
|
|
use crate::events::room::message::MessageEvent;
|
|
|
|
|
2020-04-03 13:14:46 +00:00
|
|
|
use crate::identifiers::{RoomAliasId, RoomId, UserId};
|
2020-03-28 10:58:30 +00:00
|
|
|
|
2020-04-29 08:40:27 +00:00
|
|
|
use crate::js_int::{Int, UInt};
|
2020-04-16 14:02:59 +00:00
|
|
|
use serde::{Deserialize, Serialize};
|
2020-06-17 16:52:53 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Serialize, Deserialize, Clone)]
|
2020-03-28 10:58:30 +00:00
|
|
|
/// `RoomName` allows the calculation of a text room name.
|
|
|
|
pub struct RoomName {
|
|
|
|
/// The displayed name of the room.
|
|
|
|
name: Option<String>,
|
|
|
|
/// The canonical alias of the room ex. `#room-name:example.com` and port number.
|
|
|
|
canonical_alias: Option<RoomAliasId>,
|
|
|
|
/// List of `RoomAliasId`s the room has been given.
|
|
|
|
aliases: Vec<RoomAliasId>,
|
2020-04-17 01:00:50 +00:00
|
|
|
/// Users which can be used to generate a room name if the room does not have
|
|
|
|
/// one. Required if room name or canonical aliases are not set or empty.
|
|
|
|
pub heroes: Vec<String>,
|
|
|
|
/// Number of users whose membership status is `join`.
|
|
|
|
/// Required if field has changed since last sync; otherwise, it may be
|
|
|
|
/// omitted.
|
|
|
|
pub joined_member_count: Option<UInt>,
|
|
|
|
/// Number of users whose membership status is `invite`.
|
|
|
|
/// Required if field has changed since last sync; otherwise, it may be
|
|
|
|
/// omitted.
|
|
|
|
pub invited_member_count: Option<UInt>,
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
|
2020-06-17 16:52:53 +00:00
|
|
|
#[derive(Debug, PartialEq, Eq, Serialize, Deserialize, Clone)]
|
2020-04-05 16:41:00 +00:00
|
|
|
pub struct PowerLevels {
|
|
|
|
/// The level required to ban a user.
|
|
|
|
pub ban: Int,
|
|
|
|
/// The level required to send specific event types.
|
|
|
|
///
|
|
|
|
/// This is a mapping from event type to power level required.
|
2020-04-23 08:52:47 +00:00
|
|
|
pub events: BTreeMap<EventType, Int>,
|
2020-04-05 16:41:00 +00:00
|
|
|
/// The default level required to send message events.
|
|
|
|
pub events_default: Int,
|
|
|
|
/// The level required to invite a user.
|
|
|
|
pub invite: Int,
|
|
|
|
/// The level required to kick a user.
|
|
|
|
pub kick: Int,
|
|
|
|
/// The level required to redact an event.
|
|
|
|
pub redact: Int,
|
|
|
|
/// The default level required to send state events.
|
|
|
|
pub state_default: Int,
|
|
|
|
/// The default power level for every user in the room.
|
|
|
|
pub users_default: Int,
|
|
|
|
/// The power level requirements for specific notification types.
|
|
|
|
///
|
|
|
|
/// This is a mapping from `key` to power level for that notifications key.
|
|
|
|
pub notifications: Int,
|
|
|
|
}
|
|
|
|
|
2020-05-14 11:58:09 +00:00
|
|
|
#[derive(Debug, PartialEq, Serialize, Deserialize, Clone)]
|
|
|
|
/// Encryption info of the room.
|
|
|
|
pub struct EncryptionInfo {
|
|
|
|
/// The encryption algorithm that should be used to encrypt messages in the
|
|
|
|
/// room.
|
|
|
|
algorithm: Algorithm,
|
|
|
|
/// How long should a session be used before it is rotated.
|
|
|
|
rotation_period_ms: u64,
|
|
|
|
/// The maximum amount of messages that should be encrypted using the same
|
|
|
|
/// session.
|
|
|
|
rotation_period_messages: u64,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl EncryptionInfo {
|
|
|
|
/// The encryption algorithm that should be used to encrypt messages in the
|
|
|
|
/// room.
|
|
|
|
pub fn algorithm(&self) -> &Algorithm {
|
|
|
|
&self.algorithm
|
|
|
|
}
|
|
|
|
|
|
|
|
/// How long should a session be used before it is rotated.
|
|
|
|
pub fn rotation_period(&self) -> u64 {
|
|
|
|
self.rotation_period_ms
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The maximum amount of messages that should be encrypted using the same
|
|
|
|
/// session.
|
|
|
|
pub fn rotation_period_messages(&self) -> u64 {
|
|
|
|
self.rotation_period_messages
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl From<&EncryptionEvent> for EncryptionInfo {
|
|
|
|
fn from(event: &EncryptionEvent) -> Self {
|
|
|
|
EncryptionInfo {
|
|
|
|
algorithm: event.content.algorithm.clone(),
|
|
|
|
rotation_period_ms: event
|
|
|
|
.content
|
|
|
|
.rotation_period_ms
|
2020-05-14 12:52:32 +00:00
|
|
|
.map_or(604_800_000, Into::into),
|
2020-05-14 11:58:09 +00:00
|
|
|
rotation_period_messages: event.content.rotation_period_msgs.map_or(100, Into::into),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-17 16:52:53 +00:00
|
|
|
#[derive(Debug, PartialEq, Eq, Serialize, Deserialize, Clone)]
|
2020-04-23 10:45:00 +00:00
|
|
|
pub struct Tombstone {
|
|
|
|
/// A server-defined message.
|
|
|
|
body: String,
|
|
|
|
/// The room that is now active.
|
|
|
|
replacement: RoomId,
|
|
|
|
}
|
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
#[derive(Debug, PartialEq, Eq)]
|
|
|
|
enum MemberDirection {
|
|
|
|
Entering,
|
|
|
|
Exiting,
|
|
|
|
}
|
|
|
|
|
2020-06-17 16:52:53 +00:00
|
|
|
#[derive(Debug, PartialEq, Serialize, Deserialize, Clone)]
|
2020-04-23 10:45:00 +00:00
|
|
|
/// A Matrix room.
|
2020-03-28 10:58:30 +00:00
|
|
|
pub struct Room {
|
|
|
|
/// The unique id of the room.
|
2020-04-03 13:07:40 +00:00
|
|
|
pub room_id: RoomId,
|
2020-03-28 10:58:30 +00:00
|
|
|
/// The name of the room, clients use this to represent a room.
|
|
|
|
pub room_name: RoomName,
|
|
|
|
/// The mxid of our own user.
|
|
|
|
pub own_user_id: UserId,
|
|
|
|
/// The mxid of the room creator.
|
|
|
|
pub creator: Option<UserId>,
|
2020-06-10 16:12:27 +00:00
|
|
|
// TODO: Track banned members, e.g. for /unban support?
|
|
|
|
/// The map of invited room members.
|
|
|
|
pub invited_members: HashMap<UserId, RoomMember>,
|
|
|
|
/// The map of joined room members.
|
|
|
|
pub joined_members: HashMap<UserId, RoomMember>,
|
2020-04-30 10:28:36 +00:00
|
|
|
/// A queue of messages, holds no more than 10 of the most recent messages.
|
2020-04-29 11:58:31 +00:00
|
|
|
///
|
|
|
|
/// This is helpful when using a `StateStore` to avoid multiple requests
|
|
|
|
/// to the server for messages.
|
2020-05-07 01:17:21 +00:00
|
|
|
#[cfg(feature = "messages")]
|
|
|
|
#[cfg_attr(docsrs, doc(cfg(feature = "messages")))]
|
2020-04-29 11:58:31 +00:00
|
|
|
#[serde(with = "super::message::ser_deser")]
|
|
|
|
pub messages: MessageQueue,
|
2020-03-28 10:58:30 +00:00
|
|
|
/// A list of users that are currently typing.
|
|
|
|
pub typing_users: Vec<UserId>,
|
2020-04-05 16:41:00 +00:00
|
|
|
/// The power level requirements for specific actions in this room
|
|
|
|
pub power_levels: Option<PowerLevels>,
|
2020-05-14 11:58:09 +00:00
|
|
|
/// Optional encryption info, will be `Some` if the room is encrypted.
|
|
|
|
pub encrypted: Option<EncryptionInfo>,
|
2020-03-29 19:54:26 +00:00
|
|
|
/// Number of unread notifications with highlight flag set.
|
|
|
|
pub unread_highlight: Option<UInt>,
|
|
|
|
/// Number of unread notifications.
|
|
|
|
pub unread_notifications: Option<UInt>,
|
2020-04-23 10:45:00 +00:00
|
|
|
/// The tombstone state of this room.
|
|
|
|
pub tombstone: Option<Tombstone>,
|
2020-06-09 09:10:09 +00:00
|
|
|
/// The map of disambiguated display names for users who have the same display name
|
|
|
|
disambiguated_display_names: HashMap<UserId, String>,
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2020-06-09 22:28:56 +00:00
|
|
|
/// Calculate the canonical display name of a room, taking into account its name, aliases and
|
|
|
|
/// members.
|
|
|
|
///
|
|
|
|
/// The display name is calculated according to [this algorithm][spec].
|
|
|
|
///
|
|
|
|
/// [spec]:
|
|
|
|
/// <https://matrix.org/docs/spec/client_server/latest#calculating-the-display-name-for-a-room>
|
2020-06-10 16:12:27 +00:00
|
|
|
pub fn calculate_name(
|
|
|
|
&self,
|
2020-06-10 20:45:08 +00:00
|
|
|
own_user_id: &UserId,
|
2020-06-10 16:12:27 +00:00
|
|
|
invited_members: &HashMap<UserId, RoomMember>,
|
|
|
|
joined_members: &HashMap<UserId, RoomMember>,
|
|
|
|
) -> String {
|
2020-03-28 10:58:30 +00:00
|
|
|
if let Some(name) = &self.name {
|
2020-04-23 10:53:34 +00:00
|
|
|
let name = name.trim();
|
|
|
|
name.to_string()
|
2020-03-28 10:58:30 +00:00
|
|
|
} else if let Some(alias) = &self.canonical_alias {
|
2020-04-23 10:53:34 +00:00
|
|
|
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()
|
2020-03-28 10:58:30 +00:00
|
|
|
} else {
|
2020-04-27 10:12:45 +00:00
|
|
|
let joined = self.joined_member_count.unwrap_or(UInt::MIN);
|
|
|
|
let invited = self.invited_member_count.unwrap_or(UInt::MIN);
|
2020-04-17 01:00:50 +00:00
|
|
|
let heroes = UInt::new(self.heroes.len() as u64).unwrap();
|
|
|
|
let one = UInt::new(1).unwrap();
|
2020-03-28 10:58:30 +00:00
|
|
|
|
2020-04-27 10:12:45 +00:00
|
|
|
let invited_joined = if invited + joined == UInt::MIN {
|
|
|
|
UInt::MIN
|
2020-04-23 10:53:34 +00:00
|
|
|
} else {
|
|
|
|
invited + joined - one
|
|
|
|
};
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
let members = joined_members.values().chain(invited_members.values());
|
|
|
|
|
2020-06-10 10:04:58 +00:00
|
|
|
// TODO: This should use `self.heroes` but it is always empty??
|
2020-04-23 10:53:34 +00:00
|
|
|
if heroes >= invited_joined {
|
|
|
|
let mut names = members
|
2020-06-10 20:45:08 +00:00
|
|
|
.filter(|m| m.user_id != *own_user_id)
|
2020-04-23 10:53:34 +00:00
|
|
|
.take(3)
|
2020-04-23 13:55:59 +00:00
|
|
|
.map(|mem| {
|
|
|
|
mem.display_name
|
|
|
|
.clone()
|
2020-04-27 10:12:51 +00:00
|
|
|
.unwrap_or_else(|| mem.user_id.localpart().to_string())
|
2020-04-23 13:55:59 +00:00
|
|
|
})
|
2020-04-23 10:53:34 +00:00
|
|
|
.collect::<Vec<String>>();
|
2020-04-23 11:05:59 +00:00
|
|
|
// stabilize ordering
|
2020-03-28 10:58:30 +00:00
|
|
|
names.sort();
|
|
|
|
names.join(", ")
|
2020-04-23 10:53:34 +00:00
|
|
|
} else if heroes < invited_joined && invited + joined > one {
|
|
|
|
let mut names = members
|
2020-06-10 20:45:08 +00:00
|
|
|
.filter(|m| m.user_id != *own_user_id)
|
2020-04-23 10:53:34 +00:00
|
|
|
.take(3)
|
2020-04-23 13:55:59 +00:00
|
|
|
.map(|mem| {
|
|
|
|
mem.display_name
|
|
|
|
.clone()
|
2020-04-27 10:12:51 +00:00
|
|
|
.unwrap_or_else(|| mem.user_id.localpart().to_string())
|
2020-04-23 13:55:59 +00:00
|
|
|
})
|
2020-04-23 10:53:34 +00:00
|
|
|
.collect::<Vec<String>>();
|
2020-04-17 01:00:50 +00:00
|
|
|
names.sort();
|
2020-06-10 10:04:58 +00:00
|
|
|
|
|
|
|
// TODO: What length does the spec want us to use here and in the `else`?
|
2020-04-17 01:00:50 +00:00
|
|
|
format!("{}, and {} others", names.join(", "), (joined + invited))
|
|
|
|
} else {
|
2020-06-10 16:12:27 +00:00
|
|
|
"Empty room".to_string()
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Room {
|
|
|
|
/// Create a new room.
|
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
|
|
|
/// * `room_id` - The unique id of the room.
|
|
|
|
///
|
|
|
|
/// * `own_user_id` - The mxid of our own user.
|
2020-04-03 13:07:40 +00:00
|
|
|
pub fn new(room_id: &RoomId, own_user_id: &UserId) -> Self {
|
2020-03-28 10:58:30 +00:00
|
|
|
Room {
|
2020-04-03 13:07:40 +00:00
|
|
|
room_id: room_id.clone(),
|
2020-03-28 10:58:30 +00:00
|
|
|
room_name: RoomName::default(),
|
2020-04-03 13:07:40 +00:00
|
|
|
own_user_id: own_user_id.clone(),
|
2020-03-28 10:58:30 +00:00
|
|
|
creator: None,
|
2020-06-10 16:12:27 +00:00
|
|
|
invited_members: HashMap::new(),
|
|
|
|
joined_members: HashMap::new(),
|
2020-05-07 01:17:21 +00:00
|
|
|
#[cfg(feature = "messages")]
|
2020-04-29 11:58:31 +00:00
|
|
|
messages: MessageQueue::new(),
|
2020-03-28 10:58:30 +00:00
|
|
|
typing_users: Vec::new(),
|
2020-04-05 16:41:00 +00:00
|
|
|
power_levels: None,
|
2020-05-14 11:58:09 +00:00
|
|
|
encrypted: None,
|
2020-03-29 19:54:26 +00:00
|
|
|
unread_highlight: None,
|
|
|
|
unread_notifications: None,
|
2020-04-23 10:45:00 +00:00
|
|
|
tombstone: None,
|
2020-06-09 09:10:09 +00:00
|
|
|
disambiguated_display_names: HashMap::new(),
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-02 22:22:52 +00:00
|
|
|
/// Return the display name of the room.
|
2020-05-07 10:51:53 +00:00
|
|
|
pub fn display_name(&self) -> String {
|
2020-06-15 15:11:12 +00:00
|
|
|
self.room_name.calculate_name(
|
|
|
|
&self.own_user_id,
|
|
|
|
&self.invited_members,
|
|
|
|
&self.joined_members,
|
|
|
|
)
|
2020-03-31 19:28:48 +00:00
|
|
|
}
|
|
|
|
|
2020-04-01 13:37:00 +00:00
|
|
|
/// Is the room a encrypted room.
|
|
|
|
pub fn is_encrypted(&self) -> bool {
|
2020-05-14 11:58:09 +00:00
|
|
|
self.encrypted.is_some()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Get the encryption info if any of the room.
|
|
|
|
///
|
|
|
|
/// Returns None if the room is not encrypted.
|
|
|
|
pub fn encryption_info(&self) -> Option<&EncryptionInfo> {
|
|
|
|
self.encrypted.as_ref()
|
2020-04-01 13:37:00 +00:00
|
|
|
}
|
|
|
|
|
2020-06-09 13:29:37 +00:00
|
|
|
/// Get the disambiguated display name for a member of this room.
|
2020-06-09 13:35:43 +00:00
|
|
|
///
|
2020-06-09 21:02:01 +00:00
|
|
|
/// If a member has no display name set, returns the MXID as a fallback. Additionally, we
|
|
|
|
/// return the MXID even if there is no such member in the room.
|
2020-06-09 14:11:41 +00:00
|
|
|
///
|
2020-06-09 13:35:43 +00:00
|
|
|
/// When displaying a room member's display name, clients *must* use this method to obtain the
|
|
|
|
/// name instead of displaying the `RoomMember::display_name` directly. This is because
|
2020-06-09 21:02:01 +00:00
|
|
|
/// multiple members can share the same display name in which case the display name has to be
|
2020-06-09 13:35:43 +00:00
|
|
|
/// disambiguated.
|
2020-06-09 14:11:41 +00:00
|
|
|
pub fn member_display_name<'a>(&'a self, id: &'a UserId) -> Cow<'a, str> {
|
2020-06-09 21:02:01 +00:00
|
|
|
let disambiguated_name = self
|
|
|
|
.disambiguated_display_names
|
2020-05-14 15:09:18 +00:00
|
|
|
.get(id)
|
2020-06-09 21:02:01 +00:00
|
|
|
.map(|s| s.as_str().into());
|
|
|
|
|
|
|
|
if let Some(name) = disambiguated_name {
|
|
|
|
// The display name of the member is non-unique so we return a disambiguated version.
|
|
|
|
name
|
2020-06-10 16:12:27 +00:00
|
|
|
} else if let Some(member) = self
|
|
|
|
.joined_members
|
|
|
|
.get(id)
|
|
|
|
.or_else(|| self.invited_members.get(id))
|
|
|
|
{
|
2020-06-09 21:02:01 +00:00
|
|
|
// The display name of the member is unique so we can return it directly if it is set.
|
|
|
|
// If not, we return his MXID.
|
2020-06-10 16:12:27 +00:00
|
|
|
member.name().into()
|
2020-06-09 21:02:01 +00:00
|
|
|
} else {
|
|
|
|
// There is no member with the requested MXID in the room. We still return the MXID.
|
|
|
|
id.as_ref().into()
|
|
|
|
}
|
2020-05-13 14:18:07 +00:00
|
|
|
}
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
/// Process the member event of an entering user.
|
2020-07-01 14:08:25 +00:00
|
|
|
fn add_member(&mut self, event: &MemberEvent) {
|
2020-06-10 10:01:01 +00:00
|
|
|
let new_member = RoomMember::new(event);
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
match event.membership_change() {
|
2020-07-01 12:53:58 +00:00
|
|
|
MembershipChange::Joined => {
|
|
|
|
// Since the member is now joined, he shouldn't be tracked as an invited member any
|
|
|
|
// longer.
|
|
|
|
if self.invited_members.contains_key(&new_member.user_id) {
|
|
|
|
self.invited_members.remove(&new_member.user_id);
|
|
|
|
}
|
|
|
|
|
|
|
|
self.joined_members
|
|
|
|
.insert(new_member.user_id.clone(), new_member.clone())
|
|
|
|
}
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
MembershipChange::Invited => self
|
|
|
|
.invited_members
|
|
|
|
.insert(new_member.user_id.clone(), new_member.clone()),
|
2020-07-01 12:53:58 +00:00
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
_ => {
|
|
|
|
panic!("Room::add_member called on an event that is neither a join nor an invite.")
|
|
|
|
}
|
|
|
|
};
|
2020-04-05 16:41:00 +00:00
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
// Perform display name disambiguations, if necessary.
|
2020-07-01 14:08:25 +00:00
|
|
|
let disambiguations =
|
|
|
|
self.disambiguation_updates(&new_member, MemberDirection::Entering);
|
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
for (id, name) in disambiguations.into_iter() {
|
|
|
|
match name {
|
|
|
|
None => self.disambiguated_display_names.remove(&id),
|
|
|
|
Some(name) => self.disambiguated_display_names.insert(id, name),
|
|
|
|
};
|
2020-06-10 16:12:27 +00:00
|
|
|
}
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
/// Process the member event of a leaving user.
|
|
|
|
///
|
|
|
|
/// Returns true if this made a change to the room's state, false otherwise.
|
|
|
|
fn remove_member(&mut self, event: &MemberEvent) -> bool {
|
|
|
|
let leaving_member = RoomMember::new(event);
|
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
// Perform display name disambiguations, if necessary.
|
|
|
|
let disambiguations =
|
|
|
|
self.disambiguation_updates(&leaving_member, MemberDirection::Exiting);
|
|
|
|
for (id, name) in disambiguations.into_iter() {
|
|
|
|
match name {
|
|
|
|
None => self.disambiguated_display_names.remove(&id),
|
|
|
|
Some(name) => self.disambiguated_display_names.insert(id, name),
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
if self.joined_members.contains_key(&leaving_member.user_id) {
|
|
|
|
self.joined_members.remove(&leaving_member.user_id);
|
|
|
|
true
|
|
|
|
} else if self.invited_members.contains_key(&leaving_member.user_id) {
|
|
|
|
self.invited_members.remove(&leaving_member.user_id);
|
|
|
|
true
|
|
|
|
} else {
|
|
|
|
false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
/// Given a room `member`, return the list of members which have the same display name.
|
2020-06-10 16:12:27 +00:00
|
|
|
///
|
2020-06-15 15:11:12 +00:00
|
|
|
/// The `inclusive` parameter controls whether the passed member should be included in the
|
|
|
|
/// list or not.
|
2020-07-01 10:06:59 +00:00
|
|
|
fn shares_displayname_with(&self, member: &RoomMember, inclusive: bool) -> Vec<RoomMember> {
|
2020-06-10 16:12:27 +00:00
|
|
|
let members = self
|
|
|
|
.invited_members
|
2020-05-13 14:18:07 +00:00
|
|
|
.iter()
|
2020-06-10 16:12:27 +00:00
|
|
|
.chain(self.joined_members.iter());
|
|
|
|
|
|
|
|
// Find all other users that share the same display name as the joining user.
|
2020-06-15 15:11:12 +00:00
|
|
|
members
|
2020-06-09 13:17:38 +00:00
|
|
|
.filter(|(_, existing_member)| {
|
2020-06-10 16:12:27 +00:00
|
|
|
member
|
2020-05-13 14:18:07 +00:00
|
|
|
.display_name
|
|
|
|
.as_ref()
|
2020-06-09 13:17:38 +00:00
|
|
|
.and_then(|new_member_name| {
|
|
|
|
existing_member
|
|
|
|
.display_name
|
|
|
|
.as_ref()
|
2020-06-10 14:36:51 +00:00
|
|
|
.map(|existing_member_name| new_member_name == existing_member_name)
|
2020-06-09 13:17:38 +00:00
|
|
|
})
|
2020-05-13 14:18:07 +00:00
|
|
|
.unwrap_or(false)
|
|
|
|
})
|
2020-06-10 16:12:27 +00:00
|
|
|
// If not an inclusive search, do not consider the member for which we are disambiguating.
|
|
|
|
.filter(|(id, _)| inclusive || **id != member.user_id)
|
2020-07-01 10:06:59 +00:00
|
|
|
.map(|(_, member)| member)
|
2020-05-13 14:18:07 +00:00
|
|
|
.cloned()
|
2020-06-15 15:11:12 +00:00
|
|
|
.collect()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Given a room member, generate a map of all display name disambiguations which are necessary
|
|
|
|
/// in order to make that member's display name unique.
|
|
|
|
///
|
|
|
|
/// The `inclusive` parameter controls whether or not the member for which we are
|
|
|
|
/// disambiguating should be considered a current member of the room.
|
|
|
|
///
|
|
|
|
/// Returns a map from MXID to disambiguated name.
|
|
|
|
fn member_disambiguations(
|
|
|
|
&self,
|
|
|
|
member: &RoomMember,
|
|
|
|
inclusive: bool,
|
|
|
|
) -> HashMap<UserId, String> {
|
|
|
|
let users_with_same_name = self.shares_displayname_with(member, inclusive);
|
2020-07-01 10:06:59 +00:00
|
|
|
let disambiguate_with = |members: Vec<RoomMember>, f: fn(&RoomMember) -> String| {
|
2020-06-15 15:11:12 +00:00
|
|
|
members
|
|
|
|
.into_iter()
|
2020-07-01 10:06:59 +00:00
|
|
|
.map(|ref m| (m.user_id.clone(), f(m)))
|
2020-06-15 15:11:12 +00:00
|
|
|
.collect::<HashMap<UserId, String>>()
|
|
|
|
};
|
|
|
|
|
|
|
|
match users_with_same_name.len() {
|
2020-07-01 13:56:29 +00:00
|
|
|
0 | 1 => HashMap::new(),
|
2020-06-15 15:11:12 +00:00
|
|
|
_ => disambiguate_with(users_with_same_name, |m: &RoomMember| m.unique_name()),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Calculate disambiguation updates needed when a room member either enters or exits.
|
|
|
|
fn disambiguation_updates(
|
|
|
|
&self,
|
|
|
|
member: &RoomMember,
|
|
|
|
when: MemberDirection,
|
|
|
|
) -> HashMap<UserId, Option<String>> {
|
|
|
|
let before;
|
|
|
|
let after;
|
|
|
|
|
|
|
|
match when {
|
|
|
|
MemberDirection::Entering => {
|
|
|
|
before = self.member_disambiguations(member, false);
|
|
|
|
after = self.member_disambiguations(member, true);
|
|
|
|
}
|
|
|
|
MemberDirection::Exiting => {
|
|
|
|
before = self.member_disambiguations(member, true);
|
|
|
|
after = self.member_disambiguations(member, false);
|
2020-05-13 14:18:07 +00:00
|
|
|
}
|
|
|
|
}
|
2020-06-15 15:11:12 +00:00
|
|
|
|
|
|
|
let mut res = before;
|
|
|
|
res.extend(after.clone());
|
|
|
|
|
|
|
|
res.into_iter()
|
|
|
|
.map(|(user_id, name)| {
|
|
|
|
if !after.contains_key(&user_id) {
|
|
|
|
(user_id, None)
|
|
|
|
} else {
|
|
|
|
(user_id, Some(name))
|
|
|
|
}
|
|
|
|
})
|
|
|
|
.collect()
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Add to the list of `RoomAliasId`s.
|
2020-04-03 13:07:40 +00:00
|
|
|
fn push_room_alias(&mut self, alias: &RoomAliasId) -> bool {
|
2020-03-28 10:58:30 +00:00
|
|
|
self.room_name.push_alias(alias.clone());
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
|
|
|
/// RoomAliasId is `#alias:hostname` and `port`
|
|
|
|
fn canonical_alias(&mut self, alias: &RoomAliasId) -> bool {
|
|
|
|
self.room_name.set_canonical(alias.clone());
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
2020-04-23 13:55:59 +00:00
|
|
|
fn set_room_name(&mut self, name: &str) -> bool {
|
2020-03-28 10:58:30 +00:00
|
|
|
self.room_name.set_name(name);
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
2020-04-05 16:41:00 +00:00
|
|
|
fn set_room_power_level(&mut self, event: &PowerLevelsEvent) -> 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
|
|
|
|
}
|
|
|
|
|
2020-04-17 01:00:50 +00:00
|
|
|
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();
|
2020-04-23 10:45:00 +00:00
|
|
|
self.room_name.invited_member_count = *invited_member_count;
|
|
|
|
self.room_name.joined_member_count = *joined_member_count;
|
2020-04-17 01:00:50 +00:00
|
|
|
}
|
|
|
|
|
2020-05-07 19:21:06 +00:00
|
|
|
pub(crate) fn set_unread_notice_count(&mut self, notifications: &UnreadNotificationsCount) {
|
|
|
|
self.unread_highlight = notifications.highlight_count;
|
|
|
|
self.unread_notifications = notifications.notification_count;
|
|
|
|
}
|
|
|
|
|
2020-03-29 19:54:26 +00:00
|
|
|
/// Handle a room.member updating the room state if necessary.
|
|
|
|
///
|
|
|
|
/// Returns true if the joined member list changed, false otherwise.
|
|
|
|
pub fn handle_membership(&mut self, event: &MemberEvent) -> bool {
|
2020-06-10 14:36:51 +00:00
|
|
|
use MembershipChange::*;
|
|
|
|
|
2020-03-29 19:54:26 +00:00
|
|
|
match event.membership_change() {
|
2020-07-01 13:44:44 +00:00
|
|
|
Invited | Joined => {
|
2020-07-01 14:08:25 +00:00
|
|
|
self.add_member(event);
|
|
|
|
|
|
|
|
true
|
2020-07-01 13:44:44 +00:00
|
|
|
}
|
2020-06-10 16:12:27 +00:00
|
|
|
Kicked | Banned | KickedAndBanned | InvitationRejected | Left => {
|
|
|
|
self.remove_member(event)
|
|
|
|
}
|
2020-06-10 14:36:51 +00:00
|
|
|
ProfileChanged => {
|
2020-07-01 13:44:44 +00:00
|
|
|
self.update_member_profile(event)
|
2020-03-29 19:54:26 +00:00
|
|
|
}
|
2020-06-10 14:36:51 +00:00
|
|
|
|
|
|
|
// Not interested in other events.
|
|
|
|
_ => false,
|
2020-03-29 19:54:26 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-29 11:58:31 +00:00
|
|
|
/// Handle a room.message event and update the `MessageQueue` if necessary.
|
|
|
|
///
|
|
|
|
/// Returns true if `MessageQueue` was added to.
|
2020-05-07 01:17:21 +00:00
|
|
|
#[cfg(feature = "messages")]
|
|
|
|
#[cfg_attr(docsrs, doc(cfg(feature = "messages")))]
|
2020-04-29 11:58:31 +00:00
|
|
|
pub fn handle_message(&mut self, event: &MessageEvent) -> bool {
|
|
|
|
self.messages.push(event.clone())
|
|
|
|
}
|
|
|
|
|
2020-03-28 10:58:30 +00:00
|
|
|
/// Handle a room.aliases event, updating the room state if necessary.
|
2020-03-28 12:58:02 +00:00
|
|
|
///
|
2020-03-28 10:58:30 +00:00
|
|
|
/// Returns true if the room name changed, false otherwise.
|
|
|
|
pub fn handle_room_aliases(&mut self, event: &AliasesEvent) -> bool {
|
|
|
|
match event.content.aliases.as_slice() {
|
2020-04-03 13:07:40 +00:00
|
|
|
[alias] => self.push_room_alias(alias),
|
|
|
|
[alias, ..] => self.push_room_alias(alias),
|
2020-03-28 10:58:30 +00:00
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Handle a room.canonical_alias event, updating the room state if necessary.
|
2020-03-28 12:58:02 +00:00
|
|
|
///
|
2020-03-28 10:58:30 +00:00
|
|
|
/// Returns true if the room name changed, false otherwise.
|
|
|
|
pub fn handle_canonical(&mut self, event: &CanonicalAliasEvent) -> bool {
|
|
|
|
match &event.content.alias {
|
|
|
|
Some(name) => self.canonical_alias(&name),
|
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Handle a room.name event, updating the room state if necessary.
|
2020-03-28 12:58:02 +00:00
|
|
|
///
|
2020-03-28 10:58:30 +00:00
|
|
|
/// Returns true if the room name changed, false otherwise.
|
|
|
|
pub fn handle_room_name(&mut self, event: &NameEvent) -> bool {
|
|
|
|
match event.content.name() {
|
2020-04-23 13:55:59 +00:00
|
|
|
Some(name) => self.set_room_name(name),
|
2020-03-28 10:58:30 +00:00
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-07 00:46:06 +00:00
|
|
|
/// Handle a room.name event, updating the room state if necessary.
|
|
|
|
///
|
|
|
|
/// Returns true if the room name changed, false otherwise.
|
|
|
|
pub fn handle_stripped_room_name(&mut self, event: &StrippedRoomName) -> bool {
|
|
|
|
match event.content.name() {
|
|
|
|
Some(name) => self.set_room_name(name),
|
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-28 19:05:37 +00:00
|
|
|
/// Handle a room.power_levels event, updating the room state if necessary.
|
|
|
|
///
|
|
|
|
/// Returns true if the room name changed, false otherwise.
|
|
|
|
pub fn handle_power_level(&mut self, event: &PowerLevelsEvent) -> bool {
|
2020-04-05 16:41:00 +00:00
|
|
|
// NOTE: this is always true, we assume that if we get an event their is an update.
|
|
|
|
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);
|
2020-03-28 19:05:37 +00:00
|
|
|
}
|
2020-04-05 16:41:00 +00:00
|
|
|
|
|
|
|
for user in event.content.users.keys() {
|
2020-06-10 16:12:27 +00:00
|
|
|
if let Some(member) = self.joined_members.get_mut(user) {
|
2020-07-01 13:44:44 +00:00
|
|
|
if Room::update_member_power(member, event, max_power) {
|
2020-04-05 16:41:00 +00:00
|
|
|
updated = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-07-01 13:44:44 +00:00
|
|
|
|
2020-04-05 16:41:00 +00:00
|
|
|
updated
|
2020-03-28 19:05:37 +00:00
|
|
|
}
|
|
|
|
|
2020-04-23 10:45:00 +00:00
|
|
|
fn handle_tombstone(&mut self, event: &TombstoneEvent) -> bool {
|
|
|
|
self.tombstone = Some(Tombstone {
|
|
|
|
body: event.content.body.clone(),
|
|
|
|
replacement: event.content.replacement_room.clone(),
|
|
|
|
});
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
2020-05-14 11:58:09 +00:00
|
|
|
fn handle_encryption_event(&mut self, event: &EncryptionEvent) -> bool {
|
|
|
|
self.encrypted = Some(event.into());
|
2020-04-01 14:06:28 +00:00
|
|
|
true
|
|
|
|
}
|
|
|
|
|
2020-03-28 10:58:30 +00:00
|
|
|
/// Receive a timeline event for this room and update the room state.
|
|
|
|
///
|
|
|
|
/// Returns true if the joined member list changed, false otherwise.
|
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
|
|
|
/// * `event` - The event of the room.
|
|
|
|
pub fn receive_timeline_event(&mut self, event: &RoomEvent) -> bool {
|
|
|
|
match event {
|
|
|
|
// update to the current members of the room
|
2020-04-29 11:58:31 +00:00
|
|
|
RoomEvent::RoomMember(member) => self.handle_membership(member),
|
2020-04-05 12:23:59 +00:00
|
|
|
// finds all events related to the name of the room for later use
|
2020-04-29 11:58:31 +00:00
|
|
|
RoomEvent::RoomName(name) => self.handle_room_name(name),
|
|
|
|
RoomEvent::RoomCanonicalAlias(c_alias) => self.handle_canonical(c_alias),
|
|
|
|
RoomEvent::RoomAliases(alias) => self.handle_room_aliases(alias),
|
2020-03-28 10:58:30 +00:00
|
|
|
// power levels of the room members
|
2020-04-29 11:58:31 +00:00
|
|
|
RoomEvent::RoomPowerLevels(power) => self.handle_power_level(power),
|
|
|
|
RoomEvent::RoomTombstone(tomb) => self.handle_tombstone(tomb),
|
|
|
|
RoomEvent::RoomEncryption(encrypt) => self.handle_encryption_event(encrypt),
|
2020-05-07 01:17:21 +00:00
|
|
|
#[cfg(feature = "messages")]
|
2020-04-29 11:58:31 +00:00
|
|
|
RoomEvent::RoomMessage(msg) => self.handle_message(msg),
|
2020-03-28 10:58:30 +00:00
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Receive a state event for this room and update the room state.
|
|
|
|
///
|
2020-04-07 12:55:42 +00:00
|
|
|
/// Returns true if the state of the `Room` has changed, false otherwise.
|
2020-03-28 10:58:30 +00:00
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
|
|
|
/// * `event` - The event of the room.
|
|
|
|
pub fn receive_state_event(&mut self, event: &StateEvent) -> bool {
|
|
|
|
match event {
|
2020-04-29 11:58:31 +00:00
|
|
|
// update to the current members of the room
|
|
|
|
StateEvent::RoomMember(member) => self.handle_membership(member),
|
|
|
|
// finds all events related to the name of the room for later use
|
|
|
|
StateEvent::RoomName(name) => self.handle_room_name(name),
|
|
|
|
StateEvent::RoomCanonicalAlias(c_alias) => self.handle_canonical(c_alias),
|
|
|
|
StateEvent::RoomAliases(alias) => self.handle_room_aliases(alias),
|
|
|
|
// power levels of the room members
|
|
|
|
StateEvent::RoomPowerLevels(power) => self.handle_power_level(power),
|
|
|
|
StateEvent::RoomTombstone(tomb) => self.handle_tombstone(tomb),
|
|
|
|
StateEvent::RoomEncryption(encrypt) => self.handle_encryption_event(encrypt),
|
2020-03-28 10:58:30 +00:00
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-05 20:13:14 +00:00
|
|
|
/// Receive a stripped state event for this room and update the room state.
|
|
|
|
///
|
|
|
|
/// Returns true if the state of the `Room` has changed, false otherwise.
|
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
|
|
|
/// * `event` - The `AnyStrippedStateEvent` sent by the server for invited but not
|
|
|
|
/// joined rooms.
|
2020-05-07 00:46:06 +00:00
|
|
|
pub fn receive_stripped_state_event(&mut self, event: &AnyStrippedStateEvent) -> bool {
|
|
|
|
match event {
|
|
|
|
AnyStrippedStateEvent::RoomName(n) => self.handle_stripped_room_name(n),
|
|
|
|
_ => false,
|
|
|
|
}
|
2020-05-05 20:13:14 +00:00
|
|
|
}
|
|
|
|
|
2020-07-01 13:44:44 +00:00
|
|
|
/// Receive a presence event for a member of the current room.
|
2020-03-28 10:58:30 +00:00
|
|
|
///
|
2020-07-01 13:44:44 +00:00
|
|
|
/// Returns true if the event causes a change to the member's presence, false otherwise.
|
2020-03-28 10:58:30 +00:00
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
2020-07-01 12:02:37 +00:00
|
|
|
/// * `event` - The presence event to receive and process.
|
2020-03-28 10:58:30 +00:00
|
|
|
pub fn receive_presence_event(&mut self, event: &PresenceEvent) -> bool {
|
2020-07-01 13:44:44 +00:00
|
|
|
let PresenceEvent {
|
|
|
|
content:
|
|
|
|
PresenceEventContent {
|
|
|
|
avatar_url,
|
|
|
|
currently_active,
|
|
|
|
displayname,
|
|
|
|
last_active_ago,
|
|
|
|
presence,
|
|
|
|
status_msg,
|
|
|
|
},
|
|
|
|
..
|
|
|
|
} = event;
|
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
if let Some(member) = self.joined_members.get_mut(&event.sender) {
|
2020-07-01 13:44:44 +00:00
|
|
|
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 {
|
|
|
|
|
|
|
|
// Everything is the same, nothing to do.
|
2020-03-28 10:58:30 +00:00
|
|
|
false
|
|
|
|
} else {
|
2020-07-01 13:44:44 +00:00
|
|
|
// Something changed, do the update.
|
|
|
|
|
|
|
|
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);
|
|
|
|
member.status_msg = status_msg.clone();
|
|
|
|
|
2020-03-28 10:58:30 +00:00
|
|
|
true
|
|
|
|
}
|
|
|
|
} else {
|
2020-07-01 13:44:44 +00:00
|
|
|
// This is probably an error as we have a `PresenceEvent` for a user
|
|
|
|
// we don't know about.
|
2020-03-28 10:58:30 +00:00
|
|
|
false
|
|
|
|
}
|
2020-07-01 13:44:44 +00:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Process an update of a member's profile.
|
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
|
|
|
/// * `event` - The profile update event for a specified room member.
|
|
|
|
// TODO: NEXT: Add disambiguation handling here
|
|
|
|
pub(crate) fn update_member_profile(&mut self, event: &MemberEvent) -> bool {
|
|
|
|
let user_id = if let Ok(id) = UserId::try_from(event.state_key.as_str()) {
|
|
|
|
id
|
|
|
|
} else {
|
|
|
|
return false;
|
|
|
|
};
|
|
|
|
|
|
|
|
if let Some(member) = self.joined_members.get_mut(&user_id) {
|
|
|
|
member.display_name = event.content.displayname.clone();
|
|
|
|
member.avatar_url = event.content.avatar_url.clone();
|
|
|
|
true
|
|
|
|
} else if let Some(member) = self.invited_members.get_mut(&user_id) {
|
|
|
|
member.display_name = event.content.displayname.clone();
|
|
|
|
member.avatar_url = event.content.avatar_url.clone();
|
|
|
|
true
|
|
|
|
} else {
|
|
|
|
false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Process an update of a member's power level.
|
|
|
|
///
|
|
|
|
/// # Arguments
|
|
|
|
///
|
|
|
|
/// * `event` - The power level event to process.
|
|
|
|
/// * `max_power` - Maximum power level allowed.
|
|
|
|
pub fn update_member_power(member: &mut RoomMember, event: &PowerLevelsEvent, 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::from(0) {
|
|
|
|
member.power_level_norm = Some((member.power_level.unwrap() * Int::from(100)) / max_power);
|
|
|
|
}
|
|
|
|
|
|
|
|
changed
|
2020-03-28 10:58:30 +00:00
|
|
|
}
|
|
|
|
}
|
2020-03-31 13:01:48 +00:00
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
2020-04-03 13:07:40 +00:00
|
|
|
use super::*;
|
2020-06-10 16:12:27 +00:00
|
|
|
use crate::events::{room::encryption::EncryptionEventContent, UnsignedData};
|
2020-05-14 11:58:09 +00:00
|
|
|
use crate::identifiers::{EventId, UserId};
|
2020-05-11 08:43:21 +00:00
|
|
|
use crate::{BaseClient, Session};
|
2020-06-21 18:01:04 +00:00
|
|
|
use matrix_sdk_test::{async_test, sync_response, EventBuilder, EventsJson, SyncResponseFile};
|
2020-03-31 13:01:48 +00:00
|
|
|
|
2020-05-14 11:58:09 +00:00
|
|
|
use std::time::SystemTime;
|
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
|
|
use wasm_bindgen_test::*;
|
2020-03-31 13:01:48 +00:00
|
|
|
|
|
|
|
use std::convert::TryFrom;
|
2020-04-05 16:41:00 +00:00
|
|
|
use std::ops::Deref;
|
2020-05-07 10:51:53 +00:00
|
|
|
|
2020-05-22 13:23:58 +00:00
|
|
|
async fn get_client() -> BaseClient {
|
2020-03-31 13:01:48 +00:00
|
|
|
let session = Session {
|
|
|
|
access_token: "1234".to_owned(),
|
2020-03-31 19:18:27 +00:00
|
|
|
user_id: UserId::try_from("@example:localhost").unwrap(),
|
2020-03-31 13:01:48 +00:00
|
|
|
device_id: "DEVICEID".to_owned(),
|
|
|
|
};
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = BaseClient::new().unwrap();
|
|
|
|
client.restore_login(session).await.unwrap();
|
|
|
|
client
|
2020-05-07 14:22:18 +00:00
|
|
|
}
|
2020-03-31 13:01:48 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
fn get_room_id() -> RoomId {
|
|
|
|
RoomId::try_from("!SVkFJHzfwvuaIEawgC:localhost").unwrap()
|
|
|
|
}
|
2020-03-31 13:01:48 +00:00
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[async_test]
|
2020-05-07 14:22:18 +00:00
|
|
|
async fn user_presence() {
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = get_client().await;
|
2020-05-07 14:22:18 +00:00
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
let mut response = sync_response(SyncResponseFile::Default);
|
2020-03-31 13:01:48 +00:00
|
|
|
|
2020-05-07 10:51:53 +00:00
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
2020-03-31 13:01:48 +00:00
|
|
|
|
2020-05-07 10:51:53 +00:00
|
|
|
let rooms_lock = &client.joined_rooms();
|
2020-05-06 11:57:58 +00:00
|
|
|
let rooms = rooms_lock.read().await;
|
2020-03-31 13:01:48 +00:00
|
|
|
let room = &rooms
|
2020-04-03 13:07:40 +00:00
|
|
|
.get(&RoomId::try_from("!SVkFJHzfwvuaIEawgC:localhost").unwrap())
|
2020-03-31 13:01:48 +00:00
|
|
|
.unwrap()
|
2020-04-14 22:10:10 +00:00
|
|
|
.read()
|
2020-04-01 01:08:25 +00:00
|
|
|
.await;
|
2020-03-31 13:01:48 +00:00
|
|
|
|
2020-06-15 15:04:10 +00:00
|
|
|
assert_eq!(1, room.joined_members.len());
|
2020-04-05 16:41:00 +00:00
|
|
|
assert!(room.deref().power_levels.is_some())
|
2020-03-31 13:01:48 +00:00
|
|
|
}
|
2020-04-07 00:59:44 +00:00
|
|
|
|
2020-07-01 11:51:21 +00:00
|
|
|
#[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 = UserId::try_from("@example:localhost").unwrap();
|
|
|
|
let user_id2 = UserId::try_from("@example2:localhost").unwrap();
|
|
|
|
|
|
|
|
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, RoomEvent::RoomMember)
|
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
let mut member2_invite_sync_response = event_builder
|
|
|
|
.add_custom_joined_event(&room_id, member2_invite_event, RoomEvent::RoomMember)
|
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
let mut member2_join_sync_response = event_builder
|
|
|
|
.add_custom_joined_event(&room_id, member2_join_event, RoomEvent::RoomMember)
|
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
|
|
|
|
// Test that `user` is either joined or invited to `room` but not both.
|
|
|
|
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()
|
|
|
|
);
|
|
|
|
};
|
|
|
|
|
|
|
|
// First member joins.
|
|
|
|
client
|
|
|
|
.receive_sync_response(&mut member1_join_sync_response)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
// The first member is not *both* invited and joined but it *is* one of those.
|
|
|
|
invited_or_joined_but_not_both(&client, &room_id, &user_id1).await;
|
|
|
|
|
|
|
|
// First member invites second member.
|
|
|
|
client
|
|
|
|
.receive_sync_response(&mut member2_invite_sync_response)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
// Neither member is *both* invited and joined, but they are both *at least one* of those.
|
|
|
|
invited_or_joined_but_not_both(&client, &room_id, &user_id1).await;
|
|
|
|
invited_or_joined_but_not_both(&client, &room_id, &user_id2).await;
|
|
|
|
|
|
|
|
// Second member joins.
|
|
|
|
client
|
|
|
|
.receive_sync_response(&mut member2_join_sync_response)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
// Repeat the previous test.
|
|
|
|
invited_or_joined_but_not_both(&client, &room_id, &user_id1).await;
|
|
|
|
invited_or_joined_but_not_both(&client, &room_id, &user_id2).await;
|
|
|
|
}
|
|
|
|
|
2020-05-14 15:09:30 +00:00
|
|
|
#[async_test]
|
|
|
|
async fn test_member_display_name() {
|
2020-06-09 13:17:21 +00:00
|
|
|
// Initialize
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
let client = get_client().await;
|
2020-05-14 15:09:30 +00:00
|
|
|
let room_id = get_room_id();
|
2020-06-09 13:17:21 +00:00
|
|
|
let user_id1 = UserId::try_from("@example:localhost").unwrap();
|
2020-06-15 15:11:12 +00:00
|
|
|
let user_id2 = UserId::try_from("@example2:localhost").unwrap();
|
|
|
|
let user_id3 = UserId::try_from("@example3:localhost").unwrap();
|
2020-06-09 13:17:21 +00:00
|
|
|
|
|
|
|
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,
|
2020-06-15 15:11:12 +00:00
|
|
|
"sender": format!("{}", user_id2),
|
|
|
|
"state_key": format!("{}", user_id2),
|
2020-06-09 13:17:21 +00:00
|
|
|
"type": "m.room.member",
|
2020-06-15 15:11:12 +00:00
|
|
|
"prev_content": {
|
|
|
|
"avatar_url": null,
|
|
|
|
"displayname": "example",
|
|
|
|
"membership": "invite"
|
|
|
|
},
|
2020-06-09 13:17:21 +00:00
|
|
|
"unsigned": {
|
|
|
|
"age": 1989321234,
|
2020-06-15 15:11:12 +00:00
|
|
|
"replaces_state": "$1622a2311315tkjoA:localhost"
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
2020-07-01 11:53:10 +00:00
|
|
|
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"
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
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"
|
2020-06-09 13:17:21 +00:00
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
let mut event_builder = EventBuilder::new();
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
let mut member1_join_sync_response = event_builder
|
2020-06-24 08:42:58 +00:00
|
|
|
.add_room_event(EventsJson::Member, RoomEvent::RoomMember)
|
2020-05-14 15:09:30 +00:00
|
|
|
.build_sync_response();
|
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
let mut member2_join_sync_response = event_builder
|
2020-07-01 11:53:10 +00:00
|
|
|
.add_custom_joined_event(&room_id, member2_join_event.clone(), RoomEvent::RoomMember)
|
2020-06-09 13:17:21 +00:00
|
|
|
.build_sync_response();
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
let mut member3_join_sync_response = event_builder
|
|
|
|
.add_custom_joined_event(&room_id, member3_join_event, RoomEvent::RoomMember)
|
|
|
|
.build_sync_response();
|
|
|
|
|
2020-07-01 11:53:10 +00:00
|
|
|
let mut member2_and_member3_leave_sync_response = event_builder
|
2020-06-15 15:11:12 +00:00
|
|
|
.add_custom_joined_event(&room_id, member2_leave_event, RoomEvent::RoomMember)
|
2020-07-01 11:53:10 +00:00
|
|
|
.add_custom_joined_event(&room_id, member3_leave_event, RoomEvent::RoomMember)
|
2020-06-15 15:11:12 +00:00
|
|
|
.build_sync_response();
|
|
|
|
|
2020-07-01 11:53:10 +00:00
|
|
|
let mut member2_rejoins_when_invited_sync_response = event_builder
|
|
|
|
.add_custom_joined_event(&room_id, member1_invites_member2_event, RoomEvent::RoomMember)
|
|
|
|
.add_custom_joined_event(&room_id, member2_join_event, RoomEvent::RoomMember)
|
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
let mut member1_name_change_sync_response = event_builder
|
|
|
|
.add_room_event(EventsJson::MemberNameChange, RoomEvent::RoomMember)
|
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
let mut member2_name_change_sync_response = event_builder
|
|
|
|
.add_custom_joined_event(&room_id, member2_name_change_event, RoomEvent::RoomMember)
|
2020-06-15 15:11:12 +00:00
|
|
|
.build_sync_response();
|
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
// First member with display name "example" joins
|
|
|
|
client
|
|
|
|
.receive_sync_response(&mut member1_join_sync_response)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
// First member's disambiguated display name is "example"
|
|
|
|
{
|
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
2020-06-09 14:11:41 +00:00
|
|
|
let display_name1 = room.member_display_name(&user_id1);
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
assert_eq!("example", display_name1);
|
2020-03-31 13:01:48 +00:00
|
|
|
}
|
2020-04-05 16:41:00 +00:00
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
// Second and third member with display name "example" join
|
2020-06-09 13:17:21 +00:00
|
|
|
client
|
|
|
|
.receive_sync_response(&mut member2_join_sync_response)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
2020-06-15 15:11:12 +00:00
|
|
|
client
|
|
|
|
.receive_sync_response(&mut member3_join_sync_response)
|
|
|
|
.await
|
|
|
|
.unwrap();
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-15 15:11:12 +00:00
|
|
|
// All of their display names are now disambiguated with MXIDs
|
2020-06-09 13:17:21 +00:00
|
|
|
{
|
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
2020-06-09 14:11:41 +00:00
|
|
|
let display_name1 = room.member_display_name(&user_id1);
|
|
|
|
let display_name2 = room.member_display_name(&user_id2);
|
2020-06-15 15:11:12 +00:00
|
|
|
let display_name3 = room.member_display_name(&user_id3);
|
2020-05-14 15:09:30 +00:00
|
|
|
|
2020-06-09 13:17:21 +00:00
|
|
|
assert_eq!(format!("example ({})", user_id1), display_name1);
|
|
|
|
assert_eq!(format!("example ({})", user_id2), display_name2);
|
2020-06-15 15:11:12 +00:00
|
|
|
assert_eq!(format!("example ({})", user_id3), display_name3);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Second and third member leave. The first's display name is now just "example" again.
|
|
|
|
client
|
2020-07-01 11:53:10 +00:00
|
|
|
.receive_sync_response(&mut member2_and_member3_leave_sync_response)
|
2020-06-15 15:11:12 +00:00
|
|
|
.await
|
|
|
|
.unwrap();
|
2020-07-01 11:53:10 +00:00
|
|
|
|
|
|
|
{
|
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
|
|
|
|
|
|
|
let display_name1 = room.member_display_name(&user_id1);
|
|
|
|
|
|
|
|
assert_eq!("example", display_name1);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Second member rejoins after being invited by first member. Both of their names are
|
|
|
|
// disambiguated.
|
2020-06-15 15:11:12 +00:00
|
|
|
client
|
2020-07-01 11:53:10 +00:00
|
|
|
.receive_sync_response(&mut member2_rejoins_when_invited_sync_response)
|
2020-06-15 15:11:12 +00:00
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
{
|
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
|
|
|
|
|
|
|
let display_name1 = room.member_display_name(&user_id1);
|
2020-07-01 11:53:10 +00:00
|
|
|
let display_name2 = room.member_display_name(&user_id2);
|
2020-06-15 15:11:12 +00:00
|
|
|
|
2020-07-01 11:53:10 +00:00
|
|
|
assert_eq!(format!("example ({})", user_id1), display_name1);
|
|
|
|
assert_eq!(format!("example ({})", user_id2), display_name2);
|
|
|
|
}
|
|
|
|
|
|
|
|
// First member changes his display name to "changed". None of the display names are
|
|
|
|
// disambiguated.
|
|
|
|
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.member_display_name(&user_id1);
|
|
|
|
let display_name2 = room.member_display_name(&user_id2);
|
|
|
|
|
|
|
|
assert_eq!("changed", display_name1);
|
|
|
|
assert_eq!("example", display_name2);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Second member *also* changes his display name to "changed". Again, both display name are
|
|
|
|
// disambiguated.
|
|
|
|
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.member_display_name(&user_id1);
|
|
|
|
let display_name2 = room.member_display_name(&user_id2);
|
|
|
|
|
|
|
|
assert_eq!(format!("changed ({})", user_id1), display_name1);
|
|
|
|
assert_eq!(format!("changed ({})", user_id2), display_name2);
|
2020-06-09 13:17:21 +00:00
|
|
|
}
|
2020-03-31 13:01:48 +00:00
|
|
|
}
|
2020-04-07 00:59:44 +00:00
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[async_test]
|
2020-05-07 14:22:18 +00:00
|
|
|
async fn room_events() {
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = get_client().await;
|
2020-05-07 14:22:18 +00:00
|
|
|
let room_id = get_room_id();
|
|
|
|
let user_id = UserId::try_from("@example:localhost").unwrap();
|
2020-04-07 20:11:35 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
let mut response = EventBuilder::default()
|
2020-06-21 18:01:04 +00:00
|
|
|
.add_room_event(EventsJson::Member, RoomEvent::RoomMember)
|
|
|
|
.add_room_event(EventsJson::PowerLevels, RoomEvent::RoomPowerLevels)
|
2020-05-07 14:22:18 +00:00
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
2020-04-07 12:55:42 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
2020-04-07 20:11:35 +00:00
|
|
|
|
2020-06-10 16:12:27 +00:00
|
|
|
assert_eq!(room.joined_members.len(), 1);
|
2020-04-07 20:11:35 +00:00
|
|
|
assert!(room.power_levels.is_some());
|
|
|
|
assert_eq!(
|
|
|
|
room.power_levels.as_ref().unwrap().kick,
|
2020-04-29 08:40:27 +00:00
|
|
|
crate::js_int::Int::new(50).unwrap()
|
2020-04-07 20:11:35 +00:00
|
|
|
);
|
2020-06-10 16:12:27 +00:00
|
|
|
let admin = room.joined_members.get(&user_id).unwrap();
|
2020-04-29 08:40:27 +00:00
|
|
|
assert_eq!(
|
|
|
|
admin.power_level.unwrap(),
|
|
|
|
crate::js_int::Int::new(100).unwrap()
|
|
|
|
);
|
2020-04-07 00:59:44 +00:00
|
|
|
}
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[async_test]
|
2020-05-07 14:22:18 +00:00
|
|
|
async fn calculate_aliases() {
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = get_client().await;
|
2020-05-07 14:22:18 +00:00
|
|
|
|
|
|
|
let room_id = get_room_id();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
let mut response = EventBuilder::default()
|
2020-06-21 18:01:04 +00:00
|
|
|
.add_state_event(EventsJson::Aliases, StateEvent::RoomAliases)
|
2020-05-07 14:22:18 +00:00
|
|
|
.build_sync_response();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
|
|
|
|
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 10:51:53 +00:00
|
|
|
assert_eq!("tutorial", room.display_name());
|
2020-04-23 13:55:59 +00:00
|
|
|
}
|
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[async_test]
|
2020-05-07 14:22:18 +00:00
|
|
|
async fn calculate_alias() {
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = get_client().await;
|
2020-05-07 14:22:18 +00:00
|
|
|
|
|
|
|
let room_id = get_room_id();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
let mut response = EventBuilder::default()
|
2020-06-21 18:01:04 +00:00
|
|
|
.add_state_event(EventsJson::Alias, StateEvent::RoomCanonicalAlias)
|
2020-05-07 14:22:18 +00:00
|
|
|
.build_sync_response();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
|
|
|
|
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 10:51:53 +00:00
|
|
|
assert_eq!("tutorial", room.display_name());
|
2020-04-23 13:55:59 +00:00
|
|
|
}
|
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[async_test]
|
2020-05-07 14:22:18 +00:00
|
|
|
async fn calculate_name() {
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = get_client().await;
|
2020-05-07 14:22:18 +00:00
|
|
|
|
|
|
|
let room_id = get_room_id();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
let mut response = EventBuilder::default()
|
2020-06-21 18:01:04 +00:00
|
|
|
.add_state_event(EventsJson::Name, StateEvent::RoomName)
|
2020-05-07 14:22:18 +00:00
|
|
|
.build_sync_response();
|
|
|
|
|
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 14:22:18 +00:00
|
|
|
let room = client.get_joined_room(&room_id).await.unwrap();
|
|
|
|
let room = room.read().await;
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 10:51:53 +00:00
|
|
|
assert_eq!("room name", room.display_name());
|
2020-04-23 13:55:59 +00:00
|
|
|
}
|
|
|
|
|
2020-05-08 14:12:21 +00:00
|
|
|
#[async_test]
|
2020-04-23 13:55:59 +00:00
|
|
|
async fn calculate_room_names_from_summary() {
|
2020-05-08 14:12:21 +00:00
|
|
|
let mut response = sync_response(SyncResponseFile::DefaultWithSummary);
|
2020-04-23 13:55:59 +00:00
|
|
|
|
|
|
|
let session = Session {
|
|
|
|
access_token: "1234".to_owned(),
|
|
|
|
user_id: UserId::try_from("@example:localhost").unwrap(),
|
|
|
|
device_id: "DEVICEID".to_owned(),
|
|
|
|
};
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = BaseClient::new().unwrap();
|
|
|
|
client.restore_login(session).await.unwrap();
|
2020-05-07 10:51:53 +00:00
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-05-07 10:51:53 +00:00
|
|
|
let mut room_names = vec![];
|
|
|
|
for room in client.joined_rooms().read().await.values() {
|
|
|
|
room_names.push(room.read().await.display_name())
|
2020-05-07 11:09:22 +00:00
|
|
|
}
|
2020-04-23 13:55:59 +00:00
|
|
|
|
2020-06-10 20:45:08 +00:00
|
|
|
assert_eq!(vec!["example2"], room_names);
|
2020-04-23 13:55:59 +00:00
|
|
|
}
|
2020-05-14 11:58:09 +00:00
|
|
|
|
|
|
|
#[async_test]
|
2020-05-14 13:12:14 +00:00
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
2020-05-14 11:58:09 +00:00
|
|
|
async fn encryption_info_test() {
|
|
|
|
let mut response = sync_response(SyncResponseFile::DefaultWithSummary);
|
|
|
|
let user_id = UserId::try_from("@example:localhost").unwrap();
|
|
|
|
|
|
|
|
let session = Session {
|
|
|
|
access_token: "1234".to_owned(),
|
|
|
|
user_id: user_id.clone(),
|
|
|
|
device_id: "DEVICEID".to_owned(),
|
|
|
|
};
|
2020-05-22 13:23:58 +00:00
|
|
|
let client = BaseClient::new().unwrap();
|
|
|
|
client.restore_login(session).await.unwrap();
|
2020-05-14 11:58:09 +00:00
|
|
|
client.receive_sync_response(&mut response).await.unwrap();
|
|
|
|
|
|
|
|
let event = EncryptionEvent {
|
2020-05-14 12:50:20 +00:00
|
|
|
event_id: EventId::try_from("$h29iv0s8:example.com").unwrap(),
|
2020-05-14 11:58:09 +00:00
|
|
|
origin_server_ts: SystemTime::now(),
|
|
|
|
sender: user_id,
|
|
|
|
state_key: "".into(),
|
|
|
|
unsigned: UnsignedData::default(),
|
|
|
|
content: EncryptionEventContent {
|
|
|
|
algorithm: Algorithm::MegolmV1AesSha2,
|
|
|
|
rotation_period_ms: Some(100_000u32.into()),
|
|
|
|
rotation_period_msgs: Some(100u32.into()),
|
|
|
|
},
|
|
|
|
prev_content: None,
|
|
|
|
room_id: None,
|
|
|
|
};
|
|
|
|
|
|
|
|
let room_id = get_room_id();
|
|
|
|
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(), &Algorithm::MegolmV1AesSha2);
|
|
|
|
assert_eq!(encryption_info.rotation_period(), 100_000);
|
|
|
|
assert_eq!(encryption_info.rotation_period_messages(), 100);
|
|
|
|
}
|
2020-03-31 13:01:48 +00:00
|
|
|
}
|