GetShopDataGridConfig

DREDGE-Mods · Sep 13, 2024 · 1e77ada · 1e77ada
1 parent cf242c7
commit 1e77ada
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 2 deletions.
diff --git a/Winch/Data/Shop/ModdedShopData.cs b/Winch/Data/Shop/ModdedShopData.cs
@@ -17,5 +17,19 @@ internal void Populate()
         }
     }
 
-    public static implicit operator ShopRestocker.ShopDataGridConfig(ModdedShopData shopData) => new ModdedShopDataGridConfig(shopData);
+    private ModdedShopDataGridConfig shopDataGridConfig;
+    public ModdedShopDataGridConfig ToShopDataGridConfig()
+    {
+        if (shopDataGridConfig == null)
+            shopDataGridConfig = new ModdedShopDataGridConfig(this);
+        else
+        {
+            shopDataGridConfig.gridKey = gridKey;
+            shopDataGridConfig.shopData = this;
+        }
+        return shopDataGridConfig;
+    }
+
+    public static implicit operator ShopRestocker.ShopDataGridConfig(ModdedShopData shopData) => shopData.ToShopDataGridConfig();
+    public static implicit operator ModdedShopDataGridConfig(ModdedShopData shopData) => shopData.ToShopDataGridConfig();
 }
diff --git a/Winch/Util/ShopUtil.cs b/Winch/Util/ShopUtil.cs
@@ -36,6 +36,7 @@ internal static bool PopulateDialogueLinkedShopDataFromMetaWithConverter(Dialogu
     }
 
     internal static List<string> KeepInStockItems = new();
+    internal static Dictionary<GridKey, ModdedShopDataGridConfig> ModdedShopDataGridConfigDict = new();
     internal static Dictionary<GridKey, ShopDataGridConfig> AllShopDataGridConfigDict = new();
     internal static Dictionary<string, ModdedShopData> ModdedShopDataDict = new();
     internal static Dictionary<string, ShopData> AllShopDataDict = new();
@@ -67,13 +68,38 @@ public static ShopData GetShopData(string id)
         return null;
     }
 
+    public static ModdedShopDataGridConfig GetModdedShopDataGridConfig(GridKey gridKey)
+    {
+        if (gridKey == GridKey.NONE)
+            return null;
+
+        if (ModdedShopDataGridConfigDict.TryGetValue(gridKey, out ModdedShopDataGridConfig shop))
+            return shop;
+        else
+            return null;
+    }
+
+    public static ShopDataGridConfig GetShopDataGridConfig(GridKey gridKey)
+    {
+        if (gridKey == GridKey.NONE)
+            return null;
+
+        if (AllShopDataGridConfigDict.TryGetValue(gridKey, out var shop))
+            return shop;
+
+        if (ModdedShopDataGridConfigDict.TryGetValue(gridKey, out var moddedShop))
+            return moddedShop;
+
+        return null;
+    }
+
     internal static void AddModdedShopData(ShopRestocker restocker)
     {
         restocker.itemIdsToKeep.AddRange(KeepInStockItems);
         foreach (var shopData in ModdedShopDataDict.Values)
         {
             shopData.Populate();
-            restocker.shopDataGridConfigs.Add(shopData);
+            restocker.shopDataGridConfigs.Add(shopData.ToShopDataGridConfig());
         }
     }
 
@@ -117,6 +143,7 @@ internal static void AddShopDataFromMeta(string metaPath)
         if (PopulateShopDataFromMetaWithConverter(shopData, meta))
         {
             ModdedShopDataDict.Add(id, shopData);
+            ModdedShopDataGridConfigDict.Add(shopData.gridKey, shopData.ToShopDataGridConfig());
         }
         else
         {