RobustToolbox/Robust.Server/GameObjects/EntitySystems/VisibilitySystem.cs

using Robust.Shared.GameObjects;

namespace Robust.Server.GameObjects
{
    public sealed class VisibilitySystem : EntitySystem
    {
        public override void Initialize()
        {
            base.Initialize();
            SubscribeLocalEvent<EntParentChangedMessage>(OnParentChange);
            EntityManager.EntityInitialized += OnEntityInit;
        }

        public override void Shutdown()
        {
            base.Shutdown();
            EntityManager.EntityInitialized -= OnEntityInit;
        }

        public void AddLayer(VisibilityComponent component, int layer, bool refresh = true)
        {
            if ((layer & component.Layer) == layer) return;
            component.Layer |= layer;

            if (refresh)
                RefreshVisibility(component.Owner, visibilityComponent: component);
        }

        public void RemoveLayer(VisibilityComponent component, int layer, bool refresh = true)
        {
            if ((layer & component.Layer) != layer) return;
            component.Layer &= ~layer;

            if (refresh)
                RefreshVisibility(component.Owner, visibilityComponent: component);
        }

        public void SetLayer(VisibilityComponent component, int layer, bool refresh = true)
        {
            if (component.Layer == layer) return;
            component.Layer = layer;

            if (refresh)
                RefreshVisibility(component.Owner, visibilityComponent: component);
        }

        private void OnParentChange(ref EntParentChangedMessage ev)
        {
            RefreshVisibility(ev.Entity);
        }

        private void OnEntityInit(object? sender, EntityUid uid)
        {
            RefreshVisibility(uid);
        }

        public void RefreshVisibility(EntityUid uid, MetaDataComponent? metaDataComponent = null, VisibilityComponent? visibilityComponent = null)
        {
            if (!Resolve(uid, ref metaDataComponent, false))
            {
                // This means it's deleting or some shit; I'd love to make it a GetComponent<T> in future.
                return;
            }

            var visMask = 1;
            metaDataComponent.VisibilityMask = GetVisibilityMask(uid, ref visMask, visibilityComponent);
        }

        public void RefreshVisibility(VisibilityComponent visibilityComponent)
        {
            RefreshVisibility(visibilityComponent.Owner);
        }

        private int GetVisibilityMask(EntityUid uid, ref int visMask, VisibilityComponent? visibilityComponent = null, TransformComponent? xform = null)
        {
            if (Resolve(uid, ref visibilityComponent, false))
            {
                visMask |= visibilityComponent.Layer;
            }

            if (Resolve(uid, ref xform) && xform.ParentUid.IsValid())
            {
                GetVisibilityMask(xform.ParentUid, ref visMask);
            }

            return visMask;
        }
    }
}