Merge pull request #576 from TheTrackerCouncil/keysanity-minimal-ui

Added changes for keysanity minimal ui
TheTrackerCouncil · Sep 23, 2024 · a477c0d · a477c0d
2 parents d1fa1cc + 97d7e0a
commit a477c0d
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/src/TrackerCouncil.Smz3.Data/Configuration/ConfigTypes/UIGridLocation.cs b/src/TrackerCouncil.Smz3.Data/Configuration/ConfigTypes/UIGridLocation.cs
@@ -39,4 +39,9 @@ public UIGridLocation() { }
     /// Collection of object identifiers to look up for this location
     /// </summary>
     public ICollection<string> Identifiers { get; set; } = new List<string>();
+
+    /// <summary>
+    /// Map of identifier to images to use
+    /// </summary>
+    public Dictionary<string, string> ReplacementImages { get; set; } = new();
 }
diff --git a/src/TrackerCouncil.Smz3.UI/Services/TrackerWindowService.cs b/src/TrackerCouncil.Smz3.UI/Services/TrackerWindowService.cs
@@ -573,9 +573,14 @@ private TrackerWindowPanelViewModel GetItemPanelViewModel(UIGridLocation gridLoc
             items[item] = fileName;
         }
 
+        var replacementImages =
+            gridLocation.ReplacementImages.ToDictionary(x => x.Key,
+                x => uiService.GetSpritePath("Items", x.Value, out _));
+
         var model = new TrackerWindowItemPanelViewModel()
         {
             Items = items,
+            ItemReplacementImages = replacementImages,
             ConnectedItems = connectedItems,
             LabelImage = labelImage,
             IsLabelActive = items.Keys.Any(x => x.State.TrackingState > 0),

diff --git a/src/TrackerCouncil.Smz3.UI/ViewModels/TrackerWindowItemPanelViewModel.cs b/src/TrackerCouncil.Smz3.UI/ViewModels/TrackerWindowItemPanelViewModel.cs
@@ -34,6 +34,8 @@ public class TrackerWindowItemPanelViewModel : TrackerWindowPanelViewModel
 
     public Dictionary<Item, string>? Items { get; set; }
 
+    public Dictionary<string, string?> ItemReplacementImages { get; set; } = [];
+
     public List<Item>? ConnectedItems { get; set; }
 
     [Reactive] public int ItemCount { get; set; }
@@ -59,7 +61,7 @@ public void UpdateItem(Item? item, string? path)
         if (Items != null && item != null && !string.IsNullOrEmpty(path))
         {
             var key = Items.Keys.First(x => x.Type == item.Type);
-            Items[key] = path;
+            Items[key] = GetItemPath(item, path);
         }
 
         if (ConnectedItems != null)
@@ -95,7 +97,7 @@ public override List<TrackerWindowPanelImage> GetMainImages()
 
                 images.Add(new TrackerWindowPanelImage
                 {
-                    ImagePath = item.Value,
+                    ImagePath = GetItemPath(item.Key ,item.Value),
                     IsActive = isActive
                 });
 
@@ -200,4 +202,13 @@ public override List<MenuItem> GetMenuItems()
 
         return menuItems;
     }
+
+    private string GetItemPath(Item item, string desiredPath)
+    {
+        if (ItemReplacementImages.TryGetValue(item.Name, out var replacementImage))
+        {
+            return replacementImage ?? desiredPath;
+        }
+        return desiredPath;
+    }
 }