From 4340b1899470c06f170817dbc1200040619fbf8d Mon Sep 17 00:00:00 2001 From: TheArcaneBrony Date: Sat, 14 Oct 2023 23:19:25 +0200 Subject: Handle floats etc in requests --- .../State/RoomInfo/RoomPowerLevelEventContent.cs | 30 +++++++++++++++++++--- 1 file changed, 27 insertions(+), 3 deletions(-) (limited to 'LibMatrix/EventTypes/Spec/State/RoomInfo/RoomPowerLevelEventContent.cs') diff --git a/LibMatrix/EventTypes/Spec/State/RoomInfo/RoomPowerLevelEventContent.cs b/LibMatrix/EventTypes/Spec/State/RoomInfo/RoomPowerLevelEventContent.cs index 7f37271..a1bf517 100644 --- a/LibMatrix/EventTypes/Spec/State/RoomInfo/RoomPowerLevelEventContent.cs +++ b/LibMatrix/EventTypes/Spec/State/RoomInfo/RoomPowerLevelEventContent.cs @@ -3,13 +3,15 @@ using LibMatrix.Interfaces; namespace LibMatrix.EventTypes.Spec.State; -[MatrixEvent(EventName = "m.room.power_levels")] +[MatrixEvent(EventName = EventId)] public class RoomPowerLevelEventContent : EventContent { + public const string EventId = "m.room.power_levels"; + [JsonPropertyName("ban")] public long? Ban { get; set; } = 50; [JsonPropertyName("events_default")] - public long EventsDefault { get; set; } = 0; + public long? EventsDefault { get; set; } = 0; [JsonPropertyName("events")] public Dictionary? Events { get; set; } // = null!; @@ -46,10 +48,32 @@ public class RoomPowerLevelEventContent : EventContent { } public bool IsUserAdmin(string userId) { + if(userId is null) throw new ArgumentNullException(nameof(userId)); return Users.TryGetValue(userId, out var level) && level >= Events.Max(x => x.Value); } public bool UserHasPermission(string userId, string eventType) { - return Users.TryGetValue(userId, out var level) && level >= Events.GetValueOrDefault(eventType, EventsDefault); + if(userId is null) throw new ArgumentNullException(nameof(userId)); + return Users.TryGetValue(userId, out var level) && level >= Events.GetValueOrDefault(eventType, EventsDefault ?? 0); + } + + public long GetUserPowerLevel(string userId) { + if(userId is null) throw new ArgumentNullException(nameof(userId)); + return Users.TryGetValue(userId, out var level) ? level : UsersDefault ?? UsersDefault ?? 0; + } + + public long GetEventPowerLevel(string eventType) { + return Events.TryGetValue(eventType, out var level) ? level : EventsDefault ?? EventsDefault ?? 0; + } + + public void SetUserPowerLevel(string userId, long powerLevel) { + if(userId is null) throw new ArgumentNullException(nameof(userId)); + Users ??= new(); + if (Users.TryGetValue(userId, out var level)) { + Users[userId] = powerLevel; + } + else { + Users.Add(userId, powerLevel); + } } } -- cgit 1.4.1