Optional server whitelists (#5979)

2022-01-04 06:37:06 -07:00
parent 66deacc322
commit af785f873f
13 changed files with 1958 additions and 1 deletions
--- a/Content.Server/Whitelist/WhitelistCommands.cs
+++ b/Content.Server/Whitelist/WhitelistCommands.cs
@@ -0,0 +1,105 @@
+using Content.Server.Administration;
+using Content.Server.Database;
+using Content.Shared.Administration;
+using Content.Shared.CCVar;
+using Robust.Server.Player;
+using Robust.Shared.Configuration;
+using Robust.Shared.Console;
+using Robust.Shared.IoC;
+using Robust.Shared.Localization;
+using Robust.Shared.Network;
+using Robust.Shared.Player;
+
+namespace Content.Server.Whitelist;
+
+[AdminCommand(AdminFlags.Ban)]
+public class AddWhitelistCommand : IConsoleCommand
+{
+    public string Command => "whitelistadd";
+    public string Description => Loc.GetString("command-whitelistadd-description");
+    public string Help => Loc.GetString("command-whitelistadd-help");
+    public async void Execute(IConsoleShell shell, string argStr, string[] args)
+    {
+        if (args.Length != 1)
+            return;
+
+        var db = IoCManager.Resolve<IServerDbManager>();
+        var loc = IoCManager.Resolve<IPlayerLocator>();
+
+        var name = args[0];
+        var data = await loc.LookupIdByNameAsync(name);
+
+        if (data != null)
+        {
+            var guid = data.UserId;
+            var isWhitelisted = await db.GetWhitelistStatusAsync(guid);
+            if (isWhitelisted)
+                return;
+
+            await db.AddToWhitelistAsync(guid);
+        }
+    }
+}
+
+[AdminCommand(AdminFlags.Ban)]
+public class RemoveWhitelistCommand : IConsoleCommand
+{
+    public string Command => "whitelistremove";
+    public string Description => Loc.GetString("command-whitelistremove-description");
+    public string Help => Loc.GetString("command-whitelistremove-help");
+    public async void Execute(IConsoleShell shell, string argStr, string[] args)
+    {
+        if (args.Length != 1)
+            return;
+
+        var db = IoCManager.Resolve<IServerDbManager>();
+        var loc = IoCManager.Resolve<IPlayerLocator>();
+
+        var name = args[0];
+        var data = await loc.LookupIdByNameAsync(name);
+
+        if (data != null)
+        {
+            var guid = data.UserId;
+            var isWhitelisted = await db.GetWhitelistStatusAsync(guid);
+            if (!isWhitelisted)
+                return;
+
+            await db.RemoveFromWhitelistAsync(guid);
+        }
+    }
+}
+
+[AdminCommand(AdminFlags.Ban)]
+public class KickNonWhitelistedCommand : IConsoleCommand
+{
+    public string Command => "kicknonwhitelisted";
+    public string Description => Loc.GetString("command-kicknonwhitelisted-description");
+    public string Help => Loc.GetString("command-kicknonwhitelisted-help");
+    public async void Execute(IConsoleShell shell, string argStr, string[] args)
+    {
+        if (args.Length != 0)
+            return;
+
+        var cfg = IoCManager.Resolve<IConfigurationManager>();
+
+        if (!cfg.GetCVar(CCVars.WhitelistEnabled))
+            return;
+
+        var player = IoCManager.Resolve<IPlayerManager>();
+        var db = IoCManager.Resolve<IServerDbManager>();
+        var net = IoCManager.Resolve<IServerNetManager>();
+
+        foreach (var session in player.NetworkedSessions)
+        {
+            if (await db.GetAdminDataForAsync(session.UserId) is not null)
+                continue;
+
+            if (!await db.GetWhitelistStatusAsync(session.UserId))
+            {
+                net.DisconnectChannel(session.ConnectedClient, Loc.GetString("whitelist-not-whitelisted"));
+            }
+        }
+
+    }
+}