base: Split out the new room and member structs from the state store
parent
d4ebe8cc83
commit
a08f857e49
|
@ -58,8 +58,9 @@ use crate::{
|
||||||
AccountData, Ephemeral, JoinedRoom, LeftRoom, Presence, Rooms, State, SyncResponse,
|
AccountData, Ephemeral, JoinedRoom, LeftRoom, Presence, Rooms, State, SyncResponse,
|
||||||
Timeline,
|
Timeline,
|
||||||
},
|
},
|
||||||
|
rooms::{InnerSummary, Room, RoomType},
|
||||||
session::Session,
|
session::Session,
|
||||||
store::{InnerSummary, Room, RoomType, StateChanges, Store},
|
store::{StateChanges, Store},
|
||||||
};
|
};
|
||||||
|
|
||||||
pub type Token = String;
|
pub type Token = String;
|
||||||
|
|
|
@ -45,10 +45,12 @@ pub use matrix_sdk_common::*;
|
||||||
mod client;
|
mod client;
|
||||||
mod error;
|
mod error;
|
||||||
pub mod responses;
|
pub mod responses;
|
||||||
|
mod rooms;
|
||||||
mod session;
|
mod session;
|
||||||
mod store;
|
mod store;
|
||||||
|
|
||||||
pub use store::{Room, RoomMember, Store};
|
pub use rooms::{InnerSummary, Room, RoomMember};
|
||||||
|
pub use store::Store;
|
||||||
|
|
||||||
pub use client::{BaseClient, BaseClientConfig, RoomState, RoomStateType};
|
pub use client::{BaseClient, BaseClientConfig, RoomState, RoomStateType};
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,373 @@
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
use std::{
|
||||||
|
convert::TryFrom,
|
||||||
|
sync::{Arc, Mutex as SyncMutex},
|
||||||
|
};
|
||||||
|
|
||||||
|
use futures::{
|
||||||
|
future,
|
||||||
|
stream::{self, Stream, StreamExt},
|
||||||
|
};
|
||||||
|
use matrix_sdk_common::{
|
||||||
|
api::r0::sync::sync_events::RoomSummary as RumaSummary,
|
||||||
|
events::{room::encryption::EncryptionEventContent, AnySyncStateEvent, EventType},
|
||||||
|
identifiers::{RoomAliasId, RoomId, UserId},
|
||||||
|
};
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
use tracing::info;
|
||||||
|
|
||||||
|
use crate::{responses::UnreadNotificationsCount, store::Store};
|
||||||
|
|
||||||
|
use super::RoomMember;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct Room {
|
||||||
|
room_id: Arc<RoomId>,
|
||||||
|
own_user_id: Arc<UserId>,
|
||||||
|
inner: Arc<SyncMutex<InnerSummary>>,
|
||||||
|
store: Store,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug, Default, Serialize, Deserialize)]
|
||||||
|
pub struct SomeSummary {
|
||||||
|
heroes: Vec<String>,
|
||||||
|
joined_member_count: u64,
|
||||||
|
invited_member_count: u64,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Signals to the `BaseClient` which `RoomState` to send to `EventEmitter`.
|
||||||
|
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
|
||||||
|
pub enum RoomType {
|
||||||
|
/// Represents a joined room, the `joined_rooms` HashMap will be used.
|
||||||
|
Joined,
|
||||||
|
/// Represents a left room, the `left_rooms` HashMap will be used.
|
||||||
|
Left,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Room {
|
||||||
|
pub fn new(own_user_id: &UserId, store: Store, room_id: &RoomId, room_type: RoomType) -> Self {
|
||||||
|
let room_id = Arc::new(room_id.clone());
|
||||||
|
|
||||||
|
Self {
|
||||||
|
own_user_id: Arc::new(own_user_id.clone()),
|
||||||
|
room_id: room_id.clone(),
|
||||||
|
store,
|
||||||
|
inner: Arc::new(SyncMutex::new(InnerSummary {
|
||||||
|
room_id,
|
||||||
|
room_type,
|
||||||
|
encryption: None,
|
||||||
|
summary: Default::default(),
|
||||||
|
last_prev_batch: None,
|
||||||
|
members_synced: false,
|
||||||
|
name: None,
|
||||||
|
canonical_alias: None,
|
||||||
|
avatar_url: None,
|
||||||
|
topic: None,
|
||||||
|
notification_counts: Default::default(),
|
||||||
|
})),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn are_members_synced(&self) -> bool {
|
||||||
|
self.inner.lock().unwrap().members_synced
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_j_members(&self) -> impl Stream<Item = RoomMember> + '_ {
|
||||||
|
let joined = self.store.get_joined_user_ids(self.room_id()).await;
|
||||||
|
let invited = self.store.get_invited_user_ids(self.room_id()).await;
|
||||||
|
|
||||||
|
let x = move |u| async move {
|
||||||
|
let presence = self.store.get_presence_event(&u).await;
|
||||||
|
let power = self
|
||||||
|
.store
|
||||||
|
.get_state_event(self.room_id(), EventType::RoomPowerLevels, "")
|
||||||
|
.await
|
||||||
|
.map(|e| {
|
||||||
|
if let AnySyncStateEvent::RoomPowerLevels(e) = e {
|
||||||
|
Some(e)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.flatten();
|
||||||
|
|
||||||
|
self.store
|
||||||
|
.get_member_event(self.room_id(), &u)
|
||||||
|
.await
|
||||||
|
.map(|m| RoomMember {
|
||||||
|
user_id: UserId::try_from(m.state_key.as_str()).unwrap().into(),
|
||||||
|
event: m.into(),
|
||||||
|
presence: presence.into(),
|
||||||
|
power_levles: power.into(),
|
||||||
|
})
|
||||||
|
};
|
||||||
|
|
||||||
|
joined.chain(invited).filter_map(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Calculate the canonical display name of the 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>
|
||||||
|
pub async fn calculate_name(&self) -> String {
|
||||||
|
let inner = self.inner.lock().unwrap();
|
||||||
|
|
||||||
|
if let Some(name) = &inner.name {
|
||||||
|
let name = name.trim();
|
||||||
|
name.to_string()
|
||||||
|
} else if let Some(alias) = &inner.canonical_alias {
|
||||||
|
let alias = alias.alias().trim();
|
||||||
|
alias.to_string()
|
||||||
|
} else {
|
||||||
|
let joined = inner.summary.joined_member_count;
|
||||||
|
let invited = inner.summary.invited_member_count;
|
||||||
|
let heroes_count = inner.summary.heroes.len() as u64;
|
||||||
|
let invited_joined = (invited + joined).saturating_sub(1);
|
||||||
|
|
||||||
|
let members = self.get_j_members().await;
|
||||||
|
|
||||||
|
// info!(
|
||||||
|
// "Calculating name for {}, hero count {} members {:#?}",
|
||||||
|
// self.room_id(),
|
||||||
|
// heroes_count,
|
||||||
|
// members
|
||||||
|
// );
|
||||||
|
// TODO: This should use `self.heroes` but it is always empty??
|
||||||
|
//
|
||||||
|
let own_user_id = self.own_user_id.clone();
|
||||||
|
|
||||||
|
let is_own_member = |m: &RoomMember| m.user_id() == &*own_user_id;
|
||||||
|
|
||||||
|
if !inner.summary.heroes.is_empty() {
|
||||||
|
let mut names = stream::iter(inner.summary.heroes.iter())
|
||||||
|
.take(3)
|
||||||
|
.filter_map(|u| async move {
|
||||||
|
let user_id = UserId::try_from(u.as_str()).ok()?;
|
||||||
|
self.get_member(&user_id).await
|
||||||
|
})
|
||||||
|
.map(|mem| {
|
||||||
|
mem.display_name()
|
||||||
|
.map(|d| d.to_string())
|
||||||
|
.unwrap_or_else(|| mem.user_id().localpart().to_string())
|
||||||
|
})
|
||||||
|
.collect::<Vec<String>>()
|
||||||
|
.await;
|
||||||
|
names.sort();
|
||||||
|
names.join(", ")
|
||||||
|
} else if heroes_count >= invited_joined {
|
||||||
|
let mut names = members
|
||||||
|
.filter(|m| future::ready(is_own_member(m)))
|
||||||
|
.take(3)
|
||||||
|
.map(|mem| {
|
||||||
|
mem.display_name()
|
||||||
|
.map(|d| d.to_string())
|
||||||
|
.unwrap_or_else(|| mem.user_id().localpart().to_string())
|
||||||
|
})
|
||||||
|
.collect::<Vec<String>>()
|
||||||
|
.await;
|
||||||
|
// stabilize ordering
|
||||||
|
names.sort();
|
||||||
|
names.join(", ")
|
||||||
|
} else if heroes_count < invited_joined && invited + joined > 1 {
|
||||||
|
let mut names = members
|
||||||
|
.filter(|m| future::ready(is_own_member(m)))
|
||||||
|
.take(3)
|
||||||
|
.map(|mem| {
|
||||||
|
mem.display_name()
|
||||||
|
.map(|d| d.to_string())
|
||||||
|
.unwrap_or_else(|| mem.user_id().localpart().to_string())
|
||||||
|
})
|
||||||
|
.collect::<Vec<String>>()
|
||||||
|
.await;
|
||||||
|
names.sort();
|
||||||
|
|
||||||
|
// TODO: What length does the spec want us to use here and in
|
||||||
|
// the `else`?
|
||||||
|
format!("{}, and {} others", names.join(", "), (joined + invited))
|
||||||
|
} else {
|
||||||
|
"Empty room".to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn own_user_id(&self) -> &UserId {
|
||||||
|
&self.own_user_id
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn clone_summary(&self) -> InnerSummary {
|
||||||
|
(*self.inner.lock().unwrap()).clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn joined_user_ids(&self) -> impl Stream<Item = UserId> {
|
||||||
|
self.store.get_joined_user_ids(&self.room_id).await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn is_encrypted(&self) -> bool {
|
||||||
|
self.inner.lock().unwrap().encryption.is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn update_summary(&self, summary: InnerSummary) {
|
||||||
|
let mut inner = self.inner.lock().unwrap();
|
||||||
|
*inner = summary;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_member(&self, user_id: &UserId) -> Option<RoomMember> {
|
||||||
|
let presence = self.store.get_presence_event(user_id).await;
|
||||||
|
let power = self
|
||||||
|
.store
|
||||||
|
.get_state_event(self.room_id(), EventType::RoomPowerLevels, "")
|
||||||
|
.await
|
||||||
|
.map(|e| {
|
||||||
|
if let AnySyncStateEvent::RoomPowerLevels(e) = e {
|
||||||
|
Some(e)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.flatten();
|
||||||
|
|
||||||
|
self.store
|
||||||
|
.get_member_event(&self.room_id, user_id)
|
||||||
|
.await
|
||||||
|
.map(|e| RoomMember {
|
||||||
|
user_id: UserId::try_from(e.state_key.as_str()).unwrap().into(),
|
||||||
|
event: e.into(),
|
||||||
|
presence: presence.into(),
|
||||||
|
power_levles: power.into(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn room_id(&self) -> &RoomId {
|
||||||
|
&self.room_id
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn last_prev_batch(&self) -> Option<String> {
|
||||||
|
self.inner.lock().unwrap().last_prev_batch.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn display_name(&self) -> String {
|
||||||
|
self.calculate_name().await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize)]
|
||||||
|
pub struct InnerSummary {
|
||||||
|
pub room_id: Arc<RoomId>,
|
||||||
|
pub room_type: RoomType,
|
||||||
|
|
||||||
|
pub name: Option<String>,
|
||||||
|
pub canonical_alias: Option<RoomAliasId>,
|
||||||
|
pub avatar_url: Option<String>,
|
||||||
|
pub topic: Option<String>,
|
||||||
|
|
||||||
|
pub notification_counts: UnreadNotificationsCount,
|
||||||
|
pub summary: SomeSummary,
|
||||||
|
pub members_synced: bool,
|
||||||
|
|
||||||
|
pub encryption: Option<EncryptionEventContent>,
|
||||||
|
pub last_prev_batch: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl InnerSummary {
|
||||||
|
pub fn mark_as_joined(&mut self) {
|
||||||
|
self.room_type = RoomType::Joined;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mark_as_left(&mut self) {
|
||||||
|
self.room_type = RoomType::Left;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mark_members_synced(&mut self) {
|
||||||
|
self.members_synced = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mark_members_missing(&mut self) {
|
||||||
|
self.members_synced = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_prev_batch(&mut self, prev_batch: Option<&str>) -> bool {
|
||||||
|
if self.last_prev_batch.as_deref() != prev_batch {
|
||||||
|
self.last_prev_batch = prev_batch.map(|p| p.to_string());
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn handle_state_event(&mut self, event: &AnySyncStateEvent) -> bool {
|
||||||
|
match event {
|
||||||
|
AnySyncStateEvent::RoomEncryption(encryption) => {
|
||||||
|
info!("MARKING ROOM {} AS ENCRYPTED", self.room_id);
|
||||||
|
self.encryption = Some(encryption.content.clone());
|
||||||
|
true
|
||||||
|
}
|
||||||
|
AnySyncStateEvent::RoomName(n) => {
|
||||||
|
self.name = n.content.name().map(|n| n.to_string());
|
||||||
|
true
|
||||||
|
}
|
||||||
|
AnySyncStateEvent::RoomCanonicalAlias(a) => {
|
||||||
|
self.canonical_alias = a.content.alias.clone();
|
||||||
|
true
|
||||||
|
}
|
||||||
|
AnySyncStateEvent::RoomTopic(t) => {
|
||||||
|
self.topic = Some(t.content.topic.clone());
|
||||||
|
true
|
||||||
|
}
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn is_encrypted(&self) -> bool {
|
||||||
|
self.encryption.is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn update_notification_count(&mut self, notification_counts: UnreadNotificationsCount) {
|
||||||
|
self.notification_counts = notification_counts;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn update(&mut self, summary: &RumaSummary) -> bool {
|
||||||
|
let mut changed = false;
|
||||||
|
|
||||||
|
info!("UPDAGING SUMMARY FOR {} WITH {:#?}", self.room_id, summary);
|
||||||
|
|
||||||
|
if !summary.is_empty() {
|
||||||
|
if !summary.heroes.is_empty() {
|
||||||
|
self.summary.heroes = summary.heroes.clone();
|
||||||
|
changed = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(joined) = summary.joined_member_count {
|
||||||
|
self.summary.joined_member_count = joined.into();
|
||||||
|
changed = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(invited) = summary.invited_member_count {
|
||||||
|
self.summary.invited_member_count = invited.into();
|
||||||
|
changed = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
changed
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn serialize(&self) -> Vec<u8> {
|
||||||
|
serde_json::to_vec(&self).unwrap()
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,56 @@
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use matrix_sdk_common::{
|
||||||
|
events::{
|
||||||
|
presence::PresenceEvent,
|
||||||
|
room::{member::MemberEventContent, power_levels::PowerLevelsEventContent},
|
||||||
|
SyncStateEvent,
|
||||||
|
},
|
||||||
|
identifiers::UserId,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct RoomMember {
|
||||||
|
pub(crate) user_id: Arc<UserId>,
|
||||||
|
pub(crate) event: Arc<SyncStateEvent<MemberEventContent>>,
|
||||||
|
pub(crate) presence: Arc<Option<PresenceEvent>>,
|
||||||
|
pub(crate) power_levles: Arc<Option<SyncStateEvent<PowerLevelsEventContent>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl RoomMember {
|
||||||
|
pub fn user_id(&self) -> &UserId {
|
||||||
|
&self.user_id
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn display_name(&self) -> Option<&str> {
|
||||||
|
self.event.content.displayname.as_deref()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn power_level(&self) -> i64 {
|
||||||
|
self.power_levles
|
||||||
|
.as_ref()
|
||||||
|
.as_ref()
|
||||||
|
.map(|e| {
|
||||||
|
e.content
|
||||||
|
.users
|
||||||
|
.get(&self.user_id())
|
||||||
|
.map(|p| (*p).into())
|
||||||
|
.unwrap_or(e.content.users_default.into())
|
||||||
|
})
|
||||||
|
.unwrap_or(0)
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,5 @@
|
||||||
|
mod joined;
|
||||||
|
mod members;
|
||||||
|
|
||||||
|
pub use joined::{InnerSummary, Room, RoomType};
|
||||||
|
pub use members::RoomMember;
|
|
@ -1,32 +1,19 @@
|
||||||
use std::{
|
use std::{collections::BTreeMap, convert::TryFrom, path::Path};
|
||||||
collections::BTreeMap,
|
|
||||||
convert::TryFrom,
|
|
||||||
path::Path,
|
|
||||||
sync::{Arc, Mutex as SyncMutex},
|
|
||||||
};
|
|
||||||
|
|
||||||
use futures::{
|
use futures::stream::{self, Stream};
|
||||||
future,
|
|
||||||
stream::{self, Stream, StreamExt},
|
|
||||||
};
|
|
||||||
use matrix_sdk_common::{
|
use matrix_sdk_common::{
|
||||||
api::r0::sync::sync_events::RoomSummary as RumaSummary,
|
|
||||||
events::{
|
events::{
|
||||||
presence::PresenceEvent,
|
presence::PresenceEvent, room::member::MemberEventContent, AnyBasicEvent,
|
||||||
room::{
|
AnyStrippedStateEvent, AnySyncStateEvent, EventContent, EventType, SyncStateEvent,
|
||||||
encryption::EncryptionEventContent, member::MemberEventContent,
|
|
||||||
power_levels::PowerLevelsEventContent,
|
|
||||||
},
|
},
|
||||||
AnyBasicEvent, AnyStrippedStateEvent, AnySyncStateEvent, EventContent, EventType,
|
identifiers::{RoomId, UserId},
|
||||||
SyncStateEvent,
|
|
||||||
},
|
|
||||||
identifiers::{RoomAliasId, RoomId, UserId},
|
|
||||||
};
|
};
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
use sled::{transaction::TransactionResult, Config, Db, Transactional, Tree};
|
use sled::{transaction::TransactionResult, Config, Db, Transactional, Tree};
|
||||||
use tracing::info;
|
use tracing::info;
|
||||||
|
|
||||||
|
use crate::{rooms::InnerSummary, Session};
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct Store {
|
pub struct Store {
|
||||||
inner: Db,
|
inner: Db,
|
||||||
|
@ -41,8 +28,6 @@ pub struct Store {
|
||||||
presence: Tree,
|
presence: Tree,
|
||||||
}
|
}
|
||||||
|
|
||||||
use crate::{responses::UnreadNotificationsCount, Session};
|
|
||||||
|
|
||||||
#[derive(Debug, Default)]
|
#[derive(Debug, Default)]
|
||||||
pub struct StateChanges {
|
pub struct StateChanges {
|
||||||
pub session: Option<Session>,
|
pub session: Option<Session>,
|
||||||
|
@ -145,376 +130,6 @@ impl From<Session> for StateChanges {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
pub struct Room {
|
|
||||||
room_id: Arc<RoomId>,
|
|
||||||
own_user_id: Arc<UserId>,
|
|
||||||
inner: Arc<SyncMutex<InnerSummary>>,
|
|
||||||
store: Store,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, Default, Serialize, Deserialize)]
|
|
||||||
pub struct SomeSummary {
|
|
||||||
heroes: Vec<String>,
|
|
||||||
joined_member_count: u64,
|
|
||||||
invited_member_count: u64,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Signals to the `BaseClient` which `RoomState` to send to `EventEmitter`.
|
|
||||||
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
|
|
||||||
pub enum RoomType {
|
|
||||||
/// Represents a joined room, the `joined_rooms` HashMap will be used.
|
|
||||||
Joined,
|
|
||||||
/// Represents a left room, the `left_rooms` HashMap will be used.
|
|
||||||
Left,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Room {
|
|
||||||
pub fn new(own_user_id: &UserId, store: Store, room_id: &RoomId, room_type: RoomType) -> Self {
|
|
||||||
let room_id = Arc::new(room_id.clone());
|
|
||||||
|
|
||||||
Self {
|
|
||||||
own_user_id: Arc::new(own_user_id.clone()),
|
|
||||||
room_id: room_id.clone(),
|
|
||||||
store,
|
|
||||||
inner: Arc::new(SyncMutex::new(InnerSummary {
|
|
||||||
room_id,
|
|
||||||
room_type,
|
|
||||||
encryption: None,
|
|
||||||
summary: Default::default(),
|
|
||||||
last_prev_batch: None,
|
|
||||||
members_synced: false,
|
|
||||||
name: None,
|
|
||||||
canonical_alias: None,
|
|
||||||
avatar_url: None,
|
|
||||||
topic: None,
|
|
||||||
notification_counts: Default::default(),
|
|
||||||
})),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn are_members_synced(&self) -> bool {
|
|
||||||
self.inner.lock().unwrap().members_synced
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_j_members(&self) -> impl Stream<Item = RoomMember> + '_ {
|
|
||||||
let joined = self.store.get_joined_user_ids(self.room_id()).await;
|
|
||||||
let invited = self.store.get_invited_user_ids(self.room_id()).await;
|
|
||||||
|
|
||||||
let x = move |u| async move {
|
|
||||||
let presence = self.store.get_presence_event(&u).await;
|
|
||||||
let power = self
|
|
||||||
.store
|
|
||||||
.get_state_event(self.room_id(), EventType::RoomPowerLevels, "")
|
|
||||||
.await
|
|
||||||
.map(|e| {
|
|
||||||
if let AnySyncStateEvent::RoomPowerLevels(e) = e {
|
|
||||||
Some(e)
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.flatten();
|
|
||||||
|
|
||||||
self.store
|
|
||||||
.get_member_event(self.room_id(), &u)
|
|
||||||
.await
|
|
||||||
.map(|m| RoomMember {
|
|
||||||
user_id: UserId::try_from(m.state_key.as_str()).unwrap().into(),
|
|
||||||
event: m.into(),
|
|
||||||
presence: presence.into(),
|
|
||||||
power_levles: power.into(),
|
|
||||||
})
|
|
||||||
};
|
|
||||||
|
|
||||||
joined.chain(invited).filter_map(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Calculate the canonical display name of the 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>
|
|
||||||
pub async fn calculate_name(&self) -> String {
|
|
||||||
let inner = self.inner.lock().unwrap();
|
|
||||||
|
|
||||||
if let Some(name) = &inner.name {
|
|
||||||
let name = name.trim();
|
|
||||||
name.to_string()
|
|
||||||
} else if let Some(alias) = &inner.canonical_alias {
|
|
||||||
let alias = alias.alias().trim();
|
|
||||||
alias.to_string()
|
|
||||||
} else {
|
|
||||||
let joined = inner.summary.joined_member_count;
|
|
||||||
let invited = inner.summary.invited_member_count;
|
|
||||||
let heroes_count = inner.summary.heroes.len() as u64;
|
|
||||||
let invited_joined = (invited + joined).saturating_sub(1);
|
|
||||||
|
|
||||||
let members = self.get_j_members().await;
|
|
||||||
|
|
||||||
// info!(
|
|
||||||
// "Calculating name for {}, hero count {} members {:#?}",
|
|
||||||
// self.room_id(),
|
|
||||||
// heroes_count,
|
|
||||||
// members
|
|
||||||
// );
|
|
||||||
// TODO: This should use `self.heroes` but it is always empty??
|
|
||||||
//
|
|
||||||
let own_user_id = self.own_user_id.clone();
|
|
||||||
|
|
||||||
let is_own_member = |m: &RoomMember| m.user_id() == &*own_user_id;
|
|
||||||
|
|
||||||
if !inner.summary.heroes.is_empty() {
|
|
||||||
let mut names = stream::iter(inner.summary.heroes.iter())
|
|
||||||
.take(3)
|
|
||||||
.filter_map(|u| async move {
|
|
||||||
let user_id = UserId::try_from(u.as_str()).ok()?;
|
|
||||||
self.get_member(&user_id).await
|
|
||||||
})
|
|
||||||
.map(|mem| {
|
|
||||||
mem.display_name()
|
|
||||||
.map(|d| d.to_string())
|
|
||||||
.unwrap_or_else(|| mem.user_id().localpart().to_string())
|
|
||||||
})
|
|
||||||
.collect::<Vec<String>>()
|
|
||||||
.await;
|
|
||||||
names.sort();
|
|
||||||
names.join(", ")
|
|
||||||
} else if heroes_count >= invited_joined {
|
|
||||||
let mut names = members
|
|
||||||
.filter(|m| future::ready(is_own_member(m)))
|
|
||||||
.take(3)
|
|
||||||
.map(|mem| {
|
|
||||||
mem.display_name()
|
|
||||||
.map(|d| d.to_string())
|
|
||||||
.unwrap_or_else(|| mem.user_id().localpart().to_string())
|
|
||||||
})
|
|
||||||
.collect::<Vec<String>>()
|
|
||||||
.await;
|
|
||||||
// stabilize ordering
|
|
||||||
names.sort();
|
|
||||||
names.join(", ")
|
|
||||||
} else if heroes_count < invited_joined && invited + joined > 1 {
|
|
||||||
let mut names = members
|
|
||||||
.filter(|m| future::ready(is_own_member(m)))
|
|
||||||
.take(3)
|
|
||||||
.map(|mem| {
|
|
||||||
mem.display_name()
|
|
||||||
.map(|d| d.to_string())
|
|
||||||
.unwrap_or_else(|| mem.user_id().localpart().to_string())
|
|
||||||
})
|
|
||||||
.collect::<Vec<String>>()
|
|
||||||
.await;
|
|
||||||
names.sort();
|
|
||||||
|
|
||||||
// TODO: What length does the spec want us to use here and in
|
|
||||||
// the `else`?
|
|
||||||
format!("{}, and {} others", names.join(", "), (joined + invited))
|
|
||||||
} else {
|
|
||||||
"Empty room".to_string()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn own_user_id(&self) -> &UserId {
|
|
||||||
&self.own_user_id
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn clone_summary(&self) -> InnerSummary {
|
|
||||||
(*self.inner.lock().unwrap()).clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn joined_user_ids(&self) -> impl Stream<Item = UserId> {
|
|
||||||
self.store.get_joined_user_ids(&self.room_id).await
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn is_encrypted(&self) -> bool {
|
|
||||||
self.inner.lock().unwrap().encryption.is_some()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn update_summary(&self, summary: InnerSummary) {
|
|
||||||
let mut inner = self.inner.lock().unwrap();
|
|
||||||
*inner = summary;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_member(&self, user_id: &UserId) -> Option<RoomMember> {
|
|
||||||
let presence = self.store.get_presence_event(user_id).await;
|
|
||||||
let power = self
|
|
||||||
.store
|
|
||||||
.get_state_event(self.room_id(), EventType::RoomPowerLevels, "")
|
|
||||||
.await
|
|
||||||
.map(|e| {
|
|
||||||
if let AnySyncStateEvent::RoomPowerLevels(e) = e {
|
|
||||||
Some(e)
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.flatten();
|
|
||||||
|
|
||||||
self.store
|
|
||||||
.get_member_event(&self.room_id, user_id)
|
|
||||||
.await
|
|
||||||
.map(|e| RoomMember {
|
|
||||||
user_id: UserId::try_from(e.state_key.as_str()).unwrap().into(),
|
|
||||||
event: e.into(),
|
|
||||||
presence: presence.into(),
|
|
||||||
power_levles: power.into(),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn room_id(&self) -> &RoomId {
|
|
||||||
&self.room_id
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn last_prev_batch(&self) -> Option<String> {
|
|
||||||
self.inner.lock().unwrap().last_prev_batch.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn display_name(&self) -> String {
|
|
||||||
self.calculate_name().await
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
|
||||||
pub struct RoomMember {
|
|
||||||
user_id: Arc<UserId>,
|
|
||||||
event: Arc<SyncStateEvent<MemberEventContent>>,
|
|
||||||
presence: Arc<Option<PresenceEvent>>,
|
|
||||||
power_levles: Arc<Option<SyncStateEvent<PowerLevelsEventContent>>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl RoomMember {
|
|
||||||
pub fn user_id(&self) -> &UserId {
|
|
||||||
&self.user_id
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn display_name(&self) -> Option<&str> {
|
|
||||||
self.event.content.displayname.as_deref()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn power_level(&self) -> i64 {
|
|
||||||
self.power_levles
|
|
||||||
.as_ref()
|
|
||||||
.as_ref()
|
|
||||||
.map(|e| {
|
|
||||||
e.content
|
|
||||||
.users
|
|
||||||
.get(&self.user_id())
|
|
||||||
.map(|p| (*p).into())
|
|
||||||
.unwrap_or(e.content.users_default.into())
|
|
||||||
})
|
|
||||||
.unwrap_or(0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, Serialize, Deserialize)]
|
|
||||||
pub struct InnerSummary {
|
|
||||||
room_id: Arc<RoomId>,
|
|
||||||
room_type: RoomType,
|
|
||||||
|
|
||||||
name: Option<String>,
|
|
||||||
canonical_alias: Option<RoomAliasId>,
|
|
||||||
avatar_url: Option<String>,
|
|
||||||
topic: Option<String>,
|
|
||||||
|
|
||||||
notification_counts: UnreadNotificationsCount,
|
|
||||||
summary: SomeSummary,
|
|
||||||
members_synced: bool,
|
|
||||||
|
|
||||||
encryption: Option<EncryptionEventContent>,
|
|
||||||
last_prev_batch: Option<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl InnerSummary {
|
|
||||||
pub fn mark_as_joined(&mut self) {
|
|
||||||
self.room_type = RoomType::Joined;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn mark_as_left(&mut self) {
|
|
||||||
self.room_type = RoomType::Left;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn mark_members_synced(&mut self) {
|
|
||||||
self.members_synced = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn mark_members_missing(&mut self) {
|
|
||||||
self.members_synced = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn set_prev_batch(&mut self, prev_batch: Option<&str>) -> bool {
|
|
||||||
if self.last_prev_batch.as_deref() != prev_batch {
|
|
||||||
self.last_prev_batch = prev_batch.map(|p| p.to_string());
|
|
||||||
true
|
|
||||||
} else {
|
|
||||||
false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn handle_state_event(&mut self, event: &AnySyncStateEvent) -> bool {
|
|
||||||
match event {
|
|
||||||
AnySyncStateEvent::RoomEncryption(encryption) => {
|
|
||||||
info!("MARKING ROOM {} AS ENCRYPTED", self.room_id);
|
|
||||||
self.encryption = Some(encryption.content.clone());
|
|
||||||
true
|
|
||||||
}
|
|
||||||
AnySyncStateEvent::RoomName(n) => {
|
|
||||||
self.name = n.content.name().map(|n| n.to_string());
|
|
||||||
true
|
|
||||||
}
|
|
||||||
AnySyncStateEvent::RoomCanonicalAlias(a) => {
|
|
||||||
self.canonical_alias = a.content.alias.clone();
|
|
||||||
true
|
|
||||||
}
|
|
||||||
AnySyncStateEvent::RoomTopic(t) => {
|
|
||||||
self.topic = Some(t.content.topic.clone());
|
|
||||||
true
|
|
||||||
}
|
|
||||||
_ => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn is_encrypted(&self) -> bool {
|
|
||||||
self.encryption.is_some()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn update_notification_count(&mut self, notification_counts: UnreadNotificationsCount) {
|
|
||||||
self.notification_counts = notification_counts;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn update(&mut self, summary: &RumaSummary) -> bool {
|
|
||||||
let mut changed = false;
|
|
||||||
|
|
||||||
info!("UPDAGING SUMMARY FOR {} WITH {:#?}", self.room_id, summary);
|
|
||||||
|
|
||||||
if !summary.is_empty() {
|
|
||||||
if !summary.heroes.is_empty() {
|
|
||||||
self.summary.heroes = summary.heroes.clone();
|
|
||||||
changed = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(joined) = summary.joined_member_count {
|
|
||||||
self.summary.joined_member_count = joined.into();
|
|
||||||
changed = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(invited) = summary.invited_member_count {
|
|
||||||
self.summary.invited_member_count = invited.into();
|
|
||||||
changed = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
changed
|
|
||||||
}
|
|
||||||
|
|
||||||
fn serialize(&self) -> Vec<u8> {
|
|
||||||
serde_json::to_vec(&self).unwrap()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Store {
|
impl Store {
|
||||||
fn open_helper(db: Db) -> Self {
|
fn open_helper(db: Db) -> Self {
|
||||||
let session = db.open_tree("session").unwrap();
|
let session = db.open_tree("session").unwrap();
|
||||||
|
|
Loading…
Reference in New Issue