tbd-station-14/Content.Client/Weapons/Melee/Components/MeleeLungeComponent.cs

using Robust.Client.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Maths;

namespace Content.Client.Weapons.Melee.Components
{
    [RegisterComponent]
    public sealed class MeleeLungeComponent : Component
    {
        private const float ResetTime = 0.3f;
        private const float BaseOffset = 0.25f;

        private Angle _angle;
        private float _time;

        public void SetData(Angle angle)
        {
            _angle = angle;
            _time = 0;
        }

        public void Update(float frameTime)
        {
            _time += frameTime;

            var offset = Vector2.Zero;
            var deleteSelf = false;

            if (_time > ResetTime)
            {
                deleteSelf = true;
            }
            else
            {
                offset = _angle.RotateVec((0, -BaseOffset));
                offset *= (ResetTime - _time) / ResetTime;
            }

            var entMan = IoCManager.Resolve<IEntityManager>();

            if (entMan.TryGetComponent(Owner, out ISpriteComponent? spriteComponent))
            {
                spriteComponent.Offset = offset;
            }

            if (deleteSelf)
            {
                entMan.RemoveComponent<MeleeLungeComponent>(Owner);
            }
        }
    }
}