/*
 * Decompiled with CFR 0.152.
 * 
 * Could not load the following classes:
 *  com.google.common.collect.Lists
 *  com.google.common.collect.Maps
 *  com.google.common.collect.Streams
 *  com.mojang.datafixers.DSL
 *  com.mojang.datafixers.DataFix
 *  com.mojang.datafixers.OpticFinder
 *  com.mojang.datafixers.TypeRewriteRule
 *  com.mojang.datafixers.Typed
 *  com.mojang.datafixers.schemas.Schema
 *  com.mojang.datafixers.types.Type
 *  com.mojang.datafixers.types.templates.List$ListType
 *  com.mojang.datafixers.util.Pair
 *  com.mojang.serialization.Dynamic
 */
package net.minecraft.util.datafix.fixes;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Streams;
import com.mojang.datafixers.DSL;
import com.mojang.datafixers.DataFix;
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.List;
import com.mojang.datafixers.util.Pair;
import com.mojang.serialization.Dynamic;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.function.Function;
import java.util.stream.IntStream;
import net.minecraft.util.datafix.fixes.References;

public class ChunkBedBlockEntityInjecterFix
extends DataFix {
    public ChunkBedBlockEntityInjecterFix(Schema outputSchema, boolean changesType) {
        super(outputSchema, changesType);
    }

    public TypeRewriteRule makeRule() {
        Type chunkType = this.getOutputSchema().getType(References.CHUNK);
        Type levelType = chunkType.findFieldType("Level");
        Type tileEntitiesType = levelType.findFieldType("TileEntities");
        if (!(tileEntitiesType instanceof List.ListType)) {
            throw new IllegalStateException("Tile entity type is not a list type.");
        }
        List.ListType tileEntityListType = (List.ListType)tileEntitiesType;
        return this.cap(levelType, tileEntityListType);
    }

    private <TE> TypeRewriteRule cap(Type<?> levelType, List.ListType<TE> tileEntityListType) {
        Type tileEntityType = tileEntityListType.getElement();
        OpticFinder levelF = DSL.fieldFinder((String)"Level", levelType);
        OpticFinder tileEntitiesF = DSL.fieldFinder((String)"TileEntities", tileEntityListType);
        int bedId = 416;
        return TypeRewriteRule.seq((TypeRewriteRule)this.fixTypeEverywhere("InjectBedBlockEntityType", (Type)this.getInputSchema().findChoiceType(References.BLOCK_ENTITY), (Type)this.getOutputSchema().findChoiceType(References.BLOCK_ENTITY), ops -> v -> v), (TypeRewriteRule)this.fixTypeEverywhereTyped("BedBlockEntityInjecter", this.getOutputSchema().getType(References.CHUNK), input -> {
            Typed level = input.getTyped(levelF);
            Dynamic levelTag = (Dynamic)level.get(DSL.remainderFinder());
            int chunkX = levelTag.get("xPos").asInt(0);
            int chunkZ = levelTag.get("zPos").asInt(0);
            ArrayList tileEntities = Lists.newArrayList((Iterable)((Iterable)level.getOrCreate(tileEntitiesF)));
            List sectionTags = levelTag.get("Sections").asList(Function.identity());
            for (Dynamic sectionTag : sectionTags) {
                int pos = sectionTag.get("Y").asInt(0);
                Streams.mapWithIndex((IntStream)sectionTag.get("Blocks").asIntStream(), (block, index) -> {
                    if (416 == (block & 0xFF) << 4) {
                        int p = (int)index;
                        int xx = p & 0xF;
                        int yy = p >> 8 & 0xF;
                        int zz = p >> 4 & 0xF;
                        HashMap bedTag = Maps.newHashMap();
                        bedTag.put(sectionTag.createString("id"), sectionTag.createString("minecraft:bed"));
                        bedTag.put(sectionTag.createString("x"), sectionTag.createInt(xx + (chunkX << 4)));
                        bedTag.put(sectionTag.createString("y"), sectionTag.createInt(yy + (pos << 4)));
                        bedTag.put(sectionTag.createString("z"), sectionTag.createInt(zz + (chunkZ << 4)));
                        bedTag.put(sectionTag.createString("color"), sectionTag.createShort((short)14));
                        return bedTag;
                    }
                    return null;
                }).forEachOrdered(bedTag -> {
                    if (bedTag != null) {
                        tileEntities.add(((Pair)tileEntityType.read(sectionTag.createMap(bedTag)).result().orElseThrow(() -> new IllegalStateException("Could not parse newly created bed block entity."))).getFirst());
                    }
                });
            }
            if (!tileEntities.isEmpty()) {
                return input.set(levelF, level.set(tileEntitiesF, (Object)tileEntities));
            }
            return input;
        }));
    }
}