feat: export guilds from old bot and import into new one

2024-10-24 23:35:44 +02:00 · 2024-10-24 23:35:44 +02:00 · 9302ba200f
commit 9302ba200f
parent 301744dd4e
10 changed files with 1326 additions and 0 deletions
--- a/Catalogger.GoImporter/GuildImport.cs
+++ b/Catalogger.GoImporter/GuildImport.cs
@ -0,0 +1,129 @@
+// Copyright (C) 2021-present sam (starshines.gay)
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero 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 Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+using System.Diagnostics;
+using System.Text.Json;
+using Catalogger.Backend.Database;
+using Catalogger.Backend.Database.Models;
+using NodaTime.Extensions;
+using Serilog;
+
+namespace Catalogger.GoImporter;
+
+public static class GuildImport
+{
+    public static async Task DoImportAsync(DatabaseContext db, string filename)
+    {
+        var rawData = await File.OpenText(filename).ReadToEndAsync();
+        var data = JsonSerializer.Deserialize<GuildsExport>(rawData, Program.JsonOptions);
+        if (data == null)
+        {
+            Log.Error("Guilds JSON dump was deserialized as null, aborting");
+            return;
+        }
+
+        var watch = new Stopwatch();
+        watch.Start();
+
+        await using var tx = await db.Database.BeginTransactionAsync();
+
+        foreach (var g in data.Guilds)
+            ImportGuild(db, g);
+        foreach (var i in data.Invites)
+            ImportInvite(db, i);
+        foreach (var w in data.Watchlist)
+            ImportWatchlistEntry(db, w);
+
+        await db.SaveChangesAsync();
+        await tx.CommitAsync();
+
+        Log.Information(
+            "Imported guilds, invites, and watchlist entries in {Duration}",
+            watch.Elapsed.ToDuration()
+        );
+    }
+
+    private static void ImportGuild(DatabaseContext db, GoGuild guild)
+    {
+        var channels = new Guild.ChannelConfig
+        {
+            IgnoredChannels = guild.IgnoredChannels.ToList(),
+            IgnoredUsers = guild.IgnoredUsers.ToList(),
+            GuildUpdate = guild.Channels.TryParse("GUILD_UPDATE"),
+            GuildEmojisUpdate = guild.Channels.TryParse("GUILD_EMOJIS_UPDATE"),
+            GuildRoleCreate = guild.Channels.TryParse("GUILD_ROLE_CREATE"),
+            GuildRoleUpdate = guild.Channels.TryParse("GUILD_ROLE_UPDATE"),
+            GuildRoleDelete = guild.Channels.TryParse("GUILD_ROLE_DELETE"),
+            ChannelCreate = guild.Channels.TryParse("CHANNEL_CREATE"),
+            ChannelUpdate = guild.Channels.TryParse("CHANNEL_UPDATE"),
+            ChannelDelete = guild.Channels.TryParse("CHANNEL_DELETE"),
+            GuildMemberAdd = guild.Channels.TryParse("GUILD_MEMBER_ADD"),
+            GuildMemberUpdate = guild.Channels.TryParse("GUILD_MEMBER_UPDATE"),
+            GuildKeyRoleUpdate = guild.Channels.TryParse("GUILD_KEY_ROLE_UPDATE"),
+            GuildMemberNickUpdate = guild.Channels.TryParse("GUILD_MEMBER_NICK_UPDATE"),
+            GuildMemberAvatarUpdate = guild.Channels.TryParse("GUILD_MEMBER_NICK_UPDATE"),
+            GuildMemberTimeout = guild.Channels.TryParse("GUILD_MEMBER_KICK"),
+            GuildMemberRemove = guild.Channels.TryParse("GUILD_MEMBER_REMOVE"),
+            GuildMemberKick = guild.Channels.TryParse("GUILD_MEMBER_KICK"),
+            GuildBanAdd = guild.Channels.TryParse("GUILD_BAN_ADD"),
+            GuildBanRemove = guild.Channels.TryParse("GUILD_BAN_REMOVE"),
+            InviteCreate = guild.Channels.TryParse("INVITE_CREATE"),
+            InviteDelete = guild.Channels.TryParse("INVITE_DELETE"),
+            MessageUpdate = guild.Channels.TryParse("MESSAGE_UPDATE"),
+            MessageDelete = guild.Channels.TryParse("MESSAGE_DELETE"),
+            MessageDeleteBulk = guild.Channels.TryParse("MESSAGE_DELETE_BULK"),
+        };
+
+        foreach (var (key, value) in guild.Redirects)
+            if (ulong.TryParse(key, out var fromId) && ulong.TryParse(value, out var toId))
+                channels.Redirects[fromId] = toId;
+
+        var dbGuild = new Guild
+        {
+            Id = guild.Id,
+            BannedSystems = guild.BannedSystems.ToList(),
+            KeyRoles = guild.KeyRoles.ToList(),
+            Channels = channels,
+        };
+
+        db.Guilds.Add(dbGuild);
+    }
+
+    private static void ImportInvite(DatabaseContext db, GoInvite invite)
+    {
+        var dbInvite = new Invite
+        {
+            GuildId = invite.GuildId,
+            Code = invite.Code,
+            Name = invite.Name,
+        };
+        db.Invites.Add(dbInvite);
+    }
+
+    private static void ImportWatchlistEntry(DatabaseContext db, GoWatchlistEntry watchlistEntry)
+    {
+        var dbWatchlist = new Watchlist
+        {
+            GuildId = watchlistEntry.GuildId,
+            UserId = watchlistEntry.UserId,
+            ModeratorId = watchlistEntry.ModeratorId,
+            AddedAt = watchlistEntry.AddedAt,
+            Reason = watchlistEntry.Reason,
+        };
+    }
+
+    private static ulong TryParse(this Dictionary<string, string> dict, string key) =>
+        dict.TryGetValue(key, out var raw) && ulong.TryParse(raw, out var id) ? id : 0;
+}