106 lines
4.0 KiB
C#
106 lines
4.0 KiB
C#
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;
|
|
using Robust.Shared.Replays;
|
|
using Robust.Shared.Serialization.Markdown.Mapping;
|
|
|
|
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!;
|
|
[Dependency] private readonly IReplayRecordingManager _replay = 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));
|
|
_replay.OnRecordingStarted += OnStartReplayRecording;
|
|
}
|
|
|
|
private void OnStartReplayRecording((MappingDataNode, List<object>) initReplayData)
|
|
{
|
|
// replays will need information about currently loaded extra resources
|
|
foreach (var (path, data) in ContentRoot.GetAllFiles())
|
|
{
|
|
initReplayData.Item2.Add(new ReplayResourceUploadMsg { RelativePath = path, Data = data });
|
|
}
|
|
}
|
|
|
|
/// <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);
|
|
}
|
|
|
|
_replay.QueueReplayMessage(new ReplayResourceUploadMsg { RelativePath = msg.RelativePath, Data = msg.Data });
|
|
|
|
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 = new 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);
|
|
}
|
|
}
|