48 lines
1.5 KiB
Java
48 lines
1.5 KiB
Java
package net.minecraft.client;
|
|
|
|
import java.util.function.Consumer;
|
|
import net.fabricmc.api.EnvType;
|
|
import net.fabricmc.api.Environment;
|
|
import net.minecraft.client.multiplayer.ClientPacketListener;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.nbt.CompoundTag;
|
|
import net.minecraft.network.protocol.game.ServerboundBlockEntityTagQueryPacket;
|
|
import net.minecraft.network.protocol.game.ServerboundEntityTagQueryPacket;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
@Environment(EnvType.CLIENT)
|
|
public class DebugQueryHandler {
|
|
private final ClientPacketListener connection;
|
|
private int transactionId = -1;
|
|
@Nullable
|
|
private Consumer<CompoundTag> callback;
|
|
|
|
public DebugQueryHandler(ClientPacketListener connection) {
|
|
this.connection = connection;
|
|
}
|
|
|
|
public boolean handleResponse(int transactionId, @Nullable CompoundTag tag) {
|
|
if (this.transactionId == transactionId && this.callback != null) {
|
|
this.callback.accept(tag);
|
|
this.callback = null;
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private int startTransaction(Consumer<CompoundTag> callback) {
|
|
this.callback = callback;
|
|
return ++this.transactionId;
|
|
}
|
|
|
|
public void queryEntityTag(int entId, Consumer<CompoundTag> tag) {
|
|
int i = this.startTransaction(tag);
|
|
this.connection.send(new ServerboundEntityTagQueryPacket(i, entId));
|
|
}
|
|
|
|
public void queryBlockEntityTag(BlockPos pos, Consumer<CompoundTag> tag) {
|
|
int i = this.startTransaction(tag);
|
|
this.connection.send(new ServerboundBlockEntityTagQueryPacket(i, pos));
|
|
}
|
|
}
|