75 lines
3.0 KiB
C#
75 lines
3.0 KiB
C#
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Content.Server.Construction.Components;
|
|
using Content.Shared.Construction;
|
|
using JetBrains.Annotations;
|
|
using Robust.Server.Containers;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Serialization.Manager.Attributes;
|
|
|
|
namespace Content.Server.Construction.Completions
|
|
{
|
|
[UsedImplicitly]
|
|
[DataDefinition]
|
|
public class BuildComputer : IGraphAction
|
|
{
|
|
[DataField("container")] public string Container { get; private set; } = string.Empty;
|
|
|
|
public void PerformAction(EntityUid uid, EntityUid? userUid, IEntityManager entityManager)
|
|
{
|
|
if (!entityManager.TryGetComponent(uid, out ContainerManagerComponent? containerManager))
|
|
{
|
|
Logger.Warning($"Computer entity {uid} did not have a container manager! Aborting build computer action.");
|
|
return;
|
|
}
|
|
|
|
var containerSystem = entityManager.EntitySysManager.GetEntitySystem<ContainerSystem>();
|
|
|
|
if (!containerSystem.TryGetContainer(uid, Container, out var container, containerManager))
|
|
{
|
|
Logger.Warning($"Computer entity {uid} did not have the specified '{Container}' container! Aborting build computer action.");
|
|
return;
|
|
}
|
|
|
|
if (container.ContainedEntities.Count != 1)
|
|
{
|
|
Logger.Warning($"Computer entity {uid} did not have exactly one item in the specified '{Container}' container! Aborting build computer action.");
|
|
}
|
|
|
|
var board = container.ContainedEntities[0];
|
|
|
|
if (!board.TryGetComponent(out ComputerBoardComponent? boardComponent))
|
|
{
|
|
Logger.Warning($"Computer entity {uid} had an invalid entity in container \"{Container}\"! Aborting build computer action.");
|
|
return;
|
|
}
|
|
|
|
container.Remove(board);
|
|
|
|
var transform = entityManager.GetComponent<TransformComponent>(uid);
|
|
var computer = entityManager.SpawnEntity(boardComponent.Prototype, transform.Coordinates);
|
|
computer.Transform.LocalRotation = transform.LocalRotation;
|
|
|
|
var computerContainer = containerSystem.EnsureContainer<Container>(computer.Uid, Container);
|
|
|
|
// In case it already existed and there are any entities inside the container, delete them.
|
|
foreach (var ent in computerContainer.ContainedEntities.ToArray())
|
|
{
|
|
computerContainer.ForceRemove(ent);
|
|
IoCManager.Resolve<IEntityManager>().DeleteEntity(ent.Uid);
|
|
}
|
|
|
|
computerContainer.Insert(board);
|
|
|
|
// We only add this container. If some construction needs to take other containers into account, fix this.
|
|
entityManager.EntitySysManager.GetEntitySystem<ConstructionSystem>().AddContainer(computer.Uid, Container);
|
|
|
|
// Delete the original entity.
|
|
entityManager.DeleteEntity(uid);
|
|
}
|
|
}
|
|
}
|