import datetime
import time
from typing import Optional

import discord
import humanize
import lavalink
from discord.ext import commands
from discord.ext import tasks
from discord.ext.commands import Context

from bot import TuneBot
from context import CustomContext
from utils.database import AutoJoin
from utils.EmbedGenerator import EmbedGenerator
from utils.classes import BaseCog
from utils.paginator import HelpPaginator


class InformationCog(BaseCog, name="Information"):
    @commands.command(name="ping", aliases=["pong"])
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def ping(self, ctx: Context):
        """Test the latency"""
        avatar = ctx.author.avatar.with_static_format("jpeg")
        emoji = discord.utils.get(ctx.bot.emojis, name="loading")
        start = time.monotonic()
        msg = await ctx.send(
            embed=discord.Embed(description=f"{emoji} Calculating ping")
        )
        millis = (time.monotonic() - start) * 1000
        heartbeat = ctx.bot.latency * 1000
        embed = discord.Embed(color=discord.Color.blue())
        embed.add_field(
            name=":heartbeat: Heartbeat", value=f"`{heartbeat:,.2f}ms`", inline=True
        )
        embed.add_field(
            name=":file_cabinet: ACK", value=f"`{millis:,.2f}ms`", inline=True
        )
        embed.set_footer(text=f"Requested by: {ctx.author}", icon_url=f"{avatar}")
        await msg.edit(embed=embed)

    @commands.command(name="invite")
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def invite(self, ctx: Context):
        """Gets the invite link"""
        await EmbedGenerator.Message(
            ctx, "Add our bot to your server:", self.bot.invite_link
        )

    @commands.command(name="wlinfo")
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def wlinfo(self, ctx: CustomContext):
        """Retrieve various node/server/player information"""
        player = self.bot.lavalink.player_manager.get(ctx.guild.id)
        node = player.node

        used = humanize.naturalsize(node.stats.memory_used)
        total = humanize.naturalsize(node.stats.memory_allocated)
        free = humanize.naturalsize(node.stats.memory_free)
        cpu = node.stats.cpu_cores

        fmt = (
            f"**WaveLink:** `{lavalink.__version__}`\n\n"
            f"Connected to `{len(self.bot.lavalink.nodes)}` nodes.\n"
            f"Best available Node `{self.bot.lavalink.get_best_node().__repr__()}`\n"
            f"`{len(self.bot.lavalink.players)}` players are distributed on nodes.\n"
            f"`{node.stats.players}` players are distributed on server.\n"
            f"`{node.stats.playing_players}` players are playing on server.\n\n"
            f"Server Memory: `{used}/{total}` | `({free} free)`\n"
            f"Server CPU: `{cpu}`\n\n"
            f"Server Uptime: `{datetime.timedelta(milliseconds=node.stats.uptime)}`"
        )
        await ctx.send(fmt)
        AutoJoin.get_channels()

    @commands.command(name="help", aliases=["about", "info"])
    @commands.cooldown(1, 1, commands.BucketType.user)
    async def about(
        self,
        ctx: Context,
        command: Optional[str] = commands.Option(
            description="Show help for a command or category"
        ),
    ):
        """Retrieve a list of possible commands"""
        if command:
            entity = self.bot.get_cog(command) or self.bot.get_command(command)

            if entity is None:
                clean = command.replace("@", "@\u200b")
                return await ctx.send(f'Command or category "{clean}" not found.')
            elif isinstance(entity, discord.ext.commands.Command):
                p = await HelpPaginator.from_command(ctx, entity)
            else:
                p = await HelpPaginator.from_cog(ctx, entity)
            return await p.paginate()

        info = self.bot.config["info"]
        title = info["name"] + " Help"
        descr = info["description"]
        color = self.bot.colors["embed"]

        embed = discord.Embed(color=color, title=title, description=descr)

        display_cogs = {
            "Information": ":information_source:",
            "Music": ":musical_note:",
            "Settings": ":gear:",
        }

        for cog_name, cog_icon in display_cogs.items():
            cog = self.bot.cogs.get(cog_name)
            if not cog:
                continue
            cogname_str = f"{cog_icon} {cog_name}"
            commands = [f"`{cmd.name}`" for cmd in cog.get_commands() if not cmd.hidden]
            commands_str = ", ".join(commands)
            embed.add_field(name=cogname_str, value=commands_str, inline=False)

        avatar = ctx.author.avatar.with_static_format("jpeg")
        embed.set_footer(text=f"Requested by: {ctx.author}", icon_url=avatar)
        await ctx.send(embed=embed)


def setup(bot: TuneBot):
    bot.remove_command("help")
    bot.add_cog(InformationCog(bot))