package net.minecraft.commands;

import java.util.Optional;
import java.util.stream.Stream;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.Registry;
import net.minecraft.resources.ResourceKey;
import net.minecraft.world.flag.FeatureFlagSet;

public interface CommandBuildContext extends HolderLookup.Provider {
	static CommandBuildContext simple(final HolderLookup.Provider access, final FeatureFlagSet enabledFeatures) {
		return new CommandBuildContext() {
			@Override
			public Stream<ResourceKey<? extends Registry<?>>> listRegistryKeys() {
				return access.listRegistryKeys();
			}

			@Override
			public <T> Optional<HolderLookup.RegistryLookup<T>> lookup(final ResourceKey<? extends Registry<? extends T>> key) {
				return access.lookup(key).map(lookup -> lookup.filterFeatures(enabledFeatures));
			}

			@Override
			public FeatureFlagSet enabledFeatures() {
				return enabledFeatures;
			}
		};
	}

	FeatureFlagSet enabledFeatures();
}