From 69351d2cdf9de1b4c01ef5740c73983cb3a94fe2 Mon Sep 17 00:00:00 2001
From: Noah M <noahcodes@protonmail.com>
Date: Wed, 18 May 2022 01:09:58 -0500
Subject: [PATCH] Implement button to delete all beatmap videos

---
 osu.Game/Beatmaps/BeatmapManager.cs           | 12 ++++++
 .../MaintenanceSettingsStrings.cs             |  5 +++
 .../Sections/Maintenance/GeneralSettings.cs   | 14 +++++++
 .../MassDeleteConfirmationDialog.cs           |  7 ++++
 osu.Game/Stores/RealmArchiveModelManager.cs   | 38 +++++++++++++++++++
 5 files changed, 76 insertions(+)

diff --git a/osu.Game/Beatmaps/BeatmapManager.cs b/osu.Game/Beatmaps/BeatmapManager.cs
index 5f7de0d762..2a5df9e206 100644
--- a/osu.Game/Beatmaps/BeatmapManager.cs
+++ b/osu.Game/Beatmaps/BeatmapManager.cs
@@ -319,6 +319,18 @@ namespace osu.Game.Beatmaps
             });
         }
 
+        public void DeleteVideos(Expression<Func<BeatmapSetInfo, bool>>? filter = null, bool silent = false)
+        {
+            realm.Write(r =>
+            {
+                var items = r.All<BeatmapSetInfo>().Where(s => !s.DeletePending && !s.Protected);
+
+                if (filter != null)
+                    items = items.Where(filter);
+                beatmapModelManager.DeleteVideos(items.ToList(), silent);
+            });
+        }
+
         public void UndeleteAll()
         {
             realm.Run(r => beatmapModelManager.Undelete(r.All<BeatmapSetInfo>().Where(s => s.DeletePending).ToList()));
diff --git a/osu.Game/Localisation/MaintenanceSettingsStrings.cs b/osu.Game/Localisation/MaintenanceSettingsStrings.cs
index a0e1a9ddab..7a04bcd1ca 100644
--- a/osu.Game/Localisation/MaintenanceSettingsStrings.cs
+++ b/osu.Game/Localisation/MaintenanceSettingsStrings.cs
@@ -29,6 +29,11 @@ namespace osu.Game.Localisation
         /// </summary>
         public static LocalisableString DeleteAllBeatmaps => new TranslatableString(getKey(@"delete_all_beatmaps"), @"Delete ALL beatmaps");
 
+        /// <summary>
+        /// "Delete ALL beatmap videos"
+        /// </summary>
+        public static LocalisableString DeleteAllBeatmapVideos => new TranslatableString(getKey(@"delete_all_beatmap_videos"), @"Delete ALL beatmap videos");
+
         /// <summary>
         /// "Import scores from stable"
         /// </summary>
diff --git a/osu.Game/Overlays/Settings/Sections/Maintenance/GeneralSettings.cs b/osu.Game/Overlays/Settings/Sections/Maintenance/GeneralSettings.cs
index be4b0decd9..c2ebb59ecc 100644
--- a/osu.Game/Overlays/Settings/Sections/Maintenance/GeneralSettings.cs
+++ b/osu.Game/Overlays/Settings/Sections/Maintenance/GeneralSettings.cs
@@ -28,6 +28,7 @@ namespace osu.Game.Overlays.Settings.Sections.Maintenance
         private SettingsButton deleteSkinsButton;
         private SettingsButton restoreButton;
         private SettingsButton undeleteButton;
+        private SettingsButton deleteBeatmapVideosButton;
 
         [BackgroundDependencyLoader(permitNulls: true)]
         private void load(BeatmapManager beatmaps, ScoreManager scores, SkinManager skins, [CanBeNull] CollectionManager collectionManager, [CanBeNull] LegacyImportManager legacyImportManager, IDialogOverlay dialogOverlay)
@@ -58,6 +59,19 @@ namespace osu.Game.Overlays.Settings.Sections.Maintenance
                 }
             });
 
+            Add(deleteBeatmapVideosButton = new DangerousSettingsButton
+            {
+                Text = MaintenanceSettingsStrings.DeleteAllBeatmapVideos,
+                Action = () =>
+                {
+                    dialogOverlay?.Push(new MassVideoDeleteConfirmationDialog(() =>
+                    {
+                        deleteBeatmapVideosButton.Enabled.Value = false;
+                        Task.Run(() => beatmaps.DeleteVideos()).ContinueWith(t => Schedule(() => deleteBeatmapsButton.Enabled.Value = true));
+                    }));
+                }
+            });
+
             if (legacyImportManager?.SupportsImportFromStable == true)
             {
                 Add(importScoresButton = new SettingsButton
diff --git a/osu.Game/Overlays/Settings/Sections/Maintenance/MassDeleteConfirmationDialog.cs b/osu.Game/Overlays/Settings/Sections/Maintenance/MassDeleteConfirmationDialog.cs
index c481c80d82..274b1060ca 100644
--- a/osu.Game/Overlays/Settings/Sections/Maintenance/MassDeleteConfirmationDialog.cs
+++ b/osu.Game/Overlays/Settings/Sections/Maintenance/MassDeleteConfirmationDialog.cs
@@ -29,4 +29,11 @@ namespace osu.Game.Overlays.Settings.Sections.Maintenance
             };
         }
     }
+    public class MassVideoDeleteConfirmationDialog : MassDeleteConfirmationDialog
+    {
+        public MassVideoDeleteConfirmationDialog(Action deleteAction) : base(deleteAction)
+        {
+            BodyText = "All beatmap videos? This cannot be undone!";
+        }
+    }
 }
diff --git a/osu.Game/Stores/RealmArchiveModelManager.cs b/osu.Game/Stores/RealmArchiveModelManager.cs
index cc8229b436..f70aa41cfb 100644
--- a/osu.Game/Stores/RealmArchiveModelManager.cs
+++ b/osu.Game/Stores/RealmArchiveModelManager.cs
@@ -132,6 +132,44 @@ namespace osu.Game.Stores
             notification.State = ProgressNotificationState.Completed;
         }
 
+        /// <summary>
+        /// Delete videos from a list of items.
+        /// This will post notifications tracking progress.
+        /// </summary>
+        public void DeleteVideos(List<TModel> items, bool silent = false)
+        {
+            if (items.Count == 0) return;
+
+            var notification = new ProgressNotification
+            {
+                Progress = 0,
+                Text = $"Preparing to delete all {HumanisedModelName} videos...",
+                CompletionText = $"Deleted all {HumanisedModelName} videos!",
+                State = ProgressNotificationState.Active,
+            };
+            if (!silent)
+                PostNotification?.Invoke(notification);
+
+            int i = 0;
+
+            foreach (var b in items)
+            {
+                if (notification.State == ProgressNotificationState.Cancelled)
+                    // user requested abort
+                    return;
+
+                notification.Text = $"Deleting videos from {HumanisedModelName}s ({++i} of {items.Count})";
+
+                var video = b.Files.FirstOrDefault(f => f.Filename.EndsWith(".mp4") || f.Filename.EndsWith(".avi") || f.Filename.EndsWith(".mov") || f.Filename.EndsWith(".flv"));
+                if (video != null)
+                    DeleteFile(b, video);
+
+                notification.Progress = (float)i / items.Count;
+            }
+
+            notification.State = ProgressNotificationState.Completed;
+        }
+
         /// <summary>
         /// Restore multiple items that were previously deleted.
         /// This will post notifications tracking progress.