discord-havoc/cogs/radio.py

import logging
import traceback
from typing import Optional
from discord.ext import bridge, commands, tasks
from util.radio_util import get_now_playing
import discord
from disc_havoc import Havoc

class Radio(commands.Cog):
    """Radio Cog for Havoc"""
    def __init__(self, bot: Havoc) -> None:
        self.bot: Havoc = bot
        self.channels: dict[str, tuple] = {
            'sfm': (1145182936002482196, 1221615558492029050), # Tuple: Guild Id, Chan Id
        }        
        self.STREAM_URL: str = "https://relay.sfm.codey.lol/aces.ogg"
        self.LAST_NP_TRACK: Optional[str] = None
        try:
            self.radio_state_loop.cancel()
        except Exception as e:
            logging.debug("Failed to cancel radio_state_loop: %s", 
                          str(e))
     
    @commands.Cog.listener()
    async def on_ready(self) -> None:
        """Run on Bot Ready"""
        await self.radio_init()
         
    def is_radio_chan(): # type: ignore
        """Check if channel is radio chan"""
        def predicate(ctx):
            try:
                return ctx.channel.id == 1221615558492029050
            except:
                traceback.print_exc()
                return False
        return commands.check(predicate)     
    
    @bridge.bridge_command()
    @commands.is_owner()    
    async def reinitradio(self, ctx) -> None:
        """
        Reinitialize serious.FM
        
        Args:
            ctx (Any): Discord context
        Returns:
            None
        """
        loop: discord.asyncio.AbstractEventLoop = self.bot.loop
        loop.create_task(self.radio_init())
        await ctx.respond("Done!", ephemeral=True)
    
    async def radio_init(self) -> None:
        """Init Radio"""
        try:
            (radio_guild, radio_chan) = self.channels['sfm']
            guild: Optional[discord.Guild] = self.bot.get_guild(radio_guild)
            if not guild:
                return
            channel = guild.get_channel(radio_chan)             
            if not isinstance(channel, discord.VoiceChannel):
                return
            if not self.bot.voice_clients:
                await channel.connect()
            try:
                try:
                    self.radio_state_loop.cancel()
                except Exception as e:
                    logging.debug("Failed to cancel radio_state_loop: %s", 
                                  str(e))
                self.radio_state_loop.start()
                logging.info("radio_state_loop task started!")
            except:
                logging.critical("Could not start task...")
                traceback.print_exc()
        except:
            traceback.print_exc()
            return
    
    @tasks.loop(seconds=5.0)
    async def radio_state_loop(self) -> None:
        """Radio State Loop"""
        try:
            (radio_guild, radio_chan) = self.channels['sfm']
            try:
                vc: discord.VoiceProtocol = self.bot.voice_clients[-1]
            except:
                logging.debug("No voice client, establishing new VC connection...")
                guild: Optional[discord.Guild] = self.bot.get_guild(radio_guild)
                if not guild:
                    return
                channel = guild.get_channel(radio_chan)
                if not isinstance(channel, discord.VoiceChannel):
                    return
                await channel.connect()
                vc = self.bot.voice_clients[-1]
            
            if not vc.is_playing() or vc.is_paused():  # type: ignore
                """
                Mypy does not seem aware of the is_playing, play, and is_paused methods,
                but they exist.
                """
                logging.info("Detected VC not playing... playing!")
                source: discord.FFmpegAudio = discord.FFmpegOpusAudio(self.STREAM_URL,
                                                 before_options="-timeout 3000000")
                vc.play(source, # type: ignore
                        after=lambda e: logging.info("Error: %s", e) if e\
                            else None) 
            # Get Now Playing (disabled)
            # np_track: Optional[str] = await get_now_playing()
            # if np_track and not self.LAST_NP_TRACK == np_track:
            #     self.LAST_NP_TRACK = np_track
            #     if isinstance(vc.channel, discord.VoiceChannel):
            #         await vc.channel.set_status(f"Now playing: {np_track}")
        except:
            traceback.print_exc()

        # pylint: enable=superfluous-parens
        
    def cog_unload(self) -> None:
        """Run on Cog Unload"""          
        self.radio_state_loop.cancel()
    
def setup(bot) -> None:
    """Run on Cog Load"""
    bot.add_cog(Radio(bot))