335 lines
11 KiB
C#
335 lines
11 KiB
C#
|
|
||
|
|
using Content.Shared.Containers.ItemSlots;
|
||
|
|
using Content.Shared.Interaction;
|
||
|
|
using Content.Shared.Lock;
|
||
|
|
using Content.Shared.Popups;
|
||
|
|
using Robust.Shared.Containers;
|
||
|
|
using Robust.Shared.Serialization;
|
||
|
|
using System.Diagnostics.CodeAnalysis;
|
||
|
|
using Content.Shared.Verbs;
|
||
|
|
using Content.Shared.Storage.EntitySystems;
|
||
|
|
using Content.Shared.Storage;
|
||
|
|
using Robust.Shared.Random;
|
||
|
|
using Robust.Shared.Audio.Systems;
|
||
|
|
|
||
|
|
namespace Content.Shared.CrystallPunk.LockKey;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
///
|
||
|
|
/// </summary>
|
||
|
|
public sealed class SharedCPLockKeySystem : 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 DepthCompexity = 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<CPKeyComponent, AfterInteractEvent>(OnKeyInteract);
|
||
|
|
SubscribeLocalEvent<CPKeyRingComponent, AfterInteractEvent>(OnKeyRingInteract);
|
||
|
|
SubscribeLocalEvent<CPKeyComponent, GetVerbsEvent<UtilityVerb>>(OnKeyToLockVerb);
|
||
|
|
SubscribeLocalEvent<CPLockpickComponent, GetVerbsEvent<UtilityVerb>>(OnLockpickToLockVerb);
|
||
|
|
}
|
||
|
|
private void OnKeyRingInteract(Entity<CPKeyRingComponent> 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<CPKeyComponent>(item.Key, out var keyComp))
|
||
|
|
continue;
|
||
|
|
|
||
|
|
if (keyComp.LockShape != lockEnt.Value.Comp.LockShape)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
TryUseKeyOnLock(args.User, args.Target.Value, new Entity<CPKeyComponent>(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<CPKeyComponent> 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<CPLockComponent>(lockEnt.Value.Owner, lockEnt.Value.Comp));
|
||
|
|
args.Handled = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnLockpickToLockVerb(Entity<CPLockpickComponent> 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<CPLockComponent>(lockItem, out var lockItemComp))
|
||
|
|
return;
|
||
|
|
|
||
|
|
var target = args.Target;
|
||
|
|
var user = args.User;
|
||
|
|
|
||
|
|
for (int i = DepthCompexity; i >= -DepthCompexity; 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<CPLockpickComponent> lockpick, CPLockComponent 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<CPKeyComponent> 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<CPLockComponent>(lockItem, out var lockItemComp))
|
||
|
|
return;
|
||
|
|
|
||
|
|
var target = args.Target;
|
||
|
|
var user = args.User;
|
||
|
|
|
||
|
|
var verb = new UtilityVerb()
|
||
|
|
{
|
||
|
|
Act = () =>
|
||
|
|
{
|
||
|
|
TryUseKeyOnLock(user, target, key, new Entity<CPLockComponent>(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<CPLockComponent>(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<CPLockComponent>(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<CPLockComponent>? 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<CPLockComponent>(slot.Item, out var lockComp))
|
||
|
|
{
|
||
|
|
lockEnt = new Entity<CPLockComponent>(slot.Item.Value, lockComp);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
lockEnt = null;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
lockEnt = null;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
private bool TryUseKeyOnLock(EntityUid user, EntityUid target, Entity<CPKeyComponent> keyEnt, Entity<CPLockComponent> 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
|
||
|
|
}
|