Fully customise creation/close embeds

Author: kyb3r

core/thread.py

@@ -123,27 +123,41 @@ async def setup(self, *, creator=None, category=None):
             channel.send(mention, embed=info_embed)
         )
 
+        self.bot.loop.create_task(msg.pin()) # pin message
         self.ready = True
 
         # Once thread is ready, tell the recipient.
         thread_creation_response = self.bot.config.get(
             'thread_creation_response',
-            'The moderation team will get back to you as soon as possible!'
+            'The staff team will get back to you as soon as possible.'
         )
 
         embed = discord.Embed(
             color=self.bot.mod_color,
             description=thread_creation_response,
-            timestamp=datetime.utcnow(),
+            timestamp=channel.created_at,
         )
-        embed.set_footer(text='Your message has been sent',
+
+        footer = 'Your message has been sent'
+        if not self.bot.config.get('disable_recipient_thread_close'):
+            footer = 'Click the lock to close the thread'
+
+        footer = self.bot.config.get('thread_creation_footer', footer)
+            
+
+        embed.set_footer(text=footer,
                          icon_url=self.bot.guild.icon_url)
-        embed.set_author(name='Thread Created')
 
-        if creator is None:
-            self.bot.loop.create_task(recipient.send(embed=embed))
+        embed.title = self.bot.config.get('thread_creation_title', 'Thread Created')
 
-        await msg.pin()
+
+        if creator is None:
+            msg = await recipient.send(embed=embed)
+            if not self.bot.config.get('disable_recipient_thread_close'):
+                close_emoji = self.bot.config.get('close_emoji', '🔒')
+                close_emoji = await self.bot.convert_emoji(close_emoji)
+                await msg.add_reaction(close_emoji) 
+        
 
     def _close_after(self, closer, silent, delete_channel, message):
         return self.bot.loop.create_task(
@@ -227,12 +241,17 @@ async def _close(self, closer, silent=False, delete_channel=True,
             user = f"{self.recipient} (`{self.id}`)"
         else:
             user = f'`{self.id}`'
+        
+        if self.id == closer.id:
+            _closer = 'the Recipient'
+        else:
+            _closer = f'{closer} ({closer.id})'
 
         embed.title = user
 
         event = 'Thread Closed as Scheduled' if scheduled else 'Thread Closed'
         # embed.set_author(name=f'Event: {event}', url=log_url)
-        embed.set_footer(text=f'{event} by {closer} ({closer.id})')
+        embed.set_footer(text=f'{event} by {_closer}')
         embed.timestamp = datetime.utcnow()
 
         tasks = [
@@ -244,14 +263,18 @@ async def _close(self, closer, silent=False, delete_channel=True,
 
         # Thread closed message
 
-        embed = discord.Embed(title='Thread Closed',
+        embed = discord.Embed(title=self.bot.config.get('thread_close_title', 'Thread Closed'),
                               color=discord.Color.red(),
                               timestamp=datetime.utcnow())
 
         if not message:
-            message = f'{closer.mention} has closed this Modmail thread.'
+            if self.id == closer.id:
+                message = 'You have closed this modmail thread.'
+            else:
+                message = f'{closer.mention} has closed this Modmail thread.'
         embed.description = message
-        embed.set_footer(text='Replying will create a new thread',
+        footer = self.bot.config.get('thread_close_footer', 'Replying will create a new thread')
+        embed.set_footer(text=footer,
                          icon_url=self.bot.guild.icon_url)
 
         if not silent and self.recipient is not None: