From bf50a9b8f8d92d0a23a498822513d65d9fade133 Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Tue, 18 Jan 2022 14:30:41 +0900
Subject: [PATCH] Also backup the realm database before migration

---
 osu.Game/Database/EFToRealmMigrator.cs   |  8 ++++++--
 osu.Game/Database/RealmContextFactory.cs | 11 +++++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/osu.Game/Database/EFToRealmMigrator.cs b/osu.Game/Database/EFToRealmMigrator.cs
index d8a3b66c7c..9fad29b3f4 100644
--- a/osu.Game/Database/EFToRealmMigrator.cs
+++ b/osu.Game/Database/EFToRealmMigrator.cs
@@ -75,6 +75,9 @@ namespace osu.Game.Database
                 if (!realm.All<BeatmapSetInfo>().Any(s => !s.Protected))
                 {
                     Logger.Log($"Migrating {existingBeatmapSets.Count} beatmaps", LoggingTarget.Database);
+                    string migration = $"before_beatmap_migration_{DateTimeOffset.UtcNow.ToUnixTimeSeconds()}";
+                    realmContextFactory.CreateBackup($"client.{migration}.realm");
+                    efContextFactory.CreateBackup($"client.{migration}.db");
 
                     foreach (var beatmapSet in existingBeatmapSets)
                     {
@@ -131,7 +134,6 @@ namespace osu.Game.Database
                     }
                 }
 
-                efContextFactory.CreateBackup($"client.before_beatmap_migration_{DateTimeOffset.UtcNow.ToUnixTimeSeconds()}.db");
                 ef.Context.RemoveRange(existingBeatmapSets);
                 // Intentionally don't clean up the files, so they don't get purged by EF.
 
@@ -187,6 +189,9 @@ namespace osu.Game.Database
                 if (!realm.All<ScoreInfo>().Any())
                 {
                     Logger.Log($"Migrating {existingScores.Count} scores", LoggingTarget.Database);
+                    string migration = $"before_score_migration_{DateTimeOffset.UtcNow.ToUnixTimeSeconds()}";
+                    realmContextFactory.CreateBackup($"client.{migration}.realm");
+                    efContextFactory.CreateBackup($"client.{migration}.db");
 
                     foreach (var score in existingScores)
                     {
@@ -222,7 +227,6 @@ namespace osu.Game.Database
                     }
                 }
 
-                efContextFactory.CreateBackup($"client.before_scores_migration_{DateTimeOffset.UtcNow.ToUnixTimeSeconds()}.db");
                 db.Context.RemoveRange(existingScores);
                 // Intentionally don't clean up the files, so they don't get purged by EF.
 
diff --git a/osu.Game/Database/RealmContextFactory.cs b/osu.Game/Database/RealmContextFactory.cs
index 01c54d6ee2..c86a9e54e1 100644
--- a/osu.Game/Database/RealmContextFactory.cs
+++ b/osu.Game/Database/RealmContextFactory.cs
@@ -2,6 +2,7 @@
 // See the LICENCE file in the repository root for full licence text.
 
 using System;
+using System.IO;
 using System.Linq;
 using System.Reflection;
 using System.Threading;
@@ -353,6 +354,16 @@ namespace osu.Game.Database
         private string? getRulesetShortNameFromLegacyID(long rulesetId) =>
             efContextFactory?.Get().RulesetInfo.FirstOrDefault(r => r.ID == rulesetId)?.ShortName;
 
+        public void CreateBackup(string filename)
+        {
+            using (BlockAllOperations())
+            {
+                using (var source = storage.GetStream(Filename))
+                using (var destination = storage.GetStream(filename, FileAccess.Write, FileMode.CreateNew))
+                    source.CopyTo(destination);
+            }
+        }
+
         /// <summary>
         /// Flush any active contexts and block any further writes.
         /// </summary>