mirror of
https://github.com/neoarz/Syntrel.git
synced 2025-12-25 11:40:12 +01:00
71 lines
2.9 KiB
Python
71 lines
2.9 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from discord.ext.commands import Context
|
|
|
|
from .translate import translate_command
|
|
from .codepreview import codepreview_command
|
|
|
|
class Utilities(commands.GroupCog, name="utils"):
|
|
def __init__(self, bot) -> None:
|
|
self.bot = bot
|
|
super().__init__()
|
|
|
|
@commands.group(name="utilities", aliases=["utils"], invoke_without_command=True)
|
|
async def utilities_group(self, context: Context):
|
|
embed = discord.Embed(
|
|
title="Utilities Commands",
|
|
description="Use `.utils <subcommand>` or `/utils <subcommand>`.",
|
|
color=0x7289DA
|
|
)
|
|
embed.set_author(name="Utilities", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
|
|
embed.add_field(name="Available", value="translate, codepreview", inline=False)
|
|
await context.send(embed=embed)
|
|
|
|
async def _invoke_hybrid(self, context: Context, name: str, **kwargs):
|
|
command = self.bot.get_command(name)
|
|
if command is not None:
|
|
await context.invoke(command, **kwargs)
|
|
else:
|
|
await context.send(f"Unknown utilities command: {name}")
|
|
|
|
def _require_group_prefix(context: Context) -> bool:
|
|
if getattr(context, "interaction", None):
|
|
return True
|
|
group = getattr(getattr(context, "cog", None), "qualified_name", "").lower()
|
|
if not group:
|
|
return True
|
|
prefix = context.prefix or ""
|
|
content = context.message.content.strip().lower()
|
|
return content.startswith(f"{prefix}{group} ")
|
|
|
|
@utilities_group.command(name="translate")
|
|
async def utilities_group_translate(self, context: Context, text: str = None, to_lang: str = "en", from_lang: str = None):
|
|
await self._invoke_hybrid(context, "translate", text=text, to_lang=to_lang, from_lang=from_lang)
|
|
|
|
@utilities_group.command(name="codepreview")
|
|
async def utilities_group_codepreview(self, context: Context, url: str = None):
|
|
await self._invoke_hybrid(context, "codepreview", url=url)
|
|
|
|
@commands.check(_require_group_prefix)
|
|
@commands.hybrid_command(
|
|
name="translate",
|
|
description="Translate text to another language"
|
|
)
|
|
async def translate(self, context, text: str = None, to_lang: str = "en", from_lang: str = None):
|
|
return await translate_command()(self, context, text=text, to_lang=to_lang, from_lang=from_lang)
|
|
|
|
@commands.check(_require_group_prefix)
|
|
@commands.hybrid_command(
|
|
name="codepreview",
|
|
description="Preview code from GitHub URLs"
|
|
)
|
|
async def codepreview(self, context, url: str = None):
|
|
return await codepreview_command()(self, context, url=url)
|
|
|
|
async def setup(bot) -> None:
|
|
cog = Utilities(bot)
|
|
await bot.add_cog(cog)
|
|
|
|
bot.logger.info("Loaded extension 'utilities.translate'")
|
|
bot.logger.info("Loaded extension 'utilities.codepreview'")
|