222 lines
8.6 KiB
Java
222 lines
8.6 KiB
Java
package net.citizensnpcs.nms.v1_12_R1.util;
|
|
|
|
import java.lang.invoke.MethodHandle;
|
|
import java.lang.reflect.Field;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.entity.EntityType;
|
|
import org.bukkit.entity.Player;
|
|
|
|
import com.google.common.collect.ForwardingMap;
|
|
import com.google.common.collect.ForwardingSet;
|
|
|
|
import net.citizensnpcs.Settings.Setting;
|
|
import net.citizensnpcs.api.CitizensAPI;
|
|
import net.citizensnpcs.api.event.NPCLinkToPlayerEvent;
|
|
import net.citizensnpcs.api.event.NPCSeenByPlayerEvent;
|
|
import net.citizensnpcs.api.npc.NPC;
|
|
import net.citizensnpcs.nms.v1_12_R1.entity.EntityHumanNPC;
|
|
import net.citizensnpcs.npc.ai.NPCHolder;
|
|
import net.citizensnpcs.util.NMS;
|
|
import net.minecraft.server.v1_12_R1.Entity;
|
|
import net.minecraft.server.v1_12_R1.EntityPlayer;
|
|
import net.minecraft.server.v1_12_R1.EntityTrackerEntry;
|
|
import net.minecraft.server.v1_12_R1.PacketPlayOutAnimation;
|
|
|
|
public class PlayerlistTrackerEntry extends EntityTrackerEntry {
|
|
private final Entity tracker;
|
|
|
|
private Map<EntityPlayer, Boolean> trackingMap;
|
|
|
|
public PlayerlistTrackerEntry(Entity entity, int i, int j, int k, boolean flag) {
|
|
super(entity, i, j, k, flag);
|
|
this.tracker = getTracker(this);
|
|
if (TRACKING_MAP_SETTER != null) {
|
|
try {
|
|
Map<EntityPlayer, Boolean> delegate = (Map<EntityPlayer, Boolean>) TRACKING_MAP_GETTER.invoke(this);
|
|
trackingMap = delegate;
|
|
TRACKING_MAP_SETTER.invoke(this, new ForwardingMap<EntityPlayer, Boolean>() {
|
|
@Override
|
|
protected Map<EntityPlayer, Boolean> delegate() {
|
|
return delegate;
|
|
}
|
|
|
|
@Override
|
|
public Boolean put(EntityPlayer player, Boolean value) {
|
|
Boolean res = super.put(player, value);
|
|
if (res == null) {
|
|
updateLastPlayer(player);
|
|
}
|
|
return res;
|
|
}
|
|
});
|
|
} catch (Throwable e) {
|
|
e.printStackTrace();
|
|
}
|
|
} else {
|
|
try {
|
|
Set<EntityPlayer> delegate = super.trackedPlayers;
|
|
TRACKING_SET_SETTER.invoke(this, new ForwardingSet<EntityPlayer>() {
|
|
@Override
|
|
public boolean add(EntityPlayer player) {
|
|
boolean res = super.add(player);
|
|
if (res) {
|
|
updateLastPlayer(player);
|
|
}
|
|
return res;
|
|
}
|
|
|
|
@Override
|
|
protected Set<EntityPlayer> delegate() {
|
|
return delegate;
|
|
}
|
|
});
|
|
} catch (Throwable e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
|
|
public PlayerlistTrackerEntry(EntityTrackerEntry entry) {
|
|
this(getTracker(entry), getE(entry), getF(entry), getG(entry), getU(entry));
|
|
}
|
|
|
|
private boolean isTracked(EntityPlayer player) {
|
|
return trackingMap != null ? trackingMap.containsKey(player) : trackedPlayers.contains(player);
|
|
}
|
|
|
|
public void updateLastPlayer(EntityPlayer lastUpdatedPlayer) {
|
|
final EntityPlayer entityplayer = lastUpdatedPlayer;
|
|
if (lastUpdatedPlayer != null) {
|
|
Bukkit.getScheduler().scheduleSyncDelayedTask(CitizensAPI.getPlugin(),
|
|
() -> Bukkit.getPluginManager().callEvent(
|
|
new NPCLinkToPlayerEvent(((NPCHolder) tracker).getNPC(), entityplayer.getBukkitEntity())));
|
|
}
|
|
|
|
if (lastUpdatedPlayer == null || tracker.dead || tracker.getBukkitEntity().getType() != EntityType.PLAYER)
|
|
return;
|
|
NMS.sendTabListAdd(entityplayer.getBukkitEntity(), (Player) tracker.getBukkitEntity());
|
|
lastUpdatedPlayer = null;
|
|
NPC npc = ((NPCHolder) tracker).getNPC();
|
|
if (npc.data().get(NPC.Metadata.RESET_YAW_ON_SPAWN, Setting.RESET_YAW_ON_SPAWN.asBoolean())) {
|
|
Bukkit.getScheduler().scheduleSyncDelayedTask(CitizensAPI.getPlugin(),
|
|
() -> NMSImpl.sendPacket(entityplayer.getBukkitEntity(), new PacketPlayOutAnimation(tracker, 0)),
|
|
1);
|
|
}
|
|
|
|
if (!Setting.DISABLE_TABLIST.asBoolean())
|
|
return;
|
|
|
|
Bukkit.getScheduler().scheduleSyncDelayedTask(CitizensAPI.getPlugin(),
|
|
() -> NMS.sendTabListRemove(entityplayer.getBukkitEntity(), (Player) tracker.getBukkitEntity()),
|
|
Setting.TABLIST_REMOVE_PACKET_DELAY.asTicks());
|
|
}
|
|
|
|
@Override
|
|
public void updatePlayer(final EntityPlayer entityplayer) {
|
|
if (entityplayer instanceof EntityHumanNPC)
|
|
return;
|
|
|
|
if (!isTracked(entityplayer) && tracker instanceof NPCHolder) {
|
|
NPC npc = ((NPCHolder) tracker).getNPC();
|
|
NPCSeenByPlayerEvent event = new NPCSeenByPlayerEvent(npc, entityplayer.getBukkitEntity());
|
|
Bukkit.getPluginManager().callEvent(event);
|
|
if (event.isCancelled())
|
|
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();
|
|
}
|
|
}
|
|
}
|
|
|
|
super.updatePlayer(entityplayer);
|
|
}
|
|
|
|
private static int getE(EntityTrackerEntry entry) {
|
|
try {
|
|
return (Integer) E.get(entry);
|
|
} catch (IllegalArgumentException e) {
|
|
e.printStackTrace();
|
|
} catch (IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private static int getF(EntityTrackerEntry entry) {
|
|
try {
|
|
return (Integer) F.get(entry);
|
|
} catch (IllegalArgumentException e) {
|
|
e.printStackTrace();
|
|
} catch (IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private static int getG(EntityTrackerEntry entry) {
|
|
try {
|
|
return (Integer) G.get(entry);
|
|
} catch (IllegalArgumentException e) {
|
|
e.printStackTrace();
|
|
} catch (IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private static Entity getTracker(EntityTrackerEntry entry) {
|
|
try {
|
|
return (Entity) TRACKER.get(entry);
|
|
} catch (IllegalArgumentException e) {
|
|
e.printStackTrace();
|
|
} catch (IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private static boolean getU(EntityTrackerEntry entry) {
|
|
try {
|
|
return (Boolean) U.get(entry);
|
|
} catch (IllegalArgumentException e) {
|
|
e.printStackTrace();
|
|
} catch (IllegalAccessException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private static Field E = NMS.getField(EntityTrackerEntry.class, "e");
|
|
private static Field F = NMS.getField(EntityTrackerEntry.class, "f");
|
|
private static Field G = NMS.getField(EntityTrackerEntry.class, "g");
|
|
private static Field TRACKER = NMS.getField(EntityTrackerEntry.class, "tracker");
|
|
private static MethodHandle TRACKING_MAP_GETTER;
|
|
private static MethodHandle TRACKING_MAP_SETTER;
|
|
private static final MethodHandle TRACKING_RANGE_SETTER = NMS.getFirstFinalSetter(EntityTrackerEntry.class,
|
|
int.class);
|
|
|
|
private static final MethodHandle TRACKING_SET_SETTER = NMS.getFirstFinalSetter(EntityTrackerEntry.class,
|
|
Set.class);
|
|
private static Field U = NMS.getField(EntityTrackerEntry.class, "u");
|
|
static {
|
|
try {
|
|
// Old paper versions override the tracked player set to be a map
|
|
if (EntityTrackerEntry.class.getField("trackedPlayerMap") != null) {
|
|
TRACKING_MAP_SETTER = NMS.getFirstSetter(EntityTrackerEntry.class, Map.class);
|
|
TRACKING_MAP_GETTER = NMS.getFirstGetter(EntityTrackerEntry.class, Map.class);
|
|
}
|
|
} catch (Exception e) {
|
|
}
|
|
}
|
|
}
|