tbd-station-14/Content.Server/Climbing/ClimbSystem.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Content.Server.Climbing.Components;
using Content.Server.Popups;
using Content.Server.Stunnable;
using Content.Shared.ActionBlocker;
using Content.Shared.Buckle.Components;
using Content.Shared.Climbing;
using Content.Shared.Damage;
using Content.Shared.GameTicking;
using Content.Shared.Verbs;
using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Localization;
using Robust.Shared.Player;

namespace Content.Server.Climbing
{
    [UsedImplicitly]
    internal sealed class ClimbSystem : SharedClimbSystem
    {
        private readonly HashSet<ClimbingComponent> _activeClimbers = new();

        [Dependency] private readonly ActionBlockerSystem _actionBlockerSystem = default!;
        [Dependency] private readonly StunSystem _stunSystem = default!;
        [Dependency] private readonly DamageableSystem _damageableSystem = default!;
        [Dependency] private readonly PopupSystem _popupSystem = default!;

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

            SubscribeLocalEvent<RoundRestartCleanupEvent>(Reset);
            SubscribeLocalEvent<ClimbableComponent, GetVerbsEvent<AlternativeVerb>>(AddClimbVerb);
            SubscribeLocalEvent<ClimbingComponent, BuckleChangeEvent>(OnBuckleChange);
            SubscribeLocalEvent<GlassTableComponent, ClimbedOnEvent>(OnGlassClimbed);
        }

        public void ForciblySetClimbing(EntityUid uid, ClimbingComponent? component = null)
        {
            if (!Resolve(uid, ref component, false))
                return;
            component.IsClimbing = true;
            UnsetTransitionBoolAfterBufferTime(uid, component);
        }

        private void AddClimbVerb(EntityUid uid, ClimbableComponent component, GetVerbsEvent<AlternativeVerb> args)
        {
            if (!args.CanAccess || !args.CanInteract || !_actionBlockerSystem.CanMove(args.User))
                return;

            // Check that the user climb.
            if (!EntityManager.TryGetComponent(args.User, out ClimbingComponent? climbingComponent) ||
                climbingComponent.IsClimbing)
                return;

            // Add a climb verb
            AlternativeVerb verb = new();
            verb.Act = () => component.TryClimb(args.User, args.Target);
            verb.Text = Loc.GetString("comp-climbable-verb-climb");
            // TODO VERBS ICON add a climbing icon?
            args.Verbs.Add(verb);
        }

        private void OnBuckleChange(EntityUid uid, ClimbingComponent component, BuckleChangeEvent args)
        {
            if (args.Buckling)
                component.IsClimbing = false;
        }

        private void OnGlassClimbed(EntityUid uid, GlassTableComponent component, ClimbedOnEvent args)
        {
            if (TryComp<PhysicsComponent>(args.Climber, out var physics) && physics.Mass <= component.MassLimit)
                return;
            _damageableSystem.TryChangeDamage(args.Climber, component.ClimberDamage);
            _damageableSystem.TryChangeDamage(uid, component.TableDamage);
            _stunSystem.TryParalyze(args.Climber, TimeSpan.FromSeconds(component.StunTime), true);

            // Not shown to the user, since they already get a 'you climb on the glass table' popup
            _popupSystem.PopupEntity(Loc.GetString("glass-table-shattered-others",
                    ("table", uid), ("climber", args.Climber)), args.Climber,
                Filter.Pvs(uid).RemoveWhereAttachedEntity(puid => puid == args.Climber));
        }

        public void AddActiveClimber(ClimbingComponent climbingComponent)
        {
            _activeClimbers.Add(climbingComponent);
        }

        public void RemoveActiveClimber(ClimbingComponent climbingComponent)
        {
            _activeClimbers.Remove(climbingComponent);
        }

        public void UnsetTransitionBoolAfterBufferTime(EntityUid uid, ClimbingComponent? component = null)
        {
            if (!Resolve(uid, ref component, false))
                return;
            component.Owner.SpawnTimer((int) (SharedClimbingComponent.BufferTime * 1000), () =>
            {
                if (component.Deleted) return;
                component.OwnerIsTransitioning = false;
            });
        }

        public override void Update(float frameTime)
        {
            foreach (var climber in _activeClimbers.ToArray())
            {
                climber.Update();
            }
        }

        public void Reset(RoundRestartCleanupEvent ev)
        {
            _activeClimbers.Clear();
        }
    }
}