mostly working idfk
This commit is contained in:
WhatDidYouExpect
parent
b0ba03f97d
commit
4e111b410d
12 changed files with 822 additions and 291 deletions
189
assets/cogs/internal/base_commands.py
Normal file
189
assets/cogs/internal/base_commands.py
Normal file
|
|
@ -0,0 +1,189 @@
|
|||
import os
|
||||
import subprocess
|
||||
from typing import Dict, List
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
import discord.ext
|
||||
import discord.ext.commands
|
||||
from modules.volta.main import _ , set_language
|
||||
from modules.permission import requires_admin
|
||||
from modules.sentenceprocessing import send_message
|
||||
from modules.settings import instance as settings_manager
|
||||
import requests
|
||||
|
||||
settings = settings_manager.settings
|
||||
|
||||
def get_git_origin_raw():
|
||||
try:
|
||||
result = subprocess.run(
|
||||
["git", "config", "--get", "remote.origin.url"],
|
||||
capture_output=True, text=True, check=True
|
||||
)
|
||||
return result.stdout.strip()
|
||||
except Exception:
|
||||
return "http://forgejo.expect.ovh/gooberinc/goober"
|
||||
|
||||
class BaseCommands(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot: discord.ext.commands.Bot = bot
|
||||
|
||||
|
||||
|
||||
@commands.command()
|
||||
async def help(self, ctx: commands.Context) -> None:
|
||||
embed: discord.Embed = discord.Embed(
|
||||
title=f"{_('command_help_embed_title')}",
|
||||
description=f"{_('command_help_embed_desc')}",
|
||||
color=discord.Colour(0x000000),
|
||||
)
|
||||
|
||||
command_categories = {
|
||||
f"{_('command_help_categories_general')}": [
|
||||
"mem",
|
||||
"talk",
|
||||
"about",
|
||||
"ping",
|
||||
"impact",
|
||||
"demotivator",
|
||||
"help",
|
||||
],
|
||||
f"{_('command_help_categories_admin')}": ["stats", "retrain", "setlanguage"],
|
||||
}
|
||||
|
||||
custom_commands: List[str] = []
|
||||
for cog_name, cog in self.bot.cogs.items():
|
||||
for command in cog.get_commands():
|
||||
if (
|
||||
command.name
|
||||
not in command_categories[f"{_('command_help_categories_general')}"]
|
||||
and command.name
|
||||
not in command_categories[f"{_('command_help_categories_admin')}"]
|
||||
):
|
||||
custom_commands.append(command.name)
|
||||
|
||||
if custom_commands:
|
||||
embed.add_field(
|
||||
name=_('command_help_categories_custom'),
|
||||
value="\n".join(
|
||||
[
|
||||
f"{settings['bot']['prefix']}{command}"
|
||||
for command in custom_commands
|
||||
]
|
||||
),
|
||||
inline=False,
|
||||
)
|
||||
|
||||
for category, commands_list in command_categories.items():
|
||||
commands_in_category: str = "\n".join(
|
||||
[f"{settings['bot']['prefix']}{command}" for command in commands_list]
|
||||
)
|
||||
embed.add_field(name=category, value=commands_in_category, inline=False)
|
||||
|
||||
|
||||
await send_message(ctx, embed=embed)
|
||||
|
||||
@requires_admin()
|
||||
@commands.command()
|
||||
async def setlanguage(self, ctx: commands.Context, locale: str) -> None:
|
||||
await ctx.defer()
|
||||
set_language(locale)
|
||||
await ctx.send(":thumbsup:")
|
||||
|
||||
@commands.command()
|
||||
async def ping(self, ctx: commands.Context) -> None:
|
||||
await ctx.defer()
|
||||
latency: int = round(self.bot.latency * 1000)
|
||||
|
||||
embed: discord.Embed = discord.Embed(
|
||||
title="Pong!!",
|
||||
description=(
|
||||
settings["bot"]["misc"]["ping_line"],
|
||||
f"`{_('command_ping_embed_desc')}: {latency}ms`\n",
|
||||
),
|
||||
color=discord.Colour(0x000000),
|
||||
)
|
||||
embed.set_footer(
|
||||
text=f"{_('command_ping_footer')} {ctx.author.name}",
|
||||
icon_url=ctx.author.display_avatar.url,
|
||||
)
|
||||
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
@commands.command()
|
||||
async def about(self, ctx: commands.Context) -> None:
|
||||
embed: discord.Embed = discord.Embed(
|
||||
title=f"{_('command_about_embed_title')}",
|
||||
description="",
|
||||
color=discord.Colour(0x000000),
|
||||
)
|
||||
|
||||
embed.add_field(
|
||||
name=_('command_about_embed_field1'),
|
||||
value=settings['name'],
|
||||
inline=False,
|
||||
)
|
||||
|
||||
embed.add_field(name="Git", value=get_git_origin_raw())
|
||||
await send_message(ctx, embed=embed)
|
||||
|
||||
@commands.command()
|
||||
async def stats(self, ctx: commands.Context) -> None:
|
||||
memory_file: str = "memory.json"
|
||||
file_size: int = os.path.getsize(memory_file)
|
||||
|
||||
with open(memory_file, "r") as file:
|
||||
line_count: int = sum(1 for _ in file)
|
||||
|
||||
embed: discord.Embed = discord.Embed(
|
||||
title=f"{_('command_stats_embed_title')}",
|
||||
description=f"{_('command_stats_embed_desc')}",
|
||||
color=discord.Colour(0x000000),
|
||||
)
|
||||
embed.add_field(
|
||||
name=f"{_('command_stats_embed_field1name')}",
|
||||
value=f"placeholder",
|
||||
inline=False,
|
||||
)
|
||||
|
||||
with open(settings["splash_text_loc"], "r") as f:
|
||||
splash_text = "".join(f.readlines())
|
||||
|
||||
embed.add_field(
|
||||
name=_('command_stats_embed_field3name'),
|
||||
value=_('command_stats_embed_field3value').format(
|
||||
NAME=settings["name"],
|
||||
PREFIX=settings["bot"]["prefix"],
|
||||
ownerid=settings["bot"]["owner_ids"][0],
|
||||
PING_LINE=settings["bot"]["misc"]["ping_line"],
|
||||
showmemenabled=settings["bot"]["allow_show_mem_command"],
|
||||
USERTRAIN_ENABLED=settings["bot"]["user_training"],
|
||||
song=settings["bot"]["misc"]["active_song"],
|
||||
splashtext=splash_text
|
||||
),
|
||||
inline=False,
|
||||
)
|
||||
|
||||
|
||||
await send_message(ctx, embed=embed)
|
||||
|
||||
@commands.command()
|
||||
async def mem(self, ctx: commands.Context) -> None:
|
||||
if not settings["bot"]["allow_show_mem_command"]:
|
||||
return
|
||||
|
||||
with open(settings["bot"]["active_memory"], "rb") as f:
|
||||
data: bytes = f.read()
|
||||
|
||||
response = requests.post(
|
||||
"https://litterbox.catbox.moe/resources/internals/api.php",
|
||||
data={"reqtype": "fileupload", "time": "1h"},
|
||||
files={"fileToUpload": data},
|
||||
)
|
||||
|
||||
await send_message(ctx, response.text)
|
||||
|
||||
|
||||
async def setup(bot: discord.ext.commands.Bot):
|
||||
print("Setting up base_commands")
|
||||
bot.remove_command("help")
|
||||
await bot.add_cog(BaseCommands(bot))
|
||||
Loading…
Add table
Add a link
Reference in a new issue