From d2e28974e638807c8b0f2408868c01c7c4fe8de1 Mon Sep 17 00:00:00 2001
From: HitPanda <104197232+EnefFlow@users.noreply.github.com>
Date: Mon, 18 Sep 2023 21:12:57 +0300
Subject: [PATCH] department ban (#415)

---
 .../Administration/Managers/RoleBanManager.cs | 96 +++++++++++++++++++
 .../Commands/UtkaJobBanCommand.cs             | 17 +++-
 2 files changed, 111 insertions(+), 2 deletions(-)

diff --git a/Content.Server/Administration/Managers/RoleBanManager.cs b/Content.Server/Administration/Managers/RoleBanManager.cs
index 963b962e22..5af9fd28a6 100644
--- a/Content.Server/Administration/Managers/RoleBanManager.cs
+++ b/Content.Server/Administration/Managers/RoleBanManager.cs
@@ -30,6 +30,7 @@ public sealed class RoleBanManager
     [Dependency] private readonly IPlayerLocator _playerLocator = default!;
     [Dependency] private readonly UtkaTCPWrapper _utkaSockets = default!; // WD
     [Dependency] private readonly IEntitySystemManager _systems = default!; // WD
+    [Dependency] private readonly IBanManager _banManager = default!; // WD
 
     private const string JobPrefix = "Job:";
 
@@ -67,6 +68,16 @@ public sealed class RoleBanManager
         return true;
     }
 
+    public void SendRoleBans(LocatedPlayerData located)
+    {
+        if (!_playerManager.TryGetSessionById(located.UserId, out var player))
+        {
+            return;
+        }
+
+        _banManager.SendRoleBans(player);
+    }
+
     public HashSet<string>? GetRoleBans(NetUserId playerUserId)
     {
         return _cachedRoleBans.TryGetValue(playerUserId, out var roleBans) ? roleBans.Select(banDef => banDef.Role).ToHashSet() : null;
@@ -121,6 +132,89 @@ public sealed class RoleBanManager
     }
 
     //WD start
+    public async void UtkaCreateDepartmentBan(string admin, string target, DepartmentPrototype department, string reason, uint minutes, bool isGlobalBan)
+    {
+        var located = await _playerLocator.LookupIdByNameOrIdAsync(target);
+        if (located == null)
+        {
+            UtkaSendResponse(false);
+            return;
+        }
+
+        var targetUid = located.UserId;
+        var targetHWid = located.LastHWId;
+        var targetAddress = located.LastAddress;
+
+        DateTimeOffset? expires = null;
+        if (minutes > 0)
+        {
+            expires = DateTimeOffset.Now + TimeSpan.FromMinutes(minutes);
+        }
+
+        (IPAddress, int)? addressRange = null;
+        if (targetAddress != null)
+        {
+            if (targetAddress.IsIPv4MappedToIPv6)
+                targetAddress = targetAddress.MapToIPv4();
+
+            // Ban /64 for IPv4, /32 for IPv4.
+            var cidr = targetAddress.AddressFamily == AddressFamily.InterNetworkV6 ? 64 : 32;
+            addressRange = (targetAddress, cidr);
+        }
+
+        var cfg = UnsafePseudoIoC.ConfigurationManager;
+        var serverName = cfg.GetCVar(CCVars.AdminLogsServerName);
+
+        if (isGlobalBan)
+        {
+            serverName = "unknown";
+        }
+
+        var locatedPlayer = await _playerLocator.LookupIdByNameOrIdAsync(admin);
+        if (locatedPlayer == null)
+        {
+            UtkaSendResponse(false);
+            return;
+        }
+        var player = locatedPlayer.UserId;
+
+        UtkaSendResponse(true);
+
+        _systems.TryGetEntitySystem<GameTicker>(out var ticker);
+        int? roundId = ticker == null || ticker.RoundId == 0 ? null : ticker.RoundId;
+        var playtime = (await _db.GetPlayTimes(targetUid)).Find(p => p.Tracker == PlayTimeTrackingShared.TrackerOverall)?.TimeSpent ?? TimeSpan.Zero;
+
+        foreach (var job in department.Roles)
+        {
+            var role = string.Concat(JobPrefix, job);
+
+            var banDef = new ServerRoleBanDef(
+                null,
+                targetUid,
+                addressRange,
+                targetHWid,
+                DateTimeOffset.Now,
+                expires,
+                roundId,
+                playtime,
+                reason,
+                NoteSeverity.High,
+                player,
+                null,
+                role,
+                serverName);
+
+            if (!await AddRoleBan(banDef))
+                continue;
+
+            var banId = await UtkaGetBanId(reason, role, targetUid);
+
+            UtkaSendJobBanEvent(admin, target, minutes, job, isGlobalBan, reason, banId);
+        }
+
+        SendRoleBans(located);
+    }
+
     public async void UtkaCreateJobBan(string admin, string target, string job, string reason, uint minutes, bool isGlobalBan)
     {
         if (!_prototypeManager.TryIndex<JobPrototype>(job, out _))
@@ -205,6 +299,8 @@ public sealed class RoleBanManager
 
         UtkaSendJobBanEvent(admin, target, minutes, job, isGlobalBan, reason, banId);
         UtkaSendResponse(true);
+
+        SendRoleBans(located);
     }
     //WD end
 
diff --git a/Content.Server/UtkaIntegration/Commands/UtkaJobBanCommand.cs b/Content.Server/UtkaIntegration/Commands/UtkaJobBanCommand.cs
index dbc25ed752..37efbcf0c4 100644
--- a/Content.Server/UtkaIntegration/Commands/UtkaJobBanCommand.cs
+++ b/Content.Server/UtkaIntegration/Commands/UtkaJobBanCommand.cs
@@ -1,16 +1,21 @@
 using Content.Server.Administration.Managers;
 using Content.Server.UtkaIntegration.TCP;
+using Content.Shared.Roles;
+using Robust.Shared.Prototypes;
 
 namespace Content.Server.UtkaIntegration;
 
 public sealed class UtkaJobBanCommand : IUtkaCommand
 {
+    [Dependency] private readonly IPrototypeManager _prototypeManager = default!;
+
     public string Name => "jobban";
     public Type RequestMessageType => typeof(UtkaJobBanRequest);
     public void Execute(UtkaTCPSession session, UtkaBaseMessage baseMessage)
     {
-        if (baseMessage is not UtkaJobBanRequest message) return;
+        IoCManager.InjectDependencies(this);
 
+        if (baseMessage is not UtkaJobBanRequest message) return;
         var target = message.Ckey!;
         var job = message.Type!;
         var reason = message.Reason!;
@@ -18,6 +23,14 @@ public sealed class UtkaJobBanCommand : IUtkaCommand
         var isGlobalBan = (bool) message.Global!;
         var admin = message.ACkey!;
 
-        IoCManager.Resolve<RoleBanManager>().UtkaCreateJobBan(admin, target, job, reason, minutes, isGlobalBan);
+        var roleBanManager = IoCManager.Resolve<RoleBanManager>();
+
+        if (_prototypeManager.TryIndex<DepartmentPrototype>(job, out var departmentProto))
+            roleBanManager.UtkaCreateDepartmentBan(admin, target, departmentProto, reason, minutes, isGlobalBan);
+
+        else
+            roleBanManager.UtkaCreateJobBan(admin, target, job, reason, minutes, isGlobalBan);
     }
 }
+
+