blob: 9d50e8acc93a13a1296211705268dca111760d4b (
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
82
83
|
/*
* Copyright (C) 2023 NotEnoughUpdates contributors
*
* This file is part of NotEnoughUpdates.
*
* NotEnoughUpdates is free software: you can redistribute it
* and/or modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation, either
* version 3 of the License, or (at your option) any later version.
*
* NotEnoughUpdates is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
*/
package io.github.moulberry.notenoughupdates.miscfeatures;
import com.google.gson.JsonObject;
import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
import io.github.moulberry.notenoughupdates.autosubscribe.NEUAutoSubscribe;
import net.minecraft.item.ItemArmor;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.JsonToNBT;
import net.minecraft.nbt.NBTException;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraftforge.event.world.WorldEvent;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
@NEUAutoSubscribe
public class DefaultArmorColour {
private static Map<String, Integer> armorColourCache = new HashMap<>();
private static Set<String> erroredItems = new HashSet<>();
public static int getDefaultArmorColour(ItemArmor item, ItemStack stack) {
JsonObject itemJson = NotEnoughUpdates.INSTANCE.manager
.createItemResolutionQuery()
.withItemStack(stack)
.resolveToItemListJson();
if (itemJson == null) return item.getColor(stack);
String internalname = itemJson.get("internalname").getAsString();
if (armorColourCache.containsKey(internalname)) return armorColourCache.get(internalname);
if (erroredItems.contains(internalname)) return item.getColor(stack);
if (itemJson.has("nbttag")) {
try {
NBTTagCompound nbt = JsonToNBT.getTagFromJson(itemJson.get("nbttag").getAsString());
NBTTagCompound display;
if (nbt.hasKey("display") && (display = nbt.getCompoundTag("display")).hasKey("color")) {
int colour = display.getInteger("color");
if (colour != 0) {
armorColourCache.put(internalname, colour);
return colour;
}
}
} catch (NBTException exception) {
erroredItems.add(internalname);
System.out.println("[NEU] Ran into NBTException whilst converting Json into NBT with the JsonObject: " + itemJson);
exception.printStackTrace();
}
}
return item.getColor(stack);
}
@SubscribeEvent
public void onWorldChange(WorldEvent.Unload event) {
armorColourCache.clear();
}
}
|
