package net.minecraft.client.renderer.block.model.multipart;

import com.google.common.base.MoreObjects;
import com.google.common.base.Splitter;
import java.util.List;
import java.util.Locale;
import java.util.Optional;
import java.util.function.Predicate;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.Util;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.block.state.StateDefinition;
import net.minecraft.world.level.block.state.properties.Property;

@Environment(EnvType.CLIENT)
public class KeyValueCondition implements Condition {
	private static final Splitter PIPE_SPLITTER = Splitter.on('|').omitEmptyStrings();
	private final String key;
	private final String value;

	public KeyValueCondition(String key, String value) {
		this.key = key;
		this.value = value;
	}

	@Override
	public Predicate<BlockState> getPredicate(StateDefinition<Block, BlockState> stateDefinition) {
		Property<?> property = stateDefinition.getProperty(this.key);
		if (property == null) {
			throw new RuntimeException(String.format(Locale.ROOT, "Unknown property '%s' on '%s'", this.key, stateDefinition.getOwner()));
		} else {
			String string = this.value;
			boolean bl = !string.isEmpty() && string.charAt(0) == '!';
			if (bl) {
				string = string.substring(1);
			}

			List<String> list = PIPE_SPLITTER.splitToList(string);
			if (list.isEmpty()) {
				throw new RuntimeException(String.format(Locale.ROOT, "Empty value '%s' for property '%s' on '%s'", this.value, this.key, stateDefinition.getOwner()));
			} else {
				Predicate<BlockState> predicate;
				if (list.size() == 1) {
					predicate = this.getBlockStatePredicate(stateDefinition, property, string);
				} else {
					predicate = Util.anyOf(list.stream().map(stringx -> this.getBlockStatePredicate(stateDefinition, property, stringx)).toList());
				}

				return bl ? predicate.negate() : predicate;
			}
		}
	}

	private Predicate<BlockState> getBlockStatePredicate(StateDefinition<Block, BlockState> definition, Property<?> property, String value) {
		Optional<?> optional = property.getValue(value);
		if (optional.isEmpty()) {
			throw new RuntimeException(
				String.format(Locale.ROOT, "Unknown value '%s' for property '%s' on '%s' in '%s'", value, this.key, definition.getOwner(), this.value)
			);
		} else {
			return blockState -> blockState.getValue(property).equals(optional.get());
		}
	}

	public String toString() {
		return MoreObjects.toStringHelper(this).add("key", this.key).add("value", this.value).toString();
	}
}