Change Thief Syndie & Chameleon kit contents, add Syndie codeword paper (#30446)

* Initial commit * more like bYE * Fix exception during test
2024-08-05 10:03:24 +02:00
parent 8197382444
commit bf06d0e869
7 changed files with 161 additions and 9 deletions
--- a/Content.Server/Traitor/Systems/TraitorCodePaperSystem.cs
+++ b/Content.Server/Traitor/Systems/TraitorCodePaperSystem.cs
@@ -0,0 +1,93 @@
+using System.Diagnostics.CodeAnalysis;
+using Content.Server.GameTicking;
+using Content.Server.GameTicking.Rules;
+using Content.Server.GameTicking.Rules.Components;
+using Content.Server.Paper;
+using Content.Server.Traitor.Components;
+using Robust.Shared.Random;
+using Robust.Shared.Utility;
+using System.Linq;
+
+namespace Content.Server.Traitor.Systems;
+
+public sealed class TraitorCodePaperSystem : EntitySystem
+{
+    [Dependency] private readonly GameTicker _gameTicker = default!;
+    [Dependency] private readonly TraitorRuleSystem _traitorRuleSystem = default!;
+    [Dependency] private readonly IRobustRandom _random = default!;
+    [Dependency] private readonly PaperSystem _paper = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+        SubscribeLocalEvent<TraitorCodePaperComponent, MapInitEvent>(OnMapInit);
+    }
+
+    private void OnMapInit(EntityUid uid, TraitorCodePaperComponent component, MapInitEvent args)
+    {
+        SetupPaper(uid, component);
+    }
+
+    private void SetupPaper(EntityUid uid, TraitorCodePaperComponent? component = null)
+    {
+        if (!Resolve(uid, ref component))
+            return;
+
+        if (HasComp<PaperComponent>(uid))
+        {
+            if (TryGetTraitorCode(out var paperContent, component))
+            {
+                _paper.SetContent(uid, paperContent);
+            }
+        }
+    }
+
+    private bool TryGetTraitorCode([NotNullWhen(true)] out string? traitorCode, TraitorCodePaperComponent component)
+    {
+        traitorCode = null;
+
+        var codesMessage = new FormattedMessage();
+        List<string> codeList = new();
+        // Find the first nuke that matches the passed location.
+        if (_gameTicker.IsGameRuleAdded<TraitorRuleComponent>())
+        {
+            var ruleEnts = _gameTicker.GetAddedGameRules();
+            foreach (var ruleEnt in ruleEnts)
+            {
+                if (TryComp(ruleEnt, out TraitorRuleComponent? traitorComp))
+                {
+                    codeList.AddRange(traitorComp.Codewords.ToList());
+                }
+            }
+        }
+        if (codeList.Count == 0)
+        {
+            if (component.FakeCodewords)
+                codeList = _traitorRuleSystem.GenerateTraitorCodewords(new TraitorRuleComponent()).ToList();
+            else
+                codeList = [Loc.GetString("traitor-codes-none")];
+        }
+
+        _random.Shuffle(codeList);
+
+        int i = 0;
+        foreach (var code in codeList)
+        {
+            i++;
+            if (i > component.CodewordAmount && !component.CodewordShowAll)
+                break;
+
+            codesMessage.PushNewline();
+            codesMessage.AddMarkup(code);
+        }
+
+        if (!codesMessage.IsEmpty)
+        {
+            if (i == 1) 
+                traitorCode = Loc.GetString("traitor-codes-message-singular") + codesMessage;
+            else
+                traitorCode = Loc.GetString("traitor-codes-message-plural") + codesMessage;
+        }
+        return !codesMessage.IsEmpty;
+    }
+}