package net.minecraft.world.level.levelgen.structure.templatesystem.loader;

import com.mojang.datafixers.DataFixer;
import com.mojang.logging.LogUtils;
import java.util.Optional;
import java.util.stream.Stream;
import net.minecraft.core.HolderGetter;
import net.minecraft.resources.FileToIdConverter;
import net.minecraft.resources.Identifier;
import net.minecraft.server.packs.resources.ResourceManager;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureTemplate;
import org.slf4j.Logger;

public class ResourceManagerTemplateSource extends TemplateSource {
	private static final Logger LOGGER = LogUtils.getLogger();
	private ResourceManager resourceManager;
	private final FileToIdConverter fileToIdConverter;

	public ResourceManagerTemplateSource(
		final DataFixer fixerUpper, final HolderGetter<Block> blockLookup, final ResourceManager resourceManager, final FileToIdConverter fileToIdConverter
	) {
		super(fixerUpper, blockLookup);
		this.resourceManager = resourceManager;
		this.fileToIdConverter = fileToIdConverter;
	}

	public void setResourceManager(final ResourceManager resourceManager) {
		this.resourceManager = resourceManager;
	}

	@Override
	public Optional<StructureTemplate> load(final Identifier id) {
		Identifier identifier = this.fileToIdConverter.idToFile(id);
		return this.load(() -> this.resourceManager.open(identifier), false, e -> LOGGER.error("Couldn't load structure {}", id, e));
	}

	@Override
	public Stream<Identifier> list() {
		return this.fileToIdConverter.listMatchingResources(this.resourceManager).keySet().stream().map(this.fileToIdConverter::fileToId);
	}
}