crystall-punk-14/Content.Server/_CP14/RoundStatistic/CP14RoundStatTrackerSystem.cs

using System.Text;
using Content.Server.GameTicking;
using Content.Shared._CP14.RoundStatistic;
using Content.Shared.GameTicking;
using Robust.Shared.Prototypes;

namespace Content.Server._CP14.RoundStatistic;

public sealed partial class CP14RoundStatTrackerSystem : EntitySystem
{
    [Dependency] private readonly IPrototypeManager _proto = default!;

    private readonly Dictionary<ProtoId<CP14RoundStatTrackerPrototype>, int> _tracking = new();

    public override void Initialize()
    {
        base.Initialize();
        InitializeDemiplaneDeath();

        SubscribeLocalEvent<RoundRestartCleanupEvent>(OnRoundReset);
        SubscribeLocalEvent<RoundEndTextAppendEvent>(OnRoundEndTextAppend);
        ClearStatistic();
    }

    private void OnRoundReset(RoundRestartCleanupEvent ev)
    {
        ClearStatistic();
    }

    private void OnRoundEndTextAppend(RoundEndTextAppendEvent ev)
    {
        //TODO: Move to separate UI Text block
        var sb = new StringBuilder();

        sb.Append($"[head=3]{Loc.GetString("cp14-tracker-header")}[/head] \n");
        foreach (var pair in _tracking)
        {
            if (!_proto.TryIndex(pair.Key, out var indexedTracker))
                continue;

            sb.Append($"- {Loc.GetString(indexedTracker.Text)}: {pair.Value}\n");
        }
        ev.AddLine(sb.ToString());
    }

    private void ClearStatistic()
    {
        _tracking.Clear();

        foreach (var statTracker in _proto.EnumeratePrototypes<CP14RoundStatTrackerPrototype>())
        {
            _tracking.Add(statTracker.ID, 0);
        }
    }

    public void TrackAdd(ProtoId<CP14RoundStatTrackerPrototype> proto, int dif)
    {
        _tracking[proto] += Math.Max(dif, 0);
    }

    public int? GetTrack(ProtoId<CP14RoundStatTrackerPrototype> proto)
    {
        if (!_tracking.TryGetValue(proto, out var stat))
        {
            Log.Error($"Failed to get round statistic: {proto}");
            return null;
        }

        return stat;
    }
}