1 files changed, 97 insertions, 0 deletions
diff --git a/src/texturePacks/java/moe/nea/firmament/mixins/custommodels/SupplyFakeModelPatch.java b/src/texturePacks/java/moe/nea/firmament/mixins/custommodels/SupplyFakeModelPatch.java
new file mode 100644
index 0000000..75cedf8
--- /dev/null
+++ b/src/texturePacks/java/moe/nea/firmament/mixins/custommodels/SupplyFakeModelPatch.java
@@ -0,0 +1,97 @@
+package moe.nea.firmament.mixins.custommodels;
+
+import com.google.gson.JsonObject;
+import com.llamalad7.mixinextras.injector.ModifyReturnValue;
+import com.llamalad7.mixinextras.sugar.Local;
+import moe.nea.firmament.Firmament;
+import moe.nea.firmament.features.texturepack.CustomSkyBlockTextures;
+import moe.nea.firmament.features.texturepack.HeadModelChooser;
+import moe.nea.firmament.features.texturepack.PredicateModel;
+import moe.nea.firmament.util.ErrorUtil;
+import net.minecraft.client.item.ItemAsset;
+import net.minecraft.client.item.ItemAssetsLoader;
+import net.minecraft.client.render.item.model.BasicItemModel;
+import net.minecraft.client.render.item.model.ItemModel;
+import net.minecraft.resource.Resource;
+import net.minecraft.resource.ResourceManager;
+import net.minecraft.resource.ResourcePack;
+import net.minecraft.util.Identifier;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+
+import java.io.InputStreamReader;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.Executor;
+import java.util.stream.Collector;
+import java.util.stream.Collectors;
+
+@Mixin(ItemAssetsLoader.class)
+public class SupplyFakeModelPatch {
+
+	@ModifyReturnValue(
+		method = "load",
+		at = @At("RETURN")
+	)
+	private static CompletableFuture<ItemAssetsLoader.Result> injectFakeGeneratedModels(
+		CompletableFuture<ItemAssetsLoader.Result> original,
+		@Local(argsOnly = true) ResourceManager resourceManager,
+		@Local(argsOnly = true) Executor executor
+	) {
+		return original.thenCompose(oldModels -> CompletableFuture.supplyAsync(() -> supplyExtraModels(resourceManager, oldModels), executor));
+	}
+
+	private static ItemAssetsLoader.Result supplyExtraModels(ResourceManager resourceManager, ItemAssetsLoader.Result oldModels) {
+		if (!CustomSkyBlockTextures.TConfig.INSTANCE.getEnableLegacyMinecraftCompat()) return oldModels;
+		Map<Identifier, ItemAsset> newModels = new HashMap<>(oldModels.contents());
+		var resources = resourceManager.findResources(
+			"models/item",
+			id -> (id.getNamespace().equals("firmskyblock") || id.getNamespace().equals("cittofirmgenerated"))
+				      && id.getPath().endsWith(".json"));
+		for (Map.Entry<Identifier, Resource> model : resources.entrySet()) {
+			var resource = model.getValue();
+			var itemModelId = model.getKey().withPath(it -> it.substring("models/item/".length(), it.length() - ".json".length()));
+			var genericModelId = itemModelId.withPrefixedPath("item/");
+			var itemAssetId = itemModelId.withPrefixedPath("items/");
+			// TODO: inject tint indexes based on the json data here
+			ItemModel.Unbaked unbakedModel = new BasicItemModel.Unbaked(genericModelId, List.of());
+			// TODO: add a filter using the pack.mcmeta to opt out of this behaviour
+			try (var is = resource.getInputStream()) {
+				var jsonObject = Firmament.INSTANCE.getGson().fromJson(new InputStreamReader(is), JsonObject.class);
+				unbakedModel = PredicateModel.Unbaked.fromLegacyJson(jsonObject, unbakedModel);
+				unbakedModel = HeadModelChooser.Unbaked.fromLegacyJson(jsonObject, unbakedModel);
+			} catch (Exception e) {
+				ErrorUtil.INSTANCE.softError("Could not create resource for fake model supplication: " + model.getKey(), e);
+			}
+			if (resourceManager.getResource(itemAssetId.withSuffixedPath(".json"))
+			                   .map(Resource::getPack)
+			                   .map(it -> isResourcePackNewer(resourceManager, it, resource.getPack()))
+			                   .orElse(true)) {
+				newModels.put(itemModelId, new ItemAsset(
+					unbakedModel,
+					new ItemAsset.Properties(true)
+				));
+			}
+		}
+		return new ItemAssetsLoader.Result(newModels);
+	}
+
+	private static boolean isResourcePackNewer(
+		ResourceManager manager,
+		ResourcePack null_, ResourcePack proposal) {
+		var pack = manager.streamResourcePacks()
+		                  .filter(it -> it == null_ || it == proposal)
+		                  .collect(findLast());
+		return pack.orElse(null_) != null_;
+	}
+
+	private static <T> Collector<T, ?, Optional<T>> findLast() {
+		return Collectors.reducing(Optional.empty(), Optional::of,
+		                           (left, right) -> right.isPresent() ? right : left);
+
+	}
+
+}