ADD traitor codewords highlight

Content.Client/UserInterface/Systems/Chat/Widgets/ChatBox.xaml.cs

@@ -1,9 +1,11 @@
 using Content.Client.UserInterface.Systems.Chat.Controls;
 using Content.Shared.Chat;
 using Content.Shared.Input;
+using Content.Shared.SS220.ShowCodewords;
 using Robust.Client.Audio;
 using Robust.Client.AutoGenerated;
 using Robust.Client.GameObjects;
+using Robust.Client.Player;
 using Robust.Client.UserInterface;
 using Robust.Client.UserInterface.Controls;
 using Robust.Client.UserInterface.XAML;
@@ -65,6 +67,20 @@ private void OnMessageAdded(ChatMessage msg)
 
         var color = msg.MessageColorOverride ?? msg.Channel.TextColor();
 
+        //ss220 codewords highlight add start
+        var playerManager = IoCManager.Resolve<IPlayerManager>();
+        var entManager = IoCManager.Resolve<IEntityManager>();
+
+        if (playerManager.LocalEntity != null &&
+            entManager.TryGetComponent<ShowCodewordsComponent>(playerManager.LocalEntity, out var showCodewordsComponent))
+        {
+            foreach (var codeword in showCodewordsComponent.CodeWords)
+            {
+                msg.WrappedMessage = msg.WrappedMessage.Replace(codeword, $"[color=pink]{codeword}[/color]");
+            }
+        }
+        //ss220 codewords highlight add end
+
         AddLine(msg.WrappedMessage, color);
     }
 

Content.Server/GameTicking/Rules/TraitorRuleSystem.cs

@@ -18,6 +18,7 @@
 using System.Text;
 using Content.Shared.GameTicking.Components;
 using Content.Server.SS220.MindSlave;
+using Content.Shared.SS220.ShowCodewords;
 
 namespace Content.Server.GameTicking.Rules;
 
@@ -106,6 +107,13 @@ public bool MakeTraitor(EntityUid traitor, TraitorRuleComponent component, bool
 
         component.TraitorMinds.Add(mindId);
 
+        //ss220 codewords highlight add start
+
+        var showCodewordsComp = EnsureComp<ShowCodewordsComponent>(traitor);
+        showCodewordsComp.CodeWords = component.Codewords;
+
+        //ss220 codewords highlight add end
+
         // Assign briefing
         _roleSystem.MindAddRole(mindId, new RoleBriefingComponent
         {

Content.Shared/SS220/ShowCodewords/ShowCodewordsComponent.cs

@@ -0,0 +1,11 @@
+using Robust.Shared.GameStates;
+
+namespace Content.Shared.SS220.ShowCodewords;
+
+[RegisterComponent, NetworkedComponent]
+[AutoGenerateComponentState]
+public sealed partial class ShowCodewordsComponent : Component
+{
+    [DataField("codewords"), ViewVariables(VVAccess.ReadWrite), AutoNetworkedField]
+    public string[] CodeWords;
+}