]> git.smokeofanarchy.ru Git - space-station-14.git/commitdiff
Another batch of DoAfter fixes (#14351)
authorkeronshb <54602815+keronshb@users.noreply.github.com>
Sun, 5 Mar 2023 05:26:03 +0000 (00:26 -0500)
committerGitHub <noreply@github.com>
Sun, 5 Mar 2023 05:26:03 +0000 (21:26 -0800)
Content.Server/Kitchen/EntitySystems/SharpSystem.cs
Content.Server/Nutrition/Components/DrinkComponent.cs
Content.Server/Nutrition/Components/FoodComponent.cs
Content.Server/Nutrition/EntitySystems/DrinkSystem.cs
Content.Server/Nutrition/EntitySystems/FoodSystem.cs
Content.Server/Repairable/RepairableSystem.cs
Content.Server/Storage/EntitySystems/DumpableSystem.cs
Content.Shared/DoAfter/SharedDoAfterSystem.cs
Content.Shared/Interaction/SharedInteractionSystem.cs
Content.Shared/Radio/Components/EncryptionKeyHolderComponent.cs
Content.Shared/Radio/EntitySystems/EncryptionKeySystem.cs

index 2b53f3459396dcd8a1a92369d933fc2fe2e026a6..41426d7215a7360a69583c68a2ceada85e8b86c8 100644 (file)
@@ -77,9 +77,15 @@ public sealed class SharpSystem : EntitySystem
 
     private void OnDoAfter(EntityUid uid, SharpComponent component, DoAfterEvent args)
     {
-        if (args.Handled || args.Cancelled || !TryComp<ButcherableComponent>(args.Args.Target, out var butcher))
+        if (args.Handled || !TryComp<ButcherableComponent>(args.Args.Target, out var butcher))
             return;
 
+        if (args.Cancelled)
+        {
+            component.Butchering.Remove(args.Args.Target.Value);
+            return;
+        }
+
         component.Butchering.Remove(args.Args.Target.Value);
 
         if (_containerSystem.IsEntityInContainer(args.Args.Target.Value))
index 14c9712fa8376c7789466886ab38f21801d7ddca..60784ac4123b600b1199c0c04aff4a841e348c99 100644 (file)
@@ -41,6 +41,12 @@ namespace Content.Server.Nutrition.Components
         [DataField("forceDrink")]
         public bool ForceDrink;
 
+        /// <summary>
+        /// Is the entity currently drinking or trying to make someone else drink?
+        /// </summary>
+        [DataField("drinking")]
+        public bool Drinking;
+
         /// <summary>
         /// How long it takes to drink this yourself.
         /// </summary>
index c5c12e0fa680fb47a2de43580bf84397452df2fd..7e9effdfda6d20f6b896242ca59df0838bcb6aac 100644 (file)
@@ -42,11 +42,16 @@ namespace Content.Server.Nutrition.Components
 
         /// <summary>
         /// Is this entity being forcefed?
-        /// Prevents the entity from being forced to eat multiple times if not self
         /// </summary>
         [DataField("forceFeed")]
         public bool ForceFeed;
 
+        /// <summary>
+        /// Is this entity eating or being fed?
+        /// </summary>
+        [DataField(("eating"))]
+        public bool Eating;
+
         /// <summary>
         /// How long it takes to eat the food personally.
         /// </summary>
index 7cbd2b302343124cc5fbdfe4a1ae4db7bddc00b7..d919f926cefdb46fd083c3a05d11834f5dae8b82 100644 (file)
@@ -217,7 +217,7 @@ namespace Content.Server.Nutrition.EntitySystems
 
         private bool TryDrink(EntityUid user, EntityUid target, DrinkComponent drink, EntityUid item)
         {
-            if (!EntityManager.HasComponent<BodyComponent>(target) || drink.ForceDrink)
+            if (!EntityManager.HasComponent<BodyComponent>(target) || drink.Drinking)
                 return false;
 
             if (!drink.Opened)
@@ -241,6 +241,7 @@ namespace Content.Server.Nutrition.EntitySystems
             if (!_interactionSystem.InRangeUnobstructed(user, item, popup: true))
                 return true;
 
+            drink.Drinking = true;
             drink.ForceDrink = user != target;
 
             if (drink.ForceDrink)
@@ -286,11 +287,10 @@ namespace Content.Server.Nutrition.EntitySystems
         /// </summary>
         private void OnDoAfter(EntityUid uid, DrinkComponent component, DoAfterEvent<DrinkData> args)
         {
-            //Special cancel if they're force feeding someone.
-            //Allows self to drink multiple times but prevents force feeding drinks to others rapidly.
-            if (args.Cancelled && component.ForceDrink)
+            if (args.Cancelled)
             {
                 component.ForceDrink = false;
+                component.Drinking = false;
                 return;
             }
 
@@ -374,6 +374,7 @@ namespace Content.Server.Nutrition.EntitySystems
             //TODO: Grab the stomach UIDs somehow without using Owner
             _stomachSystem.TryTransferSolution(firstStomach.Value.Comp.Owner, drained, firstStomach.Value.Comp);
 
+            component.Drinking = false;
             component.ForceDrink = false;
             args.Handled = true;
         }
index bd31a9793c26e47c660738fd5e58e4651e6074c4..e0e2a5f0f4b84845849515a08587ab58a467127b 100644 (file)
@@ -86,8 +86,8 @@ namespace Content.Server.Nutrition.EntitySystems
             if (food == user || EntityManager.TryGetComponent<MobStateComponent>(food, out var mobState) && _mobStateSystem.IsAlive(food, mobState)) // Suppresses eating alive mobs
                 return false;
 
-            // Target can't be fed or they're already forcefeeding
-            if (!EntityManager.HasComponent<BodyComponent>(target) || foodComp.ForceFeed)
+            // Target can't be fed or they're already eating
+            if (!EntityManager.HasComponent<BodyComponent>(target) || foodComp.Eating)
                 return false;
 
             if (!_solutionContainerSystem.TryGetSolution(food, foodComp.SolutionName, out var foodSolution))
@@ -111,6 +111,7 @@ namespace Content.Server.Nutrition.EntitySystems
             if (!_interactionSystem.InRangeUnobstructed(user, food, popup: true))
                 return true;
 
+            foodComp.Eating = true;
             foodComp.ForceFeed = user != target;
 
             if (foodComp.ForceFeed)
@@ -152,8 +153,9 @@ namespace Content.Server.Nutrition.EntitySystems
         private void OnDoAfter(EntityUid uid, FoodComponent component, DoAfterEvent<FoodData> args)
         {
             //Prevents the target from being force fed food but allows the user to chow down
-            if (args.Cancelled && component.ForceFeed)
+            if (args.Cancelled)
             {
+                component.Eating = false;
                 component.ForceFeed = false;
                 return;
             }
@@ -167,6 +169,8 @@ namespace Content.Server.Nutrition.EntitySystems
             if (!_bodySystem.TryGetBodyOrganComponents<StomachComponent>(args.Args.Target.Value, out var stomachs, body))
                 return;
 
+            component.Eating = false;
+
             var transferAmount = component.TransferAmount != null ? FixedPoint2.Min((FixedPoint2) component.TransferAmount, args.AdditionalData.FoodSolution.Volume) : args.AdditionalData.FoodSolution.Volume;
 
             var split = _solutionContainerSystem.SplitSolution(uid, args.AdditionalData.FoodSolution, transferAmount);
index 6013d2660120ed681e04b95520904747be68a604..d9d53686d3c6d64aff5799d8e4d278e9dc6dd287 100644 (file)
@@ -1,5 +1,4 @@
 using Content.Server.Administration.Logs;
-using Content.Server.Tools;
 using Content.Shared.Damage;
 using Content.Shared.Database;
 using Content.Shared.Interaction;
@@ -18,24 +17,12 @@ namespace Content.Server.Repairable
         public override void Initialize()
         {
             SubscribeLocalEvent<RepairableComponent, InteractUsingEvent>(Repair);
+            SubscribeLocalEvent<RepairableComponent, RepairFinishedEvent>(OnRepairFinished);
         }
 
-        public async void Repair(EntityUid uid, RepairableComponent component, InteractUsingEvent args)
+        private void OnRepairFinished(EntityUid uid, RepairableComponent component, RepairFinishedEvent args)
         {
-            // Only try repair the target if it is damaged
-            if (!EntityManager.TryGetComponent(component.Owner, out DamageableComponent? damageable) || damageable.TotalDamage == 0)
-                return;
-
-            float delay = component.DoAfterDelay;
-
-            // Add a penalty to how long it takes if the user is repairing itself
-            if (args.User == args.Target)
-                delay *= component.SelfRepairPenalty;
-
-            var toolEvData = new ToolEventData(null);
-
-            // Can the tool actually repair this, does it have enough fuel?
-            if (!_toolSystem.UseTool(args.Used, args.User, uid, delay, component.QualityNeeded, toolEvData, component.FuelCost))
+            if (!EntityManager.TryGetComponent(uid, out DamageableComponent? damageable) || damageable.TotalDamage == 0)
                 return;
 
             if (component.Damage != null)
@@ -43,6 +30,7 @@ namespace Content.Server.Repairable
                 var damageChanged = _damageableSystem.TryChangeDamage(uid, component.Damage, true, false, origin: args.User);
                 _adminLogger.Add(LogType.Healed, $"{ToPrettyString(args.User):user} repaired {ToPrettyString(uid):target} by {damageChanged?.Total}");
             }
+
             else
             {
                 // Repair all damage
@@ -51,12 +39,43 @@ namespace Content.Server.Repairable
             }
 
 
-            component.Owner.PopupMessage(args.User,
+            uid.PopupMessage(args.User,
                 Loc.GetString("comp-repairable-repair",
-                    ("target", component.Owner),
+                    ("target", uid),
                     ("tool", args.Used)));
+        }
+
+        public async void Repair(EntityUid uid, RepairableComponent component, InteractUsingEvent args)
+        {
+            // Only try repair the target if it is damaged
+            if (!EntityManager.TryGetComponent(uid, out DamageableComponent? damageable) || damageable.TotalDamage == 0)
+                return;
+
+            float delay = component.DoAfterDelay;
+
+            // Add a penalty to how long it takes if the user is repairing itself
+            if (args.User == args.Target)
+                delay *= component.SelfRepairPenalty;
+
+            var toolEvData = new ToolEventData(new RepairFinishedEvent(args.User, args.Used), component.FuelCost, targetEntity:uid);
+
+            // Can the tool actually repair this, does it have enough fuel?
+            if (!_toolSystem.UseTool(args.Used, args.User, uid, delay, component.QualityNeeded, toolEvData, component.FuelCost))
+                return;
 
             args.Handled = true;
         }
     }
+
+    public sealed class RepairFinishedEvent : EntityEventArgs
+    {
+        public EntityUid User;
+        public EntityUid Used;
+
+        public RepairFinishedEvent(EntityUid user, EntityUid used)
+        {
+            User = user;
+            Used = used;
+        }
+    }
 }
index c0e1c22a1b73567cc9dc5fc893bbd9d04ffca115..a59e439bec2c4609ca6f411dacc69415968feea8 100644 (file)
@@ -106,6 +106,7 @@ namespace Content.Server.Storage.EntitySystems
 
             _doAfterSystem.DoAfter(new DoAfterEventArgs(userUid, delay, target: targetUid, used: storageUid)
             {
+                RaiseOnTarget = false,
                 BreakOnTargetMove = true,
                 BreakOnUserMove = true,
                 BreakOnStun = true,
index 18c92cc7000f7e2aacf4332ca9515da57e1d3262..3f73e27c02e79be34cdc0b80ef2ec5eac8b4138e 100644 (file)
@@ -103,6 +103,10 @@ public abstract class SharedDoAfterSystem : EntitySystem
 
         foreach (var (_, comp) in EntityManager.EntityQuery<ActiveDoAfterComponent, DoAfterComponent>())
         {
+            //Don't run the doafter if its comp or owner is deleted.
+            if (EntityManager.Deleted(comp.Owner) || comp.Deleted)
+                continue;
+
             foreach (var doAfter in comp.DoAfters.Values.ToArray())
             {
                 Run(comp.Owner, comp, doAfter);
index 64f0a947db5e8a910949eb8bb2edf98390d34549..a2396ce2c7d1ed3ac9581158bbd85b4d974d6b60 100644 (file)
@@ -769,6 +769,7 @@ namespace Content.Shared.Interaction
                 return;
 
             // all interactions should only happen when in range / unobstructed, so no range check is needed
+            //TODO: See why this is firing off multiple times
             var interactUsingEvent = new InteractUsingEvent(user, used, target, clickLocation);
             RaiseLocalEvent(target, interactUsingEvent, true);
             DoContactInteraction(user, used, interactUsingEvent);
index ca5d1f647aa5a4eba5b792bbdc105b21fa0fb2fc..c5cf05ea811b6129fd726bb1ce76c64f2a6a8837 100644 (file)
@@ -42,6 +42,12 @@ public sealed class EncryptionKeyHolderComponent : Component
     public Container KeyContainer = default!;
     public const string KeyContainerName = "key_slots";
 
+    /// <summary>
+    /// Blocks multiple attempts to remove the key
+    /// </summary>
+    [DataField("removing")]
+    public bool Removing;
+
     /// <summary>
     ///     Combined set of radio channels provided by all contained keys.
     /// </summary>
index 772255eab85b4a6cd6dd5c43b2fd55d4b4cdcb55..e3c60b99f97bc503645b3fc5e623e4cf5f32d6d7 100644 (file)
@@ -37,6 +37,28 @@ public sealed class EncryptionKeySystem : EntitySystem
         SubscribeLocalEvent<EncryptionKeyHolderComponent, InteractUsingEvent>(OnInteractUsing);
         SubscribeLocalEvent<EncryptionKeyHolderComponent, EntInsertedIntoContainerMessage>(OnContainerModified);
         SubscribeLocalEvent<EncryptionKeyHolderComponent, EntRemovedFromContainerMessage>(OnContainerModified);
+        SubscribeLocalEvent<EncryptionKeyHolderComponent, EncryptionRemovalFinishedEvent>(OnKeyRemoval);
+        SubscribeLocalEvent<EncryptionKeyHolderComponent, EncryptionRemovalCancelledEvent>(OnKeyCancelled);
+    }
+
+    private void OnKeyCancelled(EntityUid uid, EncryptionKeyHolderComponent component, EncryptionRemovalCancelledEvent args)
+    {
+        component.Removing = false;
+    }
+
+    private void OnKeyRemoval(EntityUid uid, EncryptionKeyHolderComponent component, EncryptionRemovalFinishedEvent args)
+    {
+        var contained = component.KeyContainer.ContainedEntities.ToArray();
+        _container.EmptyContainer(component.KeyContainer, entMan: EntityManager);
+        foreach (var ent in contained)
+        {
+            _hands.PickupOrDrop(args.User, ent);
+        }
+
+        // if tool use ever gets predicted this needs changing.
+        _popupSystem.PopupEntity(Loc.GetString("headset-encryption-keys-all-extracted"), uid, args.User);
+        _audio.PlayPvs(component.KeyExtractionSound, uid);
+        component.Removing = false;
     }
 
     public void UpdateChannels(EntityUid uid, EncryptionKeyHolderComponent component)
@@ -67,7 +89,7 @@ public sealed class EncryptionKeySystem : EntitySystem
 
     private void OnInteractUsing(EntityUid uid, EncryptionKeyHolderComponent component, InteractUsingEvent args)
     {
-        if (!TryComp<ContainerManagerComponent>(uid, out var storage))
+        if (!TryComp<ContainerManagerComponent>(uid, out var storage) || args.Handled || component.Removing)
             return;
 
         if (TryComp<EncryptionKeyComponent>(args.Used, out var key))
@@ -99,7 +121,7 @@ public sealed class EncryptionKeySystem : EntitySystem
 
         if (!TryComp<ToolComponent>(args.Used, out var tool) || !tool.Qualities.Contains(component.KeysExtractionMethod))
             return;
-
+        
         args.Handled = true;
 
         if (component.KeyContainer.ContainedEntities.Count == 0)
@@ -109,21 +131,13 @@ public sealed class EncryptionKeySystem : EntitySystem
             return;
         }
 
-        var toolEvData = new ToolEventData(null);
+        //This is honestly the poor mans fix because the InteractUsingEvent fires off 12 times
+        component.Removing = true;
 
-        if(!_toolSystem.UseTool(args.Used, args.User, uid, 0f, new[] { component.KeysExtractionMethod }, toolEvData, toolComponent: tool))
-            return;
+        var toolEvData = new ToolEventData(new EncryptionRemovalFinishedEvent(args.User), cancelledEv: new EncryptionRemovalCancelledEvent(), targetEntity: uid);
 
-        var contained = component.KeyContainer.ContainedEntities.ToArray();
-        _container.EmptyContainer(component.KeyContainer, entMan: EntityManager);
-        foreach (var ent in contained)
-        {
-            _hands.PickupOrDrop(args.User, ent);
-        }
-
-        // if tool use ever gets predicted this needs changing.
-        _popupSystem.PopupEntity(Loc.GetString("headset-encryption-keys-all-extracted"), uid, args.User);
-        _audio.PlayPvs(component.KeyExtractionSound, args.Target);
+        if(!_toolSystem.UseTool(args.Used, args.User, uid, 1f, new[] { component.KeysExtractionMethod }, toolEvData, toolComponent: tool))
+            return;
     }
 
     private void OnStartup(EntityUid uid, EncryptionKeyHolderComponent component, ComponentStartup args)
@@ -195,4 +209,19 @@ public sealed class EncryptionKeySystem : EntitySystem
             examineEvent.PushMarkup(msg);
         }
     }
+
+    public sealed class EncryptionRemovalFinishedEvent : EntityEventArgs
+    {
+        public EntityUid User;
+
+        public EncryptionRemovalFinishedEvent(EntityUid user)
+        {
+            User = user;
+        }
+    }
+
+    public sealed class EncryptionRemovalCancelledEvent : EntityEventArgs
+    {
+
+    }
 }