tbd-station-14/Content.Shared/Rootable/SharedRootableSystem.cs

using Content.Shared.Actions;
using Content.Shared.Actions.Components;
using Content.Shared.Alert;
using Content.Shared.Cloning.Events;
using Content.Shared.Coordinates;
using Content.Shared.Fluids.Components;
using Content.Shared.Gravity;
using Content.Shared.Mobs;
using Content.Shared.Movement.Systems;
using Content.Shared.Slippery;
using Content.Shared.Toggleable;
using Content.Shared.Trigger.Components.Effects;
using Robust.Shared.Audio.Systems;
using Robust.Shared.Physics.Components;
using Robust.Shared.Physics.Events;
using Robust.Shared.Physics.Systems;
using Robust.Shared.Timing;

namespace Content.Shared.Rootable;

/// <summary>
/// Adds an action to toggle rooting to the ground, primarily for the Diona species.
/// Being rooted prevents weighlessness and slipping, but causes any floor contents to transfer its reagents to the bloodstream.
/// </summary>
public abstract class SharedRootableSystem : EntitySystem
{
    [Dependency] private readonly IGameTiming _timing = default!;
    [Dependency] private readonly SharedActionsSystem _actions = default!;
    [Dependency] private readonly SharedGravitySystem _gravity = default!;
    [Dependency] private readonly SharedPhysicsSystem _physics = default!;
    [Dependency] private readonly MovementSpeedModifierSystem _movementSpeedModifier = default!;
    [Dependency] private readonly AlertsSystem _alerts = default!;
    [Dependency] private readonly SharedAudioSystem _audio = default!;

    protected EntityQuery<PuddleComponent> PuddleQuery;
    protected EntityQuery<PhysicsComponent> PhysicsQuery;

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

        PuddleQuery = GetEntityQuery<PuddleComponent>();
        PhysicsQuery = GetEntityQuery<PhysicsComponent>();

        SubscribeLocalEvent<RootableComponent, MapInitEvent>(OnRootableMapInit);
        SubscribeLocalEvent<RootableComponent, ComponentShutdown>(OnRootableShutdown);
        SubscribeLocalEvent<RootableComponent, StartCollideEvent>(OnStartCollide);
        SubscribeLocalEvent<RootableComponent, EndCollideEvent>(OnEndCollide);
        SubscribeLocalEvent<RootableComponent, ToggleActionEvent>(OnRootableToggle);
        SubscribeLocalEvent<RootableComponent, MobStateChangedEvent>(OnMobStateChanged);
        SubscribeLocalEvent<RootableComponent, IsWeightlessEvent>(OnIsWeightless);
        SubscribeLocalEvent<RootableComponent, SlipAttemptEvent>(OnSlipAttempt);
        SubscribeLocalEvent<RootableComponent, RefreshMovementSpeedModifiersEvent>(OnRefreshMovementSpeed);
        SubscribeLocalEvent<RootableComponent, CloningEvent>(OnCloning);
    }

    private void OnCloning(Entity<RootableComponent> ent, ref CloningEvent args)
    {
        if (!args.Settings.EventComponents.Contains(Factory.GetRegistration(ent.Comp.GetType()).Name))
            return;

        var cloneComp = EnsureComp<RootableComponent>(args.CloneUid);
        cloneComp.TransferRate = ent.Comp.TransferRate;
        cloneComp.TransferFrequency = ent.Comp.TransferFrequency;
        cloneComp.SpeedModifier = ent.Comp.SpeedModifier;
        cloneComp.RootSound = ent.Comp.RootSound;
        Dirty(args.CloneUid, cloneComp);
    }

    private void OnRootableMapInit(Entity<RootableComponent> entity, ref MapInitEvent args)
    {
        if (!TryComp(entity, out ActionsComponent? comp))
            return;

        entity.Comp.NextUpdate = _timing.CurTime;
        _actions.AddAction(entity, ref entity.Comp.ActionEntity, entity.Comp.Action, component: comp);
    }

    private void OnRootableShutdown(Entity<RootableComponent> entity, ref ComponentShutdown args)
    {
        if (!TryComp(entity, out ActionsComponent? comp))
            return;

        var actions = new Entity<ActionsComponent?>(entity, comp);
        _actions.RemoveAction(actions, entity.Comp.ActionEntity);
        _alerts.ClearAlert(entity.Owner, entity.Comp.RootedAlert);
    }

    private void OnRootableToggle(Entity<RootableComponent> entity, ref ToggleActionEvent args)
    {
        args.Handled = TryToggleRooting((entity, entity));
    }

    private void OnMobStateChanged(Entity<RootableComponent> entity, ref MobStateChangedEvent args)
    {
        if (entity.Comp.Rooted)
            TryToggleRooting((entity, entity));
    }

    public bool TryToggleRooting(Entity<RootableComponent?> entity)
    {
        if (!Resolve(entity, ref entity.Comp))
            return false;

        entity.Comp.Rooted = !entity.Comp.Rooted;
        _movementSpeedModifier.RefreshMovementSpeedModifiers(entity);
        _gravity.RefreshWeightless(entity.Owner);
        Dirty(entity);

        if (entity.Comp.Rooted)
        {
            _alerts.ShowAlert(entity.Owner, entity.Comp.RootedAlert);
            var curTime = _timing.CurTime;
            if (curTime > entity.Comp.NextUpdate)
            {
                entity.Comp.NextUpdate = curTime;
            }
        }
        else
        {
            _alerts.ClearAlert(entity.Owner, entity.Comp.RootedAlert);
        }
        _audio.PlayPredicted(entity.Comp.RootSound, entity.Owner.ToCoordinates(), entity);

        return true;
    }

    private void OnIsWeightless(Entity<RootableComponent> ent, ref IsWeightlessEvent args)
    {
        if (args.Handled || !ent.Comp.Rooted)
            return;

        // do not cancel weightlessness if the person is in off-grid.
        if (!_gravity.EntityOnGravitySupportingGridOrMap(ent.Owner))
            return;

        args.IsWeightless = false;
        args.Handled = true;
    }

    private void OnSlipAttempt(Entity<RootableComponent> ent, ref SlipAttemptEvent args)
    {
        if (!ent.Comp.Rooted)
            return;

        if (args.SlipCausingEntity != null && HasComp<DamageOnTriggerComponent>(args.SlipCausingEntity))
            return;

        args.NoSlip = true;
    }

    private void OnStartCollide(Entity<RootableComponent> entity, ref StartCollideEvent args)
    {
        if (!PuddleQuery.HasComp(args.OtherEntity))
            return;

        entity.Comp.PuddleEntity = args.OtherEntity;

        if (entity.Comp.NextUpdate < _timing.CurTime) // To prevent constantly moving to new puddles resetting the timer
            entity.Comp.NextUpdate = _timing.CurTime;
    }

    private void OnEndCollide(Entity<RootableComponent> entity, ref EndCollideEvent args)
    {
        if (entity.Comp.PuddleEntity != args.OtherEntity)
            return;

        var exists = Exists(args.OtherEntity);

        if (!PhysicsQuery.TryComp(entity, out var body))
            return;

        foreach (var ent in _physics.GetContactingEntities(entity, body))
        {
            if (exists && ent == args.OtherEntity)
                continue;

            if (!PuddleQuery.HasComponent(ent))
                continue;

            entity.Comp.PuddleEntity = ent;
            return; // New puddle found, no need to continue
        }

        entity.Comp.PuddleEntity = null;
    }

    private void OnRefreshMovementSpeed(Entity<RootableComponent> entity, ref RefreshMovementSpeedModifiersEvent args)
    {
        if (entity.Comp.Rooted)
            args.ModifySpeed(entity.Comp.SpeedModifier);
    }
}