import random
import wavelink
import pendulum

import onnxruntime as rt

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

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))