Files
tbd-station-14/Content.Server/Construction/Completions/MoveContainer.cs
2022-02-16 18:23:23 +11:00

38 lines
1.4 KiB
C#

using System.Linq;
using System.Threading.Tasks;
using Content.Shared.Construction;
using JetBrains.Annotations;
using Robust.Server.Containers;
using Robust.Shared.Containers;
using Robust.Shared.GameObjects;
using Robust.Shared.Serialization.Manager.Attributes;
namespace Content.Server.Construction.Completions
{
[UsedImplicitly]
[DataDefinition]
public sealed class MoveContainer : IGraphAction
{
[DataField("from")] public string? FromContainer { get; } = null;
[DataField("to")] public string? ToContainer { get; } = null;
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 (from.Remove(contained))
to.Insert(contained);
}
}
}
}