From c1a359e856f93e454e039a4fe353ebf4b4811fff Mon Sep 17 00:00:00 2001
From: SeaswimmerTheFsh <seaswimmerthefsh@gmail.com>
Date: Mon, 18 Dec 2023 16:17:29 -0500
Subject: [PATCH] feat(moderation): added importer for moving db data between
 bots

---
 moderation/importers/moderation.py | 95 ++++++++++++++++++++++++++++++
 moderation/moderation.py           | 11 ++++
 2 files changed, 106 insertions(+)
 create mode 100644 moderation/importers/moderation.py

diff --git a/moderation/importers/moderation.py b/moderation/importers/moderation.py
new file mode 100644
index 0000000..35d2c2e
--- /dev/null
+++ b/moderation/importers/moderation.py
@@ -0,0 +1,95 @@
+from datetime import timedelta
+import json
+from redbot.core import commands
+from discord import Message, ButtonStyle, Interaction, ui
+from ..database import connect, create_guild_table, mysql_log
+
+class ImportModerationView(ui.View):
+    def __init__(self, timeout, ctx, message):
+        super().__init__()
+        self.ctx: commands.Context = ctx
+        self.message: Message = message
+
+    @ui.button(label="Yes", style=ButtonStyle.success)
+    async def import_button_y(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument
+        await self.message.delete()
+        await interaction.response.send_message("Deleting original table...", ephemeral=True)
+
+        database = await connect()
+        cursor = database.cursor()
+
+        query = f"DROP TABLE IF EXISTS moderation_{self.ctx.guild.id};"
+        cursor.execute(query)
+
+        cursor.close()
+        database.commit()
+
+        await interaction.edit_original_response(content="Creating new table...")
+
+        await create_guild_table(self.ctx.guild)
+
+        await interaction.edit_original_response(content="Importing moderations...")
+
+        file = await self.ctx.message.attachments[0].read()
+        data = sorted(json.loads(file), key=lambda x: x['case'])
+
+        user_mod_types = [
+            'NOTE',
+            'WARN',
+            'MUTE',
+            'UNMUTE',
+            'KICK',
+            'BAN',
+            'UNBAN'
+        ]
+
+        channel_mod_types = [
+            'SLOWMODE',
+            'LOCKDOWN'
+        ]
+
+        failed_cases = []
+
+        for case in data:
+            if 'target_type' not in case or not case['target_type']:
+                if case['moderation_type'] in user_mod_types:
+                    case['target_type'] = 'USER'
+                elif case['moderation_type'] in channel_mod_types:
+                    case['target_type'] = 'CHANNEL'
+
+            if 'role_id' not in case or not case['role_id']:
+                case['role_id'] = 0
+
+            if 'changes' not in case or not case['changes']:
+                case['changes'] = []
+
+            if 'metadata' not in case or not case['metadata']:
+                case['metadata'] = {}
+
+            await mysql_log(
+                self.ctx.guild.id,
+                case['moderator_id'],
+                case['moderation_type'],
+                case['target_type'],
+                case['target_id'],
+                case['role_id'],
+                case['duration'],
+                case['reason'],
+                timestamp=case['timestamp'],
+                resolved=case['resolved'],
+                resolved_by=case['resolved_by'],
+                resolved_reason=case['resolved_reason'],
+                changes=case['changes'],
+                metadata=case['metadata'],
+                database=database
+            )
+
+        await interaction.edit_original_response(content="Import complete.")
+        if failed_cases:
+            await interaction.edit_original_response(content=f"Import complete.\n*Failed to import the following cases:*\n```{failed_cases}```")
+
+    @ui.button(label="No", style=ButtonStyle.danger)
+    async def import_button_n(self, interaction: Interaction, button: ui.Button): # pylint: disable=unused-argument
+        await self.message.edit("Import cancelled.", view=None)
+        await self.message.delete(10)
+        await self.ctx.message.delete(10)
diff --git a/moderation/moderation.py b/moderation/moderation.py
index a3620d8..7d97f4e 100644
--- a/moderation/moderation.py
+++ b/moderation/moderation.py
@@ -25,6 +25,7 @@ from .embed_factory import embed_factory
 from .logger import logger
 from .utils import check_conf, check_permissions, check_moddable, fetch_channel_dict, fetch_user_dict, generate_dict, log
 from .importers.galacticbot import ImportGalacticBotView
+from .importers.moderation import ImportModerationView
 
 class Moderation(commands.Cog):
     """Custom moderation cog.
@@ -1339,6 +1340,16 @@ class Moderation(commands.Cog):
     async def moderationset_import(self, ctx: commands.Context):
         """Import moderations from other bots."""
 
+    moderationset_import.command(name="moderation")
+    @checks.admin()
+    async def moderationset_import_moderation(self, ctx: commands.Context):
+        """Import moderations from another bot using this cog."""
+        if ctx.message.attachments and ctx.message.attachments[0].content_type == 'application/json; charset=utf-8':
+            message = await ctx.send("Are you sure you want to import moderations from another bot?\n**This will overwrite any moderations that already exist in this guild's moderation table.**\n*The import process will block the rest of your bot until it is complete.*")
+            await message.edit(view=ImportModerationView(60, ctx, message))
+        else:
+            await ctx.send("Please provide a valid moderation export file.")
+
     @moderationset_import.command(name="galacticbot")
     @checks.admin()
     async def moderationset_import_galacticbot(self, ctx: commands.Context):