Fix some AI (#206)

space-syndicate · Jan 1, 2025 · 5418be6 · 5418be6
1 parent 19c520b
commit 5418be6
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/Content.Shared/_CorvaxNext/TurretControl/Components/TurretControllerComponent.cs b/Content.Shared/_CorvaxNext/TurretControl/Components/TurretControllerComponent.cs
@@ -1,4 +1,10 @@
+using Robust.Shared.Prototypes;
+
 namespace Content.Shared._CorvaxNext.TurretControl.Components;
 
 [RegisterComponent]
-public sealed partial class TurretControllerComponent : Component;
+public sealed partial class TurretControllerComponent : Component
+{
+    [DataField]
+    public ComponentRegistry RequiredComponents = [];
+}
diff --git a/Content.Shared/_CorvaxNext/TurretControl/TurretControlSystem.cs b/Content.Shared/_CorvaxNext/TurretControl/TurretControlSystem.cs
@@ -26,9 +26,13 @@ public override void Initialize()
 
     private void OnGetVerbs(Entity<TurretControllableComponent> entity, ref GetVerbsEvent<Verb> e)
     {
-        if (!HasComp<TurretControllerComponent>(e.User))
+        if (!TryComp<TurretControllerComponent>(e.User, out var controller))
             return;
 
+        foreach (var component in controller.RequiredComponents)
+            if (!HasComp(e.User, component.Value.Component.GetType()))
+                return;
+
         if (!TryComp<NpcFactionMemberComponent>(entity, out var factionMember))
             return;
 

diff --git a/Resources/Prototypes/Entities/Mobs/Player/silicon.yml b/Resources/Prototypes/Entities/Mobs/Player/silicon.yml
@@ -439,6 +439,8 @@
     speechVerb: Robotic
     speechSounds: Borg
   - type: TurretController # Corvax-Next-TurretControl
+    requiredComponents:
+    - type: StationAiHeld
   - type: Tag
     tags:
     - HideContextMenu