|
| 1 | +// |
| 2 | +// Hyperverse - A minecraft world management plugin |
| 3 | +// |
| 4 | +// This program is free software: you can redistribute it and/or modify |
| 5 | +// it under the terms of the GNU General Public License as published by |
| 6 | +// the Free Software Foundation, either version 3 of the License, or |
| 7 | +// (at your option) any later version. |
| 8 | +// |
| 9 | +// This program is distributed in the hope that it will be useful, |
| 10 | +// but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 11 | +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 | +// GNU General Public License for more details. |
| 13 | +// |
| 14 | +// You should have received a copy of the GNU General Public License |
| 15 | +// along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 16 | +// |
| 17 | + |
| 18 | +package org.incendo.hyperverse.platform.v1_21_5; |
| 19 | + |
| 20 | +import co.aikar.taskchain.TaskChainFactory; |
| 21 | +import com.google.inject.Inject; |
| 22 | +import io.papermc.lib.PaperLib; |
| 23 | +import java.io.InputStream; |
| 24 | +import java.io.OutputStream; |
| 25 | +import java.lang.reflect.Field; |
| 26 | +import java.nio.file.Files; |
| 27 | +import java.nio.file.Path; |
| 28 | +import java.util.Objects; |
| 29 | +import java.util.Optional; |
| 30 | + |
| 31 | +import net.minecraft.BlockUtil; |
| 32 | +import net.minecraft.core.BlockPos; |
| 33 | +import net.minecraft.nbt.CompoundTag; |
| 34 | +import net.minecraft.nbt.DoubleTag; |
| 35 | +import net.minecraft.nbt.IntTag; |
| 36 | +import net.minecraft.nbt.ListTag; |
| 37 | +import net.minecraft.nbt.NbtAccounter; |
| 38 | +import net.minecraft.nbt.NbtIo; |
| 39 | +import net.minecraft.nbt.NumericTag; |
| 40 | +import net.minecraft.nbt.StringTag; |
| 41 | +import net.minecraft.resources.ResourceKey; |
| 42 | +import net.minecraft.server.level.ServerLevel; |
| 43 | +import net.minecraft.server.level.ServerPlayer; |
| 44 | +import net.minecraft.world.entity.Entity; |
| 45 | +import net.minecraft.world.level.Level; |
| 46 | +import net.minecraft.world.level.border.WorldBorder; |
| 47 | +import net.minecraft.world.level.dimension.DimensionType; |
| 48 | +import net.minecraft.world.level.entity.EntityLookup; |
| 49 | +import net.minecraft.world.level.entity.PersistentEntitySectionManager; |
| 50 | +import net.minecraft.world.level.portal.PortalForcer; |
| 51 | +import net.minecraft.world.phys.Vec3; |
| 52 | +import org.apache.logging.log4j.core.Filter; |
| 53 | +import org.apache.logging.log4j.core.Logger; |
| 54 | +import org.apache.logging.log4j.core.filter.RegexFilter; |
| 55 | +import org.bukkit.Bukkit; |
| 56 | +import org.bukkit.Location; |
| 57 | +import org.bukkit.World; |
| 58 | +import org.bukkit.craftbukkit.CraftWorld; |
| 59 | +import org.bukkit.craftbukkit.entity.CraftEntity; |
| 60 | +import org.bukkit.craftbukkit.entity.CraftPlayer; |
| 61 | +import org.bukkit.entity.Player; |
| 62 | +import org.bukkit.event.player.PlayerRespawnEvent; |
| 63 | +import org.jetbrains.annotations.NotNull; |
| 64 | +import org.jetbrains.annotations.Nullable; |
| 65 | +import org.incendo.hyperverse.util.HyperConfigShouldGroupProfiles; |
| 66 | +import org.incendo.hyperverse.util.NMS; |
| 67 | + |
| 68 | +@SuppressWarnings("unused") |
| 69 | +public class NMSImpl implements NMS { |
| 70 | + |
| 71 | + private final TaskChainFactory taskFactory; |
| 72 | + private Field entitySectionManager; |
| 73 | + private Field entityLookup; |
| 74 | + private org.apache.logging.log4j.core.Logger worldServerLogger; |
| 75 | + |
| 76 | + @Inject public NMSImpl(final TaskChainFactory taskFactory, final @HyperConfigShouldGroupProfiles boolean hyperConfiguration) { |
| 77 | + this.taskFactory = taskFactory; |
| 78 | + if (hyperConfiguration) { |
| 79 | + try { |
| 80 | + final Field field = ServerLevel.class.getDeclaredField("LOGGER"); |
| 81 | + field.setAccessible(true); |
| 82 | + this.worldServerLogger = (Logger) field.get(null); |
| 83 | + } catch (final Exception e) { |
| 84 | + e.printStackTrace(); |
| 85 | + } |
| 86 | + try { |
| 87 | + final RegexFilter regexFilter = RegexFilter |
| 88 | + .createFilter("[\\S\\s]*Force-added player with duplicate UUID[\\S\\s]*", null, false, |
| 89 | + Filter.Result.DENY, Filter.Result.ACCEPT); |
| 90 | + this.worldServerLogger.addFilter(regexFilter); |
| 91 | + } catch (IllegalAccessException e) { |
| 92 | + e.printStackTrace(); |
| 93 | + } |
| 94 | + } |
| 95 | + } |
| 96 | + |
| 97 | + @Override @Nullable public Location getOrCreateNetherPortal(@NotNull final org.bukkit.entity.Entity entity, |
| 98 | + @NotNull final Location origin) { |
| 99 | + final ServerLevel worldServer = Objects.requireNonNull(((CraftWorld) origin.getWorld()).getHandle()); |
| 100 | + final PortalForcer portalTravelAgent = Objects.requireNonNull(worldServer.getPortalForcer()); |
| 101 | + final Entity nmsEntity = Objects.requireNonNull(((CraftEntity) entity).getHandle()); |
| 102 | + final BlockPos blockPosition = new BlockPos(origin.getBlockX(), origin.getBlockY(), origin.getBlockZ()); |
| 103 | + final WorldBorder worldBorder = worldServer.getWorldBorder(); |
| 104 | + Optional<BlockPos> existingPortalPosition = Objects.requireNonNull(portalTravelAgent, "travel agent") |
| 105 | + .findClosestPortalPosition(Objects.requireNonNull(blockPosition, "position"), worldBorder,128); |
| 106 | + if (existingPortalPosition.isPresent()) { |
| 107 | + BlockPos bottomLeft = existingPortalPosition.get(); |
| 108 | + return new Location(origin.getWorld(), bottomLeft.getX(), bottomLeft.getY(), bottomLeft.getZ()); |
| 109 | + } |
| 110 | + Optional<BlockUtil.FoundRectangle> createdPortal = portalTravelAgent.createPortal(blockPosition, |
| 111 | + nmsEntity.getDirection().getAxis(), nmsEntity, |
| 112 | + 16); |
| 113 | + if (createdPortal.isEmpty()) { |
| 114 | + return null; |
| 115 | + } |
| 116 | + final BlockUtil.FoundRectangle rectangle = createdPortal.get(); |
| 117 | + return new Location(origin.getWorld(), rectangle.minCorner.getX() + 1D, rectangle.minCorner.getY() - 1D, |
| 118 | + rectangle.minCorner.getZ() + 1D); |
| 119 | + } |
| 120 | + |
| 121 | + @Override @Nullable public Location getDimensionSpawn(@NotNull final Location origin) { |
| 122 | + if (Objects.requireNonNull(origin.getWorld()).getEnvironment() |
| 123 | + == World.Environment.THE_END) { |
| 124 | + return new Location(origin.getWorld(), 100, 50, 0); |
| 125 | + } |
| 126 | + return origin.getWorld().getSpawnLocation(); |
| 127 | + } |
| 128 | + |
| 129 | + @Override |
| 130 | + public void writePlayerData(@NotNull final Player player, @NotNull final Path file) { |
| 131 | + final CompoundTag playerTag = new CompoundTag(); |
| 132 | + final net.minecraft.world.entity.player.Player entityPlayer = ((CraftPlayer) player).getHandle(); |
| 133 | + entityPlayer.save(playerTag); |
| 134 | + |
| 135 | + // Handle the Optional return type of getCompound |
| 136 | + CompoundTag hyperverse = playerTag.getCompound("hyperverse").orElse(new CompoundTag()); |
| 137 | + // If "hyperverse" didn't exist, we need to add it to playerTag |
| 138 | + playerTag.put("hyperverse", hyperverse); |
| 139 | + |
| 140 | + hyperverse.putLong("writeTime", System.currentTimeMillis()); |
| 141 | + hyperverse.putString("version", Bukkit.getPluginManager().getPlugin("Hyperverse").getDescription().getVersion()); |
| 142 | + |
| 143 | + taskFactory.newChain().async(() -> { |
| 144 | + try (final OutputStream outputStream = Files.newOutputStream(file)) { |
| 145 | + NbtIo.writeCompressed(playerTag, outputStream); |
| 146 | + } catch (final Exception e) { |
| 147 | + e.printStackTrace(); |
| 148 | + } |
| 149 | + }).execute(); |
| 150 | + } |
| 151 | + |
| 152 | + @Override |
| 153 | + public void readPlayerData(@NotNull final Player player, @NotNull final Path file, @NotNull final Runnable whenDone) { |
| 154 | + final Location originLocation = player.getLocation().clone(); |
| 155 | + taskFactory.newChain().asyncFirst(() -> { |
| 156 | + try (final InputStream inputStream = Files.newInputStream(file)) { |
| 157 | + return Optional.of(NbtIo.readCompressed(inputStream, NbtAccounter.unlimitedHeap())); |
| 158 | + } catch (final Exception e) { |
| 159 | + e.printStackTrace(); |
| 160 | + } |
| 161 | + return Optional.empty(); |
| 162 | + }).syncLast((optionalCompound) -> { |
| 163 | + if (!optionalCompound.isPresent()) { |
| 164 | + return; |
| 165 | + } |
| 166 | + final CompoundTag compound = (CompoundTag) optionalCompound.get(); |
| 167 | + PaperLib.getChunkAtAsync(originLocation).thenAccept(chunk -> { |
| 168 | + // Health and hunger don't update properly, so we give them a little help |
| 169 | + final float health = compound.getFloat("Health").orElse(20.0f); // Default to 20 (full health) |
| 170 | + final int foodLevel = compound.getInt("foodLevel").orElse(20); // Default to 20 (full food) |
| 171 | + |
| 172 | + // Restore bed spawn |
| 173 | + final String spawnWorld = compound.getString("SpawnWorld").orElse(player.getWorld().getName()); // Default to current world |
| 174 | + final int spawnX = compound.getInt("SpawnX").orElse((int) originLocation.getX()); |
| 175 | + final int spawnY = compound.getInt("SpawnY").orElse((int) originLocation.getY()); |
| 176 | + final int spawnZ = compound.getInt("SpawnZ").orElse((int) originLocation.getZ()); |
| 177 | + final Location spawnLocation = new Location(Bukkit.getWorld(spawnWorld), spawnX, spawnY, spawnZ); |
| 178 | + |
| 179 | + final ServerPlayer entityPlayer = ((CraftPlayer) player).getHandle(); |
| 180 | + |
| 181 | + // We re-write the extra Bukkit data as to not mess up the profile |
| 182 | + ((CraftPlayer) player).setExtraData(compound); |
| 183 | + // Set the position to the player's current position |
| 184 | + Vec3 pos = entityPlayer.position(); |
| 185 | + compound.put("Pos", doubleList(pos.x, pos.y, pos.z)); |
| 186 | + // Set the world to the player's current world |
| 187 | + compound.putString("world", player.getWorld().getName()); |
| 188 | + // Store persistent values |
| 189 | + ((CraftPlayer) player).storeBukkitValues(compound); |
| 190 | + |
| 191 | + // We start by doing a total reset |
| 192 | + entityPlayer.reset(); |
| 193 | + entityPlayer.load(compound); |
| 194 | + |
| 195 | + entityPlayer.effectsDirty = true; |
| 196 | + entityPlayer.onUpdateAbilities(); |
| 197 | + player.teleport(originLocation); |
| 198 | + |
| 199 | + final ServerLevel worldServer = ((CraftWorld) originLocation.getWorld()).getHandle(); |
| 200 | + final DimensionType dimensionManager = worldServer.dimensionType(); |
| 201 | + |
| 202 | + // Prevent annoying message |
| 203 | + // Spigot-obf = decouple() |
| 204 | + entityPlayer.unRide(); |
| 205 | + worldServer.removePlayerImmediately(entityPlayer, Entity.RemovalReason.CHANGED_DIMENSION); |
| 206 | + // worldServer.removePlayer above should remove the player from the map, but that doesn't always happen |
| 207 | + try { |
| 208 | + if (this.entitySectionManager == null) { |
| 209 | + this.entitySectionManager = worldServer.getClass().getDeclaredField("entityManager"); |
| 210 | + this.entitySectionManager.setAccessible(true); |
| 211 | + } |
| 212 | + final PersistentEntitySectionManager<Entity> esm = (PersistentEntitySectionManager<Entity>) this.entitySectionManager.get(worldServer); |
| 213 | + if (this.entityLookup == null) { |
| 214 | + this.entityLookup = esm.getClass().getDeclaredField("visibleEntityStorage"); |
| 215 | + } |
| 216 | + final EntityLookup<Entity> lookup = (EntityLookup<Entity>) this.entityLookup.get(esm); |
| 217 | + lookup.remove(entityPlayer); |
| 218 | + } catch (final NoSuchFieldException | IllegalAccessException e) { |
| 219 | + e.printStackTrace(); |
| 220 | + } |
| 221 | + |
| 222 | + // pre 1.18 code = PlayerList#moveToWorld |
| 223 | + entityPlayer.server.getPlayerList().remove(entityPlayer); |
| 224 | + worldServer.getServer().getPlayerList().respawn(entityPlayer, true, |
| 225 | + Entity.RemovalReason.CHANGED_DIMENSION, PlayerRespawnEvent.RespawnReason.PLUGIN, originLocation); |
| 226 | + |
| 227 | + // Apply health and foodLevel |
| 228 | + player.setHealth(health); |
| 229 | + player.setFoodLevel(foodLevel); |
| 230 | + player.setPortalCooldown(40); |
| 231 | + player.setBedSpawnLocation(spawnLocation, true); |
| 232 | + }); |
| 233 | + }).execute(whenDone); |
| 234 | + } |
| 235 | + |
| 236 | + @Override |
| 237 | + @Nullable |
| 238 | + public Location findBedRespawn(@NotNull final Location spawnLocation) { |
| 239 | + final CraftWorld craftWorld = (CraftWorld) spawnLocation.getWorld(); |
| 240 | + if (craftWorld == null) { |
| 241 | + return null; |
| 242 | + } |
| 243 | + |
| 244 | + // Create a BlockPos for the spawn location |
| 245 | + BlockPos blockPos = new BlockPos(spawnLocation.getBlockX(), spawnLocation.getBlockY(), spawnLocation.getBlockZ()); |
| 246 | + |
| 247 | + // Get the dimension ResourceKey from the ServerLevel |
| 248 | + ServerLevel serverLevel = craftWorld.getHandle(); |
| 249 | + ResourceKey<Level> dimension = serverLevel.dimension(); |
| 250 | + |
| 251 | + // Create a RespawnConfig instance with the required arguments |
| 252 | + ServerPlayer.RespawnConfig respawnConfig = new ServerPlayer.RespawnConfig( |
| 253 | + dimension, // The dimension of the world |
| 254 | + blockPos, // The position to check for respawn |
| 255 | + 0.0f, // Yaw (rotation angle), default to 0 |
| 256 | + true // Forced flag, set to true to match previous behavior |
| 257 | + ); |
| 258 | + |
| 259 | + // Call the method with the correct arguments |
| 260 | + return ServerPlayer.findRespawnAndUseSpawnBlock( |
| 261 | + serverLevel, |
| 262 | + respawnConfig, |
| 263 | + false // Assuming the boolean is respawnAfterDeath |
| 264 | + ) |
| 265 | + .map(ServerPlayer.RespawnPosAngle::position) |
| 266 | + .map(pos -> new Location(spawnLocation.getWorld(), pos.x(), pos.y(), pos.z())) |
| 267 | + .orElse(null); |
| 268 | + } |
| 269 | + |
| 270 | + private static ListTag doubleList(final double... values) { |
| 271 | + final ListTag tagList = new ListTag(); |
| 272 | + for (final double d : values) { |
| 273 | + tagList.add(DoubleTag.valueOf(d)); |
| 274 | + } |
| 275 | + return tagList; |
| 276 | + } |
| 277 | + |
| 278 | +} |
0 commit comments