93 lines
3.0 KiB
C#
93 lines
3.0 KiB
C#
using System.Linq;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Prying.Components;
|
|
using Content.Shared.Tools.Components;
|
|
|
|
namespace Content.Shared.Tools.Systems;
|
|
|
|
public abstract partial class SharedToolSystem : EntitySystem
|
|
{
|
|
public void InitializeMultipleTool()
|
|
{
|
|
SubscribeLocalEvent<MultipleToolComponent, ComponentStartup>(OnMultipleToolStartup);
|
|
SubscribeLocalEvent<MultipleToolComponent, ActivateInWorldEvent>(OnMultipleToolActivated);
|
|
SubscribeLocalEvent<MultipleToolComponent, AfterAutoHandleStateEvent>(OnMultipleToolHandleState);
|
|
}
|
|
|
|
private void OnMultipleToolHandleState(EntityUid uid, MultipleToolComponent component, ref AfterAutoHandleStateEvent args)
|
|
{
|
|
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);
|
|
}
|
|
|
|
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.UseSound;
|
|
tool.Qualities = current.Behavior;
|
|
|
|
// TODO: Replace this with a better solution later
|
|
if (TryComp<PryingComponent>(uid, out var pcomp))
|
|
{
|
|
if (current.Behavior.Contains("Prying"))
|
|
{
|
|
pcomp.Enabled = true;
|
|
}
|
|
else
|
|
{
|
|
pcomp.Enabled = false;
|
|
}
|
|
}
|
|
|
|
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);
|
|
}
|
|
}
|
|
|