Files
tbd-station-14/Content.Server/Construction/Completions/MoveContainer.cs
2023-12-28 01:30:03 -04:00

35 lines
1.3 KiB
C#

using System.Linq;
using Content.Shared.Construction;
using JetBrains.Annotations;
using Robust.Server.Containers;
using Robust.Shared.Containers;
namespace Content.Server.Construction.Completions
{
[UsedImplicitly]
[DataDefinition]
public sealed partial class MoveContainer : IGraphAction
{
[DataField("from")] public string? FromContainer { get; private set; }
[DataField("to")] public string? ToContainer { get; private set; }
public void PerformAction(EntityUid uid, EntityUid? userUid, IEntityManager entityManager)
{
if (string.IsNullOrEmpty(FromContainer) || string.IsNullOrEmpty(ToContainer))
return;
var containerSystem = entityManager.EntitySysManager.GetEntitySystem<ContainerSystem>();
var containerManager = entityManager.EnsureComponent<ContainerManagerComponent>(uid);
var from = containerSystem.EnsureContainer<Container>(uid, FromContainer, containerManager);
var to = containerSystem.EnsureContainer<Container>(uid, ToContainer, containerManager);
foreach (var contained in from.ContainedEntities.ToArray())
{
if (containerSystem.Remove(contained, from))
containerSystem.Insert(contained, to);
}
}
}
}