94 lines
3.2 KiB
C#
94 lines
3.2 KiB
C#
using System.Linq;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Tools.Components;
|
|
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Shared.Tools;
|
|
|
|
public abstract class SharedToolSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _protoMan = default!;
|
|
[Dependency] private readonly SharedAudioSystem _audioSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<MultipleToolComponent, ComponentStartup>(OnMultipleToolStartup);
|
|
SubscribeLocalEvent<MultipleToolComponent, ActivateInWorldEvent>(OnMultipleToolActivated);
|
|
SubscribeLocalEvent<MultipleToolComponent, ComponentGetState>(OnMultipleToolGetState);
|
|
SubscribeLocalEvent<MultipleToolComponent, ComponentHandleState>(OnMultipleToolHandleState);
|
|
}
|
|
|
|
private void OnMultipleToolHandleState(EntityUid uid, MultipleToolComponent component, ref ComponentHandleState args)
|
|
{
|
|
if (args.Current is not MultipleToolComponentState state)
|
|
return;
|
|
|
|
component.CurrentEntry = state.Selected;
|
|
SetMultipleTool(uid, component);
|
|
}
|
|
|
|
private void OnMultipleToolStartup(EntityUid uid, MultipleToolComponent multiple, ComponentStartup args)
|
|
{
|
|
// Only set the multiple tool if we have a tool component.
|
|
if(EntityManager.TryGetComponent(uid, out ToolComponent? tool))
|
|
SetMultipleTool(uid, multiple, tool);
|
|
}
|
|
|
|
private void OnMultipleToolActivated(EntityUid uid, MultipleToolComponent multiple, ActivateInWorldEvent args)
|
|
{
|
|
if (args.Handled)
|
|
return;
|
|
|
|
args.Handled = CycleMultipleTool(uid, multiple, args.User);
|
|
}
|
|
|
|
private void OnMultipleToolGetState(EntityUid uid, MultipleToolComponent multiple, ref ComponentGetState args)
|
|
{
|
|
args.State = new MultipleToolComponentState(multiple.CurrentEntry);
|
|
}
|
|
|
|
public bool CycleMultipleTool(EntityUid uid, MultipleToolComponent? multiple = null, EntityUid? user = null)
|
|
{
|
|
if (!Resolve(uid, ref multiple))
|
|
return false;
|
|
|
|
if (multiple.Entries.Length == 0)
|
|
return false;
|
|
|
|
multiple.CurrentEntry = (uint) ((multiple.CurrentEntry + 1) % multiple.Entries.Length);
|
|
SetMultipleTool(uid, multiple, playSound: true, user: user);
|
|
|
|
return true;
|
|
}
|
|
|
|
public virtual void SetMultipleTool(EntityUid uid,
|
|
MultipleToolComponent? multiple = null,
|
|
ToolComponent? tool = null,
|
|
bool playSound = false,
|
|
EntityUid? user = null)
|
|
{
|
|
if (!Resolve(uid, ref multiple, ref tool))
|
|
return;
|
|
|
|
Dirty(multiple);
|
|
|
|
if (multiple.Entries.Length <= multiple.CurrentEntry)
|
|
{
|
|
multiple.CurrentQualityName = Loc.GetString("multiple-tool-component-no-behavior");
|
|
return;
|
|
}
|
|
|
|
var current = multiple.Entries[multiple.CurrentEntry];
|
|
tool.UseSound = current.Sound;
|
|
tool.Qualities = current.Behavior;
|
|
|
|
if (playSound && current.ChangeSound != null)
|
|
_audioSystem.PlayPredicted(current.ChangeSound, uid, user);
|
|
|
|
if (_protoMan.TryIndex(current.Behavior.First(), out ToolQualityPrototype? quality))
|
|
multiple.CurrentQualityName = Loc.GetString(quality.Name);
|
|
}
|
|
}
|
|
|