package net.minecraft.data;

import java.nio.file.Path;
import java.util.function.UnaryOperator;
import net.minecraft.core.Registry;
import net.minecraft.core.registries.Registries;
import net.minecraft.resources.Identifier;
import net.minecraft.resources.ResourceKey;

public class PackOutput {
	private final Path outputFolder;

	public PackOutput(final Path outputFolder) {
		this.outputFolder = outputFolder;
	}

	public Path getOutputFolder() {
		return this.outputFolder;
	}

	public Path getOutputFolder(final PackOutput.Target target) {
		return this.getOutputFolder().resolve(target.directory);
	}

	public PackOutput.PathProvider createPathProvider(final PackOutput.Target target, final String kind) {
		return new PackOutput.PathProvider(this, target, kind);
	}

	public PackOutput.PathProvider createRegistryElementsPathProvider(final ResourceKey<? extends Registry<?>> registryKey) {
		return this.createPathProvider(PackOutput.Target.DATA_PACK, Registries.elementsDirPath(registryKey));
	}

	public PackOutput.PathProvider createRegistryTagsPathProvider(final ResourceKey<? extends Registry<?>> registryKey) {
		return this.createPathProvider(PackOutput.Target.DATA_PACK, Registries.tagsDirPath(registryKey));
	}

	public PackOutput.PathProvider createRegistryComponentPathProvider(final ResourceKey<? extends Registry<?>> registryKey) {
		return this.createPathProvider(PackOutput.Target.REPORTS, Registries.componentsDirPath(registryKey));
	}

	public static class PathProvider {
		private final Path root;
		private final String kind;

		private PathProvider(final PackOutput output, final PackOutput.Target target, final String kind) {
			this.root = output.getOutputFolder(target);
			this.kind = kind;
		}

		public Path file(final Identifier element, final String extension) {
			return element.withPath((UnaryOperator<String>)(path -> this.kind + "/" + path + "." + extension)).resolveAgainst(this.root);
		}

		public Path json(final Identifier element) {
			return element.withPath((UnaryOperator<String>)(path -> this.kind + "/" + path + ".json")).resolveAgainst(this.root);
		}

		public Path json(final ResourceKey<?> element) {
			return this.json(element.identifier());
		}
	}

	public static enum Target {
		DATA_PACK("data"),
		RESOURCE_PACK("assets"),
		REPORTS("reports");

		private final String directory;

		private Target(final String directory) {
			this.directory = directory;
		}
	}
}