68 lines
2.3 KiB
C#
68 lines
2.3 KiB
C#
using Content.Server.Popups;
|
|
using Content.Shared.DoAfter;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Mech.Components;
|
|
using Content.Shared.Mech.Equipment.Components;
|
|
|
|
namespace Content.Server.Mech.Systems;
|
|
|
|
/// <summary>
|
|
/// Handles the insertion of mech equipment into mechs.
|
|
/// </summary>
|
|
public sealed class MechEquipmentSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly MechSystem _mech = default!;
|
|
[Dependency] private readonly SharedDoAfterSystem _doAfter = default!;
|
|
[Dependency] private readonly PopupSystem _popup = default!;
|
|
|
|
/// <inheritdoc/>
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<MechEquipmentComponent, AfterInteractEvent>(OnUsed);
|
|
SubscribeLocalEvent<MechEquipmentComponent, InsertEquipmentEvent>(OnInsertEquipment);
|
|
}
|
|
|
|
private void OnUsed(EntityUid uid, MechEquipmentComponent component, AfterInteractEvent args)
|
|
{
|
|
if (args.Handled || !args.CanReach || args.Target == null)
|
|
return;
|
|
|
|
var mech = args.Target.Value;
|
|
if (!TryComp<MechComponent>(mech, out var mechComp))
|
|
return;
|
|
|
|
if (mechComp.Broken)
|
|
return;
|
|
|
|
if (args.User == mechComp.PilotSlot.ContainedEntity)
|
|
return;
|
|
|
|
if (mechComp.EquipmentContainer.ContainedEntities.Count >= mechComp.MaxEquipmentAmount)
|
|
return;
|
|
|
|
if (mechComp.EquipmentWhitelist != null && !mechComp.EquipmentWhitelist.IsValid(args.Used))
|
|
return;
|
|
|
|
_popup.PopupEntity(Loc.GetString("mech-equipment-begin-install", ("item", uid)), mech);
|
|
|
|
var doAfterEventArgs = new DoAfterArgs(EntityManager, args.User, component.InstallDuration, new InsertEquipmentEvent(), uid, target: mech, used: uid)
|
|
{
|
|
BreakOnTargetMove = true,
|
|
BreakOnUserMove = true
|
|
};
|
|
|
|
_doAfter.TryStartDoAfter(doAfterEventArgs);
|
|
}
|
|
|
|
private void OnInsertEquipment(EntityUid uid, MechEquipmentComponent component, InsertEquipmentEvent args)
|
|
{
|
|
if (args.Handled || args.Cancelled || args.Args.Target == null)
|
|
return;
|
|
|
|
_popup.PopupEntity(Loc.GetString("mech-equipment-finish-install", ("item", uid)), args.Args.Target.Value);
|
|
_mech.InsertEquipment(args.Args.Target.Value, uid);
|
|
|
|
args.Handled = true;
|
|
}
|
|
}
|