mirror of
https://github.com/Matthww/TuneBot.git
synced 2026-09-21 22:47:51 +00:00
Converted information cog
This commit is contained in:
@@ -1,125 +1,16 @@
|
||||
import datetime
|
||||
import time
|
||||
from typing import Optional
|
||||
|
||||
import discord
|
||||
import humanize
|
||||
import lavalink
|
||||
from discord.ext import commands
|
||||
from discord.ext.commands import Context
|
||||
|
||||
from bot import TuneBot
|
||||
from context import CustomContext
|
||||
import typing
|
||||
from cogs.information.interactions import InfoCommands
|
||||
from utils.classes import BaseCog
|
||||
from utils.EmbedGenerator import EmbedGenerator
|
||||
from utils.paginator import HelpPaginator
|
||||
|
||||
if typing.TYPE_CHECKING:
|
||||
from bot import TuneBot
|
||||
|
||||
|
||||
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)
|
||||
|
||||
@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)
|
||||
pass
|
||||
|
||||
|
||||
async def setup(bot: TuneBot):
|
||||
async def setup(bot: "TuneBot"):
|
||||
bot.remove_command("help")
|
||||
await bot.add_cog(InformationCog(bot))
|
||||
bot.tree.add_command(InfoCommands(bot), override=True)
|
||||
|
||||
@@ -0,0 +1,78 @@
|
||||
import typing
|
||||
|
||||
from discord import app_commands
|
||||
from discord import Interaction
|
||||
import humanize
|
||||
import lavalink
|
||||
|
||||
import cogs.music.helper as music_helper
|
||||
from utils.embed import create_embed
|
||||
|
||||
if typing.TYPE_CHECKING:
|
||||
from bot import TuneBot
|
||||
|
||||
|
||||
class InfoCommands(app_commands.Group):
|
||||
def __init__(self, bot: "TuneBot"):
|
||||
super().__init__(name="bot", description="Some extra info")
|
||||
self.bot: "TuneBot" = bot
|
||||
|
||||
@app_commands.command(name="invite", description="I'd happily join your server")
|
||||
async def invite(self, ctx: Interaction):
|
||||
embed = create_embed(ctx.user)
|
||||
embed.title = "My invite link"
|
||||
embed.description = f"[{self.bot.invite_link}]({self.bot.invite_link})"
|
||||
await ctx.response.send_message(embed=embed)
|
||||
|
||||
@app_commands.command(
|
||||
name="help", description="In case you cannot find what you want"
|
||||
)
|
||||
async def about(self, ctx: Interaction):
|
||||
embed = create_embed(ctx.user)
|
||||
|
||||
info = self.bot.config["info"]
|
||||
embed.title = info["name"] + " help"
|
||||
embed.description = info["description"]
|
||||
|
||||
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)
|
||||
|
||||
await ctx.response.send_message(embed=embed)
|
||||
|
||||
@app_commands.command(
|
||||
name="stats", description="Some node/server/player information"
|
||||
)
|
||||
async def wlinfo(self, ctx: Interaction):
|
||||
embed = create_embed(ctx.user)
|
||||
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])
|
||||
|
||||
info = self.bot.config["info"]
|
||||
embed.title = info["name"] + " stats"
|
||||
|
||||
embed.description = (
|
||||
f"**Lavalink:** `{lavalink.__version__}`\n\n"
|
||||
f"Connected to `{len(self.bot.lavalink.node_manager.available_nodes)}` nodes.\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"
|
||||
)
|
||||
await ctx.response.send_message(embed=embed)
|
||||
Reference in New Issue
Block a user