mirror of https://github.com/stijndcl/didier
200 lines
8.1 KiB
Python
200 lines
8.1 KiB
Python
from data import constants
|
|
from decorators import help
|
|
import discord
|
|
from discord.ext import commands
|
|
from enums.help_categories import Category
|
|
from functions import checks, config, timeFormatters
|
|
from functions.database import memes, githubs, twitch, dadjoke
|
|
import json
|
|
import os
|
|
|
|
|
|
class ModCommands(commands.Cog):
|
|
|
|
def __init__(self, client):
|
|
self.client = client
|
|
self.utilsCog = self.client.get_cog('Utils')
|
|
|
|
@commands.command(name="Remove", aliases=["Rm"], hidden=True)
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def remove(self, ctx, message: str):
|
|
spl = message.split("/")
|
|
channel = self.client.get_channel(int(spl[-2]))
|
|
message = await channel.fetch_message(int(spl[-1]))
|
|
await message.delete()
|
|
|
|
# Load a cog
|
|
@commands.group(name="Load", usage="[Cog]", case_insensitive=True, invoke_without_command=True)
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def load(self, ctx, extension: str):
|
|
try:
|
|
self.client.load_extension("cogs.{}".format(extension))
|
|
await self.sendDm(constants.myId, "Loaded **{}**".format(extension))
|
|
except discord.ext.commands.errors.ExtensionAlreadyLoaded:
|
|
await self.sendDm(constants.myId, "**{}** has already been loaded".format(extension))
|
|
|
|
@commands.command(name="Config", aliases=["Setup", "Set"], case_insensitive=True, usage="[Categorie] [Value]",
|
|
invoke_without_commands=True)
|
|
@commands.check(checks.isMe)
|
|
@help.Category(Category.Mod)
|
|
async def set(self, ctx, category, value):
|
|
if config.config(category, value):
|
|
await ctx.message.add_reaction("✅")
|
|
|
|
# Load all cogs except for modCommands
|
|
@load.command(name="All")
|
|
async def loadAll(self, ctx):
|
|
for file in os.listdir("./cogs"):
|
|
if file.endswith(".py") and not file == "modCommands.py":
|
|
await self.load(ctx, file[:-3])
|
|
|
|
# Unload a cog
|
|
@commands.group(name="Unload", usage="[Cog]", case_insensitive=True, invoke_without_command=True)
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def unload(self, ctx, extension: str):
|
|
try:
|
|
self.client.unload_extension("cogs.{}".format(extension))
|
|
await self.sendDm(constants.myId, "Unloaded **{}**".format(extension))
|
|
except discord.ext.commands.errors.ExtensionNotLoaded:
|
|
await self.sendDm(constants.myId, "**{}** has already been unloaded".format(extension))
|
|
|
|
# Unload all cogs except for modCommands
|
|
@unload.command(name="All")
|
|
async def unloadAll(self, ctx):
|
|
for file in os.listdir("./cogs"):
|
|
if file.endswith(".py") and not file == "modCommands.py":
|
|
await self.unload(ctx, file[:-3])
|
|
|
|
# Reloads a cog
|
|
@commands.command(name="Reload", aliases=["Update"], usage="[Cog]")
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def reload(self, ctx, cog):
|
|
await self.unload(ctx, cog)
|
|
await self.load(ctx, cog)
|
|
await ctx.message.add_reaction("✅")
|
|
|
|
# Repeat what was said
|
|
@commands.command(name="Repeat", usage="[Text]")
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def repeat(self, ctx, *text):
|
|
await self.utilsCog.removeMessage(ctx.message)
|
|
await ctx.send(" ".join(text))
|
|
|
|
# Add a reaction to a message
|
|
@commands.command(name="Reac", aliases=["Reacc"], usage="[Emoji] [Id]")
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def reac(self, ctx, emoji, messageId):
|
|
channel = ctx.channel
|
|
|
|
# Check if the URL or the Id was passed
|
|
if messageId.count("/") > 3:
|
|
spl = messageId.split("/")
|
|
channel = self.client.get_channel(int(spl[-2]))
|
|
if channel is None:
|
|
return await ctx.send("Ik kan geen kanaal zien met dit Id.")
|
|
messageId = int(spl[-1])
|
|
|
|
await self.utilsCog.removeMessage(ctx.message)
|
|
message = await channel.fetch_message(messageId)
|
|
if message is None:
|
|
return await ctx.send("Ik kan geen bericht zien met dit Id.")
|
|
await message.add_reaction(emoji)
|
|
|
|
# Adds stuff into their databases
|
|
@commands.group(name="Add", usage="[Category] [Args]", case_insensitive=True, invoke_without_command=False)
|
|
@commands.check(checks.isMe)
|
|
@help.Category(category=Category.Mod)
|
|
async def add(self, ctx):
|
|
pass
|
|
|
|
@add.command(name="Dadjoke", aliases=["Dj", "Dad"], usage="[Joke]")
|
|
async def dadjoke(self, ctx, *, joke):
|
|
dadjoke.addJoke(joke)
|
|
await ctx.send("Added ``{}``.".format(joke))
|
|
await ctx.message.add_reaction("✅")
|
|
|
|
@add.command(name="8-Ball", aliases=["8b", "Eightball", "8Ball"], usage="[Response]")
|
|
async def eightball(self, ctx, message):
|
|
with open("files/eightball.json", "r") as fp:
|
|
file = json.load(fp)
|
|
file.append(message)
|
|
with open("files/eightball.json", "w") as fp:
|
|
json.dump(file, fp)
|
|
|
|
# Adds a meme into the database
|
|
@add.command(name="Meme", aliases=["Mem"], usage="[Id] [Name] [Aantal Velden]")
|
|
async def meme(self, ctx, memeid, meme, fields):
|
|
await ctx.send(memes.insert(memeid, meme, fields))
|
|
|
|
# Adds a person's GitHub into the database
|
|
@add.command(name="GitHub", aliases=["Gh", "Git"], usage="[Id] [Link]")
|
|
async def github(self, ctx, userid, link):
|
|
# Allow tagging to work as well
|
|
if len(ctx.message.mentions) == 1:
|
|
userid = ctx.message.mentions[0].id
|
|
githubs.add(userid, link)
|
|
await ctx.send("{}'s GitHub is toegevoegd aan de database.".format(self.utilsCog.getDisplayName(ctx, userid)))
|
|
|
|
# Adds a person's Twitch into the database
|
|
@add.command(name="Twitch", aliases=["Stream", "Streamer", "Tw"])
|
|
async def twitch(self, ctx, userid, link):
|
|
# Allow tagging to work as well
|
|
if len(ctx.message.mentions) == 1:
|
|
userid = ctx.message.mentions[0].id
|
|
twitch.add(userid, link)
|
|
await ctx.send("{}'s Twitch is toegevoegd aan de database.".format(self.utilsCog.getDisplayName(ctx, userid)))
|
|
|
|
@commands.command(name="WhoIs", aliases=["Info", "Whodis"], usage="[@User]")
|
|
@help.Category(Category.Mod)
|
|
async def whois(self, ctx, user):
|
|
if ctx.message.mentions:
|
|
user = ctx.message.mentions[0].id
|
|
|
|
user = await self.client.fetch_user(int(user))
|
|
|
|
if user is None:
|
|
return
|
|
|
|
embed = discord.Embed(colour=discord.Colour.blue())
|
|
|
|
embed.set_author(name=user.display_name, icon_url=user.avatar_url)
|
|
embed.add_field(name="Discriminator", value="#{}".format(user.discriminator))
|
|
embed.add_field(name="Discord id", value=user.id)
|
|
embed.add_field(name="Bot", value="Nee" if not user.bot else "Ja")
|
|
|
|
created_local = timeFormatters.epochToDate(user.created_at.timestamp())
|
|
|
|
embed.add_field(name="Account aangemaakt", value="{}\n({} geleden)".format(
|
|
created_local["date"], timeFormatters.diffYearBasisString(round(created_local["dateDT"].timestamp()))
|
|
), inline=False)
|
|
|
|
# Check if the user is in the current guild
|
|
if ctx.guild is not None:
|
|
member_instance = ctx.guild.get_member(user.id)
|
|
|
|
if member_instance is not None:
|
|
joined_local = timeFormatters.epochToDate(member_instance.joined_at.timestamp())
|
|
|
|
embed.add_field(name="Lid geworden van {} op".format(ctx.guild.name), value="{}\n({} Geleden)".format(
|
|
joined_local["date"], timeFormatters.diffYearBasisString(round(joined_local["dateDT"].timestamp()))
|
|
))
|
|
|
|
embed.add_field(name="Mention String", value=member_instance.mention, inline=False)
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
# Send a DM to a user -- Can't re-use Utils cog in (un)load because the cog might not be loaded
|
|
async def sendDm(self, userid, message: str):
|
|
user = self.client.get_user(int(userid))
|
|
await user.send(message)
|
|
|
|
|
|
def setup(client):
|
|
client.add_cog(ModCommands(client))
|