Stunbaton now has a 3/4 chance of applying slowdown instead of paralyze

Content.Server/GameObjects/Components/Weapon/Melee/StunbatonComponent.cs

@@ -1,32 +1,75 @@
 using System.Collections.Generic;
 using Content.Server.GameObjects.Components.Mobs;
+using Content.Server.GameObjects.EntitySystems;
+using Robust.Server.GameObjects;
 using Robust.Shared.GameObjects;
 using Robust.Shared.Interfaces.GameObjects;
+using Robust.Shared.Interfaces.Random;
+using Robust.Shared.IoC;
+using Robust.Shared.Random;
 using Robust.Shared.Serialization;
 
 namespace Content.Server.GameObjects.Components.Weapon.Melee
 {
     [RegisterComponent]
-    public class StunbatonComponent : MeleeWeaponComponent
+    public class StunbatonComponent : MeleeWeaponComponent, IUse
     {
+        [Dependency] private IRobustRandom _robustRandom;
+
         public override string Name => "Stunbaton";
 
         private float _paralyzeTime = 10f;
+        private float _paralyzeChance = 0.25f;
+        private float _slowdownTime = 5f;
+
+        private bool _activated = false;
+
+        public bool Activated => _activated;
 
         public override void ExposeData(ObjectSerializer serializer)
         {
             base.ExposeData(serializer);
 
+            serializer.DataField(ref _paralyzeChance, "paralyzeChance", 0.25f);
             serializer.DataField(ref _paralyzeTime, "paralyzeTime", 10f);
+            serializer.DataField(ref _slowdownTime, "slowdownTime", 5f);
         }
 
         public override void OnHitEntities(IEnumerable<IEntity> entities)
         {
+            if (!_activated)
+                return;
+
             foreach (var entity in entities)
             {
-                if(entity.TryGetComponent(out StunnableComponent stunnable))
+                if (!entity.TryGetComponent(out StunnableComponent stunnable)) continue;
+
+                if(_robustRandom.Prob(_paralyzeChance))
                     stunnable.Paralyze(_paralyzeTime);
+                else
+                    stunnable.Slowdown(_slowdownTime);
             }
         }
+
+        public bool UseEntity(UseEntityEventArgs eventArgs)
+        {
+            var sprite = Owner.GetComponent<SpriteComponent>();
+            var item = Owner.GetComponent<ItemComponent>();
+
+            if (_activated)
+            {
+                item.EquippedPrefix = "off";
+                sprite.LayerSetState(0, "stunbaton_off");
+                _activated = false;
+            }
+            else
+            {
+                item.EquippedPrefix = "on";
+                sprite.LayerSetState(0, "stunbaton_on");
+                _activated = true;
+            }
+
+            return true;
+        }
     }
 }