62 lines
2.5 KiB
C#
62 lines
2.5 KiB
C#
using System.Linq;
|
|
using System.Text.Json;
|
|
using System.Text.Json.Nodes;
|
|
using Robust.Shared.Serialization.Markdown;
|
|
using Robust.Shared.Serialization.Markdown.Mapping;
|
|
using Robust.Shared.Serialization.Markdown.Sequence;
|
|
using Robust.Shared.Serialization.Markdown.Value;
|
|
using YamlDotNet.RepresentationModel;
|
|
|
|
namespace Content.Server.Database;
|
|
|
|
public static class DataNodeJsonExtensions
|
|
{
|
|
private static JsonNode ToJsonNode(this MappingDataNode node)
|
|
{
|
|
return new JsonObject(node.Children.Select(kvp => new KeyValuePair<string, JsonNode?>(kvp.Key, kvp.Value.ToJsonNode())));
|
|
}
|
|
|
|
private static JsonNode ToJsonNode(this SequenceDataNode node)
|
|
{
|
|
return new JsonArray(node.Select(ToJsonNode).ToArray());
|
|
}
|
|
|
|
public static JsonNode? ToJsonNode(this DataNode node)
|
|
{
|
|
return node switch
|
|
{
|
|
ValueDataNode valueDataNode => JsonValue.Create(valueDataNode.IsNull ? null : valueDataNode.Value),
|
|
MappingDataNode mappingDataNode => mappingDataNode.ToJsonNode(),
|
|
SequenceDataNode sequenceNode => sequenceNode.ToJsonNode(),
|
|
_ => throw new ArgumentOutOfRangeException(nameof(node))
|
|
};
|
|
}
|
|
|
|
public static DataNode ToDataNode(this JsonElement element)
|
|
{
|
|
return element.ValueKind switch
|
|
{
|
|
JsonValueKind.Object => new MappingDataNode(element.EnumerateObject().ToDictionary(kvp => kvp.Name, kvp => kvp.Value.ToDataNode())),
|
|
JsonValueKind.Array => new SequenceDataNode(element.EnumerateArray().Select(item => item.ToDataNode()).ToList()),
|
|
JsonValueKind.Number => new ValueDataNode(element.GetRawText()),
|
|
JsonValueKind.String => new ValueDataNode(element.GetString()),
|
|
JsonValueKind.True => new ValueDataNode("true"),
|
|
JsonValueKind.False => new ValueDataNode("false"),
|
|
JsonValueKind.Null => new ValueDataNode("null"),
|
|
_ => throw new ArgumentOutOfRangeException(nameof(element)),
|
|
};
|
|
}
|
|
|
|
public static DataNode ToDataNode(this JsonNode? node)
|
|
{
|
|
return node switch
|
|
{
|
|
null => ValueDataNode.Null(),
|
|
JsonValue value => new ValueDataNode(value.GetValue<string>()),
|
|
JsonArray array => new SequenceDataNode(array.Select(item => item.ToDataNode()).ToList()),
|
|
JsonObject obj => new MappingDataNode(obj.ToDictionary(kvp => kvp.Key, kvp => kvp.Value.ToDataNode())),
|
|
_ => throw new ArgumentOutOfRangeException(nameof(node))
|
|
};
|
|
}
|
|
}
|