import random
import wavelink

import onnxruntime as rt

from discord import ApplicationContext, DiscordException, Game, Guild
from discord.ext.commands import Cog

from discord.ext.commands import CommandOnCooldown
from Christmas.UI.Embed import Default_Embed
from Christmas.config import ChristmasConfig
from Christmas.Database import database

model = rt.InferenceSession("Christmas/Tagging/model.onnx", provider_options="CPUExecutionProvider")

class Event(Cog):
    def __init__(self, bot):
        self.bot = bot
        self.config = ChristmasConfig()
    
    @Cog.listener()
    async def on_application_command_error(self, ctx: ApplicationContext, exception: DiscordException) -> None:
        if isinstance(exception, CommandOnCooldown):
            await ctx.respond(embed=Default_Embed.cooldown(exception.retry_after), ephemeral=True)
        else:
            print(exception)
    @Cog.listener()
    async def on_ready(self) -> None:
        print("Ready!")
        await self.bot.change_presence(activity=Game(name="크리스마스에 함께!"))

    
    # connect wavelink
    @Cog.listener()
    async def on_connect(self) -> None:
        await self.bot.wait_until_ready()
        nodes = []
        for node in self.config.LAVALINK:
            nodes.append(wavelink.Node(uri=node["HOST"], password=node["PASSWORD"], identifier=node["IDENTIFIER"]))
        await wavelink.Pool.connect(nodes=nodes, client=self.bot)
    
    @Cog.listener()
    async def on_guild_join(self, guild: Guild) -> None:
        if guild.system_channel is not None:
            await guild.system_channel.send(embed=Default_Embed.guild_join())
        else:
            await random.choice(guild.text_channels).send(embed=Default_Embed.guild_join())
    
    @Cog.listener()
    async def on_guild_update(self, before: Guild, after: Guild) -> None:
        if before.name != after.name:
            await database.update_guild_name(after.id, after.name)
            
    #@Cog.listener()
    #async def on_wavelink_track_end(self, payload):
        
    #@Cog.listener()
    #async def on_wavelink_node_ready(self, node: wavelink.Node):
    #    print(f"Node {node.id} is ready!")
        
def setup(bot):
    bot.add_cog(Event(bot))