Map-Init guidebook entities. (#14823)
This commit is contained in:
@@ -1,64 +0,0 @@
|
||||
using Content.Server.Storage.Components;
|
||||
using Content.Shared.Storage;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization;
|
||||
using Robust.Shared.Serialization.Manager;
|
||||
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 : ITypeValidator<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);
|
||||
}
|
||||
}
|
||||
@@ -1,47 +0,0 @@
|
||||
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