package net.minecraft.util.datafix.fixes;

import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
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.util.Pair;
import com.mojang.serialization.Dynamic;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;
import javax.annotation.Nullable;
import net.minecraft.util.datafix.schemas.NamespacedSchema;

public class ItemStackTheFlatteningFix extends DataFix {
    private static final Map<String, String> MAP = DataFixUtils.make(Maps.newHashMap(), map -> {
        map.put("minecraft:stone.0", "minecraft:stone");
        map.put("minecraft:stone.1", "minecraft:granite");
        map.put("minecraft:stone.2", "minecraft:polished_granite");
        map.put("minecraft:stone.3", "minecraft:diorite");
        map.put("minecraft:stone.4", "minecraft:polished_diorite");
        map.put("minecraft:stone.5", "minecraft:andesite");
        map.put("minecraft:stone.6", "minecraft:polished_andesite");
        map.put("minecraft:dirt.0", "minecraft:dirt");
        map.put("minecraft:dirt.1", "minecraft:coarse_dirt");
        map.put("minecraft:dirt.2", "minecraft:podzol");
        map.put("minecraft:leaves.0", "minecraft:oak_leaves");
        map.put("minecraft:leaves.1", "minecraft:spruce_leaves");
        map.put("minecraft:leaves.2", "minecraft:birch_leaves");
        map.put("minecraft:leaves.3", "minecraft:jungle_leaves");
        map.put("minecraft:leaves2.0", "minecraft:acacia_leaves");
        map.put("minecraft:leaves2.1", "minecraft:dark_oak_leaves");
        map.put("minecraft:log.0", "minecraft:oak_log");
        map.put("minecraft:log.1", "minecraft:spruce_log");
        map.put("minecraft:log.2", "minecraft:birch_log");
        map.put("minecraft:log.3", "minecraft:jungle_log");
        map.put("minecraft:log2.0", "minecraft:acacia_log");
        map.put("minecraft:log2.1", "minecraft:dark_oak_log");
        map.put("minecraft:sapling.0", "minecraft:oak_sapling");
        map.put("minecraft:sapling.1", "minecraft:spruce_sapling");
        map.put("minecraft:sapling.2", "minecraft:birch_sapling");
        map.put("minecraft:sapling.3", "minecraft:jungle_sapling");
        map.put("minecraft:sapling.4", "minecraft:acacia_sapling");
        map.put("minecraft:sapling.5", "minecraft:dark_oak_sapling");
        map.put("minecraft:planks.0", "minecraft:oak_planks");
        map.put("minecraft:planks.1", "minecraft:spruce_planks");
        map.put("minecraft:planks.2", "minecraft:birch_planks");
        map.put("minecraft:planks.3", "minecraft:jungle_planks");
        map.put("minecraft:planks.4", "minecraft:acacia_planks");
        map.put("minecraft:planks.5", "minecraft:dark_oak_planks");
        map.put("minecraft:sand.0", "minecraft:sand");
        map.put("minecraft:sand.1", "minecraft:red_sand");
        map.put("minecraft:quartz_block.0", "minecraft:quartz_block");
        map.put("minecraft:quartz_block.1", "minecraft:chiseled_quartz_block");
        map.put("minecraft:quartz_block.2", "minecraft:quartz_pillar");
        map.put("minecraft:anvil.0", "minecraft:anvil");
        map.put("minecraft:anvil.1", "minecraft:chipped_anvil");
        map.put("minecraft:anvil.2", "minecraft:damaged_anvil");
        map.put("minecraft:wool.0", "minecraft:white_wool");
        map.put("minecraft:wool.1", "minecraft:orange_wool");
        map.put("minecraft:wool.2", "minecraft:magenta_wool");
        map.put("minecraft:wool.3", "minecraft:light_blue_wool");
        map.put("minecraft:wool.4", "minecraft:yellow_wool");
        map.put("minecraft:wool.5", "minecraft:lime_wool");
        map.put("minecraft:wool.6", "minecraft:pink_wool");
        map.put("minecraft:wool.7", "minecraft:gray_wool");
        map.put("minecraft:wool.8", "minecraft:light_gray_wool");
        map.put("minecraft:wool.9", "minecraft:cyan_wool");
        map.put("minecraft:wool.10", "minecraft:purple_wool");
        map.put("minecraft:wool.11", "minecraft:blue_wool");
        map.put("minecraft:wool.12", "minecraft:brown_wool");
        map.put("minecraft:wool.13", "minecraft:green_wool");
        map.put("minecraft:wool.14", "minecraft:red_wool");
        map.put("minecraft:wool.15", "minecraft:black_wool");
        map.put("minecraft:carpet.0", "minecraft:white_carpet");
        map.put("minecraft:carpet.1", "minecraft:orange_carpet");
        map.put("minecraft:carpet.2", "minecraft:magenta_carpet");
        map.put("minecraft:carpet.3", "minecraft:light_blue_carpet");
        map.put("minecraft:carpet.4", "minecraft:yellow_carpet");
        map.put("minecraft:carpet.5", "minecraft:lime_carpet");
        map.put("minecraft:carpet.6", "minecraft:pink_carpet");
        map.put("minecraft:carpet.7", "minecraft:gray_carpet");
        map.put("minecraft:carpet.8", "minecraft:light_gray_carpet");
        map.put("minecraft:carpet.9", "minecraft:cyan_carpet");
        map.put("minecraft:carpet.10", "minecraft:purple_carpet");
        map.put("minecraft:carpet.11", "minecraft:blue_carpet");
        map.put("minecraft:carpet.12", "minecraft:brown_carpet");
        map.put("minecraft:carpet.13", "minecraft:green_carpet");
        map.put("minecraft:carpet.14", "minecraft:red_carpet");
        map.put("minecraft:carpet.15", "minecraft:black_carpet");
        map.put("minecraft:hardened_clay.0", "minecraft:terracotta");
        map.put("minecraft:stained_hardened_clay.0", "minecraft:white_terracotta");
        map.put("minecraft:stained_hardened_clay.1", "minecraft:orange_terracotta");
        map.put("minecraft:stained_hardened_clay.2", "minecraft:magenta_terracotta");
        map.put("minecraft:stained_hardened_clay.3", "minecraft:light_blue_terracotta");
        map.put("minecraft:stained_hardened_clay.4", "minecraft:yellow_terracotta");
        map.put("minecraft:stained_hardened_clay.5", "minecraft:lime_terracotta");
        map.put("minecraft:stained_hardened_clay.6", "minecraft:pink_terracotta");
        map.put("minecraft:stained_hardened_clay.7", "minecraft:gray_terracotta");
        map.put("minecraft:stained_hardened_clay.8", "minecraft:light_gray_terracotta");
        map.put("minecraft:stained_hardened_clay.9", "minecraft:cyan_terracotta");
        map.put("minecraft:stained_hardened_clay.10", "minecraft:purple_terracotta");
        map.put("minecraft:stained_hardened_clay.11", "minecraft:blue_terracotta");
        map.put("minecraft:stained_hardened_clay.12", "minecraft:brown_terracotta");
        map.put("minecraft:stained_hardened_clay.13", "minecraft:green_terracotta");
        map.put("minecraft:stained_hardened_clay.14", "minecraft:red_terracotta");
        map.put("minecraft:stained_hardened_clay.15", "minecraft:black_terracotta");
        map.put("minecraft:silver_glazed_terracotta.0", "minecraft:light_gray_glazed_terracotta");
        map.put("minecraft:stained_glass.0", "minecraft:white_stained_glass");
        map.put("minecraft:stained_glass.1", "minecraft:orange_stained_glass");
        map.put("minecraft:stained_glass.2", "minecraft:magenta_stained_glass");
        map.put("minecraft:stained_glass.3", "minecraft:light_blue_stained_glass");
        map.put("minecraft:stained_glass.4", "minecraft:yellow_stained_glass");
        map.put("minecraft:stained_glass.5", "minecraft:lime_stained_glass");
        map.put("minecraft:stained_glass.6", "minecraft:pink_stained_glass");
        map.put("minecraft:stained_glass.7", "minecraft:gray_stained_glass");
        map.put("minecraft:stained_glass.8", "minecraft:light_gray_stained_glass");
        map.put("minecraft:stained_glass.9", "minecraft:cyan_stained_glass");
        map.put("minecraft:stained_glass.10", "minecraft:purple_stained_glass");
        map.put("minecraft:stained_glass.11", "minecraft:blue_stained_glass");
        map.put("minecraft:stained_glass.12", "minecraft:brown_stained_glass");
        map.put("minecraft:stained_glass.13", "minecraft:green_stained_glass");
        map.put("minecraft:stained_glass.14", "minecraft:red_stained_glass");
        map.put("minecraft:stained_glass.15", "minecraft:black_stained_glass");
        map.put("minecraft:stained_glass_pane.0", "minecraft:white_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.1", "minecraft:orange_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.2", "minecraft:magenta_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.3", "minecraft:light_blue_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.4", "minecraft:yellow_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.5", "minecraft:lime_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.6", "minecraft:pink_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.7", "minecraft:gray_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.8", "minecraft:light_gray_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.9", "minecraft:cyan_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.10", "minecraft:purple_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.11", "minecraft:blue_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.12", "minecraft:brown_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.13", "minecraft:green_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.14", "minecraft:red_stained_glass_pane");
        map.put("minecraft:stained_glass_pane.15", "minecraft:black_stained_glass_pane");
        map.put("minecraft:prismarine.0", "minecraft:prismarine");
        map.put("minecraft:prismarine.1", "minecraft:prismarine_bricks");
        map.put("minecraft:prismarine.2", "minecraft:dark_prismarine");
        map.put("minecraft:concrete.0", "minecraft:white_concrete");
        map.put("minecraft:concrete.1", "minecraft:orange_concrete");
        map.put("minecraft:concrete.2", "minecraft:magenta_concrete");
        map.put("minecraft:concrete.3", "minecraft:light_blue_concrete");
        map.put("minecraft:concrete.4", "minecraft:yellow_concrete");
        map.put("minecraft:concrete.5", "minecraft:lime_concrete");
        map.put("minecraft:concrete.6", "minecraft:pink_concrete");
        map.put("minecraft:concrete.7", "minecraft:gray_concrete");
        map.put("minecraft:concrete.8", "minecraft:light_gray_concrete");
        map.put("minecraft:concrete.9", "minecraft:cyan_concrete");
        map.put("minecraft:concrete.10", "minecraft:purple_concrete");
        map.put("minecraft:concrete.11", "minecraft:blue_concrete");
        map.put("minecraft:concrete.12", "minecraft:brown_concrete");
        map.put("minecraft:concrete.13", "minecraft:green_concrete");
        map.put("minecraft:concrete.14", "minecraft:red_concrete");
        map.put("minecraft:concrete.15", "minecraft:black_concrete");
        map.put("minecraft:concrete_powder.0", "minecraft:white_concrete_powder");
        map.put("minecraft:concrete_powder.1", "minecraft:orange_concrete_powder");
        map.put("minecraft:concrete_powder.2", "minecraft:magenta_concrete_powder");
        map.put("minecraft:concrete_powder.3", "minecraft:light_blue_concrete_powder");
        map.put("minecraft:concrete_powder.4", "minecraft:yellow_concrete_powder");
        map.put("minecraft:concrete_powder.5", "minecraft:lime_concrete_powder");
        map.put("minecraft:concrete_powder.6", "minecraft:pink_concrete_powder");
        map.put("minecraft:concrete_powder.7", "minecraft:gray_concrete_powder");
        map.put("minecraft:concrete_powder.8", "minecraft:light_gray_concrete_powder");
        map.put("minecraft:concrete_powder.9", "minecraft:cyan_concrete_powder");
        map.put("minecraft:concrete_powder.10", "minecraft:purple_concrete_powder");
        map.put("minecraft:concrete_powder.11", "minecraft:blue_concrete_powder");
        map.put("minecraft:concrete_powder.12", "minecraft:brown_concrete_powder");
        map.put("minecraft:concrete_powder.13", "minecraft:green_concrete_powder");
        map.put("minecraft:concrete_powder.14", "minecraft:red_concrete_powder");
        map.put("minecraft:concrete_powder.15", "minecraft:black_concrete_powder");
        map.put("minecraft:cobblestone_wall.0", "minecraft:cobblestone_wall");
        map.put("minecraft:cobblestone_wall.1", "minecraft:mossy_cobblestone_wall");
        map.put("minecraft:sandstone.0", "minecraft:sandstone");
        map.put("minecraft:sandstone.1", "minecraft:chiseled_sandstone");
        map.put("minecraft:sandstone.2", "minecraft:cut_sandstone");
        map.put("minecraft:red_sandstone.0", "minecraft:red_sandstone");
        map.put("minecraft:red_sandstone.1", "minecraft:chiseled_red_sandstone");
        map.put("minecraft:red_sandstone.2", "minecraft:cut_red_sandstone");
        map.put("minecraft:stonebrick.0", "minecraft:stone_bricks");
        map.put("minecraft:stonebrick.1", "minecraft:mossy_stone_bricks");
        map.put("minecraft:stonebrick.2", "minecraft:cracked_stone_bricks");
        map.put("minecraft:stonebrick.3", "minecraft:chiseled_stone_bricks");
        map.put("minecraft:monster_egg.0", "minecraft:infested_stone");
        map.put("minecraft:monster_egg.1", "minecraft:infested_cobblestone");
        map.put("minecraft:monster_egg.2", "minecraft:infested_stone_bricks");
        map.put("minecraft:monster_egg.3", "minecraft:infested_mossy_stone_bricks");
        map.put("minecraft:monster_egg.4", "minecraft:infested_cracked_stone_bricks");
        map.put("minecraft:monster_egg.5", "minecraft:infested_chiseled_stone_bricks");
        map.put("minecraft:yellow_flower.0", "minecraft:dandelion");
        map.put("minecraft:red_flower.0", "minecraft:poppy");
        map.put("minecraft:red_flower.1", "minecraft:blue_orchid");
        map.put("minecraft:red_flower.2", "minecraft:allium");
        map.put("minecraft:red_flower.3", "minecraft:azure_bluet");
        map.put("minecraft:red_flower.4", "minecraft:red_tulip");
        map.put("minecraft:red_flower.5", "minecraft:orange_tulip");
        map.put("minecraft:red_flower.6", "minecraft:white_tulip");
        map.put("minecraft:red_flower.7", "minecraft:pink_tulip");
        map.put("minecraft:red_flower.8", "minecraft:oxeye_daisy");
        map.put("minecraft:double_plant.0", "minecraft:sunflower");
        map.put("minecraft:double_plant.1", "minecraft:lilac");
        map.put("minecraft:double_plant.2", "minecraft:tall_grass");
        map.put("minecraft:double_plant.3", "minecraft:large_fern");
        map.put("minecraft:double_plant.4", "minecraft:rose_bush");
        map.put("minecraft:double_plant.5", "minecraft:peony");
        map.put("minecraft:deadbush.0", "minecraft:dead_bush");
        map.put("minecraft:tallgrass.0", "minecraft:dead_bush");
        map.put("minecraft:tallgrass.1", "minecraft:grass");
        map.put("minecraft:tallgrass.2", "minecraft:fern");
        map.put("minecraft:sponge.0", "minecraft:sponge");
        map.put("minecraft:sponge.1", "minecraft:wet_sponge");
        map.put("minecraft:purpur_slab.0", "minecraft:purpur_slab");
        map.put("minecraft:stone_slab.0", "minecraft:stone_slab");
        map.put("minecraft:stone_slab.1", "minecraft:sandstone_slab");
        map.put("minecraft:stone_slab.2", "minecraft:petrified_oak_slab");
        map.put("minecraft:stone_slab.3", "minecraft:cobblestone_slab");
        map.put("minecraft:stone_slab.4", "minecraft:brick_slab");
        map.put("minecraft:stone_slab.5", "minecraft:stone_brick_slab");
        map.put("minecraft:stone_slab.6", "minecraft:nether_brick_slab");
        map.put("minecraft:stone_slab.7", "minecraft:quartz_slab");
        map.put("minecraft:stone_slab2.0", "minecraft:red_sandstone_slab");
        map.put("minecraft:wooden_slab.0", "minecraft:oak_slab");
        map.put("minecraft:wooden_slab.1", "minecraft:spruce_slab");
        map.put("minecraft:wooden_slab.2", "minecraft:birch_slab");
        map.put("minecraft:wooden_slab.3", "minecraft:jungle_slab");
        map.put("minecraft:wooden_slab.4", "minecraft:acacia_slab");
        map.put("minecraft:wooden_slab.5", "minecraft:dark_oak_slab");
        map.put("minecraft:coal.0", "minecraft:coal");
        map.put("minecraft:coal.1", "minecraft:charcoal");
        map.put("minecraft:fish.0", "minecraft:cod");
        map.put("minecraft:fish.1", "minecraft:salmon");
        map.put("minecraft:fish.2", "minecraft:clownfish");
        map.put("minecraft:fish.3", "minecraft:pufferfish");
        map.put("minecraft:cooked_fish.0", "minecraft:cooked_cod");
        map.put("minecraft:cooked_fish.1", "minecraft:cooked_salmon");
        map.put("minecraft:skull.0", "minecraft:skeleton_skull");
        map.put("minecraft:skull.1", "minecraft:wither_skeleton_skull");
        map.put("minecraft:skull.2", "minecraft:zombie_head");
        map.put("minecraft:skull.3", "minecraft:player_head");
        map.put("minecraft:skull.4", "minecraft:creeper_head");
        map.put("minecraft:skull.5", "minecraft:dragon_head");
        map.put("minecraft:golden_apple.0", "minecraft:golden_apple");
        map.put("minecraft:golden_apple.1", "minecraft:enchanted_golden_apple");
        map.put("minecraft:fireworks.0", "minecraft:firework_rocket");
        map.put("minecraft:firework_charge.0", "minecraft:firework_star");
        map.put("minecraft:dye.0", "minecraft:ink_sac");
        map.put("minecraft:dye.1", "minecraft:rose_red");
        map.put("minecraft:dye.2", "minecraft:cactus_green");
        map.put("minecraft:dye.3", "minecraft:cocoa_beans");
        map.put("minecraft:dye.4", "minecraft:lapis_lazuli");
        map.put("minecraft:dye.5", "minecraft:purple_dye");
        map.put("minecraft:dye.6", "minecraft:cyan_dye");
        map.put("minecraft:dye.7", "minecraft:light_gray_dye");
        map.put("minecraft:dye.8", "minecraft:gray_dye");
        map.put("minecraft:dye.9", "minecraft:pink_dye");
        map.put("minecraft:dye.10", "minecraft:lime_dye");
        map.put("minecraft:dye.11", "minecraft:dandelion_yellow");
        map.put("minecraft:dye.12", "minecraft:light_blue_dye");
        map.put("minecraft:dye.13", "minecraft:magenta_dye");
        map.put("minecraft:dye.14", "minecraft:orange_dye");
        map.put("minecraft:dye.15", "minecraft:bone_meal");
        map.put("minecraft:silver_shulker_box.0", "minecraft:light_gray_shulker_box");
        map.put("minecraft:fence.0", "minecraft:oak_fence");
        map.put("minecraft:fence_gate.0", "minecraft:oak_fence_gate");
        map.put("minecraft:wooden_door.0", "minecraft:oak_door");
        map.put("minecraft:boat.0", "minecraft:oak_boat");
        map.put("minecraft:lit_pumpkin.0", "minecraft:jack_o_lantern");
        map.put("minecraft:pumpkin.0", "minecraft:carved_pumpkin");
        map.put("minecraft:trapdoor.0", "minecraft:oak_trapdoor");
        map.put("minecraft:nether_brick.0", "minecraft:nether_bricks");
        map.put("minecraft:red_nether_brick.0", "minecraft:red_nether_bricks");
        map.put("minecraft:netherbrick.0", "minecraft:nether_brick");
        map.put("minecraft:wooden_button.0", "minecraft:oak_button");
        map.put("minecraft:wooden_pressure_plate.0", "minecraft:oak_pressure_plate");
        map.put("minecraft:noteblock.0", "minecraft:note_block");
        map.put("minecraft:bed.0", "minecraft:white_bed");
        map.put("minecraft:bed.1", "minecraft:orange_bed");
        map.put("minecraft:bed.2", "minecraft:magenta_bed");
        map.put("minecraft:bed.3", "minecraft:light_blue_bed");
        map.put("minecraft:bed.4", "minecraft:yellow_bed");
        map.put("minecraft:bed.5", "minecraft:lime_bed");
        map.put("minecraft:bed.6", "minecraft:pink_bed");
        map.put("minecraft:bed.7", "minecraft:gray_bed");
        map.put("minecraft:bed.8", "minecraft:light_gray_bed");
        map.put("minecraft:bed.9", "minecraft:cyan_bed");
        map.put("minecraft:bed.10", "minecraft:purple_bed");
        map.put("minecraft:bed.11", "minecraft:blue_bed");
        map.put("minecraft:bed.12", "minecraft:brown_bed");
        map.put("minecraft:bed.13", "minecraft:green_bed");
        map.put("minecraft:bed.14", "minecraft:red_bed");
        map.put("minecraft:bed.15", "minecraft:black_bed");
        map.put("minecraft:banner.15", "minecraft:white_banner");
        map.put("minecraft:banner.14", "minecraft:orange_banner");
        map.put("minecraft:banner.13", "minecraft:magenta_banner");
        map.put("minecraft:banner.12", "minecraft:light_blue_banner");
        map.put("minecraft:banner.11", "minecraft:yellow_banner");
        map.put("minecraft:banner.10", "minecraft:lime_banner");
        map.put("minecraft:banner.9", "minecraft:pink_banner");
        map.put("minecraft:banner.8", "minecraft:gray_banner");
        map.put("minecraft:banner.7", "minecraft:light_gray_banner");
        map.put("minecraft:banner.6", "minecraft:cyan_banner");
        map.put("minecraft:banner.5", "minecraft:purple_banner");
        map.put("minecraft:banner.4", "minecraft:blue_banner");
        map.put("minecraft:banner.3", "minecraft:brown_banner");
        map.put("minecraft:banner.2", "minecraft:green_banner");
        map.put("minecraft:banner.1", "minecraft:red_banner");
        map.put("minecraft:banner.0", "minecraft:black_banner");
        map.put("minecraft:grass.0", "minecraft:grass_block");
        map.put("minecraft:brick_block.0", "minecraft:bricks");
        map.put("minecraft:end_bricks.0", "minecraft:end_stone_bricks");
        map.put("minecraft:golden_rail.0", "minecraft:powered_rail");
        map.put("minecraft:magma.0", "minecraft:magma_block");
        map.put("minecraft:quartz_ore.0", "minecraft:nether_quartz_ore");
        map.put("minecraft:reeds.0", "minecraft:sugar_cane");
        map.put("minecraft:slime.0", "minecraft:slime_block");
        map.put("minecraft:stone_stairs.0", "minecraft:cobblestone_stairs");
        map.put("minecraft:waterlily.0", "minecraft:lily_pad");
        map.put("minecraft:web.0", "minecraft:cobweb");
        map.put("minecraft:snow.0", "minecraft:snow_block");
        map.put("minecraft:snow_layer.0", "minecraft:snow");
        map.put("minecraft:record_11.0", "minecraft:music_disc_11");
        map.put("minecraft:record_13.0", "minecraft:music_disc_13");
        map.put("minecraft:record_blocks.0", "minecraft:music_disc_blocks");
        map.put("minecraft:record_cat.0", "minecraft:music_disc_cat");
        map.put("minecraft:record_chirp.0", "minecraft:music_disc_chirp");
        map.put("minecraft:record_far.0", "minecraft:music_disc_far");
        map.put("minecraft:record_mall.0", "minecraft:music_disc_mall");
        map.put("minecraft:record_mellohi.0", "minecraft:music_disc_mellohi");
        map.put("minecraft:record_stal.0", "minecraft:music_disc_stal");
        map.put("minecraft:record_strad.0", "minecraft:music_disc_strad");
        map.put("minecraft:record_wait.0", "minecraft:music_disc_wait");
        map.put("minecraft:record_ward.0", "minecraft:music_disc_ward");
    });
    private static final Set<String> IDS = MAP.keySet().stream().map(string -> string.substring(0, string.indexOf(46))).collect(Collectors.toSet());
    private static final Set<String> DAMAGE_IDS = Sets.newHashSet(
        "minecraft:bow",
        "minecraft:carrot_on_a_stick",
        "minecraft:chainmail_boots",
        "minecraft:chainmail_chestplate",
        "minecraft:chainmail_helmet",
        "minecraft:chainmail_leggings",
        "minecraft:diamond_axe",
        "minecraft:diamond_boots",
        "minecraft:diamond_chestplate",
        "minecraft:diamond_helmet",
        "minecraft:diamond_hoe",
        "minecraft:diamond_leggings",
        "minecraft:diamond_pickaxe",
        "minecraft:diamond_shovel",
        "minecraft:diamond_sword",
        "minecraft:elytra",
        "minecraft:fishing_rod",
        "minecraft:flint_and_steel",
        "minecraft:golden_axe",
        "minecraft:golden_boots",
        "minecraft:golden_chestplate",
        "minecraft:golden_helmet",
        "minecraft:golden_hoe",
        "minecraft:golden_leggings",
        "minecraft:golden_pickaxe",
        "minecraft:golden_shovel",
        "minecraft:golden_sword",
        "minecraft:iron_axe",
        "minecraft:iron_boots",
        "minecraft:iron_chestplate",
        "minecraft:iron_helmet",
        "minecraft:iron_hoe",
        "minecraft:iron_leggings",
        "minecraft:iron_pickaxe",
        "minecraft:iron_shovel",
        "minecraft:iron_sword",
        "minecraft:leather_boots",
        "minecraft:leather_chestplate",
        "minecraft:leather_helmet",
        "minecraft:leather_leggings",
        "minecraft:shears",
        "minecraft:shield",
        "minecraft:stone_axe",
        "minecraft:stone_hoe",
        "minecraft:stone_pickaxe",
        "minecraft:stone_shovel",
        "minecraft:stone_sword",
        "minecraft:wooden_axe",
        "minecraft:wooden_hoe",
        "minecraft:wooden_pickaxe",
        "minecraft:wooden_shovel",
        "minecraft:wooden_sword"
    );

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

    @Override
    public TypeRewriteRule makeRule() {
        Type<?> type = this.getInputSchema().getType(References.ITEM_STACK);
        OpticFinder<Pair<String, String>> opticFinder = DSL.fieldFinder("id", DSL.named(References.ITEM_NAME.typeName(), NamespacedSchema.namespacedString()));
        OpticFinder<?> opticFinder1 = type.findField("tag");
        return this.fixTypeEverywhereTyped("ItemInstanceTheFlatteningFix", type, typed -> {
            Optional<Pair<String, String>> optional = typed.getOptional(opticFinder);
            if (optional.isEmpty()) {
                return typed;
            } else {
                Typed<?> typed1 = typed;
                Dynamic<?> dynamic = typed.get(DSL.remainderFinder());
                int _int = dynamic.get("Damage").asInt(0);
                String string = updateItem(optional.get().getSecond(), _int);
                if (string != null) {
                    typed1 = typed.set(opticFinder, Pair.of(References.ITEM_NAME.typeName(), string));
                }

                if (DAMAGE_IDS.contains(optional.get().getSecond())) {
                    Typed<?> typed2 = typed.getOrCreateTyped(opticFinder1);
                    Dynamic<?> dynamic1 = typed2.get(DSL.remainderFinder());
                    if (_int != 0) dynamic1 = dynamic1.set("Damage", dynamic1.createInt(_int)); // CraftBukkit
                    typed1 = typed1.set(opticFinder1, typed2.set(DSL.remainderFinder(), dynamic1));
                }

                return typed1.set(DSL.remainderFinder(), dynamic.remove("Damage"));
            }
        });
    }

    @Nullable
    public static String updateItem(@Nullable String item, int dataValue) {
        if (IDS.contains(item)) {
            String string = MAP.get(item + "." + dataValue);
            return string == null ? MAP.get(item + ".0") : string;
        } else {
            return null;
        }
    }
}
