package net.minecraft.server.packs.resources;

import com.google.common.collect.Lists;
import com.mojang.logging.LogUtils;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.packs.PackResources;
import net.minecraft.server.packs.PackType;
import net.minecraft.util.Unit;
import org.slf4j.Logger;

public class ReloadableResourceManager implements ResourceManager, AutoCloseable {
	private static final Logger LOGGER = LogUtils.getLogger();
	private CloseableResourceManager resources;
	private final List<PreparableReloadListener> listeners = Lists.<PreparableReloadListener>newArrayList();
	private final PackType type;

	public ReloadableResourceManager(PackType type) {
		this.type = type;
		this.resources = new MultiPackResourceManager(type, List.of());
	}

	public void close() {
		this.resources.close();
	}

	public void registerReloadListener(PreparableReloadListener listener) {
		this.listeners.add(listener);
	}

	public ReloadInstance createReload(Executor backgroundExecutor, Executor gameExecutor, CompletableFuture<Unit> waitingFor, List<PackResources> resourcePacks) {
		LOGGER.info("Reloading ResourceManager: {}", LogUtils.defer(() -> resourcePacks.stream().map(PackResources::packId).collect(Collectors.joining(", "))));
		this.resources.close();
		this.resources = new MultiPackResourceManager(this.type, resourcePacks);
		return SimpleReloadInstance.create(this.resources, this.listeners, backgroundExecutor, gameExecutor, waitingFor, LOGGER.isDebugEnabled());
	}

	@Override
	public Optional<Resource> getResource(ResourceLocation resourceLocation) {
		return this.resources.getResource(resourceLocation);
	}

	@Override
	public Set<String> getNamespaces() {
		return this.resources.getNamespaces();
	}

	@Override
	public List<Resource> getResourceStack(ResourceLocation location) {
		return this.resources.getResourceStack(location);
	}

	@Override
	public Map<ResourceLocation, Resource> listResources(String path, Predicate<ResourceLocation> filter) {
		return this.resources.listResources(path, filter);
	}

	@Override
	public Map<ResourceLocation, List<Resource>> listResourceStacks(String path, Predicate<ResourceLocation> filter) {
		return this.resources.listResourceStacks(path, filter);
	}

	@Override
	public Stream<PackResources> listPacks() {
		return this.resources.listPacks();
	}
}