package net.minecraft.util.datafix.fixes;

import com.google.common.collect.ImmutableMap;
import com.mojang.datafixers.schemas.Schema;
import java.util.Map;

public class EntityCodSalmonFix extends SimplestEntityRenameFix {
	public static final Map<String, String> RENAMED_IDS = ImmutableMap.<String, String>builder()
		.put("minecraft:salmon_mob", "minecraft:salmon")
		.put("minecraft:cod_mob", "minecraft:cod")
		.build();
	public static final Map<String, String> RENAMED_EGG_IDS = ImmutableMap.<String, String>builder()
		.put("minecraft:salmon_mob_spawn_egg", "minecraft:salmon_spawn_egg")
		.put("minecraft:cod_mob_spawn_egg", "minecraft:cod_spawn_egg")
		.build();

	public EntityCodSalmonFix(Schema outputSchema, boolean changesType) {
		super("EntityCodSalmonFix", outputSchema, changesType);
	}

	@Override
	protected String rename(String name) {
		return (String)RENAMED_IDS.getOrDefault(name, name);
	}
}