package net.minecraft.util.datafix.fixes;

import com.mojang.datafixers.DSL;
import com.mojang.datafixers.Typed;
import com.mojang.datafixers.schemas.Schema;
import com.mojang.serialization.Dynamic;

public class WeaponSmithChestLootTableFix extends NamedEntityFix {
    public WeaponSmithChestLootTableFix(Schema p_203111_, boolean p_203112_) {
        super(p_203111_, p_203112_, "WeaponSmithChestLootTableFix", References.BLOCK_ENTITY, "minecraft:chest");
    }

    @Override
    protected Typed<?> fix(Typed<?> p_203114_) {
        return p_203114_.update(
            DSL.remainderFinder(),
            p_203116_ -> {
                String s = p_203116_.get("LootTable").asString("");
                return s.equals("minecraft:chests/village_blacksmith")
                    ? p_203116_.set("LootTable", p_203116_.createString("minecraft:chests/village/village_weaponsmith"))
                    : p_203116_;
            }
        );
    }
}