CatStation/Content.Shared/Singularity/EntitySystems/SharedSingularityGeneratorSystem.cs

using Content.Shared.Emag.Systems;
using Content.Shared.Popups;
using Content.Shared.Singularity.Components;

namespace Content.Shared.Singularity.EntitySystems;

/// <summary>
/// Shared part of SingularitySingularityGeneratorSystem
/// </summary>
public abstract partial class SharedSingularityGeneratorSystem : EntitySystem
{
    #region Dependencies
    [Dependency] protected SharedPopupSystem PopupSystem = default!;
    [Dependency] private EmagSystem _emag = default!;
    #endregion Dependencies

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

        SubscribeLocalEvent<SingularityGeneratorComponent, GotEmaggedEvent>(OnEmagged);
    }

    private void OnEmagged(EntityUid uid, SingularityGeneratorComponent component, ref GotEmaggedEvent args)
    {
        if (!_emag.CompareFlag(args.Type, EmagType.Interaction))
            return;

        if (_emag.CheckFlag(uid, EmagType.Interaction))
            return;

        if (component.FailsafeDisabled)
            return;

        component.FailsafeDisabled = true;
        args.Handled = true;
    }
}