2023-03-14 20:37:05 +01:00
|
|
|
package net.citizensnpcs.nms.v1_19_R3.util;
|
2022-06-08 05:58:22 +02:00
|
|
|
|
|
|
|
import java.lang.invoke.MethodHandle;
|
2023-03-07 16:58:03 +01:00
|
|
|
import java.util.Set;
|
2023-02-22 17:06:13 +01:00
|
|
|
|
2022-06-08 05:58:22 +02:00
|
|
|
import org.bukkit.Bukkit;
|
2023-03-07 16:58:03 +01:00
|
|
|
import org.bukkit.entity.EntityType;
|
2022-06-08 05:58:22 +02:00
|
|
|
import org.bukkit.entity.Player;
|
2023-02-22 17:06:13 +01:00
|
|
|
|
2023-03-07 16:58:03 +01:00
|
|
|
import com.google.common.collect.ForwardingSet;
|
|
|
|
|
2022-12-17 04:22:14 +01:00
|
|
|
import net.citizensnpcs.Settings.Setting;
|
2022-06-08 05:58:22 +02:00
|
|
|
import net.citizensnpcs.api.CitizensAPI;
|
2022-12-18 17:36:53 +01:00
|
|
|
import net.citizensnpcs.api.npc.NPC;
|
2023-03-14 20:37:05 +01:00
|
|
|
import net.citizensnpcs.nms.v1_19_R3.entity.EntityHumanNPC;
|
2022-12-18 17:36:53 +01:00
|
|
|
import net.citizensnpcs.npc.ai.NPCHolder;
|
2022-06-08 05:58:22 +02:00
|
|
|
import net.citizensnpcs.util.NMS;
|
|
|
|
import net.minecraft.network.protocol.game.ClientboundAnimatePacket;
|
|
|
|
import net.minecraft.server.level.ChunkMap;
|
|
|
|
import net.minecraft.server.level.ChunkMap.TrackedEntity;
|
|
|
|
import net.minecraft.server.level.ServerEntity;
|
|
|
|
import net.minecraft.server.level.ServerPlayer;
|
2023-03-27 16:26:20 +02:00
|
|
|
import net.minecraft.server.network.ServerPlayerConnection;
|
2022-06-08 05:58:22 +02:00
|
|
|
import net.minecraft.world.entity.Entity;
|
|
|
|
|
2023-03-07 16:58:03 +01:00
|
|
|
public class CitizensEntityTracker extends ChunkMap.TrackedEntity {
|
2022-06-08 05:58:22 +02:00
|
|
|
private final Entity tracker;
|
|
|
|
|
2023-03-07 16:58:03 +01:00
|
|
|
public CitizensEntityTracker(ChunkMap map, Entity entity, int i, int j, boolean flag) {
|
2022-06-08 05:58:22 +02:00
|
|
|
map.super(entity, i, j, flag);
|
|
|
|
this.tracker = entity;
|
2023-03-07 16:58:03 +01:00
|
|
|
try {
|
2023-03-27 16:26:20 +02:00
|
|
|
Set<ServerPlayerConnection> set = (Set<ServerPlayerConnection>) TRACKING_SET_GETTER.invoke(this);
|
|
|
|
TRACKING_SET_SETTER.invoke(this, new ForwardingSet<ServerPlayerConnection>() {
|
2023-03-07 16:58:03 +01:00
|
|
|
@Override
|
2023-03-27 16:26:20 +02:00
|
|
|
public boolean add(ServerPlayerConnection conn) {
|
2023-03-07 16:58:03 +01:00
|
|
|
boolean res = super.add(conn);
|
|
|
|
if (res) {
|
2023-03-27 16:26:20 +02:00
|
|
|
updateLastPlayer(conn.getPlayer());
|
2023-03-07 16:58:03 +01:00
|
|
|
}
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected Set delegate() {
|
|
|
|
return set;
|
|
|
|
}
|
|
|
|
});
|
|
|
|
} catch (Throwable e) {
|
|
|
|
e.printStackTrace();
|
|
|
|
}
|
2022-06-08 05:58:22 +02:00
|
|
|
}
|
|
|
|
|
2023-03-07 16:58:03 +01:00
|
|
|
public CitizensEntityTracker(ChunkMap map, TrackedEntity entry) {
|
2022-06-08 05:58:22 +02:00
|
|
|
this(map, getTracker(entry), getTrackingDistance(entry), getE(entry), getF(entry));
|
|
|
|
}
|
|
|
|
|
2023-03-27 16:26:20 +02:00
|
|
|
public void updateLastPlayer(ServerPlayer lastUpdatedPlayer) {
|
|
|
|
if (tracker.isRemoved() || tracker.getBukkitEntity().getType() != EntityType.PLAYER)
|
2022-06-08 05:58:22 +02:00
|
|
|
return;
|
|
|
|
final ServerPlayer entityplayer = lastUpdatedPlayer;
|
2022-12-18 16:07:04 +01:00
|
|
|
boolean sendTabRemove = NMS.sendTabListAdd(entityplayer.getBukkitEntity(), (Player) tracker.getBukkitEntity());
|
2022-12-19 15:08:58 +01:00
|
|
|
if (!sendTabRemove || !Setting.DISABLE_TABLIST.asBoolean()) {
|
|
|
|
Bukkit.getScheduler().scheduleSyncDelayedTask(CitizensAPI.getPlugin(),
|
|
|
|
() -> NMSImpl.sendPacket(entityplayer.getBukkitEntity(), new ClientboundAnimatePacket(tracker, 0)),
|
|
|
|
1);
|
2022-12-17 04:22:14 +01:00
|
|
|
return;
|
2022-12-19 15:08:58 +01:00
|
|
|
}
|
|
|
|
Bukkit.getScheduler().scheduleSyncDelayedTask(CitizensAPI.getPlugin(), () -> {
|
|
|
|
NMS.sendTabListRemove(entityplayer.getBukkitEntity(), (Player) tracker.getBukkitEntity());
|
|
|
|
NMSImpl.sendPacket(entityplayer.getBukkitEntity(), new ClientboundAnimatePacket(tracker, 0));
|
2023-03-12 15:29:41 +01:00
|
|
|
}, Setting.TABLIST_REMOVE_PACKET_DELAY.asTicks());
|
2022-06-08 05:58:22 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public void updatePlayer(final ServerPlayer entityplayer) {
|
2023-03-26 20:52:56 +02:00
|
|
|
if (tracker instanceof NPCHolder) {
|
|
|
|
NPC npc = ((NPCHolder) tracker).getNPC();
|
|
|
|
if (npc.isHiddenFrom(entityplayer.getBukkitEntity()))
|
|
|
|
return;
|
|
|
|
Integer trackingRange = npc.data().<Integer> get(NPC.Metadata.TRACKING_RANGE);
|
|
|
|
if (TRACKING_RANGE_SETTER != null && trackingRange != null
|
|
|
|
&& npc.data().get("last-tracking-range", -1) != trackingRange.intValue()) {
|
|
|
|
try {
|
|
|
|
TRACKING_RANGE_SETTER.invoke(this, trackingRange);
|
|
|
|
npc.data().set("last-tracking-range", trackingRange);
|
|
|
|
} catch (Throwable e) {
|
|
|
|
e.printStackTrace();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (entityplayer instanceof EntityHumanNPC)
|
2023-03-07 16:58:03 +01:00
|
|
|
return;
|
|
|
|
|
2022-06-08 05:58:22 +02:00
|
|
|
super.updatePlayer(entityplayer);
|
|
|
|
}
|
|
|
|
|
|
|
|
private static int getE(TrackedEntity entry) {
|
|
|
|
try {
|
|
|
|
return (int) E.invoke(TRACKER_ENTRY.invoke(entry));
|
|
|
|
} catch (Throwable e) {
|
|
|
|
e.printStackTrace();
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
private static boolean getF(TrackedEntity entry) {
|
|
|
|
try {
|
|
|
|
return (boolean) F.invoke(TRACKER_ENTRY.invoke(entry));
|
|
|
|
} catch (Throwable e) {
|
|
|
|
e.printStackTrace();
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
private static Entity getTracker(TrackedEntity entry) {
|
|
|
|
try {
|
|
|
|
return (Entity) TRACKER.invoke(entry);
|
|
|
|
} catch (Throwable e) {
|
|
|
|
e.printStackTrace();
|
|
|
|
}
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
private static int getTrackingDistance(TrackedEntity entry) {
|
|
|
|
try {
|
2022-12-18 17:36:53 +01:00
|
|
|
return (Integer) TRACKING_RANGE.invoke(entry);
|
2022-06-08 05:58:22 +02:00
|
|
|
} catch (Throwable e) {
|
|
|
|
e.printStackTrace();
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
private static final MethodHandle E = NMS.getGetter(ServerEntity.class, "e");
|
|
|
|
private static final MethodHandle F = NMS.getGetter(ServerEntity.class, "f");
|
|
|
|
private static final MethodHandle TRACKER = NMS.getFirstGetter(TrackedEntity.class, Entity.class);
|
|
|
|
private static final MethodHandle TRACKER_ENTRY = NMS.getFirstGetter(TrackedEntity.class, ServerEntity.class);
|
2022-12-18 17:36:53 +01:00
|
|
|
private static final MethodHandle TRACKING_RANGE = NMS.getFirstGetter(TrackedEntity.class, int.class);
|
2023-03-26 20:52:56 +02:00
|
|
|
private static final MethodHandle TRACKING_RANGE_SETTER = NMS.getFirstFinalSetter(TrackedEntity.class, int.class);
|
2023-03-07 16:58:03 +01:00
|
|
|
private static final MethodHandle TRACKING_SET_GETTER = NMS.getFirstGetter(TrackedEntity.class, Set.class);
|
|
|
|
private static final MethodHandle TRACKING_SET_SETTER = NMS.getFirstFinalSetter(TrackedEntity.class, Set.class);
|
2022-06-08 05:58:22 +02:00
|
|
|
}
|