Files
tbd-station-14/Content.Shared/GameObjects/Components/Materials/MaterialComponent.cs
DrSmugleaf 6f012cb9ad Enable nullability in Content.Shared (#3626)
* Enable nullability in Content.Shared

* Fix null errors in server

* aye github i swear on me mom
2021-03-15 21:55:49 +01:00

71 lines
2.0 KiB
C#

using System.Collections.Generic;
using Content.Shared.Materials;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Prototypes;
using Robust.Shared.Reflection;
using Robust.Shared.Serialization;
using Robust.Shared.Serialization.Manager.Attributes;
namespace Content.Shared.GameObjects.Components.Materials
{
/// <summary>
/// Component to store data such as "this object is made out of steel".
/// This is not a storage system for say smelteries.
/// </summary>
[RegisterComponent]
public class MaterialComponent : Component, ISerializationHooks
{
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
public const string SerializationCache = "mat";
public override string Name => "Material";
[DataField("materials")] private List<MaterialDataEntry> _materials = new();
public IEnumerable<KeyValuePair<object, MaterialPrototype>> MaterialTypes
{
get
{
foreach (var entry in _materials)
{
var prototype = _prototypeManager.Index<MaterialPrototype>(entry.Value);
yield return new KeyValuePair<object, MaterialPrototype>(entry.Key, prototype);
}
}
}
[DataDefinition]
public class MaterialDataEntry : ISerializationHooks
{
public object Key = default!;
[DataField("key", required: true)]
public string StringKey = default!;
[DataField("mat", required: true)]
public string Value = default!;
void ISerializationHooks.AfterDeserialization()
{
var refl = IoCManager.Resolve<IReflectionManager>();
if (refl.TryParseEnumReference(StringKey, out var @enum))
{
Key = @enum;
return;
}
Key = StringKey;
}
}
}
public enum MaterialKeys
{
Stack,
}
}