mirror of
https://github.com/Matthww/TuneBot.git
synced 2026-09-21 19:07:52 +00:00
129 lines
5.1 KiB
Python
129 lines
5.1 KiB
Python
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.classes import BaseCog
|
|
from utils.database import AutoJoin
|
|
from utils.EmbedGenerator import EmbedGenerator
|
|
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)
|
|
nodes = self.bot.lavalink.node_manager.available_nodes
|
|
|
|
used = humanize.naturalsize(sum([n.stats.memory_used for n in nodes]))
|
|
total = humanize.naturalsize(sum([n.stats.memory_allocated for n in nodes]))
|
|
free = humanize.naturalsize(sum([n.stats.memory_free for n in nodes]))
|
|
cpu = sum([n.stats.cpu_cores for n in nodes])
|
|
|
|
fmt = (
|
|
f"**Lavalink:** `{lavalink.__version__}`\n\n"
|
|
f"Connected to `{len(self.bot.lavalink.node_manager.available_nodes)}` nodes.\n"
|
|
#f"Best available Node `{self.bot.lavalink.node_manager.find_ideal_node().name.__repr__()}`\n"
|
|
f"`{len(self.bot.lavalink.player_manager.players)}` players are distributed on nodes.\n"
|
|
f"`{sum([n.stats.players for n in nodes])}` players are distributed on server.\n"
|
|
f"`{sum([n.stats.playing_players for n in nodes])}` 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))
|