Fucking big refactor
This commit is contained in:
Ed
333
Content.Shared/_CP14/LockKey/SharedCP14LockKeySystem.cs
Normal file
333
Content.Shared/_CP14/LockKey/SharedCP14LockKeySystem.cs
Normal file
@@ -0,0 +1,333 @@
|
||||
using System.Diagnostics.CodeAnalysis;
|
||||
using Content.Shared._CP14.LockKey.Components;
|
||||
using Content.Shared.Containers.ItemSlots;
|
||||
using Content.Shared.Interaction;
|
||||
using Content.Shared.Lock;
|
||||
using Content.Shared.Popups;
|
||||
using Content.Shared.Storage;
|
||||
using Content.Shared.Verbs;
|
||||
using Robust.Shared.Audio.Systems;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.Random;
|
||||
using Robust.Shared.Serialization;
|
||||
|
||||
namespace Content.Shared._CP14.LockKey;
|
||||
|
||||
/// <summary>
|
||||
///
|
||||
/// </summary>
|
||||
public sealed class SharedCP14LockKeySystem : EntitySystem
|
||||
{
|
||||
|
||||
[Dependency] private readonly ItemSlotsSystem _itemSlots = default!;
|
||||
[Dependency] private readonly SharedAppearanceSystem _appearance = default!;
|
||||
[Dependency] private readonly SharedPopupSystem _popup = default!;
|
||||
[Dependency] private readonly LockSystem _lock = default!;
|
||||
[Dependency] private readonly IRobustRandom _random = default!;
|
||||
[Dependency] private readonly SharedAudioSystem _audio = default!;
|
||||
|
||||
private const int DepthComplexity = 2; //TODO - fix this constant duplication from KeyholeGenerationSystem.cs
|
||||
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
SubscribeLocalEvent<LockComponent, ContainerIsInsertingAttemptEvent>(OnLockInsertAttempt);
|
||||
|
||||
SubscribeLocalEvent<LockComponent, EntInsertedIntoContainerMessage>(OnLockInserted);
|
||||
SubscribeLocalEvent<LockComponent, EntRemovedFromContainerMessage>(OnLockRemoved);
|
||||
|
||||
SubscribeLocalEvent<CP14KeyComponent, AfterInteractEvent>(OnKeyInteract);
|
||||
SubscribeLocalEvent<CP14KeyRingComponent, AfterInteractEvent>(OnKeyRingInteract);
|
||||
SubscribeLocalEvent<CP14KeyComponent, GetVerbsEvent<UtilityVerb>>(OnKeyToLockVerb);
|
||||
SubscribeLocalEvent<CP14LockpickComponent, GetVerbsEvent<UtilityVerb>>(OnLockpickToLockVerb);
|
||||
}
|
||||
private void OnKeyRingInteract(Entity<CP14KeyRingComponent> keyring, ref AfterInteractEvent args)
|
||||
{
|
||||
if (args.Handled)
|
||||
return;
|
||||
|
||||
if (!args.CanReach || args.Target is not { Valid: true } target)
|
||||
return;
|
||||
|
||||
if (!TryComp<StorageComponent>(keyring, out var storageComp))
|
||||
return;
|
||||
|
||||
|
||||
if (TryComp<LockComponent>(args.Target, out var lockComp) &&
|
||||
TryGetLockFromSlot(args.Target.Value, out var lockEnt))
|
||||
{
|
||||
|
||||
foreach (var item in storageComp.StoredItems)
|
||||
{
|
||||
if (!TryComp<CP14KeyComponent>(item.Key, out var keyComp))
|
||||
continue;
|
||||
|
||||
if (keyComp.LockShape != lockEnt.Value.Comp.LockShape)
|
||||
continue;
|
||||
|
||||
TryUseKeyOnLock(args.User, args.Target.Value, new Entity<CP14KeyComponent>(item.Key, keyComp), lockEnt.Value);
|
||||
args.Handled = true;
|
||||
return;
|
||||
}
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-keyring-use-nofit"), args.Target.Value, args.User);
|
||||
}
|
||||
}
|
||||
|
||||
private void OnKeyInteract(Entity<CP14KeyComponent> key, ref AfterInteractEvent args)
|
||||
{
|
||||
if (args.Handled)
|
||||
return;
|
||||
|
||||
if (!args.CanReach || args.Target is not { Valid: true } target)
|
||||
return;
|
||||
|
||||
if (TryComp<LockComponent>(args.Target, out var lockComp) &&
|
||||
TryGetLockFromSlot(args.Target.Value, out var lockEnt))
|
||||
{
|
||||
TryUseKeyOnLock(args.User, args.Target.Value, key, new Entity<CP14LockComponent>(lockEnt.Value.Owner, lockEnt.Value.Comp));
|
||||
args.Handled = true;
|
||||
}
|
||||
}
|
||||
|
||||
private void OnLockpickToLockVerb(Entity<CP14LockpickComponent> lockpick, ref GetVerbsEvent<UtilityVerb> args)
|
||||
{
|
||||
if (!args.CanInteract || !args.CanAccess)
|
||||
return;
|
||||
|
||||
if (!TryComp<LockComponent>(args.Target, out var lockComp) || !lockComp.Locked)
|
||||
return;
|
||||
|
||||
if (!TryGetLockFromSlot(args.Target, out var lockItem))
|
||||
return;
|
||||
|
||||
if (!TryComp<CP14LockComponent>(lockItem, out var lockItemComp))
|
||||
return;
|
||||
|
||||
var target = args.Target;
|
||||
var user = args.User;
|
||||
|
||||
for (int i = DepthComplexity; i >= -DepthComplexity; i--)
|
||||
{
|
||||
var height = i;
|
||||
var verb = new UtilityVerb()
|
||||
{
|
||||
Act = () =>
|
||||
{
|
||||
TryHackDoorElement(user, target, lockpick, lockItemComp, lockComp, height);
|
||||
},
|
||||
Text = Loc.GetString("cp-lock-verb-lockpick-use-text") + $" {height}",
|
||||
Message = Loc.GetString("cp-lock-verb-lockpick-use-message"),
|
||||
Category = VerbCategory.Lockpick,
|
||||
Priority = height,
|
||||
};
|
||||
|
||||
args.Verbs.Add(verb);
|
||||
}
|
||||
}
|
||||
|
||||
private bool TryHackDoorElement(EntityUid user, EntityUid target, Entity<CP14LockpickComponent> lockpick, CP14LockComponent lockEnt, LockComponent lockComp, int height)
|
||||
{
|
||||
if (lockEnt.LockShape == null)
|
||||
return true;
|
||||
|
||||
if (height == lockEnt.LockShape[lockEnt.LockpickStatus]) //Success
|
||||
{
|
||||
_audio.PlayPvs(lockpick.Comp.SuccessSound, target);
|
||||
lockEnt.LockpickStatus++;
|
||||
if (lockEnt.LockpickStatus >= lockEnt.LockShape.Count) // Final success
|
||||
{
|
||||
if (lockComp.Locked)
|
||||
{
|
||||
_lock.TryUnlock(target, user, lockComp);
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-unlock-lock", ("lock", MetaData(lockEnt.Owner).EntityName)), target, user);
|
||||
lockEnt.LockpickStatus = 0;
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
_lock.TryLock(target, user, lockComp);
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-lock-lock", ("lock", MetaData(lockEnt.Owner).EntityName)), target, user);
|
||||
lockEnt.LockpickStatus = 0;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-lockpick-success"), target, user);
|
||||
return true;
|
||||
}
|
||||
else //Fail
|
||||
{
|
||||
_audio.PlayPvs(lockpick.Comp.FailSound, target);
|
||||
if (_random.Prob(lockEnt.LockPickDamageChance)) // Damage lockpick
|
||||
{
|
||||
lockpick.Comp.Health--;
|
||||
if (lockpick.Comp.Health > 0)
|
||||
{
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-lockpick-failed-damage", ("lock", MetaData(lockEnt.Owner).EntityName)), target, user);
|
||||
} else
|
||||
{
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-lockpick-failed-break", ("lock", MetaData(lockEnt.Owner).EntityName)), target, user);
|
||||
QueueDel(lockpick);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-lockpick-failed", ("lock", MetaData(lockEnt.Owner).EntityName)), target, user);
|
||||
}
|
||||
lockEnt.LockpickeddFailMarkup = true;
|
||||
lockEnt.LockpickStatus = 0;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
private void OnKeyToLockVerb(Entity<CP14KeyComponent> key, ref GetVerbsEvent<UtilityVerb> args)
|
||||
{
|
||||
if (!args.CanInteract || !args.CanAccess)
|
||||
return;
|
||||
|
||||
if (!TryComp<LockComponent>(args.Target, out var lockComp))
|
||||
return;
|
||||
|
||||
if (!TryGetLockFromSlot(args.Target, out var lockItem))
|
||||
return;
|
||||
|
||||
if (!TryComp<CP14LockComponent>(lockItem, out var lockItemComp))
|
||||
return;
|
||||
|
||||
var target = args.Target;
|
||||
var user = args.User;
|
||||
|
||||
var verb = new UtilityVerb()
|
||||
{
|
||||
Act = () =>
|
||||
{
|
||||
TryUseKeyOnLock(user, target, key, new Entity<CP14LockComponent>(target, lockItemComp));
|
||||
},
|
||||
IconEntity = GetNetEntity(key),
|
||||
Text = Loc.GetString(lockComp.Locked ? "cp-lock-verb-use-key-text-open" : "cp-lock-verb-use-key-text-close", ("item", MetaData(args.Target).EntityName)),
|
||||
Message = Loc.GetString("cp-lock-verb-use-key-message", ("item", MetaData(args.Target).EntityName))
|
||||
};
|
||||
|
||||
args.Verbs.Add(verb);
|
||||
}
|
||||
|
||||
private void OnLockInsertAttempt(Entity<LockComponent> lockSlot, ref ContainerIsInsertingAttemptEvent args)
|
||||
{
|
||||
if (!lockSlot.Comp.Initialized)
|
||||
return;
|
||||
|
||||
if (args.Container.ID != lockSlot.Comp.LockSlotId)
|
||||
return;
|
||||
|
||||
if (!TryComp<CP14LockComponent>(args.EntityUid, out var lockComp))
|
||||
{
|
||||
args.Cancel();
|
||||
return;
|
||||
}
|
||||
|
||||
if (lockComp == null)
|
||||
return;
|
||||
|
||||
//if (lockComp.Locked)
|
||||
//{
|
||||
// _popup.PopupEntity(Loc.GetString(
|
||||
// "cp-lock-lock-insert-fail-locked",
|
||||
// ("lock", MetaData(args.EntityUid).EntityName),
|
||||
// ("target", MetaData(lockSlot).EntityName)), lockSlot);
|
||||
// args.Cancel();
|
||||
//}
|
||||
}
|
||||
|
||||
private void OnLockInserted(Entity<LockComponent> lockSlot, ref EntInsertedIntoContainerMessage args)
|
||||
{
|
||||
if (!lockSlot.Comp.Initialized)
|
||||
return;
|
||||
|
||||
if (args.Container.ID != lockSlot.Comp.LockSlotId)
|
||||
return;
|
||||
|
||||
if (!TryComp<CP14LockComponent>(args.Entity, out var lockComp))
|
||||
return;
|
||||
|
||||
_appearance.SetData(lockSlot, LockSlotVisuals.LockExist, true);
|
||||
}
|
||||
|
||||
private void OnLockRemoved(Entity<LockComponent> lockSlot, ref EntRemovedFromContainerMessage args)
|
||||
{
|
||||
if (args.Container.ID != lockSlot.Comp.LockSlotId)
|
||||
return;
|
||||
_appearance.SetData(lockSlot, LockSlotVisuals.LockExist, false);
|
||||
}
|
||||
|
||||
public bool TryGetLockFromSlot(EntityUid uid,
|
||||
[NotNullWhen(true)] out Entity<CP14LockComponent>? lockEnt,
|
||||
LockComponent? component = null)
|
||||
{
|
||||
if (!Resolve(uid, ref component, false))
|
||||
{
|
||||
lockEnt = null;
|
||||
return false;
|
||||
}
|
||||
|
||||
if (component.LockSlotId == null)
|
||||
{
|
||||
lockEnt = null;
|
||||
return false;
|
||||
}
|
||||
|
||||
if (_itemSlots.TryGetSlot(uid, component.LockSlotId, out ItemSlot? slot))
|
||||
{
|
||||
if (TryComp<CP14LockComponent>(slot.Item, out var lockComp))
|
||||
{
|
||||
lockEnt = new Entity<CP14LockComponent>(slot.Item.Value, lockComp);
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
lockEnt = null;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
lockEnt = null;
|
||||
return false;
|
||||
}
|
||||
private bool TryUseKeyOnLock(EntityUid user, EntityUid target, Entity<CP14KeyComponent> keyEnt, Entity<CP14LockComponent> lockEnt)
|
||||
{
|
||||
if (!TryComp<LockComponent>(target, out var lockComp))
|
||||
return false;
|
||||
|
||||
var keyShape = keyEnt.Comp.LockShape;
|
||||
var lockShape = lockEnt.Comp.LockShape;
|
||||
|
||||
if (keyShape == null || lockShape == null)
|
||||
return false;
|
||||
|
||||
if (keyShape == lockShape)
|
||||
{
|
||||
if (lockComp.Locked)
|
||||
{
|
||||
if(_lock.TryUnlock(target, user))
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-unlock-lock", ("lock", MetaData(lockEnt).EntityName)), lockEnt, user);
|
||||
}
|
||||
else
|
||||
{
|
||||
if (_lock.TryLock(target, user))
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-lock-lock", ("lock", MetaData(lockEnt).EntityName)), lockEnt, user);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
_popup.PopupEntity(Loc.GetString("cp-lock-key-use-nofit"), lockEnt, user);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
[Serializable, NetSerializable]
|
||||
public enum LockSlotVisuals : byte
|
||||
{
|
||||
LockExist
|
||||
}
|
||||
Reference in New Issue
Block a user