1use ruma::OwnedUserId;
16
17use super::Room;
18
19impl Room {
20 pub fn has_active_room_call(&self) -> bool {
23 self.info.read().has_active_room_call()
24 }
25
26 pub fn active_room_call_participants(&self) -> Vec<OwnedUserId> {
35 self.info.read().active_room_call_participants()
36 }
37}
38
39#[cfg(test)]
40mod tests {
41 use std::{ops::Sub, sync::Arc, time::Duration};
42
43 use assign::assign;
44 use matrix_sdk_test::{ALICE, BOB, CAROL, event_factory::EventFactory};
45 use ruma::{
46 DeviceId, EventId, MilliSecondsSinceUnixEpoch, OwnedUserId, UserId, device_id, event_id,
47 events::{
48 AnySyncStateEvent,
49 call::member::{
50 ActiveFocus, ActiveLivekitFocus, Application, CallApplicationContent,
51 CallMemberEventContent, CallMemberStateKey, Focus, LegacyMembershipData,
52 LegacyMembershipDataInit, LivekitFocus,
53 },
54 },
55 room_id,
56 time::SystemTime,
57 user_id,
58 };
59 use similar_asserts::assert_eq;
60
61 use super::super::{Room, RoomState};
62 use crate::store::MemoryStore;
63
64 fn make_room_test_helper(room_type: RoomState) -> (Arc<MemoryStore>, Room) {
65 let store = Arc::new(MemoryStore::new());
66 let user_id = user_id!("@me:example.org");
67 let room_id = room_id!("!test:localhost");
68 let (sender, _receiver) = tokio::sync::broadcast::channel(1);
69
70 (store.clone(), Room::new(user_id, store, room_id, room_type, sender))
71 }
72
73 fn timestamp(minutes_ago: u32) -> MilliSecondsSinceUnixEpoch {
74 MilliSecondsSinceUnixEpoch::from_system_time(
75 SystemTime::now().sub(Duration::from_secs((60 * minutes_ago).into())),
76 )
77 .expect("date out of range")
78 }
79
80 fn legacy_membership_for_my_call(
81 device_id: &DeviceId,
82 membership_id: &str,
83 minutes_ago: u32,
84 ) -> LegacyMembershipData {
85 let (application, foci) = foci_and_application();
86 assign!(
87 LegacyMembershipData::from(LegacyMembershipDataInit {
88 application,
89 device_id: device_id.to_owned(),
90 expires: Duration::from_millis(3_600_000),
91 foci_active: foci,
92 membership_id: membership_id.to_owned(),
93 }),
94 { created_ts: Some(timestamp(minutes_ago)) }
95 )
96 }
97
98 fn legacy_member_state_event(
99 memberships: Vec<LegacyMembershipData>,
100 ev_id: &EventId,
101 user_id: &UserId,
102 ) -> AnySyncStateEvent {
103 let content = CallMemberEventContent::new_legacy(memberships);
104 EventFactory::new()
105 .sender(user_id)
106 .event(content)
107 .state_key(CallMemberStateKey::new(user_id.to_owned(), None, false).as_ref())
108 .event_id(ev_id)
109 .server_ts(timestamp(0))
112 .into()
113 }
114
115 struct InitData<'a> {
116 device_id: &'a DeviceId,
117 minutes_ago: u32,
118 }
119
120 fn session_member_state_event(
121 ev_id: &EventId,
122 user_id: &UserId,
123 init_data: Option<InitData<'_>>,
124 ) -> AnySyncStateEvent {
125 let application = Application::Call(CallApplicationContent::new(
126 "my_call_id_1".to_owned(),
127 ruma::events::call::member::CallScope::Room,
128 ));
129 let foci_preferred = vec![Focus::Livekit(LivekitFocus::new(
130 "my_call_foci_alias".to_owned(),
131 "https://lk.org".to_owned(),
132 ))];
133 let focus_active = ActiveFocus::Livekit(ActiveLivekitFocus::new());
134
135 let (content, state_key) = match init_data {
136 Some(InitData { device_id, minutes_ago }) => {
137 let member_id = format!("{device_id}_m.call");
138 (
139 CallMemberEventContent::new(
140 application,
141 device_id.to_owned(),
142 focus_active,
143 foci_preferred,
144 Some(timestamp(minutes_ago)),
145 None,
146 ),
147 CallMemberStateKey::new(user_id.to_owned(), Some(member_id), false),
148 )
149 }
150
151 None => (
152 CallMemberEventContent::new_empty(None),
153 CallMemberStateKey::new(user_id.to_owned(), None, false),
154 ),
155 };
156
157 EventFactory::new()
158 .sender(user_id)
159 .event(content)
160 .state_key(state_key.as_ref())
161 .event_id(ev_id)
162 .server_ts(timestamp(0))
165 .into()
166 }
167
168 fn foci_and_application() -> (Application, Vec<Focus>) {
169 (
170 Application::Call(CallApplicationContent::new(
171 "my_call_id_1".to_owned(),
172 ruma::events::call::member::CallScope::Room,
173 )),
174 vec![Focus::Livekit(LivekitFocus::new(
175 "my_call_foci_alias".to_owned(),
176 "https://lk.org".to_owned(),
177 ))],
178 )
179 }
180
181 fn receive_state_events(room: &Room, events: Vec<&AnySyncStateEvent>) {
182 room.info.update_if(|info| {
183 let mut res = false;
184 for ev in events {
185 res |= info.handle_state_event(ev);
186 }
187 res
188 });
189 }
190
191 fn legacy_create_call_with_member_events_for_user(a: &UserId, b: &UserId, c: &UserId) -> Room {
195 let (_, room) = make_room_test_helper(RoomState::Joined);
196
197 let a_empty = legacy_member_state_event(Vec::new(), event_id!("$1234"), a);
198
199 let m_init_b = legacy_membership_for_my_call(device_id!("DEVICE_0"), "0", 1);
201 let b_one = legacy_member_state_event(vec![m_init_b], event_id!("$12345"), b);
202
203 let m_init_c1 = legacy_membership_for_my_call(device_id!("DEVICE_0"), "0", 10);
205 let m_init_c2 = legacy_membership_for_my_call(device_id!("DEVICE_1"), "0", 20);
207 let c_two = legacy_member_state_event(vec![m_init_c1, m_init_c2], event_id!("$123456"), c);
208
209 receive_state_events(&room, vec![&c_two, &a_empty, &b_one]);
211
212 room
213 }
214
215 fn session_create_call_with_member_events_for_user(a: &UserId, b: &UserId, c: &UserId) -> Room {
219 let (_, room) = make_room_test_helper(RoomState::Joined);
220
221 let a_empty = session_member_state_event(event_id!("$1234"), a, None);
222
223 let b_one = session_member_state_event(
225 event_id!("$12345"),
226 b,
227 Some(InitData { device_id: "DEVICE_0".into(), minutes_ago: 1 }),
228 );
229
230 let m_c1 = session_member_state_event(
231 event_id!("$123456_0"),
232 c,
233 Some(InitData { device_id: "DEVICE_0".into(), minutes_ago: 10 }),
234 );
235 let m_c2 = session_member_state_event(
236 event_id!("$123456_1"),
237 c,
238 Some(InitData { device_id: "DEVICE_1".into(), minutes_ago: 20 }),
239 );
240 receive_state_events(&room, vec![&m_c1, &m_c2, &a_empty, &b_one]);
242
243 room
244 }
245
246 #[test]
247 fn test_show_correct_active_call_state() {
248 let room_legacy = legacy_create_call_with_member_events_for_user(&ALICE, &BOB, &CAROL);
249
250 assert_eq!(
254 vec![CAROL.to_owned(), CAROL.to_owned(), BOB.to_owned()],
255 room_legacy.active_room_call_participants()
256 );
257 assert!(room_legacy.has_active_room_call());
258
259 let room_session = session_create_call_with_member_events_for_user(&ALICE, &BOB, &CAROL);
260 assert_eq!(
261 vec![CAROL.to_owned(), CAROL.to_owned(), BOB.to_owned()],
262 room_session.active_room_call_participants()
263 );
264 assert!(room_session.has_active_room_call());
265 }
266
267 #[test]
268 fn test_active_call_is_false_when_everyone_left() {
269 let room = legacy_create_call_with_member_events_for_user(&ALICE, &BOB, &CAROL);
270
271 let b_empty_membership = legacy_member_state_event(Vec::new(), event_id!("$1234_1"), &BOB);
272 let c_empty_membership =
273 legacy_member_state_event(Vec::new(), event_id!("$12345_1"), &CAROL);
274
275 receive_state_events(&room, vec![&b_empty_membership, &c_empty_membership]);
276
277 assert_eq!(Vec::<OwnedUserId>::new(), room.active_room_call_participants());
279 assert!(!room.has_active_room_call());
280 }
281}