Added function calling and lots of other stuff

src/cogs/manage_chat.py

@@ -0,0 +1,108 @@
+import discord
+import re
+import os
+from src.config import debug, curs_data
+
+
+class ManageChat(discord.Cog):
+    def __init__(self, bot: discord.Bot) -> None:
+        super().__init__()
+        self.bot = bot
+
+    @discord.slash_command(
+        name="cancel", description="Cancel the last message sent into a channel"
+    )
+    async def cancel(self, ctx: discord.ApplicationContext):
+        debug(
+            f"The user {ctx.author} ran the cancel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
+        )
+        # check if the guild is in the database
+        curs_data.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
+        if curs_data.fetchone() is None:
+            await ctx.respond(
+                "This server is not setup, please run /setup", ephemeral=True
+            )
+            return
+        # get the last message sent by the bot in the cha where the command was sent
+        last_message = await ctx.channel.fetch_message(ctx.channel.last_message_id)
+        # delete the message
+        await last_message.delete()
+        await ctx.respond("The last message has been deleted", ephemeral=True)
+
+    # add a slash command called "clear" that deletes all the messages in the channel
+    @discord.slash_command(
+        name="clear", description="Clear all the messages in the channel"
+    )
+    async def clear(self, ctx: discord.ApplicationContext):
+        debug(
+            f"The user {ctx.author.name} ran the clear command command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
+        )
+        await ctx.respond("messages deleted!", ephemeral=True)
+        return await ctx.channel.purge()
+
+    @discord.slash_command(
+        name="transcript",
+        description="Get a transcript of the messages that have been sent in this channel intoa text file",
+    )
+    @discord.option(
+        name="channel_send",
+        description="The channel to send the transcript to",
+        required=False,
+    )
+    async def transcript(
+        self, ctx: discord.ApplicationContext, channel_send: discord.TextChannel = None
+    ):
+        debug(
+            f"The user {ctx.author.name} ran the transcript command command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
+        )
+        # save all the messages in the channel in a txt file and send it
+        messages = await ctx.channel.history(limit=None).flatten()
+        messages.reverse()
+        transcript = ""
+        # defer the response
+        await ctx.defer()  # defer the response so that the bot doesn't say that it's thinking
+        for msg in messages:
+            if msg.author.bot:
+                transcript += f"Botator: {msg.content}\n"
+            else:
+                mentions = re.findall(r"<@!?\d+>", msg.content)
+                # then replace each mention with the name of the user
+                for mention in mentions:
+                    # get the user id
+                    id = mention[2:-1]
+                    # get the user
+                    user = await self.bot.fetch_user(id)
+                    # replace the mention with the name
+                    msg.content = msg.content.replace(
+                        mention, msg.guild.get_member(user.id).name
+                    )
+                transcript += f"{msg.author.name}: {msg.content}\n"
+        # save the transcript in a txt file called transcript.txt. If the file already exists, delete it and create a new one
+        # check if the file exists
+        if os.path.exists("transcript.txt"):
+            os.remove("transcript.txt")
+        f = open("transcript.txt", "w")
+        f.write(transcript)
+        f.close()
+        last_message: discord.Message = await ctx.channel.fetch_message(
+            ctx.channel.last_message_id
+        )
+        new_file_name = f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt"
+        # rename the file with the name of the channel and the date in this format: transcript_servername_channelname_dd-month-yyyy.txt ex : transcript_Botator_Testing_12-may-2021.txt
+        os.rename(
+            "transcript.txt",
+            new_file_name,
+        )
+        # send the file in a private message to the user who ran the command
+        # TODO: rework so as to give the choice of a private send or a public send
+        if channel_send is None:
+            await ctx.respond(
+                file=discord.File(new_file_name),
+                ephemeral=True,
+            )
+        else:
+            await channel_send.send(file=discord.File(new_file_name))
+            await ctx.respond("Transcript sent!", ephemeral=True, delete_after=5)
+        await ctx.author.send(file=discord.File(new_file_name))
+        # delete the file
+        os.remove(new_file_name)