tbd-station-14/Content.Client/GameObjects/EntitySystems/MeleeWeaponSystem.cs

using System.Linq;
using Content.Client.GameObjects.Components.Mobs;
using Content.Client.GameObjects.Components.Weapons.Melee;
using Content.Shared.GameObjects.Components.Weapons.Melee;
using JetBrains.Annotations;
using Robust.Client.Interfaces.GameObjects.Components;
using Robust.Shared.GameObjects;
using Robust.Shared.GameObjects.Systems;
using Robust.Shared.Interfaces.Network;
using Robust.Shared.IoC;
using Robust.Shared.Log;
using Robust.Shared.Maths;
using Robust.Shared.Prototypes;
using Robust.Shared.Timers;
using static Content.Shared.GameObjects.EntitySystemMessages.MeleeWeaponSystemMessages;

namespace Content.Client.GameObjects.EntitySystems
{
    [UsedImplicitly]
    public sealed class MeleeWeaponSystem : EntitySystem
    {
#pragma warning disable 649
        [Dependency] private readonly IPrototypeManager _prototypeManager;
#pragma warning restore 649

        public override void Initialize()
        {
            base.Initialize();

            EntityQuery = new TypeEntityQuery(typeof(MeleeWeaponArcAnimationComponent));
        }

        public override void RegisterMessageTypes()
        {
            base.RegisterMessageTypes();

            RegisterMessageType<PlayMeleeWeaponAnimationMessage>();
        }

        public override void HandleNetMessage(INetChannel channel, EntitySystemMessage message)
        {
            base.HandleNetMessage(channel, message);

            switch (message)
            {
                case PlayMeleeWeaponAnimationMessage playMsg:
                    PlayWeaponArc(playMsg);
                    break;
            }
        }

        public override void FrameUpdate(float frameTime)
        {
            base.FrameUpdate(frameTime);

            foreach (var entity in RelevantEntities)
            {
                entity.GetComponent<MeleeWeaponArcAnimationComponent>().Update(frameTime);
            }
        }

        private void PlayWeaponArc(PlayMeleeWeaponAnimationMessage msg)
        {
            if (!_prototypeManager.TryIndex(msg.ArcPrototype, out MeleeWeaponAnimationPrototype weaponArc))
            {
                Logger.Error("Tried to play unknown weapon arc prototype '{0}'", msg.ArcPrototype);
                return;
            }

            var attacker = EntityManager.GetEntity(msg.Attacker);

            var lunge = attacker.EnsureComponent<MeleeLungeComponent>();
            lunge.SetData(msg.Angle);

            var entity = EntityManager.SpawnEntityAt("WeaponArc", attacker.Transform.GridPosition);
            entity.Transform.LocalRotation = msg.Angle;

            var weaponArcAnimation = entity.GetComponent<MeleeWeaponArcAnimationComponent>();
            weaponArcAnimation.SetData(weaponArc, msg.Angle);


            foreach (var hitEntity in msg.Hits.Select(u => EntityManager.GetEntity(u)))
            {
                if (!hitEntity.TryGetComponent(out ISpriteComponent sprite)) continue;

                var originalColor = sprite.Color;
                var newColor = Color.Red * originalColor;
                sprite.Color = newColor;

                Timer.Spawn(100, () =>
                {
                    // Only reset back to the original color if something else didn't change the color in the mean time.
                    if (sprite.Color == newColor)
                    {
                        sprite.Color = originalColor;
                    }
                });
            }
        }
    }
}