add "neededRank" argument for CustomCommand flag

Author: Elektryk-Andrzej

Code/FlagSystem/Flags/CustomCommandFlag.cs

@@ -1,5 +1,6 @@
 using CommandSystem;
 using JetBrains.Annotations;
+using LabApi.Features.Permissions;
 using RemoteAdmin;
 using SER.Code.Helpers.Exceptions;
 using SER.Code.Helpers.Extensions;
@@ -73,6 +74,13 @@ public class CustomCommandFlag : Flag
             "The description of the command.",
             AddDescription,
             false
+        ),
+        new(
+            "neededRank",
+            "The required remote admin rank in order to have access to this command. " +
+            "You can provide multiple ranks, and if the player has any of the listed ranks, they will be able to use the command.",
+            AddNeededRank,
+            false
         )
     ];
 
@@ -160,6 +168,7 @@ public bool Execute(ArraySegment<string> arguments, ICommandSender sender, out s
         public string Description { get; set; } = "";
         public ConsoleType ConsoleTypes { get; set; } = ConsoleType.Server;
         public string[] Usage { get; set; } = [];
+        public string[] NeededRanks { get; set; } = [];
         public string GetHelp(ArraySegment<string> arguments)
         {
             return $"Description: {Description}\n" +
@@ -173,6 +182,15 @@ public string GetHelp(ArraySegment<string> arguments)
 
     public static Result RunAttachedScript(CustomCommand requestingCommand, ScriptExecutor sender, string[] args)
     {
+        if (requestingCommand.NeededRanks.Any() && sender is IPlayerExecutor { Player: { } player })
+        {
+            if (player.UserGroup is not { } group || requestingCommand.NeededRanks.All(rank => group.Name != rank))
+            {
+                return "This command is reserved for players with a rank: " +
+                       $"{requestingCommand.NeededRanks.JoinStrings(", ")}";
+            }
+        }
+        
         if (!ScriptCommands.TryGetValue(requestingCommand, out var flag))
         {
             return "The script that was supposed to handle this command was not found.";
@@ -264,4 +282,10 @@ private Result AddDescription(string[] args)
         Command.Description = args.JoinStrings(" ");
         return true;
     }
+    
+    private Result AddNeededRank(string[] args)
+    {
+        Command.NeededRanks = args;
+        return true;
+    }
 }

Code/ScriptSystem/Structures/IPlayerExecutor.cs

@@ -0,0 +1,8 @@
+using LabApi.Features.Wrappers;
+
+namespace SER.Code.ScriptSystem.Structures;
+
+public interface IPlayerExecutor
+{
+    public Player? Player { get; }
+}

Code/ScriptSystem/Structures/PlayerConsoleExecutor.cs

@@ -1,9 +1,13 @@
-namespace SER.Code.ScriptSystem.Structures;
+using LabApi.Features.Wrappers;
 
-public class PlayerConsoleExecutor : ScriptExecutor
+namespace SER.Code.ScriptSystem.Structures;
+
+public class PlayerConsoleExecutor : ScriptExecutor, IPlayerExecutor
 {
     public required ReferenceHub Sender { get; init; }
 
+    public Player Player => Player.Get(Sender);
+
     public override void Reply(string content, Script scr)
     {
         Sender.gameConsoleTransmission.SendToClient(content, "green");

Code/ScriptSystem/Structures/RemoteAdminExecutor.cs

@@ -1,11 +1,14 @@
-using RemoteAdmin;
+using LabApi.Features.Wrappers;
+using RemoteAdmin;
 
 namespace SER.Code.ScriptSystem.Structures;
 
-public class RemoteAdminExecutor : ScriptExecutor
+public class RemoteAdminExecutor : ScriptExecutor, IPlayerExecutor
 {
     public required PlayerCommandSender Sender { get; init; }
 
+    public Player? Player => Player.Get(Sender);
+
     public override void Reply(string content, Script scr)
     {
         Sender.Print(content);