package net.minecraft.client.gui.screens;

import com.google.common.collect.Maps;
import com.mojang.logging.LogUtils;
import java.util.Map;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.screens.inventory.AnvilScreen;
import net.minecraft.client.gui.screens.inventory.BeaconScreen;
import net.minecraft.client.gui.screens.inventory.BlastFurnaceScreen;
import net.minecraft.client.gui.screens.inventory.BrewingStandScreen;
import net.minecraft.client.gui.screens.inventory.CartographyTableScreen;
import net.minecraft.client.gui.screens.inventory.ContainerScreen;
import net.minecraft.client.gui.screens.inventory.CrafterScreen;
import net.minecraft.client.gui.screens.inventory.CraftingScreen;
import net.minecraft.client.gui.screens.inventory.DispenserScreen;
import net.minecraft.client.gui.screens.inventory.EnchantmentScreen;
import net.minecraft.client.gui.screens.inventory.FurnaceScreen;
import net.minecraft.client.gui.screens.inventory.GrindstoneScreen;
import net.minecraft.client.gui.screens.inventory.HopperScreen;
import net.minecraft.client.gui.screens.inventory.LecternScreen;
import net.minecraft.client.gui.screens.inventory.LoomScreen;
import net.minecraft.client.gui.screens.inventory.MenuAccess;
import net.minecraft.client.gui.screens.inventory.MerchantScreen;
import net.minecraft.client.gui.screens.inventory.ShulkerBoxScreen;
import net.minecraft.client.gui.screens.inventory.SmithingScreen;
import net.minecraft.client.gui.screens.inventory.SmokerScreen;
import net.minecraft.client.gui.screens.inventory.StonecutterScreen;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.network.chat.Component;
import net.minecraft.world.entity.player.Inventory;
import net.minecraft.world.inventory.AbstractContainerMenu;
import net.minecraft.world.inventory.MenuType;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;

@Environment(EnvType.CLIENT)
public class MenuScreens {
	private static final Logger LOGGER = LogUtils.getLogger();
	private static final Map<MenuType<?>, MenuScreens.ScreenConstructor<?, ?>> SCREENS = Maps.<MenuType<?>, MenuScreens.ScreenConstructor<?, ?>>newHashMap();

	public static <T extends AbstractContainerMenu> void create(MenuType<T> type, Minecraft mc, int windowId, Component title) {
		MenuScreens.ScreenConstructor<T, ?> screenConstructor = getConstructor(type);
		if (screenConstructor == null) {
			LOGGER.warn("Failed to create screen for menu type: {}", BuiltInRegistries.MENU.getKey(type));
		} else {
			screenConstructor.fromPacket(title, type, mc, windowId);
		}
	}

	@Nullable
	private static <T extends AbstractContainerMenu> MenuScreens.ScreenConstructor<T, ?> getConstructor(MenuType<T> type) {
		return (MenuScreens.ScreenConstructor<T, ?>)SCREENS.get(type);
	}

	private static <M extends AbstractContainerMenu, U extends Screen & MenuAccess<M>> void register(
		MenuType<? extends M> type, MenuScreens.ScreenConstructor<M, U> factory
	) {
		MenuScreens.ScreenConstructor<?, ?> screenConstructor = (MenuScreens.ScreenConstructor<?, ?>)SCREENS.put(type, factory);
		if (screenConstructor != null) {
			throw new IllegalStateException("Duplicate registration for " + BuiltInRegistries.MENU.getKey(type));
		}
	}

	public static boolean selfTest() {
		boolean bl = false;

		for (MenuType<?> menuType : BuiltInRegistries.MENU) {
			if (!SCREENS.containsKey(menuType)) {
				LOGGER.debug("Menu {} has no matching screen", BuiltInRegistries.MENU.getKey(menuType));
				bl = true;
			}
		}

		return bl;
	}

	static {
		register(MenuType.GENERIC_9x1, ContainerScreen::new);
		register(MenuType.GENERIC_9x2, ContainerScreen::new);
		register(MenuType.GENERIC_9x3, ContainerScreen::new);
		register(MenuType.GENERIC_9x4, ContainerScreen::new);
		register(MenuType.GENERIC_9x5, ContainerScreen::new);
		register(MenuType.GENERIC_9x6, ContainerScreen::new);
		register(MenuType.GENERIC_3x3, DispenserScreen::new);
		register(MenuType.CRAFTER_3x3, CrafterScreen::new);
		register(MenuType.ANVIL, AnvilScreen::new);
		register(MenuType.BEACON, BeaconScreen::new);
		register(MenuType.BLAST_FURNACE, BlastFurnaceScreen::new);
		register(MenuType.BREWING_STAND, BrewingStandScreen::new);
		register(MenuType.CRAFTING, CraftingScreen::new);
		register(MenuType.ENCHANTMENT, EnchantmentScreen::new);
		register(MenuType.FURNACE, FurnaceScreen::new);
		register(MenuType.GRINDSTONE, GrindstoneScreen::new);
		register(MenuType.HOPPER, HopperScreen::new);
		register(MenuType.LECTERN, LecternScreen::new);
		register(MenuType.LOOM, LoomScreen::new);
		register(MenuType.MERCHANT, MerchantScreen::new);
		register(MenuType.SHULKER_BOX, ShulkerBoxScreen::new);
		register(MenuType.SMITHING, SmithingScreen::new);
		register(MenuType.SMOKER, SmokerScreen::new);
		register(MenuType.CARTOGRAPHY_TABLE, CartographyTableScreen::new);
		register(MenuType.STONECUTTER, StonecutterScreen::new);
	}

	@Environment(EnvType.CLIENT)
	interface ScreenConstructor<T extends AbstractContainerMenu, U extends Screen & MenuAccess<T>> {
		default void fromPacket(Component title, MenuType<T> type, Minecraft mc, int windowId) {
			U screen = this.create(type.create(windowId, mc.player.getInventory()), mc.player.getInventory(), title);
			mc.player.containerMenu = screen.getMenu();
			mc.setScreen(screen);
		}

		U create(T abstractContainerMenu, Inventory inventory, Component component);
	}
}