RobustToolbox/Robust.Shared/Upload/SharedNetworkResourceManager.cs

using System;
using System.Collections.Generic;
using Robust.Shared.ContentPack;
using Robust.Shared.IoC;
using Robust.Shared.Network;
using Robust.Shared.Replays;
using Robust.Shared.Serialization;
using Robust.Shared.Serialization.Markdown.Mapping;
using Robust.Shared.Utility;

namespace Robust.Shared.Upload;

/// <summary>
///     Manager that allows resources to be added at runtime by admins.
///     They will be sent to all clients automatically.
/// </summary>
public abstract class SharedNetworkResourceManager : IDisposable
{
    [Dependency] private readonly INetManager _netManager = default!;
    [Dependency] private readonly IReplayRecordingManager _replay = default!;
    [Dependency] protected readonly IResourceManager ResourceManager = default!;

    public const double BytesToMegabytes = 0.000001d;

    /// <summary>
    ///     The prefix for any and all downloaded network resources.
    /// </summary>
    private static readonly ResPath Prefix = ResPath.Root / "Uploaded";

    protected readonly MemoryContentRoot ContentRoot = new();

    public bool FileExists(ResPath path)
        => ContentRoot.FileExists(path);

    public virtual void Initialize()
    {
        _netManager.RegisterNetMessage<NetworkResourceUploadMessage>(ResourceUploadMsg);

        // Add our content root to the resource manager.
        ResourceManager.AddRoot(Prefix, ContentRoot);
        _replay.RecordingStarted += OnStartReplayRecording;
    }

    private void OnStartReplayRecording(MappingDataNode metadata, List<object> events)
    {
        // replays will need information about currently loaded extra resources
        foreach (var (path, data) in ContentRoot.GetAllFiles())
        {
            events.Add(new ReplayResourceUploadMsg { RelativePath = path, Data = data });
        }
    }

    protected virtual void ResourceUploadMsg(NetworkResourceUploadMessage msg)
    {
        ContentRoot.AddOrUpdateFile(msg.RelativePath, msg.Data);
        _replay.RecordReplayMessage(new ReplayResourceUploadMsg { RelativePath = msg.RelativePath, Data = msg.Data });
    }

    public void Dispose()
    {
        // This is called automatically when the IoCManager's dependency collection is cleared.
        // MemoryContentRoot uses a ReaderWriterLockSlim, which we need to dispose of.
        ContentRoot.Dispose();
    }

    [Serializable, NetSerializable]
    public sealed class ReplayResourceUploadMsg
    {
        public byte[] Data = default!;
        public ResPath RelativePath = default!;
    }
}