From d0faeb52f34200834dab5d6df3d0ad2219b5fedf Mon Sep 17 00:00:00 2001
From: Jonas Platte <jplatte@element.io>
Date: Wed, 13 Apr 2022 15:40:57 +0200
Subject: [PATCH] events: Add RoomPowerLevels::{for_user, max}

---
 crates/ruma-common/src/events/room/power_levels.rs | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/crates/ruma-common/src/events/room/power_levels.rs b/crates/ruma-common/src/events/room/power_levels.rs
index 8b44813b..59bf6ced 100644
--- a/crates/ruma-common/src/events/room/power_levels.rs
+++ b/crates/ruma-common/src/events/room/power_levels.rs
@@ -2,7 +2,7 @@
 //!
 //! [`m.room.power_levels`]: https://spec.matrix.org/v1.2/client-server-api/#mroompower_levels
 
-use std::collections::BTreeMap;
+use std::{cmp::max, collections::BTreeMap};
 
 use js_int::{int, Int};
 use ruma_macros::EventContent;
@@ -238,6 +238,18 @@ pub struct RoomPowerLevels {
     pub notifications: NotificationPowerLevels,
 }
 
+impl RoomPowerLevels {
+    /// Get the power level of a specific user.
+    pub fn for_user(&self, user_id: &UserId) -> Int {
+        self.users.get(user_id).map_or(self.users_default, |pl| *pl)
+    }
+
+    /// Get the maximum power level of any user.
+    pub fn max(&self) -> Int {
+        self.users.values().fold(self.users_default, |max_pl, user_pl| max(max_pl, *user_pl))
+    }
+}
+
 impl From<RoomPowerLevelsEventContent> for RoomPowerLevels {
     fn from(c: RoomPowerLevelsEventContent) -> Self {
         Self {