using System;
using Content.Shared.Hands.Components;
using JetBrains.Annotations;
using Robust.Client.GameObjects;
using Robust.Client.ResourceManagement;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;

namespace Content.Client.Hands
{
    [UsedImplicitly]
    public sealed class HandsVisualizer : AppearanceVisualizer
    {
        public override void OnChangeData(AppearanceComponent component)
        {
            base.OnChangeData(component);

            var entities = IoCManager.Resolve<IEntityManager>();
            if (!entities.TryGetComponent<ISpriteComponent>(component.Owner, out var sprite)) return;
            if (!component.TryGetData(HandsVisuals.VisualState, out HandsVisualState visualState)) return;

            foreach (HandLocation location in Enum.GetValues(typeof(HandLocation)))
            {
                var layerKey = LocationToLayerKey(location);
                if (sprite.LayerMapTryGet(layerKey, out var layer))
                {
                    sprite.RemoveLayer(layer);
                    sprite.LayerMapRemove(layer);
                }
            }

            var resourceCache = IoCManager.Resolve<IResourceCache>();
            var hands = visualState.Hands;

            foreach (var hand in hands)
            {
                var rsi = resourceCache.GetResource<RSIResource>(SharedSpriteComponent.TextureRoot / hand.RsiPath).RSI;

                var state = $"inhand-{hand.Location.ToString().ToLowerInvariant()}";
                if (hand.EquippedPrefix != null)
                    state = $"{hand.EquippedPrefix}-" + state;

                if (rsi.TryGetState(state, out var _))
                {
                    var layerKey = LocationToLayerKey(hand.Location);
                    sprite.LayerMapReserveBlank(layerKey);

                    var layer = sprite.LayerMapGet(layerKey);
                    sprite.LayerSetVisible(layer, true);
                    sprite.LayerSetRSI(layer, rsi);
                    sprite.LayerSetColor(layer, hand.Color);
                    sprite.LayerSetState(layer, state);
                }
            }
        }

        private string LocationToLayerKey(HandLocation location)
        {
            return location.ToString();
        }
    }
}