TuneBot/cogs/information.py

import discord
from discord.ext import tasks, commands

import time
from discord.ext.commands import Context
import humanize
import datetime
import lavalink
from bot import ChristmasBot
from utils.EmbedGenerator import EmbedGenerator
from utils.paginator import HelpPaginator


class InformationCog(commands.Cog, name="Information"):
    def __init__(self, bot: ChristmasBot):
        self.bot = bot
        self.is_help_msg = True
        self.update_status.start()

    @tasks.loop(seconds=30.0)
    async def update_status(self):
        await self.bot.wait_until_ready()
        title = "ck!connect | ck!help"
        if self.is_help_msg:
            title = "Tfoe broer"
            self.is_help_msg = not self.is_help_msg
        await self.bot.change_presence(activity=discord.Game(title))

    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    @commands.command(description="PONG!", aliases=["pong"])
    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", description="Gets the invite link!", slash_commands=True
    )
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def invite(self, ctx: Context):
        """Gets the invite link!"""
        embed = await EmbedGenerator.Message(
            ctx, "Add our bot to your server:", self.bot.invite_link
        )
        await ctx.send(embed=embed)

    @commands.command(
        name="wlinfo",
        description="Retrieve various Node/Server/Player information.",
        slash_commands=True,
    )
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def wlinfo(self, ctx: Context):
        """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)

    @commands.command(name="help", aliases=["about", "info"], slash_command=True)
    @commands.cooldown(1, 1, commands.BucketType.user)
    async def about(self, ctx: Context):
        """Exobot command list"""
        if None:
            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: ChristmasBot):
    bot.remove_command("help")
    bot.add_cog(InformationCog(bot))