[FEATURE][TWEAK] AccessWeaponBlocker

Content.Client/Backmen/AccessWeaponBlockerSystem/AccessWeaponBlockerComponent.cs

@@ -0,0 +1,13 @@
+using Content.Shared.Backmen.AccessGunBlockerSystem;
+
+namespace Content.Client.Backmen.AccessWeaponBlockerSystem;
+
+[RegisterComponent]
+public sealed partial class AccessWeaponBlockerComponent : SharedAccessWeaponBlockerComponent
+{
+    [ViewVariables(VVAccess.ReadWrite)]
+    public bool CanUse;
+
+    [ViewVariables(VVAccess.ReadWrite)]
+    public string AlertText = "";
+}

Content.Client/Backmen/AccessWeaponBlockerSystem/AccessWeaponBlockerSystem.cs

@@ -0,0 +1,64 @@
+using Content.Shared.Interaction.Events;
+using Content.Shared.Backmen.AccessGunBlockerSystem;
+using Content.Shared.Weapons.Melee.Events;
+using Content.Shared.Weapons.Ranged.Systems;
+using Robust.Shared.GameStates;
+
+namespace Content.Client.Backmen.AccessWeaponBlockerSystem;
+
+public sealed class AccessWeaponBlockerSystem : EntitySystem
+{
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, AttemptShootEvent>(OnShootAttempt);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, AttemptMeleeEvent>(OnMeleeAttempt);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, UseAttemptEvent>(OnUseAttempt);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, InteractionAttemptEvent>(OnInteractAttempt);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, ComponentHandleState>(OnAccessWeaponBlockerHandleState);
+    }
+
+    private void OnUseAttempt(EntityUid uid, AccessWeaponBlockerComponent component, ref UseAttemptEvent args)
+    {
+        if (component.CanUse)
+            return;
+
+        args.Cancel();
+    }
+
+    private void OnInteractAttempt(EntityUid uid, AccessWeaponBlockerComponent component, ref InteractionAttemptEvent args)
+    {
+        if (component.CanUse)
+            return;
+
+        args.Cancelled = true;
+    }
+
+    private void OnAccessWeaponBlockerHandleState(EntityUid uid, AccessWeaponBlockerComponent component, ref ComponentHandleState args)
+    {
+        if (args.Current is not AccessWeaponBlockerComponentState state)
+            return;
+
+        component.CanUse = state.CanUse;
+        component.AlertText = state.AlertText;
+    }
+
+    private void OnMeleeAttempt(EntityUid uid, AccessWeaponBlockerComponent component, ref AttemptMeleeEvent args)
+    {
+        if (component.CanUse)
+            return;
+
+        args.Cancelled = true;
+        args.Message = component.AlertText;
+    }
+
+    private void OnShootAttempt(EntityUid uid, AccessWeaponBlockerComponent component, ref AttemptShootEvent args)
+    {
+        if (component.CanUse)
+            return;
+
+        args.Cancelled = true;
+        args.Message = component.AlertText;
+    }
+}

Content.Server/Backmen/AccessWeaponBlockerSystem/AccessWeaponBlockerComponent.cs

@@ -0,0 +1,20 @@
+using Content.Shared.Access;
+using Content.Shared.Backmen.AccessGunBlockerSystem;
+using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.Set;
+
+namespace Content.Server.Backmen.AccessWeaponBlockerSystem;
+
+[RegisterComponent]
+public sealed partial class AccessWeaponBlockerComponent : SharedAccessWeaponBlockerComponent
+{
+    [ViewVariables(VVAccess.ReadWrite)]
+    public bool CanUse;
+
+    [ViewVariables(VVAccess.ReadWrite)]
+    [DataField("alertText")]
+    public string AlertText = "";
+
+    [ViewVariables(VVAccess.ReadWrite),
+     DataField("access", customTypeSerializer: typeof(PrototypeIdHashSetSerializer<AccessLevelPrototype>))]
+    public HashSet<string> Access = new();
+}

Content.Server/Backmen/AccessWeaponBlockerSystem/AccessWeaponBlockerSystem.cs

@@ -0,0 +1,74 @@
+using Content.Shared.Hands;
+using Content.Shared.Access.Components;
+using Content.Shared.Backmen.AccessGunBlockerSystem;
+using Content.Shared.Weapons.Melee.Events;
+using Content.Shared.Weapons.Ranged.Systems;
+using Robust.Shared.GameStates;
+using Content.Shared.Inventory;
+using Content.Shared.PDA;
+
+namespace Content.Server.Backmen.AccessWeaponBlockerSystem;
+
+public sealed class AccessWeaponBlockerSystem : EntitySystem
+{
+    [Dependency] private readonly InventorySystem _inventorySystem = default!;
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, AttemptShootEvent>(OnShootAttempt);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, AttemptMeleeEvent>(OnMeleeAttempt);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, ComponentGetState>(OnGetState);
+        SubscribeLocalEvent<AccessWeaponBlockerComponent, GotEquippedHandEvent>(OnGotEquippedHand);
+    }
+
+
+    private void OnGotEquippedHand(EntityUid uid, AccessWeaponBlockerComponent component, ref GotEquippedHandEvent args)
+    {
+        if (!_inventorySystem.TryGetSlotEntity(args.User, "id", out var slotCardUid))
+            return;
+        var accessEntity = TryComp<PdaComponent>(slotCardUid, out var pda) && pda.ContainedId is { } pdaSlot
+            ? pdaSlot
+            : slotCardUid.Value;
+        component.CanUse = IsAnyAccess(accessEntity, component);
+        Dirty(uid, component);
+    }
+
+    private bool IsAnyAccess(EntityUid accessEntity, AccessWeaponBlockerComponent component)
+    {
+        if (!TryComp<AccessComponent>(accessEntity, out var access))
+            return false;
+        foreach (var accessTag in access.Tags)
+        {
+            if (component.Access.Contains(accessTag))
+                return true;
+        }
+        return false;
+    }
+    private void OnGetState(EntityUid uid, AccessWeaponBlockerComponent component, ref ComponentGetState args)
+    {
+        args.State = new AccessWeaponBlockerComponentState()
+        {
+            CanUse = component.CanUse,
+            AlertText = component.AlertText
+        };
+    }
+
+    private void OnMeleeAttempt(EntityUid uid, AccessWeaponBlockerComponent component, ref AttemptMeleeEvent args)
+    {
+        if (component.CanUse)
+            return;
+
+        args.Cancelled = true;
+        args.Message = component.AlertText;
+    }
+
+    private void OnShootAttempt(EntityUid uid, AccessWeaponBlockerComponent component, ref AttemptShootEvent args)
+    {
+        if (component.CanUse)
+            return;
+
+        args.Cancelled = true;
+        args.Message = component.AlertText;
+    }
+}

Content.Shared/Backmen/AccessWeaponBlockerSystem/SharedAccessWeaponBlockerComponent.cs

@@ -0,0 +1,17 @@
+using Robust.Shared.GameStates;
+using Robust.Shared.Serialization;
+
+namespace Content.Shared.Backmen.AccessWeaponBlockerSystem;;
+
+[NetworkedComponent]
+public abstract partial class SharedAccessWeaponBlockerComponent : Component
+{
+
+}
+
+[Serializable, NetSerializable]
+public sealed class AccessWeaponBlockerComponentState : ComponentState
+{
+    public bool CanUse;
+    public string AlertText = "";
+}

Content.Shared/Backmen/AccessWeaponBlockerSystem/SharedFactionWeaponBlockerSystem.cs

@@ -0,0 +1,6 @@
+namespace Content.Shared.Backmen.AccessWeaponBlockerSystem;
+
+public sealed class SharedFactionWeaponBlockerSystem : EntitySystem
+{
+
+}