package net.minecraft.world.item.crafting;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.Multimap;
import com.google.common.collect.ImmutableMultimap.Builder;
import java.util.Collection;
import java.util.Map;
import java.util.stream.Stream;
import net.minecraft.resources.ResourceKey;
import net.minecraft.world.level.Level;
import org.jspecify.annotations.Nullable;

public class RecipeMap {
	public static final RecipeMap EMPTY = new RecipeMap(ImmutableMultimap.of(), Map.of());
	private final Multimap<RecipeType<?>, RecipeHolder<?>> byType;
	private final Map<ResourceKey<Recipe<?>>, RecipeHolder<?>> byKey;

	private RecipeMap(final Multimap<RecipeType<?>, RecipeHolder<?>> byType, final Map<ResourceKey<Recipe<?>>, RecipeHolder<?>> byKey) {
		this.byType = byType;
		this.byKey = byKey;
	}

	public static RecipeMap create(final Iterable<RecipeHolder<?>> recipes) {
		Builder<RecipeType<?>, RecipeHolder<?>> byType = ImmutableMultimap.builder();
		com.google.common.collect.ImmutableMap.Builder<ResourceKey<Recipe<?>>, RecipeHolder<?>> byKey = ImmutableMap.builder();

		for (RecipeHolder<?> recipe : recipes) {
			byType.put(recipe.value().getType(), recipe);
			byKey.put(recipe.id(), recipe);
		}

		return new RecipeMap(byType.build(), byKey.build());
	}

	public <I extends RecipeInput, T extends Recipe<I>> Collection<RecipeHolder<T>> byType(final RecipeType<T> type) {
		return (Collection<RecipeHolder<T>>)this.byType.get(type);
	}

	public Collection<RecipeHolder<?>> values() {
		return this.byKey.values();
	}

	@Nullable
	public RecipeHolder<?> byKey(final ResourceKey<Recipe<?>> recipeId) {
		return (RecipeHolder<?>)this.byKey.get(recipeId);
	}

	public <I extends RecipeInput, T extends Recipe<I>> Stream<RecipeHolder<T>> getRecipesFor(final RecipeType<T> type, final I container, final Level level) {
		return container.isEmpty() ? Stream.empty() : this.byType(type).stream().filter(r -> r.value().matches(container, level));
	}
}