aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/main/java/at/hannibal2/skyhanni/utils/ItemResolutionQuery.java5
-rw-r--r--src/main/java/at/hannibal2/skyhanni/utils/IteratorUtils.kt14
2 files changed, 16 insertions, 3 deletions
diff --git a/src/main/java/at/hannibal2/skyhanni/utils/ItemResolutionQuery.java b/src/main/java/at/hannibal2/skyhanni/utils/ItemResolutionQuery.java
index 1f8988613..548db2f08 100644
--- a/src/main/java/at/hannibal2/skyhanni/utils/ItemResolutionQuery.java
+++ b/src/main/java/at/hannibal2/skyhanni/utils/ItemResolutionQuery.java
@@ -1,7 +1,6 @@
package at.hannibal2.skyhanni.utils;
import at.hannibal2.skyhanni.config.ConfigManager;
-import com.google.common.collect.Iterables;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;
import net.minecraft.client.Minecraft;
@@ -155,14 +154,14 @@ public class ItemResolutionQuery {
private String resolveEnchantedBookNameFromNBT() {
NBTTagCompound enchantments = getExtraAttributes().getCompoundTag("enchantments");
- String enchantName = Iterables.getOnlyElement(enchantments.getKeySet(), null);
+ String enchantName = IteratorUtils.INSTANCE.getOnlyElement(enchantments.getKeySet(), null);
if (enchantName == null || enchantName.isEmpty()) return null;
return enchantName.toUpperCase(Locale.ROOT) + ";" + enchantments.getInteger(enchantName);
}
private String resolveRuneName() {
NBTTagCompound runes = getExtraAttributes().getCompoundTag("runes");
- String runeName = Iterables.getOnlyElement(runes.getKeySet(), null);
+ String runeName = IteratorUtils.INSTANCE.getOnlyElement(runes.getKeySet(), null);
if (runeName == null || runeName.isEmpty()) return null;
return runeName.toUpperCase(Locale.ROOT) + "_RUNE;" + runes.getInteger(runeName);
}
diff --git a/src/main/java/at/hannibal2/skyhanni/utils/IteratorUtils.kt b/src/main/java/at/hannibal2/skyhanni/utils/IteratorUtils.kt
new file mode 100644
index 000000000..ca55a665f
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/utils/IteratorUtils.kt
@@ -0,0 +1,14 @@
+package at.hannibal2.skyhanni.utils
+
+object IteratorUtils {
+
+ fun <T> getOnlyElement(it: Iterator<T>, defaultValue: T): T {
+ if (!it.hasNext()) return defaultValue
+ val ret = it.next()
+ return if (it.hasNext()) defaultValue else ret
+ }
+
+ fun <T> getOnlyElement(it: Iterable<T>, defaultValue: T): T {
+ return getOnlyElement(it.iterator(), defaultValue)
+ }
+} \ No newline at end of file