From b788852ea8c4f4aae55983716be9cf9c6a792a51 Mon Sep 17 00:00:00 2001
From: destoer <destoer1234@protonmail.com>
Date: Sun, 3 Mar 2024 12:49:00 +0000
Subject: [PATCH] implement basic warden countdown

---
 lang/en.json                 |  7 ++++
 src/Jail.cs                  |  5 ++-
 src/LastRequest/LRBase.cs    |  2 +-
 src/Lib/Timer.cs             |  9 +++--
 src/SpecialDay/SpecialDay.cs |  2 +-
 src/Warden/Warden.cs         |  2 ++
 src/Warden/WardenCmd.cs      | 66 ++++++++++++++++++++++++++++++++++++
 7 files changed, 88 insertions(+), 5 deletions(-)

diff --git a/lang/en.json b/lang/en.json
index 19ffc57..029a7f3 100644
--- a/lang/en.json
+++ b/lang/en.json
@@ -62,6 +62,13 @@
     "warden.give_pardon": "{0} has been pardoned",
     "warden.give_freeday": "{0} has been given a freeday",
 
+    "warden.countdown_cmd": "countdown",
+    "warden.countdown_abort_cmd": "countdown_abort",
+    "warden.countdown_usage": "!countdown: <time> <msg>",
+    "warden.countdown_restrict": "You must be the warden to call a countdown",
+    "warden.countdown_abort": "Warden aborted the countdown",
+
+
     "logs.ct": "CT",
     "logs.t": "T",
     "logs.logs_cmd": "logs",
diff --git a/src/Jail.cs b/src/Jail.cs
index 43a82eb..6f75a02 100644
--- a/src/Jail.cs
+++ b/src/Jail.cs
@@ -309,7 +309,10 @@ void RegisterCommands()
         AddLocalizedCmd("warden.fire_guard_cmd","admin : Remove all guards apart from warden",warden.FireGuardCmd);
 
         AddLocalizedCmd("warden.give_freeday_cmd","give t a freeday",warden.GiveFreedayCmd);
-        AddLocalizedCmd("warden.give_pardon_cmd","give t a freeday",warden.GivePardonCmd);
+        AddLocalizedCmd("warden.give_pardon_cmd","give t a pardon",warden.GivePardonCmd);
+
+        AddLocalizedCmd("warden.countdown_cmd","start a countdown",warden.CountdownCmd);
+        AddLocalizedCmd("warden.countdown_abort_cmd","abort a countdown",warden.CountdownAbortCmd);  
 
         // reg lr commands
         AddLocalizedCmd("lr.start_lr_cmd","start an lr",lr.LRCmd);
diff --git a/src/LastRequest/LRBase.cs b/src/LastRequest/LRBase.cs
index eaa16f8..a5606ac 100644
--- a/src/LastRequest/LRBase.cs
+++ b/src/LastRequest/LRBase.cs
@@ -278,7 +278,7 @@ public void CountdownStart()
             }
         }
 
-        countdown.Start(lrName,5,this,PrintCountdown,manager.ActivateLR);
+        countdown.Start($"{lrName} starts in",5,this,PrintCountdown,manager.ActivateLR);
     }
 
     public void FailSafeActivate()
diff --git a/src/Lib/Timer.cs b/src/Lib/Timer.cs
index 4cdd906..e575661 100644
--- a/src/Lib/Timer.cs
+++ b/src/Lib/Timer.cs
@@ -28,7 +28,7 @@
 public class Countdown<T>
 {
     public void Start(String countdownName, int countdownDelay,
-        T countdownData,Action<T,int>? countdownPrintFunc, Action <T> countdownCallback)
+        T countdownData,Action<T,int>? countdownPrintFunc, Action <T>? countdownCallback)
     {
         this.delay = countdownDelay;
         this.callback = countdownCallback;
@@ -59,6 +59,11 @@ void Tick()
             {
                 callback(data);
             }
+
+            else
+            {
+                Chat.PrintCentreAll("Countdown over");
+            }
         }
 
         // countdown still active
@@ -73,7 +78,7 @@ void Tick()
             // default print
             else
             {
-                Chat.PrintCentreAll($"{name} is starting in {delay} seconds");
+                Chat.PrintCentreAll($"{name}: {delay}");
             }
         }
     }
diff --git a/src/SpecialDay/SpecialDay.cs b/src/SpecialDay/SpecialDay.cs
index afbb8d9..bb417c9 100644
--- a/src/SpecialDay/SpecialDay.cs
+++ b/src/SpecialDay/SpecialDay.cs
@@ -171,7 +171,7 @@ public void SetupSD(CCSPlayerController? invoke, ChatMenuOption option)
             activeSD.SetupCommon();
 
             // start the countdown for enable
-            countdown.Start($"{name} specialday",delay,0,null,StartSD);
+            countdown.Start($"{name} starts in",delay,0,null,StartSD);
         }
     }
 
diff --git a/src/Warden/Warden.cs b/src/Warden/Warden.cs
index 8edb1c7..5f710a7 100644
--- a/src/Warden/Warden.cs
+++ b/src/Warden/Warden.cs
@@ -201,6 +201,8 @@ public void SetupPlayerGuns(CCSPlayerController? player)
         return jailPlayers[player.Slot];
     }
     
+    Countdown<int> chatCountdown = new Countdown<int>();
+
     const int INAVLID_SLOT = -3;   
 
     int wardenSlot = INAVLID_SLOT;
diff --git a/src/Warden/WardenCmd.cs b/src/Warden/WardenCmd.cs
index 6582741..c0dc32c 100644
--- a/src/Warden/WardenCmd.cs
+++ b/src/Warden/WardenCmd.cs
@@ -104,6 +104,70 @@ public void WardayCmd(CCSPlayerController? player, CommandInfo command)
     }
 
 
+    public void CountdownCmd(CCSPlayerController? player, CommandInfo command)
+    {
+        if(!player.IsLegal())
+        {
+            return;
+        }
+
+        // must be warden
+        if(!IsWarden(player))
+        {
+            player.LocalizePrefix(WARDEN_PREFIX,"warden.countdown_restrict");
+            return;
+        }
+
+        // must specify location
+        if(command.ArgCount < 2)
+        {
+            player.LocalizePrefix(WARDEN_PREFIX,"warden.countdown_usage");
+            return;
+        }
+
+        // attempt the start the warday
+        String str = command.ArgByIndex(2);
+
+        // attempt to parse optional delay
+        int delay = 20;
+
+        if(command.ArgCount >= 3)
+        {
+            if(Int32.TryParse(command.ArgByIndex(1),out int delayOpt))
+            {
+                delay = delayOpt;
+            }   
+
+            else
+            {
+                player.LocalizePrefix(WARDEN_PREFIX,"warden.countdown_usage");
+                return;      
+            }    
+        }
+
+        chatCountdown.Start(str,delay,0,null,null);
+    }
+
+
+    public void CountdownAbortCmd(CCSPlayerController? player, CommandInfo command)
+    {
+        if(!player.IsLegal())
+        {
+            return;
+        }
+
+        // must be warden
+        if(!IsWarden(player))
+        {
+            player.LocalizePrefix(WARDEN_PREFIX,"warden.countdown_restrict");
+            return;
+        }
+
+        Chat.LocalizeAnnounce(WARDEN_PREFIX,"warden.countdown_abort");
+
+        chatCountdown.Kill();
+    }
+
     (JailPlayer?, CCSPlayerController?) GiveTInternal(CCSPlayerController? invoke, String name, String playerName)
     {
         if(!IsWarden(invoke))
@@ -409,4 +473,6 @@ public void CmdCtGuns(CCSPlayerController? player, CommandInfo command)
         player.GunMenuInternal(true,CtGuns);     
     }
 
+
+
 }
\ No newline at end of file