353 lines
12 KiB
Python
353 lines
12 KiB
Python
|
import discord
|
||
|
|
from discord import app_commands
|
||
|
|
from discord.ext import commands
|
||
|
|
from discord.ext.commands import Context
|
||
|
|
|
||
|
|
from helpers import checks, db_manager
|
||
|
|
|
||
|
|
|
||
|
|
class Owner(commands.Cog, name="owner"):
|
||
|
|
def __init__(self, bot):
|
||
|
|
self.bot = bot
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.command(
|
||
|
|
name="sync",
|
||
|
|
description="Synchonizes the slash commands.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(scope="The scope of the sync. Can be `global` or `guild`")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def sync(self, context: Context, scope: str) -> None:
|
||
|
|
if scope == "global":
|
||
|
|
await context.bot.tree.sync()
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Slash Commands Sync",
|
||
|
|
description="Slash commands have been globally synchronized.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
elif scope == "guild":
|
||
|
|
context.bot.tree.copy_global_to(guild=context.guild)
|
||
|
|
await context.bot.tree.sync(guild=context.guild)
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Slash Commands Sync",
|
||
|
|
description="Slash commands have been synchronized in this guild.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Invalid Scope",
|
||
|
|
description="The scope must be `global` or `guild`.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.command(
|
||
|
|
name="unsync",
|
||
|
|
description="Unsynchonizes the slash commands.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(scope="The scope of the sync. Can be `global`, `current_guild` or `guild`")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def unsync(self, context: Context, scope: str) -> None:
|
||
|
|
if scope == "global":
|
||
|
|
context.bot.tree.clear_commands(guild=None)
|
||
|
|
await context.bot.tree.sync()
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Slash Commands Unsync",
|
||
|
|
description="Slash commands have been globally unsynchronized.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
elif scope == "guild":
|
||
|
|
context.bot.tree.clear_commands(guild=context.guild)
|
||
|
|
await context.bot.tree.sync(guild=context.guild)
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Slash Commands Unsync",
|
||
|
|
description="Slash commands have been unsynchronized in this guild.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Invalid Scope",
|
||
|
|
description="The scope must be `global` or `guild`.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="load",
|
||
|
|
description="Load a cog",
|
||
|
|
)
|
||
|
|
@app_commands.describe(cog="The name of the cog to load")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def load(self, context: Context, cog: str) -> None:
|
||
|
|
|
||
|
|
try:
|
||
|
|
await self.bot.load_extension(f"cogs.{cog}")
|
||
|
|
except Exception:
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Error!",
|
||
|
|
description=f"Could not load the `{cog}` cog.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Load",
|
||
|
|
description=f"Successfully loaded the `{cog}` cog.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="unload",
|
||
|
|
description="Unloads a cog.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(cog="The name of the cog to unload")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def unload(self, context: Context, cog: str) -> None:
|
||
|
|
|
||
|
|
try:
|
||
|
|
await self.bot.unload_extension(f"cogs.{cog}")
|
||
|
|
except Exception:
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Error!",
|
||
|
|
description=f"Could not unload the `{cog}` cog.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Unload",
|
||
|
|
description=f"Successfully unloaded the `{cog}` cog.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="reload",
|
||
|
|
description="Reloads a cog.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(cog="The name of the cog to reload")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def reload(self, context: Context, cog: str) -> None:
|
||
|
|
|
||
|
|
try:
|
||
|
|
await self.bot.reload_extension(f"cogs.{cog}")
|
||
|
|
except Exception:
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Error!",
|
||
|
|
description=f"Could not reload the `{cog}` cog.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Reload",
|
||
|
|
description=f"Successfully reloaded the `{cog}` cog.",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="shutdown",
|
||
|
|
description="Make the bot shutdown.",
|
||
|
|
)
|
||
|
|
@checks.is_owner()
|
||
|
|
async def shutdown(self, context: Context) -> None:
|
||
|
|
|
||
|
|
embed = discord.Embed(
|
||
|
|
description="Shutting down. Bye! :wave:",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
await self.bot.close()
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="say",
|
||
|
|
description="The bot will say anything you want.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(message="The message that should be repeated by the bot")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def say(self, context: Context, *, message: str) -> None:
|
||
|
|
|
||
|
|
await context.send(message)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="embed",
|
||
|
|
description="The bot will say anything you want, but within embeds.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(message="The message that should be repeated by the bot")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def embed(self, context: Context, *, message: str) -> None:
|
||
|
|
|
||
|
|
embed = discord.Embed(
|
||
|
|
description=message,
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_group(
|
||
|
|
name="blacklist",
|
||
|
|
description="Get the list of all blacklisted users.",
|
||
|
|
)
|
||
|
|
@checks.is_owner()
|
||
|
|
async def blacklist(self, context: Context) -> None:
|
||
|
|
|
||
|
|
if context.invoked_subcommand is None:
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Blacklist",
|
||
|
|
description="You need to specify a subcommand.\n\n**Subcommands:**\n`add` - Add a user to the blacklist.\n`remove` - Remove a user from the blacklist.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@blacklist.command(
|
||
|
|
base="blacklist",
|
||
|
|
name="show",
|
||
|
|
description="Shows the list of all blacklisted users.",
|
||
|
|
)
|
||
|
|
@checks.is_owner()
|
||
|
|
async def blacklist_show(self, context: Context) -> None:
|
||
|
|
|
||
|
|
blacklisted_users = await db_manager.get_blacklisted_users()
|
||
|
|
if len(blacklisted_users) == 0:
|
||
|
|
embed = discord.Embed(
|
||
|
|
description="There are currently no blacklisted users.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Blacklisted users",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
users = []
|
||
|
|
for bluser in blacklisted_users:
|
||
|
|
user = self.bot.get_user(int(bluser[0])) or await self.bot.fetch_user(int(bluser[0]))
|
||
|
|
users.append(
|
||
|
|
f"• {user.mention} ({user}) - Blacklisted <t:{bluser[1]}>")
|
||
|
|
embed.description = "\n".join(users)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@blacklist.command(
|
||
|
|
base="blacklist",
|
||
|
|
name="add",
|
||
|
|
description="Lets you add a user from not being able to use the bot.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(user="The user that should be added to the blacklist")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def blacklist_add(self, context: Context, user: discord.User) -> None:
|
||
|
|
|
||
|
|
user_id = user.id
|
||
|
|
if await db_manager.is_blacklisted(user_id):
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Error!",
|
||
|
|
description=f"**{user.name}** is already in the blacklist.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
total = await db_manager.add_user_to_blacklist(user_id)
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="User Blacklisted",
|
||
|
|
description=f"**{user.name}** has been successfully added to the blacklist",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
embed.set_footer(
|
||
|
|
text=f"There {'is' if total == 1 else 'are'} now {total} {'user' if total == 1 else 'users'} in the blacklist"
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@blacklist.command(
|
||
|
|
base="blacklist",
|
||
|
|
name="remove",
|
||
|
|
description="Lets you remove a user from not being able to use the bot.",
|
||
|
|
)
|
||
|
|
@app_commands.describe(user="The user that should be removed from the blacklist.")
|
||
|
|
@checks.is_owner()
|
||
|
|
async def blacklist_remove(self, context: Context, user: discord.User) -> None:
|
||
|
|
|
||
|
|
user_id = user.id
|
||
|
|
if not await db_manager.is_blacklisted(user_id):
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Error!",
|
||
|
|
description=f"**{user.name}** is not in the blacklist.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
return
|
||
|
|
total = await db_manager.remove_user_from_blacklist(user_id)
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="User removed from blacklist",
|
||
|
|
description=f"**{user.name}** has been successfully removed from the blacklist",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
embed.set_footer(
|
||
|
|
text=f"There {'is' if total == 1 else 'are'} now {total} {'user' if total == 1 else 'users'} in the blacklist"
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
@commands.hybrid_command(
|
||
|
|
name="leave",
|
||
|
|
description="Lets you remove a the bot from a server.",
|
||
|
|
)
|
||
|
|
@checks.is_owner()
|
||
|
|
@app_commands.describe(guild_id="Supply a guild ID.")
|
||
|
|
async def leave(self, context, guild_id, reason: str = "This BOT is not allowed to be on other Guild other that the guild it was made for") -> None:
|
||
|
|
|
||
|
|
try:
|
||
|
|
await self.bot.get_guild(int(guild_id)).leave()
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="BOT Removed frome guild",
|
||
|
|
description=f"I left: {guild_id}",
|
||
|
|
color=0x9C84EF
|
||
|
|
)
|
||
|
|
embed.add_field(
|
||
|
|
name="Reason:",
|
||
|
|
value=reason
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
except Exception as e:
|
||
|
|
embed = discord.Embed(
|
||
|
|
title="Error!",
|
||
|
|
description="An error occurred while trying to remove the bot. Make sure GUILD_ID is an existing ID.",
|
||
|
|
color=0xE02B2B
|
||
|
|
)
|
||
|
|
await context.send(embed=embed)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
|
||
|
|
async def setup(bot):
|
||
|
|
await bot.add_cog(Owner(bot))
|