ss14-wega/Content.Client/_Wega/VeilCult/VeilCultSystem.cs

using System.Numerics;
using Content.Shared.Veil.Cult;
using Content.Shared.Veil.Cult.Components;
using Content.Shared.StatusIcon.Components;
using Robust.Client.GameObjects;
using Robust.Shared.Prototypes;

namespace Content.Client.Veil.Cult
{
    public sealed class VeilCultSystem : SharedVeilCultSystem
    {
        [Dependency] private IPrototypeManager _prototype = default!;
        [Dependency] private SpriteSystem _sprite = default!;

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

            SubscribeLocalEvent<VeilCultistComponent, GetStatusIconsEvent>(GetCultistIcons);
            SubscribeLocalEvent<VeilCogDisplayComponent, ComponentStartup>(GetHalo);
            SubscribeLocalEvent<VeilCogDisplayComponent, ComponentShutdown>(RemoveHalo);
            SubscribeLocalEvent<EnchantedComponent, ComponentRemove>(OnEnchantRemove);
        }

        private void GetCultistIcons(Entity<VeilCultistComponent> ent, ref GetStatusIconsEvent args)
        {
            var iconPrototype = _prototype.Index(ent.Comp.StatusIcon);
            args.StatusIcons.Add(iconPrototype);
        }

        private void GetHalo(EntityUid uid, VeilCogDisplayComponent component, ComponentStartup args)
        {
            if (!TryComp<SpriteComponent>(uid, out var sprite))
                return;

            if (_sprite.LayerMapTryGet(uid, CogKey.Halo, out _, false))
                return;

            var bounds = _sprite.GetLocalBounds((uid, sprite));
            var adj = bounds.Height / 2 + 1.0f / 32 * 6.0f;

            var layerData = new PrototypeLayerData
            {
                Shader = "unshaded",
                RsiPath = "_Wega/Interface/Misc/veilcult_cog.rsi",
                State = "halo",
                Offset = new Vector2(0.0f, adj)
            };

            var layer = _sprite.AddLayer(uid, layerData, null);
            _sprite.LayerMapSet(uid, CogKey.Halo, layer);
        }

        private void RemoveHalo(EntityUid uid, VeilCogDisplayComponent component, ComponentShutdown args)
        {
            if (_sprite.LayerMapTryGet(uid, CogKey.Halo, out var layer, true))
                _sprite.RemoveLayer(uid, layer);
        }

        private void OnEnchantRemove(EntityUid uid, EnchantedComponent comp, ComponentRemove args)
        {
            if (!HasComp<SpriteComponent>(uid))
                return;

            if (_sprite.LayerExists(uid, 1))
                _sprite.RemoveLayer(uid, 1);
        }

        private enum CogKey
        {
            Halo
        }
    }
}