Adds Damage on tool interaction component (#1060) (#1102)

Fixed issue with Welding Tool ignoring the tools list.
2020-07-03 22:27:52 +01:00
parent 77368886e9
commit 68a515ee09
2 changed files with 72 additions and 0 deletions
--- a/Content.Server/GameObjects/Components/Damage/DamageOnToolInteractComponent.cs
+++ b/Content.Server/GameObjects/Components/Damage/DamageOnToolInteractComponent.cs
@@ -0,0 +1,67 @@
+using Content.Server.GameObjects.Components.Interactable;
+using Content.Server.GameObjects.EntitySystems;
+using Content.Shared.GameObjects.Components.Interactable;
+using Robust.Shared.GameObjects;
+using Robust.Shared.Serialization;
+using System.Collections.Generic;
+
+namespace Content.Server.GameObjects.Components.Damage
+{
+    [RegisterComponent]
+    class DamageOnToolInteractComponent : Component, IInteractUsing
+    {
+        public override string Name => "DamageOnToolInteract";
+
+        /* Set in YAML */
+        protected int Damage;
+        private List<ToolQuality> _tools = new List<ToolQuality>();
+
+        public override void ExposeData(ObjectSerializer serializer)
+        {
+            base.ExposeData(serializer);
+
+            serializer.DataField(ref Damage, "damage", 0);
+
+            serializer.DataField(ref _tools, "tools", new List<ToolQuality>());
+        }
+
+        public override void Initialize()
+        {
+            base.Initialize();
+            Owner.EnsureComponent<DamageableComponent>();
+        }
+
+        public bool InteractUsing(InteractUsingEventArgs eventArgs)
+        {
+            if (eventArgs.Using.TryGetComponent<ToolComponent>(out var tool))
+            {
+                foreach (var toolQuality in _tools)
+                {
+                    if (tool.HasQuality(ToolQuality.Welding) && toolQuality == ToolQuality.Welding)
+                    {
+                    if (eventArgs.Using.TryGetComponent<WelderComponent>(out WelderComponent welder))
+                    {
+                            if (welder.WelderLit) return CallDamage(eventArgs, tool);
+                    }
+                        break; //If the tool quality is welding and its not lit or its not actually a welder that can be lit then its pointless to continue. 
+                }
+
+                    if (tool.HasQuality(toolQuality)) return CallDamage(eventArgs, tool);
+                }
+            }
+            return false;
+        }
+
+        protected bool CallDamage(InteractUsingEventArgs eventArgs, ToolComponent tool)
+        {
+            if (eventArgs.Target.TryGetComponent<DamageableComponent>(out var damageable))
+            {
+                if(tool.HasQuality(ToolQuality.Welding)) damageable.TakeDamage(Shared.GameObjects.DamageType.Heat, Damage, eventArgs.Using, eventArgs.User);
+                else
+                damageable.TakeDamage(Shared.GameObjects.DamageType.Brute, Damage, eventArgs.Using, eventArgs.User);
+                return true;
+            }
+                return false;
+        }
+    }
+}