From d6c4a29dcbde369acba8293b9c9a2bcf4d2cec0d Mon Sep 17 00:00:00 2001
From: Nemanja <98561806+EmoGarbage404@users.noreply.github.com>
Date: Sun, 17 Jul 2022 17:57:52 -0400
Subject: [PATCH] make nukies actual antags (#9844)

---
 Content.Server/GameTicking/Rules/NukeopsRuleSystem.cs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Content.Server/GameTicking/Rules/NukeopsRuleSystem.cs b/Content.Server/GameTicking/Rules/NukeopsRuleSystem.cs
index eaaf5246d8..7862cf051e 100644
--- a/Content.Server/GameTicking/Rules/NukeopsRuleSystem.cs
+++ b/Content.Server/GameTicking/Rules/NukeopsRuleSystem.cs
@@ -22,6 +22,7 @@ using Robust.Shared.Map;
 using Robust.Shared.Prototypes;
 using Robust.Shared.Random;
 using Robust.Shared.Utility;
+using Content.Server.Traitor;
 
 namespace Content.Server.GameTicking.Rules;
 
@@ -242,20 +243,24 @@ public sealed class NukeopsRuleSystem : GameRuleSystem
         for (var i = 0; i < operatives.Count; i++)
         {
             string name;
+            string role;
             StartingGearPrototype gear;
 
             switch (i)
             {
                 case 0:
                     name = $"Commander " + _random.PickAndTake<string>(syndicateNamesElite);
+                    role = NukeopsCommanderPrototypeId;
                     gear = commanderGear;
                     break;
                 case 1:
                     name = $"Agent " + _random.PickAndTake<string>(syndicateNamesNormal);
+                    role = NukeopsPrototypeId;
                     gear = medicGear;
                     break;
                 default:
                     name = $"Operator " + _random.PickAndTake<string>(syndicateNamesNormal);
+                    role = NukeopsPrototypeId;
                     gear = starterGear;
                     break;
             }
@@ -266,6 +271,7 @@ public sealed class NukeopsRuleSystem : GameRuleSystem
                 CharacterName = name
             };
             newMind.ChangeOwningPlayer(session.UserId);
+            newMind.AddRole(new TraitorRole(newMind, _prototypeManager.Index<AntagPrototype>(role)));
 
             var mob = EntityManager.SpawnEntity("MobHuman", _random.Pick(spawns));
             EntityManager.GetComponent<MetaDataComponent>(mob).EntityName = name;