tbd-station-14/Content.Shared/GameObjects/Components/Body/Mechanism/SharedMechanismComponent.cs

#nullable enable
using System.Collections.Generic;
using Content.Shared.GameObjects.Components.Body.Part;
using Robust.Shared.GameObjects;
using Robust.Shared.Interfaces.GameObjects;
using Robust.Shared.Serialization;

namespace Content.Shared.GameObjects.Components.Body.Mechanism
{
    public abstract class SharedMechanismComponent : Component, IMechanism
    {
        public override string Name => "Mechanism";

        private IBodyPart? _part;

        protected readonly Dictionary<int, object> OptionsCache = new Dictionary<int, object>();

        protected IBody? BodyCache;

        protected int IdHash;

        protected IEntity? PerformerCache;

        public IBody? Body => Part?.Body;

        public IBodyPart? Part
        {
            get => _part;
            set
            {
                if (_part == value)
                {
                    return;
                }

                var old = _part;
                _part = value;

                if (value != null)
                {
                    OnPartAdd(old, value);
                }
                else if (old != null)
                {
                    OnPartRemove(old);
                }
            }
        }

        public string Description { get; set; } = string.Empty;

        public string ExamineMessage { get; set; } = string.Empty;

        public int MaxDurability { get; set; }

        public int CurrentDurability { get; set; }

        public int DestroyThreshold { get; set; }

        // TODO BODY
        public int Resistance { get; set; }

        // TODO BODY OnSizeChanged
        public int Size { get; set; }

        public BodyPartCompatibility Compatibility { get; set; }

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

            serializer.DataField(this, m => m.Description, "description", string.Empty);

            serializer.DataField(this, m => m.ExamineMessage, "examineMessage", string.Empty);

            serializer.DataField(this, m => m.MaxDurability, "maxDurability", 10);

            serializer.DataField(this, m => m.CurrentDurability, "currentDurability", MaxDurability);

            serializer.DataField(this, m => m.DestroyThreshold, "destroyThreshold", -MaxDurability);

            serializer.DataField(this, m => m.Resistance, "resistance", 0);

            serializer.DataField(this, m => m.Size, "size", 1);

            serializer.DataField(this, m => m.Compatibility, "compatibility", BodyPartCompatibility.Universal);
        }

        public virtual void OnBodyAdd(IBody? old, IBody current) { }

        public virtual void OnBodyRemove(IBody old) { }

        protected virtual void OnPartAdd(IBodyPart? old, IBodyPart current)
        {
            Owner.Transform.AttachParent(current.Owner);
        }

        protected virtual void OnPartRemove(IBodyPart old)
        {
            Owner.Transform.AttachToGridOrMap();
        }
    }
}