prevent borgs unlocking eachother and robotics console

Content.Shared/Lock/LockSystem.cs

@@ -224,6 +224,7 @@ public bool CanToggleLock(EntityUid uid, EntityUid user, bool quiet = true)
 
         var ev = new LockToggleAttemptEvent(user, quiet);
         RaiseLocalEvent(uid, ref ev, true);
+        RaiseLocalEvent(user, ref ev, true);
         return !ev.Cancelled;
     }
 

Content.Shared/Lock/LockingWhitelistComponent.cs

@@ -0,0 +1,17 @@
+using Content.Shared.Whitelist;
+using Robust.Shared.GameStates;
+
+namespace Content.Shared.Lock;
+
+/// <summary>
+/// Adds whitelist and blacklist for this mob to lock things.
+/// </summary>
+[RegisterComponent, NetworkedComponent, Access(typeof(LockingWhitelistSystem))]
+public sealed partial class LockingWhitelistComponent : Component
+{
+    [DataField]
+    public EntityWhitelist? Whitelist;
+
+    [DataField]
+    public EntityWhitelist? Blacklist;
+}

Content.Shared/Lock/LockingWhitelistSystem.cs

@@ -0,0 +1,21 @@
+using Content.Shared.Whitelist;
+
+namespace Content.Shared.Lock;
+
+public sealed class LockingWhitelistSystem : EntitySystem
+{
+    [Dependency] private readonly EntityWhitelistSystem _whitelist = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<LockingWhitelistComponent, LockToggleAttemptEvent>(OnLockToggleAttempt);
+    }
+
+    private void OnLockToggleAttempt(Entity<LockingWhitelistComponent> ent, ref LockToggleAttemptEvent args)
+    {
+        if (args.User == ent.Owner && !_whitelist.IsAllowed(ent, ent.Comp.Blacklist, ent.Comp.Whitelist))
+            args.Cancelled = true;
+    }
+}

Content.Shared/Silicons/Borgs/Components/BorgChassisComponent.cs

@@ -19,6 +19,12 @@ public sealed partial class BorgChassisComponent : Component
     [DataField("activated"), ViewVariables(VVAccess.ReadWrite), AutoNetworkedField]
     public bool Activated;
 
+    /// <summary>
+    /// Things the borg is not allowed to lock or unlock.
+    /// </summary>
+    [DataField]
+    public EntityWhitelist? LockBlacklist;
+
     #region Brain
     /// <summary>
     /// A whitelist for which entities count as valid brains

Content.Shared/Silicons/Borgs/SharedBorgSystem.cs

@@ -1,5 +1,6 @@
 using Content.Shared.Access.Components;
 using Content.Shared.Containers.ItemSlots;
+using Content.Shared.Lock;
 using Content.Shared.Movement.Components;
 using Content.Shared.Movement.Systems;
 using Content.Shared.Popups;

Content.Shared/Whitelist/EntityWhitelistSystem.cs

@@ -23,6 +23,42 @@ public bool IsValid(EntityWhitelist list, [NotNullWhen(true)] EntityUid? uid)
         return uid != null && IsValid(list, uid.Value);
     }
 
+    /// <summary>
+    /// Checks whether a given entity satisfies a whitelist.
+    /// If the whitelist is null it returns null.
+    /// </summary>
+    public bool? Match(EntityWhitelist? list, [NotNullWhen(true)] EntityUid? uid)
+    {
+        if (list == null)
+            return null;
+
+        return IsValid(list, uid);
+    }
+
+    /// <summary>
+    /// Checks whether a given entity is allowed by a whitelist and not blocked by a blacklist.
+    /// If a blacklist is provided and it matches then this returns false.
+    /// If a whitelist is provided and it does not match then this returns false.
+    /// If either list is null it does not get checked.
+    /// </summary>
+    public bool IsAllowed([NotNullWhen(true)] EntityUid? uid, EntityWhitelist? blacklist = null, EntityWhitelist? whitelist = null)
+    {
+        if (uid == null)
+            return false;
+
+        return Match(blacklist, uid) != true
+            && Match(whitelist, uid) != false;
+    }
+
+    /// <summary>
+    /// Checks whether a given entity is on a blacklist.
+    /// If the blacklist is null it cannot be on it, so it returns false.
+    /// </summary>
+    public bool IsBlacklisted(EntityWhitelist? list, EntityUid? uid)
+    {
+        return Match(list, uid) != true;
+    }
+
     /// <summary>
     /// Checks whether a given entity satisfies a whitelist.
     /// </summary>

Resources/Prototypes/Entities/Mobs/Cyborgs/base_borg_chassis.yml

@@ -100,6 +100,11 @@
     doAfterDelay: 10
     allowSelfRepair: false
   - type: BorgChassis
+  - type: LockingWhitelist
+    blacklist:
+      components:
+      - BorgChassis
+      - RoboticsConsole
   - type: WiresPanel
   - type: ActivatableUIRequiresPanel
   - type: NameIdentifier