Fix bug with critter feeder

daviscook477 · Jan 9, 2020 · a860405 · a860405
1 parent 899776c
commit a860405
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/RollerSnake/RollerSnakePatches.cs b/src/RollerSnake/RollerSnakePatches.cs
@@ -38,6 +38,23 @@ public static void Postfix(ref Immigration __instance)
             }
         }
 
+        [HarmonyPatch(typeof(CreatureFeederConfig))]
+        [HarmonyPatch(nameof(CreatureFeederConfig.ConfigurePost))]
+        public static class CreatureFeederConfig_ConfigurePost_Patch
+        {
+            public static void Postfix(BuildingDef def)
+            {
+                List<Tag> tagList = def.BuildingComplete.GetComponent<Storage>().storageFilters;
+                Tag[] target_species = new Tag[1]
+                {
+                    BaseRollerSnakeConfig.SpeciesId
+                };
+                foreach (KeyValuePair<Tag, Diet> collectDiet in DietManager.CollectDiets(target_species))
+                    tagList.Add(collectDiet.Key);
+                def.BuildingComplete.GetComponent<Storage>().storageFilters = tagList;
+            }
+        }
+
         [HarmonyPatch(typeof(CodexEntryGenerator), "GenerateCreatureEntries")]
         public class CodexEntryGenerator_GenerateCreatureEntries_Patch
         {