using System.Diagnostics.CodeAnalysis; using System.Linq; using Content.Shared.Administration.Logs; using Content.Shared.Database; using Content.Shared.Hands.EntitySystems; using Content.Shared.Interaction; using Content.Shared.Storage.Components; using Robust.Shared.Containers; using Robust.Shared.GameStates; using Robust.Shared.Network; using Robust.Shared.Timing; namespace Content.Shared.Storage.EntitySystems; /// /// This handles /// public sealed class BinSystem : EntitySystem { [Dependency] private readonly IGameTiming _timing = default!; [Dependency] private readonly INetManager _net = default!; [Dependency] private readonly ISharedAdminLogManager _admin = default!; [Dependency] private readonly SharedContainerSystem _container = default!; [Dependency] private readonly SharedHandsSystem _hands = default!; public const string BinContainerId = "bin-container"; /// public override void Initialize() { SubscribeLocalEvent(OnGetState); SubscribeLocalEvent(OnHandleState); SubscribeLocalEvent(OnStartup); SubscribeLocalEvent(OnMapInit); SubscribeLocalEvent(OnEntRemoved); SubscribeLocalEvent(OnInteractHand); SubscribeLocalEvent(OnAfterInteractUsing); } private void OnGetState(EntityUid uid, BinComponent component, ref ComponentGetState args) { args.State = new BinComponentState(component.Items, component.Whitelist, component.MaxItems); } private void OnHandleState(EntityUid uid, BinComponent component, ref ComponentHandleState args) { if (args.Current is not BinComponentState state) return; component.Items = new List(state.Items); component.Whitelist = state.Whitelist; component.MaxItems = state.MaxItems; } private void OnStartup(EntityUid uid, BinComponent component, ComponentStartup args) { component.ItemContainer = _container.EnsureContainer(uid, BinContainerId); } private void OnMapInit(EntityUid uid, BinComponent component, MapInitEvent args) { // don't spawn on the client. if (_net.IsClient) return; var xform = Transform(uid); foreach (var id in component.InitialContents) { var ent = Spawn(id, xform.Coordinates); if (!TryInsertIntoBin(uid, ent, component)) { Logger.Error($"Entity {ToPrettyString(ent)} was unable to be initialized into bin {ToPrettyString(uid)}"); return; } } } private void OnEntRemoved(EntityUid uid, BinComponent component, EntRemovedFromContainerMessage args) { component.Items.Remove(args.Entity); } private void OnInteractHand(EntityUid uid, BinComponent component, InteractHandEvent args) { if (args.Handled || !_timing.IsFirstTimePredicted) return; EntityUid? toGrab = component.Items.LastOrDefault(); if (!TryRemoveFromBin(uid, toGrab, component)) return; _hands.TryPickupAnyHand(args.User, toGrab.Value); _admin.Add(LogType.Pickup, LogImpact.Low, $"{ToPrettyString(uid):player} removed {ToPrettyString(toGrab.Value)} from bin {ToPrettyString(uid)}."); args.Handled = true; } private void OnAfterInteractUsing(EntityUid uid, BinComponent component, AfterInteractUsingEvent args) { if (args.Handled || !args.CanReach) return; if (!_timing.IsFirstTimePredicted) return; if (!TryInsertIntoBin(uid, args.Used, component)) return; _admin.Add(LogType.Pickup, LogImpact.Low, $"{ToPrettyString(uid):player} inserted {ToPrettyString(args.User)} into bin {ToPrettyString(uid)}."); args.Handled = true; } /// /// Inserts an entity at the top of the bin /// /// /// /// /// public bool TryInsertIntoBin(EntityUid uid, EntityUid toInsert, BinComponent? component = null) { if (!Resolve(uid, ref component)) return false; if (component.Items.Count >= component.MaxItems) return false; if (component.Whitelist != null && !component.Whitelist.IsValid(toInsert)) return false; component.ItemContainer.Insert(toInsert); component.Items.Add(toInsert); Dirty(component); return true; } /// /// Tries to remove an entity from the top of the bin. /// /// /// /// /// public bool TryRemoveFromBin(EntityUid uid, [NotNullWhen(true)] EntityUid? toRemove, BinComponent? component = null) { if (!Resolve(uid, ref component)) return false; if (toRemove != component.Items.Last()) return false; if (!component.ItemContainer.Remove(toRemove.Value)) return false; component.Items.Remove(toRemove.Value); Dirty(component); return true; } }