package net.minecraft.util.datafix.fixes;

import com.mojang.datafixers.DSL;
import com.mojang.datafixers.DataFix;
import com.mojang.datafixers.DataFixUtils;
import com.mojang.datafixers.OpticFinder;
import com.mojang.datafixers.TypeRewriteRule;
import com.mojang.datafixers.Typed;
import com.mojang.datafixers.schemas.Schema;
import com.mojang.datafixers.types.Type;
import com.mojang.datafixers.types.templates.CompoundList.CompoundListType;
import com.mojang.datafixers.util.Pair;
import com.mojang.serialization.Dynamic;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;
import net.minecraft.util.datafix.schemas.NamespacedSchema;

public class NewVillageFix extends DataFix {
    public NewVillageFix(Schema p_16476_, boolean p_16477_) {
        super(p_16476_, p_16477_);
    }

    @Override
    protected TypeRewriteRule makeRule() {
        CompoundListType<String, ?> compoundlisttype = DSL.compoundList(DSL.string(), this.getInputSchema().getType(References.STRUCTURE_FEATURE));
        OpticFinder<? extends List<? extends Pair<String, ?>>> opticfinder = compoundlisttype.finder();
        return this.cap(compoundlisttype);
    }

    private <SF> TypeRewriteRule cap(CompoundListType<String, SF> p_16499_) {
        Type<?> type = this.getInputSchema().getType(References.CHUNK);
        Type<?> type1 = this.getInputSchema().getType(References.STRUCTURE_FEATURE);
        OpticFinder<?> opticfinder = type.findField("Level");
        OpticFinder<?> opticfinder1 = opticfinder.type().findField("Structures");
        OpticFinder<?> opticfinder2 = opticfinder1.type().findField("Starts");
        OpticFinder<List<Pair<String, SF>>> opticfinder3 = p_16499_.finder();
        return TypeRewriteRule.seq(
            this.fixTypeEverywhereTyped(
                "NewVillageFix",
                type,
                p_16483_ -> p_16483_.updateTyped(
                    opticfinder,
                    p_145526_ -> p_145526_.updateTyped(
                        opticfinder1,
                        p_145530_ -> p_145530_.updateTyped(
                                opticfinder2,
                                p_145533_ -> p_145533_.update(
                                    opticfinder3,
                                    p_145544_ -> p_145544_.stream()
                                        .filter(p_145546_ -> !Objects.equals(p_145546_.getFirst(), "Village"))
                                        .map(p_145535_ -> p_145535_.mapFirst(p_145542_ -> p_145542_.equals("New_Village") ? "Village" : p_145542_))
                                        .collect(Collectors.toList())
                                )
                            )
                            .update(
                                DSL.remainderFinder(),
                                p_145550_ -> p_145550_.update(
                                    "References",
                                    p_145552_ -> {
                                        Optional<? extends Dynamic<?>> optional = p_145552_.get("New_Village").result();
                                        return DataFixUtils.orElse(
                                                optional.map(p_145540_ -> p_145552_.remove("New_Village").set("Village", (Dynamic<?>)p_145540_)), p_145552_
                                            )
                                            .remove("Village");
                                    }
                                )
                            )
                    )
                )
            ),
            this.fixTypeEverywhereTyped(
                "NewVillageStartFix",
                type1,
                p_16497_ -> p_16497_.update(
                    DSL.remainderFinder(),
                    p_145537_ -> p_145537_.update(
                        "id",
                        p_145548_ -> Objects.equals(NamespacedSchema.ensureNamespaced(p_145548_.asString("")), "minecraft:new_village")
                            ? p_145548_.createString("minecraft:village")
                            : p_145548_
                    )
                )
            )
        );
    }
}