using System.Collections.Generic; using Content.Server.Hands.Components; using Content.Server.Interaction; using Content.Server.Storage.Components; using Content.Shared.Verbs; using Content.Shared.Movement; using JetBrains.Annotations; using Robust.Server.GameObjects; using Robust.Server.Player; using Robust.Shared.Containers; using Robust.Shared.GameObjects; using Robust.Shared.IoC; using Robust.Shared.Timing; using Robust.Shared.Localization; namespace Content.Server.Storage.EntitySystems { [UsedImplicitly] internal sealed class StorageSystem : EntitySystem { [Dependency] private readonly IGameTiming _gameTiming = default!; private readonly List _sessionCache = new(); /// public override void Initialize() { base.Initialize(); SubscribeLocalEvent(HandleEntityRemovedFromContainer); SubscribeLocalEvent(HandleEntityInsertedIntoContainer); SubscribeLocalEvent(AddToggleOpenVerb); SubscribeLocalEvent(AddOpenUiVerb); SubscribeLocalEvent(OnRelayMovement); } private void OnRelayMovement(EntityUid uid, EntityStorageComponent component, RelayMovementEntityEvent args) { if (EntityManager.HasComponent(args.Entity.Uid)) { if (_gameTiming.CurTime < component.LastInternalOpenAttempt + EntityStorageComponent.InternalOpenAttemptDelay) { return; } component.LastInternalOpenAttempt = _gameTiming.CurTime; component.TryOpenStorage(args.Entity); } } /// public override void Update(float frameTime) { foreach (var component in EntityManager.EntityQuery()) { CheckSubscribedEntities(component); } } private void AddToggleOpenVerb(EntityUid uid, EntityStorageComponent component, GetInteractionVerbsEvent args) { if (!args.CanAccess || !args.CanInteract) return; if (!component.CanOpen(args.User, silent: true)) return; Verb verb = new(); if (component.Open) { verb.Text = Loc.GetString("verb-common-close"); verb.IconTexture = "/Textures/Interface/VerbIcons/close.svg.192dpi.png"; } else { verb.Text = Loc.GetString("verb-common-open"); verb.IconTexture = "/Textures/Interface/VerbIcons/open.svg.192dpi.png"; } verb.Act = () => component.ToggleOpen(args.User); args.Verbs.Add(verb); } private void AddOpenUiVerb(EntityUid uid, ServerStorageComponent component, GetActivationVerbsEvent args) { if (!args.CanAccess || !args.CanInteract) return; if (EntityManager.TryGetComponent(uid, out LockComponent? lockComponent) && lockComponent.Locked) return; // Get the session for the user var session = args.User.GetComponentOrNull()?.PlayerSession; if (session == null) return; // Does this player currently have the storage UI open? var uiOpen = component.SubscribedSessions.Contains(session); Verb verb = new(); verb.Act = () => component.OpenStorageUI(args.User); if (uiOpen) { verb.Text = Loc.GetString("verb-common-close-ui"); verb.IconTexture = "/Textures/Interface/VerbIcons/close.svg.192dpi.png"; } else { verb.Text = Loc.GetString("verb-common-open-ui"); verb.IconTexture = "/Textures/Interface/VerbIcons/open.svg.192dpi.png"; } args.Verbs.Add(verb); } private static void HandleEntityRemovedFromContainer(EntRemovedFromContainerMessage message) { var oldParentEntity = message.Container.Owner; if (oldParentEntity.TryGetComponent(out ServerStorageComponent? storageComp)) { storageComp.HandleEntityMaybeRemoved(message); } } private static void HandleEntityInsertedIntoContainer(EntInsertedIntoContainerMessage message) { var oldParentEntity = message.Container.Owner; if (oldParentEntity.TryGetComponent(out ServerStorageComponent? storageComp)) { storageComp.HandleEntityMaybeInserted(message); } } private void CheckSubscribedEntities(ServerStorageComponent storageComp) { // We have to cache the set of sessions because Unsubscribe modifies the original. _sessionCache.Clear(); _sessionCache.AddRange(storageComp.SubscribedSessions); if (_sessionCache.Count == 0) return; var storagePos = storageComp.Owner.Transform.WorldPosition; var storageMap = storageComp.Owner.Transform.MapID; foreach (var session in _sessionCache) { var attachedEntity = session.AttachedEntity; // The component manages the set of sessions, so this invalid session should be removed soon. if (attachedEntity == null || !attachedEntity.IsValid()) continue; if (storageMap != attachedEntity.Transform.MapID) continue; var distanceSquared = (storagePos - attachedEntity.Transform.WorldPosition).LengthSquared; if (distanceSquared > InteractionSystem.InteractionRangeSquared) { storageComp.UnsubscribeSession(session); } } } } }