CatStation/Content.Client/NPC/HTN/HTNSystem.cs

using Content.Shared.NPC;
using Robust.Client.Graphics;
using Robust.Client.ResourceManagement;

namespace Content.Client.NPC.HTN;

public sealed partial class HTNSystem : EntitySystem
{
    /*
     * Mainly handles clientside debugging for HTN NPCs.
     */
    public bool EnableOverlay
    {
        get => _enableOverlay;
        set
        {
            var overlayManager = IoCManager.Resolve<IOverlayManager>();
            _enableOverlay = value;

            if (_enableOverlay)
            {
                overlayManager.AddOverlay(new HTNOverlay(EntityManager, IoCManager.Resolve<IResourceCache>()));
                RaiseNetworkEvent(new RequestHTNMessage()
                {
                    Enabled = true,
                });
            }
            else
            {
                overlayManager.RemoveOverlay<HTNOverlay>();
                RaiseNetworkEvent(new RequestHTNMessage()
                {
                    Enabled = false,
                });
            }
        }
    }

    private bool _enableOverlay;

    public override void Initialize()
    {
        base.Initialize();
        SubscribeNetworkEvent<HTNMessage>(OnHTNMessage);
    }

    private void OnHTNMessage(HTNMessage ev)
    {
        if (!TryComp<HTNComponent>(GetEntity(ev.Uid), out var htn))
            return;

        htn.DebugText = ev.Text;
    }
}