Makes tools and welders ECS, add ToolQualityPrototype. (#4741)

Content.Server/Tools/ToolSystem.MultipleTool.cs

@@ -0,0 +1,108 @@
+using System;
+using System.Linq;
+using Content.Server.Tools.Components;
+using Content.Shared.Interaction;
+using Content.Shared.Tools;
+using Content.Shared.Tools.Components;
+using Robust.Server.GameObjects;
+using Robust.Shared.Audio;
+using Robust.Shared.GameObjects;
+using Robust.Shared.GameStates;
+using Robust.Shared.IoC;
+using Robust.Shared.Localization;
+using Robust.Shared.Player;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.Tools
+{
+    public partial class ToolSystem
+    {
+        private void InitializeMultipleTools()
+        {
+            SubscribeLocalEvent<MultipleToolComponent, ComponentStartup>(OnMultipleToolStartup);
+            SubscribeLocalEvent<MultipleToolComponent, UseInHandEvent>(OnMultipleToolUsedInHand);
+            SubscribeLocalEvent<MultipleToolComponent, ActivateInWorldEvent>(OnMultipleToolActivated);
+            SubscribeLocalEvent<MultipleToolComponent, ComponentGetState>(OnMultipleToolGetState);
+        }
+
+        private void OnMultipleToolStartup(EntityUid uid, MultipleToolComponent multiple, ComponentStartup args)
+        {
+            // Only set the multiple tool if we have a tool component.
+            if(EntityManager.TryGetComponent(uid, out ToolComponent? tool))
+                SetMultipleTool(uid, multiple, tool);
+        }
+
+        private void OnMultipleToolUsedInHand(EntityUid uid, MultipleToolComponent multiple, UseInHandEvent args)
+        {
+            if (args.Handled)
+                return;
+
+            args.Handled = CycleMultipleTool(uid, multiple);
+        }
+
+        private void OnMultipleToolActivated(EntityUid uid, MultipleToolComponent multiple, ActivateInWorldEvent args)
+        {
+            if (args.Handled)
+                return;
+
+            args.Handled = CycleMultipleTool(uid, multiple);
+        }
+
+        private void OnMultipleToolGetState(EntityUid uid, MultipleToolComponent multiple, ref ComponentGetState args)
+        {
+            args.State = new MultipleToolComponentState(multiple.CurrentQualityName);
+        }
+
+        public bool CycleMultipleTool(EntityUid uid, MultipleToolComponent? multiple = null)
+        {
+            if (!Resolve(uid, ref multiple))
+                return false;
+
+            if (multiple.Entries.Length == 0)
+                return false;
+
+            multiple.CurrentEntry = (multiple.CurrentEntry + 1) % multiple.Entries.Length;
+            SetMultipleTool(uid, multiple);
+
+            var current = multiple.Entries[multiple.CurrentEntry];
+
+            if(current.ChangeSound is {} changeSound)
+                SoundSystem.Play(Filter.Pvs(uid), changeSound.GetSound(), uid);
+
+            return true;
+        }
+
+        public void SetMultipleTool(EntityUid uid, MultipleToolComponent? multiple = null, ToolComponent? tool = null, SpriteComponent? sprite = null)
+        {
+            if (!Resolve(uid, ref multiple, ref tool))
+                return;
+
+            // Sprite is optional.
+            Resolve(uid, ref sprite);
+
+            if (multiple.Entries.Length == 0)
+            {
+                multiple.CurrentQualityName = Loc.GetString("multiple-tool-component-no-behavior");
+                multiple.Dirty();
+                return;
+            }
+
+            var current = multiple.Entries[multiple.CurrentEntry];
+
+            tool.UseSound = current.Sound;
+            tool.Qualities = current.Behavior;
+
+            if (_prototypeManager.TryIndex(current.Behavior.First(), out ToolQualityPrototype? quality))
+            {
+                multiple.CurrentQualityName = Loc.GetString(quality.Name);
+            }
+
+            multiple.Dirty();
+
+            if (current.Sprite == null || sprite == null)
+                return;
+
+            sprite.LayerSetSprite(0, current.Sprite);
+        }
+    }
+}