85 lines
2.9 KiB
C#
85 lines
2.9 KiB
C#
using Content.Server._DV.Cargo.Components;
|
|
using Content.Server._DV.Cargo.Systems;
|
|
using Content.Server.CartridgeLoader;
|
|
using Content.Shared.CartridgeLoader;
|
|
using Content.Shared.CartridgeLoader.Cartridges;
|
|
using Content.Server._DV.Mail.Components;
|
|
using Content.Server._NF.SectorServices; // Frontier
|
|
|
|
namespace Content.Server._DV.CartridgeLoader.Cartridges;
|
|
|
|
public sealed class MailMetricsCartridgeSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly CartridgeLoaderSystem _cartridgeLoader = default!;
|
|
[Dependency] private readonly SectorServiceSystem _sectorService = default!; // Frontier
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<MailMetricsCartridgeComponent, CartridgeUiReadyEvent>(OnUiReady);
|
|
SubscribeLocalEvent<LogisticStatsUpdatedEvent>(OnLogisticsStatsUpdated);
|
|
SubscribeLocalEvent<MailComponent, MapInitEvent>(OnMapInit);
|
|
}
|
|
|
|
private void OnUiReady(Entity<MailMetricsCartridgeComponent> ent, ref CartridgeUiReadyEvent args)
|
|
{
|
|
UpdateUI(args.Loader); // Frontier: remove station as first arg
|
|
}
|
|
|
|
private void OnLogisticsStatsUpdated(LogisticStatsUpdatedEvent args)
|
|
{
|
|
UpdateAllCartridges(); // Frontier: remove station
|
|
}
|
|
|
|
private void OnMapInit(EntityUid uid, MailComponent mail, MapInitEvent args)
|
|
{
|
|
UpdateAllCartridges(); // Frontier: remove station, no owner check
|
|
}
|
|
|
|
private void UpdateAllCartridges() // Frontier: remove station
|
|
{
|
|
var query = EntityQueryEnumerator<MailMetricsCartridgeComponent, CartridgeComponent>();
|
|
while (query.MoveNext(out var uid, out var comp, out var cartridge))
|
|
{
|
|
if (cartridge.LoaderUid is not { } loader)
|
|
continue;
|
|
UpdateUI(loader);
|
|
}
|
|
}
|
|
|
|
private void UpdateUI(EntityUid loader)
|
|
{
|
|
//if (_station.GetOwningStation(loader) is { } station) // Frontier
|
|
// ent.Comp.Station = station; // Frontier
|
|
|
|
if (!TryComp<SectorLogisticStatsComponent>(_sectorService.GetServiceEntity(), out var logiStats)) // Frontier
|
|
return; // Frontier
|
|
|
|
// Get station's logistic stats
|
|
var unopenedMailCount = GetUnopenedMailCount(); // Frontier: no station arg
|
|
|
|
// Send logistic stats to cartridge client
|
|
var state = new MailMetricUiState(logiStats.Metrics, unopenedMailCount);
|
|
_cartridgeLoader.UpdateCartridgeUiState(loader, state);
|
|
}
|
|
|
|
|
|
private int GetUnopenedMailCount() // Frontier: remove EntityUid param
|
|
{
|
|
var unopenedMail = 0;
|
|
|
|
var query = EntityQueryEnumerator<MailComponent>();
|
|
|
|
while (query.MoveNext(out var _, out var comp))
|
|
{
|
|
// Frontier: remove station check, add profitable check (consider only possible profit as unopened)
|
|
if (comp.IsLocked && comp.IsProfitable)
|
|
unopenedMail++;
|
|
// End Frontier
|
|
}
|
|
|
|
return unopenedMail;
|
|
}
|
|
}
|