71 lines
2.6 KiB
C#
71 lines
2.6 KiB
C#
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using Robust.Shared.IoC;
|
||
using Robust.Shared.Log;
|
||
using Robust.Shared.Prototypes;
|
||
|
||
namespace Content.Shared.Actions
|
||
{
|
||
/// <summary>
|
||
/// Provides access to all configured actions by action type.
|
||
/// </summary>
|
||
public class ActionManager
|
||
{
|
||
[Dependency]
|
||
private readonly IPrototypeManager _prototypeManager = default!;
|
||
|
||
private Dictionary<ActionType, ActionPrototype> _typeToAction;
|
||
private Dictionary<ItemActionType, ItemActionPrototype> _typeToItemAction;
|
||
|
||
public void Initialize()
|
||
{
|
||
_typeToAction = new Dictionary<ActionType, ActionPrototype>();
|
||
foreach (var action in _prototypeManager.EnumeratePrototypes<ActionPrototype>())
|
||
{
|
||
if (!_typeToAction.TryAdd(action.ActionType, action))
|
||
{
|
||
Logger.ErrorS("action",
|
||
"Found action with duplicate actionType {0} - all actions must have" +
|
||
" a unique actionType, this one will be skipped", action.ActionType);
|
||
}
|
||
}
|
||
|
||
_typeToItemAction = new Dictionary<ItemActionType, ItemActionPrototype>();
|
||
foreach (var action in _prototypeManager.EnumeratePrototypes<ItemActionPrototype>())
|
||
{
|
||
if (!_typeToItemAction.TryAdd(action.ActionType, action))
|
||
{
|
||
Logger.ErrorS("action",
|
||
"Found itemAction with duplicate actionType {0} - all actions must have" +
|
||
" a unique actionType, this one will be skipped", action.ActionType);
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <returns>all action prototypes of all types</returns>
|
||
public IEnumerable<BaseActionPrototype> EnumerateActions()
|
||
{
|
||
return _typeToAction.Values.Concat<BaseActionPrototype>(_typeToItemAction.Values);
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Tries to get the action of the indicated type
|
||
/// </summary>
|
||
/// <returns>true if found</returns>
|
||
public bool TryGet(ActionType actionType, out ActionPrototype action)
|
||
{
|
||
return _typeToAction.TryGetValue(actionType, out action);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Tries to get the item action of the indicated type
|
||
/// </summary>
|
||
/// <returns>true if found</returns>
|
||
public bool TryGet(ItemActionType actionType, out ItemActionPrototype action)
|
||
{
|
||
return _typeToItemAction.TryGetValue(actionType, out action);
|
||
}
|
||
}
|
||
}
|