tbd-station-14/Content.Shared/GameObjects/Components/Mobs/SharedOverlayEffectsComponent.cs

using System;
using System.Collections.Generic;
using System.Linq;
using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.Interfaces.Timing;
using Robust.Shared.IoC;
using Robust.Shared.Serialization;
using Robust.Shared.ViewVariables;

namespace Content.Shared.GameObjects.Components.Mobs
{
    /// <summary>
    /// Full screen overlays; Blindness, death, flash, alcohol etc.
    /// </summary>
    public abstract class SharedOverlayEffectsComponent : Component
    {
        public override string Name => "OverlayEffectsUI";

        public sealed override uint? NetID => ContentNetIDs.OVERLAYEFFECTS;
    }

    [Serializable, NetSerializable]
    public class OverlayContainer : IEquatable<string>, IEquatable<OverlayContainer>
    {
        [ViewVariables(VVAccess.ReadOnly)]
        public string ID { get; }

        [ViewVariables(VVAccess.ReadWrite)]
        public List<OverlayParameter> Parameters { get; } = new List<OverlayParameter>();

        public OverlayContainer([NotNull] string id)
        {
            ID = id;
        }

        public OverlayContainer(SharedOverlayID id) : this(id.ToString())
        {
        }

        public OverlayContainer(SharedOverlayID id, params OverlayParameter[] parameters) : this(id)
        {
            Parameters.AddRange(parameters);
        }

        public bool TryGetOverlayParameter<T>(out T parameter) where T : OverlayParameter
        {
            var found = Parameters.FirstOrDefault(p => p is T);
            if (found != null)
            {
                parameter = found as T;
                return true;
            }

            parameter = default;
            return false;
        }

        public bool Equals(string other)
        {
            return ID == other;
        }

        public bool Equals(OverlayContainer other)
        {
            return ID == other?.ID;
        }

        public override int GetHashCode()
        {
            return ID != null ? ID.GetHashCode() : 0;
        }

    }

    [Serializable, NetSerializable]
    public class OverlayEffectComponentMessage : ComponentMessage
    {
        public List<OverlayContainer> Overlays;

        public OverlayEffectComponentMessage(List<OverlayContainer> overlays)
        {
            Directed = true;
            Overlays = overlays;
        }
    }

    [Serializable, NetSerializable]
    public class ResendOverlaysMessage : ComponentMessage
    {
    }

    [Serializable, NetSerializable]
    public abstract class OverlayParameter
    {
    }

    [Serializable, NetSerializable]
    public class TimedOverlayParameter : OverlayParameter
    {
        [ViewVariables(VVAccess.ReadOnly)]
        public int Length { get; set; }

        public double StartedAt { get; private set; }

        public TimedOverlayParameter(int length)
        {
            Length = length;
            StartedAt = IoCManager.Resolve<IGameTiming>().CurTime.TotalMilliseconds;
        }
    }

    public enum SharedOverlayID
    {
        GradientCircleMaskOverlay,
        CircleMaskOverlay,
        FlashOverlay,
        RadiationPulseOverlay,
    }
}