BauSystem2.0/BauSystem_15/src/de/steamwar/bausystem/utils/NMSWrapper15.java

/*
 * This file is a part of the SteamWar software.
 *
 * Copyright (C) 2022  SteamWar.de-Serverteam
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

package de.steamwar.bausystem.utils;

import com.comphenix.tinyprotocol.Reflection;
import de.steamwar.bausystem.entities.DetonatorEntity15;
import de.steamwar.bausystem.entities.SimulatorEntity15;
import de.steamwar.bausystem.entities.TraceEntity15;
import de.steamwar.bausystem.features.detonator.AbstractDetonatorEntity;
import de.steamwar.bausystem.features.other.NoClipCommand;
import de.steamwar.bausystem.features.simulator.AbstractSimulatorEntity;
import de.steamwar.bausystem.features.tracer.AbstractTraceEntity;
import de.steamwar.bausystem.features.warp.AbstractWarpEntity;
import de.steamwar.bausystem.entities.WarpEntity15;
import net.minecraft.server.v1_15_R1.*;
import org.bukkit.Bukkit;
import org.bukkit.GameMode;
import org.bukkit.World;
import org.bukkit.craftbukkit.v1_15_R1.entity.CraftEntity;
import org.bukkit.craftbukkit.v1_15_R1.entity.CraftPlayer;
import org.bukkit.craftbukkit.v1_15_R1.inventory.CraftItemStack;
import org.bukkit.entity.Player;
import org.bukkit.entity.TNTPrimed;
import org.bukkit.inventory.ItemStack;
import org.bukkit.util.Vector;

import java.util.ArrayList;
import java.util.List;
import java.util.function.LongSupplier;

public class NMSWrapper15 implements NMSWrapper.INMSWrapper {

    private static final Reflection.FieldAccessor<EnumGamemode> playerGameMode = Reflection.getField(PlayerInteractManager.class, EnumGamemode.class, 0);
    @Override
    @SuppressWarnings("deprecation")
    public void setInternalGameMode(Player player, GameMode gameMode) {
        playerGameMode.set(((CraftPlayer) player).getHandle().playerInteractManager, EnumGamemode.getById(gameMode.getValue()));
    }

    @Override
    public void init(LongSupplier longSupplier) {
        SystemUtils.a = () -> System.nanoTime() + longSupplier.getAsLong();
    }

    private static final List<Packet<?>> packets = new ArrayList<>();
    private static final Vec3D noMotion = new Vec3D(0, 0, 0);
    @Override
    public void createTickCache(World world) {
        packets.clear();
        world.getEntities().stream().filter(entity -> !(entity instanceof Player)).forEach(entity -> {
            packets.add(new PacketPlayOutEntityVelocity(entity.getEntityId(), noMotion));
            packets.add(new PacketPlayOutEntityTeleport(((CraftEntity) entity).getHandle()));

            if (entity instanceof TNTPrimed) {
                net.minecraft.server.v1_15_R1.Entity serverEntity = ((CraftEntity) entity).getHandle();
                packets.add(new PacketPlayOutEntityMetadata(serverEntity.getId(), serverEntity.getDataWatcher(), true));
            }
        });
    }

    @Override
    public void sendTickPackets() {
        Bukkit.getOnlinePlayers().forEach(player -> {
            PlayerConnection connection = ((CraftPlayer) player).getHandle().playerConnection;
            for (Packet<?> p : packets) {
                connection.sendPacket(p);
            }
        });
    }

    private static final Reflection.FieldAccessor<Integer> gameStateChangeReason = Reflection.getField(NoClipCommand.gameStateChange, int.class, 0);
    @Override
    public void setGameStateChangeReason(Object packet) {
        gameStateChangeReason.set(packet, 3);
    }

    private static final int threshold = 2048;
    @Override
    public boolean checkItemStack(ItemStack item) {
        net.minecraft.server.v1_15_R1.ItemStack nmsItem = CraftItemStack.asNMSCopy(item);
        NBTTagCompound tag = nmsItem.getTag();
        if (tag != null && tag.hasKey("BlockEntityTag")) {
            NBTTagCompound blockTag = tag.getCompound("BlockEntityTag");
            if (blockTag.hasKey("Items")) {
                return drillDown(blockTag.getList("Items", 10), 0, 0) > threshold;
            }
        }

        return false;
    }

    private int drillDown(NBTTagList items, int layer, int start) {
        if (layer > 2) return start + threshold;
        int invalid = start;
        for (NBTBase nbtBase : items) {
            if (!(nbtBase instanceof NBTTagCompound))
                continue;
            NBTTagCompound slot = (NBTTagCompound) nbtBase;
            if (slot.hasKey("tag")) {
                invalid += slot.getByte("Count");
                NBTTagCompound iTag = slot.getCompound("tag");
                if (iTag.hasKey("BlockEntityTag")) {
                    NBTTagCompound blockTag = iTag.getCompound("BlockEntityTag");
                    if (blockTag.hasKey("Items")) {
                        invalid = drillDown(blockTag.getList("Items", 10), layer + 1, invalid);
                    }
                }
            }
            if (invalid > threshold)
                break;
        }
        return invalid;
    }

    @Override
    public AbstractWarpEntity createWarp(World world, Vector position, String name) {
        return new WarpEntity15(world, position, name);
    }

    @Override
    public AbstractSimulatorEntity createSimulator(World world, Vector tntPosition, boolean highlight) {
        return new SimulatorEntity15(world, tntPosition, highlight);
    }

    @Override
    public AbstractDetonatorEntity constructDetonator(World world, Vector position) {
        return new DetonatorEntity15(world, position);
    }

    @Override
    public AbstractTraceEntity createTrace(World world, Vector tntPosition, boolean tnt) {
        return new TraceEntity15(world, tntPosition, tnt);
    }
}