1 files changed, 56 insertions, 0 deletions
diff --git a/src/main/java/com/anthonyhilyard/iceberg/mixin/ConfigMenusForgeIEntryDataMixin.java b/src/main/java/com/anthonyhilyard/iceberg/mixin/ConfigMenusForgeIEntryDataMixin.java
new file mode 100644
index 0000000..9bd54f6
--- /dev/null
+++ b/src/main/java/com/anthonyhilyard/iceberg/mixin/ConfigMenusForgeIEntryDataMixin.java
@@ -0,0 +1,56 @@
+package com.anthonyhilyard.iceberg.mixin;
+
+import java.util.Map;
+
+import com.anthonyhilyard.iceberg.util.ConfigMenusForgeHelper;
+import com.electronwill.nightconfig.core.UnmodifiableConfig;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Maps;
+
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Overwrite;
+
+import fuzs.configmenusforge.client.gui.data.IEntryData;
+import net.minecraftforge.fml.config.IConfigSpec;
+import net.minecraftforge.fml.config.ModConfig;
+
+@Mixin(IEntryData.class)
+public interface ConfigMenusForgeIEntryDataMixin
+{
+	/**
+	 * @author Iceberg
+	 * @reason Overwriting makeValueToDataMap to allow classes other than ForgeConfigSpec to be supported.
+	 */
+	@Overwrite(remap = false)
+	public static Map<Object, IEntryData> makeValueToDataMap(ModConfig config)
+	{
+		if (checkInvalid(config))
+		{
+			return ImmutableMap.of();
+		}
+		Map<Object, IEntryData> allData = Maps.newHashMap();
+		UnmodifiableConfig spec = config.getSpec();
+		ConfigMenusForgeHelper.makeValueToDataMap(spec, ConfigMenusForgeHelper.getValues(spec), config.getConfigData(), allData, "");
+		return ImmutableMap.copyOf(allData);
+	}
+
+	/**
+	 * @author Iceberg
+	 * @reason Overwriting checkInvalid to allow classes other than ForgeConfigSpec to be supported.
+	 */
+	@Overwrite(remap = false)
+	public static boolean checkInvalid(ModConfig config)
+	{
+		IConfigSpec<?> spec = config.getSpec();
+
+		// True / false means the config class has been cached, null means it's new.
+		Boolean cachedValue = ConfigMenusForgeHelper.cachedValidity(spec.getClass());
+		if (cachedValue == null)
+		{
+			// It's not cached, so do the lookup via MethodHandles API and cache the results.
+			ConfigMenusForgeHelper.cacheClass(spec.getClass());
+		}
+
+		return config.getConfigData() == null || !ConfigMenusForgeHelper.cachedValidity(spec.getClass()) || !ConfigMenusForgeHelper.isLoaded(spec);
+	}
+}