package net.minecraft.client.resources;

import java.nio.file.FileSystems;
import java.nio.file.Files;
import java.nio.file.LinkOption;
import java.nio.file.Path;
import java.util.Map;
import java.util.Optional;
import java.util.function.BiConsumer;
import java.util.function.Function;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.SharedConstants;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.packs.BuiltInMetadata;
import net.minecraft.server.packs.PackLocationInfo;
import net.minecraft.server.packs.PackResources;
import net.minecraft.server.packs.PackSelectionConfig;
import net.minecraft.server.packs.PackType;
import net.minecraft.server.packs.VanillaPackResources;
import net.minecraft.server.packs.VanillaPackResourcesBuilder;
import net.minecraft.server.packs.metadata.pack.PackMetadataSection;
import net.minecraft.server.packs.repository.BuiltInPackSource;
import net.minecraft.server.packs.repository.KnownPack;
import net.minecraft.server.packs.repository.Pack;
import net.minecraft.server.packs.repository.PackSource;
import net.minecraft.server.packs.repository.Pack.Position;
import net.minecraft.server.packs.repository.Pack.ResourcesSupplier;
import net.minecraft.world.level.validation.DirectoryValidator;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class ClientPackSource extends BuiltInPackSource {
	private static final PackMetadataSection VERSION_METADATA_SECTION = new PackMetadataSection(
		Component.translatable("resourcePack.vanilla.description"), SharedConstants.getCurrentVersion().getPackVersion(PackType.CLIENT_RESOURCES), Optional.empty()
	);
	private static final BuiltInMetadata BUILT_IN_METADATA = BuiltInMetadata.of(PackMetadataSection.TYPE, VERSION_METADATA_SECTION);
	public static final String HIGH_CONTRAST_PACK = "high_contrast";
	private static final Map<String, Component> SPECIAL_PACK_NAMES = Map.of(
		"programmer_art", Component.translatable("resourcePack.programmer_art.name"), "high_contrast", Component.translatable("resourcePack.high_contrast.name")
	);
	private static final PackLocationInfo VANILLA_PACK_INFO = new PackLocationInfo(
		"vanilla", Component.translatable("resourcePack.vanilla.name"), PackSource.BUILT_IN, Optional.of(CORE_PACK_INFO)
	);
	private static final PackSelectionConfig VANILLA_SELECTION_CONFIG = new PackSelectionConfig(true, Position.BOTTOM, false);
	private static final PackSelectionConfig BUILT_IN_SELECTION_CONFIG = new PackSelectionConfig(false, Position.TOP, false);
	private static final ResourceLocation PACKS_DIR = ResourceLocation.withDefaultNamespace("resourcepacks");
	@Nullable
	private final Path externalAssetDir;

	public ClientPackSource(Path assetIndex, DirectoryValidator validator) {
		super(PackType.CLIENT_RESOURCES, createVanillaPackSource(assetIndex), PACKS_DIR, validator);
		this.externalAssetDir = this.findExplodedAssetPacks(assetIndex);
	}

	private static PackLocationInfo createBuiltInPackLocation(String id, Component title) {
		return new PackLocationInfo(id, title, PackSource.BUILT_IN, Optional.of(KnownPack.vanilla(id)));
	}

	@Nullable
	private Path findExplodedAssetPacks(Path assetIndex) {
		if (SharedConstants.IS_RUNNING_IN_IDE && assetIndex.getFileSystem() == FileSystems.getDefault()) {
			Path path = assetIndex.getParent().resolve("resourcepacks");
			if (Files.isDirectory(path, new LinkOption[0])) {
				return path;
			}
		}

		return null;
	}

	private static VanillaPackResources createVanillaPackSource(Path assetIndex) {
		VanillaPackResourcesBuilder vanillaPackResourcesBuilder = new VanillaPackResourcesBuilder()
			.setMetadata(BUILT_IN_METADATA)
			.exposeNamespace("minecraft", "realms");
		return vanillaPackResourcesBuilder.applyDevelopmentConfig().pushJarResources().pushAssetPath(PackType.CLIENT_RESOURCES, assetIndex).build(VANILLA_PACK_INFO);
	}

	@Override
	protected Component getPackTitle(String id) {
		Component component = (Component)SPECIAL_PACK_NAMES.get(id);
		return (Component)(component != null ? component : Component.literal(id));
	}

	@Nullable
	@Override
	protected Pack createVanillaPack(PackResources resources) {
		return Pack.readMetaAndCreate(VANILLA_PACK_INFO, fixedResources(resources), PackType.CLIENT_RESOURCES, VANILLA_SELECTION_CONFIG);
	}

	@Nullable
	@Override
	protected Pack createBuiltinPack(String id, ResourcesSupplier resources, Component title) {
		return Pack.readMetaAndCreate(createBuiltInPackLocation(id, title), resources, PackType.CLIENT_RESOURCES, BUILT_IN_SELECTION_CONFIG);
	}

	@Override
	protected void populatePackList(BiConsumer<String, Function<String, Pack>> populator) {
		super.populatePackList(populator);
		if (this.externalAssetDir != null) {
			this.discoverPacksInPath(this.externalAssetDir, populator);
		}
	}
}