You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

226 lines
9.6 KiB
Python

import discord, toml
from discord.ext import commands
from discord.commands import option
with open("config.toml", "r") as f:
config = toml.load(f)
guild_ids=config["bot"]["guild_ids"]
with open("cogs/essentials/config.toml", "r") as f:
esconf = toml.load(f)
class EssentialCommands(commands.Cog):
def __init__(self, bot):
self.bot = bot
@discord.Cog.listener()
async def on_member_join(self, member):
# Send a welcome message with an embed to the specified channel
if not esconf["events"]["welcome_enabled"]:
return 0
welcome_title = esconf['events']['welcome_title']
welcome_message = esconf['events']['welcome_content']
welcome_channel = self.bot.get_channel(esconf["events"]['welcome_channel_id'])
embed = discord.Embed(title=welcome_title.format(member=member), description=welcome_message.format(member=member), color=0x00ff00)
await welcome_channel.send(embed=embed)
@discord.Cog.listener()
async def on_member_remove(self, member):
# Send a goodbye message with an embed to the specified channel
if not esconf["events"]["goodbye_enabled"]:
return 0
gb_title = esconf['events']['goodbye_title']
gb_message = esconf['events']['goodbye_content']
gb_channel = self.bot.get_channel(esconf["events"]['welcome_channel_id'])
embed = discord.Embed(title=gb_title.format(member=member), description=gb_message.format(member=member), color=0xff0000)
await gb_channel.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
async def ping(self, ctx):
if not esconf["commands"]["ping"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.respond('Pong!')
@commands.slash_command(guild_ids=guild_ids)
@option('msg', description="The message to say.")
async def say(self, ctx, message):
if not esconf["commands"]["say"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.send(message)
@commands.slash_command(guild_ids=guild_ids)
async def serverinfo(self, ctx):
if not esconf["commands"]["serverinfo"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.defer()
embed = discord.Embed(title=f"{ctx.guild.name} Info", colour=discord.Colour.blurple())
embed.add_field(name="Roles", value=len(ctx.guild.roles), inline=False)
embed.add_field(name="Members", value=ctx.guild.member_count, inline=False)
embed.add_field(name="Owner", value=ctx.guild.owner.mention, inline=False)
embed.add_field(name="Text Channels", value=len(ctx.guild.text_channels), inline=False)
embed.add_field(name="Voice Channels", value=len(ctx.guild.voice_channels), inline=False)
embed.add_field(name="Boost Level", value=ctx.guild.premium_tier, inline=False)
embed.add_field(name="Created", value=ctx.guild.created_at, inline=False)
embed.add_field(name="ID", value=ctx.guild.id, inline=False)
embed.add_field(name="Max File Size", value=f"{str(int(ctx.guild.filesize_limit / (1024 * 1024)))} MB", inline=False)
icon = ctx.guild.icon
embed.set_thumbnail(url=icon)
await ctx.followup.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
@option('ammount', description="The of messages to clear.", required=True)
@commands.has_permissions(manage_messages=True)
async def clean(self, ctx, amount: int = 10):
if not esconf["commands"]["clean"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.defer()
try:
await ctx.channel.purge(limit=amount + 1)
except:
await ctx.followup.send(f"Could not purge all of the messages.")
else:
await ctx.followup.send(f"Purged {amount} messages.")
@commands.slash_command(guild_ids=guild_ids)
@commands.has_permissions(manage_channels=True)
async def lock(self, ctx, *, reason=None):
if not esconf["commands"]["lock"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.defer()
channel = ctx.channel
overwrite = channel.overwrites_for(ctx.guild.default_role)
overwrite.send_messages = False
await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
embed = discord.Embed(title="Locked", description=f"Reason: {reason}")
await ctx.followup.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
@commands.has_permissions(manage_channels=True)
async def unlock(self, ctx, *, reason=None):
if not esconf["commands"]["unlock"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.defer()
channel = ctx.channel
overwrite = channel.overwrites_for(ctx.guild.default_role)
overwrite.send_messages = True
await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
embed = discord.Embed(title="Unlocked", description=f"Reason: {reason}")
await ctx.followup.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
@commands.has_permissions(manage_channels=True)
async def lockdown(self, ctx, *, reason=None):
if not esconf["commands"]["lockdown"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.defer()
for channel in ctx.guild.channels:
if isinstance(channel, discord.TextChannel):
overwrite = channel.overwrites_for(ctx.guild.default_role)
overwrite.send_messages = False
await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
embed = discord.Embed(title="Server Locked Down", description=f"Reason: {reason}")
await ctx.followup.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
@commands.has_permissions(manage_channels=True)
async def unlockdown(self, ctx):
if not esconf["commands"]["unlockdown"]:
await ctx.respond("This command is disabled.")
return 0
await ctx.defer()
for channel in ctx.guild.channels:
if isinstance(channel, discord.TextChannel):
overwrite = channel.overwrites_for(ctx.guild.default_role)
overwrite.send_messages = None
await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
embed = discord.Embed(title="Server Unlocked", description="All channels have been unlocked.")
await ctx.followup.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
@option('user', description="The username of the user you want the avatars picture incl 1234 numbers", required=True)
async def avatar(self, ctx, member: discord.Member):
if not esconf["commands"]["avatar"]:
await ctx.respond("This command is disabled.")
return 0
try:
avatar_url = member.avatar.url
except:
await ctx.respond(f"Unable to get {member.name}'s avatar as he probably has a default avatar.")
else:
await ctx.respond(f"{member.name}'s avatar: {avatar_url}")
@commands.slash_command(guild_ids=guild_ids)
@option('member', description="The username of the user to run the check on.")
async def whois(self, ctx, member: discord.Member = None):
if not esconf["commands"]["whois"]:
await ctx.respond("This command is disabled.")
return 0
member = member or ctx.author
roles = [role.mention for role in member.roles[1:]]
if not roles:
roles = ["None"]
embed = discord.Embed(title=f"{member.display_name}'s Info", color=member.color)
try:
embed.set_thumbnail(url=member.avatar.url)
except: pass
embed.add_field(name="Name", value=str(member), inline=False)
embed.add_field(name="ID", value=member.id, inline=False)
embed.add_field(name="Account Created", value=member.created_at.strftime("%d/%m/%Y %H:%M:%S"), inline=False)
embed.add_field(name="Join Date", value=member.joined_at.strftime("%d/%m/%Y %H:%M:%S"), inline=False)
if not member.activity == None:
embed.add_field(name="Activity", value=member.activity, inline=False)
embed.add_field(name="Roles", value=" ".join(roles), inline=False)
await ctx.send(embed=embed)
@commands.slash_command(guild_ids=guild_ids)
@option('title', description="The biggest text.", required=True)
@option('content', description="The actuall text to put in the embed.", required=True)
@option('footer', description="The lower text.", required=False)
@option('color', description="The color as a word example: Color: dark_blue", required=False)
async def embed(self, ctx, title: str, content: str, color: str = None, footer: str = None):
if not esconf["commands"]["embed"]:
await ctx.respond("This command is disabled.")
return 0
embed = discord.Embed(title=title, description=content)
if color:
try:
atr = getattr(discord.Color, color)
embed.color = atr()
except Exception as e:
bot.api.logger("inf", f"Command ivoked a error wrong color: {e}")
ctx.send("ERR: Invalid color.")
return 1
if footer:
embed.set_footer(text=footer)
await ctx.send(embed=embed)
def setup(bot):
bot.add_cog(EssentialCommands(bot))