From c4819bb6339cb97b22a1f398ff430024edd5cb75 Mon Sep 17 00:00:00 2001
From: mii443 <mii@sfc.wide.ad.jp>
Date: Mon, 26 May 2025 00:02:00 +0900
Subject: [PATCH] check users before restoring instance

---
 src/events/ready.rs | 51 +++++++++++++++++++++++++++++++++++++++++++++
 src/tts/instance.rs | 35 +++++++++++++++++++++++++++++++
 2 files changed, 86 insertions(+)

diff --git a/src/events/ready.rs b/src/events/ready.rs
index 69f6901..f22b410 100644
--- a/src/events/ready.rs
+++ b/src/events/ready.rs
@@ -55,6 +55,57 @@ async fn restore_tts_instances(ctx: &Context) {
             let mut failed_count = 0;
 
             for (guild_id, instance) in instances {
+                // Check if there are users in the voice channel before reconnecting
+                let should_reconnect = match guild_id.channels(&ctx.http).await {
+                    Ok(channels) => {
+                        if let Some(channel) = channels.get(&instance.voice_channel) {
+                            match channel.members(&ctx.cache) {
+                                Ok(members) => {
+                                    let user_count =
+                                        members.iter().filter(|member| !member.user.bot).count();
+                                    user_count > 0
+                                }
+                                Err(_) => {
+                                    // If we can't get members, assume there are no users
+                                    tracing::warn!(
+                                        "Failed to get members for voice channel {} in guild {}",
+                                        instance.voice_channel,
+                                        guild_id
+                                    );
+                                    false
+                                }
+                            }
+                        } else {
+                            // Channel doesn't exist anymore
+                            tracing::warn!(
+                                "Voice channel {} no longer exists in guild {}",
+                                instance.voice_channel,
+                                guild_id
+                            );
+                            false
+                        }
+                    }
+                    Err(_) => {
+                        // If we can't get channels, assume reconnection should not happen
+                        tracing::warn!("Failed to get channels for guild {}", guild_id);
+                        false
+                    }
+                };
+
+                if !should_reconnect {
+                    // Remove instance from database as the channel is empty or doesn't exist
+                    failed_count += 1;
+                    tracing::info!("Skipping reconnection for guild {} - no users in voice channel or channel doesn't exist", guild_id);
+
+                    if let Err(db_err) = database.remove_tts_instance(guild_id).await {
+                        tracing::error!(
+                            "Failed to remove empty TTS instance from database: {}",
+                            db_err
+                        );
+                    }
+                    continue;
+                }
+
                 // Try to reconnect to voice channel
                 match instance.reconnect(ctx).await {
                     Ok(_) => {
diff --git a/src/tts/instance.rs b/src/tts/instance.rs
index 35d1ff3..786ea53 100644
--- a/src/tts/instance.rs
+++ b/src/tts/instance.rs
@@ -55,6 +55,41 @@ impl TTSInstance {
                     self.voice_channel,
                     self.guild
                 );
+
+                // Double-check if there are users in the voice channel after connection
+                match self.guild.channels(&ctx.http).await {
+                    Ok(channels) => {
+                        if let Some(channel) = channels.get(&self.voice_channel) {
+                            match channel.members(&ctx.cache) {
+                                Ok(members) => {
+                                    let user_count =
+                                        members.iter().filter(|member| !member.user.bot).count();
+                                    if user_count == 0 {
+                                        tracing::info!("No users found in voice channel after reconnection, disconnecting from guild {}", self.guild);
+                                        // Disconnect if no users are present
+                                        let _ = manager.remove(self.guild).await;
+                                        return Err(
+                                            "No users in voice channel after reconnection".into()
+                                        );
+                                    }
+                                }
+                                Err(_) => {
+                                    tracing::warn!(
+                                        "Failed to verify members after reconnection for guild {}",
+                                        self.guild
+                                    );
+                                }
+                            }
+                        }
+                    }
+                    Err(_) => {
+                        tracing::warn!(
+                            "Failed to get channels after reconnection for guild {}",
+                            self.guild
+                        );
+                    }
+                }
+
                 Ok(())
             }
             Err(e) => {