im scared to lose my work :|

Author: RealCyGuy

cogs/modmail.py

@@ -873,7 +873,7 @@ async def pareply(self, ctx, *, msg: str = ""):
         async with ctx.typing():
             await ctx.thread.reply(ctx.message, anonymous=True, plain=True)
 
-    @commands.command()
+    @commands.group(invoke_without_command=True)
     @checks.has_permissions(PermissionLevel.SUPPORTER)
     @checks.thread_only()
     async def note(self, ctx, *, msg: str = ""):
@@ -887,6 +887,19 @@ async def note(self, ctx, *, msg: str = ""):
             msg = await ctx.thread.note(ctx.message)
             await msg.pin()
 
+    @note.command(name="persistent", aliases=["persist"])
+    @checks.has_permissions(PermissionLevel.SUPPORTER)
+    @checks.thread_only()
+    async def note_persistent(self, ctx, *, msg: str = ""):
+        """
+        Take a persistent note about the current user.
+        """
+        ctx.message.content = msg
+        async with ctx.typing():
+            msg = await ctx.thread.note(ctx.message, persistent=True)
+            await msg.pin()
+        await self.bot.api.create_note(recipient=ctx.thread.recipient, message=ctx.message)
+
     @commands.command()
     @checks.has_permissions(PermissionLevel.SUPPORTER)
     @checks.thread_only()

core/clients.py

@@ -401,6 +401,25 @@ async def search_by_text(self, text: str, limit: Optional[int]):
             {"messages": {"$slice": 5}},
         ).to_list(limit)
 
+    async def create_note(self, recipient: Member, message: Message):
+        await self.db.notes.insert_one(
+            {
+                "recipient": str(recipient.id),
+                "author": {
+                    "id": str(message.author.id),
+                    "name": message.author.name,
+                    "discriminator": message.author.discriminator,
+                    "avatar_url": str(message.author.avatar_url),
+                },
+            }
+        )
+
+    async def delete_note(self):
+        pass
+
+    async def find_notes(self, recipient: Member):
+        return await self.db.notes.find({"recipient": str(recipient.id)}).to_list(None)
+
     def get_plugin_partition(self, cog):
         cls_name = cog.__class__.__name__
         return self.db.plugins[cls_name]

core/thread.py

@@ -190,6 +190,13 @@ async def send_recipient_genesis_message():
                     close_emoji = await self.bot.convert_emoji(close_emoji)
                     await self.bot.add_reaction(msg, close_emoji)
 
+        async def send_persistent_notes():
+            notes = await self.bot.api.find_notes()
+            for note in notes:
+                message = discord.Message()
+                await self.note(note.message)
+                pass
+
         await asyncio.gather(send_genesis_message(), send_recipient_genesis_message())
         self.bot.dispatch("thread_ready", self)
 
@@ -629,11 +636,11 @@ async def edit_dm_message(self, message: discord.Message, content: str) -> None:
             self.bot.api.edit_message(message.id, content), linked_message.edit(embed=embed)
         )
 
-    async def note(self, message: discord.Message) -> None:
+    async def note(self, message: discord.Message, persistent=False) -> None:
         if not message.content and not message.attachments:
             raise MissingRequiredArgument(SimpleNamespace(name="msg"))
 
-        msg = await self.send(message, self.channel, note=True)
+        msg = await self.send(message, self.channel, note=True, persistent_note=persistent)
 
         self.bot.loop.create_task(
             self.bot.api.append_log(
@@ -719,6 +726,7 @@ async def send(
         note: bool = False,
         anonymous: bool = False,
         plain: bool = False,
+        persistent_note: bool = False,
     ) -> None:
 
         self.bot.loop.create_task(
@@ -780,7 +788,7 @@ async def send(
         else:
             # Special note messages
             embed.set_author(
-                name=f"Note ({author.name})",
+                name=f"{'Persistent' if persistent_note else ''} Note ({author.name})",
                 icon_url=system_avatar_url,
                 url=f"https://discordapp.com/users/{author.id}#{message.id}",
             )