using Content.Client.AME.Components;
using Robust.Client.GameObjects;
using static Content.Shared.AME.SharedAMEShieldComponent;

namespace Content.Client.AME;

public sealed class AMEShieldingVisualizerSystem : VisualizerSystem<AMEShieldingVisualsComponent>
{
    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<AMEShieldingVisualsComponent, ComponentInit>(OnComponentInit);
    }

    private void OnComponentInit(EntityUid uid, AMEShieldingVisualsComponent component, ComponentInit args)
    {
        if(TryComp<SpriteComponent>(uid, out var sprite))
        {
            sprite.LayerMapSet(AMEShieldingVisualsLayer.Core, sprite.AddLayerState("core"));
            sprite.LayerSetVisible(AMEShieldingVisualsLayer.Core, false);
            sprite.LayerMapSet(AMEShieldingVisualsLayer.CoreState, sprite.AddLayerState("core_weak"));
            sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, false);
        }
    }

    protected override void OnAppearanceChange(EntityUid uid, AMEShieldingVisualsComponent component, ref AppearanceChangeEvent args)
    {
        if(args.Sprite == null)
            return;

        if(args.Component.TryGetData<string>(AMEShieldVisuals.Core, out var core))
        {
            if (core == "isCore")
            {
                args.Sprite.LayerSetState(AMEShieldingVisualsLayer.Core, "core");
                args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.Core, true);

            }
            else
            {
                args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.Core, false);
            }
        }

        if(args.Component.TryGetData<string>(AMEShieldVisuals.CoreState, out var coreState))
        {
            switch(coreState)
            {
                case "weak":
                    args.Sprite.LayerSetState(AMEShieldingVisualsLayer.CoreState, "core_weak");
                    args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, true);
                    break;
                case "strong":
                    args.Sprite.LayerSetState(AMEShieldingVisualsLayer.CoreState, "core_strong");
                    args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, true);
                    break;
                case "off":
                    args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, false);
                    break;
            }
        }
    }
}

public enum AMEShieldingVisualsLayer : byte
{
    Core,
    CoreState,
}