From 282bc688a04b8d64c032a1c94b2d0388fc55f587 Mon Sep 17 00:00:00 2001
From: MSWS <imodmaker@gmail.com>
Date: Sun, 4 Feb 2024 20:55:48 -0800
Subject: [PATCH] Add rebel notifications

---
 mod/Jailbreak.Rebel/RebelManager.cs | 36 +++++++++++++++++++++++++----
 1 file changed, 32 insertions(+), 4 deletions(-)

diff --git a/mod/Jailbreak.Rebel/RebelManager.cs b/mod/Jailbreak.Rebel/RebelManager.cs
index 426976d2..9fadd81f 100644
--- a/mod/Jailbreak.Rebel/RebelManager.cs
+++ b/mod/Jailbreak.Rebel/RebelManager.cs
@@ -27,6 +27,7 @@ public RebelManager(IRebelNotifications notifs, ILogService logs)
     public void Start(BasePlugin parent)
     {
         parent.RegisterEventHandler<EventPlayerDisconnect>(OnPlayerDisconnect);
+        parent.RegisterEventHandler<EventPlayerDeath>(OnPlayerDeath);
         parent.RegisterEventHandler<EventRoundStart>(OnRoundStart);
 
         parent.AddTimer(1f, () =>
@@ -43,6 +44,7 @@ public void Start(BasePlugin parent)
                 }
 
                 ApplyRebelColor(player);
+                SendTimeLeft(player);
             }
         }, TimerFlags.REPEAT);
     }
@@ -70,6 +72,15 @@ HookResult OnPlayerDisconnect(EventPlayerDisconnect @event, GameEventInfo info)
         return HookResult.Continue;
     }
 
+    HookResult OnPlayerDeath(EventPlayerDeath @event, GameEventInfo info)
+    {
+        var player = @event.Userid;
+        if (!player.IsReal())
+            return HookResult.Continue;
+        rebelTimes.Remove(player);
+        return HookResult.Continue;
+    }
+
     public ISet<CCSPlayerController> GetActiveRebels()
     {
         return rebelTimes.Keys.ToHashSet();
@@ -116,11 +127,9 @@ private float GetRebelTimePercentage(CCSPlayerController player)
             return 0;
         return (float)(100 - (120 - x) * (Math.Sqrt(120 - x)) / 13f) / 100;
     }
-
-    private void ApplyRebelColor(CCSPlayerController player)
+    
+    private Color GetRebelColor(CCSPlayerController player)
     {
-        if (!player.IsReal() || player.Pawn.Value == null)
-            return;
         var percent = GetRebelTimePercentage(player);
         var percentRGB = 255 - (int)Math.Round(percent * 255.0);
         var color = Color.FromArgb(254, 255, percentRGB, percentRGB);
@@ -129,8 +138,27 @@ private void ApplyRebelColor(CCSPlayerController player)
             color = Color.FromArgb(254, 255, 255, 255);
         }
 
+        return color;
+    }
+
+    private void ApplyRebelColor(CCSPlayerController player)
+    {
+        if (!player.IsReal() || player.Pawn.Value == null)
+            return;
+        var color = GetRebelColor(player);
+
         player.Pawn.Value.RenderMode = RenderMode_t.kRenderTransColor;
         player.Pawn.Value.Render = color;
         Utilities.SetStateChanged(player.Pawn.Value, "CBaseModelEntity", "m_clrRender");
     }
+
+    private void SendTimeLeft(CCSPlayerController player)
+    {
+        var timeLeft = GetRebelTimeLeft(player);
+        var formattedTime = TimeSpan.FromSeconds(timeLeft).ToString(@"mm\:ss");
+        var color = GetRebelColor(player);
+        var formattedColor = $"<font color=\"#{color.R:X2}{color.G:X2}{color.B:X2}\">";
+        
+        player.PrintToCenterHtml($"You are a rebel for {formattedColor}{formattedTime}</font> more seconds.");
+    }
 }
\ No newline at end of file