Make monkeys better (#19407)

Co-authored-by: deltanedas <@deltanedas:kde.org>

Content.Server/Traitor/Systems/AutoTraitorSystem.cs

@@ -0,0 +1,76 @@
+using Content.Server.GameTicking.Rules;
+using Content.Server.Traitor.Components;
+using Content.Shared.Mind;
+using Content.Shared.Mind.Components;
+
+namespace Content.Server.Traitor.Systems;
+
+/// <summary>
+/// Makes entities with <see cref="AutoTraitorComponent"/> a traitor either immediately if they have a mind or when a mind is added.
+/// </summary>
+public sealed class AutoTraitorSystem : EntitySystem
+{
+    [Dependency] private readonly TraitorRuleSystem _traitorRule = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<AutoTraitorComponent, MapInitEvent>(OnMapInit);
+        SubscribeLocalEvent<AutoTraitorComponent, MindAddedMessage>(OnMindAdded);
+    }
+
+    private void OnMapInit(EntityUid uid, AutoTraitorComponent comp, MapInitEvent args)
+    {
+        TryMakeTraitor(uid, comp);
+    }
+
+    private void OnMindAdded(EntityUid uid, AutoTraitorComponent comp, MindAddedMessage args)
+    {
+        TryMakeTraitor(uid, comp);
+    }
+
+    /// <summary>
+    /// Sets the GiveUplink field.
+    /// </summary>
+    public void SetGiveUplink(EntityUid uid, bool giveUplink, AutoTraitorComponent? comp = null)
+    {
+        if (!Resolve(uid, ref comp))
+            return;
+
+        comp.GiveUplink = giveUplink;
+    }
+
+    /// <summary>
+    /// Sets the GiveObjectives field.
+    /// </summary>
+    public void SetGiveObjectives(EntityUid uid, bool giveObjectives, AutoTraitorComponent? comp = null)
+    {
+        if (!Resolve(uid, ref comp))
+            return;
+
+        comp.GiveObjectives = giveObjectives;
+    }
+
+    /// <summary>
+    /// Checks if there is a mind, then makes it a traitor using the options.
+    /// </summary>
+    public bool TryMakeTraitor(EntityUid uid, AutoTraitorComponent? comp = null)
+    {
+        if (!Resolve(uid, ref comp))
+            return false;
+
+        if (!TryComp<MindContainerComponent>(uid, out var mindContainer) || mindContainer.Mind == null)
+            return false;
+
+        var mindId = mindContainer.Mind.Value;
+        if (!TryComp<MindComponent>(mindId, out var mind) || mind.Session == null)
+            return false;
+
+        var session = mind.Session;
+        _traitorRule.MakeTraitor(session, giveUplink: comp.GiveUplink, giveObjectives: comp.GiveObjectives);
+        // prevent spamming anything if it fails
+        RemComp<AutoTraitorComponent>(uid);
+        return true;
+    }
+}