package net.minecraft.network.protocol;

import com.mojang.logging.LogUtils;
import net.minecraft.CrashReport;
import net.minecraft.CrashReportCategory;
import net.minecraft.CrashReportDetail;
import net.minecraft.ReportedException;
import net.minecraft.network.PacketListener;
import net.minecraft.server.RunningOnDifferentThreadException;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.thread.BlockableEventLoop;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;

public class PacketUtils {
	private static final Logger LOGGER = LogUtils.getLogger();

	/**
	 * Ensures that the given packet is handled on the main thread. If the current thread is not the main thread, this method
	 * throws {@link net.minecraft.server.RunningOnDifferentThreadException}, which is caught and ignored in the outer call ({@link net.minecraft.network.Connection#channelRead0(io.netty.channel.ChannelHandlerContext, net.minecraft.network.protocol.Packet)}). Additionally, it then re-schedules the packet to be handled on the main thread,
	 * which will then end up back here, but this time on the main thread.
	 */
	public static <T extends PacketListener> void ensureRunningOnSameThread(Packet<T> packet, T processor, ServerLevel level) throws RunningOnDifferentThreadException {
		ensureRunningOnSameThread(packet, processor, level.getServer());
	}

	/**
	 * Ensures that the given packet is handled on the main thread. If the current thread is not the main thread, this method
	 * throws {@link net.minecraft.server.RunningOnDifferentThreadException}, which is caught and ignored in the outer call ({@link net.minecraft.network.Connection#channelRead0(io.netty.channel.ChannelHandlerContext, net.minecraft.network.protocol.Packet)}). Additionally, it then re-schedules the packet to be handled on the main thread,
	 * which will then end up back here, but this time on the main thread.
	 */
	public static <T extends PacketListener> void ensureRunningOnSameThread(Packet<T> packet, T processor, BlockableEventLoop<?> executor) throws RunningOnDifferentThreadException {
		if (!executor.isSameThread()) {
			executor.executeIfPossible(() -> {
				if (processor.shouldHandleMessage(packet)) {
					try {
						packet.handle(processor);
					} catch (Exception var4) {
						if (var4 instanceof ReportedException reportedException && reportedException.getCause() instanceof OutOfMemoryError) {
							throw makeReportedException(var4, packet, processor);
						}

						processor.onPacketError(packet, var4);
					}
				} else {
					LOGGER.debug("Ignoring packet due to disconnection: {}", packet);
				}
			});
			throw RunningOnDifferentThreadException.RUNNING_ON_DIFFERENT_THREAD;
		}
	}

	public static <T extends PacketListener> ReportedException makeReportedException(Exception exception, Packet<T> packet, T packetListener) {
		if (exception instanceof ReportedException reportedException) {
			fillCrashReport(reportedException.getReport(), packetListener, packet);
			return reportedException;
		} else {
			CrashReport crashReport = CrashReport.forThrowable(exception, "Main thread packet handler");
			fillCrashReport(crashReport, packetListener, packet);
			return new ReportedException(crashReport);
		}
	}

	public static <T extends PacketListener> void fillCrashReport(CrashReport crashReport, T packetListener, @Nullable Packet<T> packet) {
		if (packet != null) {
			CrashReportCategory crashReportCategory = crashReport.addCategory("Incoming Packet");
			crashReportCategory.setDetail("Type", (CrashReportDetail<String>)(() -> packet.type().toString()));
			crashReportCategory.setDetail("Is Terminal", (CrashReportDetail<String>)(() -> Boolean.toString(packet.isTerminal())));
			crashReportCategory.setDetail("Is Skippable", (CrashReportDetail<String>)(() -> Boolean.toString(packet.isSkippable())));
		}

		packetListener.fillCrashReport(crashReport);
	}
}