ss14-wega/Content.Server/_Wega/Hallucinations/HallucinationsSystem.cs

using Content.Shared.Administration.Logs;
using Content.Shared.Database;
using Content.Shared.Hallucinations;
using Content.Shared.StatusEffect;
using Robust.Server.GameObjects;
using Robust.Shared.Prototypes;
using Robust.Shared.Random;
using Robust.Shared.Timing;

namespace Content.Server.Hallucinations;

// TODO: Full refactor this shit
public sealed partial class HallucinationsSystem : EntitySystem
{
    [Dependency] private readonly IEntityManager _entityManager = default!;
    [Dependency] private readonly IGameTiming _timing = default!;
    [Dependency] private readonly IPrototypeManager _proto = default!;
    [Dependency] private readonly IRobustRandom _random = default!;
    [Dependency] private readonly ISharedAdminLogManager _adminLogger = default!;
    [Dependency] private readonly SharedEyeSystem _eye = default!;
    [Dependency] private readonly StatusEffectsSystem _status = default!;
    [Dependency] private readonly VisibilitySystem _visibilitySystem = default!;

    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<HallucinationsComponent, MapInitEvent>(OnHallucinationsInit);
        SubscribeLocalEvent<HallucinationsComponent, ComponentShutdown>(OnHallucinationsShutdown);
    }

    private void OnHallucinationsInit(EntityUid uid, HallucinationsComponent component, MapInitEvent args)
    {
        component.Layer = _random.Next(100, 150);
        if (!_entityManager.TryGetComponent<EyeComponent>(uid, out var eye))
            return;
        UpdatePreset(component);
        _eye.SetVisibilityMask(uid, eye.VisibilityMask | component.Layer, eye);
        _adminLogger.Add(LogType.Action, LogImpact.Medium,
        $"{ToPrettyString(uid):player} began to hallucinate.");
    }

    /// <summary>
    /// Updates hallucinations component settings to match prototype
    /// </summary>
    /// <param name="component">Active HallucinationsComponent</param>
    public void UpdatePreset(HallucinationsComponent component)
    {
        if (component.Proto == null)
            return;
        var preset = component.Proto;

        component.Spawns = preset.Entities;
        component.Range = preset.Range;
        component.SpawnRate = preset.SpawnRate;
        component.MinChance = preset.MinChance;
        component.MaxChance = preset.MaxChance;
        component.MaxSpawns = preset.MaxSpawns;
        component.IncreaseChance = preset.IncreaseChance;
    }

    private void OnHallucinationsShutdown(EntityUid uid, HallucinationsComponent component, ComponentShutdown args)
    {
        if (!_entityManager.TryGetComponent<EyeComponent>(uid, out var eye))
            return;
        _eye.SetVisibilityMask(uid, eye.VisibilityMask & ~component.Layer, eye);
        _adminLogger.Add(LogType.Action, LogImpact.Medium,
        $"{ToPrettyString(uid):player} stopped hallucinating.");
    }

    /// <summary>
    ///     Attempts to start hallucinations for target
    /// </summary>
    /// <param name="target">The target.</param>
    /// <param name="key">Status effect key.</param>
    /// <param name="time">Duration of hallucinations effect.</param>
    /// <param name="refresh">Refresh active effects.</param>
    /// <param name="proto">Hallucinations pack prototype.</param>
    public bool StartHallucinations(EntityUid target, string key, TimeSpan time, bool refresh, string proto)
    {
        if (proto == null)
            return false;
        if (!_proto.TryIndex<HallucinationsPrototype>(proto, out var prototype))
            return false;
        if (!_status.TryAddStatusEffect<HallucinationsComponent>(target, key, time, refresh))
            return false;

        var hallucinations = _entityManager.GetComponent<HallucinationsComponent>(target);
        hallucinations.Proto = prototype;
        UpdatePreset(hallucinations);
        hallucinations.CurChance = prototype.MinChance;

        return true;
    }

    public override void Update(float frameTime)
    {
        base.Update(frameTime);

        var query = EntityQueryEnumerator<HallucinationsComponent, TransformComponent>();
        while (query.MoveNext(out var uid, out var stat, out var xform))
        {
            if (_timing.CurTime < stat.NextSecond)
                continue;
            var rate = stat.SpawnRate;
            stat.NextSecond = _timing.CurTime + TimeSpan.FromSeconds(rate);

            if (stat.CurChance < stat.MaxChance && stat.CurChance + stat.IncreaseChance <= 1)
                stat.CurChance = stat.CurChance + stat.IncreaseChance;

            if (!_random.Prob(stat.CurChance))
                continue;

            var selectedEntity = _random.Pick(stat.Spawns);
            int spawnCount = _random.Next(3, 6);
            stat.SpawnedCount = 0;

            var range = stat.Range * 4;

            UpdatePreset(stat);

            for (int i = 0; i < spawnCount; i++)
            {
                var newCoords = Transform(uid).MapPosition.Offset(_random.NextVector2(stat.Range));

                if (stat.SpawnedCount >= stat.MaxSpawns)
                    continue;

                stat.SpawnedCount++;

                var hallucination = Spawn(selectedEntity, newCoords);
                EnsureComp<VisibilityComponent>(hallucination, out var visibility);
                _visibilitySystem.SetLayer(hallucination, (ushort)stat.Layer, false);
                _visibilitySystem.RefreshVisibility(hallucination, visibilityComponent: visibility);
            }
        }
    }
}