TuneBot/cogs/information.py

import discord
from discord.ext import tasks, commands

import time
import humanize
import datetime
import lavalink
from utils import metadata
from utils.paginator import HelpPaginator
from discord import Status


class InformationCog(commands.Cog, name="Information"):
    def __init__(self, bot: commands.Bot):
        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 = (await metadata.fetch_metadata())["name"]
            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):
        """Test the latency"""
        avatar = ctx.author.avatar_url_as(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):
        """Gets the invite link!"""
        color = self.bot.colors["embed"]
        avatar = ctx.author.avatar_url_as(static_format='jpeg')
        client_id = ctx.bot.user.id
        link = f"https://discord.com/oauth2/authorize?scope=bot&client_id={client_id}&permissions=70642768"
        embed = discord.Embed(color=color)
        embed.add_field(name='Add our bot to your server:', value=link)
        embed.set_footer(text=f"Requested by: {ctx.author}",
                         icon_url=f"{avatar}")
        await ctx.send(embed=embed)

    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    @commands.command()
    async def wlinfo(self, ctx):
        """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"])
    @commands.cooldown(1, 1, commands.BucketType.user)
    async def about(self, ctx, command: str = None):
        """Exobot command list"""
        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_url_as(static_format='jpeg')
        embed.set_footer(text=f"Requested by: {ctx.author}", icon_url=avatar)
        await ctx.send(embed=embed)


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