package net.minecraft.resources;

import java.util.List;
import java.util.Map;
import net.minecraft.core.Registry;
import net.minecraft.core.registries.Registries;
import net.minecraft.server.packs.resources.Resource;
import net.minecraft.server.packs.resources.ResourceManager;

public class FileToIdConverter {
	private final String prefix;
	private final String extension;

	public FileToIdConverter(String prefix, String extenstion) {
		this.prefix = prefix;
		this.extension = extenstion;
	}

	public static FileToIdConverter json(String name) {
		return new FileToIdConverter(name, ".json");
	}

	public static FileToIdConverter registry(ResourceKey<? extends Registry<?>> registryKey) {
		return json(Registries.elementsDirPath(registryKey));
	}

	public ResourceLocation idToFile(ResourceLocation id) {
		return id.withPath(this.prefix + "/" + id.getPath() + this.extension);
	}

	public ResourceLocation fileToId(ResourceLocation file) {
		String string = file.getPath();
		return file.withPath(string.substring(this.prefix.length() + 1, string.length() - this.extension.length()));
	}

	public Map<ResourceLocation, Resource> listMatchingResources(ResourceManager resourceManager) {
		return resourceManager.listResources(this.prefix, resourceLocation -> resourceLocation.getPath().endsWith(this.extension));
	}

	public Map<ResourceLocation, List<Resource>> listMatchingResourceStacks(ResourceManager resourceManager) {
		return resourceManager.listResourceStacks(this.prefix, resourceLocation -> resourceLocation.getPath().endsWith(this.extension));
	}
}