Upstream/Minestom
1
0
Fork 0
mirror of https://github.com/Minestom/Minestom.git synced 2025-01-08 17:37:42 +01:00
Code Issues Projects Releases Wiki Activity

Cleanup ItemTag

Browse Source
This commit is contained in:
themode 2021-04-03 16:03:03 +02:00
parent c8fbf47bd9
commit 89b35a0569
1 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/main/java/net/minestom/server/item/ItemTag.java
View File

@ -3,34 +3,39 @@ package net.minestom.server.item;
import org.jetbrains.annotations.NotNull;
import org.jglrxavpok.hephaistos.nbt.NBTCompound;
public abstract class ItemTag<T> {
import java.util.function.BiConsumer;
import java.util.function.Function;
public class ItemTag<T> {
private final String key;
private final Function<NBTCompound, T> readFunction;
private final BiConsumer<NBTCompound, T> writeConsumer;
private ItemTag(@NotNull String key) {
private ItemTag(@NotNull String key,
@NotNull Function<NBTCompound, T> readFunction,
@NotNull BiConsumer<NBTCompound, T> writeConsumer) {
this.key = key;
this.readFunction = readFunction;
this.writeConsumer = writeConsumer;
}
public @NotNull String getKey() {
return key;
}
protected abstract T read(@NotNull NBTCompound nbtCompound);
protected T read(@NotNull NBTCompound nbtCompound) {
return readFunction.apply(nbtCompound);
}
protected abstract void write(@NotNull NBTCompound nbtCompound, @NotNull T value);
protected void write(@NotNull NBTCompound nbtCompound, @NotNull T value) {
this.writeConsumer.accept(nbtCompound, value);
}
public static @NotNull ItemTag<Integer> Integer(@NotNull String key) {
return new ItemTag<>(key) {
@Override
protected Integer read(@NotNull NBTCompound nbtCompound) {
return nbtCompound.getInt(key);
}
@Override
protected void write(@NotNull NBTCompound nbtCompound, @NotNull Integer value) {
nbtCompound.setInt(key, value);
}
};
return new ItemTag<>(key,
nbtCompound -> nbtCompound.getInt(key),
(nbtCompound, integer) -> nbtCompound.setInt(key, integer));
}
}