package net.minecraft.client.gui.screens.worldselection;

import com.google.common.collect.Lists;
import it.unimi.dsi.fastutil.objects.Object2BooleanLinkedOpenHashMap;
import it.unimi.dsi.fastutil.objects.Object2BooleanMap;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Consumer;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.ChatFormatting;
import net.minecraft.client.gui.components.AbstractWidget;
import net.minecraft.client.gui.components.Button;
import net.minecraft.client.gui.components.MultiLineTextWidget;
import net.minecraft.client.gui.layouts.HeaderAndFooterLayout;
import net.minecraft.client.gui.layouts.LinearLayout;
import net.minecraft.client.gui.screens.Screen;
import net.minecraft.client.gui.screens.worldselection.SwitchGrid.Builder;
import net.minecraft.client.resources.language.I18n;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.server.packs.repository.Pack;
import net.minecraft.server.packs.repository.PackRepository;
import net.minecraft.server.packs.repository.PackSource;

@Environment(EnvType.CLIENT)
public class ExperimentsScreen extends Screen {
	private static final Component TITLE = Component.translatable("selectWorld.experiments");
	private static final Component INFO = Component.translatable("selectWorld.experiments.info").withStyle(ChatFormatting.RED);
	private static final int MAIN_CONTENT_WIDTH = 310;
	private final HeaderAndFooterLayout layout = new HeaderAndFooterLayout(this);
	private final Screen parent;
	private final PackRepository packRepository;
	private final Consumer<PackRepository> output;
	private final Object2BooleanMap<Pack> packs = new Object2BooleanLinkedOpenHashMap<>();

	public ExperimentsScreen(Screen parent, PackRepository packRepository, Consumer<PackRepository> output) {
		super(TITLE);
		this.parent = parent;
		this.packRepository = packRepository;
		this.output = output;

		for (Pack pack : packRepository.getAvailablePacks()) {
			if (pack.getPackSource() == PackSource.FEATURE) {
				this.packs.put(pack, packRepository.getSelectedPacks().contains(pack));
			}
		}
	}

	@Override
	protected void init() {
		this.layout.addTitleHeader(TITLE, this.font);
		LinearLayout linearLayout = this.layout.addToContents(LinearLayout.vertical());
		linearLayout.addChild(new MultiLineTextWidget(INFO, this.font).setMaxWidth(310), layoutSettings -> layoutSettings.paddingBottom(15));
		Builder builder = SwitchGrid.builder(310).withInfoUnderneath(2, true).withRowSpacing(4);
		this.packs
			.forEach(
				(pack, boolean_) -> builder.addSwitch(
						getHumanReadableTitle(pack), () -> this.packs.getBoolean(pack), boolean_x -> this.packs.put(pack, boolean_x.booleanValue())
					)
					.withInfo(pack.getDescription())
			);
		builder.build(linearLayout::addChild);
		LinearLayout linearLayout2 = this.layout.addToFooter(LinearLayout.horizontal().spacing(8));
		linearLayout2.addChild(Button.builder(CommonComponents.GUI_DONE, button -> this.onDone()).build());
		linearLayout2.addChild(Button.builder(CommonComponents.GUI_CANCEL, button -> this.onClose()).build());
		this.layout.visitWidgets(guiEventListener -> {
			AbstractWidget var10000 = this.addRenderableWidget(guiEventListener);
		});
		this.repositionElements();
	}

	private static Component getHumanReadableTitle(Pack pack) {
		String string = "dataPack." + pack.getId() + ".name";
		return (Component)(I18n.exists(string) ? Component.translatable(string) : pack.getTitle());
	}

	@Override
	protected void repositionElements() {
		this.layout.arrangeElements();
	}

	@Override
	public Component getNarrationMessage() {
		return CommonComponents.joinForNarration(super.getNarrationMessage(), INFO);
	}

	@Override
	public void onClose() {
		this.minecraft.setScreen(this.parent);
	}

	private void onDone() {
		List<Pack> list = new ArrayList(this.packRepository.getSelectedPacks());
		List<Pack> list2 = new ArrayList();
		this.packs.forEach((pack, boolean_) -> {
			list.remove(pack);
			if (boolean_) {
				list2.add(pack);
			}
		});
		list.addAll(Lists.reverse(list2));
		this.packRepository.setSelected(list.stream().map(Pack::getId).toList());
		this.output.accept(this.packRepository);
	}
}