// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.core.cosmetics;

import java.util.Iterator;
import java.util.Collections;
import java.io.IOException;
import org.bson.BsonValue;
import org.bson.BsonArray;
import java.nio.file.Files;
import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
import org.bson.BsonDocument;
import java.util.function.Function;
import java.nio.file.Path;
import com.hypixel.hytale.assetstore.AssetPack;
import javax.annotation.Nonnull;
import java.util.Map;

public class CosmeticRegistry
{
    public static final String MODEL = "Characters/Player.blockymodel";
    public static final String SKIN_GRADIENTSET_ID = "Skin";
    @Nonnull
    private final Map<String, Emote> emotes;
    @Nonnull
    private final Map<String, PlayerSkinTintColor> eyeColors;
    @Nonnull
    private final Map<String, PlayerSkinGradientSet> gradientSets;
    @Nonnull
    private final Map<String, PlayerSkinPart> bodyCharacteristics;
    @Nonnull
    private final Map<String, PlayerSkinPart> underwear;
    @Nonnull
    private final Map<String, PlayerSkinPart> eyebrows;
    @Nonnull
    private final Map<String, PlayerSkinPart> ears;
    @Nonnull
    private final Map<String, PlayerSkinPart> eyes;
    @Nonnull
    private final Map<String, PlayerSkinPart> faces;
    @Nonnull
    private final Map<String, PlayerSkinPart> mouths;
    @Nonnull
    private final Map<String, PlayerSkinPart> facialHair;
    @Nonnull
    private final Map<String, PlayerSkinPart> pants;
    @Nonnull
    private final Map<String, PlayerSkinPart> overpants;
    @Nonnull
    private final Map<String, PlayerSkinPart> undertops;
    @Nonnull
    private final Map<String, PlayerSkinPart> overtops;
    @Nonnull
    private final Map<String, PlayerSkinPart> haircuts;
    @Nonnull
    private final Map<String, PlayerSkinPart> shoes;
    @Nonnull
    private final Map<String, PlayerSkinPart> headAccessory;
    @Nonnull
    private final Map<String, PlayerSkinPart> faceAccessory;
    @Nonnull
    private final Map<String, PlayerSkinPart> earAccessory;
    @Nonnull
    private final Map<String, PlayerSkinPart> gloves;
    @Nonnull
    private final Map<String, PlayerSkinPart> capes;
    @Nonnull
    private final Map<String, PlayerSkinPart> skinFeatures;
    
    public CosmeticRegistry(@Nonnull final AssetPack pack) {
        final Path assetsDirectory = pack.getRoot();
        this.emotes = this.load(assetsDirectory, "Emotes.json", Emote::new);
        this.eyeColors = this.load(assetsDirectory, "EyeColors.json", PlayerSkinTintColor::new);
        this.gradientSets = this.load(assetsDirectory, "GradientSets.json", PlayerSkinGradientSet::new);
        this.bodyCharacteristics = this.load(assetsDirectory, "BodyCharacteristics.json", PlayerSkinPart::new);
        this.underwear = this.load(assetsDirectory, "Underwear.json", PlayerSkinPart::new);
        this.eyes = this.load(assetsDirectory, "Eyes.json", PlayerSkinPart::new);
        this.faces = this.load(assetsDirectory, "Faces.json", PlayerSkinPart::new);
        this.eyebrows = this.load(assetsDirectory, "Eyebrows.json", PlayerSkinPart::new);
        this.ears = this.load(assetsDirectory, "Ears.json", PlayerSkinPart::new);
        this.mouths = this.load(assetsDirectory, "Mouths.json", PlayerSkinPart::new);
        this.facialHair = this.load(assetsDirectory, "FacialHair.json", PlayerSkinPart::new);
        this.pants = this.load(assetsDirectory, "Pants.json", PlayerSkinPart::new);
        this.overpants = this.load(assetsDirectory, "Overpants.json", PlayerSkinPart::new);
        this.undertops = this.load(assetsDirectory, "Undertops.json", PlayerSkinPart::new);
        this.overtops = this.load(assetsDirectory, "Overtops.json", PlayerSkinPart::new);
        this.haircuts = this.load(assetsDirectory, "Haircuts.json", PlayerSkinPart::new);
        this.shoes = this.load(assetsDirectory, "Shoes.json", PlayerSkinPart::new);
        this.headAccessory = this.load(assetsDirectory, "HeadAccessory.json", PlayerSkinPart::new);
        this.faceAccessory = this.load(assetsDirectory, "FaceAccessory.json", PlayerSkinPart::new);
        this.earAccessory = this.load(assetsDirectory, "EarAccessory.json", PlayerSkinPart::new);
        this.gloves = this.load(assetsDirectory, "Gloves.json", PlayerSkinPart::new);
        this.capes = this.load(assetsDirectory, "Capes.json", PlayerSkinPart::new);
        this.skinFeatures = this.load(assetsDirectory, "SkinFeatures.json", PlayerSkinPart::new);
    }
    
    @Nonnull
    private <T> Map<String, T> load(@Nonnull final Path assetsDirectory, @Nonnull final String file, @Nonnull final Function<BsonDocument, T> func) {
        final Map<String, T> map = new Object2ObjectOpenHashMap<String, T>();
        final Path path = assetsDirectory.resolve("Cosmetics").resolve("CharacterCreator").resolve(file);
        try {
            final BsonArray bsonArray = BsonArray.parse(Files.readString(path));
            for (final BsonValue bsonValue : bsonArray) {
                final BsonDocument doc = bsonValue.asDocument();
                map.put(doc.getString("Id").getValue(), func.apply(doc));
            }
        }
        catch (final IOException e) {
            e.printStackTrace();
        }
        return Collections.unmodifiableMap((Map<? extends String, ? extends T>)map);
    }
    
    @Nonnull
    public Map<String, Emote> getEmotes() {
        return this.emotes;
    }
    
    @Nonnull
    public Map<String, PlayerSkinTintColor> getEyeColors() {
        return this.eyeColors;
    }
    
    @Nonnull
    public Map<String, PlayerSkinGradientSet> getGradientSets() {
        return this.gradientSets;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getBodyCharacteristics() {
        return this.bodyCharacteristics;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getUnderwear() {
        return this.underwear;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getEyebrows() {
        return this.eyebrows;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getEars() {
        return this.ears;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getEyes() {
        return this.eyes;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getFaces() {
        return this.faces;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getMouths() {
        return this.mouths;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getFacialHairs() {
        return this.facialHair;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getPants() {
        return this.pants;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getOverpants() {
        return this.overpants;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getUndertops() {
        return this.undertops;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getOvertops() {
        return this.overtops;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getHaircuts() {
        return this.haircuts;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getShoes() {
        return this.shoes;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getHeadAccessories() {
        return this.headAccessory;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getFaceAccessories() {
        return this.faceAccessory;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getEarAccessories() {
        return this.earAccessory;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getGloves() {
        return this.gloves;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getSkinFeatures() {
        return this.skinFeatures;
    }
    
    @Nonnull
    public Map<String, PlayerSkinPart> getCapes() {
        return this.capes;
    }
    
    public Map<String, ?> getByType(@Nonnull final CosmeticType type) {
        return switch (type) {
            default -> throw new MatchException(null, null);
            case EMOTES -> this.getEmotes();
            case SKIN_TONES -> this.getGradientSets().get("Skin").getGradients();
            case EYE_COLORS -> this.getEyeColors();
            case GRADIENT_SETS -> this.getGradientSets();
            case BODY_CHARACTERISTICS -> this.getBodyCharacteristics();
            case UNDERWEAR -> this.getUnderwear();
            case EYEBROWS -> this.getEyebrows();
            case EARS -> this.getEars();
            case EYES -> this.getEyes();
            case FACE -> this.getFaces();
            case MOUTHS -> this.getMouths();
            case FACIAL_HAIR -> this.getFacialHairs();
            case PANTS -> this.getPants();
            case OVERPANTS -> this.getOverpants();
            case UNDERTOPS -> this.getUndertops();
            case OVERTOPS -> this.getOvertops();
            case HAIRCUTS -> this.getHaircuts();
            case SHOES -> this.getShoes();
            case HEAD_ACCESSORY -> this.getHeadAccessories();
            case FACE_ACCESSORY -> this.getFaceAccessories();
            case EAR_ACCESSORY -> this.getEarAccessories();
            case GLOVES -> this.getGloves();
            case CAPES -> this.getCapes();
            case SKIN_FEATURES -> this.getSkinFeatures();
        };
    }
}