Uninstall ROMs when install flat is enabled

Author: matthew-pye

Games/RomMGameInfo.Plugin.cs

@@ -26,7 +26,7 @@ public EmulatorMapping Mapping
 
         public InstallController GetInstallController(Game game, RomM romm, GameInstallInfo GameData) => new RomMInstallController(game, romm, GameData);
 
-        public UninstallController GetUninstallController(Game game, RomM romm) => new RomMUninstallController(game, romm);
+        public UninstallController GetUninstallController(Game game, RomM romm, EmulatorMapping mapping) => new RomMUninstallController(game, romm, mapping);
 
         protected IEnumerable<string> GetDescriptionLines()
         {

Games/RomMUninstallController.cs

@@ -1,5 +1,6 @@
 using Playnite.SDK.Models;
 using Playnite.SDK.Plugins;
+using RomM.Settings;
 using System.IO;
 using System.Windows;
 
@@ -8,24 +9,38 @@ namespace RomM.Games
     internal class RomMUninstallController : UninstallController
     {
         private readonly IRomM _romM;
+        private EmulatorMapping _mapping;
 
-        internal RomMUninstallController(Game game, IRomM romM) : base(game)
+        internal RomMUninstallController(Game game, IRomM romM, EmulatorMapping mapping) : base(game)
         {
             Name = "Uninstall";
             _romM = romM;
+            _mapping = mapping;
         }
 
         public override void Uninstall(UninstallActionArgs args)
         {
-            if (new DirectoryInfo(Game.InstallDirectory).Exists)
+            if(_mapping.InstallFlat)
             {
-                Directory.Delete(Game.InstallDirectory, true);
+                foreach (var RomFile in Game.Roms)
+                {
+                    if(File.Exists(RomFile.Path))
+                        File.Delete(RomFile.Path);
+                }
             }
             else
             {
-                _romM.Playnite.Dialogs.ShowMessage($"\"{Game.Name}\" folder could not be found. Marking as uninstalled.", "Game not found", MessageBoxButton.OK);
+                if (new DirectoryInfo(Game.InstallDirectory).Exists)
+                {
+                    Directory.Delete(Game.InstallDirectory, true);
+                }
+                else
+                {
+                    _romM.Playnite.Dialogs.ShowMessage($"\"{Game.Name}\" folder could not be found. Marking as uninstalled.", "Game not found", MessageBoxButton.OK);
+                }
             }
-            Game.Roms.Clear();
+
+                Game.Roms.Clear();
             InvokeOnUninstalled(new GameUninstalledEventArgs());
         }
     }

RomM.cs

@@ -490,7 +490,28 @@ public override IEnumerable<UninstallController> GetUninstallActions(GetUninstal
         {
             if (args.Game.PluginId == Id)
             {
-                yield return new RomMUninstallController(args.Game, this);
+                EmulatorMapping mapping = null;
+
+                try
+                {
+                    var splitID = args.Game.GameId.Split(':');
+                    string sidecarPath = $"{ROMDataPath}{splitID[1]}.json";
+                    if (File.Exists(sidecarPath))
+                    { 
+                            var existingJson = File.ReadAllText(sidecarPath);
+                            var localROM = JsonConvert.DeserializeObject<RomMRomLocal>(existingJson);
+                            mapping = Settings.Mappings.FirstOrDefault(x => x.MappingId == localROM.MappingID);
+                    }
+                }
+                catch (Exception)
+                {
+                    Logger.Error($"{args.Game.Name} GameID is malformed or json file is corrupted!");
+                }
+
+                if (mapping == null)
+                    yield return null;
+
+                yield return new RomMUninstallController(args.Game, this, mapping);
             }
         }
         public override void OnGameInstalled(OnGameInstalledEventArgs args)