From 20bcc00662d77f0445f373720cbe3f92f3c0658e Mon Sep 17 00:00:00 2001
From: Bogdan <mynameisbogdan@users.noreply.github.com>
Date: Sun, 25 Jun 2023 05:14:08 +0300
Subject: [PATCH] Fix apprise server url migration

---
 .../031_apprise_server_urlFixture.cs          | 62 +++++++++++++++++++
 .../Migration/031_apprise_server_url.cs       | 45 +++++++++-----
 2 files changed, 90 insertions(+), 17 deletions(-)
 create mode 100644 src/NzbDrone.Core.Test/Datastore/Migration/031_apprise_server_urlFixture.cs

diff --git a/src/NzbDrone.Core.Test/Datastore/Migration/031_apprise_server_urlFixture.cs b/src/NzbDrone.Core.Test/Datastore/Migration/031_apprise_server_urlFixture.cs
new file mode 100644
index 000000000..907bd7e47
--- /dev/null
+++ b/src/NzbDrone.Core.Test/Datastore/Migration/031_apprise_server_urlFixture.cs
@@ -0,0 +1,62 @@
+using System.Collections.Generic;
+using System.Linq;
+using FluentAssertions;
+using NUnit.Framework;
+using NzbDrone.Common.Serializer;
+using NzbDrone.Core.Datastore.Migration;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.Datastore.Migration
+{
+    [TestFixture]
+    public class apprise_server_urlFixture : MigrationTest<apprise_server_url>
+    {
+        [Test]
+        public void should_rename_server_url_setting_for_apprise()
+        {
+            var db = WithMigrationTestDb(c =>
+            {
+                c.Insert.IntoTable("Notifications").Row(new
+                {
+                    Name = "Apprise",
+                    Implementation = "Apprise",
+                    Settings = new
+                    {
+                        BaseUrl = "http://localhost:8000",
+                        NotificationType = 0
+                    }.ToJson(),
+                    ConfigContract = "AppriseSettings",
+                    OnHealthIssue = true,
+                    IncludeHealthWarnings = true,
+                    OnApplicationUpdate = true,
+                    OnGrab = true,
+                    IncludeManualGrabs = true
+                });
+            });
+
+            var items = db.Query<NotificationDefinition31>("SELECT * FROM \"Notifications\"");
+
+            items.Should().HaveCount(1);
+
+            items.First().Settings.Should().NotContainKey("baseUrl");
+            items.First().Settings.Should().ContainKey("serverUrl");
+            items.First().Settings.GetValueOrDefault("serverUrl").Should().Be("http://localhost:8000");
+        }
+    }
+
+    public class NotificationDefinition31
+    {
+        public int Id { get; set; }
+        public int Priority { get; set; }
+        public string Name { get; set; }
+        public string Implementation { get; set; }
+        public Dictionary<string, string> Settings { get; set; }
+        public string ConfigContract { get; set; }
+        public bool OnHealthIssue { get; set; }
+        public bool IncludeHealthWarnings { get; set; }
+        public bool OnApplicationUpdate { get; set; }
+        public bool OnGrab { get; set; }
+        public bool IncludeManualGrabs { get; set; }
+        public List<int> Tags { get; set; }
+    }
+}
diff --git a/src/NzbDrone.Core/Datastore/Migration/031_apprise_server_url.cs b/src/NzbDrone.Core/Datastore/Migration/031_apprise_server_url.cs
index 409390649..f49586e7e 100644
--- a/src/NzbDrone.Core/Datastore/Migration/031_apprise_server_url.cs
+++ b/src/NzbDrone.Core/Datastore/Migration/031_apprise_server_url.cs
@@ -1,3 +1,4 @@
+using System.Collections.Generic;
 using System.Data;
 using Dapper;
 using FluentMigrator;
@@ -17,33 +18,43 @@ namespace NzbDrone.Core.Datastore.Migration
 
         private void MigrateToServerUrl(IDbConnection conn, IDbTransaction tran)
         {
-            using var selectCommand = conn.CreateCommand();
-            selectCommand.Transaction = tran;
-            selectCommand.CommandText = "SELECT \"Id\", \"Settings\" FROM \"Notifications\" WHERE \"Implementation\" = 'Apprise'";
+            var updatedNotifications = new List<object>();
 
-            using var reader = selectCommand.ExecuteReader();
-
-            while (reader.Read())
+            using (var selectCommand = conn.CreateCommand())
             {
-                var id = reader.GetInt32(0);
-                var settings = reader.GetString(1);
+                selectCommand.Transaction = tran;
+                selectCommand.CommandText = "SELECT \"Id\", \"Settings\" FROM \"Notifications\" WHERE \"Implementation\" = 'Apprise'";
+
+                using var reader = selectCommand.ExecuteReader();
 
-                if (!string.IsNullOrWhiteSpace(settings))
+                while (reader.Read())
                 {
-                    var jsonObject = Json.Deserialize<JObject>(settings);
+                    var id = reader.GetInt32(0);
+                    var settings = reader.GetString(1);
 
-                    if (jsonObject.ContainsKey("baseUrl"))
+                    if (!string.IsNullOrWhiteSpace(settings))
                     {
-                        jsonObject.Add("serverUrl", jsonObject.Value<string>("baseUrl"));
-                        jsonObject.Remove("baseUrl");
+                        var jsonObject = Json.Deserialize<JObject>(settings);
+
+                        if (jsonObject.ContainsKey("baseUrl"))
+                        {
+                            jsonObject.Add("serverUrl", jsonObject.Value<string>("baseUrl"));
+                            jsonObject.Remove("baseUrl");
+                        }
+
+                        settings = jsonObject.ToJson();
                     }
 
-                    settings = jsonObject.ToJson();
+                    updatedNotifications.Add(new
+                    {
+                        Id = id,
+                        Settings = settings
+                    });
                 }
-
-                var parameters = new { Settings = settings, Id = id };
-                conn.Execute("UPDATE Notifications SET Settings = @Settings WHERE Id = @Id", parameters, transaction: tran);
             }
+
+            var updateNotificationsSql = "UPDATE \"Notifications\" SET \"Settings\" = @Settings WHERE \"Id\" = @Id";
+            conn.Execute(updateNotificationsSql, updatedNotifications, transaction: tran);
         }
     }
 }