PermissionsManager and DefaultPermissionsProvider improvements

LabApi/Features/Permissions/PermissionsManager.cs

@@ -2,6 +2,7 @@
 using LabApi.Features.Wrappers;
 using System;
 using System.Collections.Generic;
+using System.Diagnostics.CodeAnalysis;
 using System.Linq;
 
 namespace LabApi.Features.Permissions;
@@ -31,13 +32,7 @@ public static void RegisterProvider<T>()
             return;
         }
 
-        if (Activator.CreateInstance<T>() is not IPermissionsProvider provider)
-        {
-            Logger.Error($"{LoggerPrefix} Failed to create an instance of the permission provider of type {typeof(T).FullName}.");
-            return;
-        }
-
-        PermissionProviders.Add(typeof(T), provider);
+        PermissionProviders.Add(typeof(T), new T());
     }
 
     /// <summary>
@@ -62,13 +57,38 @@ public static void UnregisterProvider<T>()
     /// <returns>The registered <see cref="IPermissionsProvider"/> of the given type <typeparamref name="T"/>; otherwise, null.</returns>
     public static IPermissionsProvider? GetProvider<T>()
         where T : IPermissionsProvider, new()
+        => GetProvider(typeof(T));
+
+    /// <summary>
+    /// Retrieves the registered <see cref="IPermissionsProvider"/> of the given type <typeparamref name="T"/>.
+    /// </summary>
+    /// <typeparam name="T">The type of the permission provider to retrieve.</typeparam>
+    /// <returns>The registered <see cref="IPermissionsProvider"/> of the given type <typeparamref name="T"/>; otherwise, null.</returns>
+    public static bool TryGetProvider<T>([NotNullWhen(true)] out T? provider)
+        where T : class, IPermissionsProvider, new()
     {
-        if (PermissionProviders.TryGetValue(typeof(T), out IPermissionsProvider provider))
+        provider = GetProvider(typeof(T)) as T;
+        return provider != null;
+    }
+
+    /// <summary>
+    /// Retrieves the registered <see cref="IPermissionsProvider"/> of the given type <paramref name="providerType"/>.
+    /// </summary>
+    /// <param name="providerType">The type of the permission provider to retrieve.</param>
+    /// <returns>The registered <see cref="IPermissionsProvider"/> of the given type <paramref name="providerType"/>; otherwise, null.</returns>
+    public static IPermissionsProvider? GetProvider(Type providerType)
+    {
+        if (providerType == null)
+        {
+            throw new ArgumentNullException(nameof(providerType));
+        }
+
+        if (PermissionProviders.TryGetValue(providerType, out IPermissionsProvider provider))
         {
             return provider;
         }
 
-        Logger.Warn($"{LoggerPrefix} The permission provider of type {typeof(T).FullName} is not registered.");
+        Logger.Warn($"{LoggerPrefix} The permission provider of type {providerType.FullName} is not registered.");
         return null;
     }
 

LabApi/Features/Permissions/Providers/DefaultPermissionsProvider.cs

@@ -5,6 +5,7 @@
 using Serialization;
 using System;
 using System.Collections.Generic;
+using System.Diagnostics.CodeAnalysis;
 using System.IO;
 using System.Linq;
 
@@ -46,6 +47,10 @@ public DefaultPermissionsProvider()
         {
             // We deserialize the permissions from the file.
             _permissionsDictionary = YamlParser.Deserializer.Deserialize<Dictionary<string, PermissionGroup>>(File.ReadAllText(_permissions.FullName));
+            foreach (PermissionGroup permissionGroup in _permissionsDictionary.Values)
+            {
+                permissionGroup.IsRuntime = false;
+            }
 
             // We then reload the permissions to fill the special permissions.
             ReloadPermissions();
@@ -110,9 +115,34 @@ public void RemovePermissions(Player player, params string[] permissions)
     /// <inheritdoc />
     void IPermissionsProvider.ReloadPermissions() => ReloadPermissions();
 
-    private PermissionGroup GetPlayerGroup(Player player) => _permissionsDictionary.GetValueOrDefault(player.PermissionsGroupName ?? "default") ?? PermissionGroup.Default;
+    /// <summary>
+    /// Gets the <see cref="PermissionGroup"/> the player is a part of.
+    /// </summary>
+    /// <param name="player">The player whose <see cref="PermissionGroup"/> to find.</param>
+    /// <returns>The <see cref="PermissionGroup"/> the player is a part of, otherwise <see cref="PermissionGroup.Default"/>.</returns>
+    public PermissionGroup GetPlayerGroup(Player player) => GetPermissionGroup(player.PermissionsGroupName ?? "default");
+
+    /// <summary>
+    /// Gets the <see cref="PermissionGroup"/> from a <see cref="UserGroup.Name"/>.
+    /// </summary>
+    /// <param name="groupName">A <see cref="UserGroup.Name"/> to find the <see cref="PermissionGroup"/> of.</param>
+    /// <returns>A <see cref="PermissionGroup"/> if one is defined, <see cref="PermissionGroup.Default"/> otherwise.</returns>
+    public PermissionGroup GetPermissionGroup(string groupName) => _permissionsDictionary.GetValueOrDefault(groupName) ?? PermissionGroup.Default;
 
-    private string[] GetPermissions(PermissionGroup group)
+    /// <summary>
+    /// Tries to get the <see cref="PermissionGroup"/> from a <see cref="UserGroup.Name"/>.
+    /// </summary>
+    /// <param name="groupName">A <see cref="UserGroup.Name"/> to find the <see cref="PermissionGroup"/> of.</param>
+    /// <param name="permissionGroup">The found <see cref="PermissionGroup"/> when true, null otherwise.</param>
+    /// <returns>Whether a <see cref="PermissionGroup"/> with the registry name of <paramref name="groupName"/> was found.</returns>
+    public bool TryGetPermissionGroup(string groupName, [NotNullWhen(true)] out PermissionGroup? permissionGroup) => _permissionsDictionary.TryGetValue(groupName, out permissionGroup);
+
+    /// <summary>
+    /// Gets the <see cref="string"/> array of permissions a <see cref="PermissionGroup"/> grants.
+    /// </summary>
+    /// <param name="group">The <see cref="PermissionGroup"/>, permissions of which will be returned.</param>
+    /// <returns>A <see cref="string"/> array of permissions this group grants.</returns>
+    public string[] GetPermissions(PermissionGroup group)
     {
         List<string> permissions = ListPool<string>.Shared.Rent();
 
@@ -132,6 +162,36 @@ private string[] GetPermissions(PermissionGroup group)
         return [.. permissions];
     }
 
+    /// <summary>
+    /// Adds a new permission group.
+    /// </summary>
+    /// <param name="groupName">A <see cref="UserGroup.Name"/> the <paramref name="group"/> is linked to.</param>
+    /// <param name="group">The group to add.</param>
+    /// <returns>Whether the group was successfully added. False if group with this name is already registered or if the <paramref name="group"/>'s <see cref="PermissionGroup.IsRuntime"/> is set to false.</returns>
+    public bool AddPermissionGroup(string groupName, PermissionGroup group)
+        => group.IsRuntime && _permissionsDictionary.TryAdd(groupName, group);
+
+    /// <summary>
+    /// Removes a permission group if it exists.
+    /// </summary>
+    /// <param name="groupName">A <see cref="UserGroup.Name"/> which links to a <see cref="PermissionGroup"/> that is to be removed.</param>
+    /// <param name="group">The group which was removed or null.</param>
+    /// <returns>Whether the group was found and removed successfully. False if group with this name could not be found or if the <paramref name="group"/>'s <see cref="PermissionGroup.IsRuntime"/> is set to false.</returns>
+    public bool RemovePermissionGroup(string groupName, [NotNullWhen(true)] out PermissionGroup? group)
+    {
+        if (!_permissionsDictionary.TryGetValue(groupName, out group))
+        {
+            return false;
+        }
+
+        if (!group.IsRuntime)
+        {
+            return false;
+        }
+
+        return _permissionsDictionary.Remove(groupName, out group);
+    }
+
     private bool HasPermission(PermissionGroup group, string permission)
     {
         if (group.IsRoot)
@@ -208,5 +268,5 @@ private void ReloadPermissions()
         }
     }
 
-    private void SavePermissions() => File.WriteAllText(_permissions.FullName, YamlParser.Serializer.Serialize(_permissionsDictionary));
+    private void SavePermissions() => File.WriteAllText(_permissions.FullName, YamlParser.Serializer.Serialize(_permissionsDictionary.Where(kvp => !kvp.Value.IsRuntime).ToDictionary(kvp => kvp.Key, kvp => kvp.Value)));
 }

LabApi/Features/Permissions/Providers/PermissionGroup.cs

@@ -11,7 +11,7 @@ public class PermissionGroup
     /// <summary>
     /// Gets the default permission group.
     /// </summary>
-    public static PermissionGroup Default => new([], []);
+    public static PermissionGroup Default => new([], []) { IsRuntime = false };
 
     /// <summary>
     /// Generates the default permission groups based on the available groups in the RA settings.
@@ -64,6 +64,12 @@ public PermissionGroup(string[] inheritedGroups, string[] permissions)
     [YamlIgnore]
     public bool IsRoot { get; set; } = false;
 
+    /// <summary>
+    /// An internal bool indicating whether the permission was created at runtime and should not be saved.
+    /// </summary>
+    [YamlIgnore]
+    public bool IsRuntime { get; internal set; } = true;
+
     /// <summary>
     /// An internal dictionary that saves special permissions. (x.*).
     /// </summary>