tbd-station-14/Content.Client/GameObjects/Components/HUD/Inventory/ClientInventoryComponent.cs

using Content.Shared.GameObjects;
using Robust.Client.Interfaces.GameObjects.Components;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Shared.GameObjects;
using Robust.Shared.Interfaces.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Serialization;
using Robust.Shared.Utility;
using System;
using System.Collections.Generic;
using System.Linq;
using Content.Client.GameObjects.Components.Clothing;
using Robust.Shared.Interfaces.Reflection;
using static Content.Shared.GameObjects.Components.Inventory.EquipmentSlotDefines;
using static Content.Shared.GameObjects.SharedInventoryComponent.ClientInventoryMessage;
using Content.Client.GameObjects.Components.Mobs;
using Content.Client.GameObjects.Components.Actor;
using Content.Client.UserInterface;
using Robust.Client.GameObjects;
using Robust.Client.UserInterface.CustomControls;
using Robust.Shared.Interfaces.Network;

namespace Content.Client.GameObjects
{
    /// <summary>
    /// A character UI which shows items the user has equipped within his inventory
    /// </summary>
    public class ClientInventoryComponent : SharedInventoryComponent
    {
        private readonly Dictionary<Slots, IEntity> _slots = new Dictionary<Slots, IEntity>();

        [Dependency]
#pragma warning disable 649
        private readonly IGameHud _gameHud;
#pragma warning restore 649

        /// <summary>
        /// Holds the godot control for the inventory window
        /// </summary>
        private InventoryWindow _window;

        public SS14Window Window => _window;

        private string _templateName = "HumanInventory"; //stored for serialization purposes

        /// <summary>
        /// Inventory template after being loaded from instance creator and string name
        /// </summary>
        private Inventory _inventory;

        private ISpriteComponent _sprite;

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

            _window.Dispose();
        }

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

            //Loads inventory template
            var reflectionManager = IoCManager.Resolve<IReflectionManager>();
            var type = reflectionManager.LooseGetType(_templateName);
            DebugTools.Assert(type != null);
            _inventory = (Inventory) Activator.CreateInstance(type);

            //Creates godot control class for inventory
            _window = new InventoryWindow(this);
            _window.CreateInventory(_inventory);
            _window.OnClose += () => _gameHud.InventoryButtonDown = false;

            if (Owner.TryGetComponent(out _sprite))
            {
                foreach (var mask in _inventory.SlotMasks.OrderBy(s => _inventory.SlotDrawingOrder(s)))
                {
                    if (mask == Slots.NONE)
                    {
                        continue;
                    }

                    _sprite.LayerMapReserveBlank(mask);
                }
            }

            // Component state already came in but we couldn't set anything visually because, well, we didn't initialize yet.
            foreach (var (slot, entity) in _slots)
            {
                _setSlot(slot, entity);
            }
        }

        public override void ExposeData(ObjectSerializer serializer)
        {
            base.ExposeData(serializer);

            serializer.DataField(ref _templateName, "Template", "HumanInventory");
        }

        public override void HandleComponentState(ComponentState curState, ComponentState nextState)
        {
            base.HandleComponentState(curState, nextState);

            if (curState == null)
                return;

            var cast = (InventoryComponentState) curState;

            var doneSlots = new HashSet<Slots>();

            foreach (var (slot, entityUid) in cast.Entities)
            {
                if (_slots.ContainsKey(slot))
                {
                    _slots.Remove(slot);
                    _clearSlot(slot);
                }

                var entity = Owner.EntityManager.GetEntity(entityUid);
                _slots[slot] = entity;
                _setSlot(slot, entity);
                doneSlots.Add(slot);
            }

            foreach (var slot in _slots.Keys.ToList())
            {
                if (!doneSlots.Contains(slot))
                {
                    _clearSlot(slot);
                    _slots.Remove(slot);
                }
            }
        }

        private void _setSlot(Slots slot, IEntity entity)
        {
            if (_sprite != null && entity.TryGetComponent(out ClothingComponent clothing))
            {
                var flag = SlotMasks[slot];
                var data = clothing.GetEquippedStateInfo(flag);
                if (data == null)
                {
                    _sprite.LayerSetVisible(slot, false);
                }
                else
                {
                    var (rsi, state) = data.Value;
                    _sprite.LayerSetVisible(slot, true);
                    _sprite.LayerSetRSI(slot, rsi);
                    _sprite.LayerSetState(slot, state);
                }
            }

            _window?.AddToSlot(slot, entity);
        }

        private void _clearSlot(Slots slot)
        {
            _window?.RemoveFromSlot(slot);
            _sprite?.LayerSetVisible(slot, false);
        }

        public void SendUnequipMessage(Slots slot)
        {
            var unequipmessage = new ClientInventoryMessage(slot, ClientInventoryUpdate.Unequip);
            SendNetworkMessage(unequipmessage);
        }

        public void SendEquipMessage(Slots slot)
        {
            var equipmessage = new ClientInventoryMessage(slot, ClientInventoryUpdate.Equip);
            SendNetworkMessage(equipmessage);
        }

        public override void HandleMessage(ComponentMessage message, INetChannel netChannel = null,
            IComponent component = null)
        {
            base.HandleMessage(message, netChannel, component);

            switch (message)
            {
                case PlayerAttachedMsg _:
                    _gameHud.InventoryButtonVisible = true;
                    _gameHud.InventoryButtonToggled = b =>
                    {
                        if (b)
                        {
                            Window.Open();
                        }
                        else
                        {
                            Window.Close();
                        }
                    };

                    break;

                case PlayerDetachedMsg _:
                    _gameHud.InventoryButtonVisible = false;
                    _window.Close();

                    break;
            }
        }

        /// <summary>
        /// Temporary window to hold the basis for inventory hud
        /// </summary>
        private sealed class InventoryWindow : SS14Window
        {
            private List<Slots> IndexedSlots;

            private Dictionary<Slots, InventoryButton>
                InventorySlots = new Dictionary<Slots, InventoryButton>(); //ordered dictionary?

            private readonly ClientInventoryComponent InventoryComponent;
            private readonly GridContainer _gridContainer;

            public InventoryWindow(ClientInventoryComponent inventory)
            {
                Title = "Inventory";

                InventoryComponent = inventory;
                _gridContainer = new GridContainer();
                Contents.AddChild(_gridContainer);

                Size = CombinedMinimumSize;
            }

            /// <summary>
            ///     Creates a grid container filled with slot buttons loaded from an inventory template
            /// </summary>
            public void CreateInventory(Inventory inventory)
            {
                _gridContainer.Columns = inventory.Columns;

                IndexedSlots = new List<Slots>(inventory.SlotMasks);

                foreach (var slot in IndexedSlots)
                {
                    var newButton = new InventoryButton(slot);

                    if (slot == Slots.NONE)
                    {
                        //TODO: Re-enable when godot grid container maintains grid with invisible elements
                        //newbutton.Visible = false;
                    }
                    else
                    {
                        // Store slot button and give it the default onpress behavior for empty elements
                        newButton.GetChild<Button>("Button").OnPressed += AddToInventory;
                        InventorySlots.Add(slot, newButton);
                    }

                    if (SlotNames.ContainsKey(slot))
                    {
                        var button = newButton.GetChild<Button>("Button");
                        button.Text = button.ToolTip = SlotNames[slot];
                    }

                    _gridContainer.AddChild(newButton);
                }
            }

            /// <summary>
            /// Adds the item we have equipped to the slot texture and prepares the slot button for removal
            /// </summary>
            public void AddToSlot(Slots slot, IEntity entity)
            {
                var button = InventorySlots[slot];

                button.EntityUid = entity.Uid;
                var theButton = button.GetChild<Button>("Button");
                theButton.OnPressed += RemoveFromInventory;
                theButton.OnPressed -= AddToInventory;
                theButton.Text = "";

                //Gets entity sprite and assigns it to button texture
                if (entity.TryGetComponent(out ISpriteComponent sprite))
                {
                    var view = button.GetChild<SpriteView>("SpriteView");
                    view.Sprite = sprite;
                }
            }

            /// <summary>
            /// Remove element from the UI and update its button to blank texture and prepare for insertion again
            /// </summary>
            public void RemoveFromSlot(Slots slot)
            {
                var button = InventorySlots[slot];
                button.GetChild<SpriteView>("SpriteView").Sprite = null;
                button.EntityUid = EntityUid.Invalid;
                var theButton = button.GetChild<Button>("Button");
                theButton.OnPressed -= RemoveFromInventory;
                theButton.OnPressed += AddToInventory;
                theButton.Text = SlotNames[slot];
            }

            private void RemoveFromInventory(BaseButton.ButtonEventArgs args)
            {
                args.Button.Pressed = false;
                var control = (InventoryButton) args.Button.Parent;

                InventoryComponent.SendUnequipMessage(control.Slot);
            }

            private void AddToInventory(BaseButton.ButtonEventArgs args)
            {
                args.Button.Pressed = false;
                var control = (InventoryButton) args.Button.Parent;

                InventoryComponent.SendEquipMessage(control.Slot);
            }
        }

        private sealed class InventoryButton : MarginContainer
        {
            public Slots Slot { get; }
            public EntityUid EntityUid { get; set; }

            public InventoryButton(Slots slot)
            {
                Slot = slot;

                CustomMinimumSize = (50, 50);

                AddChild(new Button("Button")
                {
                    ClipText = true
                });
                AddChild(new SpriteView("SpriteView")
                {
                    MouseFilter = MouseFilterMode.Ignore
                });
            }
        }
    }
}