Killed history

NadekoBot.Core/Services/Settings/BotConfigMigrator.cs

@@ -0,0 +1,122 @@
+using System.Collections.Generic;
+using System.Data.Common;
+using System.Globalization;
+using Microsoft.EntityFrameworkCore;
+using NadekoBot.Core.Common.Configs;
+using Serilog;
+using SixLabors.ImageSharp.PixelFormats;
+
+namespace NadekoBot.Core.Services
+{
+    public sealed class BotConfigMigrator : IConfigMigrator
+    {
+        private readonly DbService _db;
+        private readonly BotConfigService _bss;
+
+        public BotConfigMigrator(DbService dbService, BotConfigService bss)
+        {
+            _db = dbService;
+            _bss = bss;
+        }
+
+        public void EnsureMigrated()
+        {
+            using var uow = _db.GetDbContext();
+            using var conn = uow._context.Database.GetDbConnection();
+            
+            // check if bot config exists
+            using (var checkTableCommand = conn.CreateCommand())
+            {
+                // make sure table still exists
+                checkTableCommand.CommandText =
+                    "SELECT name FROM sqlite_master WHERE type='table' AND name='BotConfig';";
+                var checkReader = checkTableCommand.ExecuteReader();
+                if (!checkReader.HasRows)
+                    return;
+            }
+            
+            MigrateBotConfig(conn);
+
+            using var dropBlockedTable = conn.CreateCommand();
+            dropBlockedTable.CommandText = "DROP TABLE IF EXISTS BlockedCmdOrMdl;";
+            dropBlockedTable.ExecuteNonQuery();
+        }
+
+        private void MigrateBotConfig(DbConnection conn)
+        {
+            
+
+            using (var checkMigratedCommand = conn.CreateCommand())
+            {
+                checkMigratedCommand.CommandText =
+                    "UPDATE BotConfig SET HasMigratedBotSettings = 1 WHERE HasMigratedBotSettings = 0;";
+                var changedRows = checkMigratedCommand.ExecuteNonQuery();
+                if (changedRows == 0)
+                    return;
+            }
+
+            Log.Information("Migrating bot settings...");
+
+            var blockedCommands = new HashSet<string>();
+            using (var cmdCom = conn.CreateCommand())
+            {
+                cmdCom.CommandText = $"SELECT Name from BlockedCmdOrMdl WHERE BotConfigId is not NULL";
+                var cmdReader = cmdCom.ExecuteReader();
+                while (cmdReader.Read())
+                    blockedCommands.Add(cmdReader.GetString(0));
+            }
+
+            var blockedModules = new HashSet<string>();
+            using (var mdlCom = conn.CreateCommand())
+            {
+                mdlCom.CommandText = $"SELECT Name from BlockedCmdOrMdl WHERE BotConfigId is NULL";
+                var mdlReader = mdlCom.ExecuteReader();
+                while (mdlReader.Read())
+                    blockedModules.Add(mdlReader.GetString(0));
+            }
+
+            using var com = conn.CreateCommand();
+            com.CommandText = $@"SELECT DefaultPrefix, ForwardMessages, ForwardToAllOwners,
+OkColor, ErrorColor, ConsoleOutputType, DMHelpString, HelpString, RotatingStatuses, Locale, GroupGreets
+FROM BotConfig";
+
+            using var reader = com.ExecuteReader();
+            if (!reader.Read())
+                return;
+
+            _bss.ModifyConfig((x) =>
+            {
+                x.Prefix = reader.GetString(0);
+                x.ForwardMessages = reader.GetBoolean(1);
+                x.ForwardToAllOwners = reader.GetBoolean(2);
+                x.Color = new ColorConfig()
+                {
+                    Ok = Rgba32.TryParseHex(reader.GetString(3), out var okColor)
+                        ? okColor
+                        : Rgba32.ParseHex("00e584"),
+                    Error = Rgba32.TryParseHex(reader.GetString(4), out var errorColor)
+                        ? errorColor
+                        : Rgba32.ParseHex("ee281f"),
+                };
+                x.ConsoleOutputType = (ConsoleOutputType) reader.GetInt32(5);
+                x.DmHelpText = reader.IsDBNull(6) ? string.Empty : reader.GetString(6);
+                x.HelpText = reader.IsDBNull(7) ? string.Empty : reader.GetString(7);
+                x.RotateStatuses = reader.GetBoolean(8);
+                try
+                {
+                    x.DefaultLocale = new CultureInfo(reader.GetString(9));
+                }
+                catch
+                {
+                    x.DefaultLocale = new CultureInfo("en-US");
+                }
+
+                x.GroupGreets = reader.GetBoolean(10);
+                x.Blocked.Commands = blockedCommands;
+                x.Blocked.Modules = blockedModules;
+            });
+
+            Log.Information("Data written to data/bot.yml");
+        }
+    }
+}