civ14/Content.Server/_Shitmed/Autodoc/AutodocSafetyWireAction.cs

// SPDX-FileCopyrightText: 2024 deltanedas <39013340+deltanedas@users.noreply.github.com>
// SPDX-FileCopyrightText: 2024 deltanedas <@deltanedas:kde.org>
// SPDX-FileCopyrightText: 2025 Aiden <28298836+Aidenkrz@users.noreply.github.com>
//
// SPDX-License-Identifier: AGPL-3.0-or-later

using Content.Server.Wires;
using Content.Shared._Shitmed.Autodoc.Components;
using Content.Shared._Shitmed.Autodoc.Systems;
using Content.Shared.Wires;

namespace Content.Server._Shitmed.Autodoc;

public sealed partial class AutodocSafetyWireAction : ComponentWireAction<AutodocComponent>
{
    public override Color Color { get; set; } = Color.Red;
    public override string Name { get; set; } = "wire-name-autodoc-safety";

    public override StatusLightState? GetLightState(Wire wire, AutodocComponent comp)
        => comp.RequireSleeping ? StatusLightState.On : StatusLightState.Off;

    public override object StatusKey { get; } = AutodocWireStatus.SafetyIndicator;

    public override bool Cut(EntityUid user, Wire wire, AutodocComponent comp)
    {
        var uid = wire.Owner;
        EntityManager.System<SharedAutodocSystem>().SetSafety((uid, comp), false);
        return true;
    }

    public override bool Mend(EntityUid user, Wire wire, AutodocComponent comp)
    {
        var uid = wire.Owner;
        EntityManager.System<SharedAutodocSystem>().SetSafety((uid, comp), true);
        return true;
    }

    public override void Pulse(EntityUid user, Wire wire, AutodocComponent comp)
    {
        var uid = wire.Owner;
        EntityManager.System<SharedAutodocSystem>().SetSafety((uid, comp), !comp.RequireSleeping);
    }
}