Predict gas valves (#33836)

* Predict gas valves * wawawewa * Fix imports before I get yelled at * soff
2025-05-15 04:06:37 +10:00
parent 9a6dbb27cd
commit c09aa6039c
7 changed files with 137 additions and 104 deletions
--- a/Content.Shared/Atmos/Piping/Binary/Systems/SharedGasValveSystem.cs
+++ b/Content.Shared/Atmos/Piping/Binary/Systems/SharedGasValveSystem.cs
@@ -0,0 +1,68 @@
+using Content.Shared.Atmos.Piping.Binary.Components;
+using Content.Shared.Examine;
+using Content.Shared.Interaction;
+using Robust.Shared.Audio;
+using Robust.Shared.Audio.Systems;
+
+namespace Content.Shared.Atmos.Piping.Binary.Systems;
+
+public abstract class SharedGasValveSystem : EntitySystem
+{
+    [Dependency] private readonly SharedAppearanceSystem _appearance = default!;
+    [Dependency] private readonly SharedAudioSystem _audio = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<GasValveComponent, ComponentStartup>(OnStartup);
+        SubscribeLocalEvent<GasValveComponent, ActivateInWorldEvent>(OnActivate);
+        SubscribeLocalEvent<GasValveComponent, ExaminedEvent>(OnExamined);
+    }
+
+    private void OnStartup(Entity<GasValveComponent> ent, ref ComponentStartup args)
+    {
+        // We call set in startup so it sets the appearance, node state, etc.
+        Set(ent.Owner, ent.Comp, ent.Comp.Open);
+    }
+
+    public virtual void Set(EntityUid uid, GasValveComponent component, bool value)
+    {
+        component.Open = value;
+        Dirty(uid, component);
+
+        if (TryComp<AppearanceComponent>(uid, out var appearance))
+        {
+            _appearance.SetData(uid, FilterVisuals.Enabled, component.Open, appearance);
+        }
+    }
+
+    public void Toggle(EntityUid uid, GasValveComponent component)
+    {
+        Set(uid, component, !component.Open);
+    }
+
+    private void OnActivate(Entity<GasValveComponent> ent, ref ActivateInWorldEvent args)
+    {
+        if (args.Handled || !args.Complex)
+            return;
+
+        Toggle(ent.Owner, ent.Comp);
+        _audio.PlayPredicted(ent.Comp.ValveSound, ent.Owner, args.User, AudioParams.Default.WithVariation(0.25f));
+        args.Handled = true;
+    }
+
+    private void OnExamined(Entity<GasValveComponent> ent, ref ExaminedEvent args)
+    {
+        var valve = ent.Comp;
+        if (!Transform(ent).Anchored)
+            return;
+
+        if (Loc.TryGetString("gas-valve-system-examined", out var str,
+                ("statusColor", valve.Open ? "green" : "orange"),
+                ("open", valve.Open)))
+        {
+            args.PushMarkup(str);
+        }
+    }
+}