roxbot/Roxbot/cogs/customcommands.py

import discord
from Roxbot import checks, load_config
from discord.ext.commands import group
from Roxbot.settings import guild_settings


def blacklisted(user):
	with open("Roxbot/blacklist.txt", "r") as fp:
		for line in fp.readlines():
			if user.id+"\n" == line:
				return True
	return False


class CustomCommands():
	def __init__(self, bot_client):
		self.bot = bot_client

	async def on_message(self, message):
		settings = guild_settings.get(message.guild)
		msg = message.content.lower()
		channel = message.channel

		if blacklisted(message.author) or type(message.channel) != discord.TextChannel:
			return
		if message.author == self.bot.user:
			return

		if msg.startswith(self.bot.command_prefix):
			if msg.split(self.bot.command_prefix)[1] in settings.custom_commands["1"]:
				return await channel.send(settings.custom_commands["1"][msg.split(self.bot.command_prefix)[1]])
		else:
			for command in settings.custom_commands["0"]:
				if msg == command:
					return await channel.send(settings.custom_commands["0"][command])

	@group(pass_context=True, aliases=["cc"])
	@checks.is_owner_or_admin()
	async def custom(self, ctx):
		"A group of commands to manage custom commands for your server."
		if ctx.invoked_subcommand is None:
			return await ctx.send('Missing Argument')

	@custom.command(pass_context=True)
	async def add(self, ctx, command, output, prefix_required = "0"):
		"Adds a custom command to the list of custom commands."
		settings = guild_settings.get(ctx.guild)
		command = command.lower()
		output = output
		zero = settings.custom_commands["0"]
		one = settings.custom_commands["1"]

		if ctx.message.mentions or ctx.message.mention_everyone or ctx.message.role_mentions:
			return await ctx.send("Custom Commands cannot mention people/roles/everyone.")
		elif len(output) > 1800:
			return await ctx.send("The output is too long")
		elif command in self.bot.commands and prefix_required == "1":
			return await ctx.send("This is already the name of a built in command.")
		elif command in zero or command in one:
			return await ctx.send("Custom Command already exists.")
		elif prefix_required != "1" and prefix_required != "0":
			return await ctx.send("No prefix setting set.")
		elif len(command.split(" ")) > 1 and prefix_required == "1":
			return await ctx.send("Custom commands with a prefix can only be one word with no spaces.")

		settings.custom_commands[prefix_required][command] = output
		settings.update(settings.custom_commands, "custom_commands")
		return await ctx.send("{} has been added with the output: '{}'".format(command, output))

	@custom.command(pass_context=True)
	async def edit(self, ctx, command, edit):
		"Edits an existing custom command."
		settings = guild_settings.get(ctx.guild)
		zero = settings.custom_commands["0"]
		one = settings.custom_commands["1"]

		if ctx.message.mentions or ctx.message.mention_everyone or ctx.message.role_mentions:
			return await ctx.send("Custom Commands cannot mention people/roles/everyone.")

		if command in zero:
			settings.custom_commands["0"][command] = edit
			settings.update(settings.custom_commands, "custom_commands")
			return await ctx.send("Edit made. {} now outputs {}".format(command, edit))
		elif command in one:
			settings.custom_commands["1"][command] = edit
			settings.update(settings.custom_commands, "custom_commands")
			return await ctx.send("Edit made. {} now outputs {}".format(command, edit))
		else:
			return await ctx.send("That Custom Command doesn't exist.")

	@custom.command(pass_context=True)
	async def remove(self, ctx, command):
		"Removes a custom command."
		settings = guild_settings.get(ctx.guild)
		command = command.lower()
		if command in settings.custom_commands["1"]:
			settings.custom_commands["1"].pop(command)
			settings.update(settings.custom_commands, "custom_commands")
			return await ctx.send("Removed {} custom command".format(command))
		elif command in settings.custom_commands["0"]:
			settings.custom_commands["0"].pop(command)
			settings.update(settings.custom_commands, "custom_commands")
			return await ctx.send("Removed {} custom command".format(command))
		else:
			return await ctx.send("Custom Command doesn't exist.")


	@custom.command(pass_context=True)
	async def list(self, ctx, debug="0"):
		"Lists all custom commands for this server."
		if debug != "0" and debug != "1":
			debug = "0"
		settings = guild_settings.get(ctx.guild)
		l = settings.custom_commands
		listzero = ""
		listone = ""

		for command in l["0"]:
			if debug == "1":
				command += " - {}".format(l["0"][command])
			listzero = listzero + "- " + command + "\n"
		for command in l["1"]:
			if debug == "1":
				command += " - {}".format(l["1"][command])
			listone = listone + "- " + command + "\n"
		if not listone:
			listone = "There are no commands setup.\n"
		if not listzero:
			listzero = "There are no commands setup.\n"

		# TODO: Sort out a way to shorten this if it goes over 2000 characters.
		
		em = discord.Embed(title="Here is the list of Custom Commands", color=load_config.embedcolour)
		em.add_field(name="Commands that require Prefix:", value=listone, inline=False)
		em.add_field(name="Commands that don't:", value=listzero, inline=False)
		return await ctx.send(embed=em)

def setup(bot_client):
	bot_client.add_cog(CustomCommands(bot_client))