Move board spawning out of DoorSystem (#11772)
This commit is contained in:
95
Content.Server/Containers/ContainerFillComponent.cs
Normal file
95
Content.Server/Containers/ContainerFillComponent.cs
Normal file
@@ -0,0 +1,95 @@
|
||||
using Content.Server.Storage.Components;
|
||||
using Content.Shared.Storage;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization.Manager;
|
||||
using Robust.Shared.Serialization.Markdown;
|
||||
using Robust.Shared.Serialization.Markdown.Mapping;
|
||||
using Robust.Shared.Serialization.Markdown.Sequence;
|
||||
using Robust.Shared.Serialization.Markdown.Validation;
|
||||
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.List;
|
||||
using Robust.Shared.Serialization.TypeSerializers.Interfaces;
|
||||
|
||||
namespace Content.Server.Containers;
|
||||
|
||||
/// <summary>
|
||||
/// Component for spawning entity prototypes into containers on map init.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Unlike <see cref="StorageFillComponent"/> this is deterministic and supports arbitrary containers. While this
|
||||
/// could maybe be merged with that component, it would require significant changes to <see
|
||||
/// cref="EntitySpawnCollection.GetSpawns"/>, which is also used by several other systems.
|
||||
/// </remarks>
|
||||
[RegisterComponent]
|
||||
public sealed class ContainerFillComponent : Component
|
||||
{
|
||||
[DataField("containers", customTypeSerializer:typeof(ContainerFillSerializer))]
|
||||
public readonly Dictionary<string, List<string>> Containers = new();
|
||||
|
||||
/// <summary>
|
||||
/// If true, entities spawned via the construction system will not have entities spawned into containers managed
|
||||
/// by the construction system.
|
||||
/// </summary>
|
||||
[DataField("ignoreConstructionSpawn")]
|
||||
public bool IgnoreConstructionSpawn = true;
|
||||
}
|
||||
|
||||
// all of this exists just to validate prototype ids.
|
||||
// it would be nice if you could specify only a type validator and not have to re-implement everything else.
|
||||
// or a dictionary serializer that accepts a custom type serializer for the dictionary values
|
||||
public sealed class ContainerFillSerializer : ITypeSerializer<Dictionary<string, List<string>>, MappingDataNode>
|
||||
{
|
||||
private static PrototypeIdListSerializer<EntityPrototype> ListSerializer => new();
|
||||
|
||||
public ValidationNode Validate(
|
||||
ISerializationManager serializationManager,
|
||||
MappingDataNode node,
|
||||
IDependencyCollection dependencies,
|
||||
ISerializationContext? context = null)
|
||||
{
|
||||
var mapping = new Dictionary<ValidationNode, ValidationNode>();
|
||||
|
||||
foreach (var (key, val) in node.Children)
|
||||
{
|
||||
var keyVal = serializationManager.ValidateNode<string>(key, context);
|
||||
|
||||
var listVal = (val is SequenceDataNode seq)
|
||||
? ListSerializer.Validate(serializationManager, seq, dependencies, context)
|
||||
: new ErrorNode(val, "ContainerFillComponent prototypes must be a sequence/list");
|
||||
|
||||
mapping.Add(keyVal, listVal);
|
||||
}
|
||||
|
||||
return new ValidatedMappingNode(mapping);
|
||||
}
|
||||
|
||||
public Dictionary<string, List<string>> Copy(
|
||||
ISerializationManager serializationManager,
|
||||
Dictionary<string, List<string>> source,
|
||||
Dictionary<string, List<string>> target,
|
||||
bool skipHook,
|
||||
ISerializationContext? context = null)
|
||||
{
|
||||
serializationManager.Copy(source, ref target, context, skipHook);
|
||||
return target;
|
||||
}
|
||||
|
||||
public Dictionary<string, List<string>> Read(
|
||||
ISerializationManager serializationManager,
|
||||
MappingDataNode node,
|
||||
IDependencyCollection dependencies,
|
||||
bool skipHook,
|
||||
ISerializationContext? context = null,
|
||||
Dictionary<string, List<string>>? value = null)
|
||||
{
|
||||
return serializationManager.Read(node, context, skipHook, value);
|
||||
}
|
||||
|
||||
public DataNode Write(ISerializationManager serializationManager,
|
||||
Dictionary<string, List<string>> value,
|
||||
IDependencyCollection dependencies,
|
||||
bool alwaysWrite = false,
|
||||
ISerializationContext? context = null)
|
||||
{
|
||||
return serializationManager.WriteValue(value, alwaysWrite, context);
|
||||
}
|
||||
}
|
||||
47
Content.Server/Containers/ContainerFillSystem.cs
Normal file
47
Content.Server/Containers/ContainerFillSystem.cs
Normal file
@@ -0,0 +1,47 @@
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.Map;
|
||||
|
||||
namespace Content.Server.Containers;
|
||||
|
||||
public sealed class ContainerFillSystem : EntitySystem
|
||||
{
|
||||
[Dependency] private readonly SharedContainerSystem _containerSystem = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
SubscribeLocalEvent<ContainerFillComponent, MapInitEvent>(OnMapInit);
|
||||
}
|
||||
|
||||
private void OnMapInit(EntityUid uid, ContainerFillComponent component, MapInitEvent args)
|
||||
{
|
||||
if (!TryComp(uid, out ContainerManagerComponent? containerComp))
|
||||
{
|
||||
Logger.Error($"Entity {ToPrettyString(uid)} with a {nameof(ContainerFillComponent)} has no {nameof(ContainerManagerComponent)}.");
|
||||
return;
|
||||
}
|
||||
|
||||
var xform = Transform(uid);
|
||||
var coords = new EntityCoordinates(uid, Vector2.Zero);
|
||||
|
||||
foreach (var (contaienrId, prototypes) in component.Containers)
|
||||
{
|
||||
if (!_containerSystem.TryGetContainer(uid, contaienrId, out var container, containerComp))
|
||||
{
|
||||
Logger.Error($"Entity {ToPrettyString(uid)} with a {nameof(ContainerFillComponent)} is missing a container ({contaienrId}).");
|
||||
continue;
|
||||
}
|
||||
|
||||
foreach (var proto in prototypes)
|
||||
{
|
||||
var ent = Spawn(proto, coords);
|
||||
if (!container.Insert(ent, EntityManager, null, xform))
|
||||
{
|
||||
Logger.Error($"Entity {ToPrettyString(uid)} with a {nameof(ContainerFillComponent)} failed to insert an entity: {ToPrettyString(ent)}.");
|
||||
Transform(ent).AttachToGridOrMap();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user