civ14/Content.Shared/Body/Prototypes/BodyPrototypeSerializer.cs

using System.Linq;
using Content.Shared.Body.Organ;
using Content.Shared.Prototypes;
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.Markdown.Value;
using Robust.Shared.Serialization.TypeSerializers.Interfaces;

namespace Content.Shared.Body.Prototypes;

[TypeSerializer]
public sealed class BodyPrototypeSerializer : ITypeReader<BodyPrototype, MappingDataNode>
{
    private (ValidationNode Node, List<string> Connections) ValidateSlot(MappingDataNode slot, IDependencyCollection dependencies)
    {
        var nodes = new List<ValidationNode>();
        var prototypes = dependencies.Resolve<IPrototypeManager>();
        var factory = dependencies.Resolve<IComponentFactory>();

        var connections = new List<string>();
        if (slot.TryGet("connections", out SequenceDataNode? connectionsNode))
        {
            foreach (var node in connectionsNode)
            {
                if (node is not ValueDataNode connection)
                {
                    nodes.Add(new ErrorNode(node, $"Connection is not a value data node"));
                    continue;
                }

                connections.Add(connection.Value);
            }
        }

        if (slot.TryGet("organs", out MappingDataNode? organsNode))
        {
            foreach (var (key, value) in organsNode)
            {
                if (key is not ValueDataNode)
                {
                    nodes.Add(new ErrorNode(key, $"Key is not a value data node"));
                    continue;
                }

                if (value is not ValueDataNode organ)
                {
                    nodes.Add(new ErrorNode(value, $"Value is not a value data node"));
                    continue;
                }

                if (!prototypes.TryIndex(organ.Value, out EntityPrototype? organPrototype))
                {
                    nodes.Add(new ErrorNode(value, $"No organ entity prototype found with id {organ.Value}"));
                    continue;
                }

                if (!organPrototype.HasComponent<OrganComponent>(factory))
                {
                    nodes.Add(new ErrorNode(value, $"Organ {organ.Value} does not have a body component"));
                }
            }
        }

        var validation = new ValidatedSequenceNode(nodes);
        return (validation, connections);
    }

    public ValidationNode Validate(ISerializationManager serializationManager, MappingDataNode node,
        IDependencyCollection dependencies, ISerializationContext? context = null)
    {
        var nodes = new List<ValidationNode>();

        if (!node.TryGet("root", out ValueDataNode? root))
            nodes.Add(new ErrorNode(node, $"No root value data node found"));

        if (!node.TryGet("slots", out MappingDataNode? slots))
        {
            nodes.Add(new ErrorNode(node, $"No slots mapping data node found"));
        }
        else if (root != null)
        {
            if (!slots.TryGet(root.Value, out MappingDataNode? _))
            {
                nodes.Add(new ErrorNode(slots, $"No slot found with id {root.Value}"));
                return new ValidatedSequenceNode(nodes);
            }

            foreach (var (key, value) in slots)
            {
                if (key is not ValueDataNode)
                {
                    nodes.Add(new ErrorNode(key, $"Key is not a value data node"));
                    continue;
                }

                if (value is not MappingDataNode slot)
                {
                    nodes.Add(new ErrorNode(value, $"Slot is not a mapping data node"));
                    continue;
                }

                var result = ValidateSlot(slot, dependencies);
                nodes.Add(result.Node);

                foreach (var connection in result.Connections)
                {
                    if (!slots.TryGet(connection, out MappingDataNode? _))
                        nodes.Add(new ErrorNode(slots, $"No slot found with id {connection}"));
                }
            }
        }

        return new ValidatedSequenceNode(nodes);
    }

    public BodyPrototype Read(ISerializationManager serializationManager, MappingDataNode node,
        IDependencyCollection dependencies,
        SerializationHookContext hookCtx, ISerializationContext? context = null,
        ISerializationManager.InstantiationDelegate<BodyPrototype>? instanceProvider = null)
    {
        var id = node.Get<ValueDataNode>("id").Value;
        var name = node.Get<ValueDataNode>("name").Value;
        var root = node.Get<ValueDataNode>("root").Value;
        var slotNodes = node.Get<MappingDataNode>("slots");
        var allConnections = new Dictionary<string, (string? Part, HashSet<string>? Connections, Dictionary<string, string>? Organs)>();

        foreach (var (keyNode, valueNode) in slotNodes)
        {
            var slotId = ((ValueDataNode) keyNode).Value;
            var slot = ((MappingDataNode) valueNode);

            string? part = null;
            if (slot.TryGet<ValueDataNode>("part", out var value))
            {
                part = value.Value;
            }

            HashSet<string>? connections = null;
            if (slot.TryGet("connections", out SequenceDataNode? slotConnectionsNode))
            {
                connections = new HashSet<string>();

                foreach (var connection in slotConnectionsNode.Cast<ValueDataNode>())
                {
                    connections.Add(connection.Value);
                }
            }

            Dictionary<string, string>? organs = null;
            if (slot.TryGet("organs", out MappingDataNode? slotOrgansNode))
            {
                organs = new Dictionary<string, string>();

                foreach (var (organKeyNode, organValueNode) in slotOrgansNode)
                {
                    organs.Add(((ValueDataNode) organKeyNode).Value, ((ValueDataNode) organValueNode).Value);
                }
            }

            allConnections.Add(slotId, (part, connections, organs));
        }

        foreach (var (slotId, (_, connections, _)) in allConnections)
        {
            if (connections == null)
                continue;

            foreach (var connection in connections)
            {
                var other = allConnections[connection];
                other.Connections ??= new HashSet<string>();
                other.Connections.Add(slotId);
                allConnections[connection] = other;
            }
        }

        var slots = new Dictionary<string, BodyPrototypeSlot>();

        foreach (var (slotId, (part, connections, organs)) in allConnections)
        {
            var slot = new BodyPrototypeSlot(part, connections ?? new HashSet<string>(), organs ?? new Dictionary<string, string>());
            slots.Add(slotId, slot);
        }

        return new BodyPrototype(id, name, root, slots);
    }
}