From 2522e8a7a009b88a2a122fcc4632647e7e2375ea Mon Sep 17 00:00:00 2001
From: Phoenix/HotaruBlaze <DeriousHD@gmail.com>
Date: Fri, 12 Jun 2026 05:57:25 +0100
Subject: [PATCH] Add timer and better handling of status

---
 src/handlers.rs | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/src/handlers.rs b/src/handlers.rs
index 99f0452..6b6de91 100644
--- a/src/handlers.rs
+++ b/src/handlers.rs
@@ -12,6 +12,8 @@ use serenity::{
     },
 };
 use std::sync::Arc;
+use std::time::Duration;
+use tokio::time::sleep;
 
 pub struct Handler {
     pub db: Arc<Database>,
@@ -19,11 +21,37 @@ pub struct Handler {
 
 #[async_trait]
 impl serenity::client::EventHandler for Handler {
-    async fn ready(&self, _ctx: Context, ready: Ready) {
+    async fn ready(&self, ctx: Context, ready: Ready) {
         let shard_info = ready
             .shard
             .map_or("N/A".to_string(), |s| format!("{}/{}", s.id.0, s.total));
         println!("{} is connected! [Shard {}]", ready.user.name, shard_info);
+
+        let db = Arc::clone(&self.db);
+        tokio::spawn(async move {
+            let mut last_guild_count = 0i64;
+            let mut last_ban_count = 0i64;
+            loop {
+                let guild_count = ctx.cache.guild_count() as i64;
+                let ban_count = db.total_bans().unwrap_or(0);
+                if guild_count != last_guild_count || ban_count != last_ban_count {
+                    ctx.set_presence(
+                        Some(ActivityData::watching(format!(
+                            "{} guilds | {} bans",
+                            guild_count, ban_count
+                        ))),
+                        OnlineStatus::Online,
+                    );
+                    println!(
+                        "Status updated: {} guilds | {} total bans",
+                        guild_count, ban_count
+                    );
+                    last_guild_count = guild_count;
+                    last_ban_count = ban_count;
+                }
+                sleep(Duration::from_secs(300)).await;
+            }
+        });
     }
 
     async fn guild_create(&self, ctx: Context, guild: Guild, _is_new: Option<bool>) {