59 lines
2.0 KiB
C#
59 lines
2.0 KiB
C#
using Content.Server.GameObjects.Components.NodeContainer.NodeGroups;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.Interfaces.Reflection;
|
|
using Robust.Shared.IoC;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Reflection;
|
|
|
|
namespace Content.Server.GameObjects.Components.NodeContainer.Nodes
|
|
{
|
|
public interface INodeFactory
|
|
{
|
|
/// <summary>
|
|
/// Performs reflection to associate <see cref="Node"/> implementations with the
|
|
/// string specified in their <see cref="NodeAttribute"/>.
|
|
/// </summary>
|
|
void Initialize();
|
|
|
|
/// <summary>
|
|
/// Returns a new <see cref="Node"/> instance.
|
|
/// </summary>
|
|
Node MakeNode(string nodeName, NodeGroupID groupID, IEntity owner);
|
|
}
|
|
|
|
public class NodeFactory : INodeFactory
|
|
{
|
|
private readonly Dictionary<string, Type> _groupTypes = new Dictionary<string, Type>();
|
|
|
|
#pragma warning disable 649
|
|
[Dependency] private readonly IReflectionManager _reflectionManager;
|
|
[Dependency] private readonly IDynamicTypeFactory _typeFactory;
|
|
#pragma warning restore 649
|
|
|
|
public void Initialize()
|
|
{
|
|
var nodeTypes = _reflectionManager.GetAllChildren<Node>();
|
|
foreach (var nodeType in nodeTypes)
|
|
{
|
|
var att = nodeType.GetCustomAttribute<NodeAttribute>();
|
|
if (att != null)
|
|
{
|
|
_groupTypes.Add(att.Name, nodeType);
|
|
}
|
|
}
|
|
}
|
|
|
|
public Node MakeNode(string nodeName, NodeGroupID groupID, IEntity owner)
|
|
{
|
|
if (_groupTypes.TryGetValue(nodeName, out var type))
|
|
{
|
|
var newNode = _typeFactory.CreateInstance<Node>(type);
|
|
newNode.Initialize(groupID, owner);
|
|
return newNode;
|
|
}
|
|
throw new ArgumentException($"{nodeName} did not have an associated {nameof(Node)}.");
|
|
}
|
|
}
|
|
}
|