Remove deleted designs from automated design sets.

Ottermandias · Sep 29, 2023 · 869618e · 869618e
1 parent db6687d
commit 869618e
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 2 deletions.
diff --git a/Glamourer/Automation/AutoDesignManager.cs b/Glamourer/Automation/AutoDesignManager.cs
@@ -13,12 +13,13 @@
 using Glamourer.Structs;
 using Newtonsoft.Json;
 using Newtonsoft.Json.Linq;
+using OtterGui;
 using OtterGui.Filesystem;
 using Penumbra.GameData.Actors;
 
 namespace Glamourer.Automation;
 
-public class AutoDesignManager : ISavable, IReadOnlyList<AutoDesignSet>
+public class AutoDesignManager : ISavable, IReadOnlyList<AutoDesignSet>, IDisposable
 {
     public const int CurrentVersion = 1;
 
@@ -28,6 +29,7 @@ public class AutoDesignManager : ISavable, IReadOnlyList<AutoDesignSet>
     private readonly DesignManager     _designs;
     private readonly ActorService      _actors;
     private readonly AutomationChanged _event;
+    private readonly DesignChanged     _designEvent;
 
     private readonly List<AutoDesignSet>                        _data    = new();
     private readonly Dictionary<ActorIdentifier, AutoDesignSet> _enabled = new();
@@ -36,17 +38,22 @@ public IReadOnlyDictionary<ActorIdentifier, AutoDesignSet> EnabledSets
         => _enabled;
 
     public AutoDesignManager(JobService jobs, ActorService actors, SaveService saveService, DesignManager designs, AutomationChanged @event,
-        FixedDesignMigrator migrator, DesignFileSystem fileSystem)
+        FixedDesignMigrator migrator, DesignFileSystem fileSystem, DesignChanged designEvent)
     {
         _jobs        = jobs;
         _actors      = actors;
         _saveService = saveService;
         _designs     = designs;
         _event       = @event;
+        _designEvent = designEvent;
+        _designEvent.Subscribe(OnDesignChange, DesignChanged.Priority.AutoDesignManager);
         Load();
         migrator.ConsumeMigratedData(_actors, fileSystem, this);
     }
 
+    public void Dispose()
+        => _designEvent.Unsubscribe(OnDesignChange);
+
     public IEnumerator<AutoDesignSet> GetEnumerator()
         => _data.GetEnumerator();
 
@@ -538,4 +545,27 @@ static ActorIdentifier[] CreateNpcs(ActorManager manager, ActorIdentifier identi
             _                  => Array.Empty<ActorIdentifier>(),
         };
     }
+
+    private void OnDesignChange(DesignChanged.Type type, Design design, object? data)
+    {
+        if (type is not DesignChanged.Type.Deleted)
+            return;
+
+        foreach (var (set, idx) in this.WithIndex())
+        {
+            var deleted = 0;
+            for (var i = 0; i < set.Designs.Count; ++i)
+            {
+                if (set.Designs[i].Design != design)
+                    continue;
+
+                DeleteDesign(set, i--);
+                ++deleted;
+            }
+
+            if (deleted > 0)
+                Glamourer.Log.Information(
+                    $"Removed {deleted} automated designs from automated design set {idx} due to deletion of {design.Incognito}.");
+        }
+    }
 }
diff --git a/Glamourer/Events/DesignChanged.cs b/Glamourer/Events/DesignChanged.cs
@@ -81,6 +81,9 @@ public enum Priority
 
         /// <seealso cref="Gui.Tabs.DesignTab.DesignFileSystemSelector.OnDesignChange"/>
         DesignFileSystemSelector = -1,
+
+        /// <seealso cref="Automation.AutoDesignManager.OnDesignChange"/>
+        AutoDesignManager = 1,
     }
 
     public DesignChanged()