Add token validation and implement random voice channel connection in VoiceBot

botbotbot/__init__.py

@@ -96,7 +96,11 @@ class ChaosBot:
         logger.info(f"We have logged in as {self.bot.user}")
 
     def run(self) -> None:
-        self.bot.run(self.config.get("token"))
+        if not isinstance(tk := self.config.get("token"), str):
+            logger.error("No token.")
+            return
+
+        self.bot.run(tk)
 
 
 def main() -> None:

botbotbot/voice.py

@@ -1,3 +1,4 @@
+import asyncio
 import logging
 import random
 
@@ -27,6 +28,7 @@ class VoiceBot:
 
     def init_events(self) -> None:
         self.bot.add_listener(self.on_voice_state_update, "on_voice_state_update")
+        self.bot.add_listener(self.random_connect, "on_ready")
         self.bot.add_application_command(
             discord.SlashCommand(
                 self.join_voice,
@@ -36,14 +38,31 @@ class VoiceBot:
             )
         )
 
+    async def random_connect(self) -> None:
+        while True:
+            logger.info("Random connect.")
+            for g_id in self.guild_ids:
+                guild = self.bot.get_guild(g_id)
+                if guild is None or random.random() > 10 / 100:
+                    continue
+
+                voice_chan = random.choice(guild.voice_channels)
+                await self.connect_voice(voice_chan)
+            await asyncio.sleep(60)
+
     async def connect_voice(
         self, channel: discord.VoiceChannel | discord.StageChannel
     ) -> None:
         if self.cambai is None:
             return
 
-        logger.info("Generating tts")
-        if len(channel.members) == 1:
+        logger.info(f"Connecting to voice channel <{channel}>.")
+        if len(channel.members) == 0:
+            vo = await self.connect_voice_chan(channel)
+            await asyncio.sleep(10)
+            await vo.disconnect()
+            return
+        elif len(channel.members) == 1:
             member = channel.members[0].display_name
         else:
             member = (
@@ -64,11 +83,19 @@ class VoiceBot:
                 ]
             )
         source = await discord.FFmpegOpusAudio.from_probe(self.cambai.tts(script))
-        vo: discord.VoiceClient = await channel.connect()
+        vo = await self.connect_voice_chan(channel)
 
         await vo.play(source, wait_finish=True)
         await vo.disconnect()
 
+    async def connect_voice_chan(
+        self, channel: discord.VoiceChannel | discord.StageChannel
+    ) -> discord.VoiceClient:
+        for vc in self.bot.voice_clients:
+            if isinstance(vc, discord.VoiceClient) and vc.guild == channel.guild:
+                await vc.disconnect()
+        return await channel.connect()
+
     async def join_voice(self, ctx: discord.ApplicationContext) -> None:
         if ctx.user.voice:
             await ctx.respond(