Add pick random system (#15409)
Co-authored-by: deltanedas <@deltanedas:kde.org>
This commit is contained in:
61
Content.Server/Storage/EntitySystems/PickRandomSystem.cs
Normal file
61
Content.Server/Storage/EntitySystems/PickRandomSystem.cs
Normal file
@@ -0,0 +1,61 @@
|
||||
using Content.Server.Storage.Components;
|
||||
using Content.Shared.Database;
|
||||
using Content.Shared.Hands.EntitySystems;
|
||||
using Content.Shared.Verbs;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.Random;
|
||||
using System.Linq;
|
||||
|
||||
namespace Content.Server.Storage.EntitySystems;
|
||||
|
||||
// TODO: move this to shared for verb prediction if/when storage is in shared
|
||||
public sealed class PickRandomSystem : EntitySystem
|
||||
{
|
||||
[Dependency] private readonly SharedContainerSystem _container = default!;
|
||||
[Dependency] private readonly SharedHandsSystem _hands = default!;
|
||||
[Dependency] private readonly IRobustRandom _random = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
SubscribeLocalEvent<PickRandomComponent, GetVerbsEvent<AlternativeVerb>>(OnGetAlternativeVerbs);
|
||||
}
|
||||
|
||||
private void OnGetAlternativeVerbs(EntityUid uid, PickRandomComponent comp, GetVerbsEvent<AlternativeVerb> args)
|
||||
{
|
||||
if (!args.CanAccess || !args.CanInteract || !TryComp<ServerStorageComponent>(uid, out var storage))
|
||||
return;
|
||||
|
||||
var user = args.User;
|
||||
|
||||
// alt-click / alt-z to pick an item
|
||||
args.Verbs.Add(new AlternativeVerb
|
||||
{
|
||||
Act = (() => {
|
||||
TryPick(uid, comp, storage, user);
|
||||
}),
|
||||
Impact = LogImpact.Low,
|
||||
Text = Loc.GetString(comp.VerbText),
|
||||
Disabled = !(storage.StoredEntities?.Any(item => comp.Whitelist?.IsValid(item, EntityManager) ?? true) ?? false),
|
||||
Message = Loc.GetString(comp.EmptyText, ("storage", uid))
|
||||
});
|
||||
}
|
||||
|
||||
private void TryPick(EntityUid uid, PickRandomComponent comp, ServerStorageComponent storage, EntityUid user)
|
||||
{
|
||||
if (storage.StoredEntities == null)
|
||||
return;
|
||||
|
||||
var entities = storage.StoredEntities.Where(item => comp.Whitelist?.IsValid(item, EntityManager) ?? true);
|
||||
if (!entities.Any())
|
||||
return;
|
||||
|
||||
var picked = _random.Pick(entities.ToList());
|
||||
// if it fails to go into a hand of the user, will be on the storage
|
||||
_container.AttachParentToContainerOrGrid(Transform(picked));
|
||||
|
||||
// TODO: try to put in hands, failing that put it on the storage
|
||||
_hands.TryPickupAnyHand(user, picked);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user