aboutsummaryrefslogtreecommitdiff
path: root/src/texturePacks/java/moe/nea/firmament/mixins/custommodels/SupplyFakeModelPatch.java
blob: ddfb9adde10f36f5396bc6e6e7643697bb23c279 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package moe.nea.firmament.mixins.custommodels;

import com.llamalad7.mixinextras.injector.ModifyReturnValue;
import com.llamalad7.mixinextras.sugar.Local;
import net.minecraft.client.item.ItemAsset;
import net.minecraft.client.item.ItemAssetsLoader;
import net.minecraft.client.render.item.model.BasicItemModel;
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.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) {
		Map<Identifier, ItemAsset> newModels = new HashMap<>(oldModels.contents());
		var resources = resourceManager.findResources(
			"models/item",
			id -> id.getNamespace().equals("firmskyblock")
				      && id.getPath().endsWith(".json")
				      && !id.getPath().substring("models/item/".length()).contains("/"));
		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()));
			// TODO: parse json file here and make use of it in order to generate predicate files.
			// TODO: add a filter using the pack.mcmeta to opt out of this behaviour
			var genericModelId = itemModelId.withPrefixedPath("item/");
			if (resourceManager.getResource(itemModelId)
			                   .map(Resource::getPack)
			                   .map(it -> isResourcePackNewer(resourceManager, it, resource.getPack()))
			                   .orElse(true)) {
				newModels.put(itemModelId, new ItemAsset(
					// TODO: inject tint indexes based on the json data here
					new BasicItemModel.Unbaked(genericModelId, List.of()),
					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) == proposal;
	}

	private static <T> Collector<T, ?, Optional<T>> findLast() {
		return Collectors.reducing(Optional.empty(), Optional::of,
		                           (left, right) -> right.isPresent() ? right : left);

	}

}