перенос файлов сервера из папки White в _White

Content.Server/_White/Harpy/HarpysingerSystem.cs

@@ -0,0 +1,134 @@
+using Content.Server.Instruments;
+using Content.Server.Speech.Components;
+using Content.Server.UserInterface;
+using Content.Shared.ActionBlocker;
+using Content.Shared.White.Harpy;
+using Content.Shared.Inventory;
+using Content.Shared.Inventory.Events;
+using Content.Shared.Mobs;
+using Content.Shared.Popups;
+using Content.Shared.StatusEffect;
+using Content.Shared.UserInterface;
+using Content.Shared.Zombies;
+using Robust.Server.GameObjects;
+using Robust.Shared.Player;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server._White.Harpy
+{
+    public sealed class HarpySingerSystem : EntitySystem
+    {
+        [Dependency] private readonly InstrumentSystem _instrument = default!;
+        [Dependency] private readonly SharedPopupSystem _popupSystem = default!;
+        [Dependency] private readonly InventorySystem _inventorySystem = default!;
+        [Dependency] private readonly ActionBlockerSystem _blocker = default!;
+        [Dependency] private readonly IPrototypeManager _prototype = default!;
+
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<InstrumentComponent, MobStateChangedEvent>(OnMobStateChangedEvent);
+            SubscribeLocalEvent<GotEquippedEvent>(OnEquip);
+            SubscribeLocalEvent<EntityZombifiedEvent>(OnZombified);
+            //SubscribeLocalEvent<InstrumentComponent, KnockedDownEvent>(OnKnockedDown); //WD EDIT
+            //SubscribeLocalEvent<InstrumentComponent, StunnedEvent>(OnStunned); //WD EDIT
+            SubscribeLocalEvent<InstrumentComponent, SleepStateChangedEvent>(OnSleep);
+            SubscribeLocalEvent<InstrumentComponent, StatusEffectAddedEvent>(OnStatusEffect);
+            //SubscribeLocalEvent<InstrumentComponent, DamageChangedEvent>(OnDamageChanged); //WD EDIT
+
+            // This is intended to intercept the UI event and stop the MIDI UI from opening if the
+            // singer is unable to sing. Thus it needs to run before the ActivatableUISystem.
+            SubscribeLocalEvent<HarpySingerComponent, OpenUiActionEvent>(OnInstrumentOpen, before: new[] { typeof(ActivatableUISystem) });
+        }
+
+        private void OnEquip(GotEquippedEvent args)
+        {
+            // Check if an item that makes the singer mumble is equipped to their face
+            // (not their pockets!). As of writing, this should just be the muzzle.
+            if (TryComp<AddAccentClothingComponent>(args.Equipment, out var accent) &&
+                accent.ReplacementPrototype == "mumble" &&
+                args.Slot == "mask")
+            {
+                CloseMidiUi(args.Equipee);
+            }
+        }
+
+        private void OnMobStateChangedEvent(EntityUid uid, InstrumentComponent component, MobStateChangedEvent args)
+        {
+            if (args.NewMobState is MobState.Critical or MobState.Dead)
+                CloseMidiUi(args.Target);
+        }
+
+        private void OnZombified(ref EntityZombifiedEvent args)
+        {
+            CloseMidiUi(args.Target);
+        }
+
+        /*
+        private void OnKnockedDown(EntityUid uid, InstrumentComponent component, ref KnockedDownEvent args)
+        {
+            CloseMidiUi(uid);
+        }
+
+        private void OnStunned(EntityUid uid, InstrumentComponent component, ref StunnedEvent args)
+        {
+            CloseMidiUi(uid);
+        }
+        */
+
+        private void OnSleep(EntityUid uid, InstrumentComponent component, SleepStateChangedEvent args)
+        {
+            if (args.FellAsleep)
+                CloseMidiUi(uid);
+        }
+
+        private void OnStatusEffect(EntityUid uid, InstrumentComponent component, StatusEffectAddedEvent args)
+        {
+            if (args.Key == "Muted")
+                CloseMidiUi(uid);
+        }
+
+        /// <summary>
+        /// Almost a copy of Content.Server.Damage.ForceSay.DamageForceSaySystem.OnDamageChanged.
+        /// Done so because DamageForceSaySystem doesn't output an event, and my understanding is
+        /// that we don't want to change upstream code more than necessary to avoid merge conflicts
+        /// and maintenance overhead. It still reuses the values from DamageForceSayComponent, so
+        /// any tweaks to that will keep ForceSay consistent with singing interruptions.
+        /// </summary>
+
+        /// <summary>
+        /// Closes the MIDI UI if it is open.
+        /// </summary>
+        private void CloseMidiUi(EntityUid uid)
+        {
+            if (HasComp<ActiveInstrumentComponent>(uid) &&
+                TryComp<ActorComponent>(uid, out var actor))
+            {
+                _instrument.ToggleInstrumentUi(uid, actor.PlayerSession);
+            }
+        }
+
+        /// <summary>
+        /// Prevent the player from opening the MIDI UI under some circumstances.
+        /// </summary>
+        private void OnInstrumentOpen(EntityUid uid, HarpySingerComponent component, OpenUiActionEvent args)
+        {
+            // CanSpeak covers all reasons you can't talk, including being incapacitated
+            // (crit/dead), asleep, or for any reason mute inclding glimmer or a mime's vow.
+            var canNotSpeak = !_blocker.CanSpeak(uid);
+            var zombified = TryComp<ZombieComponent>(uid, out var _);
+            var muzzled = _inventorySystem.TryGetSlotEntity(uid, "mask", out var maskUid) &&
+                TryComp<AddAccentClothingComponent>(maskUid, out var accent) &&
+                accent.ReplacementPrototype == "mumble";
+
+            // Set this event as handled when the singer should be incapable of singing in order
+            // to stop the ActivatableUISystem event from opening the MIDI UI.
+            args.Handled = canNotSpeak || muzzled || zombified;
+
+            // Tell the user that they can not sing.
+            if (args.Handled)
+                _popupSystem.PopupEntity(Loc.GetString("no-sing-while-no-speak"), uid, uid, PopupType.Medium);
+        }
+    }
+}