Enso-Bot/cogs/guild.py

# Ensō~Chan - A Multi Purpose Discord Bot That Has Everything Your Server Needs!
# Copyright (C) 2020  Hamothy

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.

import asyncio
import datetime
import io
import random

import asyncpg
import discord
from discord import Embed, TextChannel
from discord import File
from discord.ext.commands import has_permissions, Cog, group, bot_has_permissions, BadArgument, MissingRequiredArgument, \
    command


# Method to ask the user if they want to be anonymous or not
def anon_or_not(self, author):
    # Set up embed to let the user how to start sending modmail
    AnonModMailEmbed = Embed(title="**Want to send it Anonymously?**",
                             colour=self.bot.admin_colour,
                             timestamp=datetime.datetime.utcnow())

    AnonModMailEmbed.set_thumbnail(url=author.avatar_url)
    AnonModMailEmbed.set_footer(text=f"Sent by {author}")

    fields = [(self.bot.blank_space, "**We understand that for some things,"
                                     "you may want to remain Anonymous."
                                     "\nUse the reactions below to choose!**", False),
              (self.bot.blank_space, "**Use :white_check_mark: for** `Yes`", True),
              (self.bot.blank_space, "**Use :x: for** `No`", True),
              (self.bot.blank_space, self.bot.blank_space, True),
              (self.bot.blank_space,
               "The Staff will not know who is sending this"
               "\nPurely negative feedback will not be considered.", True)]

    for name, value, inline in fields:
        AnonModMailEmbed.add_field(name=name, value=value, inline=inline)

    return AnonModMailEmbed


# Method to send an embed to to let the user know to type into chat
def send_instructions(self, author):
    # Set up embed to let the user know that they have aborted the modmail
    SendModMailEmbed = Embed(title="**Please enter a message for it to be sent to the staff!**",
                             colour=self.bot.admin_colour,
                             timestamp=datetime.datetime.utcnow())

    SendModMailEmbed.set_thumbnail(url=author.avatar_url)
    SendModMailEmbed.set_footer(text=f"Sent by {author}")

    fields = [("**Make sure that the message is above **50** and below **1024** characters!**",
               "**Include as much detail as possible :P**",
               False)]

    for name, value, inline in fields:
        SendModMailEmbed.add_field(name=name, value=value, inline=inline)

    return SendModMailEmbed


# Method to let the user know that the message must be above 50 characters
def error_handling(self, author):
    # Set up embed to let the user know that the message must be above 50 characters
    ErrorHandlingEmbed = Embed(
        title="Uh Oh! Please make sure the message is above **50** and below **1024** characters!",
        colour=self.bot.admin_colour,
        timestamp=datetime.datetime.utcnow())

    ErrorHandlingEmbed.set_thumbnail(url=author.avatar_url)
    ErrorHandlingEmbed.set_footer(text=f"Sent by {author}")

    fields = [("Please enter in a message which is above **50** and below **1024** characters!",
               "**This helps us reduce spam and allows you to include more detail in your mail!**",
               False)]

    for name, value, inline in fields:
        ErrorHandlingEmbed.add_field(name=name, value=value, inline=inline)

    return ErrorHandlingEmbed


# Method to send an embed into chat to let the user know that their mail has been sent successfully
def message_sent_confirmation(self, author):
    # Set up embed to let the user know that they have sent the mail
    ConfirmationEmbed = Embed(title="**Message relayed to Staff!!**",
                              colour=self.bot.admin_colour,
                              timestamp=datetime.datetime.utcnow())

    ConfirmationEmbed.set_thumbnail(url=author.avatar_url)
    ConfirmationEmbed.set_footer(text=f"Sent by {author}")

    fields = [("Thank you for your input! The staff team appreciate it very much!",
               f"\n As mentioned previously, please don't be hesistant to DM the Staff for anything! :P",
               False)]

    for name, value, inline in fields:
        ConfirmationEmbed.add_field(name=name, value=value, inline=inline)

    return ConfirmationEmbed


# Method to actually allow the message to be sent to #mod-mail
def send_modmail(self, msg, author):
    embed = Embed(title="Modmail",
                  colour=self.bot.admin_colour,
                  timestamp=datetime.datetime.utcnow())

    if self.anon:

        embed.set_thumbnail(url=random.choice(self.avatars))
        embed.set_footer(text=f"Sent By Anon Member")

        fields = [("Member", "Anon Member", False),
                  ("Message", msg.content, False)]
    else:

        embed.set_thumbnail(url=author.avatar_url)
        embed.set_footer(text=f"Sent By {author}")

        fields = [("Member", author, False),
                  ("Message", msg.content, False)]

    for name, value, inline in fields:
        embed.add_field(name=name, value=value, inline=inline)

    return embed


async def wait_for_msg(self, check, user_channel):
    """
    Method to check if the user actually types in a message
    If not, delete the channel
    """

    try:
        # Wait for the message from the author
        mod_message = await self.bot.wait_for('message', check=check, timeout=300.0)

    # Delete channel if user does not send a message within 5 minutes
    except asyncio.TimeoutError:
        await user_channel.delete()
        return None
    else:
        return mod_message


# Set up the Cog
class Guild(Cog):
    """All Guild Systems (Modmail/Modlogs/RolePersist)"""

    def __init__(self, bot):
        self.bot = bot
        self.anon = None
        self.avatars = ["https://cdn.discordapp.com/embed/avatars/0.png",
                        "https://cdn.discordapp.com/embed/avatars/1.png",
                        "https://cdn.discordapp.com/embed/avatars/2.png",
                        "https://cdn.discordapp.com/embed/avatars/3.png",
                        "https://cdn.discordapp.com/embed/avatars/4.png"]

    @Cog.listener()
    async def on_ready(self):
        """Printing out that Cog is ready on startup"""
        print(f"{self.__class__.__name__} Cog has been loaded\n-----")

    @command(name="modstatus", aliases=["logsstatus"])
    @bot_has_permissions(embed_links=True)
    async def all_statuses(self, ctx):
        """Status of all the moderation systems (Modlogs/Modmail/RolePersist)"""
        desc = ""

        # Get status of mod
        if self.bot.get_roles_persist(ctx.guild.id) == 0:
            desc += f"**{self.bot.cross} Role Persist**\n"
        else:
            desc += f"**{self.bot.tick} Role Persist**\n"

        # Get status of modlogs
        ml_channel = self.bot.get_modlog_for_guild(ctx.guild.id)

        if ml_channel:
            channel = ctx.guild.get_channel(ml_channel)
            desc += f"**{self.bot.tick} Modlogs | {channel.mention}**\n"
        else:
            desc += f"**{self.bot.cross} Modlogs**\n"

        # Get status of modmail
        modmail = self.bot.get_modmail(ctx.guild.id)
        if modmail:
            modmail_channel = ctx.guild.get_channel(modmail["modmail_channel_id"])
            modmail_logging = ctx.guild.get_channel(modmail["modmail_logging_channel_id"])

            desc += f"**{self.bot.tick} Modmail | Channel:  {modmail_channel.mention} | Logging: {modmail_logging.mention}**\n"
        else:
            desc += f"**{self.bot.cross} Modmail**\n"

        embed = Embed(title="Moderation Systems",
                      description=desc,
                      colour=self.bot.random_colour(),
                      timestamp=datetime.datetime.utcnow())
        embed.set_footer(text=f"Requested by {ctx.author}", icon_url=ctx.author.avatar_url)

        await ctx.send(embed=embed)

    @group(name="rolepersist", case_insensitive=True, usage="`<enable|disable>`")
    @has_permissions(manage_guild=True)
    @bot_has_permissions(manage_roles=True)
    async def roles_persist(self, ctx):
        """Role Persist! Keep user roles when they leave/join!"""

    @roles_persist.command(name="status")
    async def rp_status(self, ctx):
        """Showing the status of the role persist within the guild"""

        if self.bot.get_roles_persist(ctx.guild.id) == 0:
            await self.bot.generate_embed(ctx, desc=f"**Role Persist is currently disabled within {ctx.guild}**")
        else:
            await self.bot.generate_embed(ctx, desc=f"**Role Persist is currently enabled within {ctx.guild}**")

    @roles_persist.command(name="enable")
    async def rp_enable(self, ctx):
        """Enabling role persist within the guild"""

        if self.bot.get_roles_persist(ctx.guild.id) == 0:
            await self.bot.update_role_persist(ctx.guild.id, value=1)
            await self.bot.generate_embed(ctx, desc=f"**Role Persist has been enabled within {ctx.guild}!**")
        else:
            await self.bot.generate_embed(ctx, desc=f"**Role Persist is already enabled within {ctx.guild}!**")

    @roles_persist.command(name="disable")
    async def rp_disable(self, ctx):
        """Disabling role persist within the guild"""

        if self.bot.get_roles_persist(ctx.guild.id) == 1:
            await self.bot.update_role_persist(ctx.guild.id, value=0)
            await self.bot.generate_embed(ctx, desc=f"**Role Persist has been disabled within {ctx.guild}!**")
        else:
            await self.bot.generate_embed(ctx, desc=f"**Role Persist is already disabled within {ctx.guild}!**")

    @group(name="modlogs", case_insensitive=True, usage="`<setup|update|delete>`")
    @has_permissions(manage_guild=True)
    @bot_has_permissions(embed_links=True)
    async def modlogs(self, ctx):
        """
        Log updates in your server! (Nicknames/Deleted Msgs/etc!)
        """

    @modlogs.command(name="setup")
    async def mlsetup(self, ctx, user_channel: TextChannel):
        """Setup a channel for Kick/Ban/Mute actions to be logged"""

        # Setup pool connection
        pool = self.bot.db
        async with pool.acquire() as conn:

            # Get the row of the guild from database
            try:
                select_query = """SELECT * FROM guilds WHERE guild_id = $1"""
                result = await conn.fetchrow(select_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print("PostGres Error: Guild Record Could Not Be Retrieved For Modlog Setup", e)

            # Throw error if the modlog channel already exists
            else:
                if result["modlogs"]:
                    text = "**Modlogs Channel** already set up!" \
                           f"\nDo **{ctx.prefix}help modlogs** to find out more!"
                    await self.bot.generate_embed(ctx, desc=text)

                # Set up the modlogs channel within the guild
                else:
                    mod_log_setup = True
                    await self.bot.storage_modlog_for_guild(ctx, user_channel.id, mod_log_setup)

            # Release the connection back to the pool
            finally:
                await pool.release(conn)

    @modlogs.command(name="update")
    async def mlupdate(self, ctx, user_channel: TextChannel):
        """Change the channel that your modlogs are sent to"""

        # Setup up pool connection
        pool = self.bot.db
        async with pool.acquire() as conn:

            # Get the guilds row from the guilds table
            try:
                select_query = """SELECT * FROM guilds WHERE guild_id = $1"""
                result = await conn.fetchrow(select_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print("PostGres Error: Guild Record Could Not Be Retrieved For Modlog Update", e)

            # Throw error if the modlog channel already exists
            else:
                if not result["modlogs"]:
                    text = "**Modlogs Channel** not set up!" \
                           f"\nDo **{ctx.prefix}help modlogs** to find out more!"
                    await self.bot.generate_embed(ctx, desc=text)

                # Update the modlog channel within the database and cache
                else:
                    mod_log_setup = False
                    await self.bot.storage_modlog_for_guild(ctx, user_channel.id, mod_log_setup)

            # Release the connection back to the pool
            finally:
                await pool.release(conn)

    @modlogs.command("delete")
    async def mldelete(self, ctx):
        """Delete the existing modlogs channel"""

        # Setup up pool connection
        pool = self.bot.db
        async with pool.acquire() as conn:

            # Get the guilds row from the guilds table
            try:
                select_query = """SELECT * FROM guilds WHERE guild_id = $1"""
                result = await conn.fetchrow(select_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print("PostGres Error: Guild Record Could Not Be Retrieved For Modlog Delete", e)

            # Throw error that modlogs have not been setup
            else:
                if not result["modlogs"]:
                    text = "**Modlogs Channel** not set up!" \
                           f"\nDo **{ctx.prefix}help modlogs** to find out more!"
                    await self.bot.generate_embed(ctx, desc=text)
                    return

            # Update the existing modlogs for guild
            try:
                update = """UPDATE guilds SET modlogs = NULL WHERE guild_id = $1"""
                await conn.execute(update, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print(f"PostGres Error: Guild Modlogs Could Not Be Deleted For {ctx.guild.id}", e)

            # Delete channel from cache
            else:
                self.bot.remove_modlog_channel(ctx.guild.id)

        # Release the connection back to the pool
        await pool.release(conn)

        text = "**Modlogs System** successfully deleted!" \
               f"\nDo **{ctx.prefix}help modlogs** to setup Modlogs again!"
        await self.bot.generate_embed(ctx, desc=text)

    @group(name="modmail", case_insensitive=True, usage="`<setup|update|delete>`")
    @bot_has_permissions(manage_channels=True, embed_links=True, add_reactions=True, manage_messages=True,
                         attach_files=True, read_message_history=True, manage_roles=True)
    @has_permissions(manage_guild=True)
    async def mod_mail(self, ctx):
        """
        Modmail! Allow your members to send mail to the staff team!
        """

    @mod_mail.command(name="setup")
    async def mmsetup(self, ctx, modmail: TextChannel, modmail_logging: TextChannel):
        """
        Setup Modmail System
        First Argument: Input Channel(Mention or ID) where members can send modmail
        Second Argument: Input Channel(Mention or ID) where the members mail should be sent
        """

        # Setup up pool connection
        pool = self.bot.db
        async with pool.acquire() as conn:

            # Get the author's row from the members table
            try:
                select_query = """SELECT * FROM moderatormail WHERE guild_id = $1"""
                result = await conn.fetchrow(select_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print("PostGres Error: ModeratorMail Record Could Not Be Retrieved For Modmail Setup", e)

            # Throw error if the guild already exists
            else:
                if result:
                    text = "**Modmail System** already set up!" \
                           f"\nDo **{ctx.prefix}help modmail** to find out more!"
                    await self.bot.generate_embed(ctx, desc=text)
                    return

            # Release the connection back to the pool
            finally:
                await pool.release(conn)

        # Set up embed to let the user how to start sending modmail
        desc = "React to this message if you want to send a message to the Staff Team!" \
               "\n\n**React with ✅**" \
               "\n\nWe encourage all suggestions/thoughts and opinions on the server!" \
               "\nAs long as it is **valid** criticism." \
               "\n\n\n**Purely negative feedback will not be considered.**"

        ModMail = Embed(title="**Welcome to Modmail!**",
                        description=desc,
                        colour=self.bot.admin_colour,
                        timestamp=datetime.datetime.utcnow())
        ModMail.set_thumbnail(url=self.bot.user.avatar_url)

        # Send modmail embed to the specified channel and auto add the ✅ reaction
        modmail_message = await modmail.send(embed=ModMail)
        try:
            await modmail_message.add_reaction('✅')
        except Exception as e:
            print(e)

        # Setup up pool connection
        async with pool.acquire() as conn:

            # Insert the information about the modmail system into database
            try:
                insert_query = """INSERT INTO moderatormail (guild_id, modmail_channel_id, message_id, modmail_logging_channel_id) 
                                  VALUES ($1, $2, $3, $4)"""
                await conn.execute(insert_query, ctx.guild.id, modmail.id, modmail_message.id, modmail_logging.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print(f"PostGres Error: Modmail System Record Could Not Be Inserted For Guild {ctx.guild.id}", e)

            # Send confirmation message
            else:
                text = "**Modmail System** is successfully set up!" \
                       f"\nRefer to **{ctx.prefix}help modmail** for more information"
                await self.bot.generate_embed(ctx, desc=text)

                # Store into cache
                self.bot.cache_store_modmail(ctx.guild.id, modmail.id, modmail_message.id, modmail_logging.id)

            # Release connection back into pool
            finally:
                await pool.release(conn)

    @mod_mail.command(name="update")
    async def mmupdate(self, ctx, modmail_logging_channel: TextChannel):
        """
        Update the Channel that the Modmail is logged to
        You can Mention or use the Channel ID
        """

        # Setup up pool connection
        pool = self.bot.db
        async with pool.acquire() as conn:

            # Get the moderatormail record from the guilds table
            try:
                select_query = """SELECT * FROM moderatormail WHERE guild_id = $1"""
                result = await conn.fetchrow(select_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print("PostGres Error: ModeratorMail Record Could Not Be Retrieved For Modmail Update", e)

            # Throw error if the guild already exists
            else:
                if not result:
                    text = "**Modmail System** not set up!" \
                           f"\nDo **{ctx.prefix}help modmail** to find out more!"
                    await self.bot.generate_embed(ctx, desc=text)
                    return

            # Release connection back to pool
            finally:
                await pool.release(conn)

        # Setup up pool connection and cursor
        async with pool.acquire() as conn:

            # Update the modmail channel in the database
            try:
                update_query = """UPDATE moderatormail SET modmail_logging_channel_id = $1 WHERE guild_id = $2"""
                await conn.execute(update_query, modmail_logging_channel.id, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print(f"PostGres Error: Modmail System Record Could Not Be Updated For Guild {ctx.guild.id}", e)

            # Send confirmation that the channel has been updated
            else:
                text = "**Channel Updated**" \
                       f"\nNew Modmail will be sent to {modmail_logging_channel.mention}"
                await self.bot.generate_embed(ctx, desc=text)

                # Update cache
                self.bot.update_modmail(ctx.guild.id, modmail_logging_channel.id)

            # Release connection back to pool
            finally:
                await pool.release(conn)

    @mod_mail.command(name="delete")
    async def mmdelete(self, ctx):
        """Delete the entire modmail system from the guild"""

        # Setup up pool connection
        pool = self.bot.db
        async with pool.acquire() as conn:

            # Get the moderatormail record from the guilds table
            try:
                select_query = """SELECT * FROM moderatormail WHERE guild_id = $1"""
                result = await conn.fetchrow(select_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print("PostGres Error: ModeratorMail Record Could Not Be Retrieved For Modmail Delete", e)

            else:
                # Throw error if modmail system does not exist already
                if not result:
                    text = "**Modmail System** not set up!" \
                           f"\nDo **{ctx.prefix}help modmail** to find out more!"
                    await self.bot.generate_embed(ctx, desc=text)
                    return

            # Release connection back to pool
            finally:
                await pool.release(conn)

        # Setup up pool connection
        async with pool.acquire() as conn:

            # Remove the moderatormail record from the database
            try:
                delete_query = """DELETE FROM moderatormail WHERE guild_id = $1"""
                await conn.execute(delete_query, ctx.guild.id)

            # Catch errors
            except asyncpg.PostgresError as e:
                print(f"PostGres Error: ModeratorMail Record Could Not Be Deleted for Guild {ctx.guild.id}", e)

            # Sending confirmation message that the modmail system has been deleted
            else:
                text = "**Modmail System** successfully deleted!" \
                       f"\nDo **{ctx.prefix}help modmail** to find out more!"
                await self.bot.generate_embed(ctx, desc=text)

                # Delete from cache
                self.bot.delete_modmail(ctx.guild.id)

            # Release connection back to pool
            finally:
                await pool.release(conn)

    @Cog.listener()
    async def on_raw_reaction_add(self, payload):
        """Listen for reactions for modmail channel"""

        # Don't count reactions that are made by the bot
        # Don't count other reactions other than ✅ and ❌
        if payload.member.bot or str(payload.emoji) not in ['✅', '❌']:
            return

        # Get the modmail information from cache
        modmail = self.bot.get_modmail(payload.guild_id)
        if modmail:
            channel_id = modmail["modmail_channel_id"]
            message_id = modmail["message_id"]
            modmail_channel_id = modmail["modmail_logging_channel_id"]
        else:
            return

        # Bunch of checks to make sure it has the right guild, channel, message and reaction
        if payload.channel_id == channel_id and payload.message_id == message_id and payload.emoji.name == "✅":

            # Get the guild
            guild = self.bot.get_guild(payload.guild_id)
            # Get the member
            member = guild.get_member(payload.user_id)
            # Get the setup modmail channel
            channel = guild.get_channel(payload.channel_id)
            # Get the modmail logging channel
            modmail_channel = guild.get_channel(modmail_channel_id)

            # Fetch the message and remove the reaction
            reaction = await channel.fetch_message(message_id)
            await reaction.remove_reaction('✅', member)

            # Setting up the channel permissions for the new channel that will be created
            overwrites = {
                guild.default_role: discord.PermissionOverwrite(read_messages=False, send_messages=False),
                guild.me: discord.PermissionOverwrite(read_messages=True, send_messages=True, embed_links=True,
                                                      add_reactions=True, manage_messages=True),
                member: discord.PermissionOverwrite(read_messages=True, send_messages=True)
            }

            # Saving this for later within when discord.py 1.4 comes out
            # user_channel = await guild.create_category_channel("Member", overwrites=overwrites, position=7)

            # Create the text channel
            user_channel = await guild.create_text_channel("Member", overwrites=overwrites,
                                                           position=0)

            # Mention the user to make sure that they get pinged
            mention = await user_channel.send(member.mention)
            await mention.delete()

            try:

                # Send the embed if they want to remain anonymous or not
                Anon_or_Not = await user_channel.send(embed=anon_or_not(self, member))
                # Add reactions to the message
                await Anon_or_Not.add_reaction('✅')
                await Anon_or_Not.add_reaction('❌')

                # Checking if the user reacted with ✅ with response to sending staff a message
                def emoji_check(reaction, user):
                    return user == member and str(reaction.emoji) in ['✅', '❌']

                try:
                    # Wait for the user to add a reaction
                    reaction, user = await self.bot.wait_for('reaction_add', check=emoji_check, timeout=60.0)

                # Delete channel if user does not react within 60 seconds
                except asyncio.TimeoutError as ex:
                    print(ex)
                    await user_channel.delete()
                else:

                    # Making sure that the reply is from the author
                    def check(m):
                        return m.author == payload.member and user_channel.id == instructions.channel.id

                    # Checking if user wants to be Anonymous or not
                    if str(reaction.emoji) == "✅":
                        self.anon = True

                    if str(reaction.emoji) == "❌":
                        self.anon = False

                    # Delete the old embed
                    await Anon_or_Not.delete()

                    # Tell the user to type their mail into the chat
                    instructions = await user_channel.send(embed=send_instructions(self, member))

                    # Wait for the message from the author
                    msg = await wait_for_msg(self, check, user_channel)
                    if not msg: return

                    # Making sure that the message is below 50 characters and the message was sent in the channel
                    while len(msg.content) <= 50 and msg.channel == user_channel:
                        await user_channel.send(embed=error_handling(self, member))

                        # Wait for the message from the author
                        msg = await wait_for_msg(self, check, user_channel)
                        if not msg: return

                    # As long as the message is above 50 characters and in the correct channel
                    if len(msg.content) > 50 and msg.channel == user_channel:
                        # Delete the previous embed
                        await instructions.delete()

                        # Store all text in the channel in a bytesio object
                        text = ""
                        async for message in user_channel.history(limit=300):
                            text += "".join(f"{message.created_at} : {message.content}\n")
                        text_bytes = str.encode(text)

                        file = io.BytesIO(text_bytes)
                        file_name = "Anon.txt" if self.anon else f"{member.name}.txt"

                        # Send the message to the modmail channel
                        await modmail_channel.send(embed=send_modmail(self, msg, member),
                                                   file=File(file, file_name))

                        # Make sure the user knows that their message has been sent
                        await user_channel.send(embed=message_sent_confirmation(self, member))

                        # Let the user read the message for 5 seconds
                        await asyncio.sleep(5)

                        # Delete the channel and then stop the function
                        await user_channel.delete()

                    # If the user types anywhere else, delete the channel
                    else:
                        await user_channel.delete()

            except Exception as ex:
                print(ex)

                # Send out an error message if the user waited too long
                await user_channel.send(
                    "Sorry! Something seems to have gone wrong and the modmail will be aborting."
                    "\nRemember to make sure it's under **1024** characters!!")

                await asyncio.sleep(5)
                await user_channel.delete()

    @mlsetup.error
    @mlupdate.error
    @mmsetup.error
    @mmupdate.error
    async def mlsetup_command_error(self, ctx, exc):
        """Catching error if channel is not recognised"""

        if isinstance(exc, BadArgument):
            text = "**Channel Not Detected... Aborting Process**"
            await self.bot.generate_embed(ctx, desc=text)
        elif isinstance(exc, MissingRequiredArgument):
            text = "Required Argument(s) Missing!" \
                   f"\nUse **{ctx.prefix}help** to find how to use **{ctx.command}**"
            await self.bot.generate_embed(ctx, desc=text)


def setup(bot):
    bot.add_cog(Guild(bot))