Adds Network Resource Uploading for admins. (#6904)

Co-authored-by: Pieter-Jan Briers <pieterjan.briers@gmail.com>
Co-authored-by: Pieter-Jan Briers <pieterjan.briers+git@gmail.com>

Content.Server/Administration/NetworkResourceManager.cs

@@ -0,0 +1,90 @@
+using Content.Server.Administration.Managers;
+using Content.Server.Database;
+using Content.Shared.Administration;
+using Content.Shared.CCVar;
+using Robust.Server.Player;
+using Robust.Shared.Configuration;
+using Robust.Shared.Network;
+
+namespace Content.Server.Administration;
+
+public sealed class NetworkResourceManager : SharedNetworkResourceManager
+{
+    [Dependency] private readonly IPlayerManager _playerManager = default!;
+    [Dependency] private readonly IAdminManager _adminManager = default!;
+    [Dependency] private readonly IServerNetManager _serverNetManager = default!;
+    [Dependency] private readonly IConfigurationManager _cfgManager = default!;
+    [Dependency] private readonly IServerDbManager _serverDb = default!;
+
+    [ViewVariables] public bool Enabled { get; private set; } = true;
+    [ViewVariables] public float SizeLimit { get; private set; } = 0f;
+    [ViewVariables] public bool StoreUploaded { get; set; } = true;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        _serverNetManager.Connected += ServerNetManagerOnConnected;
+        _cfgManager.OnValueChanged(CCVars.ResourceUploadingEnabled, value => Enabled = value, true);
+        _cfgManager.OnValueChanged(CCVars.ResourceUploadingLimitMb, value => SizeLimit = value, true);
+        _cfgManager.OnValueChanged(CCVars.ResourceUploadingStoreEnabled, value => StoreUploaded = value, true);
+
+        AutoDelete(_cfgManager.GetCVar(CCVars.ResourceUploadingStoreDeletionDays));
+    }
+
+    /// <summary>
+    ///     Callback for when a client attempts to upload a resource.
+    /// </summary>
+    /// <param name="msg"></param>
+    /// <exception cref="NotImplementedException"></exception>
+    protected override async void ResourceUploadMsg(NetworkResourceUploadMessage msg)
+    {
+        // Do not allow uploading any new resources if it has been disabled.
+        // Note: Any resources uploaded before being disabled will still be kept and sent.
+        if (!Enabled)
+            return;
+
+        if (!_playerManager.TryGetSessionByChannel(msg.MsgChannel, out var session))
+            return;
+
+        // +QUERY only for now.
+        if (!_adminManager.HasAdminFlag(session, AdminFlags.Query))
+            return;
+
+        // Ensure the data is under the current size limit, if it's currently enabled.
+        if (SizeLimit > 0f && msg.Data.Length * BytesToMegabytes > SizeLimit)
+            return;
+
+        ContentRoot.AddOrUpdateFile(msg.RelativePath, msg.Data);
+
+        // Now we broadcast the message!
+        foreach (var channel in _serverNetManager.Channels)
+        {
+            channel.SendMessage(msg);
+        }
+
+        if (!StoreUploaded)
+            return;
+
+        await _serverDb.AddUploadedResourceLogAsync(session.UserId, DateTime.Now, msg.RelativePath.ToString(), msg.Data);
+    }
+
+    private void ServerNetManagerOnConnected(object? sender, NetChannelArgs e)
+    {
+        foreach (var (path, data) in ContentRoot.GetAllFiles())
+        {
+            var msg = _serverNetManager.CreateNetMessage<NetworkResourceUploadMessage>();
+            msg.RelativePath = path;
+            msg.Data = data;
+            e.Channel.SendMessage(msg);
+        }
+    }
+
+    private async void AutoDelete(int days)
+    {
+        if (days <= 0)
+            return; // auto-deletion disabled...
+
+        await _serverDb.PurgeUploadedResourceLogAsync(days);
+    }
+}