common/src/main/java/com/viaversion/viarewind/api/rewriter/VREntityRewriter.java - master - minecraftbackup/ViaRewind

common/src/main/java/com/viaversion/viarewind/api/rewriter/VREntityRewriter.java
80 строк3 KB
FlorianMichael
Improve spectating client entity id tracking
20 сен 2024, 15:36
20 сен 2024, 15:368945193
/*
 * This file is part of ViaRewind - https://github.com/ViaVersion/ViaRewind
 * Copyright (C) 2018-2024 ViaVersion and contributors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.viaversion.viarewind.api.rewriter;

import com.viaversion.viabackwards.api.BackwardsProtocol;
import com.viaversion.viabackwards.api.rewriters.LegacyEntityRewriter;
import com.viaversion.viarewind.ViaRewind;
import com.viaversion.viaversion.api.connection.UserConnection;
import com.viaversion.viaversion.api.minecraft.ClientWorld;
import com.viaversion.viaversion.api.minecraft.entitydata.EntityDataType;
import com.viaversion.viaversion.api.minecraft.entitydata.types.EntityDataTypes1_8;
import com.viaversion.viaversion.api.protocol.packet.ClientboundPacketType;
import com.viaversion.viaversion.api.protocol.remapper.PacketHandler;
import com.viaversion.viaversion.api.protocol.remapper.PacketHandlers;
import com.viaversion.viaversion.api.type.Types;
import com.viaversion.viaversion.data.entity.EntityTrackerBase;

public abstract class VREntityRewriter<C extends ClientboundPacketType, T extends BackwardsProtocol<C, ?, ?, ?>> extends LegacyEntityRewriter<C, T> {

	public VREntityRewriter(T protocol) {
		super(protocol, EntityDataTypes1_8.STRING, EntityDataTypes1_8.BYTE);
	}

	public VREntityRewriter(T protocol, EntityDataType displayType, EntityDataType displayVisibilityType) {
		super(protocol, displayType, displayVisibilityType);
	}

	protected void registerJoinGame1_8(final C packetType) {
		protocol.registerClientbound(packetType, new PacketHandlers() {
			@Override
			protected void register() {
				map(Types.INT); // Entity id
				map(Types.UNSIGNED_BYTE); // Game mode
				map(Types.BYTE); // Dimension
				handler(playerTrackerHandler());
				handler(getDimensionHandler());
			}
		});
	}

	protected void removeEntities(final UserConnection connection, final int[] entities) {
		final EntityTrackerBase tracker = tracker(connection);
		for (int entityId : entities) {
			tracker.removeEntity(entityId);
		}
	}

	protected PacketHandler getDimensionHandler() {
		return wrapper -> {
			final int dimension = wrapper.get(Types.BYTE, 0);
			final ClientWorld clientWorld = wrapper.user().getClientWorld(protocol.getClass());
			clientWorld.setEnvironment(dimension);
		};
	}

	@Override
	protected Object getDisplayVisibilityDataValue() {
		return (byte) 1;
	}

	@Override
	protected boolean alwaysShowOriginalMobName() {
		return ViaRewind.getConfig().alwaysShowOriginalMobName();
	}
}