package net.minecraft.util.datafix.schemas;

import com.mojang.datafixers.DSL.TypeReference;
import com.mojang.datafixers.schemas.Schema;
import com.mojang.datafixers.types.Type;
import com.mojang.datafixers.types.templates.Const.PrimitiveType;
import com.mojang.serialization.DataResult;
import com.mojang.serialization.DynamicOps;
import com.mojang.serialization.codecs.PrimitiveCodec;
import net.minecraft.resources.Identifier;

public class NamespacedSchema extends Schema {
	public static final PrimitiveCodec<String> NAMESPACED_STRING_CODEC = new PrimitiveCodec<String>() {
		@Override
		public <T> DataResult<String> read(final DynamicOps<T> ops, final T input) {
			return ops.getStringValue(input).map(NamespacedSchema::ensureNamespaced);
		}

		public <T> T write(final DynamicOps<T> ops, final String value) {
			return ops.createString(value);
		}

		public String toString() {
			return "NamespacedString";
		}
	};
	private static final Type<String> NAMESPACED_STRING = new PrimitiveType<>(NAMESPACED_STRING_CODEC);

	public NamespacedSchema(final int versionKey, final Schema parent) {
		super(versionKey, parent);
	}

	public static String ensureNamespaced(final String input) {
		Identifier identifier = Identifier.tryParse(input);
		return identifier != null ? identifier.toString() : input;
	}

	public static Type<String> namespacedString() {
		return NAMESPACED_STRING;
	}

	@Override
	public Type<?> getChoiceType(final TypeReference type, final String choiceName) {
		return super.getChoiceType(type, ensureNamespaced(choiceName));
	}
}