crystall-punk-14/Content.Client/GameObjects/Components/Items/ItemComponent.cs

using Content.Client.GameObjects.Components.Disposal;
using Content.Shared.GameObjects;
using Content.Shared.GameObjects.Components.Items;
using Content.Shared.Interfaces.GameObjects.Components;
using Robust.Client.Graphics;
using Robust.Client.ResourceManagement;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Maths;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.Utility;
using Robust.Shared.ViewVariables;

namespace Content.Client.GameObjects.Components.Items
{
    [RegisterComponent]
    [ComponentReference(typeof(IItemComponent))]
    public class ItemComponent : Component, IItemComponent, IDraggable
    {
        [Dependency] private readonly IResourceCache _resourceCache = default!;

        public override string Name => "Item";
        public override uint? NetID => ContentNetIDs.ITEM;

        [ViewVariables]
        [DataField("sprite")]
        protected ResourcePath? RsiPath;

        [ViewVariables(VVAccess.ReadWrite)]
        [DataField("color")]
        protected Color Color = Color.White;

        [DataField("HeldPrefix")]
        private string? _equippedPrefix;

        [ViewVariables(VVAccess.ReadWrite)]
        public string? EquippedPrefix
        {
            get => _equippedPrefix;
            set
            {
                _equippedPrefix = value;

                if (!Owner.TryGetContainer(out var container))
                    return;

                if (container.Owner.TryGetComponent(out HandsComponent? hands))
                    hands.RefreshInHands();
            }
        }

        public (RSI rsi, RSI.StateId stateId, Color color)? GetInHandStateInfo(HandLocation hand)
        {
            var rsi = GetRSI();

            if (rsi == null)
            {
                return null;
            }

            var handName = hand.ToString().ToLowerInvariant();
            var stateId = EquippedPrefix != null ? $"{EquippedPrefix}-inhand-{handName}" : $"inhand-{handName}";

            if (rsi.TryGetState(stateId, out _))
            {
                return (rsi, stateId, Color);
            }

            return null;
        }

        protected RSI? GetRSI()
        {
            if (RsiPath == null)
            {
                return null;
            }

            return _resourceCache.GetResource<RSIResource>(SharedSpriteComponent.TextureRoot / RsiPath).RSI;
        }

        public override void HandleComponentState(ComponentState? curState, ComponentState? nextState)
        {
            if (curState is not ItemComponentState state)
                return;

            EquippedPrefix = state.EquippedPrefix;
        }

        bool IDraggable.CanDrop(CanDropEventArgs args)
        {
            return args.Target.HasComponent<DisposalUnitComponent>();
        }

        bool IDraggable.Drop(DragDropEventArgs args)
        {
            // TODO: Shared item class
            return false;
        }
    }
}