tbd-station-14/Content.Server/GameObjects/Components/Mobs/ServerStatusEffectsComponent.cs

using System;
using System.Collections.Generic;
using Content.Server.GameObjects.Components.Atmos;
using Content.Server.GameObjects.Components.Buckle;
using Content.Server.GameObjects.Components.Movement;
using Content.Server.GameObjects.EntitySystems;
using Content.Shared.GameObjects.Components.Mobs;
using Content.Shared.GameObjects.Components.Pulling;
using Content.Shared.GameObjects.EntitySystems;
using Content.Shared.Interfaces;
using Robust.Shared.GameObjects;
using Robust.Shared.GameObjects.Systems;
using Robust.Shared.Interfaces.Network;
using Robust.Shared.Players;
using Robust.Shared.ViewVariables;

namespace Content.Server.GameObjects.Components.Mobs
{
    [RegisterComponent]
    [ComponentReference(typeof(SharedStatusEffectsComponent))]
    public sealed class ServerStatusEffectsComponent : SharedStatusEffectsComponent
    {
        [ViewVariables]
        private readonly Dictionary<StatusEffect, StatusEffectStatus> _statusEffects = new Dictionary<StatusEffect, StatusEffectStatus>();

        protected override void Startup()
        {
            base.Startup();

            EntitySystem.Get<WeightlessSystem>().AddStatus(this);
        }

        public override void OnRemove()
        {
            EntitySystem.Get<WeightlessSystem>().RemoveStatus(this);

            base.OnRemove();
        }

        public override ComponentState GetComponentState()
        {
            return new StatusEffectComponentState(_statusEffects);
        }

        public override void ChangeStatusEffectIcon(StatusEffect effect, string icon)
        {
            if (_statusEffects.TryGetValue(effect, out var value) && value.Icon == icon)
            {
                return;
            }

            _statusEffects[effect] = new StatusEffectStatus()
                {Icon = icon, Cooldown = value.Cooldown};
            Dirty();
        }

        public void ChangeStatusEffectCooldown(StatusEffect effect, ValueTuple<TimeSpan, TimeSpan> cooldown)
        {
            if (_statusEffects.TryGetValue(effect, out var value)
                && value.Cooldown == cooldown)
            {
                return;
            }

            _statusEffects[effect] = new StatusEffectStatus()
            {
                Icon = value.Icon, Cooldown = cooldown
            };
            Dirty();
        }

        public override void ChangeStatusEffect(StatusEffect effect, string icon, ValueTuple<TimeSpan, TimeSpan>? cooldown)
        {
            _statusEffects[effect] = new StatusEffectStatus()
                {Icon = icon, Cooldown = cooldown};

            Dirty();
        }

        public override void RemoveStatusEffect(StatusEffect effect)
        {
            if (!_statusEffects.Remove(effect))
            {
                return;
            }

            Dirty();
        }

        public override void HandleNetworkMessage(ComponentMessage message, INetChannel netChannel, ICommonSession session = null)
        {
            base.HandleNetworkMessage(message, netChannel, session);

            if (session == null)
            {
                throw new ArgumentNullException(nameof(session));
            }

            switch (message)
            {
                case ClickStatusMessage msg:
                {
                    var player = session.AttachedEntity;

                    if (player != Owner)
                    {
                        break;
                    }

                    // TODO: Implement clicking other status effects in the HUD
                    switch (msg.Effect)
                    {
                        case StatusEffect.Buckled:
                            if (!player.TryGetComponent(out BuckleComponent buckle))
                                break;

                            buckle.TryUnbuckle(player);
                            break;
                        case StatusEffect.Piloting:
                            if (!player.TryGetComponent(out ShuttleControllerComponent controller))
                                break;

                            controller.RemoveController();
                            break;
                        case StatusEffect.Pulling:
                            EntitySystem
                                .Get<SharedPullingSystem>()
                                .GetPulled(player)?
                                .GetComponentOrNull<SharedPullableComponent>()?
                                .TryStopPull();

                            break;
                        case StatusEffect.Fire:
                            if (!player.TryGetComponent(out FlammableComponent flammable))
                                break;

                            flammable.Resist();
                            break;
                        default:
                            player.PopupMessage(msg.Effect.ToString());
                            break;
                    }

                    break;
                }
            }
        }
    }

}