CatStation/Content.Server/Objectives/Systems/HijackTradeStationConditionSystem.cs

using Content.Server.Objectives.Components;
using Content.Shared.Cargo.Components;
using Content.Shared.Mind;
using Content.Shared.Objectives.Components;
using NetCord;

namespace Content.Server.Objectives.Systems;

/// <summary>
///     Handles the Hijack Trade Station objective.
/// </summary>
public sealed class HijackTradeStationConditionSystem : EntitySystem
{
    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<HijackTradeStationConditionComponent, ObjectiveGetProgressEvent>(OnGetProgress);
    }

    private void OnGetProgress(Entity<HijackTradeStationConditionComponent> ent, ref ObjectiveGetProgressEvent args)
    {
        var enumerator = EntityQueryEnumerator<TradeStationComponent>();
        args.Progress = 0f;
        // If there's any hacked trade station, succeed.
        while (enumerator.MoveNext(out var comp))
        {
            if (!comp.Hacked)
                continue;

            args.Progress = 1f;
            return;
        }
    }
}