Files
chomens-bot-java/src/main/java/me/chayapak1/chomens_bot/plugins/PositionPlugin.java
ChomeNS 0a8efe0189 refactor: make listeners an interface
(except the one in Bot, which extends MCProtocolLib's SessionAdaptor, not sure how exactly make them complement)
2025-03-23 14:18:55 +07:00

274 lines
10 KiB
Java

package me.chayapak1.chomens_bot.plugins;
import me.chayapak1.chomens_bot.Bot;
import me.chayapak1.chomens_bot.data.entity.Rotation;
import me.chayapak1.chomens_bot.data.player.PlayerEntry;
import me.chayapak1.chomens_bot.util.MathUtilities;
import org.cloudburstmc.math.vector.Vector3d;
import org.geysermc.mcprotocollib.network.Session;
import org.geysermc.mcprotocollib.network.packet.Packet;
import org.geysermc.mcprotocollib.protocol.data.game.entity.type.EntityType;
import org.geysermc.mcprotocollib.protocol.packet.ingame.clientbound.entity.*;
import org.geysermc.mcprotocollib.protocol.packet.ingame.clientbound.entity.player.ClientboundPlayerPositionPacket;
import org.geysermc.mcprotocollib.protocol.packet.ingame.clientbound.entity.spawn.ClientboundAddEntityPacket;
import org.geysermc.mcprotocollib.protocol.packet.ingame.serverbound.level.ServerboundAcceptTeleportationPacket;
import org.geysermc.mcprotocollib.protocol.packet.ingame.serverbound.player.ServerboundMovePlayerPosPacket;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
// some part of the code used to be in a test plugin but i thought it would be useful in the future so i moved it here
public class PositionPlugin extends Bot.Listener implements TickPlugin.Listener {
private final Bot bot;
private final List<Listener> listeners = new ArrayList<>();
public Vector3d position = Vector3d.from(0, 0, 0);
public boolean isGoingDownFromHeightLimit = false; // cool variable name
private final Map<Integer, PlayerEntry> entityIdMap = new HashMap<>();
private final Map<Integer, Vector3d> positionMap = new HashMap<>();
private final Map<Integer, Rotation> rotationMap = new HashMap<>();
public PositionPlugin (Bot bot) {
this.bot = bot;
// notchian clients also does this, sends the position packet every second
bot.executor.scheduleAtFixedRate(() -> {
if (!bot.loggedIn || isGoingDownFromHeightLimit) return;
bot.session.send(new ServerboundMovePlayerPosPacket(
false,
false,
position.getX(),
position.getY(),
position.getZ()
));
}, 0, 1, TimeUnit.SECONDS);
bot.addListener(this);
bot.tick.addListener(this);
}
@Override
public void onTick() {
handleHeightLimit();
}
@Override
public void packetReceived (Session session, Packet packet) {
if (packet instanceof ClientboundPlayerPositionPacket t_packet) packetReceived(t_packet);
else if (packet instanceof ClientboundAddEntityPacket t_packet) packetReceived(t_packet);
else if (packet instanceof ClientboundRemoveEntitiesPacket t_packet) packetReceived(t_packet);
else if (packet instanceof ClientboundMoveEntityRotPacket t_packet) packetReceived(t_packet);
else if (packet instanceof ClientboundMoveEntityPosPacket t_packet) packetReceived(t_packet);
else if (packet instanceof ClientboundMoveEntityPosRotPacket t_packet) packetReceived(t_packet);
else if (packet instanceof ClientboundEntityPositionSyncPacket t_packet) packetReceived(t_packet);
}
public void packetReceived (ClientboundPlayerPositionPacket packet) {
if (bot.session != null) bot.session.send(new ServerboundAcceptTeleportationPacket(packet.getId()));
position = packet.getPosition();
for (Listener listener : listeners) { listener.positionChange(position); }
}
public void packetReceived (ClientboundAddEntityPacket packet) {
if (packet.getType() != EntityType.PLAYER) return;
final PlayerEntry entry = bot.players.getEntry(packet.getUuid());
if (entry == null) return;
entityIdMap.remove(packet.getEntityId());
positionMap.remove(packet.getEntityId());
rotationMap.remove(packet.getEntityId());
entityIdMap.put(packet.getEntityId(), entry);
positionMap.put(packet.getEntityId(), Vector3d.from(packet.getX(), packet.getY(), packet.getZ()));
rotationMap.put(packet.getEntityId(), new Rotation(packet.getYaw(), packet.getPitch()));
}
public void packetReceived (ClientboundRemoveEntitiesPacket packet) {
final int[] ids = packet.getEntityIds();
for (int id : ids) {
entityIdMap.remove(id);
positionMap.remove(id);
rotationMap.remove(id);
}
}
public void packetReceived (ClientboundEntityPositionSyncPacket packet) {
final PlayerEntry player = entityIdMap.get(packet.getId());
if (player == null) return;
positionMap.remove(packet.getId());
rotationMap.remove(packet.getId());
final Vector3d position = packet.getPosition().add(packet.getDeltaMovement());
final Rotation rotation = new Rotation(packet.getXRot(), packet.getYRot());
positionMap.put(packet.getId(), position);
rotationMap.put(packet.getId(), rotation);
for (Listener listener : listeners) listener.playerMoved(player, position, rotation);
}
public void packetReceived (ClientboundMoveEntityRotPacket packet) {
final PlayerEntry player = entityIdMap.get(packet.getEntityId());
if (player == null) return;
final Rotation rotation = new Rotation(packet.getYaw(), packet.getPitch());
rotationMap.put(packet.getEntityId(), rotation);
for (Listener listener : listeners) listener.playerMoved(player, getPlayerPosition(player.profile.getName()), rotation);
}
public void packetReceived (ClientboundMoveEntityPosPacket packet) {
final PlayerEntry player = entityIdMap.get(packet.getEntityId());
if (player == null) return;
final Vector3d lastPosition = positionMap.get(packet.getEntityId());
positionMap.remove(packet.getEntityId());
// code ported straight from minecraft
Vector3d position = lastPosition;
if (packet.getMoveX() != 0 || packet.getMoveY() != 0 || packet.getMoveZ() != 0) {
position = position.add(
packet.getMoveX() == 0 ? 0 : packet.getMoveX(),
packet.getMoveY() == 0 ? 0 : packet.getMoveY(),
packet.getMoveZ() == 0 ? 0 : packet.getMoveZ()
);
}
positionMap.put(packet.getEntityId(), position);
for (Listener listener : listeners) listener.playerMoved(player, position, getPlayerRotation(player.profile.getName()));
}
public void packetReceived (ClientboundMoveEntityPosRotPacket packet) {
final PlayerEntry player = entityIdMap.get(packet.getEntityId());
if (player == null) return;
final Vector3d lastPosition = positionMap.get(packet.getEntityId());
positionMap.remove(packet.getEntityId());
rotationMap.remove(packet.getEntityId());
Vector3d position = lastPosition;
if (packet.getMoveX() != 0 || packet.getMoveY() != 0 || packet.getMoveZ() != 0) {
position = position.add(
packet.getMoveX() == 0 ? 0 : packet.getMoveX(),
packet.getMoveY() == 0 ? 0 : packet.getMoveY(),
packet.getMoveZ() == 0 ? 0 : packet.getMoveZ()
);
}
final Rotation rotation = new Rotation(packet.getYaw(), packet.getPitch());
positionMap.put(packet.getEntityId(), position);
rotationMap.put(packet.getEntityId(), rotation);
for (Listener listener : listeners) listener.playerMoved(player, position, rotation);
}
// for now this is used in CorePlugin when placing the command block
private void handleHeightLimit () {
final double y = position.getY();
final int minY = bot.world.minY;
final int maxY = bot.world.maxY;
if (y <= maxY && y >= minY) {
if (isGoingDownFromHeightLimit) {
isGoingDownFromHeightLimit = false;
for (Listener listener : listeners) { listener.positionChange(position); }
}
return;
}
isGoingDownFromHeightLimit = true;
if (y > maxY + 500 || y < minY) {
String command = "/";
if (bot.serverFeatures.hasEssentials) command += "essentials:";
command += String.format("tp ~ %s ~", maxY);
bot.chat.send(command);
return;
}
final Vector3d newPosition = Vector3d.from(
position.getX(),
MathUtilities.clamp(position.getY() - 2, maxY, position.getY()),
position.getZ()
);
position = newPosition;
bot.session.send(new ServerboundMovePlayerPosPacket(
false,
false,
newPosition.getX(),
newPosition.getY(),
newPosition.getZ()
));
}
public Vector3d getPlayerPosition (String playerName) {
int entityId = -1;
for (Map.Entry<Integer, PlayerEntry> entry : entityIdMap.entrySet()) {
if (entry.getValue().profile.getName().equals(playerName)) entityId = entry.getKey();
}
if (entityId == -1) return null;
for (Map.Entry<Integer, Vector3d> entry : positionMap.entrySet()) {
if (entry.getKey() == entityId) return entry.getValue();
}
return null;
}
public Rotation getPlayerRotation (String playerName) {
int entityId = -1;
for (Map.Entry<Integer, PlayerEntry> entry : entityIdMap.entrySet()) {
if (entry.getValue().profile.getName().equals(playerName)) entityId = entry.getKey();
}
if (entityId == -1) return null;
for (Map.Entry<Integer, Rotation> entry : rotationMap.entrySet()) {
if (entry.getKey() == entityId) return entry.getValue();
}
return null;
}
public void addListener (Listener listener) { listeners.add(listener); }
@SuppressWarnings("unused")
public interface Listener {
default void positionChange (Vector3d position) {}
default void playerMoved (PlayerEntry player, Vector3d position, Rotation rotation) {}
}
}