2022-06-21 20:30:11 +02:00
|
|
|
from typing import Optional
|
|
|
|
|
|
|
|
import discord
|
2022-06-22 01:56:13 +02:00
|
|
|
from discord import app_commands
|
2022-06-21 20:30:11 +02:00
|
|
|
from discord.ext import commands
|
|
|
|
|
2022-06-22 00:49:00 +02:00
|
|
|
from database.crud import custom_commands
|
|
|
|
from database.exceptions.constraints import DuplicateInsertException
|
|
|
|
from database.exceptions.not_found import NoResultFoundException
|
2022-06-21 20:30:11 +02:00
|
|
|
from didier import Didier
|
2022-06-22 01:56:13 +02:00
|
|
|
from didier.data.modals.custom_commands import CreateCustomCommand
|
2022-06-21 20:30:11 +02:00
|
|
|
|
|
|
|
|
|
|
|
class Owner(commands.Cog):
|
|
|
|
"""Cog for owner-only commands"""
|
|
|
|
|
|
|
|
client: Didier
|
|
|
|
|
2022-06-22 01:56:13 +02:00
|
|
|
# Slash groups
|
|
|
|
add_slash = app_commands.Group(name="add", description="Add something new to the database")
|
|
|
|
|
2022-06-21 20:30:11 +02:00
|
|
|
def __init__(self, client: Didier):
|
|
|
|
self.client = client
|
|
|
|
|
2022-06-22 00:49:00 +02:00
|
|
|
async def cog_check(self, ctx: commands.Context) -> bool:
|
|
|
|
"""Global check for every command in this cog, so we don't have to add
|
|
|
|
is_owner() to every single command separately
|
|
|
|
"""
|
2022-06-22 02:09:16 +02:00
|
|
|
# pylint: disable=W0236 # Pylint thinks this can't be async, but it can
|
2022-06-22 00:49:00 +02:00
|
|
|
return await self.client.is_owner(ctx.author)
|
|
|
|
|
2022-06-21 20:30:11 +02:00
|
|
|
@commands.command(name="Sync")
|
|
|
|
async def sync(self, ctx: commands.Context, guild: Optional[discord.Guild] = None):
|
|
|
|
"""Sync all application-commands in Discord"""
|
2022-06-22 01:56:13 +02:00
|
|
|
if guild is not None:
|
|
|
|
self.client.tree.copy_global_to(guild=guild)
|
|
|
|
await self.client.tree.sync(guild=guild)
|
|
|
|
else:
|
|
|
|
self.client.tree.clear_commands(guild=None)
|
|
|
|
await self.client.tree.sync()
|
|
|
|
|
2022-06-21 20:30:11 +02:00
|
|
|
await ctx.message.add_reaction("🔄")
|
|
|
|
|
2022-06-22 00:49:00 +02:00
|
|
|
@commands.group(name="Add", case_insensitive=True, invoke_without_command=False)
|
2022-06-22 01:56:13 +02:00
|
|
|
async def add_msg(self, ctx: commands.Context):
|
|
|
|
"""Command group for [add X] message commands"""
|
2022-06-22 00:49:00 +02:00
|
|
|
|
2022-06-22 01:56:13 +02:00
|
|
|
@add_msg.command(name="Custom")
|
2022-06-22 00:49:00 +02:00
|
|
|
async def add_custom(self, ctx: commands.Context, name: str, *, response: str):
|
|
|
|
"""Add a new custom command"""
|
|
|
|
async with self.client.db_session as session:
|
|
|
|
try:
|
|
|
|
await custom_commands.create_command(session, name, response)
|
|
|
|
await self.client.confirm_message(ctx.message)
|
|
|
|
except DuplicateInsertException:
|
|
|
|
await ctx.reply("Er bestaat al een commando met deze naam.")
|
|
|
|
await self.client.reject_message(ctx.message)
|
|
|
|
|
2022-06-22 01:56:13 +02:00
|
|
|
@add_msg.command(name="Alias")
|
2022-06-22 00:49:00 +02:00
|
|
|
async def add_alias(self, ctx: commands.Context, command: str, alias: str):
|
|
|
|
"""Add a new alias for a custom command"""
|
|
|
|
async with self.client.db_session as session:
|
|
|
|
try:
|
|
|
|
await custom_commands.create_alias(session, command, alias)
|
|
|
|
await self.client.confirm_message(ctx.message)
|
|
|
|
except NoResultFoundException:
|
|
|
|
await ctx.reply(f'Geen commando gevonden voor "{command}".')
|
|
|
|
await self.client.reject_message(ctx.message)
|
|
|
|
except DuplicateInsertException:
|
|
|
|
await ctx.reply("Er bestaat al een commando met deze naam.")
|
|
|
|
await self.client.reject_message(ctx.message)
|
|
|
|
|
2022-06-22 01:56:13 +02:00
|
|
|
@add_slash.command(name="custom", description="Add a custom command")
|
|
|
|
async def add_custom_slash(self, interaction: discord.Interaction):
|
|
|
|
"""Slash command to add a custom command"""
|
|
|
|
if not self.client.is_owner(interaction.user):
|
|
|
|
return interaction.response.send_message(
|
|
|
|
"Je hebt geen toestemming om dit commando uit te voeren.", ephemeral=True
|
|
|
|
)
|
|
|
|
|
|
|
|
await interaction.response.defer(ephemeral=True)
|
|
|
|
modal = CreateCustomCommand()
|
|
|
|
await interaction.response.send_message(modal)
|
|
|
|
|
2022-06-22 00:49:00 +02:00
|
|
|
@commands.group(name="Edit")
|
|
|
|
async def edit(self, ctx: commands.Context):
|
|
|
|
"""Command group for [edit X] commands"""
|
|
|
|
|
2022-06-21 20:30:11 +02:00
|
|
|
|
|
|
|
async def setup(client: Didier):
|
|
|
|
"""Load the cog"""
|
|
|
|
await client.add_cog(Owner(client))
|